Merge remote-tracking branch 'qatar/master'
[ffmpeg.git] / tests / tiny_psnr.c
index 681fd13..512d47b 100644 (file)
@@ -119,33 +119,36 @@ int main(int argc,char* argv[]){
     if(argc<3){
         printf("tiny_psnr <file1> <file2> [<elem size> [<shift> [<skip bytes>]]]\n");
         printf("WAV headers are skipped automatically.\n");
-        return -1;
+        return 1;
     }
 
     f[0]= fopen(argv[1], "rb");
     f[1]= fopen(argv[2], "rb");
     if(!f[0] || !f[1]){
         fprintf(stderr, "Could not open input files.\n");
-        return -1;
+        return 1;
     }
 
     for (i = 0; i < 2; i++) {
         uint8_t *p = buf[i];
-        fread(p, 1, 12, f[i]);
+        if (fread(p, 1, 12, f[i]) != 12)
+            return 1;
         if (!memcmp(p,   "RIFF", 4) &&
             !memcmp(p+8, "WAVE", 4)) {
-            fread(p, 1, 8, f[i]);
+            if (fread(p, 1, 8, f[i]) != 8)
+                return 1;
             while (memcmp(p, "data", 4)) {
                 int s = p[4] | p[5]<<8 | p[6]<<16 | p[7]<<24;
                 fseek(f[i], s, SEEK_CUR);
-                fread(p, 1, 8, f[i]);
+                if (fread(p, 1, 8, f[i]) != 8)
+                    return 1;
             }
         } else {
             fseek(f[i], -12, SEEK_CUR);
         }
     }
 
-    fseek(f[shift<0], shift < 0 ? -shift : shift, SEEK_CUR);
+    fseek(f[shift<0], abs(shift), SEEK_CUR);
 
     fseek(f[0],skip_bytes,SEEK_CUR);
     fseek(f[1],skip_bytes,SEEK_CUR);