dirname: Parse one-character paths properly

kernel/src/fs/VFS.cpp

@@ -3,6 +3,7 @@
 #include "thread/Thread.h"
 #include <bits/modes.h>
 #include <luna/PathParser.h>
+#include <luna/Utf8.h>
 
 namespace VFS
 {
@@ -46,6 +47,8 @@ namespace VFS
 
         auto child_name = TRY(parser.basename());
 
+        TRY(validate_filename(child_name.view()));
+
         return parent_inode->create_subdirectory(child_name.chars());
     }
 
@@ -60,9 +63,41 @@ namespace VFS
 
         auto child_name = TRY(parser.basename());
 
+        TRY(validate_filename(child_name.view()));
+
         return parent_inode->create_file(child_name.chars());
     }
 
+    Result<void> validate_filename(StringView name)
+    {
+        // Forbid problematic characters that could cause trouble in shell scripts and the like.
+        if (strpbrk(name.chars(), "*?:[]\"<>\\")) return err(EINVAL);
+
+        // Forbid filenames starting with a hyphen.
+        if (!name.is_empty() && name[0] == '-') return err(EINVAL);
+
+        // Forbid filenames with leading spaces.
+        if (!name.is_empty() && name[0] == ' ') return err(EINVAL);
+
+        // Forbid filenames with trailing spaces.
+        if (!name.is_empty() && name[name.length() - 1] == ' ') return err(EINVAL);
+
+        for (const auto& c : name)
+        {
+            // Forbid filenames with control characters.
+            if (c < 32 || c == 127) return err(EINVAL);
+        }
+
+        // Forbid filenames that are not valid UTF-8.
+
+        Utf8StringDecoder decoder(name.chars());
+
+        // This will fail if the filename is invalid UTF-8.
+        TRY(decoder.code_points());
+
+        return {};
+    }
+
     bool can_execute(SharedPtr<Inode> inode, Credentials auth)
     {
         if (inode->uid() == auth.euid) { return inode->mode() & S_IXUSR; }

kernel/src/fs/VFS.h

@@ -1,6 +1,7 @@
 #pragma once
 #include <luna/SharedPtr.h>
 #include <luna/StaticString.h>
+#include <luna/StringView.h>
 #include <sys/types.h>
 
 struct Credentials;
@@ -211,6 +212,8 @@ namespace VFS
     Result<SharedPtr<Inode>> create_file(const char* path, Credentials auth,
                                          SharedPtr<VFS::Inode> working_directory = {});
 
+    Result<void> validate_filename(StringView name);
+
     bool can_execute(SharedPtr<Inode> inode, Credentials auth);
     bool can_read(SharedPtr<Inode> inode, Credentials auth);
     bool can_write(SharedPtr<Inode> inode, Credentials auth);

kernel/src/sys/mknod.cpp

@@ -22,6 +22,8 @@ Result<u64> sys_mknod(Registers*, SyscallArgs args)
     auto dirname = TRY(parser.dirname());
     auto basename = TRY(parser.basename());
 
+    TRY(VFS::validate_filename(basename.view()));
+
     auto parent = TRY(VFS::resolve_path(dirname.chars(), current->auth, current->current_directory));
     if (!VFS::can_write(parent, current->auth)) return err(EACCES);
 

libluna/src/CPath.cpp

@@ -32,6 +32,8 @@ extern "C"
         usize len = strlen(path);
         if (!len) return dot;
 
+        if (len == 1 && *path != '/') return dot;
+
         // Strip trailing slashes.
         char* it = path + len - 1;
         while (*it == '/' && it != path) { it--; }