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

2017-01-27 10:43:03 -08:00
parent 4b66ddea7e
commit 92a4dbf2e4
2 changed files with 10 additions and 1 deletions
@@ -414,6 +414,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 */

@@ -772,6 +777,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;