dll: FspLaunchReg*: testing

tst/winfsp-tests/launch-test.c

@@ -0,0 +1,166 @@
+/**
+ * @file launch-test.c
+ *
+ * @copyright 2015-2018 Bill Zissimopoulos
+ */
+/*
+ * This file is part of WinFsp.
+ *
+ * You can redistribute it and/or modify it under the terms of the GNU
+ * General Public License version 3 as published by the Free Software
+ * Foundation.
+ *
+ * Licensees holding a valid commercial license may use this file in
+ * accordance with the commercial license agreement provided with the
+ * software.
+ */
+
+#include <winfsp/launch.h>
+#include <tlib/testsuite.h>
+
+#include "winfsp-tests.h"
+
+static void launch_reg_test(void)
+{
+    NTSTATUS Result;
+    PWSTR ClassName = L"winfsp-tests-launch-reg-test";
+    FSP_LAUNCH_REG_RECORD RecordBuf = { 0 }, *Record;
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(STATUS_OBJECT_NAME_NOT_FOUND == Result);
+    ASSERT(0 == Record);
+
+    RecordBuf.Executable = 0;
+    Result = FspLaunchRegSetRecord(ClassName, &RecordBuf);
+    ASSERT(STATUS_INVALID_PARAMETER == Result);
+
+    RecordBuf.Executable = L"Executable";
+    RecordBuf.CommandLine = L"CommandLine";
+    RecordBuf.WorkDirectory = L"WorkDirectory";
+    RecordBuf.RunAs = L"RunAs";
+    RecordBuf.Security = L"Security";
+    RecordBuf.JobControl = 1;
+    RecordBuf.Credentials = 42;
+    Result = FspLaunchRegSetRecord(ClassName, &RecordBuf);
+    if (STATUS_ACCESS_DENIED == Result)
+    {
+        FspDebugLog(__FUNCTION__ ": need Administrator\n");
+        return;
+    }
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(NT_SUCCESS(Result));
+    ASSERT(0 != Record);
+    ASSERT(0 == Record->Agent);
+    ASSERT(0 == wcscmp(RecordBuf.Executable, Record->Executable));
+    ASSERT(0 == wcscmp(RecordBuf.CommandLine, Record->CommandLine));
+    ASSERT(0 == wcscmp(RecordBuf.WorkDirectory, Record->WorkDirectory));
+    ASSERT(0 == wcscmp(RecordBuf.RunAs, Record->RunAs));
+    ASSERT(0 == wcscmp(RecordBuf.Security, Record->Security));
+    ASSERT(RecordBuf.JobControl == Record->JobControl);
+    ASSERT(RecordBuf.Credentials == Record->Credentials);
+    FspLaunchRegFreeRecord(Record);
+
+    Result = FspLaunchRegSetRecord(ClassName, &RecordBuf);
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(NT_SUCCESS(Result));
+    ASSERT(0 != Record);
+    ASSERT(0 == Record->Agent);
+    ASSERT(0 == wcscmp(RecordBuf.Executable, Record->Executable));
+    ASSERT(0 == wcscmp(RecordBuf.CommandLine, Record->CommandLine));
+    ASSERT(0 == wcscmp(RecordBuf.WorkDirectory, Record->WorkDirectory));
+    ASSERT(0 == wcscmp(RecordBuf.RunAs, Record->RunAs));
+    ASSERT(0 == wcscmp(RecordBuf.Security, Record->Security));
+    ASSERT(RecordBuf.JobControl == Record->JobControl);
+    ASSERT(RecordBuf.Credentials == Record->Credentials);
+    FspLaunchRegFreeRecord(Record);
+
+    RecordBuf.Security = 0;
+    RecordBuf.Credentials = 0;
+    Result = FspLaunchRegSetRecord(ClassName, &RecordBuf);
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(NT_SUCCESS(Result));
+    ASSERT(0 != Record);
+    ASSERT(0 == Record->Agent);
+    ASSERT(0 == wcscmp(RecordBuf.Executable, Record->Executable));
+    ASSERT(0 == wcscmp(RecordBuf.CommandLine, Record->CommandLine));
+    ASSERT(0 == wcscmp(RecordBuf.WorkDirectory, Record->WorkDirectory));
+    ASSERT(0 == wcscmp(RecordBuf.RunAs, Record->RunAs));
+    ASSERT(0 == Record->Security);
+    ASSERT(RecordBuf.JobControl == Record->JobControl);
+    ASSERT(RecordBuf.Credentials == Record->Credentials);
+    FspLaunchRegFreeRecord(Record);
+
+    Result = FspLaunchRegSetRecord(ClassName, 0);
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(STATUS_OBJECT_NAME_NOT_FOUND == Result);
+    ASSERT(0 == Record);
+
+    Result = FspLaunchRegSetRecord(ClassName, 0);
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(STATUS_OBJECT_NAME_NOT_FOUND == Result);
+    ASSERT(0 == Record);
+
+    RecordBuf.Agent = L"Agent1,Agent2";
+    Result = FspLaunchRegSetRecord(ClassName, &RecordBuf);
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, L"Agent", &Record);
+    ASSERT(STATUS_OBJECT_NAME_NOT_FOUND == Result);
+    ASSERT(0 == Record);
+
+    Result = FspLaunchRegGetRecord(ClassName, L"Agent1", &Record);
+    ASSERT(NT_SUCCESS(Result));
+    ASSERT(0 != Record);
+    ASSERT(0 == wcscmp(RecordBuf.Agent, Record->Agent));
+    ASSERT(0 == wcscmp(RecordBuf.Executable, Record->Executable));
+    ASSERT(0 == wcscmp(RecordBuf.CommandLine, Record->CommandLine));
+    ASSERT(0 == wcscmp(RecordBuf.WorkDirectory, Record->WorkDirectory));
+    ASSERT(0 == wcscmp(RecordBuf.RunAs, Record->RunAs));
+    ASSERT(0 == Record->Security);
+    ASSERT(RecordBuf.JobControl == Record->JobControl);
+    ASSERT(RecordBuf.Credentials == Record->Credentials);
+    FspLaunchRegFreeRecord(Record);
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(NT_SUCCESS(Result));
+    ASSERT(0 != Record);
+    ASSERT(0 == wcscmp(RecordBuf.Agent, Record->Agent));
+    ASSERT(0 == wcscmp(RecordBuf.Executable, Record->Executable));
+    ASSERT(0 == wcscmp(RecordBuf.CommandLine, Record->CommandLine));
+    ASSERT(0 == wcscmp(RecordBuf.WorkDirectory, Record->WorkDirectory));
+    ASSERT(0 == wcscmp(RecordBuf.RunAs, Record->RunAs));
+    ASSERT(0 == Record->Security);
+    ASSERT(RecordBuf.JobControl == Record->JobControl);
+    ASSERT(RecordBuf.Credentials == Record->Credentials);
+    FspLaunchRegFreeRecord(Record);
+
+    Result = FspLaunchRegSetRecord(ClassName, 0);
+    ASSERT(NT_SUCCESS(Result));
+
+    Result = FspLaunchRegGetRecord(ClassName, 0, &Record);
+    ASSERT(STATUS_OBJECT_NAME_NOT_FOUND == Result);
+    ASSERT(0 == Record);
+}
+
+static void launch_test(void)
+{
+}
+
+void launch_tests(void)
+{
+    if (OptExternal)
+        return;
+
+    TEST(launch_reg_test);
+    TEST(launch_test);
+}

tst/winfsp-tests/winfsp-tests.c

@@ -186,6 +186,7 @@ int main(int argc, char *argv[])
     TESTSUITE(path_tests);
     TESTSUITE(dirbuf_tests);
     TESTSUITE(version_tests);
+    TESTSUITE(launch_tests);
     TESTSUITE(mount_tests);
     TESTSUITE(timeout_tests);
     TESTSUITE(memfs_tests);