Fix write_raw_image on mtd devices.
Change-Id: I0fa64d4f4deaf8f067be3bd0b0bb963cf5af3f3f
diff --git a/flashutils/flashutils.h b/flashutils/flashutils.h
index 953dd00..d5dadcb 100644
--- a/flashutils/flashutils.h
+++ b/flashutils/flashutils.h
@@ -1,3 +1,6 @@
+#ifndef FLASHUTILS_H
+#define FLASHUTILS_H
+
int restore_raw_partition(const char *partition, const char *filename);
int backup_raw_partition(const char *partition, const char *filename);
int erase_raw_partition(const char *partition);
@@ -35,4 +38,14 @@
extern int cmd_bml_mount_partition(const char *partition, const char *mount_point, const char *filesystem, int read_only);
extern int cmd_bml_get_partition_device(const char *partition, char *device);
+extern int device_flash_type();
+enum flash_type {
+ UNSUPPORTED = -1,
+ UNKNOWN = 0,
+ MTD = 1,
+ MMC = 2,
+ BML = 3
+};
+
+#endif
\ No newline at end of file