[SPOOLSS] Stub BuildOtherNamesFromMachineName as seen in wine commit cdb5f35. This...
[reactos.git] / reactos / win32ss / printing / base / spoolss / main.c
1 /*
2 * PROJECT: ReactOS Spooler Router
3 * LICENSE: GNU LGPL v2.1 or any later version as published by the Free Software Foundation
4 * PURPOSE: Main functions
5 * COPYRIGHT: Copyright 2015 Colin Finck <colin@reactos.org>
6 */
7
8 #include "precomp.h"
9
10 // Global Variables
11 HANDLE hProcessHeap;
12 LIST_ENTRY PrintProviderList;
13
14
15 static DWORD
16 _AddPrintProviderToList(PCWSTR pwszFileName)
17 {
18 DWORD dwErrorCode;
19 HINSTANCE hinstPrintProvider;
20 PInitializePrintProvidor pfnInitializePrintProvidor;
21 PSPOOLSS_PRINT_PROVIDER pPrintProvider = NULL;
22
23 // Try to load it.
24 hinstPrintProvider = LoadLibraryW(pwszFileName);
25 if (!hinstPrintProvider)
26 {
27 dwErrorCode = GetLastError();
28 ERR("LoadLibraryW failed for \"%S\" with error %lu!\n", pwszFileName, dwErrorCode);
29 goto Cleanup;
30 }
31
32 // Get the initialization routine.
33 pfnInitializePrintProvidor = (PInitializePrintProvidor)GetProcAddress(hinstPrintProvider, "InitializePrintProvidor");
34 if (!pfnInitializePrintProvidor)
35 {
36 dwErrorCode = GetLastError();
37 ERR("GetProcAddress failed for \"%S\" with error %lu!\n", pwszFileName, dwErrorCode);
38 goto Cleanup;
39 }
40
41 // Create a new SPOOLSS_PRINT_PROVIDER structure for it.
42 pPrintProvider = DllAllocSplMem(sizeof(SPOOLSS_PRINT_PROVIDER));
43 if (!pPrintProvider)
44 {
45 dwErrorCode = ERROR_NOT_ENOUGH_MEMORY;
46 ERR("DllAllocSplMem failed!\n");
47 goto Cleanup;
48 }
49
50 // Call the Print Provider initialization function.
51 if (!pfnInitializePrintProvidor(&pPrintProvider->PrintProvider, sizeof(PRINTPROVIDOR), NULL))
52 {
53 dwErrorCode = GetLastError();
54 ERR("InitializePrintProvidor failed for \"%S\" with error %lu!\n", pwszFileName, dwErrorCode);
55 goto Cleanup;
56 }
57
58 // Add this Print Provider to the list.
59 InsertTailList(&PrintProviderList, &pPrintProvider->Entry);
60
61 // Don't let the cleanup routine free this.
62 pPrintProvider = NULL;
63 dwErrorCode = ERROR_SUCCESS;
64
65 Cleanup:
66 if (pPrintProvider)
67 DllFreeSplMem(pPrintProvider);
68
69 return dwErrorCode;
70 }
71
72 static BOOL
73 _InitializePrintProviderList()
74 {
75 DWORD cbFileName;
76 DWORD cchMaxSubKey;
77 DWORD cchPrintProviderName;
78 DWORD dwErrorCode;
79 DWORD dwSubKeys;
80 DWORD i;
81 HKEY hKey = NULL;
82 HKEY hSubKey = NULL;
83 PWSTR pwszPrintProviderName = NULL;
84 WCHAR wszFileName[MAX_PATH];
85
86 // Initialize an empty list for our Print Providers.
87 InitializeListHead(&PrintProviderList);
88
89 // First add the Local Spooler.
90 // This one must exist and must be the first one in the list.
91 dwErrorCode = _AddPrintProviderToList(L"localspl");
92 if (dwErrorCode != ERROR_SUCCESS)
93 {
94 ERR("The Local Spooler could not be loaded!\n");
95 goto Cleanup;
96 }
97
98 // Now add additional Print Providers from the registry.
99 // First of all, open the key containing print providers.
100 dwErrorCode = (DWORD)RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\Print\\Providers", 0, KEY_READ, &hKey);
101 if (dwErrorCode != ERROR_SUCCESS)
102 {
103 ERR("RegOpenKeyExW failed with status %lu!\n", dwErrorCode);
104 goto Cleanup;
105 }
106
107 // Get the number of Print Providers and maximum sub key length.
108 dwErrorCode = (DWORD)RegQueryInfoKeyW(hKey, NULL, NULL, NULL, &dwSubKeys, &cchMaxSubKey, NULL, NULL, NULL, NULL, NULL, NULL);
109 if (dwErrorCode != ERROR_SUCCESS)
110 {
111 ERR("RegQueryInfoKeyW failed with status %lu!\n", dwErrorCode);
112 goto Cleanup;
113 }
114
115 // Allocate a temporary buffer for the Print Provider names.
116 pwszPrintProviderName = DllAllocSplMem((cchMaxSubKey + 1) * sizeof(WCHAR));
117 if (!pwszPrintProviderName)
118 {
119 dwErrorCode = ERROR_NOT_ENOUGH_MEMORY;
120 ERR("DllAllocSplMem failed!\n");
121 goto Cleanup;
122 }
123
124 // Loop through all available Print Providers.
125 for (i = 0; i < dwSubKeys; i++)
126 {
127 // Cleanup tasks from the previous run
128 if (hSubKey)
129 {
130 RegCloseKey(hSubKey);
131 hSubKey = NULL;
132 }
133
134 // Get the name of this Print Provider.
135 cchPrintProviderName = cchMaxSubKey + 1;
136 dwErrorCode = (DWORD)RegEnumKeyExW(hKey, i, pwszPrintProviderName, &cchPrintProviderName, NULL, NULL, NULL, NULL);
137 if (dwErrorCode != ERROR_SUCCESS)
138 {
139 ERR("RegEnumKeyExW failed for iteration %lu with status %lu!\n", i, dwErrorCode);
140 continue;
141 }
142
143 // Open this Print Provider's registry key.
144 dwErrorCode = (DWORD)RegOpenKeyExW(hKey, pwszPrintProviderName, 0, KEY_READ, &hSubKey);
145 if (dwErrorCode != ERROR_SUCCESS)
146 {
147 ERR("RegOpenKeyExW failed for Print Provider \"%S\" with status %lu!\n", pwszPrintProviderName, dwErrorCode);
148 continue;
149 }
150
151 // Get the file name of the Print Provider.
152 cbFileName = MAX_PATH * sizeof(WCHAR);
153 dwErrorCode = (DWORD)RegQueryValueExW(hKey, L"Driver", NULL, NULL, (PBYTE)wszFileName, &cbFileName);
154 if (dwErrorCode != ERROR_SUCCESS)
155 {
156 ERR("RegQueryValueExW failed with status %lu!\n", dwErrorCode);
157 continue;
158 }
159
160 // Load and add it to the list.
161 dwErrorCode = _AddPrintProviderToList(wszFileName);
162 if (dwErrorCode != ERROR_SUCCESS)
163 continue;
164 }
165
166 dwErrorCode = ERROR_SUCCESS;
167
168 Cleanup:
169 // Inside the loop
170 if (hSubKey)
171 RegCloseKey(hSubKey);
172
173 // Outside the loop
174 if (pwszPrintProviderName)
175 DllFreeSplMem(pwszPrintProviderName);
176
177 if (hKey)
178 RegCloseKey(hKey);
179
180 SetLastError(dwErrorCode);
181 return (dwErrorCode == ERROR_SUCCESS);
182 }
183
184 BOOL WINAPI
185 DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
186 {
187 switch (fdwReason)
188 {
189 case DLL_PROCESS_ATTACH:
190 DisableThreadLibraryCalls(hinstDLL);
191 hProcessHeap = GetProcessHeap();
192 break;
193 }
194
195 return TRUE;
196 }
197
198 BOOL WINAPI
199 InitializeRouter(HANDLE SpoolerStatusHandle)
200 {
201 return _InitializePrintProviderList();
202 }
203
204 BOOL WINAPI
205 SplInitializeWinSpoolDrv(PVOID* pTable)
206 {
207 HINSTANCE hWinspool;
208 int i;
209
210 hWinspool = LoadLibraryW(L"winspool.drv");
211 if (!hWinspool)
212 {
213 ERR("Could not load winspool.drv, last error is %lu!\n", GetLastError());
214 return FALSE;
215 }
216
217 // Get the function pointers which are meant to be returned by this function.
218 pTable[0] = GetProcAddress(hWinspool, "OpenPrinterW");
219 pTable[1] = GetProcAddress(hWinspool, "ClosePrinter");
220 pTable[2] = GetProcAddress(hWinspool, "SpoolerDevQueryPrintW");
221 pTable[3] = GetProcAddress(hWinspool, "SpoolerPrinterEvent");
222 pTable[4] = GetProcAddress(hWinspool, "DocumentPropertiesW");
223 pTable[5] = GetProcAddress(hWinspool, (LPSTR)212);
224 pTable[6] = GetProcAddress(hWinspool, (LPSTR)213);
225 pTable[7] = GetProcAddress(hWinspool, (LPSTR)214);
226 pTable[8] = GetProcAddress(hWinspool, (LPSTR)215);
227
228 // Verify that all calls succeeded.
229 for (i = 0; i < 9; i++)
230 if (!pTable[i])
231 return FALSE;
232
233 return TRUE;
234 }
235
236 BOOL WINAPI
237 SplIsUpgrade()
238 {
239 return FALSE;
240 }
241
242 BOOL WINAPI
243 SpoolerInit()
244 {
245 // Nothing to do here yet
246 SetLastError(ERROR_SUCCESS);
247 return TRUE;
248 }
249
250 BOOL WINAPI
251 BuildOtherNamesFromMachineName(LPVOID * ptr1, LPVOID * ptr2)
252 {
253 FIXME("(%p, %p) stub\n", ptr1, ptr2);
254
255 *ptr1 = NULL;
256 *ptr2 = NULL;
257 return FALSE;
258 }
259