| /* |
| * Mailslot regression test |
| * |
| * Copyright 2003 Mike McCormack |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| */ |
| |
| #include <stdarg.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| |
| #include <windef.h> |
| #include <winbase.h> |
| |
| #include "wine/test.h" |
| |
| static const char szmspath[] = "\\\\.\\mailslot\\wine_mailslot_test"; |
| |
| static int mailslot_test(void) |
| { |
| HANDLE hSlot, hSlot2, hWriter, hWriter2; |
| unsigned char buffer[16]; |
| DWORD count, dwMax, dwNext, dwMsgCount, dwTimeout; |
| |
| /* sanity check on GetMailslotInfo */ |
| dwMax = dwNext = dwMsgCount = dwTimeout = 0; |
| ok( !GetMailslotInfo( INVALID_HANDLE_VALUE, &dwMax, &dwNext, |
| &dwMsgCount, &dwTimeout ), "getmailslotinfo succeeded\n"); |
| |
| /* open a mailslot that doesn't exist */ |
| hWriter = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE, |
| FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter == INVALID_HANDLE_VALUE, "nonexistent mailslot\n"); |
| |
| /* open a mailslot without the right name */ |
| hSlot = CreateMailslot( "blah", 0, 0, NULL ); |
| ok( hSlot == INVALID_HANDLE_VALUE, |
| "Created mailslot with invalid name\n"); |
| ok( GetLastError() == ERROR_INVALID_NAME, |
| "error should be ERROR_INVALID_NAME\n"); |
| |
| /* open a mailslot with a null name */ |
| hSlot = CreateMailslot( NULL, 0, 0, NULL ); |
| ok( hSlot == INVALID_HANDLE_VALUE, |
| "Created mailslot with invalid name\n"); |
| ok( GetLastError() == ERROR_PATH_NOT_FOUND, |
| "error should be ERROR_PATH_NOT_FOUND\n"); |
| |
| /* valid open, but with wacky parameters ... then check them */ |
| hSlot = CreateMailslot( szmspath, -1, -1, NULL ); |
| ok( hSlot != INVALID_HANDLE_VALUE , "mailslot with valid name failed\n"); |
| dwMax = dwNext = dwMsgCount = dwTimeout = 0; |
| ok( GetMailslotInfo( hSlot, &dwMax, &dwNext, &dwMsgCount, &dwTimeout ), |
| "getmailslotinfo failed\n"); |
| ok( dwMax == ~0UL, "dwMax incorrect\n"); |
| ok( dwNext == MAILSLOT_NO_MESSAGE, "dwNext incorrect\n"); |
| ok( dwMsgCount == 0, "dwMsgCount incorrect\n"); |
| ok( dwTimeout == ~0UL, "dwTimeout incorrect\n"); |
| ok( GetMailslotInfo( hSlot, NULL, NULL, NULL, NULL ), |
| "getmailslotinfo failed\n"); |
| ok( CloseHandle(hSlot), "failed to close mailslot\n"); |
| |
| /* now open it for real */ |
| hSlot = CreateMailslot( szmspath, 0, 0, NULL ); |
| ok( hSlot != INVALID_HANDLE_VALUE , "valid mailslot failed\n"); |
| |
| /* try and read/write to it */ |
| count = 0; |
| memset(buffer, 0, sizeof buffer); |
| ok( !ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "slot read\n"); |
| ok( !WriteFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "slot write\n"); |
| |
| /* now try and openthe client, but with the wrong sharing mode */ |
| hWriter = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE, |
| 0, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter == INVALID_HANDLE_VALUE, "bad sharing mode\n"); |
| ok( GetLastError() == ERROR_SHARING_VIOLATION, |
| "error should be ERROR_SHARING_VIOLATION\n"); |
| |
| /* now open the client with the correct sharing mode */ |
| hWriter = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE, |
| FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter != INVALID_HANDLE_VALUE, "existing mailslot\n"); |
| |
| /* |
| * opening a client should make no difference to |
| * whether we can read or write the mailslot |
| */ |
| ok( !ReadFile( hSlot, buffer, sizeof buffer/2, &count, NULL), |
| "slot read\n"); |
| ok( !WriteFile( hSlot, buffer, sizeof buffer/2, &count, NULL), |
| "slot write\n"); |
| |
| /* |
| * we can't read from this client, |
| * but we should be able to write to it |
| */ |
| ok( !ReadFile( hWriter, buffer, sizeof buffer/2, &count, NULL), |
| "can read client\n"); |
| ok( WriteFile( hWriter, buffer, sizeof buffer/2, &count, NULL), |
| "can't write client\n"); |
| ok( !ReadFile( hWriter, buffer, sizeof buffer/2, &count, NULL), |
| "can read client\n"); |
| |
| /* |
| * seeing as there's something in the slot, |
| * we should be able to read it once |
| */ |
| ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "slot read\n"); |
| ok( count == (sizeof buffer/2), "short read\n" ); |
| |
| /* but not again */ |
| ok( !ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "slot read\n"); |
| |
| /* now try open another writer... should fail */ |
| hWriter2 = CreateFile(szmspath, GENERIC_READ|GENERIC_WRITE, |
| FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter2 == INVALID_HANDLE_VALUE, "two writers\n"); |
| |
| /* now try open another as a reader ... also fails */ |
| hWriter2 = CreateFile(szmspath, GENERIC_READ, |
| FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter2 == INVALID_HANDLE_VALUE, "writer + reader\n"); |
| |
| /* now try open another as a writer ... still fails */ |
| hWriter2 = CreateFile(szmspath, GENERIC_WRITE, |
| FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter2 == INVALID_HANDLE_VALUE, "writer\n"); |
| |
| /* now open another one */ |
| hSlot2 = CreateMailslot( szmspath, 0, 0, NULL ); |
| ok( hSlot2 == INVALID_HANDLE_VALUE , "opened two mailslots\n"); |
| |
| /* close the client again */ |
| ok( CloseHandle( hWriter ), "closing the client\n"); |
| |
| /* |
| * now try reopen it with slightly different permissions ... |
| * shared writing |
| */ |
| hWriter = CreateFile(szmspath, GENERIC_WRITE, |
| FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter != INVALID_HANDLE_VALUE, "sharing writer\n"); |
| |
| /* |
| * now try open another as a writer ... |
| * but don't share with the first ... fail |
| */ |
| hWriter2 = CreateFile(szmspath, GENERIC_WRITE, |
| FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter2 == INVALID_HANDLE_VALUE, "greedy writer succeeded\n"); |
| |
| /* now try open another as a writer ... and share with the first */ |
| hWriter2 = CreateFile(szmspath, GENERIC_WRITE, |
| FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL); |
| ok( hWriter2 != INVALID_HANDLE_VALUE, "2nd sharing writer\n"); |
| |
| /* check the mailslot info */ |
| dwMax = dwNext = dwMsgCount = dwTimeout = 0; |
| ok( GetMailslotInfo( hSlot, &dwMax, &dwNext, &dwMsgCount, &dwTimeout ), |
| "getmailslotinfo failed\n"); |
| ok( dwNext == MAILSLOT_NO_MESSAGE, "dwNext incorrect\n"); |
| ok( dwMax == 0, "dwMax incorrect\n"); |
| ok( dwMsgCount == 0, "dwMsgCount incorrect\n"); |
| ok( dwTimeout == 0, "dwTimeout incorrect\n"); |
| |
| /* check there's still no data */ |
| ok( !ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), "slot read\n"); |
| |
| /* write two messages */ |
| buffer[0] = 'a'; |
| ok( WriteFile( hWriter, buffer, 1, &count, NULL), "1st write failed\n"); |
| |
| /* check the mailslot info */ |
| dwNext = dwMsgCount = 0; |
| ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ), |
| "getmailslotinfo failed\n"); |
| ok( dwNext == 1, "dwNext incorrect\n"); |
| ok( dwMsgCount == 1, "dwMsgCount incorrect\n"); |
| |
| buffer[0] = 'b'; |
| buffer[1] = 'c'; |
| ok( WriteFile( hWriter2, buffer, 2, &count, NULL), "2nd write failed\n"); |
| |
| /* check the mailslot info */ |
| dwNext = dwMsgCount = 0; |
| ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ), |
| "getmailslotinfo failed\n"); |
| ok( dwNext == 1, "dwNext incorrect\n"); |
| todo_wine { |
| ok( dwMsgCount == 2, "dwMsgCount incorrect\n"); |
| } |
| |
| /* write a 3rd message with zero size */ |
| ok( WriteFile( hWriter2, buffer, 0, &count, NULL), "3rd write failed\n"); |
| |
| /* check the mailslot info */ |
| dwNext = dwMsgCount = 0; |
| ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ), |
| "getmailslotinfo failed\n"); |
| ok( dwNext == 1, "dwNext incorrect\n"); |
| todo_wine { |
| ok( dwMsgCount == 3, "dwMsgCount incorrect\n"); |
| } |
| |
| buffer[0]=buffer[1]=0; |
| |
| /* |
| * then check that they come out with the correct order and size, |
| * then the slot is empty |
| */ |
| ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "1st slot read failed\n"); |
| ok( count == 1, "failed to get 1st message\n"); |
| ok( buffer[0] == 'a', "1st message wrong\n"); |
| |
| /* check the mailslot info */ |
| dwNext = dwMsgCount = 0; |
| ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ), |
| "getmailslotinfo failed\n"); |
| ok( dwNext == 2, "dwNext incorrect\n"); |
| todo_wine { |
| ok( dwMsgCount == 2, "dwMsgCount incorrect\n"); |
| } |
| |
| /* read the second message */ |
| ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "2nd slot read failed\n"); |
| ok( count == 2, "failed to get 2nd message\n"); |
| ok( ( buffer[0] == 'b' ) && ( buffer[1] == 'c' ), "2nd message wrong\n"); |
| |
| /* check the mailslot info */ |
| dwNext = dwMsgCount = 0; |
| ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ), |
| "getmailslotinfo failed\n"); |
| todo_wine { |
| ok( dwNext == 0, "dwNext incorrect\n"); |
| ok( dwMsgCount == 1, "dwMsgCount incorrect\n"); |
| } |
| |
| /* read the 3rd (zero length) message */ |
| todo_wine { |
| ok( ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "3rd slot read failed\n"); |
| } |
| ok( count == 0, "failed to get 3rd message\n"); |
| |
| /* |
| * now there should be no more messages |
| * check the mailslot info |
| */ |
| dwNext = dwMsgCount = 0; |
| ok( GetMailslotInfo( hSlot, NULL, &dwNext, &dwMsgCount, NULL ), |
| "getmailslotinfo failed\n"); |
| ok( dwNext == MAILSLOT_NO_MESSAGE, "dwNext incorrect\n"); |
| ok( dwMsgCount == 0, "dwMsgCount incorrect\n"); |
| |
| /* check that reads fail */ |
| ok( !ReadFile( hSlot, buffer, sizeof buffer, &count, NULL), |
| "3rd slot read succeeded\n"); |
| |
| /* finally close the mailslot and its client */ |
| ok( CloseHandle( hWriter2 ), "closing 2nd client\n"); |
| ok( CloseHandle( hWriter ), "closing the client\n"); |
| ok( CloseHandle( hSlot ), "closing the mailslot\n"); |
| |
| return 0; |
| } |
| |
| START_TEST(mailslot) |
| { |
| mailslot_test(); |
| } |