From 5c9adff7f877caf3d642e25899084d29a8dec21b Mon Sep 17 00:00:00 2001
From: Nick Terrell <terrelln@fb.com>
Date: Wed, 21 Sep 2016 16:25:08 -0700
Subject: [PATCH] [pzstd] Check if input is a directory

---
 contrib/pzstd/Pzstd.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/contrib/pzstd/Pzstd.cpp b/contrib/pzstd/Pzstd.cpp
index 5dd84124d..978eb9968 100644
--- a/contrib/pzstd/Pzstd.cpp
+++ b/contrib/pzstd/Pzstd.cpp
@@ -102,6 +102,14 @@ static FILE *openInputFile(const std::string &inputFile,
     SET_BINARY_MODE(stdin);
     return stdin;
   }
+  // Check if input file is a directory
+  {
+    std::error_code ec;
+    if (is_directory(inputFile, ec)) {
+      errorHolder.setError("Output file is a directory -- ignored");
+      return nullptr;
+    }
+  }
   auto inputFd = std::fopen(inputFile.c_str(), "rb");
   if (!errorHolder.check(inputFd != nullptr, "Failed to open input file")) {
     return nullptr;