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
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");
609 static void exercizeServer(const char *pipename
, HANDLE serverThread
)
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";
622 for (loop
= 0; loop
< 3; loop
++) {
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
)
630 err
= GetLastError();
632 ok(err
== ERROR_PIPE_BUSY
|| err
== ERROR_FILE_NOT_FOUND
, "connecting to pipe\n");
634 ok(err
== ERROR_PIPE_BUSY
, "connecting to pipe\n");
635 trace("connect failed, retrying\n");
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");
656 trace("exercizeServer returning\n");
659 static void test_NamedPipe_2(void)
662 DWORD serverThreadId
;
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.
677 serverThread
= CreateThread(NULL
, 0, serverThreadMain1
, (void *)8, 0, &serverThreadId
);
678 ok(serverThread
!= INVALID_HANDLE_VALUE
, "CreateThread\n");
679 exercizeServer(PIPENAME
"serverThreadMain1", serverThread
);
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 */
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)
703 static const char obuf
[] = "Bit Bucket";
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");
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");
754 static void test_CreatePipe(void)
756 SECURITY_ATTRIBUTES pipe_attr
;
757 HANDLE piperead
, pipewrite
;
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 */
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
;
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
;
820 const char message
[] = "Test";
821 DWORD bytes_read
, bytes_written
;
823 TOKEN_PRIVILEGES
*Privileges
= NULL
;
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());
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());
870 ret
= RevertToSelf();
871 ok(ret
, "RevertToSelf failed with error %d\n", GetLastError());
875 ret
= AdjustTokenPrivileges(params
->token
, TRUE
, NULL
, 0, NULL
, NULL
);
876 ok(ret
, "AdjustTokenPrivileges failed with error %d\n", GetLastError());
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());
905 static HANDLE
make_impersonation_token(DWORD Access
, SECURITY_IMPERSONATION_LEVEL ImpersonationLevel
)
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
);
922 static void test_ImpersonateNamedPipeClient(HANDLE hClientToken
, DWORD security_flags
, BOOL revert
, void (*test_func
)(int, HANDLE
))
931 struct named_pipe_client_params params
;
933 DWORD dwBytesWritten
;
934 HANDLE hToken
= NULL
;
935 SECURITY_IMPERSONATION_LEVEL ImpersonationLevel
;
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
, ¶ms
, 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
);
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
);
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());
1004 CloseHandle(hThread
);
1005 CloseHandle(hPipeServer
);
1008 static BOOL
are_all_privileges_disabled(HANDLE hToken
)
1011 TOKEN_PRIVILEGES
*Privileges
= NULL
;
1013 BOOL all_privs_disabled
= TRUE
;
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
);
1023 HeapFree(GetProcessHeap(), 0, Privileges
);
1030 for (i
= 0; i
< Privileges
->PrivilegeCount
; i
++)
1032 if (Privileges
->Privileges
[i
].Attributes
& SE_PRIVILEGE_ENABLED
)
1034 all_privs_disabled
= FALSE
;
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
);
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
)
1064 priv_count
= get_privilege_count(hToken
);
1066 ok(priv_count
== 0, "privilege count should have been 0 instead of %d\n", priv_count
);
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");
1074 ok(0, "shouldn't happen\n");
1078 static void test_no_sqos(int call_index
, HANDLE hToken
)
1083 ok(!are_all_privileges_disabled(hToken
), "token should be a copy of the process one\n");
1087 ok(are_all_privileges_disabled(hToken
), "impersonated token should have been modified\n");
1090 ok(0, "shouldn't happen\n");
1094 static void test_static_context(int call_index
, HANDLE hToken
)
1099 ok(!are_all_privileges_disabled(hToken
), "token should be a copy of the process one\n");
1102 ok(!are_all_privileges_disabled(hToken
), "impersonated token should not have been modified\n");
1105 ok(0, "shouldn't happen\n");
1109 static void test_dynamic_context(int call_index
, HANDLE hToken
)
1114 ok(!are_all_privileges_disabled(hToken
), "token should be a copy of the process one\n");
1118 ok(are_all_privileges_disabled(hToken
), "impersonated token should have been modified\n");
1121 ok(0, "shouldn't happen\n");
1125 static void test_dynamic_context_no_token(int call_index
, HANDLE hToken
)
1130 ok(are_all_privileges_disabled(hToken
), "token should be a copy of the process one\n");
1133 ok(!are_all_privileges_disabled(hToken
), "process token modification should have been detected and impersonation token updated\n");
1136 ok(0, "shouldn't happen\n");
1140 static void test_no_sqos_revert(int call_index
, HANDLE hToken
)
1146 priv_count
= get_privilege_count(hToken
);
1148 ok(priv_count
== 0, "privilege count should have been 0 instead of %d\n", priv_count
);
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");
1156 ok(0, "shouldn't happen\n");
1160 static void test_static_context_revert(int call_index
, HANDLE hToken
)
1166 ok(are_all_privileges_disabled(hToken
), "privileges should have been disabled\n");
1170 ok(are_all_privileges_disabled(hToken
), "impersonated token should not have been modified\n");
1173 ok(0, "shouldn't happen\n");
1177 static void test_dynamic_context_revert(int call_index
, HANDLE hToken
)
1183 ok(are_all_privileges_disabled(hToken
), "privileges should have been disabled\n");
1186 ok(!are_all_privileges_disabled(hToken
), "impersonated token should now be process token\n");
1189 ok(0, "shouldn't happen\n");
1193 static void test_impersonation(void)
1195 HANDLE hClientToken
;
1196 HANDLE hProcessToken
;
1199 if( !pDuplicateTokenEx
) {
1200 skip("DuplicateTokenEx not found\n");
1204 ret
= OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY
, &hProcessToken
);
1207 skip("couldn't open process token, skipping impersonation tests\n");
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
);
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
)
1262 struct overlapped_server_args
*a
= (struct overlapped_server_args
*)arg
;
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
);
1293 static void test_overlapped(void)
1296 HANDLE thread
, pipe
;
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 */
1310 ret
= WriteFile(pipe
, "x", 1, &num
, NULL
);
1311 ok(ret
== 1, "ret %d\n", ret
);
1313 WaitForSingleObject(thread
, INFINITE
);
1315 CloseHandle(args
.pipe_created
);
1316 CloseHandle(thread
);
1319 static void test_NamedPipeHandleState(void)
1321 HANDLE server
, client
;
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);
1336 ok(ret
, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1337 ret
= GetNamedPipeHandleState(server
, &state
, &instances
, NULL
, NULL
, NULL
,
1340 ok(ret
, "GetNamedPipeHandleState failed: %d\n", GetLastError());
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,
1349 SetLastError(0xdeadbeef);
1350 ret
= GetNamedPipeHandleState(server
, &state
, &instances
,
1351 &maxCollectionCount
, &collectDataTimeout
, userName
,
1352 sizeof(userName
) / sizeof(userName
[0]));
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
);
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
);
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
);
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);
1393 ok(ret
, "GetNamedPipeHandleState failed: %d\n", GetLastError());
1394 ret
= GetNamedPipeHandleState(server
, &state
, &instances
, NULL
, NULL
, NULL
,
1397 ok(ret
, "GetNamedPipeHandleState failed: %d\n", GetLastError());
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
);
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
);
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
);
1424 ok(ret
, "SetNamedPipeHandleState failed: %d\n", GetLastError());
1426 CloseHandle(client
);
1427 CloseHandle(server
);
1434 hmod
= GetModuleHandle("advapi32.dll");
1435 pDuplicateTokenEx
= (void *) GetProcAddress(hmod
, "DuplicateTokenEx");
1437 if (test_DisconnectNamedPipe())
1439 test_CreateNamedPipe_instances_must_match();
1441 test_CreateNamedPipe(PIPE_TYPE_BYTE
);
1442 test_CreateNamedPipe(PIPE_TYPE_MESSAGE
| PIPE_READMODE_MESSAGE
);
1444 test_impersonation();
1446 test_NamedPipeHandleState();