close.c File Reference

#include "ws2_32.h"

Include dependency graph for close.c:

Go to the source code of this file.

Macros
#define	TIMEOUT_SEC   0
#define	TIMEOUT_USEC   100000
#define	TIME_SLEEP1   250
#define	THREAD_PROC_LOOPS   5
#define	LISTEN_PORT   22222
#define	LISTEN_BACKLOG   5

Functions
void	Test_CloseDuplicatedSocket ()
DWORD WINAPI	thread_proc (void *param)
void	Test_CloseWhileSelectSameSocket ()
void	Test_CloseWhileSelectDuplicatedSocket ()
	START_TEST (close)

Macro Definition Documentation

LISTEN_BACKLOG

#define LISTEN_BACKLOG   5

Definition at line 64 of file close.c.

LISTEN_PORT

#define LISTEN_PORT   22222

Definition at line 63 of file close.c.

THREAD_PROC_LOOPS

#define THREAD_PROC_LOOPS   5

Definition at line 61 of file close.c.

TIME_SLEEP1

#define TIME_SLEEP1   250

Definition at line 59 of file close.c.

TIMEOUT_SEC

#define TIMEOUT_SEC   0

Definition at line 55 of file close.c.

TIMEOUT_USEC

#define TIMEOUT_USEC   100000

Definition at line 56 of file close.c.

Function Documentation

START_TEST()

START_TEST

(

close

)

Definition at line 177 of file close.c.

{
    int err;
    WSADATA wdata;
 
    /* Start up Winsock */
    err = WSAStartup(MAKEWORD(2, 2), &wdata);
    ok(err == 0, "WSAStartup failed, iResult == %d %d\n", err, WSAGetLastError());
 
    Test_CloseDuplicatedSocket();
    Test_CloseWhileSelectSameSocket();
    Test_CloseWhileSelectDuplicatedSocket();
 
    WSACleanup();
}

Test_CloseDuplicatedSocket()

void Test_CloseDuplicatedSocket

(

)

Definition at line 10 of file close.c.

{
    char szBuf[10];
    int err;
    SOCKET sck, dup_sck;
    WSAPROTOCOL_INFOW ProtocolInfo;
    struct sockaddr_in to = { AF_INET, 2222, {{{ 0x7f, 0x00, 0x00, 0x01 }}} };
 
    /* Create the socket */
    sck = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    if(sck == INVALID_SOCKET)
    {
        skip("socket failed %d. Aborting test.\n", WSAGetLastError());
        return;
    }
 
    err = sendto(sck, szBuf, _countof(szBuf), 0, (struct sockaddr *)&to, sizeof(to));
    ok(err == _countof(szBuf), "sendto err = %d %d\n", err, WSAGetLastError());
 
    err = WSADuplicateSocketW(sck, GetCurrentProcessId(), &ProtocolInfo);
    ok(err == 0, "WSADuplicateSocketW err = %d %d\n", err, WSAGetLastError());
 
    dup_sck = WSASocketW(0, 0, 0, &ProtocolInfo, 0, 0);
    if (dup_sck == INVALID_SOCKET)
    {
        skip("WSASocketW failed %d. Aborting test.\n", WSAGetLastError());
        closesocket(sck);
        return;
    }
 
    err = sendto(dup_sck, szBuf, _countof(szBuf), 0, (struct sockaddr *)&to, sizeof(to));
    ok(err == _countof(szBuf), "sendto err = %d %d\n", err, WSAGetLastError());
 
    err = closesocket(sck);
    ok(err == 0, "closesocket sck err = %d %d\n", err, WSAGetLastError());
 
    err = sendto(dup_sck, szBuf, _countof(szBuf), 0, (struct sockaddr *)&to, sizeof(to));
    ok(err == _countof(szBuf), "sendto err = %d %d\n", err, WSAGetLastError());
 
    err = closesocket(dup_sck);
    ok(err == 0, "closesocket dup_sck err = %d %d\n", err, WSAGetLastError());
    return;
}

Referenced by START_TEST().

Test_CloseWhileSelectDuplicatedSocket()

void Test_CloseWhileSelectDuplicatedSocket

(

)

Definition at line 125 of file close.c.

