Skip to content

Commit caade60

Browse files
Marsman1996rockcarry
authored andcommitted
exit program when bmp file is oversize
1 parent d5cfd49 commit caade60

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/ffjpeg.c

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,10 @@ int main(int argc, char *argv[])
2626
bmp_save(&bmp, "decode.bmp");
2727
bmp_free(&bmp);
2828
} else if (strcmp(argv[1], "-e") == 0) {
29-
bmp_load(&bmp, argv[2]);
29+
if (bmp_load(&bmp, argv[2]) == -1) {
30+
printf("failed to load bmp file: %s !\n", argv[2]);
31+
return -1;
32+
}
3033
jfif = jfif_encode(&bmp);
3134
bmp_free(&bmp);
3235
jfif_save(jfif, "encode.jpg");

0 commit comments

Comments
 (0)