diff --git a/flash_rom.c b/flash_rom.c index b9ef45e37..b1ca7b346 100644 --- a/flash_rom.c +++ b/flash_rom.c @@ -28,6 +28,8 @@ #include #include #include +#include +#include #include #include #include @@ -332,10 +334,21 @@ int main(int argc, char *argv[]) fclose(image); printf("done\n"); } else { + struct stat image_stat; + if ((image = fopen(filename, "r")) == NULL) { perror(filename); exit(1); } + if (fstat(fileno(image), &image_stat) != 0) { + perror(filename); + exit(1); + } + if(image_stat.st_size!=flash->total_size*1024) { + perror("Image size doesnt match"); + exit(1); + } + fread(buf, sizeof(char), size, image); show_id(buf, size); fclose(image);