{
    int err;
    SOCKET sock, dup_sock;
    WSAPROTOCOL_INFOW ProtocolInfo;
    struct sockaddr_in addrin;
    HANDLE hthread;
 
    /* Create the socket */
    sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (sock == INVALID_SOCKET)
    {
        skip("socket failed %d. Aborting test.\n", WSAGetLastError());
        return;
    }
 
    memset(&addrin, 0, sizeof(struct sockaddr_in));
    addrin.sin_family = AF_INET;
    addrin.sin_addr.s_addr = inet_addr("127.0.0.1");
    addrin.sin_port = htons(LISTEN_PORT);
 
    err = bind(sock, (struct sockaddr*)(&addrin), sizeof(struct sockaddr_in));
    ok(err == 0, "bind err = %d %d\n", err, WSAGetLastError());
    err = listen(sock, LISTEN_BACKLOG);
    ok(err == 0, "listen err = %d %d\n", err, WSAGetLastError());
 
    err = WSADuplicateSocketW(sock, GetCurrentProcessId(), &ProtocolInfo);
    ok(err == 0, "WSADuplicateSocketW err = %d %d\n", err, WSAGetLastError());
 
    dup_sock = WSASocketW(0, 0, 0, &ProtocolInfo, 0, 0);
    if (dup_sock == INVALID_SOCKET)
    {
        skip("WSASocketW failed %d. Aborting test.\n", WSAGetLastError());
        closesocket(sock);
        return;
    }
 
    hthread = CreateThread(NULL, 0, thread_proc, (void*)dup_sock, 0, NULL);
    ok(hthread != NULL, "CreateThread %ld\n", GetLastError());
 
    err = closesocket(sock);
    ok(err == 0, "closesocket err = %d %d\n", err, WSAGetLastError());
 
    Sleep(TIME_SLEEP1);
    err = closesocket(dup_sock);
    ok(err == 0, "closesocket err = %d %d\n", err, WSAGetLastError());
 
    WaitForSingleObject(hthread, INFINITE);
    CloseHandle(hthread);
    return;
}

Referenced by START_TEST().

Test_CloseWhileSelectSameSocket()

void Test_CloseWhileSelectSameSocket

(

)

Definition at line 88 of file close.c.

{
    int err;
    SOCKET sock;
    struct sockaddr_in addrin;
    HANDLE hthread;
 
    /* Create the socket */
    sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (sock == INVALID_SOCKET)
    {
        skip("socket failed %d. Aborting test.\n", WSAGetLastError());
        return;
    }
 
    memset(&addrin, 0, sizeof(struct sockaddr_in));
    addrin.sin_family = AF_INET;
    addrin.sin_addr.s_addr = inet_addr("127.0.0.1");
    addrin.sin_port = htons(LISTEN_PORT);
 
    err = bind(sock, (struct sockaddr*)(&addrin), sizeof(struct sockaddr_in));
    ok(err == 0, "bind err = %d %d\n", err, WSAGetLastError());
    err = listen(sock, LISTEN_BACKLOG);
    ok(err == 0, "listen err = %d %d\n", err, WSAGetLastError());
 
    hthread = CreateThread(NULL, 0, thread_proc, (void*)sock, 0, NULL);
    ok(hthread != NULL, "CreateThread %ld\n", GetLastError());
 
    Sleep(TIME_SLEEP1);
    err = closesocket(sock);
    ok(err == 0, "closesocket err = %d %d\n", err, WSAGetLastError());
 
    WaitForSingleObject(hthread, INFINITE);
    CloseHandle(hthread);
    return;
}

Referenced by START_TEST().

thread_proc()

DWORD WINAPI thread_proc

(

void *

param

)

Definition at line 66 of file close.c.

{
    fd_set read, write, except;
    struct timeval tval;
    SOCKET sock = (SOCKET)param;
    int i;
 
    tval.tv_sec = TIMEOUT_SEC;
    tval.tv_usec = TIMEOUT_USEC;
 
    for (i = 0; i < THREAD_PROC_LOOPS; ++i)
    {
        FD_ZERO(&read); FD_ZERO(&write); FD_ZERO(&except);
        // write will be empty
        FD_SET(sock, &read); FD_SET(sock, &except);
 
        select(0, &read, &write, &except, &tval);
    }
 
    return 0;
}

Referenced by Test_CloseWhileSelectDuplicatedSocket(), and Test_CloseWhileSelectSameSocket().