2 * Unit test suite for protocol functions
4 * Copyright 2004 Hans Leidekker
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
27 #include "wine/test.h"
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 \
36 #define UDPIP_SERVICE_FLAGS (XP1_CONNECTIONLESS \
37 | XP1_MESSAGE_ORIENTED \
38 | XP1_SUPPORT_BROADCAST \
39 | XP1_SUPPORT_MULTIPOINT \
42 static void test_service_flags(int family
, int version
, int socktype
, int protocol
, DWORD testflags
)
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
;
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
)
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
);
61 static void test_WSAEnumProtocolsA(void)
65 WSAPROTOCOL_INFOA info
, *buffer
;
66 INT ptest
[] = {0xdead, IPPROTO_TCP
, 0xcafe, IPPROTO_UDP
, 0xbeef, 0};
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
);
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
);
80 buffer
= HeapAlloc( GetProcessHeap(), 0, len
);
84 ret
= WSAEnumProtocolsA( NULL
, buffer
, &len
);
85 ok( ret
!= SOCKET_ERROR
, "WSAEnumProtocolsA() failed unexpectedly: %d\n",
88 for (i
= 0; i
< ret
; i
++)
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
);
96 HeapFree( GetProcessHeap(), 0, buffer
);
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
);
106 buffer
= HeapAlloc( GetProcessHeap(), 0, len
);
110 ret
= WSAEnumProtocolsA( ptest
, buffer
, &len
);
111 ok( ret
!= SOCKET_ERROR
, "WSAEnumProtocolsA() failed unexpectedly: %d\n",
113 ok( ret
>= 2, "Expected at least 2 items, received %d\n", ret
);
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
])
122 ok(found
== 0x0A, "Expected 2 bits represented as 0xA, received 0x%x\n", found
);
124 HeapFree( GetProcessHeap(), 0, buffer
);
128 static void test_WSAEnumProtocolsW(void)
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};
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
);
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
);
147 buffer
= HeapAlloc( GetProcessHeap(), 0, len
);
151 ret
= WSAEnumProtocolsW( NULL
, buffer
, &len
);
152 ok( ret
!= SOCKET_ERROR
, "WSAEnumProtocolsW() failed unexpectedly: %d\n",
155 for (i
= 0; i
< ret
; i
++)
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
);
163 HeapFree( GetProcessHeap(), 0, buffer
);
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
);
173 buffer
= HeapAlloc( GetProcessHeap(), 0, len
);
177 ret
= WSAEnumProtocolsW( ptest
, buffer
, &len
);
178 ok( ret
!= SOCKET_ERROR
, "WSAEnumProtocolsW() failed unexpectedly: %d\n",
180 ok( ret
>= 2, "Expected at least 2 items, received %d\n", ret
);
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
])
189 ok(found
== 0x0A, "Expected 2 bits represented as 0xA, received 0x%x\n", found
);
191 HeapFree( GetProcessHeap(), 0, buffer
);
195 START_TEST( protocol
)
198 WORD version
= MAKEWORD( 2, 2 );
200 if (WSAStartup( version
, &data
)) return;
202 test_WSAEnumProtocolsA();
203 test_WSAEnumProtocolsW();