1 /* SPDX-License-Identifier: GPL-2.0+ */
2 /*
3 * (C) Copyright 2012
4 * Stefan Roese, DENX Software Engineering, sr@denx.de.
5 */
6 #ifndef _BOOTCOUNT_H__
7 #define _BOOTCOUNT_H__
8
9 #include <common.h>
10 #include <asm/io.h>
11 #include <asm/byteorder.h>
12 #include <env.h>
13
14 #ifdef CONFIG_DM_BOOTCOUNT
15
16 struct bootcount_ops {
17 /**
18 * get() - get the current bootcount value
19 *
20 * Returns the current counter value of the bootcount backing
21 * store.
22 *
23 * @dev: Device to read from
24 * @bootcount: Address to put the current bootcount value
25 */
26 int (*get)(struct udevice *dev, u32 *bootcount);
27
28 /**
29 * set() - set a bootcount value (e.g. to reset or increment)
30 *
31 * Sets the value in the bootcount backing store.
32 *
33 * @dev: Device to read from
34 * @bootcount: New bootcount value to store
35 */
36 int (*set)(struct udevice *dev, const u32 bootcount);
37 };
38
39 /* Access the operations for a bootcount device */
40 #define bootcount_get_ops(dev) ((struct bootcount_ops *)(dev)->driver->ops)
41
42 /**
43 * dm_bootcount_get() - Read the current value from a bootcount storage
44 *
45 * @dev: Device to read from
46 * @bootcount: Place to put the current bootcount
47 * @return 0 if OK, -ve on error
48 */
49 int dm_bootcount_get(struct udevice *dev, u32 *bootcount);
50
51 /**
52 * dm_bootcount_set() - Write a value to a bootcount storage
53 *
54 * @dev: Device to read from
55 * @bootcount: Value to be written to the backing storage
56 * @return 0 if OK, -ve on error
57 */
58 int dm_bootcount_set(struct udevice *dev, u32 bootcount);
59
60 #endif
61
62 /** bootcount_store() - store the current bootcount */
63 void bootcount_store(ulong);
64
65 /**
66 * bootcount_load() - load the current bootcount
67 *
68 * @return bootcount, read from the appropriate location
69 */
70 ulong bootcount_load(void);
71
72 #if defined(CONFIG_SPL_BOOTCOUNT_LIMIT) || defined(CONFIG_BOOTCOUNT_LIMIT)
73
74 #if !defined(CONFIG_SYS_BOOTCOUNT_LE) && !defined(CONFIG_SYS_BOOTCOUNT_BE)
75 # if __BYTE_ORDER == __LITTLE_ENDIAN
76 # define CONFIG_SYS_BOOTCOUNT_LE
77 # else
78 # define CONFIG_SYS_BOOTCOUNT_BE
79 # endif
80 #endif
81
82 #ifdef CONFIG_SYS_BOOTCOUNT_LE
raw_bootcount_store(volatile u32 * addr,u32 data)83 static inline void raw_bootcount_store(volatile u32 *addr, u32 data)
84 {
85 out_le32(addr, data);
86 }
87
raw_bootcount_load(volatile u32 * addr)88 static inline u32 raw_bootcount_load(volatile u32 *addr)
89 {
90 return in_le32(addr);
91 }
92 #else
raw_bootcount_store(volatile u32 * addr,u32 data)93 static inline void raw_bootcount_store(volatile u32 *addr, u32 data)
94 {
95 out_be32(addr, data);
96 }
97
raw_bootcount_load(volatile u32 * addr)98 static inline u32 raw_bootcount_load(volatile u32 *addr)
99 {
100 return in_be32(addr);
101 }
102 #endif
103
104 DECLARE_GLOBAL_DATA_PTR;
bootcount_error(void)105 static inline int bootcount_error(void)
106 {
107 unsigned long bootcount = bootcount_load();
108 unsigned long bootlimit = env_get_ulong("bootlimit", 10, 0);
109
110 if (bootlimit && bootcount > bootlimit) {
111 printf("Warning: Bootlimit (%lu) exceeded.", bootlimit);
112 if (!(gd->flags & GD_FLG_SPL_INIT))
113 printf(" Using altbootcmd.");
114 printf("\n");
115
116 return 1;
117 }
118
119 return 0;
120 }
121
bootcount_inc(void)122 static inline void bootcount_inc(void)
123 {
124 unsigned long bootcount = bootcount_load();
125
126 if (gd->flags & GD_FLG_SPL_INIT) {
127 bootcount_store(++bootcount);
128 return;
129 }
130
131 #ifndef CONFIG_SPL_BUILD
132 /* Only increment bootcount when no bootcount support in SPL */
133 #ifndef CONFIG_SPL_BOOTCOUNT_LIMIT
134 bootcount_store(++bootcount);
135 #endif
136 env_set_ulong("bootcount", bootcount);
137 #endif /* !CONFIG_SPL_BUILD */
138 }
139
140 #else
bootcount_error(void)141 static inline int bootcount_error(void) { return 0; }
bootcount_inc(void)142 static inline void bootcount_inc(void) {}
143 #endif /* CONFIG_SPL_BOOTCOUNT_LIMIT || CONFIG_BOOTCOUNT_LIMIT */
144 #endif /* _BOOTCOUNT_H__ */
145