1 files changed, 136 insertions, 0 deletions
diff --git a/mtdutils/dump_image.c b/mtdutils/dump_image.c
new file mode 100644
index 0000000..bc96abc
--- /dev/null
+++ b/mtdutils/dump_image.c
@@ -0,0 +1,136 @@
+/*
+ * Copyright (C) 2008 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+
+#include "cutils/log.h"
+#include "mtdutils.h"
+
+#ifdef LOG_TAG
+#undef LOG_TAG
+#endif
+
+#define LOG_TAG "dump_image"
+
+#define BLOCK_SIZE	2048
+#define SPARE_SIZE	(BLOCK_SIZE >> 5)
+
+void die(const char *msg, ...) {
+    int err = errno;
+    va_list args;
+    va_start(args, msg);
+    char buf[1024];
+    vsnprintf(buf, sizeof(buf), msg, args);
+    va_end(args);
+
+    if (err != 0) {
+        strlcat(buf, ": ", sizeof(buf));
+        strlcat(buf, strerror(err), sizeof(buf));
+    }
+
+    fprintf(stderr, "%s\n", buf);
+    exit(1);
+}
+
+/* Read a flash partition and write it to an image file. */
+
+int main(int argc, char **argv)
+{
+    ssize_t (*read_func) (MtdReadContext *, char *, size_t);
+    MtdReadContext *in;
+    const MtdPartition *partition;
+    char buf[BLOCK_SIZE + SPARE_SIZE];
+    size_t partition_size;
+    size_t read_size;
+    size_t total;
+    int fd;
+    int wrote;
+    int len;
+
+    if (argc != 3) {
+        fprintf(stderr, "usage: %s partition file.img\n", argv[0]);
+        return 2;
+    }
+
+    if (mtd_scan_partitions() <= 0)
+    	die("error scanning partitions");
+
+    partition = mtd_find_partition_by_name(argv[1]);
+    if (partition == NULL)
+   	 die("can't find %s partition", argv[1]);
+
+    if (mtd_partition_info(partition, &partition_size, NULL, NULL)) {
+   	 die("can't get info of partition %s", argv[1]);
+    }
+
+    if (!strcmp(argv[2], "-")) {
+	fd = fileno(stdout);
+    } else {
+	fd = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0666);
+    }
+
+    if (fd < 0)
+    	die("error opening %s", argv[2]);
+
+    in = mtd_read_partition(partition);
+    if (in == NULL) {
+    	close(fd);
+	unlink(argv[2]);
+        die("error opening %s: %s\n", argv[1], strerror(errno));
+    }
+
+    if (!strcmp(argv[1], "system") ||
+        !strcmp(argv[1], "cache") ||
+	!strcmp(argv[1], "userdata")) {
+	read_size = BLOCK_SIZE + SPARE_SIZE;
+	read_func = mtd_read_raw;
+    } else {
+	read_size = BLOCK_SIZE;
+	read_func = mtd_read_data;
+    }
+
+    total = 0;
+    while ((len = read_func(in, buf, read_size)) > 0) {
+        wrote = write(fd, buf, len);
+        if (wrote != len) {
+    		close(fd);
+		unlink(argv[2]);
+		die("error writing %s", argv[2]);
+	}
+	total += BLOCK_SIZE;
+    }
+
+    if (total != partition_size) {
+    	close(fd);
+	unlink(argv[2]);
+    	die("error reading %s", argv[1]);
+    }
+
+    mtd_read_close(in);
+
+    if (close(fd)) {
+	unlink(argv[2]);
+    	die("error closing %s", argv[2]);
+    }
+
+    return 0;
+}