dlls/ws2_32/tests/protocol.c

   1 /*
   2  * Unit test suite for protocol functions
   3  *
   4  * Copyright 2004 Hans Leidekker
   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
  23 #include <windef.h>
  24 #include <winbase.h>
  25 #include <winsock2.h>
  26
  27 #include "wine/test.h"
  28
  29 /* TCP and UDP over IP fixed set of service flags */
  30 #define TCPIP_SERVICE_FLAGS (XP1_GUARANTEED_DELIVERY \
  31                            | XP1_GUARANTEED_ORDER    \
  32                            | XP1_GRACEFUL_CLOSE      \
  33                            | XP1_EXPEDITED_DATA      \
  34                            | XP1_IFS_HANDLES)
  35
  36 #define UDPIP_SERVICE_FLAGS (XP1_CONNECTIONLESS      \
  37                            | XP1_MESSAGE_ORIENTED    \
  38                            | XP1_SUPPORT_BROADCAST   \
  39                            | XP1_SUPPORT_MULTIPOINT  \
  40                            | XP1_IFS_HANDLES)
  41
  42 static void test_service_flags(int family, int version, int socktype, int protocol, DWORD testflags)
  43 {
  44     DWORD expectedflags = 0;
  45     if (socktype == SOCK_STREAM && protocol == IPPROTO_TCP)
  46         expectedflags = TCPIP_SERVICE_FLAGS;
  47     if (socktype == SOCK_DGRAM && protocol == IPPROTO_UDP)
  48         expectedflags = UDPIP_SERVICE_FLAGS;
  49
  50     /* check if standard TCP and UDP protocols are offering the correct service flags */
  51     if ((family == AF_INET || family == AF_INET6) && version == 2 && expectedflags)
  52     {
  53         /* QOS may or may not be installed */
  54         testflags &= ~XP1_QOS_SUPPORTED;
  55         ok(expectedflags == testflags,
  56            "Incorrect flags, expected 0x%x, received 0x%x\n",
  57            expectedflags, testflags);
  58     }
  59 }
  60
  61 static void test_WSAEnumProtocolsA(void)
  62 {
  63     INT ret, i, j, found;
  64     DWORD len = 0, error;
  65     WSAPROTOCOL_INFOA info, *buffer;
  66     INT ptest[] = {0xdead, IPPROTO_TCP, 0xcafe, IPPROTO_UDP, 0xbeef, 0};
  67
  68     ret = WSAEnumProtocolsA( NULL, NULL, &len );
  69     ok( ret == SOCKET_ERROR, "WSAEnumProtocolsA() succeeded unexpectedly\n");
  70     error = WSAGetLastError();
  71     ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
  72
  73     len = 0;
  74
  75     ret = WSAEnumProtocolsA( NULL, &info, &len );
  76     ok( ret == SOCKET_ERROR, "WSAEnumProtocolsA() succeeded unexpectedly\n");
  77     error = WSAGetLastError();
  78     ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
  79
  80     buffer = HeapAlloc( GetProcessHeap(), 0, len );
  81
  82     if (buffer)
  83     {
  84         ret = WSAEnumProtocolsA( NULL, buffer, &len );
  85         ok( ret != SOCKET_ERROR, "WSAEnumProtocolsA() failed unexpectedly: %d\n",
  86             WSAGetLastError() );
  87
  88         for (i = 0; i < ret; i++)
  89         {
  90             ok( strlen( buffer[i].szProtocol ), "No protocol name found\n" );
  91             test_service_flags( buffer[i].iAddressFamily, buffer[i].iVersion,
  92                                 buffer[i].iSocketType, buffer[i].iProtocol,
  93                                 buffer[i].dwServiceFlags1);
  94         }
  95
  96         HeapFree( GetProcessHeap(), 0, buffer );
  97     }
  98
  99     /* Test invalid protocols in the list */
 100     ret = WSAEnumProtocolsA( ptest, NULL, &len );
 101     ok( ret == SOCKET_ERROR, "WSAEnumProtocolsA() succeeded unexpectedly\n");
 102     error = WSAGetLastError();
 103     ok( error == WSAENOBUFS || broken(error == WSAEFAULT) /* NT4 */,
 104        "Expected 10055, received %d\n", error);
 105
 106     buffer = HeapAlloc( GetProcessHeap(), 0, len );
 107
 108     if (buffer)
 109     {
 110         ret = WSAEnumProtocolsA( ptest, buffer, &len );
 111         ok( ret != SOCKET_ERROR, "WSAEnumProtocolsA() failed unexpectedly: %d\n",
 112             WSAGetLastError() );
 113         ok( ret >= 2, "Expected at least 2 items, received %d\n", ret);
 114
 115         for (i = found = 0; i < ret; i++)
 116             for (j = 0; j < sizeof(ptest) / sizeof(ptest[0]); j++)
 117                 if (buffer[i].iProtocol == ptest[j])
 118                 {
 119                     found |= 1 << j;
 120                     break;
 121                 }
 122         ok(found == 0x0A, "Expected 2 bits represented as 0xA, received 0x%x\n", found);
 123
 124         HeapFree( GetProcessHeap(), 0, buffer );
 125     }
 126 }
 127
 128 static void test_WSAEnumProtocolsW(void)
 129 {
 130     INT ret, i, j, found;
 131     DWORD len = 0, error;
 132     WSAPROTOCOL_INFOW info, *buffer;
 133     INT ptest[] = {0xdead, IPPROTO_TCP, 0xcafe, IPPROTO_UDP, 0xbeef, 0};
 134
 135     ret = WSAEnumProtocolsW( NULL, NULL, &len );
 136     ok( ret == SOCKET_ERROR, "WSAEnumProtocolsW() succeeded unexpectedly\n");
 137     error = WSAGetLastError();
 138     ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
 139
 140     len = 0;
 141
 142     ret = WSAEnumProtocolsW( NULL, &info, &len );
 143     ok( ret == SOCKET_ERROR, "WSAEnumProtocolsW() succeeded unexpectedly\n");
 144     error = WSAGetLastError();
 145     ok( error == WSAENOBUFS, "Expected 10055, received %d\n", error);
 146
 147     buffer = HeapAlloc( GetProcessHeap(), 0, len );
 148
 149     if (buffer)
 150     {
 151         ret = WSAEnumProtocolsW( NULL, buffer, &len );
 152         ok( ret != SOCKET_ERROR, "WSAEnumProtocolsW() failed unexpectedly: %d\n",
 153             WSAGetLastError() );
 154
 155         for (i = 0; i < ret; i++)
 156         {
 157             ok( lstrlenW( buffer[i].szProtocol ), "No protocol name found\n" );
 158             test_service_flags( buffer[i].iAddressFamily, buffer[i].iVersion,
 159                                 buffer[i].iSocketType, buffer[i].iProtocol,
 160                                 buffer[i].dwServiceFlags1);
 161         }
 162
 163         HeapFree( GetProcessHeap(), 0, buffer );
 164     }
 165
 166     /* Test invalid protocols in the list */
 167     ret = WSAEnumProtocolsW( ptest, NULL, &len );
 168     ok( ret == SOCKET_ERROR, "WSAEnumProtocolsW() succeeded unexpectedly\n");
 169     error = WSAGetLastError();
 170     ok( error == WSAENOBUFS || broken(error == WSAEFAULT) /* NT4 */,
 171        "Expected 10055, received %d\n", error);
 172
 173     buffer = HeapAlloc( GetProcessHeap(), 0, len );
 174
 175     if (buffer)
 176     {
 177         ret = WSAEnumProtocolsW( ptest, buffer, &len );
 178         ok( ret != SOCKET_ERROR, "WSAEnumProtocolsW() failed unexpectedly: %d\n",
 179             WSAGetLastError() );
 180         ok( ret >= 2, "Expected at least 2 items, received %d\n", ret);
 181
 182         for (i = found = 0; i < ret; i++)
 183             for (j = 0; j < sizeof(ptest) / sizeof(ptest[0]); j++)
 184                 if (buffer[i].iProtocol == ptest[j])
 185                 {
 186                     found |= 1 << j;
 187                     break;
 188                 }
 189         ok(found == 0x0A, "Expected 2 bits represented as 0xA, received 0x%x\n", found);
 190
 191         HeapFree( GetProcessHeap(), 0, buffer );
 192     }
 193 }
 194
 195 START_TEST( protocol )
 196 {
 197     WSADATA data;
 198     WORD version = MAKEWORD( 2, 2 );
 199
 200     if (WSAStartup( version, &data )) return;
 201
 202     test_WSAEnumProtocolsA();
 203     test_WSAEnumProtocolsW();
 204 }