Fixed https://github.com/facebook/zstd/issues/232

2017-01-25 13:02:33 +01:00
parent e98788e0dc
commit 64fa2dbc5e
2 changed files with 10 additions and 1 deletions
@@ -361,6 +361,11 @@ static int FIO_compressFilename_srcFile(cRess_t ress,
        DISPLAYLEVEL(1, "zstd: %s is a directory -- ignored \n", srcFileName);
        return 1;
    }
+    if (!UTIL_doesFileExists(srcFileName)) {
+        DISPLAYLEVEL(1, "zstd: %s is not a regular file -- ignored \n", srcFileName);
+        return 1;
+    }
+
    ress.srcFile = FIO_openSrcFile(srcFileName);
    if (!ress.srcFile) return 1;   /* srcFile could not be opened */

@@ -719,6 +724,10 @@ static int FIO_decompressSrcFile(dRess_t ress, const char* dstFileName, const ch
        DISPLAYLEVEL(1, "zstd: %s is a directory -- ignored \n", srcFileName);
        return 1;
    }
+    if (!UTIL_doesFileExists(srcFileName)) {
+        DISPLAYLEVEL(1, "zstd: %s is not a regular file -- ignored \n", srcFileName);
+        return 1;
+    }

    srcFile = FIO_openSrcFile(srcFileName);
    if (srcFile==0) return 1;