From: iceman1001 <iceman@iuse.se>
Date: Sat, 8 Oct 2016 17:25:23 +0000 (+0200)
Subject: CHG:  "hf legic restore"  - added a filesize and cardsize check
X-Git-Url: http://cvs.zerfleddert.de/cgi-bin/gitweb.cgi/proxmark3-svn/commitdiff_plain/367996567bb9f22c1e53c1e8913e57a9570eb6b7

CHG:  "hf legic restore"  - added a filesize and cardsize check
---

diff --git a/client/cmdhflegic.c b/client/cmdhflegic.c
index 60c2e858..08e283b5 100644
--- a/client/cmdhflegic.c
+++ b/client/cmdhflegic.c
@@ -966,7 +966,7 @@ int CmdLegicRestore(const char *Cmd){
 	uint8_t *data = malloc(numofbytes);
 	if (!data) {
 		PrintAndLog("Fail, cannot allocate memory");
-		return 3;		
+		return 2;		
 	}
 	memset(data, 0, numofbytes);
 	
@@ -975,15 +975,27 @@ int CmdLegicRestore(const char *Cmd){
 	// set up file
 	fnameptr += fileNlen;
 	sprintf(fnameptr, ".bin");
-	
-	PrintAndLog("Reading binary file...");
 
 	if ((f = fopen(filename,"rb")) == NULL) {
 		PrintAndLog("File %s not found or locked", filename);
-		return 2;
+		return 3;
 	}	
 	
-	// verify size of dumpfile is the same as card.
+	// verify size of dumpfile is the same as card.	
+	fseek(f, 0, SEEK_END); // seek to end of file
+	size_t filesize = ftell(f); // get current file pointer
+	fseek(f, 0, SEEK_SET); // seek back to beginning of file
+	
+	if ( filesize != numofbytes) {
+		PrintAndLog("Fail, Filesize and cardsize is not equal. [%u != %u]", filesize, numofbytes);
+		free(data);
+		fclose(f);
+		return 4;
+	}
+		
+	
+	PrintAndLog("Reading binary file...");
+
 
 	// load file
 	size_t bytes_read = fread(data, 1, numofbytes, f);