Fix write_raw_image on mtd devices.

Change-Id: I0fa64d4f4deaf8f067be3bd0b0bb963cf5af3f3f
diff --git a/flashutils/flashutils.c b/flashutils/flashutils.c
index 5c57756..b71d4fa 100644
--- a/flashutils/flashutils.c
+++ b/flashutils/flashutils.c
@@ -5,14 +5,6 @@
 
 #include "flashutils/flashutils.h"
 
-enum flash_type {
-    UNSUPPORTED = -1,
-    UNKNOWN = 0,
-    MTD = 1,
-    MMC = 2,
-    BML = 3
-};
-
 int the_flash_type = UNKNOWN;
 
 int device_flash_type()
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