mirror of
https://github.com/reactos/reactos.git
synced 2024-08-20 02:13:29 +00:00
205 lines
7.1 KiB
C
205 lines
7.1 KiB
C
/*
|
|
* Unit test suite for protocol functions
|
|
*
|
|
* Copyright 2004 Hans Leidekker
|
|
*
|
|
* 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include <windef.h>
|
|
#include <winbase.h>
|
|
#include <winsock2.h>
|
|
|
|
#include "wine/test.h"
|
|
|
|
/* TCP and UDP over IP fixed set of service flags */
|
|
#define TCPIP_SERVICE_FLAGS (XP1_GUARANTEED_DELIVERY \
|
|
| XP1_GUARANTEED_ORDER \
|
|
| XP1_GRACEFUL_CLOSE \
|
|
| XP1_EXPEDITED_DATA \
|
|
| XP1_IFS_HANDLES)
|
|
|
|
#define UDPIP_SERVICE_FLAGS (XP1_CONNECTIONLESS \
|
|
| XP1_MESSAGE_ORIENTED \
|
|
| XP1_SUPPORT_BROADCAST \
|
|
| XP1_SUPPORT_MULTIPOINT \
|
|
| XP1_IFS_HANDLES)
|
|
|
|
static void test_service_flags(int family, int version, int socktype, int protocol, DWORD testflags)
|
|
{
|
|
DWORD expectedflags = 0;
|
|
if (socktype == SOCK_STREAM && protocol == IPPROTO_TCP)
|
|
expectedflags = TCPIP_SERVICE_FLAGS;
|
|
if (socktype == SOCK_DGRAM && protocol == IPPROTO_UDP)
|
|
expectedflags = UDPIP_SERVICE_FLAGS;
|
|
|
|
/* check if standard TCP and UDP protocols are offering the correct service flags */
|
|
if ((family == AF_INET || family == AF_INET6) && version == 2 && expectedflags)
|
|
{
|
|
/* QOS may or may not be installed */
|
|
testflags &= ~XP1_QOS_SUPPORTED;
|
|
ok(expectedflags == testflags,
|
|
"Incorrect flags, expected 0x%x, received 0x%x\n",
|
|
expectedflags, testflags);
|
|
}
|
|
}
|
|
|
|
static void test_WSAEnumProtocolsA(void)
|
|
{
|
|
INT ret, i, j, found;
|
|
DWORD len = 0, error;
|
|
WSAPROTOCOL_INFOA info, *buffer;
|
|
INT ptest[] = {0xdead, IPPROTO_TCP, 0xcafe, IPPROTO_UDP, 0xbeef, 0};
|
|
|
|
ret = WSAEnumProtocolsA( NULL, NULL, &len );
|
|
ok( ret == SOCKET_ERROR, "WSAEnumProtocolsA() succeeded unexpectedly\n");
|
|
error = WSAGetLastError();
|
|
ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
|
|
|
|
len = 0;
|
|
|
|
ret = WSAEnumProtocolsA( NULL, &info, &len );
|
|
ok( ret == SOCKET_ERROR, "WSAEnumProtocolsA() succeeded unexpectedly\n");
|
|
error = WSAGetLastError();
|
|
ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
|
|
|
|
buffer = HeapAlloc( GetProcessHeap(), 0, len );
|
|
|
|
if (buffer)
|
|
{
|
|
ret = WSAEnumProtocolsA( NULL, buffer, &len );
|
|
ok( ret != SOCKET_ERROR, "WSAEnumProtocolsA() failed unexpectedly: %d\n",
|
|
WSAGetLastError() );
|
|
|
|
for (i = 0; i < ret; i++)
|
|
{
|
|
ok( strlen( buffer[i].szProtocol ), "No protocol name found\n" );
|
|
test_service_flags( buffer[i].iAddressFamily, buffer[i].iVersion,
|
|
buffer[i].iSocketType, buffer[i].iProtocol,
|
|
buffer[i].dwServiceFlags1);
|
|
}
|
|
|
|
HeapFree( GetProcessHeap(), 0, buffer );
|
|
}
|
|
|
|
/* Test invalid protocols in the list */
|
|
ret = WSAEnumProtocolsA( ptest, NULL, &len );
|
|
ok( ret == SOCKET_ERROR, "WSAEnumProtocolsA() succeeded unexpectedly\n");
|
|
error = WSAGetLastError();
|
|
ok( error == WSAENOBUFS || broken(error == WSAEFAULT) /* NT4 */,
|
|
"Expected 10055, received %d\n", error);
|
|
|
|
buffer = HeapAlloc( GetProcessHeap(), 0, len );
|
|
|
|
if (buffer)
|
|
{
|
|
ret = WSAEnumProtocolsA( ptest, buffer, &len );
|
|
ok( ret != SOCKET_ERROR, "WSAEnumProtocolsA() failed unexpectedly: %d\n",
|
|
WSAGetLastError() );
|
|
ok( ret >= 2, "Expected at least 2 items, received %d\n", ret);
|
|
|
|
for (i = found = 0; i < ret; i++)
|
|
for (j = 0; j < sizeof(ptest) / sizeof(ptest[0]); j++)
|
|
if (buffer[i].iProtocol == ptest[j])
|
|
{
|
|
found |= 1 << j;
|
|
break;
|
|
}
|
|
ok(found == 0x0A, "Expected 2 bits represented as 0xA, received 0x%x\n", found);
|
|
|
|
HeapFree( GetProcessHeap(), 0, buffer );
|
|
}
|
|
}
|
|
|
|
static void test_WSAEnumProtocolsW(void)
|
|
{
|
|
INT ret, i, j, found;
|
|
DWORD len = 0, error;
|
|
WSAPROTOCOL_INFOW info, *buffer;
|
|
INT ptest[] = {0xdead, IPPROTO_TCP, 0xcafe, IPPROTO_UDP, 0xbeef, 0};
|
|
|
|
ret = WSAEnumProtocolsW( NULL, NULL, &len );
|
|
ok( ret == SOCKET_ERROR, "WSAEnumProtocolsW() succeeded unexpectedly\n");
|
|
error = WSAGetLastError();
|
|
ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
|
|
|
|
len = 0;
|
|
|
|
ret = WSAEnumProtocolsW( NULL, &info, &len );
|
|
ok( ret == SOCKET_ERROR, "WSAEnumProtocolsW() succeeded unexpectedly\n");
|
|
error = WSAGetLastError();
|
|
ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
|
|
|
|
buffer = HeapAlloc( GetProcessHeap(), 0, len );
|
|
|
|
if (buffer)
|
|
{
|
|
ret = WSAEnumProtocolsW( NULL, buffer, &len );
|
|
ok( ret != SOCKET_ERROR, "WSAEnumProtocolsW() failed unexpectedly: %d\n",
|
|
WSAGetLastError() );
|
|
|
|
for (i = 0; i < ret; i++)
|
|
{
|
|
ok( lstrlenW( buffer[i].szProtocol ), "No protocol name found\n" );
|
|
test_service_flags( buffer[i].iAddressFamily, buffer[i].iVersion,
|
|
buffer[i].iSocketType, buffer[i].iProtocol,
|
|
buffer[i].dwServiceFlags1);
|
|
}
|
|
|
|
HeapFree( GetProcessHeap(), 0, buffer );
|
|
}
|
|
|
|
/* Test invalid protocols in the list */
|
|
ret = WSAEnumProtocolsW( ptest, NULL, &len );
|
|
ok( ret == SOCKET_ERROR, "WSAEnumProtocolsW() succeeded unexpectedly\n");
|
|
error = WSAGetLastError();
|
|
ok( error == WSAENOBUFS || broken(error == WSAEFAULT) /* NT4 */,
|
|
"Expected 10055, received %d\n", error);
|
|
|
|
buffer = HeapAlloc( GetProcessHeap(), 0, len );
|
|
|
|
if (buffer)
|
|
{
|
|
ret = WSAEnumProtocolsW( ptest, buffer, &len );
|
|
ok( ret != SOCKET_ERROR, "WSAEnumProtocolsW() failed unexpectedly: %d\n",
|
|
WSAGetLastError() );
|
|
ok( ret >= 2, "Expected at least 2 items, received %d\n", ret);
|
|
|
|
for (i = found = 0; i < ret; i++)
|
|
for (j = 0; j < sizeof(ptest) / sizeof(ptest[0]); j++)
|
|
if (buffer[i].iProtocol == ptest[j])
|
|
{
|
|
found |= 1 << j;
|
|
break;
|
|
}
|
|
ok(found == 0x0A, "Expected 2 bits represented as 0xA, received 0x%x\n", found);
|
|
|
|
HeapFree( GetProcessHeap(), 0, buffer );
|
|
}
|
|
}
|
|
|
|
START_TEST( protocol )
|
|
{
|
|
WSADATA data;
|
|
WORD version = MAKEWORD( 2, 2 );
|
|
|
|
if (WSAStartup( version, &data )) return;
|
|
|
|
test_WSAEnumProtocolsA();
|
|
test_WSAEnumProtocolsW();
|
|
}
|