1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * (C) Copyright 2005
4 * 2N Telekomunikace, a.s. <www.2n.cz>
5 * Ladislav Michl <michl@2n.cz>
6 */
7
8 #include <common.h>
9 #include <nand.h>
10 #include <errno.h>
11 #include <linux/mtd/concat.h>
12
13 #ifndef CONFIG_SYS_NAND_BASE_LIST
14 #define CONFIG_SYS_NAND_BASE_LIST { CONFIG_SYS_NAND_BASE }
15 #endif
16
17 int nand_curr_device = -1;
18
19 static struct mtd_info *nand_info[CONFIG_SYS_MAX_NAND_DEVICE];
20
21 #ifndef CONFIG_SYS_NAND_SELF_INIT
22 static struct nand_chip nand_chip[CONFIG_SYS_MAX_NAND_DEVICE];
23 static ulong base_address[CONFIG_SYS_MAX_NAND_DEVICE] = CONFIG_SYS_NAND_BASE_LIST;
24 #endif
25
26 static char dev_name[CONFIG_SYS_MAX_NAND_DEVICE][8];
27
28 static unsigned long total_nand_size; /* in kiB */
29
get_nand_dev_by_index(int dev)30 struct mtd_info *get_nand_dev_by_index(int dev)
31 {
32 if (dev < 0 || dev >= CONFIG_SYS_MAX_NAND_DEVICE || !nand_info[dev] ||
33 !nand_info[dev]->name)
34 return NULL;
35
36 return nand_info[dev];
37 }
38
nand_mtd_to_devnum(struct mtd_info * mtd)39 int nand_mtd_to_devnum(struct mtd_info *mtd)
40 {
41 int i;
42
43 for (i = 0; i < CONFIG_SYS_MAX_NAND_DEVICE; i++) {
44 if (mtd && get_nand_dev_by_index(i) == mtd)
45 return i;
46 }
47
48 return -ENODEV;
49 }
50
51 /* Register an initialized NAND mtd device with the U-Boot NAND command. */
nand_register(int devnum,struct mtd_info * mtd)52 int nand_register(int devnum, struct mtd_info *mtd)
53 {
54 if (devnum >= CONFIG_SYS_MAX_NAND_DEVICE)
55 return -EINVAL;
56
57 nand_info[devnum] = mtd;
58
59 sprintf(dev_name[devnum], "nand%d", devnum);
60 mtd->name = dev_name[devnum];
61
62 #ifdef CONFIG_MTD
63 /*
64 * Add MTD device so that we can reference it later
65 * via the mtdcore infrastructure (e.g. ubi).
66 */
67 add_mtd_device(mtd);
68 #endif
69
70 total_nand_size += mtd->size / 1024;
71
72 if (nand_curr_device == -1)
73 nand_curr_device = devnum;
74
75 return 0;
76 }
77
78 #ifndef CONFIG_SYS_NAND_SELF_INIT
nand_init_chip(int i)79 static void nand_init_chip(int i)
80 {
81 struct nand_chip *nand = &nand_chip[i];
82 struct mtd_info *mtd = nand_to_mtd(nand);
83 ulong base_addr = base_address[i];
84 int maxchips = CONFIG_SYS_NAND_MAX_CHIPS;
85
86 if (maxchips < 1)
87 maxchips = 1;
88
89 nand->IO_ADDR_R = nand->IO_ADDR_W = (void __iomem *)base_addr;
90
91 if (board_nand_init(nand))
92 return;
93
94 if (nand_scan(mtd, maxchips))
95 return;
96
97 nand_register(i, mtd);
98 }
99 #endif
100
101 #ifdef CONFIG_MTD_CONCAT
create_mtd_concat(void)102 static void create_mtd_concat(void)
103 {
104 struct mtd_info *nand_info_list[CONFIG_SYS_MAX_NAND_DEVICE];
105 int nand_devices_found = 0;
106 int i;
107
108 for (i = 0; i < CONFIG_SYS_MAX_NAND_DEVICE; i++) {
109 struct mtd_info *mtd = get_nand_dev_by_index(i);
110 if (mtd != NULL) {
111 nand_info_list[nand_devices_found] = mtd;
112 nand_devices_found++;
113 }
114 }
115 if (nand_devices_found > 1) {
116 struct mtd_info *mtd;
117 char c_mtd_name[16];
118
119 /*
120 * We detected multiple devices. Concatenate them together.
121 */
122 sprintf(c_mtd_name, "nand%d", nand_devices_found);
123 mtd = mtd_concat_create(nand_info_list, nand_devices_found,
124 c_mtd_name);
125
126 if (mtd == NULL)
127 return;
128
129 nand_register(nand_devices_found, mtd);
130 }
131
132 return;
133 }
134 #else
create_mtd_concat(void)135 static void create_mtd_concat(void)
136 {
137 }
138 #endif
139
nand_size(void)140 unsigned long nand_size(void)
141 {
142 return total_nand_size;
143 }
144
nand_init(void)145 void nand_init(void)
146 {
147 static int initialized;
148
149 /*
150 * Avoid initializing NAND Flash multiple times,
151 * otherwise it will calculate a wrong total size.
152 */
153 if (initialized)
154 return;
155 initialized = 1;
156
157 #ifdef CONFIG_SYS_NAND_SELF_INIT
158 board_nand_init();
159 #else
160 int i;
161
162 for (i = 0; i < CONFIG_SYS_MAX_NAND_DEVICE; i++)
163 nand_init_chip(i);
164 #endif
165
166 #ifdef CONFIG_SYS_NAND_SELECT_DEVICE
167 /*
168 * Select the chip in the board/cpu specific driver
169 */
170 board_nand_select_device(mtd_to_nand(get_nand_dev_by_index(nand_curr_device)),
171 nand_curr_device);
172 #endif
173
174 create_mtd_concat();
175 }
176