Previously, open_file_or_dir() will open block device successfully, however,
we should enhance such checks to make sure we are really opening a file or dir.

Signed-off-by: Wang Shilong <[email protected]>
---
v1->v2:
        this is more nature to fix the problem.
---
 utils.c | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/utils.c b/utils.c
index f499023..bf81704 100644
--- a/utils.c
+++ b/utils.c
@@ -1568,13 +1568,20 @@ int open_file_or_dir(const char *fname, DIR **dirstream)
        if (S_ISDIR(st.st_mode)) {
                *dirstream = opendir(fname);
                if (!*dirstream)
-                       return -2;
+                       return -1;
                fd = dirfd(*dirstream);
-       } else {
+       } else if (S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) {
                fd = open(fname, O_RDWR);
+       } else {
+               /*
+                * we set this on purpose, in case the caller output
+                * strerror(errno) as success
+                */
+               errno = EINVAL;
+               return -1;
        }
        if (fd < 0) {
-               fd = -3;
+               fd = -1;
                if (*dirstream)
                        closedir(*dirstream);
        }
-- 
1.8.3.1

--
To unsubscribe from this list: send the line "unsubscribe linux-btrfs" in
the body of a message to [email protected]
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to