spoolss/test: Add a test for BuildOtherNamesFromMachineName.
[wine] / dlls / spoolss / tests / spoolss.c
1 /*
2  * Unit test suite for the Spooler-Service helper DLL
3  *
4  * Copyright 2007 Detlef Riekenberg
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  *
20  */
21
22 #include <stdarg.h>
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winerror.h"
27 #include "wine/test.h"
28
29
30
31 /* ##### */
32
33 static HMODULE hwinspool;
34
35 static HMODULE hspl;
36 static BOOL   (WINAPI * pBuildOtherNamesFromMachineName)(LPWSTR **, LPDWORD);
37 static DWORD  (WINAPI * pSplInitializeWinSpoolDrv)(LPVOID *);
38
39 #define WINSPOOL_TABLESIZE   16
40
41 static LPVOID fn_spl[WINSPOOL_TABLESIZE];
42 static LPVOID fn_w2k[WINSPOOL_TABLESIZE];
43 static LPVOID fn_xp[WINSPOOL_TABLESIZE];
44
45 /* ########################### */
46
47 static LPCSTR load_functions(void)
48 {
49     LPCSTR  ptr;
50
51     ptr = "spoolss.dll";
52     hspl = LoadLibraryA(ptr);
53     if (!hspl) return ptr;
54
55     ptr = "BuildOtherNamesFromMachineName";
56     pBuildOtherNamesFromMachineName = (void *) GetProcAddress(hspl, ptr);
57     if (!pBuildOtherNamesFromMachineName) return ptr;
58
59     ptr = "SplInitializeWinSpoolDrv";
60     pSplInitializeWinSpoolDrv = (void *) GetProcAddress(hspl, ptr);
61     if (!pSplInitializeWinSpoolDrv) return ptr;
62
63
64     ptr = "winspool.drv";
65     hwinspool = LoadLibraryA(ptr);
66     if (!hwinspool) return ptr;
67
68     memset(fn_w2k, 0xff, sizeof(fn_w2k));
69     fn_w2k[0]  = (void *) GetProcAddress(hwinspool, "OpenPrinterW");
70     fn_w2k[1]  = (void *) GetProcAddress(hwinspool, "ClosePrinter");
71     fn_w2k[2]  = (void *) GetProcAddress(hwinspool, "SpoolerDevQueryPrintW");
72     fn_w2k[3]  = (void *) GetProcAddress(hwinspool, "SpoolerPrinterEvent");
73     fn_w2k[4]  = (void *) GetProcAddress(hwinspool, "DocumentPropertiesW");
74     fn_w2k[5]  = (void *) GetProcAddress(hwinspool, (LPSTR) 212);  /* LoadPrinterDriver */
75     fn_w2k[6]  = (void *) GetProcAddress(hwinspool, "SetDefaultPrinterW");
76     fn_w2k[7]  = (void *) GetProcAddress(hwinspool, "GetDefaultPrinterW");
77     fn_w2k[8]  = (void *) GetProcAddress(hwinspool, (LPSTR) 213);  /* RefCntLoadDriver */
78     fn_w2k[9]  = (void *) GetProcAddress(hwinspool, (LPSTR) 214);  /* RefCntUnloadDriver */
79     fn_w2k[10] = (void *) GetProcAddress(hwinspool, (LPSTR) 215);  /* ForceUnloadDriver */
80
81     memset(fn_xp,  0xff, sizeof(fn_xp));
82     fn_xp[0] = (void *) GetProcAddress(hwinspool, "OpenPrinterW");
83     fn_xp[1] = (void *) GetProcAddress(hwinspool, "ClosePrinter");
84     fn_xp[2] = (void *) GetProcAddress(hwinspool, "SpoolerDevQueryPrintW");
85     fn_xp[3] = (void *) GetProcAddress(hwinspool, "SpoolerPrinterEvent");
86     fn_xp[4] = (void *) GetProcAddress(hwinspool, "DocumentPropertiesW");
87     fn_xp[5] = (void *) GetProcAddress(hwinspool, (LPSTR) 212);  /* LoadPrinterDriver */
88     fn_xp[6] = (void *) GetProcAddress(hwinspool, (LPSTR) 213);  /* RefCntLoadDriver */
89     fn_xp[7] = (void *) GetProcAddress(hwinspool, (LPSTR) 214);  /* RefCntUnloadDriver */
90     fn_xp[8] = (void *) GetProcAddress(hwinspool, (LPSTR) 215);  /* ForceUnloadDriver */
91
92     return NULL;
93
94 }
95
96 /* ########################### */
97
98 static void test_BuildOtherNamesFromMachineName(void)
99 {
100     LPWSTR *buffers;
101     DWORD   numentries;
102     DWORD   res;
103
104     buffers = NULL;
105     numentries = 0;
106
107     SetLastError(0xdeadbeef);
108     res = pBuildOtherNamesFromMachineName(&buffers, &numentries);
109
110     /* An array with 3 stringpointer is returned:
111       entry_#0: "" (empty String)
112       entry_#1: <hostname> (this is the same as the computernam)
113       entry_#2: <ip-address> (string with the ip-address of <hostname>)
114     */
115     todo_wine
116     ok( res && (buffers != NULL) && (numentries == 3) && (buffers[0] != NULL) && (buffers[0][0] == '\0'),
117         "got %u with %u and %p,%u (%p:%d)\n", res, GetLastError(), buffers, numentries,
118         ((numentries > 0) && buffers) ? buffers[0] : NULL,
119         ((numentries > 0) && buffers && buffers[0]) ? lstrlenW(buffers[0]) : -1);
120
121 }
122
123 /* ########################### */
124
125 static void test_SplInitializeWinSpoolDrv(VOID)
126 {
127     DWORD   res;
128     LONG    id;
129     BOOL    is_xp;
130
131     memset(fn_spl, 0xff, sizeof(fn_spl));
132     SetLastError(0xdeadbeef);
133     res = pSplInitializeWinSpoolDrv(fn_spl);
134     ok(res, "got %u with %u (expected '!= 0')\n", res, GetLastError());
135
136     /* functions 0 to 5 are the same with "spoolss.dll" from w2k and xp */
137     is_xp = (fn_spl[6] == fn_xp[6]);
138     id = 0;
139     while (id < WINSPOOL_TABLESIZE) {
140         ok( fn_spl[id] == (is_xp ? fn_xp[id] : fn_w2k[id]),
141             "(#%02u) spoolss: %p,  xp: %p,  w2k: %p\n", id, fn_spl[id], fn_xp[id], fn_w2k[id]);
142         id++;
143     }
144 }
145
146 /* ########################### */
147
148 START_TEST(spoolss)
149 {
150     LPCSTR ptr;
151
152     /* spoolss.dll does not exist on win9x */
153     ptr = load_functions();
154     if (ptr) {
155         skip("%s not found\n", ptr);
156         return;
157     }
158
159     test_BuildOtherNamesFromMachineName();
160     test_SplInitializeWinSpoolDrv();
161
162 }