1 files changed, 194 insertions, 0 deletions
diff --git a/xpcom/tests/gtest/TestFilePreferencesWin.cpp b/xpcom/tests/gtest/TestFilePreferencesWin.cpp
new file mode 100644
index 0000000000..fdf777c6f4
--- /dev/null
+++ b/xpcom/tests/gtest/TestFilePreferencesWin.cpp
@@ -0,0 +1,194 @@
+#include "gtest/gtest.h"
+
+#include "mozilla/FilePreferences.h"
+#include "nsIFile.h"
+#include "nsXPCOMCID.h"
+
+TEST(FilePreferencesWin, Normalization)
+{
+  nsAutoString normalized;
+
+  mozilla::FilePreferences::testing::NormalizePath(u"foo"_ns, normalized);
+  ASSERT_TRUE(normalized == u"foo"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\foo"_ns, normalized);
+  ASSERT_TRUE(normalized == u"\\foo"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo"_ns, normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"foo\\some"_ns, normalized);
+  ASSERT_TRUE(normalized == u"foo\\some"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\.\\foo"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\."_ns, normalized);
+  ASSERT_TRUE(normalized == u"\\\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\.\\"_ns, normalized);
+  ASSERT_TRUE(normalized == u"\\\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\.\\."_ns, normalized);
+  ASSERT_TRUE(normalized == u"\\\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo\\bar"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar\\"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo\\bar\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar\\."_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo\\bar\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar\\.\\"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo\\bar\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar\\..\\"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar\\.."_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\foo\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\..\\bar\\..\\"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\..\\bar"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\bar"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(u"\\\\foo\\bar\\..\\..\\"_ns,
+                                                   normalized);
+  ASSERT_TRUE(normalized == u"\\\\"_ns);
+
+  mozilla::FilePreferences::testing::NormalizePath(
+      u"\\\\foo\\bar\\.\\..\\.\\..\\"_ns, normalized);
+  ASSERT_TRUE(normalized == u"\\\\"_ns);
+
+  bool result;
+
+  result = mozilla::FilePreferences::testing::NormalizePath(u"\\\\.."_ns,
+                                                            normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(u"\\\\..\\"_ns,
+                                                            normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(u"\\\\.\\..\\"_ns,
+                                                            normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(
+      u"\\\\foo\\\\bar"_ns, normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(
+      u"\\\\foo\\bar\\..\\..\\..\\..\\"_ns, normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(u"\\\\\\"_ns,
+                                                            normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(u"\\\\.\\\\"_ns,
+                                                            normalized);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::testing::NormalizePath(u"\\\\..\\\\"_ns,
+                                                            normalized);
+  ASSERT_FALSE(result);
+}
+
+TEST(FilePreferencesWin, AccessUNC)
+{
+  nsCOMPtr<nsIFile> lf = do_CreateInstance(NS_LOCAL_FILE_CONTRACTID);
+
+  nsresult rv;
+
+  mozilla::FilePreferences::testing::SetBlockUNCPaths(false);
+
+  rv = lf->InitWithPath(u"\\\\nice\\..\\evil\\share"_ns);
+  ASSERT_EQ(rv, NS_OK);
+
+  mozilla::FilePreferences::testing::SetBlockUNCPaths(true);
+
+  rv = lf->InitWithPath(u"\\\\nice\\..\\evil\\share"_ns);
+  ASSERT_EQ(rv, NS_ERROR_FILE_ACCESS_DENIED);
+
+  mozilla::FilePreferences::testing::AddDirectoryToWhitelist(u"\\\\nice"_ns);
+
+  rv = lf->InitWithPath(u"\\\\nice\\share"_ns);
+  ASSERT_EQ(rv, NS_OK);
+
+  rv = lf->InitWithPath(u"\\\\nice\\..\\evil\\share"_ns);
+  ASSERT_EQ(rv, NS_ERROR_FILE_ACCESS_DENIED);
+}
+
+TEST(FilePreferencesWin, AccessDOSDevicePath)
+{
+  const auto devicePathSpecifier = u"\\\\?\\"_ns;
+
+  nsCOMPtr<nsIFile> lf = do_CreateInstance(NS_LOCAL_FILE_CONTRACTID);
+
+  nsresult rv;
+
+  mozilla::FilePreferences::testing::SetBlockUNCPaths(true);
+
+  rv = lf->InitWithPath(devicePathSpecifier + u"evil\\z:\\share"_ns);
+  ASSERT_EQ(rv, NS_ERROR_FILE_ACCESS_DENIED);
+
+  rv = lf->InitWithPath(devicePathSpecifier + u"UNC\\evil\\share"_ns);
+  ASSERT_EQ(rv, NS_ERROR_FILE_ACCESS_DENIED);
+
+  rv = lf->InitWithPath(devicePathSpecifier + u"C:\\"_ns);
+  ASSERT_EQ(rv, NS_OK);
+
+  nsCOMPtr<nsIFile> base;
+  rv = NS_GetSpecialDirectory(NS_OS_TEMP_DIR, getter_AddRefs(base));
+  ASSERT_EQ(rv, NS_OK);
+
+  nsAutoString path;
+  rv = base->GetPath(path);
+  ASSERT_EQ(rv, NS_OK);
+
+  rv = lf->InitWithPath(devicePathSpecifier + path);
+  ASSERT_EQ(rv, NS_OK);
+}
+
+TEST(FilePreferencesWin, StartsWithDiskDesignatorAndBackslash)
+{
+  bool result;
+
+  result = mozilla::FilePreferences::StartsWithDiskDesignatorAndBackslash(
+      u"\\\\UNC\\path"_ns);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::StartsWithDiskDesignatorAndBackslash(
+      u"\\single\\backslash"_ns);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::StartsWithDiskDesignatorAndBackslash(
+      u"C:relative"_ns);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::StartsWithDiskDesignatorAndBackslash(
+      u"\\\\?\\C:\\"_ns);
+  ASSERT_FALSE(result);
+
+  result = mozilla::FilePreferences::StartsWithDiskDesignatorAndBackslash(
+      u"C:\\"_ns);
+  ASSERT_TRUE(result);
+
+  result = mozilla::FilePreferences::StartsWithDiskDesignatorAndBackslash(
+      u"c:\\"_ns);
+  ASSERT_TRUE(result);
+}