Prepare v2013.07-rc2
[karo-tx-uboot.git] / drivers / dfu / dfu_nand.c
1 /*
2  * dfu_nand.c -- DFU for NAND routines.
3  *
4  * Copyright (C) 2012-2013 Texas Instruments, Inc.
5  *
6  * Based on dfu_mmc.c which is:
7  * Copyright (C) 2012 Samsung Electronics
8  * author: Lukasz Majewski <l.majewski@samsung.com>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
23  */
24
25 #include <common.h>
26 #include <malloc.h>
27 #include <errno.h>
28 #include <div64.h>
29 #include <dfu.h>
30 #include <linux/mtd/mtd.h>
31 #include <jffs2/load_kernel.h>
32 #include <nand.h>
33
34 enum dfu_nand_op {
35         DFU_OP_READ = 1,
36         DFU_OP_WRITE,
37 };
38
39 static int nand_block_op(enum dfu_nand_op op, struct dfu_entity *dfu,
40                         u64 offset, void *buf, long *len)
41 {
42         loff_t start, lim;
43         size_t count, actual;
44         int ret;
45         nand_info_t *nand;
46
47         /* if buf == NULL return total size of the area */
48         if (buf == NULL) {
49                 *len = dfu->data.nand.size;
50                 return 0;
51         }
52
53         start = dfu->data.nand.start + offset + dfu->bad_skip;
54         lim = dfu->data.nand.start + dfu->data.nand.size - start;
55         count = *len;
56
57         if (nand_curr_device < 0 ||
58             nand_curr_device >= CONFIG_SYS_MAX_NAND_DEVICE ||
59             !nand_info[nand_curr_device].name) {
60                 printf("%s: invalid nand device\n", __func__);
61                 return -1;
62         }
63
64         nand = &nand_info[nand_curr_device];
65
66         if (op == DFU_OP_READ)
67                 ret = nand_read_skip_bad(nand, start, &count, &actual,
68                                 lim, buf);
69         else
70                 ret = nand_write_skip_bad(nand, start, &count, &actual,
71                                 lim, buf, 0);
72
73         if (ret != 0) {
74                 printf("%s: nand_%s_skip_bad call failed at %llx!\n",
75                        __func__, op == DFU_OP_READ ? "read" : "write",
76                        start);
77                 return ret;
78         }
79
80         /*
81          * Find out where we stopped writing data.  This can be deeper into
82          * the NAND than we expected due to having to skip bad blocks.  So
83          * we must take this into account for the next write, if any.
84          */
85         if (actual > count)
86                 dfu->bad_skip += actual - count;
87
88         return ret;
89 }
90
91 static inline int nand_block_write(struct dfu_entity *dfu,
92                 u64 offset, void *buf, long *len)
93 {
94         return nand_block_op(DFU_OP_WRITE, dfu, offset, buf, len);
95 }
96
97 static inline int nand_block_read(struct dfu_entity *dfu,
98                 u64 offset, void *buf, long *len)
99 {
100         return nand_block_op(DFU_OP_READ, dfu, offset, buf, len);
101 }
102
103 static int dfu_write_medium_nand(struct dfu_entity *dfu,
104                 u64 offset, void *buf, long *len)
105 {
106         int ret = -1;
107
108         switch (dfu->layout) {
109         case DFU_RAW_ADDR:
110                 ret = nand_block_write(dfu, offset, buf, len);
111                 break;
112         default:
113                 printf("%s: Layout (%s) not (yet) supported!\n", __func__,
114                        dfu_get_layout(dfu->layout));
115         }
116
117         return ret;
118 }
119
120 static int dfu_read_medium_nand(struct dfu_entity *dfu, u64 offset, void *buf,
121                 long *len)
122 {
123         int ret = -1;
124
125         switch (dfu->layout) {
126         case DFU_RAW_ADDR:
127                 ret = nand_block_read(dfu, offset, buf, len);
128                 break;
129         default:
130                 printf("%s: Layout (%s) not (yet) supported!\n", __func__,
131                        dfu_get_layout(dfu->layout));
132         }
133
134         return ret;
135 }
136
137 int dfu_fill_entity_nand(struct dfu_entity *dfu, char *s)
138 {
139         char *st;
140         int ret, dev, part;
141
142         dfu->dev_type = DFU_DEV_NAND;
143         st = strsep(&s, " ");
144         if (!strcmp(st, "raw")) {
145                 dfu->layout = DFU_RAW_ADDR;
146                 dfu->data.nand.start = simple_strtoul(s, &s, 16);
147                 s++;
148                 dfu->data.nand.size = simple_strtoul(s, &s, 16);
149         } else if (!strcmp(st, "part")) {
150                 char mtd_id[32];
151                 struct mtd_device *mtd_dev;
152                 u8 part_num;
153                 struct part_info *pi;
154
155                 dfu->layout = DFU_RAW_ADDR;
156
157                 dev = simple_strtoul(s, &s, 10);
158                 s++;
159                 part = simple_strtoul(s, &s, 10);
160
161                 sprintf(mtd_id, "%s%d,%d", "nand", dev, part - 1);
162                 printf("using id '%s'\n", mtd_id);
163
164                 mtdparts_init();
165
166                 ret = find_dev_and_part(mtd_id, &mtd_dev, &part_num, &pi);
167                 if (ret != 0) {
168                         printf("Could not locate '%s'\n", mtd_id);
169                         return -1;
170                 }
171
172                 dfu->data.nand.start = pi->offset;
173                 dfu->data.nand.size = pi->size;
174
175         } else {
176                 printf("%s: Memory layout (%s) not supported!\n", __func__, st);
177                 return -1;
178         }
179
180         dfu->read_medium = dfu_read_medium_nand;
181         dfu->write_medium = dfu_write_medium_nand;
182
183         /* initial state */
184         dfu->inited = 0;
185
186         return 0;
187 }