nand: Embed mtd_info in struct nand_chip
[pandora-u-boot.git] / common / fb_nand.c
1 /*
2  * Copyright 2014 Broadcom Corporation.
3  * Copyright 2015 Free Electrons.
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
7
8 #include <config.h>
9 #include <common.h>
10
11 #include <fastboot.h>
12 #include <image-sparse.h>
13 #include <sparse_format.h>
14
15 #include <linux/mtd/mtd.h>
16 #include <jffs2/jffs2.h>
17 #include <nand.h>
18
19 static char *response_str;
20
21 struct fb_nand_sparse {
22         struct mtd_info         *nand;
23         struct part_info        *part;
24 };
25
26 __weak int board_fastboot_erase_partition_setup(char *name)
27 {
28         return 0;
29 }
30
31 __weak int board_fastboot_write_partition_setup(char *name)
32 {
33         return 0;
34 }
35
36 static int fb_nand_lookup(const char *partname, char *response,
37                           struct mtd_info **nand,
38                           struct part_info **part)
39 {
40         struct mtd_device *dev;
41         int ret;
42         u8 pnum;
43
44         ret = mtdparts_init();
45         if (ret) {
46                 error("Cannot initialize MTD partitions\n");
47                 fastboot_fail(response_str, "cannot init mtdparts");
48                 return ret;
49         }
50
51         ret = find_dev_and_part(partname, &dev, &pnum, part);
52         if (ret) {
53                 error("cannot find partition: '%s'", partname);
54                 fastboot_fail(response_str, "cannot find partition");
55                 return ret;
56         }
57
58         if (dev->id->type != MTD_DEV_TYPE_NAND) {
59                 error("partition '%s' is not stored on a NAND device",
60                       partname);
61                 fastboot_fail(response_str, "not a NAND device");
62                 return -EINVAL;
63         }
64
65         *mtd = nand_info[dev->id->num];
66
67         return 0;
68 }
69
70 static int _fb_nand_erase(struct mtd_info *mtd, struct part_info *part)
71 {
72         nand_erase_options_t opts;
73         int ret;
74
75         memset(&opts, 0, sizeof(opts));
76         opts.offset = part->offset;
77         opts.length = part->size;
78         opts.quiet = 1;
79
80         printf("Erasing blocks 0x%llx to 0x%llx\n",
81                part->offset, part->offset + part->size);
82
83         ret = nand_erase_opts(mtd, &opts);
84         if (ret)
85                 return ret;
86
87         printf("........ erased 0x%llx bytes from '%s'\n",
88                part->size, part->name);
89
90         return 0;
91 }
92
93 static int _fb_nand_write(struct mtd_info *mtd, struct part_info *part,
94                           void *buffer, unsigned int offset,
95                           unsigned int length, size_t *written)
96 {
97         int flags = WITH_WR_VERIFY;
98
99 #ifdef CONFIG_FASTBOOT_FLASH_NAND_TRIMFFS
100         flags |= WITH_DROP_FFS;
101 #endif
102
103         return nand_write_skip_bad(mtd, offset, &length, written,
104                                    part->size - (offset - part->offset),
105                                    buffer, flags);
106 }
107
108 static int fb_nand_sparse_write(struct sparse_storage *storage,
109                                 void *priv,
110                                 unsigned int offset,
111                                 unsigned int size,
112                                 char *data)
113 {
114         struct fb_nand_sparse *sparse = priv;
115         size_t written;
116         int ret;
117
118         ret = _fb_nand_write(sparse->nand, sparse->part, data,
119                              offset * storage->block_sz,
120                              size * storage->block_sz, &written);
121         if (ret < 0) {
122                 printf("Failed to write sparse chunk\n");
123                 return ret;
124         }
125
126         return written / storage->block_sz;
127 }
128
129 void fb_nand_flash_write(const char *partname, unsigned int session_id,
130                          void *download_buffer, unsigned int download_bytes,
131                          char *response)
132 {
133         struct part_info *part;
134         struct mtd_info *mtd = NULL;
135         int ret;
136
137         /* initialize the response buffer */
138         response_str = response;
139
140         ret = fb_nand_lookup(partname, response, &mtd, &part);
141         if (ret) {
142                 error("invalid NAND device");
143                 fastboot_fail(response_str, "invalid NAND device");
144                 return;
145         }
146
147         ret = board_fastboot_write_partition_setup(part->name);
148         if (ret)
149                 return;
150
151         if (is_sparse_image(download_buffer)) {
152                 struct fb_nand_sparse sparse_priv;
153                 sparse_storage_t sparse;
154
155                 sparse_priv.nand = mtd;
156                 sparse_priv.part = part;
157
158                 sparse.block_sz = mtd->writesize;
159                 sparse.start = part->offset / sparse.block_sz;
160                 sparse.size = part->size  / sparse.block_sz;
161                 sparse.name = part->name;
162                 sparse.write = fb_nand_sparse_write;
163
164                 ret = store_sparse_image(&sparse, &sparse_priv, session_id,
165                                          download_buffer);
166         } else {
167                 printf("Flashing raw image at offset 0x%llx\n",
168                        part->offset);
169
170                 ret = _fb_nand_write(mtd, part, download_buffer, part->offset,
171                                      download_bytes, NULL);
172
173                 printf("........ wrote %u bytes to '%s'\n",
174                        download_bytes, part->name);
175         }
176
177         if (ret) {
178                 fastboot_fail(response_str, "error writing the image");
179                 return;
180         }
181
182         fastboot_okay(response_str, "");
183 }
184
185 void fb_nand_erase(const char *partname, char *response)
186 {
187         struct part_info *part;
188         struct mtd_info *mtd = NULL;
189         int ret;
190
191         /* initialize the response buffer */
192         response_str = response;
193
194         ret = fb_nand_lookup(partname, response, &mtd, &part);
195         if (ret) {
196                 error("invalid NAND device");
197                 fastboot_fail(response_str, "invalid NAND device");
198                 return;
199         }
200
201         ret = board_fastboot_erase_partition_setup(part->name);
202         if (ret)
203                 return;
204
205         ret = _fb_nand_erase(mtd, part);
206         if (ret) {
207                 error("failed erasing from device %s", mtd->name);
208                 fastboot_fail(response_str, "failed erasing from device");
209                 return;
210         }
211
212         fastboot_okay(response_str, "");
213 }