server: Debugging our own process is disallowed.
[wine] / dlls / kernel32 / tests / mailslot.c
1 /*
2  *  Mailslot regression test
3  *
4  *  Copyright 2003 Mike McCormack
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdarg.h>
22 #include <stdlib.h>
23 #include <stdio.h>
24
25 #include <windef.h>
26 #include <winbase.h>
27
28 #include "wine/test.h"
29
30 static const char szmspath[] = "\\\\.\\mailslot\\wine_mailslot_test";
31
32 static int mailslot_test(void)
33 {
34     HANDLE hSlot, hSlot2, hWriter, hWriter2;
35     unsigned char buffer[16];
36     DWORD count, dwMax, dwNext, dwMsgCount, dwTimeout;
37     BOOL ret;
38
39     /* sanity check on GetMailslotInfo */
40     dwMax = dwNext = dwMsgCount = dwTimeout = 0;
41     ok( !GetMailslotInfo( INVALID_HANDLE_VALUE, &dwMax, &dwNext,
42             &dwMsgCount, &dwTimeout ), "getmailslotinfo succeeded\n");
43
44     /* open a mailslot that doesn't exist */
45     hWriter = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE,
46                              FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
47     ok( hWriter == INVALID_HANDLE_VALUE, "nonexistent mailslot\n");
48
49     /* open a mailslot without the right name */
50     hSlot = CreateMailslot( "blah", 0, 0, NULL );
51     ok( hSlot == INVALID_HANDLE_VALUE,
52             "Created mailslot with invalid name\n");
53     ok( GetLastError() == ERROR_INVALID_NAME,
54             "error should be ERROR_INVALID_NAME\n");
55
56     /* open a mailslot with a null name */
57     hSlot = CreateMailslot( NULL, 0, 0, NULL );
58     ok( hSlot == INVALID_HANDLE_VALUE || broken(hSlot != INVALID_HANDLE_VALUE), /* win9x */
59         "Created mailslot with invalid name\n");
60     if (hSlot == INVALID_HANDLE_VALUE)
61         ok( GetLastError() == ERROR_PATH_NOT_FOUND,
62             "error should be ERROR_PATH_NOT_FOUND\n");
63     else  /* succeeds on win9x */
64         CloseHandle( hSlot );
65
66     /* valid open, but with wacky parameters ... then check them */
67     hSlot = CreateMailslot( szmspath, -1, -1, NULL );
68     ok( hSlot != INVALID_HANDLE_VALUE , "mailslot with valid name failed\n");
69     dwMax = dwNext = dwMsgCount = dwTimeout = 0;
70     ok( GetMailslotInfo( hSlot, &dwMax, &dwNext, &dwMsgCount, &dwTimeout ),
71            "getmailslotinfo failed\n");
72     ok( dwMax == ~0U, "dwMax incorrect\n");
73     ok( dwNext == MAILSLOT_NO_MESSAGE, "dwNext incorrect\n");
74     ok( dwMsgCount == 0, "dwMsgCount incorrect\n");
75     ok( dwTimeout == ~0U, "dwTimeout incorrect\n");
76     ok( GetMailslotInfo( hSlot, NULL, NULL, NULL, NULL ),
77             "getmailslotinfo failed\n");
78     ok( CloseHandle(hSlot), "failed to close mailslot\n");
79
80     /* now open it for real */
81     hSlot = CreateMailslot( szmspath, 0, 0, NULL );
82     ok( hSlot != INVALID_HANDLE_VALUE , "valid mailslot failed\n");
83
84     /* try and read/write to it */
85     count = 0;
86     memset(buffer, 0, sizeof buffer);
87     ret = ReadFile( hSlot, buffer, sizeof buffer, &count, NULL);
88     ok( !ret || broken(ret), /* win9x */ "slot read\n");
89     if (!ret) ok( GetLastError() == ERROR_SEM_TIMEOUT, "wrong error %u\n", GetLastError() );
90     else ok( count == 0, "wrong count %u\n", count );
91     ok( !WriteFile( hSlot, buffer, sizeof buffer, &count, NULL),
92             "slot write\n");
93     ok( GetLastError() == ERROR_ACCESS_DENIED, "wrong error %u\n", GetLastError() );
94
95     /* now try and open the client, but with the wrong sharing mode */
96     hWriter = CreateFile(szmspath, GENERIC_WRITE,
97                              0, NULL, OPEN_EXISTING, 0, NULL);
98     ok( hWriter != INVALID_HANDLE_VALUE /* vista */ || GetLastError() == ERROR_SHARING_VIOLATION,
99         "error should be ERROR_SHARING_VIOLATION got %p / %u\n", hWriter, GetLastError());
100     if (hWriter != INVALID_HANDLE_VALUE) CloseHandle( hWriter );
101
102     /* now open the client with the correct sharing mode */
103     hWriter = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE,
104                              FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
105     if (hWriter == INVALID_HANDLE_VALUE)  /* win9x doesn't like GENERIC_READ */
106         hWriter = CreateFile(szmspath, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
107     ok( hWriter != INVALID_HANDLE_VALUE, "existing mailslot err %u\n", GetLastError());
108
109     /*
110      * opening a client should make no difference to
111      * whether we can read or write the mailslot
112      */
113     ret = ReadFile( hSlot, buffer, sizeof buffer/2, &count, NULL);
114     ok( !ret || broken(ret), /* win9x */ "slot read\n");
115     if (!ret) ok( GetLastError() == ERROR_SEM_TIMEOUT, "wrong error %u\n", GetLastError() );
116     else ok( count == 0, "wrong count %u\n", count );
117     ok( !WriteFile( hSlot, buffer, sizeof buffer/2, &count, NULL),
118             "slot write\n");
119     ok( GetLastError() == ERROR_ACCESS_DENIED, "wrong error %u\n", GetLastError() );
120
121     /*
122      * we can't read from this client, 
123      * but we should be able to write to it
124      */
125     ok( !ReadFile( hWriter, buffer, sizeof buffer/2, &count, NULL),
126             "can read client\n");
127     ok( GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_ACCESS_DENIED,
128             "wrong error %u\n", GetLastError() );
129     ok( WriteFile( hWriter, buffer, sizeof buffer/2, &count, NULL),
130             "can't write client\n");
131     ok( !ReadFile( hWriter, buffer, sizeof buffer/2, &count, NULL),
132             "can read client\n");
133     ok( GetLastError() == ERROR_INVALID_PARAMETER || GetLastError() == ERROR_ACCESS_DENIED,
134             "wrong error %u\n", GetLastError() );
135
136     /*
137      * seeing as there's something in the slot,
138      * we should be able to read it once
139      */
140     ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL),
141             "slot read\n");
142     ok( count == (sizeof buffer/2), "short read\n" );
143
144     /* but not again */
145     ret = ReadFile( hSlot, buffer, sizeof buffer, &count, NULL);
146     ok( !ret || broken(ret), /* win9x */ "slot read\n");
147     if (!ret) ok( GetLastError() == ERROR_SEM_TIMEOUT, "wrong error %u\n", GetLastError() );
148     else ok( count == 0, "wrong count %u\n", count );
149
150     /* now try open another writer... should fail */
151     hWriter2 = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE,
152                      FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
153     /* succeeds on vista, don't test */
154     if (hWriter2 != INVALID_HANDLE_VALUE) CloseHandle( hWriter2 );
155
156     /* now try open another as a reader ... also fails */
157     hWriter2 = CreateFile(szmspath, GENERIC_READ,
158                      FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
159     /* succeeds on vista, don't test */
160     if (hWriter2 != INVALID_HANDLE_VALUE) CloseHandle( hWriter2 );
161
162     /* now try open another as a writer ... still fails */
163     hWriter2 = CreateFile(szmspath, GENERIC_WRITE,
164                      FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
165     /* succeeds on vista, don't test */
166     if (hWriter2 != INVALID_HANDLE_VALUE) CloseHandle( hWriter2 );
167
168     /* now open another one */
169     hSlot2 = CreateMailslot( szmspath, 0, 0, NULL );
170     ok( hSlot2 == INVALID_HANDLE_VALUE , "opened two mailslots\n");
171
172     /* close the client again */
173     ok( CloseHandle( hWriter ), "closing the client\n");
174
175     /*
176      * now try reopen it with slightly different permissions ...
177      * shared writing
178      */
179     hWriter = CreateFile(szmspath, GENERIC_WRITE,
180               FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
181     ok( hWriter != INVALID_HANDLE_VALUE, "sharing writer\n");
182
183     /*
184      * now try open another as a writer ...
185      * but don't share with the first ... fail
186      */
187     hWriter2 = CreateFile(szmspath, GENERIC_WRITE,
188                      FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
189     /* succeeds on vista, don't test */
190     if (hWriter2 != INVALID_HANDLE_VALUE) CloseHandle( hWriter2 );
191
192     /* now try open another as a writer ... and share with the first */
193     hWriter2 = CreateFile(szmspath, GENERIC_WRITE,
194               FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
195     ok( hWriter2 != INVALID_HANDLE_VALUE, "2nd sharing writer\n");
196
197     /* check the mailslot info */
198     dwMax = dwNext = dwMsgCount = dwTimeout = 0;
199     ok( GetMailslotInfo( hSlot, &dwMax, &dwNext, &dwMsgCount, &dwTimeout ),
200         "getmailslotinfo failed\n");
201     ok( dwNext == MAILSLOT_NO_MESSAGE, "dwNext incorrect\n");
202     ok( dwMax == 0, "dwMax incorrect\n");
203     ok( dwMsgCount == 0, "dwMsgCount incorrect\n");
204     ok( dwTimeout == 0, "dwTimeout incorrect\n");
205
206     /* check there's still no data */
207     ret = ReadFile( hSlot, buffer, sizeof buffer, &count, NULL);
208     ok( !ret || broken(ret), /* win9x */ "slot read\n");
209     if (!ret) ok( GetLastError() == ERROR_SEM_TIMEOUT, "wrong error %u\n", GetLastError() );
210     else ok( count == 0, "wrong count %u\n", count );
211
212     /* write two messages */
213     buffer[0] = 'a';
214     ok( WriteFile( hWriter, buffer, 1, &count, NULL), "1st write failed\n");
215
216     /* check the mailslot info */
217     dwNext = dwMsgCount = 0;
218     ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ),
219         "getmailslotinfo failed\n");
220     ok( dwNext == 1, "dwNext incorrect\n");
221     ok( dwMsgCount == 1, "dwMsgCount incorrect\n");
222
223     buffer[0] = 'b';
224     buffer[1] = 'c';
225     ok( WriteFile( hWriter2, buffer, 2, &count, NULL), "2nd write failed\n");
226
227     /* check the mailslot info */
228     dwNext = dwMsgCount = 0;
229     ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ),
230         "getmailslotinfo failed\n");
231     ok( dwNext == 1, "dwNext incorrect\n");
232     todo_wine {
233     ok( dwMsgCount == 2, "dwMsgCount incorrect\n");
234     }
235
236     /* write a 3rd message with zero size */
237     ok( WriteFile( hWriter2, buffer, 0, &count, NULL), "3rd write failed\n");
238
239     /* check the mailslot info */
240     dwNext = dwMsgCount = 0;
241     ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ),
242         "getmailslotinfo failed\n");
243     ok( dwNext == 1, "dwNext incorrect\n");
244     todo_wine
245         ok( dwMsgCount == 3 || broken(dwMsgCount == 2), /* win9x */
246             "dwMsgCount incorrect %u\n", dwMsgCount);
247
248     buffer[0]=buffer[1]=0;
249
250     /*
251      * then check that they come out with the correct order and size,
252      * then the slot is empty
253      */
254     ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL),
255         "1st slot read failed\n");
256     ok( count == 1, "failed to get 1st message\n");
257     ok( buffer[0] == 'a', "1st message wrong\n");
258
259     /* check the mailslot info */
260     dwNext = dwMsgCount = 0;
261     ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ),
262         "getmailslotinfo failed\n");
263     ok( dwNext == 2, "dwNext incorrect\n");
264     todo_wine {
265         ok( dwMsgCount == 2 || broken(dwMsgCount == 1), /* win9x */
266             "dwMsgCount incorrect %u\n", dwMsgCount);
267     }
268
269     /* read the second message */
270     ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL),
271         "2nd slot read failed\n");
272     ok( count == 2, "failed to get 2nd message\n");
273     ok( ( buffer[0] == 'b' ) && ( buffer[1] == 'c' ), "2nd message wrong\n");
274
275     /* check the mailslot info */
276     dwNext = dwMsgCount = 0;
277     ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ),
278         "getmailslotinfo failed\n");
279     ok( dwNext == 0 || broken(dwNext == ~0u), /* win9x */ "dwNext incorrect %u\n", dwNext);
280     todo_wine {
281         ok( dwMsgCount == 1 || broken(dwMsgCount == 0), /* win9x */
282             "dwMsgCount incorrect %u\n", dwMsgCount);
283     }
284
285     /* read the 3rd (zero length) message */
286     todo_wine {
287     ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL),
288         "3rd slot read failed\n");
289     }
290     ok( count == 0, "failed to get 3rd message\n");
291
292     /*
293      * now there should be no more messages
294      * check the mailslot info
295      */
296     dwNext = dwMsgCount = 0;
297     ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ),
298         "getmailslotinfo failed\n");
299     ok( dwNext == MAILSLOT_NO_MESSAGE, "dwNext incorrect\n");
300     ok( dwMsgCount == 0, "dwMsgCount incorrect\n");
301
302     /* check that reads fail */
303     ret = ReadFile( hSlot, buffer, sizeof buffer, &count, NULL);
304     ok( !ret || broken(ret), /* win9x */ "3rd slot read succeeded\n");
305     if (!ret) ok( GetLastError() == ERROR_SEM_TIMEOUT, "wrong error %u\n", GetLastError() );
306     else ok( count == 0, "wrong count %u\n", count );
307
308     /* finally close the mailslot and its client */
309     ok( CloseHandle( hWriter2 ), "closing 2nd client\n");
310     ok( CloseHandle( hWriter ), "closing the client\n");
311     ok( CloseHandle( hSlot ), "closing the mailslot\n");
312
313     /* test timeouts */
314     hSlot = CreateMailslot( szmspath, 0, 1000, NULL );
315     ok( hSlot != INVALID_HANDLE_VALUE , "valid mailslot failed\n");
316     count = 0;
317     memset(buffer, 0, sizeof buffer);
318     dwTimeout = GetTickCount();
319     ok( !ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), "slot read\n");
320     ok( GetLastError() == ERROR_SEM_TIMEOUT || broken(GetLastError() == ERROR_ACCESS_DENIED), /* win9x */
321         "wrong error %u\n", GetLastError() );
322     dwTimeout = GetTickCount() - dwTimeout;
323     ok( dwTimeout >= 990, "timeout too short %u\n", dwTimeout );
324     ok( CloseHandle( hSlot ), "closing the mailslot\n");
325
326     return 0;
327 }
328
329 START_TEST(mailslot)
330 {
331     mailslot_test();
332 }