fail if the device isn't an mmcblk device

1 files changed, 6 insertions, 0 deletions

diff --git a/deonebook.c b/deonebook.c
index b4300dc..23b61ea 100644
--- a/deonebook.c
+++ b/deonebook.c
@@ -1,4 +1,5 @@
 #define _XOPEN_SOURCE 700
+#include <string.h>
 #include <stdio.h>
 #include <unistd.h>
 #include "deonebook.h"
@@ -36,6 +37,11 @@ int main(int argc, char *argv[])
 			if (optind < argc) {
 				device = argv[optind];
 			}
+			if (strstr(device, "mmcblk") != device) {
+				fprintf(stderr, "device must be an mmcblk device\n");
+				return 1;
+			}
+
 			key = genkey(device);
 		}