2 * mtd - simple memory technology device manipulation tool
4 * Copyright (C) 2005 Waldemar Brodkorb <wbx@dass-it.de>
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
20 * mtd utility for the openwrt project
21 * it is mainly code from the linux-mtd project, which accepts the same
22 * command line arguments as the broadcom utility
37 #include <sys/ioctl.h>
38 #include <sys/types.h>
39 #include <sys/param.h>
40 #include <sys/mount.h>
44 #include <linux/mtd/mtd.h>
47 #define TRX_MAGIC 0x30524448 /* "HDR0" */
49 #define TRX_MAX_LEN 0x3A0000
50 #define TRX_NO_HEADER 1 /* Do not write TRX header */
53 uint32_t magic; /* "HDR0" */
54 uint32_t len; /* Length of file including header */
55 uint32_t crc32; /* 32-bit CRC from flag_version to end of file */
56 uint32_t flag_version; /* 0:15 flags, 16:31 version */
57 uint32_t offsets[3]; /* Offsets of partitions from start of header */
60 #define BUFSIZE (10 * 1024)
62 extern int mtd_open(const char *mtd, int flags);
63 extern int mtd_erase(const char *mtd);
64 extern int mtd_write(const char *trxfile, const char *mtd);
65 extern int mtd_update(const char *trxfile, const char *mtd);
68 mtd_unlock(const char *mtd)
71 struct mtd_info_user mtdInfo;
72 struct erase_info_user mtdLockInfo;
74 fd = mtd_open(mtd, O_RDWR);
76 fprintf(stderr, "Could not open mtd device: %s\n", mtd);
80 if(ioctl(fd, MEMGETINFO, &mtdInfo)) {
81 fprintf(stderr, "Could not get MTD device info from %s\n", mtd);
86 mtdLockInfo.start = 0;
87 mtdLockInfo.length = mtdInfo.size;
88 if(ioctl(fd, MEMUNLOCK, &mtdLockInfo)) {
89 fprintf(stderr, "Could not unlock MTD device: %s\n", mtd);
99 mtd_open(const char *mtd, int flags)
105 if ((fp = fopen("/proc/mtd", "r"))) {
106 while (fgets(dev, sizeof(dev), fp)) {
107 if (sscanf(dev, "mtd%d:", &i) && strstr(dev, mtd)) {
108 snprintf(dev, sizeof(dev), "/dev/mtd/%d", i);
110 return open(dev, flags);
116 return open(mtd, flags);
120 mtd_erase(const char *mtd)
123 struct mtd_info_user mtdInfo;
124 struct erase_info_user mtdEraseInfo;
126 fd = mtd_open(mtd, O_RDWR);
128 fprintf(stderr, "Could not open mtd device: %s\n", mtd);
132 if(ioctl(fd, MEMGETINFO, &mtdInfo)) {
133 fprintf(stderr, "Could not get MTD device info from %s\n", mtd);
138 mtdEraseInfo.length = mtdInfo.erasesize;
140 for (mtdEraseInfo.start = 0;
141 mtdEraseInfo.start < mtdInfo.size;
142 mtdEraseInfo.start += mtdInfo.erasesize) {
144 ioctl(fd, MEMUNLOCK, &mtdEraseInfo);
145 if(ioctl(fd, MEMERASE, &mtdEraseInfo)) {
146 fprintf(stderr, "Could not erase MTD device: %s\n", mtd);
158 mtd_write(const char *trxfile, const char *mtd)
163 size_t result,size,written;
164 struct mtd_info_user mtdInfo;
165 struct erase_info_user mtdEraseInfo;
167 unsigned char src[BUFSIZE],dest[BUFSIZE];
169 fd = mtd_open(mtd, O_RDWR);
171 fprintf(stderr, "Could not open mtd device: %s\n", mtd);
175 if(ioctl(fd, MEMGETINFO, &mtdInfo)) {
176 fprintf(stderr, "Could not get MTD device info from %s\n", mtd);
181 trxfd = open(trxfile,O_RDONLY);
183 fprintf(stderr, "Could not open trx image: %s\n", trxfile);
187 if (fstat (trxfd,&trxstat) < 0) {
188 fprintf(stderr, "Could not get trx image file status: %s\n", trxfile);
193 if(mtdInfo.size < trxstat.st_size) {
194 fprintf(stderr, "Image too big for partition: %s\n", mtd);
199 mtdEraseInfo.start = 0;
200 mtdEraseInfo.length = trxstat.st_size & ~(mtdInfo.erasesize -1);
201 if(trxstat.st_size % mtdInfo.erasesize) mtdEraseInfo.length += mtdInfo.erasesize;
203 /* erase the chunk */
204 if (ioctl (fd,MEMERASE,&mtdEraseInfo) < 0) {
205 fprintf(stderr, "Erasing mtd failed: %s\n", mtd);
209 size = trxstat.st_size;
214 if (size < BUFSIZE) i = size;
216 result = write(fd,src,i);
219 fprintf(stderr,"Error while writing image");
222 fprintf(stderr,"Error writing image");
233 mtd_update(const char *trxfile, const char *mtd)
235 if (mtd_erase("rootfs") != 0) {
236 fprintf(stderr, "Could not erase rootfs\n");
239 if (mtd_write(trxfile, mtd) != 0) {
240 fprintf(stderr, "Could not update %s with %s\n", mtd, trxfile);
246 int main(int argc, char **argv) {
247 if(argc == 3 && strcasecmp(argv[1],"unlock")==0) {
248 printf("Unlocking %s ...\n",argv[2]);
249 return mtd_unlock(argv[2]);
251 if(argc == 3 && strcasecmp(argv[1],"erase")==0) {
252 printf("Erasing %s ...\n",argv[2]);
253 return mtd_erase(argv[2]);
255 if(argc == 4 && strcasecmp(argv[1],"write")==0) {
256 printf("Writing %s to %s ...\n",argv[2],argv[3]);
257 return mtd_write(argv[2],argv[3]);
259 if(argc == 4 && strcasecmp(argv[1],"update")==0) {
260 printf("Updating %s on %s ...\n",argv[2],argv[3]);
261 return mtd_update(argv[2],argv[3]);
264 printf("no valid command given\n");
265 printf("\nmtd: modify data within a Memory Technology Device.\n");
266 printf("Copyright (C) 2005 Waldemar Brodkorb <wbx@dass-it.de>\n");
267 printf("Documented by Mike Strates [dumpedcore] <mike@dilaudid.net>\n");
268 printf("mtd has ABSOLUTELY NO WARRANTY and is licensed under the GNU GPL.\n");
269 printf("\nUsage: mtd [unlock|erase] device\n");
270 printf(" mtd [write|update] imagefile device\n");
271 printf("\n.. where device is in the format of mtdX (eg: mtd4) or its label.\n\n");
272 printf("unlock enable modification to device\n");
273 printf("erase erase all data on device\n");
274 printf("write write imagefile to device\n");
275 printf("update remove rootfs and update imagefile on device\n");
276 printf("\nExample: To write linux.trx to mtd4 labeled as linux\n");
277 printf("\n mtd unlock linux && mtd write linux.trx linux\n\n");