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
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)
46 static const char obuf[] = "Bit Bucket";
47 static const char obuf2[] = "More bits";
54 if (pipemode == PIPE_TYPE_BYTE)
55 trace("test_CreateNamedPipe starting in byte mode\n");
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) {
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);
133 todo_wine ok(readden == sizeof(obuf) + sizeof(obuf2), "peek3 got %d bytes\n", readden);
137 if (readden != sizeof(obuf) + sizeof(obuf2)) /* MacOS returns both messages */
138 ok(readden == sizeof(obuf), "peek3 got %d bytes\n", readden);
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);
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);
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);
169 todo_wine ok(readden == sizeof(obuf) + sizeof(obuf2), "peek4 got %d bytes\n", readden);
173 if (readden != sizeof(obuf) + sizeof(obuf2)) /* MacOS returns both messages */
174 ok(readden == sizeof(obuf), "peek4 got %d bytes\n", readden);
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);
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);
192 ok(readden == sizeof(obuf), "read 4 got %d bytes\n", readden);
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");
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);
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);
227 todo_wine ok(avail == sizeof(obuf) + sizeof(obuf2), "peek5 got %d bytes available\n", avail);
229 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 5a check\n");
230 ok(ReadFile(hFile, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
232 ok(readden == sizeof(obuf), "read 5 got %d bytes\n", readden);
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");
241 ok(readden == sizeof(obuf2), "peek6a got %d bytes\n", readden);
242 ok(avail == sizeof(obuf2), "peek6a got %d bytes available\n", avail);
245 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
246 ok(readden == sizeof(obuf2), "read 6a got %d bytes\n", readden);
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);
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);
263 ok(memcmp(obuf, pbuf, sizeof(obuf)) == 0, "content 6a check\n");
264 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL), "ReadFile\n");
266 ok(readden == sizeof(obuf), "read 6b got %d bytes\n", readden);
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)
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");
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");
395 HANDLE hnp = INVALID_HANDLE_VALUE;
397 /** Trivial byte echo server - disconnects after each session */
398 static DWORD CALLBACK serverThreadMain1(LPVOID arg)
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++) {
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");
448 /** Trivial byte echo server - closes after each connection */
449 static DWORD CALLBACK serverThreadMain2(LPVOID arg)
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++) {
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 */
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");
512 /** Trivial byte echo server - uses overlapped named pipe calls */
513 static DWORD CALLBACK serverThreadMain3(LPVOID arg)
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 */
533 ok(hEvent != NULL, "CreateEvent\n");
535 for (i = 0; i < NB_SERVER_LOOPS; i++) {
542 int letWFSOEwait = (i & 2);
543 int letGORwait = (i & 1);
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 || err == ERROR_PIPE_CONNECTED), "overlapped ConnectNamedPipe\n");
554 trace("overlapped ConnectNamedPipe returned.\n");
555 if (!success && (err == ERROR_IO_PENDING)) {
560 ret = WaitForSingleObjectEx(hEvent, INFINITE, TRUE);
561 } while (ret == WAIT_IO_COMPLETION);
562 ok(ret == 0, "wait ConnectNamedPipe returned %x\n", ret);
564 success = GetOverlappedResult(hnp, &oOverlap, &dummy, letGORwait);
565 if (!letGORwait && !letWFSOEwait && !success) {
566 ok(GetLastError() == ERROR_IO_INCOMPLETE, "GetOverlappedResult\n");
567 success = GetOverlappedResult(hnp, &oOverlap, &dummy, TRUE);
570 ok(success || (err == ERROR_PIPE_CONNECTED), "GetOverlappedResult ConnectNamedPipe\n");
571 trace("overlapped ConnectNamedPipe operation complete.\n");
573 /* Echo bytes once */
574 memset(buf, 0, sizeof(buf));
576 trace("Server reading...\n");
577 success = ReadFile(hnp, buf, sizeof(buf), &readden, &oOverlap);
578 trace("Server ReadFile returned...\n");
579 err = GetLastError();
580 ok(success || err == ERROR_IO_PENDING, "overlapped ReadFile\n");
581 trace("overlapped ReadFile returned.\n");
582 if (!success && (err == ERROR_IO_PENDING)) {
587 ret = WaitForSingleObjectEx(hEvent, INFINITE, TRUE);
588 } while (ret == WAIT_IO_COMPLETION);
589 ok(ret == 0, "wait ReadFile returned %x\n", ret);
591 success = GetOverlappedResult(hnp, &oOverlap, &readden, letGORwait);
592 if (!letGORwait && !letWFSOEwait && !success) {
593 ok(GetLastError() == ERROR_IO_INCOMPLETE, "GetOverlappedResult\n");
594 success = GetOverlappedResult(hnp, &oOverlap, &readden, TRUE);
597 trace("Server done reading.\n");
598 ok(success, "overlapped ReadFile\n");
600 trace("Server writing...\n");
601 success = WriteFile(hnp, buf, readden, &written, &oOverlap);
602 trace("Server WriteFile returned...\n");
603 err = GetLastError();
604 ok(success || err == ERROR_IO_PENDING, "overlapped WriteFile\n");
605 trace("overlapped WriteFile returned.\n");
606 if (!success && (err == ERROR_IO_PENDING)) {
611 ret = WaitForSingleObjectEx(hEvent, INFINITE, TRUE);
612 } while (ret == WAIT_IO_COMPLETION);
613 ok(ret == 0, "wait WriteFile returned %x\n", ret);
615 success = GetOverlappedResult(hnp, &oOverlap, &written, letGORwait);
616 if (!letGORwait && !letWFSOEwait && !success) {
617 ok(GetLastError() == ERROR_IO_INCOMPLETE, "GetOverlappedResult\n");
618 success = GetOverlappedResult(hnp, &oOverlap, &written, TRUE);
621 trace("Server done writing.\n");
622 ok(success, "overlapped WriteFile\n");
623 ok(written == readden, "write file len\n");
625 /* finish this connection, wait for next one */
626 ok(FlushFileBuffers(hnp), "FlushFileBuffers\n");
627 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe\n");
632 /** Trivial byte echo server - uses i/o completion ports */
633 static DWORD CALLBACK serverThreadMain4(LPVOID arg)
638 trace("serverThreadMain4\n");
639 /* Set up a simple echo server */
640 hnp = CreateNamedPipe(PIPENAME "serverThreadMain4", PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
641 PIPE_TYPE_BYTE | PIPE_WAIT,
642 /* nMaxInstances */ 1,
643 /* nOutBufSize */ 1024,
644 /* nInBufSize */ 1024,
645 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
646 /* lpSecurityAttrib */ NULL);
647 ok(hnp != INVALID_HANDLE_VALUE, "CreateNamedPipe failed\n");
649 hcompletion = CreateIoCompletionPort(hnp, NULL, 12345, 1);
650 ok(hcompletion != NULL, "CreateIoCompletionPort failed, error=%i\n", GetLastError());
652 for (i = 0; i < NB_SERVER_LOOPS; i++) {
665 memset(&oConnect, 0, sizeof(oConnect));
666 memset(&oRead, 0, sizeof(oRead));
667 memset(&oWrite, 0, sizeof(oWrite));
669 /* Wait for client to connect */
670 trace("Server calling overlapped ConnectNamedPipe...\n");
671 success = ConnectNamedPipe(hnp, &oConnect);
672 err = GetLastError();
673 ok(!success && (err == ERROR_IO_PENDING || err == ERROR_PIPE_CONNECTED),
674 "overlapped ConnectNamedPipe got %u err %u\n", success, err );
675 if (!success && err == ERROR_IO_PENDING) {
676 trace("ConnectNamedPipe GetQueuedCompletionStatus\n");
677 success = GetQueuedCompletionStatus(hcompletion, &dummy, &compkey, &oResult, 0);
680 ok( GetLastError() == WAIT_TIMEOUT,
681 "ConnectNamedPipe GetQueuedCompletionStatus wrong error %u\n", GetLastError());
682 success = GetQueuedCompletionStatus(hcompletion, &dummy, &compkey, &oResult, 10000);
684 ok(success, "ConnectNamedPipe GetQueuedCompletionStatus failed, errno=%i\n", GetLastError());
687 ok(compkey == 12345, "got completion key %i instead of 12345\n", (int)compkey);
688 ok(oResult == &oConnect, "got overlapped pointer %p instead of %p\n", oResult, &oConnect);
691 trace("overlapped ConnectNamedPipe operation complete.\n");
693 /* Echo bytes once */
694 memset(buf, 0, sizeof(buf));
696 trace("Server reading...\n");
697 success = ReadFile(hnp, buf, sizeof(buf), &readden, &oRead);
698 trace("Server ReadFile returned...\n");
699 err = GetLastError();
700 ok(success || err == ERROR_IO_PENDING, "overlapped ReadFile, err=%i\n", err);
701 success = GetQueuedCompletionStatus(hcompletion, &readden, &compkey,
703 ok(success, "ReadFile GetQueuedCompletionStatus failed, errno=%i\n", GetLastError());
706 ok(compkey == 12345, "got completion key %i instead of 12345\n", (int)compkey);
707 ok(oResult == &oRead, "got overlapped pointer %p instead of %p\n", oResult, &oRead);
709 trace("Server done reading.\n");
711 trace("Server writing...\n");
712 success = WriteFile(hnp, buf, readden, &written, &oWrite);
713 trace("Server WriteFile returned...\n");
714 err = GetLastError();
715 ok(success || err == ERROR_IO_PENDING, "overlapped WriteFile failed, err=%u\n", err);
716 success = GetQueuedCompletionStatus(hcompletion, &written, &compkey,
718 ok(success, "WriteFile GetQueuedCompletionStatus failed, errno=%i\n", GetLastError());
721 ok(compkey == 12345, "got completion key %i instead of 12345\n", (int)compkey);
722 ok(oResult == &oWrite, "got overlapped pointer %p instead of %p\n", oResult, &oWrite);
723 ok(written == readden, "write file len\n");
725 trace("Server done writing.\n");
727 /* finish this connection, wait for next one */
728 ok(FlushFileBuffers(hnp), "FlushFileBuffers\n");
729 success = DisconnectNamedPipe(hnp);
730 ok(success, "DisconnectNamedPipe failed, err %u\n", GetLastError());
733 ok(CloseHandle(hnp), "CloseHandle named pipe failed, err=%i\n", GetLastError());
734 ok(CloseHandle(hcompletion), "CloseHandle completion failed, err=%i\n", GetLastError());
739 static void exercizeServer(const char *pipename, HANDLE serverThread)
743 trace("exercizeServer starting\n");
744 for (i = 0; i < NB_SERVER_LOOPS; i++) {
745 HANDLE hFile=INVALID_HANDLE_VALUE;
746 static const char obuf[] = "Bit Bucket";
752 for (loop = 0; loop < 3; loop++) {
754 trace("Client connecting...\n");
755 /* Connect to the server */
756 hFile = CreateFileA(pipename, GENERIC_READ | GENERIC_WRITE, 0,
757 NULL, OPEN_EXISTING, 0, 0);
758 if (hFile != INVALID_HANDLE_VALUE)
760 err = GetLastError();
762 ok(err == ERROR_PIPE_BUSY || err == ERROR_FILE_NOT_FOUND, "connecting to pipe\n");
764 ok(err == ERROR_PIPE_BUSY, "connecting to pipe\n");
765 trace("connect failed, retrying\n");
768 ok(hFile != INVALID_HANDLE_VALUE, "client opening named pipe\n");
770 /* Make sure it can echo */
771 memset(ibuf, 0, sizeof(ibuf));
772 trace("Client writing...\n");
773 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL), "WriteFile to client end of pipe\n");
774 ok(written == sizeof(obuf), "write file len\n");
775 trace("Client reading...\n");
776 ok(ReadFile(hFile, ibuf, sizeof(obuf), &readden, NULL), "ReadFile from client end of pipe\n");
777 ok(readden == sizeof(obuf), "read file len\n");
778 ok(memcmp(obuf, ibuf, written) == 0, "content check\n");
780 trace("Client closing...\n");
781 ok(CloseHandle(hFile), "CloseHandle\n");
784 ok(WaitForSingleObject(serverThread,INFINITE) == WAIT_OBJECT_0, "WaitForSingleObject\n");
786 trace("exercizeServer returning\n");
789 static void test_NamedPipe_2(void)
792 DWORD serverThreadId;
796 trace("test_NamedPipe_2 starting\n");
797 /* Set up a ten second timeout */
798 alarm_event = CreateEvent( NULL, TRUE, FALSE, NULL );
799 alarmThread = CreateThread(NULL, 0, alarmThreadMain, (void *) 10000, 0, &alarmThreadId);
801 /* The servers we're about to exercize do try to clean up carefully,
802 * but to reduce the change of a test failure due to a pipe handle
803 * leak in the test code, we'll use a different pipe name for each server.
807 serverThread = CreateThread(NULL, 0, serverThreadMain1, (void *)8, 0, &serverThreadId);
808 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
809 exercizeServer(PIPENAME "serverThreadMain1", serverThread);
812 serverThread = CreateThread(NULL, 0, serverThreadMain2, 0, 0, &serverThreadId);
813 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
814 exercizeServer(PIPENAME "serverThreadMain2", serverThread);
817 serverThread = CreateThread(NULL, 0, serverThreadMain3, 0, 0, &serverThreadId);
818 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
819 exercizeServer(PIPENAME "serverThreadMain3", serverThread);
822 serverThread = CreateThread(NULL, 0, serverThreadMain4, 0, 0, &serverThreadId);
823 ok(serverThread != INVALID_HANDLE_VALUE, "CreateThread\n");
824 exercizeServer(PIPENAME "serverThreadMain4", serverThread);
826 ok(SetEvent( alarm_event ), "SetEvent\n");
827 CloseHandle( alarm_event );
828 trace("test_NamedPipe_2 returning\n");
831 static int test_DisconnectNamedPipe(void)
835 static const char obuf[] = "Bit Bucket";
841 SetLastError(0xdeadbeef);
842 hnp = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_WAIT,
843 /* nMaxInstances */ 1,
844 /* nOutBufSize */ 1024,
845 /* nInBufSize */ 1024,
846 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
847 /* lpSecurityAttrib */ NULL);
848 if ((hnp == INVALID_HANDLE_VALUE /* Win98 */ || !hnp /* Win95 */)
849 && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) {
851 win_skip("Named pipes are not implemented\n");
855 ok(WriteFile(hnp, obuf, sizeof(obuf), &written, NULL) == 0
856 && GetLastError() == ERROR_PIPE_LISTENING, "WriteFile to not-yet-connected pipe\n");
857 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL) == 0
858 && GetLastError() == ERROR_PIPE_LISTENING, "ReadFile from not-yet-connected pipe\n");
860 hFile = CreateFileA(PIPENAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0);
861 ok(hFile != INVALID_HANDLE_VALUE, "CreateFile failed\n");
863 /* don't try to do i/o if one side couldn't be opened, as it hangs */
864 if (hFile != INVALID_HANDLE_VALUE) {
866 /* see what happens if server calls DisconnectNamedPipe
867 * when there are bytes in the pipe
870 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL), "WriteFile\n");
871 ok(written == sizeof(obuf), "write file len\n");
872 ok(DisconnectNamedPipe(hnp), "DisconnectNamedPipe while messages waiting\n");
873 ok(WriteFile(hFile, obuf, sizeof(obuf), &written, NULL) == 0
874 && GetLastError() == ERROR_PIPE_NOT_CONNECTED, "WriteFile to disconnected pipe\n");
875 ok(ReadFile(hnp, ibuf, sizeof(ibuf), &readden, NULL) == 0
876 && GetLastError() == ERROR_PIPE_NOT_CONNECTED,
877 "ReadFile from disconnected pipe with bytes waiting\n");
878 ok(!DisconnectNamedPipe(hnp) && GetLastError() == ERROR_PIPE_NOT_CONNECTED,
879 "DisconnectNamedPipe worked twice\n");
880 ret = WaitForSingleObject(hFile, 0);
881 ok(ret == WAIT_TIMEOUT, "WaitForSingleObject returned %X\n", ret);
882 ok(CloseHandle(hFile), "CloseHandle\n");
885 ok(CloseHandle(hnp), "CloseHandle\n");
889 static void test_CreatePipe(void)
891 SECURITY_ATTRIBUTES pipe_attr;
892 HANDLE piperead, pipewrite;
899 pipe_attr.nLength = sizeof(SECURITY_ATTRIBUTES);
900 pipe_attr.bInheritHandle = TRUE;
901 pipe_attr.lpSecurityDescriptor = NULL;
902 ok(CreatePipe(&piperead, &pipewrite, &pipe_attr, 0) != 0, "CreatePipe failed\n");
903 ok(WriteFile(pipewrite,PIPENAME,sizeof(PIPENAME), &written, NULL), "Write to anonymous pipe failed\n");
904 ok(written == sizeof(PIPENAME), "Write to anonymous pipe wrote %d bytes\n", written);
905 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL), "Read from non empty pipe failed\n");
906 ok(read == sizeof(PIPENAME), "Read from anonymous pipe got %d bytes\n", read);
907 ok(CloseHandle(pipewrite), "CloseHandle for the write pipe failed\n");
908 ok(CloseHandle(piperead), "CloseHandle for the read pipe failed\n");
910 /* Now write another chunk*/
911 ok(CreatePipe(&piperead, &pipewrite, &pipe_attr, 0) != 0, "CreatePipe failed\n");
912 ok(WriteFile(pipewrite,PIPENAME,sizeof(PIPENAME), &written, NULL), "Write to anonymous pipe failed\n");
913 ok(written == sizeof(PIPENAME), "Write to anonymous pipe wrote %d bytes\n", written);
914 /* and close the write end, read should still succeed*/
915 ok(CloseHandle(pipewrite), "CloseHandle for the Write Pipe failed\n");
916 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL), "Read from broken pipe withe with pending data failed\n");
917 ok(read == sizeof(PIPENAME), "Read from anonymous pipe got %d bytes\n", read);
918 /* But now we need to get informed that the pipe is closed */
919 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL) == 0, "Broken pipe not detected\n");
920 ok(CloseHandle(piperead), "CloseHandle for the read pipe failed\n");
922 /* Try bigger chunks */
924 buffer = HeapAlloc( GetProcessHeap(), 0, size );
925 for (i = 0; i < size; i++) buffer[i] = i;
926 ok(CreatePipe(&piperead, &pipewrite, &pipe_attr, (size + 24)) != 0, "CreatePipe failed\n");
927 ok(WriteFile(pipewrite, buffer, size, &written, NULL), "Write to anonymous pipe failed\n");
928 ok(written == size, "Write to anonymous pipe wrote %d bytes\n", written);
929 /* and close the write end, read should still succeed*/
930 ok(CloseHandle(pipewrite), "CloseHandle for the Write Pipe failed\n");
931 memset( buffer, 0, size );
932 ok(ReadFile(piperead, buffer, size, &read, NULL), "Read from broken pipe withe with pending data failed\n");
933 ok(read == size, "Read from anonymous pipe got %d bytes\n", read);
934 for (i = 0; i < size; i++) ok( buffer[i] == (BYTE)i, "invalid data %x at %x\n", buffer[i], i );
935 /* But now we need to get informed that the pipe is closed */
936 ok(ReadFile(piperead,readbuf,sizeof(readbuf),&read, NULL) == 0, "Broken pipe not detected\n");
937 ok(CloseHandle(piperead), "CloseHandle for the read pipe failed\n");
938 HeapFree(GetProcessHeap(), 0, buffer);
941 struct named_pipe_client_params
943 DWORD security_flags;
948 #define PIPE_NAME "\\\\.\\pipe\\named_pipe_test"
950 static DWORD CALLBACK named_pipe_client_func(LPVOID p)
952 struct named_pipe_client_params *params = p;
955 const char message[] = "Test";
956 DWORD bytes_read, bytes_written;
958 TOKEN_PRIVILEGES *Privileges = NULL;
964 /* modify the token so we can tell if the pipe impersonation
965 * token reverts to the process token */
966 ret = AdjustTokenPrivileges(params->token, TRUE, NULL, 0, NULL, NULL);
967 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
969 ret = SetThreadToken(NULL, params->token);
970 ok(ret, "SetThreadToken failed with error %d\n", GetLastError());
975 HANDLE process_token;
977 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY|TOKEN_ADJUST_PRIVILEGES, &process_token);
978 ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
980 ret = GetTokenInformation(process_token, TokenPrivileges, NULL, 0, &Size);
981 ok(!ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER, "GetTokenInformation(TokenPrivileges) failed with %d\n", GetLastError());
982 Privileges = HeapAlloc(GetProcessHeap(), 0, Size);
983 ret = GetTokenInformation(process_token, TokenPrivileges, Privileges, Size, &Size);
984 ok(ret, "GetTokenInformation(TokenPrivileges) failed with %d\n", GetLastError());
986 ret = AdjustTokenPrivileges(process_token, TRUE, NULL, 0, NULL, NULL);
987 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
989 CloseHandle(process_token);
992 pipe = CreateFile(PIPE_NAME, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, params->security_flags, NULL);
993 ok(pipe != INVALID_HANDLE_VALUE, "CreateFile for pipe failed with error %d\n", GetLastError());
995 ret = WriteFile(pipe, message, sizeof(message), &bytes_written, NULL);
996 ok(ret, "WriteFile failed with error %d\n", GetLastError());
998 ret = ReadFile(pipe, &dummy, sizeof(dummy), &bytes_read, NULL);
999 ok(ret, "ReadFile failed with error %d\n", GetLastError());
1005 ret = RevertToSelf();
1006 ok(ret, "RevertToSelf failed with error %d\n", GetLastError());
1010 ret = AdjustTokenPrivileges(params->token, TRUE, NULL, 0, NULL, NULL);
1011 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
1016 HANDLE process_token;
1018 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &process_token);
1019 ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
1021 ret = AdjustTokenPrivileges(process_token, FALSE, Privileges, 0, NULL, NULL);
1022 ok(ret, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
1024 HeapFree(GetProcessHeap(), 0, Privileges);
1026 CloseHandle(process_token);
1029 ret = WriteFile(pipe, message, sizeof(message), &bytes_written, NULL);
1030 ok(ret, "WriteFile failed with error %d\n", GetLastError());
1032 ret = ReadFile(pipe, &dummy, sizeof(dummy), &bytes_read, NULL);
1033 ok(ret, "ReadFile failed with error %d\n", GetLastError());
1040 static HANDLE make_impersonation_token(DWORD Access, SECURITY_IMPERSONATION_LEVEL ImpersonationLevel)
1042 HANDLE ProcessToken;
1043 HANDLE Token = NULL;
1046 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_DUPLICATE, &ProcessToken);
1047 ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
1049 ret = pDuplicateTokenEx(ProcessToken, Access, NULL, ImpersonationLevel, TokenImpersonation, &Token);
1050 ok(ret, "DuplicateToken failed with error %d\n", GetLastError());
1052 CloseHandle(ProcessToken);
1057 static void test_ImpersonateNamedPipeClient(HANDLE hClientToken, DWORD security_flags, BOOL revert, void (*test_func)(int, HANDLE))
1066 struct named_pipe_client_params params;
1068 DWORD dwBytesWritten;
1069 HANDLE hToken = NULL;
1070 SECURITY_IMPERSONATION_LEVEL ImpersonationLevel;
1073 hPipeServer = CreateNamedPipe(PIPE_NAME, PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_READMODE_BYTE | PIPE_WAIT, 1, 100, 100, NMPWAIT_USE_DEFAULT_WAIT, NULL);
1074 ok(hPipeServer != INVALID_HANDLE_VALUE, "CreateNamedPipe failed with error %d\n", GetLastError());
1076 params.security_flags = security_flags;
1077 params.token = hClientToken;
1078 params.revert = revert;
1079 hThread = CreateThread(NULL, 0, named_pipe_client_func, ¶ms, 0, &dwTid);
1080 ok(hThread != NULL, "CreateThread failed with error %d\n", GetLastError());
1082 SetLastError(0xdeadbeef);
1083 ret = ImpersonateNamedPipeClient(hPipeServer);
1084 error = GetLastError();
1085 ok(ret /* win2k3 */ || (error == ERROR_CANNOT_IMPERSONATE),
1086 "ImpersonateNamedPipeClient should have failed with ERROR_CANNOT_IMPERSONATE instead of %d\n", GetLastError());
1088 ret = ConnectNamedPipe(hPipeServer, NULL);
1089 ok(ret || (GetLastError() == ERROR_PIPE_CONNECTED), "ConnectNamedPipe failed with error %d\n", GetLastError());
1091 ret = ReadFile(hPipeServer, buffer, sizeof(buffer), &dwBytesRead, NULL);
1092 ok(ret, "ReadFile failed with error %d\n", GetLastError());
1094 ret = ImpersonateNamedPipeClient(hPipeServer);
1095 ok(ret, "ImpersonateNamedPipeClient failed with error %d\n", GetLastError());
1097 ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, FALSE, &hToken);
1098 ok(ret, "OpenThreadToken failed with error %d\n", GetLastError());
1100 (*test_func)(0, hToken);
1102 ImpersonationLevel = 0xdeadbeef; /* to avoid false positives */
1103 ret = GetTokenInformation(hToken, TokenImpersonationLevel, &ImpersonationLevel, sizeof(ImpersonationLevel), &size);
1104 ok(ret, "GetTokenInformation(TokenImpersonationLevel) failed with error %d\n", GetLastError());
1105 ok(ImpersonationLevel == SecurityImpersonation, "ImpersonationLevel should have been SecurityImpersonation(%d) instead of %d\n", SecurityImpersonation, ImpersonationLevel);
1107 CloseHandle(hToken);
1111 ret = WriteFile(hPipeServer, &dummy, sizeof(dummy), &dwBytesWritten, NULL);
1112 ok(ret, "WriteFile failed with error %d\n", GetLastError());
1114 ret = ReadFile(hPipeServer, buffer, sizeof(buffer), &dwBytesRead, NULL);
1115 ok(ret, "ReadFile failed with error %d\n", GetLastError());
1117 ret = ImpersonateNamedPipeClient(hPipeServer);
1118 ok(ret, "ImpersonateNamedPipeClient failed with error %d\n", GetLastError());
1120 ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, FALSE, &hToken);
1121 ok(ret, "OpenThreadToken failed with error %d\n", GetLastError());
1123 (*test_func)(1, hToken);
1125 CloseHandle(hToken);
1129 ret = WriteFile(hPipeServer, &dummy, sizeof(dummy), &dwBytesWritten, NULL);
1130 ok(ret, "WriteFile failed with error %d\n", GetLastError());
1132 WaitForSingleObject(hThread, INFINITE);
1134 ret = ImpersonateNamedPipeClient(hPipeServer);
1135 ok(ret, "ImpersonateNamedPipeClient failed with error %d\n", GetLastError());
1139 CloseHandle(hThread);
1140 CloseHandle(hPipeServer);
1143 static BOOL are_all_privileges_disabled(HANDLE hToken)
1146 TOKEN_PRIVILEGES *Privileges = NULL;
1148 BOOL all_privs_disabled = TRUE;
1151 ret = GetTokenInformation(hToken, TokenPrivileges, NULL, 0, &Size);
1152 if (!ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER)
1154 Privileges = HeapAlloc(GetProcessHeap(), 0, Size);
1155 ret = GetTokenInformation(hToken, TokenPrivileges, Privileges, Size, &Size);
1158 HeapFree(GetProcessHeap(), 0, Privileges);
1165 for (i = 0; i < Privileges->PrivilegeCount; i++)
1167 if (Privileges->Privileges[i].Attributes & SE_PRIVILEGE_ENABLED)
1169 all_privs_disabled = FALSE;
1174 HeapFree(GetProcessHeap(), 0, Privileges);
1176 return all_privs_disabled;
1179 static DWORD get_privilege_count(HANDLE hToken)
1181 TOKEN_STATISTICS Statistics;
1182 DWORD Size = sizeof(Statistics);
1185 ret = GetTokenInformation(hToken, TokenStatistics, &Statistics, Size, &Size);
1186 ok(ret, "GetTokenInformation(TokenStatistics)\n");
1187 if (!ret) return -1;
1189 return Statistics.PrivilegeCount;
1192 static void test_no_sqos_no_token(int call_index, HANDLE hToken)
1199 priv_count = get_privilege_count(hToken);
1201 ok(priv_count == 0, "privilege count should have been 0 instead of %d\n", priv_count);
1204 priv_count = get_privilege_count(hToken);
1205 ok(priv_count > 0, "privilege count should now be > 0 instead of 0\n");
1206 ok(!are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1209 ok(0, "shouldn't happen\n");
1213 static void test_no_sqos(int call_index, HANDLE hToken)
1218 ok(!are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1222 ok(are_all_privileges_disabled(hToken), "impersonated token should have been modified\n");
1225 ok(0, "shouldn't happen\n");
1229 static void test_static_context(int call_index, HANDLE hToken)
1234 ok(!are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1237 ok(!are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1240 ok(0, "shouldn't happen\n");
1244 static void test_dynamic_context(int call_index, HANDLE hToken)
1249 ok(!are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1253 ok(are_all_privileges_disabled(hToken), "impersonated token should have been modified\n");
1256 ok(0, "shouldn't happen\n");
1260 static void test_dynamic_context_no_token(int call_index, HANDLE hToken)
1265 ok(are_all_privileges_disabled(hToken), "token should be a copy of the process one\n");
1268 ok(!are_all_privileges_disabled(hToken), "process token modification should have been detected and impersonation token updated\n");
1271 ok(0, "shouldn't happen\n");
1275 static void test_no_sqos_revert(int call_index, HANDLE hToken)
1281 priv_count = get_privilege_count(hToken);
1283 ok(priv_count == 0, "privilege count should have been 0 instead of %d\n", priv_count);
1286 priv_count = get_privilege_count(hToken);
1287 ok(priv_count > 0, "privilege count should now be > 0 instead of 0\n");
1288 ok(!are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1291 ok(0, "shouldn't happen\n");
1295 static void test_static_context_revert(int call_index, HANDLE hToken)
1301 ok(are_all_privileges_disabled(hToken), "privileges should have been disabled\n");
1305 ok(are_all_privileges_disabled(hToken), "impersonated token should not have been modified\n");
1308 ok(0, "shouldn't happen\n");
1312 static void test_dynamic_context_revert(int call_index, HANDLE hToken)
1318 ok(are_all_privileges_disabled(hToken), "privileges should have been disabled\n");
1321 ok(!are_all_privileges_disabled(hToken), "impersonated token should now be process token\n");
1324 ok(0, "shouldn't happen\n");
1328 static void test_impersonation(void)
1330 HANDLE hClientToken;
1331 HANDLE hProcessToken;
1334 if( !pDuplicateTokenEx ) {
1335 skip("DuplicateTokenEx not found\n");
1339 ret = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hProcessToken);
1342 skip("couldn't open process token, skipping impersonation tests\n");
1346 if (!get_privilege_count(hProcessToken) || are_all_privileges_disabled(hProcessToken))
1348 skip("token didn't have any privileges or they were all disabled. token not suitable for impersonation tests\n");
1349 CloseHandle(hProcessToken);
1352 CloseHandle(hProcessToken);
1354 test_ImpersonateNamedPipeClient(NULL, 0, FALSE, test_no_sqos_no_token);
1355 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1356 test_ImpersonateNamedPipeClient(hClientToken, 0, FALSE, test_no_sqos);
1357 CloseHandle(hClientToken);
1358 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1359 test_ImpersonateNamedPipeClient(hClientToken,
1360 SECURITY_SQOS_PRESENT | SECURITY_IMPERSONATION, FALSE,
1361 test_static_context);
1362 CloseHandle(hClientToken);
1363 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1364 test_ImpersonateNamedPipeClient(hClientToken,
1365 SECURITY_SQOS_PRESENT | SECURITY_CONTEXT_TRACKING | SECURITY_IMPERSONATION,
1366 FALSE, test_dynamic_context);
1367 CloseHandle(hClientToken);
1368 test_ImpersonateNamedPipeClient(NULL,
1369 SECURITY_SQOS_PRESENT | SECURITY_CONTEXT_TRACKING | SECURITY_IMPERSONATION,
1370 FALSE, test_dynamic_context_no_token);
1372 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1373 test_ImpersonateNamedPipeClient(hClientToken, 0, TRUE, test_no_sqos_revert);
1374 CloseHandle(hClientToken);
1375 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1376 test_ImpersonateNamedPipeClient(hClientToken,
1377 SECURITY_SQOS_PRESENT | SECURITY_IMPERSONATION, TRUE,
1378 test_static_context_revert);
1379 CloseHandle(hClientToken);
1380 hClientToken = make_impersonation_token(TOKEN_IMPERSONATE | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, SecurityImpersonation);
1381 test_ImpersonateNamedPipeClient(hClientToken,
1382 SECURITY_SQOS_PRESENT | SECURITY_CONTEXT_TRACKING | SECURITY_IMPERSONATION,
1383 TRUE, test_dynamic_context_revert);
1384 CloseHandle(hClientToken);
1387 struct overlapped_server_args
1389 HANDLE pipe_created;
1392 static DWORD CALLBACK overlapped_server(LPVOID arg)
1397 struct overlapped_server_args *a = arg;
1401 pipe = CreateNamedPipeA("\\\\.\\pipe\\my pipe", FILE_FLAG_OVERLAPPED | PIPE_ACCESS_DUPLEX, PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE, 1, 0, 0, 100000, NULL);
1402 ok(pipe != NULL, "pipe NULL\n");
1404 ol.hEvent = CreateEventA(0, 1, 0, 0);
1405 ok(ol.hEvent != NULL, "event NULL\n");
1406 ret = ConnectNamedPipe(pipe, &ol);
1407 err = GetLastError();
1408 ok(ret == 0, "ret %d\n", ret);
1409 ok(err == ERROR_IO_PENDING, "gle %d\n", err);
1410 SetEvent(a->pipe_created);
1412 ret = WaitForSingleObjectEx(ol.hEvent, INFINITE, 1);
1413 ok(ret == WAIT_OBJECT_0, "ret %x\n", ret);
1415 ret = GetOverlappedResult(pipe, &ol, &num, 1);
1416 ok(ret == 1, "ret %d\n", ret);
1418 /* This should block */
1419 ret = ReadFile(pipe, buf, sizeof(buf), &num, NULL);
1420 ok(ret == 1, "ret %d\n", ret);
1422 DisconnectNamedPipe(pipe);
1423 CloseHandle(ol.hEvent);
1428 static void test_overlapped(void)
1431 HANDLE thread, pipe;
1433 struct overlapped_server_args args;
1435 args.pipe_created = CreateEventA(0, 1, 0, 0);
1436 thread = CreateThread(NULL, 0, overlapped_server, &args, 0, &tid);
1438 WaitForSingleObject(args.pipe_created, INFINITE);
1439 pipe = CreateFileA("\\\\.\\pipe\\my pipe", GENERIC_READ|GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
1440 ok(pipe != INVALID_HANDLE_VALUE, "cf failed\n");
1442 /* Sleep to try to get the ReadFile in the server to occur before the following WriteFile */
1445 ret = WriteFile(pipe, "x", 1, &num, NULL);
1446 ok(ret == 1, "ret %d\n", ret);
1448 WaitForSingleObject(thread, INFINITE);
1450 CloseHandle(args.pipe_created);
1451 CloseHandle(thread);
1454 static void test_NamedPipeHandleState(void)
1456 HANDLE server, client;
1458 DWORD state, instances, maxCollectionCount, collectDataTimeout;
1459 char userName[MAX_PATH];
1461 server = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX,
1462 /* dwOpenMode */ PIPE_TYPE_BYTE | PIPE_WAIT,
1463 /* nMaxInstances */ 1,
1464 /* nOutBufSize */ 1024,
1465 /* nInBufSize */ 1024,
1466 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
1467 /* lpSecurityAttrib */ NULL);
1468 ok(server != INVALID_HANDLE_VALUE, "cf failed\n");
1469 ret = GetNamedPipeHandleState(server, NULL, NULL, NULL, NULL, NULL, 0);
1471 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1472 ret = GetNamedPipeHandleState(server, &state, &instances, NULL, NULL, NULL,
1475 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1478 ok(state == 0, "unexpected state %08x\n", state);
1479 ok(instances == 1, "expected 1 instances, got %d\n", instances);
1481 /* Some parameters have no meaning, and therefore can't be retrieved,
1484 SetLastError(0xdeadbeef);
1485 ret = GetNamedPipeHandleState(server, &state, &instances,
1486 &maxCollectionCount, &collectDataTimeout, userName,
1487 sizeof(userName) / sizeof(userName[0]));
1489 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1490 "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1491 /* A byte-mode pipe server can't be changed to message mode. */
1492 state = PIPE_READMODE_MESSAGE;
1493 SetLastError(0xdeadbeef);
1494 ret = SetNamedPipeHandleState(server, &state, NULL, NULL);
1496 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1497 "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1499 client = CreateFileA(PIPENAME, GENERIC_READ|GENERIC_WRITE, 0, NULL,
1500 OPEN_EXISTING, 0, NULL);
1501 ok(client != INVALID_HANDLE_VALUE, "cf failed\n");
1503 state = PIPE_READMODE_BYTE;
1504 ret = SetNamedPipeHandleState(client, &state, NULL, NULL);
1506 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1507 /* A byte-mode pipe client can't be changed to message mode, either. */
1508 state = PIPE_READMODE_MESSAGE;
1509 SetLastError(0xdeadbeef);
1510 ret = SetNamedPipeHandleState(server, &state, NULL, NULL);
1512 ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1513 "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1515 CloseHandle(client);
1516 CloseHandle(server);
1518 server = CreateNamedPipe(PIPENAME, PIPE_ACCESS_DUPLEX,
1519 /* dwOpenMode */ PIPE_TYPE_MESSAGE | PIPE_WAIT,
1520 /* nMaxInstances */ 1,
1521 /* nOutBufSize */ 1024,
1522 /* nInBufSize */ 1024,
1523 /* nDefaultWait */ NMPWAIT_USE_DEFAULT_WAIT,
1524 /* lpSecurityAttrib */ NULL);
1525 ok(server != INVALID_HANDLE_VALUE, "cf failed\n");
1526 ret = GetNamedPipeHandleState(server, NULL, NULL, NULL, NULL, NULL, 0);
1528 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1529 ret = GetNamedPipeHandleState(server, &state, &instances, NULL, NULL, NULL,
1532 ok(ret, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1535 ok(state == 0, "unexpected state %08x\n", state);
1536 ok(instances == 1, "expected 1 instances, got %d\n", instances);
1538 /* In contrast to byte-mode pipes, a message-mode pipe server can be
1539 * changed to byte mode.
1541 state = PIPE_READMODE_BYTE;
1542 ret = SetNamedPipeHandleState(server, &state, NULL, NULL);
1544 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1546 client = CreateFileA(PIPENAME, GENERIC_READ|GENERIC_WRITE, 0, NULL,
1547 OPEN_EXISTING, 0, NULL);
1548 ok(client != INVALID_HANDLE_VALUE, "cf failed\n");
1550 state = PIPE_READMODE_MESSAGE;
1551 ret = SetNamedPipeHandleState(client, &state, NULL, NULL);
1553 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1554 /* A message-mode pipe client can also be changed to byte mode.
1556 state = PIPE_READMODE_BYTE;
1557 ret = SetNamedPipeHandleState(client, &state, NULL, NULL);
1559 ok(ret, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1561 CloseHandle(client);
1562 CloseHandle(server);
1569 hmod = GetModuleHandle("advapi32.dll");
1570 pDuplicateTokenEx = (void *) GetProcAddress(hmod, "DuplicateTokenEx");
1572 if (test_DisconnectNamedPipe())
1574 test_CreateNamedPipe_instances_must_match();
1576 test_CreateNamedPipe(PIPE_TYPE_BYTE);
1577 test_CreateNamedPipe(PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE);
1579 test_impersonation();
1581 test_NamedPipeHandleState();