[PATCH 5/7] ws2_32/tests: Add some tests for IOCTL_AFD_BIND and IOCTL_AFD_GETSOCKNAME.
Zebediah Figura
z.figura12 at gmail.com
Tue Jun 15 22:07:27 CDT 2021
Signed-off-by: Zebediah Figura <z.figura12 at gmail.com>
---
dlls/ws2_32/tests/afd.c | 231 ++++++++++++++++++++++++++++++++++++++++
1 file changed, 231 insertions(+)
diff --git a/dlls/ws2_32/tests/afd.c b/dlls/ws2_32/tests/afd.c
index 443b6e4e42f..d4633f7a8bb 100644
--- a/dlls/ws2_32/tests/afd.c
+++ b/dlls/ws2_32/tests/afd.c
@@ -1333,6 +1333,235 @@ static void test_get_events(void)
CloseHandle(event);
}
+static void test_bind(void)
+{
+ const struct sockaddr_in invalid_addr = {.sin_family = AF_INET, .sin_addr.s_addr = inet_addr("192.0.2.0")};
+ const struct sockaddr_in bind_addr = {.sin_family = AF_INET, .sin_addr.s_addr = htonl(INADDR_LOOPBACK)};
+ struct afd_bind_params params = {0};
+ struct sockaddr_in addr, addr2;
+ struct hostent *host;
+ IO_STATUS_BLOCK io;
+ unsigned int i;
+ HANDLE event;
+ SOCKET s, s2;
+ int ret;
+
+ event = CreateEventW(NULL, TRUE, FALSE, NULL);
+ memset(&addr, 0xcc, sizeof(addr));
+
+ s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+
+ params.addr.sa_family = 0xdead;
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ ok(ret == STATUS_INVALID_ADDRESS, "got %#x\n", ret);
+
+ memcpy(¶ms.addr, &bind_addr, sizeof(bind_addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params) - 1, &addr, sizeof(addr));
+ ok(ret == STATUS_INVALID_ADDRESS, "got %#x\n", ret);
+
+ memcpy(¶ms.addr, &bind_addr, sizeof(bind_addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, offsetof(struct afd_bind_params, addr.sa_data), &addr, sizeof(addr));
+ ok(ret == STATUS_INVALID_ADDRESS, "got %#x\n", ret);
+
+ memcpy(¶ms.addr, &bind_addr, sizeof(bind_addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, offsetof(struct afd_bind_params, addr.sa_data) - 1, &addr, sizeof(addr));
+ ok(ret == STATUS_INVALID_PARAMETER, "got %#x\n", ret);
+
+ memcpy(¶ms.addr, &invalid_addr, sizeof(invalid_addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ todo_wine ok(ret == STATUS_PENDING, "got %#x\n", ret);
+ ret = WaitForSingleObject(event, 0);
+ ok(!ret, "got %#x\n", ret);
+ ok(io.Status == STATUS_INVALID_ADDRESS_COMPONENT, "got %#x\n", io.Status);
+
+ memcpy(¶ms.addr, &bind_addr, sizeof(bind_addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr) - 1);
+ ok(ret == STATUS_INVALID_PARAMETER, "got %#x\n", ret);
+
+ memcpy(¶ms.addr, &bind_addr, sizeof(bind_addr));
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ todo_wine ok(ret == STATUS_PENDING, "got %#x\n", ret);
+ ret = WaitForSingleObject(event, 0);
+ ok(!ret, "got %#x\n", ret);
+ ok(!io.Status, "got %#x\n", io.Status);
+ ok(io.Information == sizeof(addr), "got %#Ix\n", io.Information);
+ ok(addr.sin_family == AF_INET, "got family %u\n", addr.sin_family);
+ ok(addr.sin_addr.s_addr == htonl(INADDR_LOOPBACK), "got address %#08x\n", addr.sin_addr.s_addr);
+ ok(addr.sin_port, "expected nonzero port\n");
+
+ /* getsockname() returns EINVAL here. Possibly the socket name is cached (in shared memory?) */
+ memset(&addr2, 0xcc, sizeof(addr2));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr2, sizeof(addr2));
+ ok(!ret, "got %#x\n", ret);
+ ok(!memcmp(&addr, &addr2, sizeof(addr)), "addresses didn't match\n");
+
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ todo_wine ok(ret == STATUS_ADDRESS_ALREADY_ASSOCIATED, "got %#x\n", ret);
+
+ s2 = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+
+ memcpy(¶ms.addr, &addr2, sizeof(addr2));
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)s2, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ todo_wine ok(ret == STATUS_PENDING, "got %#x\n", ret);
+ ret = WaitForSingleObject(event, 0);
+ ok(!ret, "got %#x\n", ret);
+ ok(io.Status == STATUS_SHARING_VIOLATION, "got %#x\n", io.Status);
+ ok(!io.Information, "got %#Ix\n", io.Information);
+
+ closesocket(s2);
+ closesocket(s);
+
+ /* test UDP */
+
+ s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+
+ memcpy(¶ms.addr, &bind_addr, sizeof(bind_addr));
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ todo_wine ok(ret == STATUS_PENDING, "got %#x\n", ret);
+ ret = WaitForSingleObject(event, 0);
+ ok(!ret, "got %#x\n", ret);
+ ok(!io.Status, "got %#x\n", io.Status);
+ ok(io.Information == sizeof(addr), "got %#Ix\n", io.Information);
+ ok(addr.sin_family == AF_INET, "got family %u\n", addr.sin_family);
+ ok(addr.sin_addr.s_addr == htonl(INADDR_LOOPBACK), "got address %#08x\n", addr.sin_addr.s_addr);
+ ok(addr.sin_port, "expected nonzero port\n");
+
+ memset(&addr2, 0xcc, sizeof(addr2));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr2, sizeof(addr2));
+ ok(!ret, "got %#x\n", ret);
+ ok(!memcmp(&addr, &addr2, sizeof(addr)), "addresses didn't match\n");
+
+ closesocket(s);
+
+ host = gethostbyname("");
+ if (host && host->h_length == 4)
+ {
+ for (i = 0; host->h_addr_list[i]; ++i)
+ {
+ ULONG in_addr = *(ULONG *)host->h_addr_list[i];
+
+ s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+
+ ((struct sockaddr_in *)¶ms.addr)->sin_addr.s_addr = in_addr;
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io, IOCTL_AFD_BIND,
+ ¶ms, sizeof(params), &addr, sizeof(addr));
+ todo_wine ok(ret == STATUS_PENDING, "got %#x\n", ret);
+ ret = WaitForSingleObject(event, 0);
+ ok(!ret, "got %#x\n", ret);
+ ok(!io.Status, "got %#x\n", io.Status);
+ ok(io.Information == sizeof(addr), "got %#Ix\n", io.Information);
+ ok(addr.sin_family == AF_INET, "got family %u\n", addr.sin_family);
+ ok(addr.sin_addr.s_addr == in_addr, "expected address %#08x, got %#08x\n", in_addr, addr.sin_addr.s_addr);
+ ok(addr.sin_port, "expected nonzero port\n");
+
+ memset(&addr2, 0xcc, sizeof(addr2));
+ ret = NtDeviceIoControlFile((HANDLE)s, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr2, sizeof(addr2));
+ ok(!ret, "got %#x\n", ret);
+ ok(!memcmp(&addr, &addr2, sizeof(addr)), "addresses didn't match\n");
+
+ closesocket(s);
+ }
+ }
+
+ CloseHandle(event);
+}
+
+static void test_getsockname(void)
+{
+ const struct sockaddr_in bind_addr = {.sin_family = AF_INET, .sin_addr.s_addr = htonl(INADDR_LOOPBACK)};
+ struct sockaddr addr, addr2;
+ SOCKET server, client;
+ IO_STATUS_BLOCK io;
+ HANDLE event;
+ int ret, len;
+
+ event = CreateEventW(NULL, TRUE, FALSE, NULL);
+
+ tcp_socketpair(&client, &server);
+
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)client, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr, sizeof(addr) - 1);
+ ok(ret == STATUS_BUFFER_TOO_SMALL, "got %#x\n", ret);
+
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)client, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr, sizeof(addr));
+ ok(!ret, "got %#x\n", ret);
+ ok(!io.Status, "got %#x\n", io.Status);
+ ok(io.Information == sizeof(addr), "got %#Ix\n", io.Information);
+ len = sizeof(addr2);
+ ret = getsockname(client, (struct sockaddr *)&addr2, &len);
+ ok(!ret, "got error %u\n", WSAGetLastError());
+ ok(!memcmp(&addr, &addr2, sizeof(struct sockaddr)), "addresses didn't match\n");
+
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)server, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr, sizeof(addr));
+ ok(!ret, "got %#x\n", ret);
+ ok(!io.Status, "got %#x\n", io.Status);
+ ok(io.Information == sizeof(addr), "got %#Ix\n", io.Information);
+ len = sizeof(addr2);
+ ret = getsockname(server, (struct sockaddr *)&addr2, &len);
+ ok(!ret, "got error %u\n", WSAGetLastError());
+ ok(!memcmp(&addr, &addr2, sizeof(struct sockaddr)), "addresses didn't match\n");
+
+ closesocket(server);
+ closesocket(client);
+
+ client = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+
+ ret = NtDeviceIoControlFile((HANDLE)client, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr, sizeof(addr));
+ ok(ret == STATUS_INVALID_PARAMETER, "got %#x\n", ret);
+
+ ret = NtDeviceIoControlFile((HANDLE)client, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr, sizeof(addr) - 1);
+ ok(ret == STATUS_INVALID_PARAMETER, "got %#x\n", ret);
+
+ ret = bind(client, (const struct sockaddr *)&bind_addr, sizeof(bind_addr));
+ ok(!ret, "got error %u\n", WSAGetLastError());
+
+ memset(&io, 0xcc, sizeof(io));
+ memset(&addr, 0xcc, sizeof(addr));
+ ret = NtDeviceIoControlFile((HANDLE)client, event, NULL, NULL, &io,
+ IOCTL_AFD_GETSOCKNAME, NULL, 0, &addr, sizeof(addr));
+ todo_wine ok(!ret, "got %#x\n", ret);
+ todo_wine ok(!io.Status, "got %#x\n", io.Status);
+ todo_wine ok(io.Information == sizeof(addr), "got %#Ix\n", io.Information);
+ len = sizeof(addr2);
+ ret = getsockname(client, (struct sockaddr *)&addr2, &len);
+ ok(!ret, "got error %u\n", WSAGetLastError());
+ todo_wine ok(!memcmp(&addr, &addr2, sizeof(struct sockaddr)), "addresses didn't match\n");
+
+ closesocket(client);
+
+ CloseHandle(event);
+}
+
START_TEST(afd)
{
WSADATA data;
@@ -1345,6 +1574,8 @@ START_TEST(afd)
test_recv();
test_event_select();
test_get_events();
+ test_bind();
+ test_getsockname();
WSACleanup();
}
--
2.30.2
More information about the wine-devel
mailing list