Merge branch 'master' of git://git.denx.de/u-boot-sunxi

[u-boot] / include / bootcount.h

diff --git a/include/bootcount.h b/include/bootcount.h
index 06fb4d3578b5079bd3f8bb4b41a23fa8af03c31d..671adcc4101adb7c192fb0cb2ab7898f60e25b31 100644 (file)
--- a/include/bootcount.h
+++ b/include/bootcount.h
@@ -1,14 +1,17 @@
+/* SPDX-License-Identifier: GPL-2.0+ */
 /*
  * (C) Copyright 2012
  * Stefan Roese, DENX Software Engineering, sr@denx.de.
- *
- * SPDX-License-Identifier:    GPL-2.0+
  */
+#ifndef _BOOTCOUNT_H__
+#define _BOOTCOUNT_H__
 
 #include <common.h>
 #include <asm/io.h>
 #include <asm/byteorder.h>
 
+#if defined(CONFIG_SPL_BOOTCOUNT_LIMIT) || defined(CONFIG_BOOTCOUNT_LIMIT)
+
 #if !defined(CONFIG_SYS_BOOTCOUNT_LE) && !defined(CONFIG_SYS_BOOTCOUNT_BE)
 # if __BYTE_ORDER == __LITTLE_ENDIAN
 #  define CONFIG_SYS_BOOTCOUNT_LE
@@ -38,3 +41,49 @@ static inline u32 raw_bootcount_load(volatile u32 *addr)
        return in_be32(addr);
 }
 #endif
+
+DECLARE_GLOBAL_DATA_PTR;
+static inline int bootcount_error(void)
+{
+       unsigned long bootcount = bootcount_load();
+       unsigned long bootlimit = env_get_ulong("bootlimit", 10, 0);
+
+       if (bootlimit && bootcount > bootlimit) {
+               printf("Warning: Bootlimit (%lu) exceeded.", bootlimit);
+               if (!(gd->flags & GD_FLG_SPL_INIT))
+                       printf(" Using altbootcmd.");
+               printf("\n");
+
+               return 1;
+       }
+
+       return 0;
+}
+
+static inline void bootcount_inc(void)
+{
+       unsigned long bootcount = bootcount_load();
+
+       if (gd->flags & GD_FLG_SPL_INIT) {
+               bootcount_store(++bootcount);
+               return;
+       }
+
+#ifndef CONFIG_SPL_BUILD
+       /* Only increment bootcount when no bootcount support in SPL */
+#ifndef CONFIG_SPL_BOOTCOUNT_LIMIT
+       bootcount_store(++bootcount);
+#endif
+       env_set_ulong("bootcount", bootcount);
+#endif /* !CONFIG_SPL_BUILD */
+}
+
+#if defined(CONFIG_SPL_BUILD) && !defined(CONFIG_SPL_BOOTCOUNT_LIMIT)
+void bootcount_store(ulong a) {};
+ulong bootcount_load(void) { return 0; }
+#endif /* CONFIG_SPL_BUILD && !CONFIG_SPL_BOOTCOUNT_LIMIT */
+#else
+static inline int bootcount_error(void) { return 0; }
+static inline void bootcount_inc(void) {}
+#endif /* CONFIG_SPL_BOOTCOUNT_LIMIT || CONFIG_BOOTCOUNT_LIMIT */
+#endif /* _BOOTCOUNT_H__ */