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