huangcm
2024-11-25 22631b47c9e9052ebc7bdc07fa77b474bccc1312
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/* SPDX-License-Identifier: GPL-2.0+ */
/*
 * (C) Copyright 2012
 * Stefan Roese, DENX Software Engineering, sr@denx.de.
 */
#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
# else
#  define CONFIG_SYS_BOOTCOUNT_BE
# endif
#endif
 
#ifdef CONFIG_SYS_BOOTCOUNT_LE
static inline void raw_bootcount_store(volatile u32 *addr, u32 data)
{
   out_le32(addr, data);
}
 
static inline u32 raw_bootcount_load(volatile u32 *addr)
{
   return in_le32(addr);
}
#else
static inline void raw_bootcount_store(volatile u32 *addr, u32 data)
{
   out_be32(addr, data);
}
 
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__ */