[LOCALSPL] Fix parameter handling in LocalSetJob and add tests for the few ways we...
[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-2017 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 extern void func_fpSetJob(void);
31
32 const struct test winetest_testlist[] =
33 {
34 { "fpEnumPrinters", func_fpEnumPrinters },
35 { "fpGetPrintProcessorDirectory", func_fpGetPrintProcessorDirectory },
36 { "fpSetJob", func_fpSetJob },
37 { 0, 0 }
38 };
39
40 /**
41 * We don't link against winspool, so we don't have GetDefaultPrinterW.
42 * We can easily implement a similar function ourselves though.
43 */
44 PWSTR
45 GetDefaultPrinterFromRegistry(VOID)
46 {
47 static const WCHAR wszWindowsKey[] = L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Windows";
48 static const WCHAR wszDeviceValue[] = L"Device";
49
50 DWORD cbNeeded;
51 DWORD dwErrorCode;
52 HKEY hWindowsKey = NULL;
53 PWSTR pwszDevice;
54 PWSTR pwszComma;
55 PWSTR pwszReturnValue = NULL;
56
57 // Open the registry key where the default printer for the current user is stored.
58 dwErrorCode = (DWORD)RegOpenKeyExW(HKEY_CURRENT_USER, wszWindowsKey, 0, KEY_READ, &hWindowsKey);
59 if (dwErrorCode != ERROR_SUCCESS)
60 {
61 skip("RegOpenKeyExW failed with status %u!\n", dwErrorCode);
62 goto Cleanup;
63 }
64
65 // Determine the size of the required buffer.
66 dwErrorCode = (DWORD)RegQueryValueExW(hWindowsKey, wszDeviceValue, NULL, NULL, NULL, &cbNeeded);
67 if (dwErrorCode != ERROR_SUCCESS)
68 {
69 skip("RegQueryValueExW failed with status %u!\n", dwErrorCode);
70 goto Cleanup;
71 }
72
73 // Allocate it.
74 pwszDevice = HeapAlloc(GetProcessHeap(), 0, cbNeeded);
75 if (!pwszDevice)
76 {
77 skip("HeapAlloc failed!\n");
78 goto Cleanup;
79 }
80
81 // Now get the actual value.
82 dwErrorCode = RegQueryValueExW(hWindowsKey, wszDeviceValue, NULL, NULL, (PBYTE)pwszDevice, &cbNeeded);
83 if (dwErrorCode != ERROR_SUCCESS)
84 {
85 skip("RegQueryValueExW failed with status %u!\n", dwErrorCode);
86 goto Cleanup;
87 }
88
89 // We get a string "<Printer Name>,winspool,<Port>:".
90 // Extract the printer name from it.
91 pwszComma = wcschr(pwszDevice, L',');
92 if (!pwszComma)
93 {
94 skip("Found no or invalid default printer: %S!\n", pwszDevice);
95 goto Cleanup;
96 }
97
98 // Return the default printer.
99 *pwszComma = 0;
100 pwszReturnValue = pwszDevice;
101
102 Cleanup:
103 if (hWindowsKey)
104 RegCloseKey(hWindowsKey);
105
106 return pwszReturnValue;
107 }
108
109 BOOL
110 GetLocalsplFuncs(LPPRINTPROVIDOR pp)
111 {
112 HMODULE hLocalspl;
113 PInitializePrintProvidor pfnInitializePrintProvidor;
114
115 // Get us a handle to the loaded localspl.dll.
116 hLocalspl = GetModuleHandleW(L"localspl");
117 if (!hLocalspl)
118 {
119 skip("GetModuleHandleW failed with error %u!\n", GetLastError());
120 return FALSE;
121 }
122
123 // Get a pointer to its InitializePrintProvidor function.
124 pfnInitializePrintProvidor = (PInitializePrintProvidor)GetProcAddress(hLocalspl, "InitializePrintProvidor");
125 if (!pfnInitializePrintProvidor)
126 {
127 skip("GetProcAddress failed with error %u!\n", GetLastError());
128 return FALSE;
129 }
130
131 // Get localspl's function pointers.
132 if (!pfnInitializePrintProvidor(pp, sizeof(PRINTPROVIDOR), NULL))
133 {
134 skip("pfnInitializePrintProvidor failed with error %u!\n", GetLastError());
135 return FALSE;
136 }
137
138 return TRUE;
139 }
140
141 PVOID
142 GetSpoolssFunc(const char* FunctionName)
143 {
144 HMODULE hSpoolss;
145
146 // Get us a handle to the loaded spoolss.dll.
147 hSpoolss = GetModuleHandleW(L"spoolss");
148 if (!hSpoolss)
149 {
150 skip("GetModuleHandleW failed with error %u!\n", GetLastError());
151 return FALSE;
152 }
153
154 return GetProcAddress(hSpoolss, FunctionName);
155 }
156
157 // Running the tests from the injected DLL and redirecting their output to the pipe.
158 BOOL WINAPI
159 DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
160 {
161 char szTestName[150];
162 DWORD cbRead;
163 FILE* fpStdout;
164 HANDLE hCommandPipe;
165 int iOldStdout;
166
167 // We only want to run our test once when the DLL is injected to the process.
168 if (fdwReason != DLL_PROCESS_ATTACH)
169 return TRUE;
170
171 // Read the test to run from the command pipe.
172 hCommandPipe = CreateFileW(COMMAND_PIPE_NAME, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
173 if (hCommandPipe == INVALID_HANDLE_VALUE)
174 {
175 DPRINT("DLL: CreateFileW failed for the command pipe with error %lu!\n", GetLastError());
176 return FALSE;
177 }
178
179 if (!ReadFile(hCommandPipe, szTestName, sizeof(szTestName), &cbRead, NULL))
180 {
181 DPRINT("DLL: ReadFile failed for the command pipe with error %lu!\n", GetLastError());
182 return FALSE;
183 }
184
185 CloseHandle(hCommandPipe);
186
187 // Check if the test name is valid.
188 if (!find_test(szTestName))
189 {
190 DPRINT("DLL: Got invalid test name \"%s\"!\n", szTestName);
191 return FALSE;
192 }
193
194 // Backup our current stdout and set it to the output pipe.
195 iOldStdout = _dup(_fileno(stdout));
196 fpStdout = _wfreopen(OUTPUT_PIPE_NAME, L"w", stdout);
197 setbuf(stdout, NULL);
198
199 // Run the test.
200 run_test(szTestName);
201
202 // Restore stdout to the previous value.
203 fclose(fpStdout);
204 _dup2(iOldStdout, _fileno(stdout));
205
206 // Return FALSE so that our DLL is immediately unloaded.
207 return FALSE;
208 }