kernel32/tests: Remove a redundant check for ERROR_CALL_NOT_IMPLEMENTED.
[wine/hacks.git] / dlls / kernel32 / tests / pipe.c
blob3adf417026f82937f5b5ce81018863ea1a99b9af
1 /*
2 * Unit tests for named pipe functions in Wine
4 * Copyright (c) 2002 Dan Kegel
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.
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.
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
21 #include <assert.h>
22 #include <stdarg.h>
23 #include <stdio.h>
25 #include <windef.h>
26 #include <winbase.h>
27 #include <winsock.h>
28 #include <wtypes.h>
29 #include <winerror.h>
31 #include "wine/test.h"
33 #define PIPENAME "\\\\.\\PiPe\\tests_pipe.c"
35 #define NB_SERVER_LOOPS 8
37 static HANDLE alarm_event;
38 static BOOL (WINAPI *pDuplicateTokenEx)(HANDLE,DWORD,LPSECURITY_ATTRIBUTES,
39 SECURITY_IMPERSONATION_LEVEL,TOKEN_TYPE,PHANDLE);
42 static void test_CreateNamedPipe(int pipemode)
44 HANDLE hnp;
45 HANDLE hFile;
46 static const char obuf[] = "Bit Bucket";
47 static const char obuf2[] = "More bits";
48 char ibuf[32], *pbuf;
49 DWORD written;
50 DWORD readden;
51 DWORD avail;
52 DWORD lpmode;
54 if (pipemode == PIPE_TYPE_BYTE)
55 trace("test_CreateNamedPipe starting in byte mode\n");
56 else
57 trace("test_CreateNamedPipe starting in message mode\n");
58 /* Bad parameter checks */
59 hnp = CreateNamedPipe("not a named pipe", PIPE_ACCESS_DUPLEX, pipemode | PIPE_WAIT,
60 /* nMaxInstances */ 1,
61 /* nOutBufSize */ 1024,
62 /* nInBufSize */ 1024,
63 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
64 /* lpSecurityAttrib */ NULL);
65 ok(hnp == INVALID_HANDLE_VALUE && GetLastError() == ERROR_INVALID_NAME,
66 "CreateNamedPipe should fail if name doesn't start with \\\\.\\pipe\n");
68 hnp = CreateNamedPipe(NULL,
69 PIPE_ACCESS_DUPLEX, pipemode | PIPE_WAIT,
70 1, 1024, 1024, NMPWAIT_USE_DEFAULT_WAIT, NULL);
71 ok(hnp == INVALID_HANDLE_VALUE && GetLastError() == ERROR_PATH_NOT_FOUND,
72 "CreateNamedPipe should fail if name is NULL\n");
74 hFile = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
75 ok(hFile == INVALID_HANDLE_VALUE
76 && GetLastError() == ERROR_FILE_NOT_FOUND,
77 "connecting to nonexistent named pipe should fail with ERROR_FILE_NOT_FOUND\n");
79 /* Functional checks */
81 hnp = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, pipemode | PIPE_WAIT,
82 /* nMaxInstances */ 1,
83 /* nOutBufSize */ 1024,
84 /* nInBufSize */ 1024,
85 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
86 /* lpSecurityAttrib */ NULL);
87 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
89 ok(WaitNamedPipeA(PIPENAME, 2000), "WaitNamedPipe failed (%d)\n", GetLastError());
91 hFile = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
92 ok(hFile != INVALID_HANDLE_VALUE, "CreateFile failed (%d)\n", GetLastError());
94 ok(!WaitNamedPipeA(PIPENAME, 1000), "WaitNamedPipe succeeded\n");
95 ok(GetLastError() == ERROR_SEM_TIMEOUT, "wrong error %u\n", GetLastError());
97 /* don't try to do i/o if one side couldn't be opened, as it hangs */
98 if (hFile != INVALID_HANDLE_VALUE) {
99 HANDLE hFile2;
101 /* Make sure we can read and write a few bytes in both directions */
102 memset(ibuf, 0, sizeof(ibuf));
103 ok(WriteFile(hnp, obuf, sizeof(obuf), &written, NULL), "WriteFile\n");
104 ok(written == sizeof(obuf), "write file len 1\n");
105 ok(PeekNamedPipe(hFile, NULL, 0, NULL, &readden, NULL), "Peek\n");
106 ok(readden == sizeof(obuf), "peek 1 got %d bytes\n", readden);
107 ok(ReadFile(hFile, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
108 ok(readden == sizeof(obuf), "read 1 got %d bytes\n", readden);
109 ok(memcmp(obuf, ibuf, written) == 0, "content 1 check\n");
111 memset(ibuf, 0, sizeof(ibuf));
112 ok(WriteFile(hFile, obuf2, sizeof(obuf2), &written, NULL), "WriteFile\n");
113 ok(written == sizeof(obuf2), "write file len 2\n");
114 ok(PeekNamedPipe(hnp, NULL, 0, NULL, &readden, NULL), "Peek\n");
115 ok(readden == sizeof(obuf2), "peek 2 got %d bytes\n", readden);
116 ok(PeekNamedPipe(hnp, (LPVOID)1, 0, NULL, &readden, NULL), "Peek\n");
117 ok(readden == sizeof(obuf2), "peek 2 got %d bytes\n", readden);
118 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
119 ok(readden == sizeof(obuf2), "read 2 got %d bytes\n", readden);
120 ok(memcmp(obuf2, ibuf, written) == 0, "content 2 check\n");
122 /* Test reading of multiple writes */
123 memset(ibuf, 0, sizeof(ibuf));
124 ok(WriteFile(hnp, obuf, sizeof(obuf), &written, NULL), "WriteFile3a\n");
125 ok(written == sizeof(obuf), "write file len 3a\n");
126 ok(WriteFile(hnp, obuf2, sizeof(obuf2), &written, NULL), " WriteFile3b\n");
127 ok(written == sizeof(obuf2), "write file len 3b\n");
128 ok(PeekNamedPipe(hFile, ibuf, sizeof(ibuf), &readden, &avail, NULL), "Peek3\n");
129 if (pipemode == PIPE_TYPE_BYTE) {
130 if (readden != sizeof(obuf)) /* Linux only returns the first message */
131 ok(readden == sizeof(obuf) + sizeof(obuf2), "peek3 got %d bytes\n", readden);
132 else
133 todo_wine ok(readden == sizeof(obuf) + sizeof(obuf2), "peek3 got %d bytes\n", readden);
135 else
137 if (readden != sizeof(obuf) + sizeof(obuf2)) /* MacOS returns both messages */
138 ok(readden == sizeof(obuf), "peek3 got %d bytes\n", readden);
139 else
140 todo_wine ok(readden == sizeof(obuf), "peek3 got %d bytes\n", readden);
142 if (avail != sizeof(obuf)) /* older Linux kernels only return the first write here */
143 ok(avail == sizeof(obuf) + sizeof(obuf2), "peek3 got %d bytes available\n", avail);
144 pbuf = ibuf;
145 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "pipe content 3a check\n");
146 if (pipemode == PIPE_TYPE_BYTE && readden >= sizeof(obuf)+sizeof(obuf2)) {
147 pbuf += sizeof(obuf);
148 ok(memcmp(obuf2, pbuf, sizeof(obuf2)) == 0, "pipe content 3b check\n");
150 ok(ReadFile(hFile, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
151 ok(readden == sizeof(obuf) + sizeof(obuf2), "read 3 got %d bytes\n", readden);
152 pbuf = ibuf;
153 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 3a check\n");
154 pbuf += sizeof(obuf);
155 ok(memcmp(obuf2, pbuf, sizeof(obuf2)) == 0, "content 3b check\n");
157 /* Multiple writes in the reverse direction */
158 memset(ibuf, 0, sizeof(ibuf));
159 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL), "WriteFile4a\n");
160 ok(written == sizeof(obuf), "write file len 4a\n");
161 ok(WriteFile(hFile, obuf2, sizeof(obuf2), &written, NULL), " WriteFile4b\n");
162 ok(written == sizeof(obuf2), "write file len 4b\n");
163 ok(PeekNamedPipe(hnp, ibuf, sizeof(ibuf), &readden, &avail, NULL), "Peek4\n");
164 if (pipemode == PIPE_TYPE_BYTE) {
165 if (readden != sizeof(obuf)) /* Linux only returns the first message */
166 /* should return all 23 bytes */
167 ok(readden == sizeof(obuf) + sizeof(obuf2), "peek4 got %d bytes\n", readden);
168 else
169 todo_wine ok(readden == sizeof(obuf) + sizeof(obuf2), "peek4 got %d bytes\n", readden);
171 else
173 if (readden != sizeof(obuf) + sizeof(obuf2)) /* MacOS returns both messages */
174 ok(readden == sizeof(obuf), "peek4 got %d bytes\n", readden);
175 else
176 todo_wine ok(readden == sizeof(obuf), "peek4 got %d bytes\n", readden);
178 if (avail != sizeof(obuf)) /* older Linux kernels only return the first write here */
179 ok(avail == sizeof(obuf) + sizeof(obuf2), "peek4 got %d bytes available\n", avail);
180 pbuf = ibuf;
181 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "pipe content 4a check\n");
182 if (pipemode == PIPE_TYPE_BYTE && readden >= sizeof(obuf)+sizeof(obuf2)) {
183 pbuf += sizeof(obuf);
184 ok(memcmp(obuf2, pbuf, sizeof(obuf2)) == 0, "pipe content 4b check\n");
186 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
187 if (pipemode == PIPE_TYPE_BYTE) {
188 ok(readden == sizeof(obuf) + sizeof(obuf2), "read 4 got %d bytes\n", readden);
190 else {
191 todo_wine {
192 ok(readden == sizeof(obuf), "read 4 got %d bytes\n", readden);
195 pbuf = ibuf;
196 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 4a check\n");
197 if (pipemode == PIPE_TYPE_BYTE) {
198 pbuf += sizeof(obuf);
199 ok(memcmp(obuf2, pbuf, sizeof(obuf2)) == 0, "content 4b check\n");
202 /* Test reading of multiple writes after a mode change
203 (CreateFile always creates a byte mode pipe) */
204 lpmode = PIPE_READMODE_MESSAGE;
205 if (pipemode == PIPE_TYPE_BYTE) {
206 /* trying to change the client end of a byte pipe to message mode should fail */
207 ok(!SetNamedPipeHandleState(hFile, &lpmode, NULL, NULL), "Change mode\n");
209 else {
210 todo_wine {
211 ok(SetNamedPipeHandleState(hFile, &lpmode, NULL, NULL), "Change mode\n");
214 memset(ibuf, 0, sizeof(ibuf));
215 ok(WriteFile(hnp, obuf, sizeof(obuf), &written, NULL), "WriteFile5a\n");
216 ok(written == sizeof(obuf), "write file len 3a\n");
217 ok(WriteFile(hnp, obuf2, sizeof(obuf2), &written, NULL), " WriteFile5b\n");
218 ok(written == sizeof(obuf2), "write file len 3b\n");
219 ok(PeekNamedPipe(hFile, ibuf, sizeof(ibuf), &readden, &avail, NULL), "Peek5\n");
220 if (readden != sizeof(obuf) + sizeof(obuf2)) /* MacOS returns both writes */
221 ok(readden == sizeof(obuf), "peek5 got %d bytes\n", readden);
222 else
223 todo_wine ok(readden == sizeof(obuf), "peek5 got %d bytes\n", readden);
224 if (avail != sizeof(obuf)) /* older Linux kernels only return the first write here */
225 ok(avail == sizeof(obuf) + sizeof(obuf2), "peek5 got %d bytes available\n", avail);
226 else
227 todo_wine ok(avail == sizeof(obuf) + sizeof(obuf2), "peek5 got %d bytes available\n", avail);
228 pbuf = ibuf;
229 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 5a check\n");
230 ok(ReadFile(hFile, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
231 todo_wine {
232 ok(readden == sizeof(obuf), "read 5 got %d bytes\n", readden);
234 pbuf = ibuf;
235 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 5a check\n");
237 /* Multiple writes in the reverse direction */
238 /* the write of obuf2 from write4 should still be in the buffer */
239 ok(PeekNamedPipe(hnp, ibuf, sizeof(ibuf), &readden, &avail, NULL), "Peek6a\n");
240 todo_wine {
241 ok(readden == sizeof(obuf2), "peek6a got %d bytes\n", readden);
242 ok(avail == sizeof(obuf2), "peek6a got %d bytes available\n", avail);
244 if (avail > 0) {
245 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
246 ok(readden == sizeof(obuf2), "read 6a got %d bytes\n", readden);
247 pbuf = ibuf;
248 ok(memcmp(obuf2, pbuf, sizeof(obuf2)) == 0, "content 6a check\n");
250 memset(ibuf, 0, sizeof(ibuf));
251 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL), "WriteFile6a\n");
252 ok(written == sizeof(obuf), "write file len 6a\n");
253 ok(WriteFile(hFile, obuf2, sizeof(obuf2), &written, NULL), " WriteFile6b\n");
254 ok(written == sizeof(obuf2), "write file len 6b\n");
255 ok(PeekNamedPipe(hnp, ibuf, sizeof(ibuf), &readden, &avail, NULL), "Peek6\n");
256 if (readden != sizeof(obuf) + sizeof(obuf2)) /* MacOS returns both writes */
257 ok(readden == sizeof(obuf), "peek6 got %d bytes\n", readden);
258 else
259 todo_wine ok(readden == sizeof(obuf), "peek6 got %d bytes\n", readden);
260 if (avail != sizeof(obuf)) /* older Linux kernels only return the first write here */
261 ok(avail == sizeof(obuf) + sizeof(obuf2), "peek6b got %d bytes available\n", avail);
262 pbuf = ibuf;
263 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 6a check\n");
264 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
265 todo_wine {
266 ok(readden == sizeof(obuf), "read 6b got %d bytes\n", readden);
268 pbuf = ibuf;
269 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 6a check\n");
272 /* Picky conformance tests */
274 /* Verify that you can't connect to pipe again
275 * until server calls DisconnectNamedPipe+ConnectNamedPipe
276 * or creates a new pipe
277 * case 1: other client not yet closed
279 hFile2 = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
280 ok(hFile2 == INVALID_HANDLE_VALUE,
281 "connecting to named pipe after other client closes but before DisconnectNamedPipe should fail\n");
282 ok(GetLastError() == ERROR_PIPE_BUSY,
283 "connecting to named pipe before other client closes should fail with ERROR_PIPE_BUSY\n");
285 ok(CloseHandle(hFile), "CloseHandle\n");
287 /* case 2: other client already closed */
288 hFile = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
289 ok(hFile == INVALID_HANDLE_VALUE,
290 "connecting to named pipe after other client closes but before DisconnectNamedPipe should fail\n");
291 ok(GetLastError() == ERROR_PIPE_BUSY,
292 "connecting to named pipe after other client closes but before DisconnectNamedPipe should fail with ERROR_PIPE_BUSY\n");
294 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe\n");
296 /* case 3: server has called DisconnectNamedPipe but not ConnectNamed Pipe */
297 hFile = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
298 ok(hFile == INVALID_HANDLE_VALUE,
299 "connecting to named pipe after other client closes but before DisconnectNamedPipe should fail\n");
300 ok(GetLastError() == ERROR_PIPE_BUSY,
301 "connecting to named pipe after other client closes but before ConnectNamedPipe should fail with ERROR_PIPE_BUSY\n");
303 /* to be complete, we'd call ConnectNamedPipe here and loop,
304 * but by default that's blocking, so we'd either have
305 * to turn on the uncommon nonblocking mode, or
306 * use another thread.
310 ok(CloseHandle(hnp), "CloseHandle\n");
312 trace("test_CreateNamedPipe returning\n");
315 static void test_CreateNamedPipe_instances_must_match(void)
317 HANDLE hnp, hnp2;
319 /* Check no mismatch */
320 hnp = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
321 /* nMaxInstances */ 2,
322 /* nOutBufSize */ 1024,
323 /* nInBufSize */ 1024,
324 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
325 /* lpSecurityAttrib */ NULL);
326 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
328 hnp2 = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
329 /* nMaxInstances */ 2,
330 /* nOutBufSize */ 1024,
331 /* nInBufSize */ 1024,
332 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
333 /* lpSecurityAttrib */ NULL);
334 ok(hnp2 != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
336 ok(CloseHandle(hnp), "CloseHandle\n");
337 ok(CloseHandle(hnp2), "CloseHandle\n");
339 /* Check nMaxInstances */
340 hnp = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
341 /* nMaxInstances */ 1,
342 /* nOutBufSize */ 1024,
343 /* nInBufSize */ 1024,
344 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
345 /* lpSecurityAttrib */ NULL);
346 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
348 hnp2 = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
349 /* nMaxInstances */ 1,
350 /* nOutBufSize */ 1024,
351 /* nInBufSize */ 1024,
352 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
353 /* lpSecurityAttrib */ NULL);
354 ok(hnp2 == INVALID_HANDLE_VALUE
355 && GetLastError() == ERROR_PIPE_BUSY, "nMaxInstances not obeyed\n");
357 ok(CloseHandle(hnp), "CloseHandle\n");
359 /* Check PIPE_ACCESS_* */
360 hnp = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
361 /* nMaxInstances */ 2,
362 /* nOutBufSize */ 1024,
363 /* nInBufSize */ 1024,
364 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
365 /* lpSecurityAttrib */ NULL);
366 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
368 hnp2 = CreateNamedPipe(PIPENAME, PIPE_ACCESS_INBOUND, PIPE_TYPE_BYTE | PIPE_WAIT,
369 /* nMaxInstances */ 1,
370 /* nOutBufSize */ 1024,
371 /* nInBufSize */ 1024,
372 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
373 /* lpSecurityAttrib */ NULL);
374 ok(hnp2 == INVALID_HANDLE_VALUE
375 && GetLastError() == ERROR_ACCESS_DENIED, "PIPE_ACCESS_* mismatch allowed\n");
377 ok(CloseHandle(hnp), "CloseHandle\n");
379 /* etc, etc */
382 /** implementation of alarm() */
383 static DWORD CALLBACK alarmThreadMain(LPVOID arg)
385 DWORD_PTR timeout = (DWORD_PTR) arg;
386 trace("alarmThreadMain\n");
387 if (WaitForSingleObject( alarm_event, timeout ) == WAIT_TIMEOUT)
389 ok(FALSE, "alarm\n");
390 ExitProcess(1);
392 return 1;
395 HANDLE hnp = INVALID_HANDLE_VALUE;
397 /** Trivial byte echo server - disconnects after each session */
398 static DWORD CALLBACK serverThreadMain1(LPVOID arg)
400 int i;
402 trace("serverThreadMain1 start\n");
403 /* Set up a simple echo server */
404 hnp = CreateNamedPipe(PIPENAME "serverThreadMain1", PIPE_ACCESS_DUPLEX,
405 PIPE_TYPE_BYTE | PIPE_WAIT,
406 /* nMaxInstances */ 1,
407 /* nOutBufSize */ 1024,
408 /* nInBufSize */ 1024,
409 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
410 /* lpSecurityAttrib */ NULL);
412 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
413 for (i = 0; i < NB_SERVER_LOOPS; i++) {
414 char buf[512];
415 DWORD written;
416 DWORD readden;
417 DWORD success;
419 /* Wait for client to connect */
420 trace("Server calling ConnectNamedPipe...\n");
421 ok(ConnectNamedPipe(hnp, NULL)
422 || GetLastError() == ERROR_PIPE_CONNECTED, "ConnectNamedPipe\n");
423 trace("ConnectNamedPipe returned.\n");
425 /* Echo bytes once */
426 memset(buf, 0, sizeof(buf));
428 trace("Server reading...\n");
429 success = ReadFile(hnp, buf, sizeof(buf), &readden, NULL);
430 trace("Server done reading.\n");
431 ok(success, "ReadFile\n");
432 ok(readden, "short read\n");
434 trace("Server writing...\n");
435 ok(WriteFile(hnp, buf, readden, &written, NULL), "WriteFile\n");
436 trace("Server done writing.\n");
437 ok(written == readden, "write file len\n");
439 /* finish this connection, wait for next one */
440 ok(FlushFileBuffers(hnp), "FlushFileBuffers\n");
441 trace("Server done flushing.\n");
442 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe\n");
443 trace("Server done disconnecting.\n");
445 return 0;
448 /** Trivial byte echo server - closes after each connection */
449 static DWORD CALLBACK serverThreadMain2(LPVOID arg)
451 int i;
452 HANDLE hnpNext = 0;
454 trace("serverThreadMain2\n");
455 /* Set up a simple echo server */
456 hnp = CreateNamedPipe(PIPENAME "serverThreadMain2", PIPE_ACCESS_DUPLEX,
457 PIPE_TYPE_BYTE | PIPE_WAIT,
458 /* nMaxInstances */ 2,
459 /* nOutBufSize */ 1024,
460 /* nInBufSize */ 1024,
461 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
462 /* lpSecurityAttrib */ NULL);
463 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
465 for (i = 0; i < NB_SERVER_LOOPS; i++) {
466 char buf[512];
467 DWORD written;
468 DWORD readden;
469 DWORD success;
471 /* Wait for client to connect */
472 trace("Server calling ConnectNamedPipe...\n");
473 ok(ConnectNamedPipe(hnp, NULL)
474 || GetLastError() == ERROR_PIPE_CONNECTED, "ConnectNamedPipe\n");
475 trace("ConnectNamedPipe returned.\n");
477 /* Echo bytes once */
478 memset(buf, 0, sizeof(buf));
480 trace("Server reading...\n");
481 success = ReadFile(hnp, buf, sizeof(buf), &readden, NULL);
482 trace("Server done reading.\n");
483 ok(success, "ReadFile\n");
485 trace("Server writing...\n");
486 ok(WriteFile(hnp, buf, readden, &written, NULL), "WriteFile\n");
487 trace("Server done writing.\n");
488 ok(written == readden, "write file len\n");
490 /* finish this connection, wait for next one */
491 ok(FlushFileBuffers(hnp), "FlushFileBuffers\n");
492 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe\n");
494 /* Set up next echo server */
495 hnpNext =
496 CreateNamedPipe(PIPENAME "serverThreadMain2", PIPE_ACCESS_DUPLEX,
497 PIPE_TYPE_BYTE | PIPE_WAIT,
498 /* nMaxInstances */ 2,
499 /* nOutBufSize */ 1024,
500 /* nInBufSize */ 1024,
501 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
502 /* lpSecurityAttrib */ NULL);
504 ok(hnpNext != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
506 ok(CloseHandle(hnp), "CloseHandle\n");
507 hnp = hnpNext;
509 return 0;
512 /** Trivial byte echo server - uses overlapped named pipe calls */
513 static DWORD CALLBACK serverThreadMain3(LPVOID arg)
515 int i;
516 HANDLE hEvent;
518 trace("serverThreadMain3\n");
519 /* Set up a simple echo server */
520 hnp = CreateNamedPipe(PIPENAME "serverThreadMain3", PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
521 PIPE_TYPE_BYTE | PIPE_WAIT,
522 /* nMaxInstances */ 1,
523 /* nOutBufSize */ 1024,
524 /* nInBufSize */ 1024,
525 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
526 /* lpSecurityAttrib */ NULL);
527 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
529 hEvent = CreateEvent(NULL, /* security attribute */
530 TRUE, /* manual reset event */
531 FALSE, /* initial state */
532 NULL); /* name */
533 ok(hEvent != NULL, "CreateEvent\n");
535 for (i = 0; i < NB_SERVER_LOOPS; i++) {
536 char buf[512];
537 DWORD written;
538 DWORD readden;
539 DWORD dummy;
540 DWORD success;
541 OVERLAPPED oOverlap;
542 int letWFSOEwait = (i & 2);
543 int letGORwait = (i & 1);
544 DWORD err;
546 memset(&oOverlap, 0, sizeof(oOverlap));
547 oOverlap.hEvent = hEvent;
549 /* Wait for client to connect */
550 trace("Server calling overlapped ConnectNamedPipe...\n");
551 success = ConnectNamedPipe(hnp, &oOverlap);
552 err = GetLastError();
553 ok(success || err == ERROR_IO_PENDING
554 || err == ERROR_PIPE_CONNECTED, "overlapped ConnectNamedPipe\n");
555 trace("overlapped ConnectNamedPipe returned.\n");
556 if (!success && (err == ERROR_IO_PENDING) && letWFSOEwait)
557 ok(WaitForSingleObjectEx(hEvent, INFINITE, TRUE) == 0, "wait ConnectNamedPipe\n");
558 success = GetOverlappedResult(hnp, &oOverlap, &dummy, letGORwait);
559 if (!letGORwait && !letWFSOEwait && !success) {
560 ok(GetLastError() == ERROR_IO_INCOMPLETE, "GetOverlappedResult\n");
561 success = GetOverlappedResult(hnp, &oOverlap, &dummy, TRUE);
563 ok(success, "GetOverlappedResult ConnectNamedPipe\n");
564 trace("overlapped ConnectNamedPipe operation complete.\n");
566 /* Echo bytes once */
567 memset(buf, 0, sizeof(buf));
569 trace("Server reading...\n");
570 success = ReadFile(hnp, buf, sizeof(buf), NULL, &oOverlap);
571 trace("Server ReadFile returned...\n");
572 err = GetLastError();
573 ok(success || err == ERROR_IO_PENDING, "overlapped ReadFile\n");
574 trace("overlapped ReadFile returned.\n");
575 if (!success && (err == ERROR_IO_PENDING) && letWFSOEwait)
576 ok(WaitForSingleObjectEx(hEvent, INFINITE, TRUE) == 0, "wait ReadFile\n");
577 success = GetOverlappedResult(hnp, &oOverlap, &readden, letGORwait);
578 if (!letGORwait && !letWFSOEwait && !success) {
579 ok(GetLastError() == ERROR_IO_INCOMPLETE, "GetOverlappedResult\n");
580 success = GetOverlappedResult(hnp, &oOverlap, &readden, TRUE);
582 trace("Server done reading.\n");
583 ok(success, "overlapped ReadFile\n");
585 trace("Server writing...\n");
586 success = WriteFile(hnp, buf, readden, NULL, &oOverlap);
587 trace("Server WriteFile returned...\n");
588 err = GetLastError();
589 ok(success || err == ERROR_IO_PENDING, "overlapped WriteFile\n");
590 trace("overlapped WriteFile returned.\n");
591 if (!success && (err == ERROR_IO_PENDING) && letWFSOEwait)
592 ok(WaitForSingleObjectEx(hEvent, INFINITE, TRUE) == 0, "wait WriteFile\n");
593 success = GetOverlappedResult(hnp, &oOverlap, &written, letGORwait);
594 if (!letGORwait && !letWFSOEwait && !success) {
595 ok(GetLastError() == ERROR_IO_INCOMPLETE, "GetOverlappedResult\n");
596 success = GetOverlappedResult(hnp, &oOverlap, &written, TRUE);
598 trace("Server done writing.\n");
599 ok(success, "overlapped WriteFile\n");
600 ok(written == readden, "write file len\n");
602 /* finish this connection, wait for next one */
603 ok(FlushFileBuffers(hnp), "FlushFileBuffers\n");
604 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe\n");
606 return 0;
609 static void exercizeServer(const char *pipename, HANDLE serverThread)
611 int i;
613 trace("exercizeServer starting\n");
614 for (i = 0; i < NB_SERVER_LOOPS; i++) {
615 HANDLE hFile=INVALID_HANDLE_VALUE;
616 static const char obuf[] = "Bit Bucket";
617 char ibuf[32];
618 DWORD written;
619 DWORD readden;
620 int loop;
622 for (loop = 0; loop < 3; loop++) {
623 DWORD err;
624 trace("Client connecting...\n");
625 /* Connect to the server */
626 hFile = CreateFileA(pipename, GENERIC_READ | GENERIC_WRITE, 0,
627 NULL, OPEN_EXISTING, 0, 0);
628 if (hFile != INVALID_HANDLE_VALUE)
629 break;
630 err = GetLastError();
631 if (loop == 0)
632 ok(err == ERROR_PIPE_BUSY || err == ERROR_FILE_NOT_FOUND, "connecting to pipe\n");
633 else
634 ok(err == ERROR_PIPE_BUSY, "connecting to pipe\n");
635 trace("connect failed, retrying\n");
636 Sleep(200);
638 ok(hFile != INVALID_HANDLE_VALUE, "client opening named pipe\n");
640 /* Make sure it can echo */
641 memset(ibuf, 0, sizeof(ibuf));
642 trace("Client writing...\n");
643 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL), "WriteFile to client end of pipe\n");
644 ok(written == sizeof(obuf), "write file len\n");
645 trace("Client reading...\n");
646 ok(ReadFile(hFile, ibuf, sizeof(obuf), &readden, NULL), "ReadFile from client end of pipe\n");
647 ok(readden == sizeof(obuf), "read file len\n");
648 ok(memcmp(obuf, ibuf, written) == 0, "content check\n");
650 trace("Client closing...\n");
651 ok(CloseHandle(hFile), "CloseHandle\n");
654 ok(WaitForSingleObject(serverThread,INFINITE) == WAIT_OBJECT_0, "WaitForSingleObject\n");
655 CloseHandle(hnp);
656 trace("exercizeServer returning\n");
659 static void test_NamedPipe_2(void)
661 HANDLE serverThread;
662 DWORD serverThreadId;
663 HANDLE alarmThread;
664 DWORD alarmThreadId;
666 trace("test_NamedPipe_2 starting\n");
667 /* Set up a ten second timeout */
668 alarm_event = CreateEvent( NULL, TRUE, FALSE, NULL );
669 alarmThread = CreateThread(NULL, 0, alarmThreadMain, (void *) 10000, 0, &alarmThreadId);
671 /* The servers we're about to exercize do try to clean up carefully,
672 * but to reduce the change of a test failure due to a pipe handle
673 * leak in the test code, we'll use a different pipe name for each server.
676 /* Try server #1 */
677 serverThread = CreateThread(NULL, 0, serverThreadMain1, (void *)8, 0, &serverThreadId);
678 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
679 exercizeServer(PIPENAME "serverThreadMain1", serverThread);
681 /* Try server #2 */
682 serverThread = CreateThread(NULL, 0, serverThreadMain2, 0, 0, &serverThreadId);
683 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
684 exercizeServer(PIPENAME "serverThreadMain2", serverThread);
686 if( 0 ) /* overlapped pipe server doesn't work yet - it randomly fails */
688 /* Try server #3 */
689 serverThread = CreateThread(NULL, 0, serverThreadMain3, 0, 0, &serverThreadId);
690 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
691 exercizeServer(PIPENAME "serverThreadMain3", serverThread);
694 ok(SetEvent( alarm_event ), "SetEvent\n");
695 CloseHandle( alarm_event );
696 trace("test_NamedPipe_2 returning\n");
699 static int test_DisconnectNamedPipe(void)
701 HANDLE hnp;
702 HANDLE hFile;
703 static const char obuf[] = "Bit Bucket";
704 char ibuf[32];
705 DWORD written;
706 DWORD readden;
708 SetLastError(0xdeadbeef);
709 hnp = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
710 /* nMaxInstances */ 1,
711 /* nOutBufSize */ 1024,
712 /* nInBufSize */ 1024,
713 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
714 /* lpSecurityAttrib */ NULL);
715 if ((hnp == INVALID_HANDLE_VALUE /* Win98 */ || !hnp /* Win95 */)
716 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) {
718 win_skip("Named pipes are not implemented\n");
719 return 1;
722 ok(WriteFile(hnp, obuf, sizeof(obuf), &written, NULL) == 0
723 && GetLastError() == ERROR_PIPE_LISTENING, "WriteFile to not-yet-connected pipe\n");
724 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL) == 0
725 && GetLastError() == ERROR_PIPE_LISTENING, "ReadFile from not-yet-connected pipe\n");
727 hFile = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
728 ok(hFile != INVALID_HANDLE_VALUE, "CreateFile failed\n");
730 /* don't try to do i/o if one side couldn't be opened, as it hangs */
731 if (hFile != INVALID_HANDLE_VALUE) {
733 /* see what happens if server calls DisconnectNamedPipe
734 * when there are bytes in the pipe
737 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL), "WriteFile\n");
738 ok(written == sizeof(obuf), "write file len\n");
739 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe while messages waiting\n");
740 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL) == 0
741 && GetLastError() == ERROR_PIPE_NOT_CONNECTED, "WriteFile to disconnected pipe\n");
742 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL) == 0
743 && GetLastError() == ERROR_PIPE_NOT_CONNECTED,
744 "ReadFile from disconnected pipe with bytes waiting\n");
745 ok(!DisconnectNamedPipe(hnp) && GetLastError() == ERROR_PIPE_NOT_CONNECTED,
746 "DisconnectNamedPipe worked twice\n");
747 ok(CloseHandle(hFile), "CloseHandle\n");
750 ok(CloseHandle(hnp), "CloseHandle\n");
752 return 0;
754 static void test_CreatePipe(void)
756 SECURITY_ATTRIBUTES pipe_attr;
757 HANDLE piperead, pipewrite;
758 DWORD written;
759 DWORD read;
760 DWORD i, size;
761 BYTE *buffer;
762 char readbuf[32];
764 pipe_attr.nLength = sizeof(SECURITY_ATTRIBUTES);
765 pipe_attr.bInheritHandle = TRUE;
766 pipe_attr.lpSecurityDescriptor = NULL;
767 ok(CreatePipe(&piperead, &pipewrite, &pipe_attr, 0) != 0, "CreatePipe failed\n");
768 ok(WriteFile(pipewrite,PIPENAME,sizeof(PIPENAME), &written, NULL), "Write to anonymous pipe failed\n");
769 ok(written == sizeof(PIPENAME), "Write to anonymous pipe wrote %d bytes\n", written);
770 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL), "Read from non empty pipe failed\n");
771 ok(read == sizeof(PIPENAME), "Read from anonymous pipe got %d bytes\n", read);
772 ok(CloseHandle(pipewrite), "CloseHandle for the write pipe failed\n");
773 ok(CloseHandle(piperead), "CloseHandle for the read pipe failed\n");
775 /* Now write another chunk*/
776 ok(CreatePipe(&piperead, &pipewrite, &pipe_attr, 0) != 0, "CreatePipe failed\n");
777 ok(WriteFile(pipewrite,PIPENAME,sizeof(PIPENAME), &written, NULL), "Write to anonymous pipe failed\n");
778 ok(written == sizeof(PIPENAME), "Write to anonymous pipe wrote %d bytes\n", written);
779 /* and close the write end, read should still succeed*/
780 ok(CloseHandle(pipewrite), "CloseHandle for the Write Pipe failed\n");
781 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL), "Read from broken pipe withe with pending data failed\n");
782 ok(read == sizeof(PIPENAME), "Read from anonymous pipe got %d bytes\n", read);
783 /* But now we need to get informed that the pipe is closed */
784 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL) == 0, "Broken pipe not detected\n");
785 ok(CloseHandle(piperead), "CloseHandle for the read pipe failed\n");
787 /* Try bigger chunks */
788 size = 32768;
789 buffer = HeapAlloc( GetProcessHeap(), 0, size );
790 for (i = 0; i < size; i++) buffer[i] = i;
791 ok(CreatePipe(&piperead, &pipewrite, &pipe_attr, (size + 24)) != 0, "CreatePipe failed\n");
792 ok(WriteFile(pipewrite, buffer, size, &written, NULL), "Write to anonymous pipe failed\n");
793 ok(written == size, "Write to anonymous pipe wrote %d bytes\n", written);
794 /* and close the write end, read should still succeed*/
795 ok(CloseHandle(pipewrite), "CloseHandle for the Write Pipe failed\n");
796 memset( buffer, 0, size );
797 ok(ReadFile(piperead, buffer, size, &read, NULL), "Read from broken pipe withe with pending data failed\n");
798 ok(read == size, "Read from anonymous pipe got %d bytes\n", read);
799 for (i = 0; i < size; i++) ok( buffer[i] == (BYTE)i, "invalid data %x at %x\n", buffer[i], i );
800 /* But now we need to get informed that the pipe is closed */
801 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL) == 0, "Broken pipe not detected\n");
802 ok(CloseHandle(piperead), "CloseHandle for the read pipe failed\n");
803 HeapFree(GetProcessHeap(), 0, buffer);
806 struct named_pipe_client_params
808 DWORD security_flags;
809 HANDLE token;
810 BOOL revert;
813 #define PIPE_NAME "\\\\.\\pipe\\named_pipe_test"
815 static DWORD CALLBACK named_pipe_client_func(LPVOID p)
817 struct named_pipe_client_params *params = (struct named_pipe_client_params *)p;
818 HANDLE pipe;
819 BOOL ret;
820 const char message[] = "Test";
821 DWORD bytes_read, bytes_written;
822 char dummy;
823 TOKEN_PRIVILEGES *Privileges = NULL;
825 if (params->token)
827 if (params->revert)
829 /* modify the token so we can tell if the pipe impersonation
830 * token reverts to the process token */
831 ret = AdjustTokenPrivileges(params->token, TRUE, NULL, 0, NULL, NULL);
832 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
834 ret = SetThreadToken(NULL, params->token);
835 ok(ret, "SetThreadToken failed with error %d\n", GetLastError());
837 else
839 DWORD Size = 0;
840 HANDLE process_token;
842 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY|TOKEN_ADJUST_PRIVILEGES, &process_token);
843 ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
845 ret = GetTokenInformation(process_token, TokenPrivileges, NULL, 0, &Size);
846 ok(!ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER, "GetTokenInformation(TokenPrivileges) failed with %d\n", GetLastError());
847 Privileges = HeapAlloc(GetProcessHeap(), 0, Size);
848 ret = GetTokenInformation(process_token, TokenPrivileges, Privileges, Size, &Size);
849 ok(ret, "GetTokenInformation(TokenPrivileges) failed with %d\n", GetLastError());
851 ret = AdjustTokenPrivileges(process_token, TRUE, NULL, 0, NULL, NULL);
852 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
854 CloseHandle(process_token);
857 pipe = CreateFile(PIPE_NAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, params->security_flags, NULL);
858 ok(pipe != INVALID_HANDLE_VALUE, "CreateFile for pipe failed with error %d\n", GetLastError());
860 ret = WriteFile(pipe, message, sizeof(message), &bytes_written, NULL);
861 ok(ret, "WriteFile failed with error %d\n", GetLastError());
863 ret = ReadFile(pipe, &dummy, sizeof(dummy), &bytes_read, NULL);
864 ok(ret, "ReadFile failed with error %d\n", GetLastError());
866 if (params->token)
868 if (params->revert)
870 ret = RevertToSelf();
871 ok(ret, "RevertToSelf failed with error %d\n", GetLastError());
873 else
875 ret = AdjustTokenPrivileges(params->token, TRUE, NULL, 0, NULL, NULL);
876 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
879 else
881 HANDLE process_token;
883 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &process_token);
884 ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
886 ret = AdjustTokenPrivileges(process_token, FALSE, Privileges, 0, NULL, NULL);
887 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
889 HeapFree(GetProcessHeap(), 0, Privileges);
891 CloseHandle(process_token);
894 ret = WriteFile(pipe, message, sizeof(message), &bytes_written, NULL);
895 ok(ret, "WriteFile failed with error %d\n", GetLastError());
897 ret = ReadFile(pipe, &dummy, sizeof(dummy), &bytes_read, NULL);
898 ok(ret, "ReadFile failed with error %d\n", GetLastError());
900 CloseHandle(pipe);
902 return 0;
905 static HANDLE make_impersonation_token(DWORD Access, SECURITY_IMPERSONATION_LEVEL ImpersonationLevel)
907 HANDLE ProcessToken;
908 HANDLE Token = NULL;
909 BOOL ret;
911 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_DUPLICATE, &ProcessToken);
912 ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
914 ret = pDuplicateTokenEx(ProcessToken, Access, NULL, ImpersonationLevel, TokenImpersonation, &Token);
915 ok(ret, "DuplicateToken failed with error %d\n", GetLastError());
917 CloseHandle(ProcessToken);
919 return Token;
922 static void test_ImpersonateNamedPipeClient(HANDLE hClientToken, DWORD security_flags, BOOL revert, void (*test_func)(int, HANDLE))
924 HANDLE hPipeServer;
925 BOOL ret;
926 DWORD dwTid;
927 HANDLE hThread;
928 char buffer[256];
929 DWORD dwBytesRead;
930 DWORD error;
931 struct named_pipe_client_params params;
932 char dummy = 0;
933 DWORD dwBytesWritten;
934 HANDLE hToken = NULL;
935 SECURITY_IMPERSONATION_LEVEL ImpersonationLevel;
936 DWORD size;
938 hPipeServer = CreateNamedPipe(PIPE_NAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_READMODE_BYTE | PIPE_WAIT, 1, 100, 100, NMPWAIT_USE_DEFAULT_WAIT, NULL);
939 ok(hPipeServer != INVALID_HANDLE_VALUE, "CreateNamedPipe failed with error %d\n", GetLastError());
941 params.security_flags = security_flags;
942 params.token = hClientToken;
943 params.revert = revert;
944 hThread = CreateThread(NULL, 0, named_pipe_client_func, &params, 0, &dwTid);
945 ok(hThread != NULL, "CreateThread failed with error %d\n", GetLastError());
947 SetLastError(0xdeadbeef);
948 ret = ImpersonateNamedPipeClient(hPipeServer);
949 error = GetLastError();
950 ok(ret /* win2k3 */ || (error == ERROR_CANNOT_IMPERSONATE),
951 "ImpersonateNamedPipeClient should have failed with ERROR_CANNOT_IMPERSONATE instead of %d\n", GetLastError());
953 ret = ConnectNamedPipe(hPipeServer, NULL);
954 ok(ret || (GetLastError() == ERROR_PIPE_CONNECTED), "ConnectNamedPipe failed with error %d\n", GetLastError());
956 ret = ReadFile(hPipeServer, buffer, sizeof(buffer), &dwBytesRead, NULL);
957 ok(ret, "ReadFile failed with error %d\n", GetLastError());
959 ret = ImpersonateNamedPipeClient(hPipeServer);
960 ok(ret, "ImpersonateNamedPipeClient failed with error %d\n", GetLastError());
962 ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, FALSE, &hToken);
963 ok(ret, "OpenThreadToken failed with error %d\n", GetLastError());
965 (*test_func)(0, hToken);
967 ImpersonationLevel = 0xdeadbeef; /* to avoid false positives */
968 ret = GetTokenInformation(hToken, TokenImpersonationLevel, &ImpersonationLevel, sizeof(ImpersonationLevel), &size);
969 ok(ret, "GetTokenInformation(TokenImpersonationLevel) failed with error %d\n", GetLastError());
970 ok(ImpersonationLevel == SecurityImpersonation, "ImpersonationLevel should have been SecurityImpersonation(%d) instead of %d\n", SecurityImpersonation, ImpersonationLevel);
972 CloseHandle(hToken);
974 RevertToSelf();
976 ret = WriteFile(hPipeServer, &dummy, sizeof(dummy), &dwBytesWritten, NULL);
977 ok(ret, "WriteFile failed with error %d\n", GetLastError());
979 ret = ReadFile(hPipeServer, buffer, sizeof(buffer), &dwBytesRead, NULL);
980 ok(ret, "ReadFile failed with error %d\n", GetLastError());
982 ret = ImpersonateNamedPipeClient(hPipeServer);
983 ok(ret, "ImpersonateNamedPipeClient failed with error %d\n", GetLastError());
985 ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, FALSE, &hToken);
986 ok(ret, "OpenThreadToken failed with error %d\n", GetLastError());
988 (*test_func)(1, hToken);
990 CloseHandle(hToken);
992 RevertToSelf();
994 ret = WriteFile(hPipeServer, &dummy, sizeof(dummy), &dwBytesWritten, NULL);
995 ok(ret, "WriteFile failed with error %d\n", GetLastError());
997 WaitForSingleObject(hThread, INFINITE);
999 ret = ImpersonateNamedPipeClient(hPipeServer);
1000 ok(ret, "ImpersonateNamedPipeClient failed with error %d\n", GetLastError());
1002 RevertToSelf();
1004 CloseHandle(hThread);
1005 CloseHandle(hPipeServer);
1008 static BOOL are_all_privileges_disabled(HANDLE hToken)
1010 BOOL ret;
1011 TOKEN_PRIVILEGES *Privileges = NULL;
1012 DWORD Size = 0;
1013 BOOL all_privs_disabled = TRUE;
1014 DWORD i;
1016 ret = GetTokenInformation(hToken, TokenPrivileges, NULL, 0, &Size);
1017 if (!ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER)
1019 Privileges = HeapAlloc(GetProcessHeap(), 0, Size);
1020 ret = GetTokenInformation(hToken, TokenPrivileges, Privileges, Size, &Size);
1021 if (!ret)
1023 HeapFree(GetProcessHeap(), 0, Privileges);
1024 return FALSE;
1027 else
1028 return FALSE;
1030 for (i = 0; i < Privileges->PrivilegeCount; i++)
1032 if (Privileges->Privileges[i].Attributes & SE_PRIVILEGE_ENABLED)
1034 all_privs_disabled = FALSE;
1035 break;
1039 HeapFree(GetProcessHeap(), 0, Privileges);
1041 return all_privs_disabled;
1044 static DWORD get_privilege_count(HANDLE hToken)
1046 TOKEN_STATISTICS Statistics;
1047 DWORD Size = sizeof(Statistics);
1048 BOOL ret;
1050 ret = GetTokenInformation(hToken, TokenStatistics, &Statistics, Size, &Size);
1051 ok(ret, "GetTokenInformation(TokenStatistics)\n");
1052 if (!ret) return -1;
1054 return Statistics.PrivilegeCount;
1057 static void test_no_sqos_no_token(int call_index, HANDLE hToken)
1059 DWORD priv_count;
1061 switch (call_index)
1063 case 0:
1064 priv_count = get_privilege_count(hToken);
1065 todo_wine
1066 ok(priv_count == 0, "privilege count should have been 0 instead of %d\n", priv_count);
1067 break;
1068 case 1:
1069 priv_count = get_privilege_count(hToken);
1070 ok(priv_count > 0, "privilege count should now be > 0 instead of 0\n");
1071 ok(!are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1072 break;
1073 default:
1074 ok(0, "shouldn't happen\n");
1078 static void test_no_sqos(int call_index, HANDLE hToken)
1080 switch (call_index)
1082 case 0:
1083 ok(!are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1084 break;
1085 case 1:
1086 todo_wine
1087 ok(are_all_privileges_disabled(hToken), "impersonated token should have been modified\n");
1088 break;
1089 default:
1090 ok(0, "shouldn't happen\n");
1094 static void test_static_context(int call_index, HANDLE hToken)
1096 switch (call_index)
1098 case 0:
1099 ok(!are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1100 break;
1101 case 1:
1102 ok(!are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1103 break;
1104 default:
1105 ok(0, "shouldn't happen\n");
1109 static void test_dynamic_context(int call_index, HANDLE hToken)
1111 switch (call_index)
1113 case 0:
1114 ok(!are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1115 break;
1116 case 1:
1117 todo_wine
1118 ok(are_all_privileges_disabled(hToken), "impersonated token should have been modified\n");
1119 break;
1120 default:
1121 ok(0, "shouldn't happen\n");
1125 static void test_dynamic_context_no_token(int call_index, HANDLE hToken)
1127 switch (call_index)
1129 case 0:
1130 ok(are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1131 break;
1132 case 1:
1133 ok(!are_all_privileges_disabled(hToken), "process token modification should have been detected and impersonation token updated\n");
1134 break;
1135 default:
1136 ok(0, "shouldn't happen\n");
1140 static void test_no_sqos_revert(int call_index, HANDLE hToken)
1142 DWORD priv_count;
1143 switch (call_index)
1145 case 0:
1146 priv_count = get_privilege_count(hToken);
1147 todo_wine
1148 ok(priv_count == 0, "privilege count should have been 0 instead of %d\n", priv_count);
1149 break;
1150 case 1:
1151 priv_count = get_privilege_count(hToken);
1152 ok(priv_count > 0, "privilege count should now be > 0 instead of 0\n");
1153 ok(!are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1154 break;
1155 default:
1156 ok(0, "shouldn't happen\n");
1160 static void test_static_context_revert(int call_index, HANDLE hToken)
1162 switch (call_index)
1164 case 0:
1165 todo_wine
1166 ok(are_all_privileges_disabled(hToken), "privileges should have been disabled\n");
1167 break;
1168 case 1:
1169 todo_wine
1170 ok(are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1171 break;
1172 default:
1173 ok(0, "shouldn't happen\n");
1177 static void test_dynamic_context_revert(int call_index, HANDLE hToken)
1179 switch (call_index)
1181 case 0:
1182 todo_wine
1183 ok(are_all_privileges_disabled(hToken), "privileges should have been disabled\n");
1184 break;
1185 case 1:
1186 ok(!are_all_privileges_disabled(hToken), "impersonated token should now be process token\n");
1187 break;
1188 default:
1189 ok(0, "shouldn't happen\n");
1193 static void test_impersonation(void)
1195 HANDLE hClientToken;
1196 HANDLE hProcessToken;
1197 BOOL ret;
1199 if( !pDuplicateTokenEx ) {
1200 skip("DuplicateTokenEx not found\n");
1201 return;
1204 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hProcessToken);
1205 if (!ret)
1207 skip("couldn't open process token, skipping impersonation tests\n");
1208 return;
1211 if (!get_privilege_count(hProcessToken) || are_all_privileges_disabled(hProcessToken))
1213 skip("token didn't have any privileges or they were all disabled. token not suitable for impersonation tests\n");
1214 CloseHandle(hProcessToken);
1215 return;
1217 CloseHandle(hProcessToken);
1219 test_ImpersonateNamedPipeClient(NULL, 0, FALSE, test_no_sqos_no_token);
1220 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1221 test_ImpersonateNamedPipeClient(hClientToken, 0, FALSE, test_no_sqos);
1222 CloseHandle(hClientToken);
1223 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1224 test_ImpersonateNamedPipeClient(hClientToken,
1225 SECURITY_SQOS_PRESENT | SECURITY_IMPERSONATION, FALSE,
1226 test_static_context);
1227 CloseHandle(hClientToken);
1228 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1229 test_ImpersonateNamedPipeClient(hClientToken,
1230 SECURITY_SQOS_PRESENT | SECURITY_CONTEXT_TRACKING | SECURITY_IMPERSONATION,
1231 FALSE, test_dynamic_context);
1232 CloseHandle(hClientToken);
1233 test_ImpersonateNamedPipeClient(NULL,
1234 SECURITY_SQOS_PRESENT | SECURITY_CONTEXT_TRACKING | SECURITY_IMPERSONATION,
1235 FALSE, test_dynamic_context_no_token);
1237 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1238 test_ImpersonateNamedPipeClient(hClientToken, 0, TRUE, test_no_sqos_revert);
1239 CloseHandle(hClientToken);
1240 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1241 test_ImpersonateNamedPipeClient(hClientToken,
1242 SECURITY_SQOS_PRESENT | SECURITY_IMPERSONATION, TRUE,
1243 test_static_context_revert);
1244 CloseHandle(hClientToken);
1245 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1246 test_ImpersonateNamedPipeClient(hClientToken,
1247 SECURITY_SQOS_PRESENT | SECURITY_CONTEXT_TRACKING | SECURITY_IMPERSONATION,
1248 TRUE, test_dynamic_context_revert);
1249 CloseHandle(hClientToken);
1252 struct overlapped_server_args
1254 HANDLE pipe_created;
1257 static DWORD CALLBACK overlapped_server(LPVOID arg)
1259 OVERLAPPED ol;
1260 HANDLE pipe;
1261 int ret, err;
1262 struct overlapped_server_args *a = (struct overlapped_server_args*)arg;
1263 DWORD num;
1264 char buf[100];
1266 pipe = CreateNamedPipeA("\\\\.\\pipe\\my pipe", FILE_FLAG_OVERLAPPED | PIPE_ACCESS_DUPLEX, PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE, 1, 0, 0, 100000, NULL);
1267 ok(pipe != NULL, "pipe NULL\n");
1269 ol.hEvent = CreateEventA(0, 1, 0, 0);
1270 ok(ol.hEvent != NULL, "event NULL\n");
1271 ret = ConnectNamedPipe(pipe, &ol);
1272 err = GetLastError();
1273 ok(ret == 0, "ret %d\n", ret);
1274 ok(err == ERROR_IO_PENDING, "gle %d\n", err);
1275 SetEvent(a->pipe_created);
1277 ret = WaitForSingleObjectEx(ol.hEvent, INFINITE, 1);
1278 ok(ret == WAIT_OBJECT_0, "ret %x\n", ret);
1280 ret = GetOverlappedResult(pipe, &ol, &num, 1);
1281 ok(ret == 1, "ret %d\n", ret);
1283 /* This should block */
1284 ret = ReadFile(pipe, buf, sizeof(buf), &num, NULL);
1285 ok(ret == 1, "ret %d\n", ret);
1287 DisconnectNamedPipe(pipe);
1288 CloseHandle(ol.hEvent);
1289 CloseHandle(pipe);
1290 return 1;
1293 static void test_overlapped(void)
1295 DWORD tid, num;
1296 HANDLE thread, pipe;
1297 int ret;
1298 struct overlapped_server_args args;
1300 args.pipe_created = CreateEventA(0, 1, 0, 0);
1301 thread = CreateThread(NULL, 0, overlapped_server, &args, 0, &tid);
1303 WaitForSingleObject(args.pipe_created, INFINITE);
1304 pipe = CreateFileA("\\\\.\\pipe\\my pipe", GENERIC_READ|GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
1305 ok(pipe != INVALID_HANDLE_VALUE, "cf failed\n");
1307 /* Sleep to try to get the ReadFile in the server to occur before the following WriteFile */
1308 Sleep(1);
1310 ret = WriteFile(pipe, "x", 1, &num, NULL);
1311 ok(ret == 1, "ret %d\n", ret);
1313 WaitForSingleObject(thread, INFINITE);
1314 CloseHandle(pipe);
1315 CloseHandle(args.pipe_created);
1316 CloseHandle(thread);
1319 static void test_NamedPipeHandleState(void)
1321 HANDLE server, client;
1322 BOOL ret;
1323 DWORD state, instances, maxCollectionCount, collectDataTimeout;
1324 char userName[MAX_PATH];
1326 server = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX,
1327 /* dwOpenMode */ PIPE_TYPE_BYTE | PIPE_WAIT,
1328 /* nMaxInstances */ 1,
1329 /* nOutBufSize */ 1024,
1330 /* nInBufSize */ 1024,
1331 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
1332 /* lpSecurityAttrib */ NULL);
1333 ok(server != INVALID_HANDLE_VALUE, "cf failed\n");
1334 ret = GetNamedPipeHandleState(server, NULL, NULL, NULL, NULL, NULL, 0);
1335 todo_wine
1336 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1337 ret = GetNamedPipeHandleState(server, &state, &instances, NULL, NULL, NULL,
1339 todo_wine
1340 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1341 if (ret)
1343 ok(state == 0, "unexpected state %08x\n", state);
1344 ok(instances == 1, "expected 1 instances, got %d\n", instances);
1346 /* Some parameters have no meaning, and therefore can't be retrieved,
1347 * on a local pipe.
1349 SetLastError(0xdeadbeef);
1350 ret = GetNamedPipeHandleState(server, &state, &instances,
1351 &maxCollectionCount, &collectDataTimeout, userName,
1352 sizeof(userName) / sizeof(userName[0]));
1353 todo_wine
1354 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1355 "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1356 /* A byte-mode pipe server can't be changed to message mode. */
1357 state = PIPE_READMODE_MESSAGE;
1358 SetLastError(0xdeadbeef);
1359 ret = SetNamedPipeHandleState(server, &state, NULL, NULL);
1360 todo_wine
1361 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1362 "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1364 client = CreateFileA(PIPENAME, GENERIC_READ|GENERIC_WRITE, 0, NULL,
1365 OPEN_EXISTING, 0, NULL);
1366 ok(client != INVALID_HANDLE_VALUE, "cf failed\n");
1368 state = PIPE_READMODE_BYTE;
1369 ret = SetNamedPipeHandleState(client, &state, NULL, NULL);
1370 todo_wine
1371 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1372 /* A byte-mode pipe client can't be changed to message mode, either. */
1373 state = PIPE_READMODE_MESSAGE;
1374 SetLastError(0xdeadbeef);
1375 ret = SetNamedPipeHandleState(server, &state, NULL, NULL);
1376 todo_wine
1377 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1378 "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1380 CloseHandle(client);
1381 CloseHandle(server);
1383 server = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX,
1384 /* dwOpenMode */ PIPE_TYPE_MESSAGE | PIPE_WAIT,
1385 /* nMaxInstances */ 1,
1386 /* nOutBufSize */ 1024,
1387 /* nInBufSize */ 1024,
1388 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
1389 /* lpSecurityAttrib */ NULL);
1390 ok(server != INVALID_HANDLE_VALUE, "cf failed\n");
1391 ret = GetNamedPipeHandleState(server, NULL, NULL, NULL, NULL, NULL, 0);
1392 todo_wine
1393 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1394 ret = GetNamedPipeHandleState(server, &state, &instances, NULL, NULL, NULL,
1396 todo_wine
1397 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1398 if (ret)
1400 ok(state == 0, "unexpected state %08x\n", state);
1401 ok(instances == 1, "expected 1 instances, got %d\n", instances);
1403 /* In contrast to byte-mode pipes, a message-mode pipe server can be
1404 * changed to byte mode.
1406 state = PIPE_READMODE_BYTE;
1407 ret = SetNamedPipeHandleState(server, &state, NULL, NULL);
1408 todo_wine
1409 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1411 client = CreateFileA(PIPENAME, GENERIC_READ|GENERIC_WRITE, 0, NULL,
1412 OPEN_EXISTING, 0, NULL);
1413 ok(client != INVALID_HANDLE_VALUE, "cf failed\n");
1415 state = PIPE_READMODE_MESSAGE;
1416 ret = SetNamedPipeHandleState(client, &state, NULL, NULL);
1417 todo_wine
1418 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1419 /* A message-mode pipe client can also be changed to byte mode.
1421 state = PIPE_READMODE_BYTE;
1422 ret = SetNamedPipeHandleState(client, &state, NULL, NULL);
1423 todo_wine
1424 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1426 CloseHandle(client);
1427 CloseHandle(server);
1430 START_TEST(pipe)
1432 HMODULE hmod;
1434 hmod = GetModuleHandle("advapi32.dll");
1435 pDuplicateTokenEx = (void *) GetProcAddress(hmod, "DuplicateTokenEx");
1437 if (test_DisconnectNamedPipe())
1438 return;
1439 test_CreateNamedPipe_instances_must_match();
1440 test_NamedPipe_2();
1441 test_CreateNamedPipe(PIPE_TYPE_BYTE);
1442 test_CreateNamedPipe(PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE);
1443 test_CreatePipe();
1444 test_impersonation();
1445 test_overlapped();
1446 test_NamedPipeHandleState();