mirror of
https://github.com/reactos/reactos.git
synced 2024-08-26 05:08:39 +00:00
181 lines
4.6 KiB
C
181 lines
4.6 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
|
|
* PURPOSE: Test for send/sendto
|
|
* PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
|
|
*/
|
|
|
|
#include "ws2_32.h"
|
|
|
|
static
|
|
PVOID
|
|
AllocateReadOnly(
|
|
_In_ SIZE_T SizeRequested)
|
|
{
|
|
NTSTATUS Status;
|
|
SIZE_T Size = PAGE_ROUND_UP(SizeRequested);
|
|
PVOID VirtualMemory = NULL;
|
|
|
|
Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_COMMIT, PAGE_READONLY);
|
|
if (!NT_SUCCESS(Status))
|
|
return NULL;
|
|
|
|
return VirtualMemory;
|
|
}
|
|
|
|
static
|
|
VOID
|
|
FreeReadOnly(
|
|
_In_ PVOID VirtualMemory)
|
|
{
|
|
NTSTATUS Status;
|
|
SIZE_T Size = 0;
|
|
|
|
Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
|
|
ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
|
|
}
|
|
|
|
static
|
|
VOID
|
|
test_send(void)
|
|
{
|
|
SOCKET sock;
|
|
int ret;
|
|
int error;
|
|
PVOID buffer;
|
|
ULONG bufferSize;
|
|
struct sockaddr_in addr;
|
|
|
|
bufferSize = 32;
|
|
buffer = AllocateReadOnly(bufferSize);
|
|
ok(buffer != NULL, "AllocateReadOnly failed\n");
|
|
if (!buffer)
|
|
{
|
|
skip("No memory\n");
|
|
return;
|
|
}
|
|
|
|
ret = send(0, NULL, 0, 0);
|
|
error = WSAGetLastError();
|
|
ok(ret == SOCKET_ERROR, "send returned %d\n", ret);
|
|
ok(error == WSAENOTSOCK, "error = %d\n", error);
|
|
|
|
ret = send(0, buffer, bufferSize, 0);
|
|
error = WSAGetLastError();
|
|
ok(ret == SOCKET_ERROR, "send returned %d\n", ret);
|
|
ok(error == WSAENOTSOCK, "error = %d\n", error);
|
|
|
|
sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
|
ok(sock != INVALID_SOCKET, "socket failed\n");
|
|
if (sock == INVALID_SOCKET)
|
|
{
|
|
skip("No socket\n");
|
|
FreeReadOnly(buffer);
|
|
return;
|
|
}
|
|
|
|
ret = send(sock, NULL, 0, 0);
|
|
error = WSAGetLastError();
|
|
ok(ret == SOCKET_ERROR, "send returned %d\n", ret);
|
|
ok(error == WSAENOTCONN, "error = %d\n", error);
|
|
|
|
ret = send(sock, buffer, bufferSize, 0);
|
|
error = WSAGetLastError();
|
|
ok(ret == SOCKET_ERROR, "send returned %d\n", ret);
|
|
ok(error == WSAENOTCONN, "error = %d\n", error);
|
|
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_addr.s_addr = inet_addr("8.8.8.8");
|
|
addr.sin_port = htons(53);
|
|
ret = connect(sock, (const struct sockaddr *)&addr, sizeof(addr));
|
|
error = WSAGetLastError();
|
|
ok(ret == 0, "connect returned %d\n", ret);
|
|
ok(error == 0, "error = %d\n", error);
|
|
|
|
ret = send(sock, NULL, 0, 0);
|
|
error = WSAGetLastError();
|
|
ok(ret == 0, "send returned %d\n", ret);
|
|
ok(error == 0, "error = %d\n", error);
|
|
|
|
ret = send(sock, buffer, bufferSize, 0);
|
|
error = WSAGetLastError();
|
|
ok(ret == bufferSize, "send returned %d\n", ret);
|
|
ok(error == 0, "error = %d\n", error);
|
|
|
|
closesocket(sock);
|
|
|
|
FreeReadOnly(buffer);
|
|
}
|
|
|
|
static
|
|
VOID
|
|
test_sendto(void)
|
|
{
|
|
SOCKET sock;
|
|
int ret;
|
|
int error;
|
|
PVOID buffer;
|
|
ULONG bufferSize;
|
|
struct sockaddr_in addr;
|
|
|
|
bufferSize = 32;
|
|
buffer = AllocateReadOnly(bufferSize);
|
|
ok(buffer != NULL, "AllocateReadOnly failed\n");
|
|
if (!buffer)
|
|
{
|
|
skip("No memory\n");
|
|
return;
|
|
}
|
|
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_addr.s_addr = inet_addr("8.8.8.8");
|
|
addr.sin_port = htons(53);
|
|
|
|
ret = sendto(0, NULL, 0, 0, (const struct sockaddr *)&addr, sizeof(addr));
|
|
error = WSAGetLastError();
|
|
ok(ret == SOCKET_ERROR, "sendto returned %d\n", ret);
|
|
ok(error == WSAENOTSOCK, "error = %d\n", error);
|
|
|
|
ret = sendto(0, buffer, bufferSize, 0, (const struct sockaddr *)&addr, sizeof(addr));
|
|
error = WSAGetLastError();
|
|
ok(ret == SOCKET_ERROR, "sendto returned %d\n", ret);
|
|
ok(error == WSAENOTSOCK, "error = %d\n", error);
|
|
|
|
sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
ok(sock != INVALID_SOCKET, "socket failed\n");
|
|
if (sock == INVALID_SOCKET)
|
|
{
|
|
skip("No socket\n");
|
|
FreeReadOnly(buffer);
|
|
return;
|
|
}
|
|
|
|
ret = sendto(sock, NULL, 0, 0, (const struct sockaddr *)&addr, sizeof(addr));
|
|
error = WSAGetLastError();
|
|
ok(ret == 0, "sendto returned %d\n", ret);
|
|
ok(error == 0, "error = %d\n", error);
|
|
|
|
ret = sendto(sock, buffer, bufferSize, 0, (const struct sockaddr *)&addr, sizeof(addr));
|
|
error = WSAGetLastError();
|
|
ok(ret == bufferSize, "sendto returned %d\n", ret);
|
|
ok(error == 0, "error = %d\n", error);
|
|
|
|
closesocket(sock);
|
|
|
|
FreeReadOnly(buffer);
|
|
}
|
|
|
|
START_TEST(send)
|
|
{
|
|
int ret;
|
|
WSADATA wsad;
|
|
|
|
ret = WSAStartup(MAKEWORD(2, 2), &wsad);
|
|
ok(ret == 0, "WSAStartup failed with %d\n", ret);
|
|
test_send();
|
|
test_sendto();
|
|
WSACleanup();
|
|
}
|