6914dc0c1e077758cd2f347cd43ee1760d4c2a17
[reactos.git] / modules / rostests / apitests / localspl / dll / main.c
1 /*
2 * PROJECT: ReactOS Local Spooler API Tests Injected DLL
3 * LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
4 * PURPOSE: Main functions
5 * COPYRIGHT: Copyright 2015-2016 Colin Finck (colin@reactos.org)
6 */
7
8 #define __ROS_LONG64__
9
10 #define STANDALONE
11 #include <apitest.h>
12
13 #define WIN32_NO_STATUS
14 #include <io.h>
15 #include <windef.h>
16 #include <winbase.h>
17 #include <wingdi.h>
18 #include <winreg.h>
19 #include <winspool.h>
20 #include <winsplp.h>
21
22 #include "../localspl_apitest.h"
23
24 //#define NDEBUG
25 #include <debug.h>
26
27 // Test list
28 extern void func_fpEnumPrinters(void);
29 extern void func_fpGetPrintProcessorDirectory(void);
30
31 const struct test winetest_testlist[] =
32 {
33 { "fpEnumPrinters", func_fpEnumPrinters },
34 { "fpGetPrintProcessorDirectory", func_fpGetPrintProcessorDirectory },
35 { 0, 0 }
36 };
37
38 BOOL
39 GetLocalsplFuncs(LPPRINTPROVIDOR pp)
40 {
41 HMODULE hLocalspl;
42 PInitializePrintProvidor pfnInitializePrintProvidor;
43
44 // Get us a handle to the loaded localspl.dll.
45 hLocalspl = GetModuleHandleW(L"localspl");
46 if (!hLocalspl)
47 {
48 skip("GetModuleHandleW failed with error %u!\n", GetLastError());
49 return FALSE;
50 }
51
52 // Get a pointer to its InitializePrintProvidor function.
53 pfnInitializePrintProvidor = (PInitializePrintProvidor)GetProcAddress(hLocalspl, "InitializePrintProvidor");
54 if (!pfnInitializePrintProvidor)
55 {
56 skip("GetProcAddress failed with error %u!\n", GetLastError());
57 return FALSE;
58 }
59
60 // Get localspl's function pointers.
61 if (!pfnInitializePrintProvidor(pp, sizeof(PRINTPROVIDOR), NULL))
62 {
63 skip("pfnInitializePrintProvidor failed with error %u!\n", GetLastError());
64 return FALSE;
65 }
66
67 return TRUE;
68 }
69
70 PVOID
71 GetSpoolssFunc(const char* FunctionName)
72 {
73 HMODULE hSpoolss;
74
75 // Get us a handle to the loaded spoolss.dll.
76 hSpoolss = GetModuleHandleW(L"spoolss");
77 if (!hSpoolss)
78 {
79 skip("GetModuleHandleW failed with error %u!\n", GetLastError());
80 return FALSE;
81 }
82
83 return GetProcAddress(hSpoolss, FunctionName);
84 }
85
86 // Running the tests from the injected DLL and redirecting their output to the pipe.
87 BOOL WINAPI
88 DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
89 {
90 char szTestName[150];
91 DWORD cbRead;
92 FILE* fpStdout;
93 HANDLE hCommandPipe;
94 int iOldStdout;
95
96 // We only want to run our test once when the DLL is injected to the process.
97 if (fdwReason != DLL_PROCESS_ATTACH)
98 return TRUE;
99
100 // Read the test to run from the command pipe.
101 hCommandPipe = CreateFileW(COMMAND_PIPE_NAME, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
102 if (hCommandPipe == INVALID_HANDLE_VALUE)
103 {
104 DPRINT("DLL: CreateFileW failed for the command pipe with error %lu!\n", GetLastError());
105 return FALSE;
106 }
107
108 if (!ReadFile(hCommandPipe, szTestName, sizeof(szTestName), &cbRead, NULL))
109 {
110 DPRINT("DLL: ReadFile failed for the command pipe with error %lu!\n", GetLastError());
111 return FALSE;
112 }
113
114 CloseHandle(hCommandPipe);
115
116 // Check if the test name is valid.
117 if (!find_test(szTestName))
118 {
119 DPRINT("DLL: Got invalid test name \"%s\"!\n", szTestName);
120 return FALSE;
121 }
122
123 // Backup our current stdout and set it to the output pipe.
124 iOldStdout = _dup(_fileno(stdout));
125 fpStdout = _wfreopen(OUTPUT_PIPE_NAME, L"w", stdout);
126 setbuf(stdout, NULL);
127
128 // Run the test.
129 run_test(szTestName);
130
131 // Restore stdout to the previous value.
132 fclose(fpStdout);
133 _dup2(iOldStdout, _fileno(stdout));
134
135 // Return FALSE so that our DLL is immediately unloaded.
136 return FALSE;
137 }