dsocket.c File Reference

#include <ws2_32.h>

Include dependency graph for dsocket.c:

Go to the source code of this file.

Functions
INT WSAAPI	WsSockStartup (VOID)
VOID WSPAPI	WsSockCleanup (VOID)
PWSSOCKET WSAAPI	WsSockAllocate (VOID)
INT WSAAPI	WsSockInitialize (IN PWSSOCKET Socket, IN PTCATALOG_ENTRY CatalogEntry)
PWSSOCKET WSAAPI	WsSockGetSocketNoExport (IN SOCKET Handle)
PWSSOCKET WSAAPI	WsSockFindIfsSocket (IN SOCKET Handle)
PWSSOCKET WSAAPI	WsSockGetSocket (IN SOCKET Handle)
INT WSAAPI	WsSockAddApiReference (IN SOCKET Handle)
BOOL WSAAPI	WsSockDeleteSockets (IN LPVOID Context, IN PWAH_HANDLE Handle)
VOID WSAAPI	WsSockDelete (IN PWSSOCKET Socket)
VOID WSAAPI	WsSockDereference (IN PWSSOCKET Socket)
INT WSAAPI	WsSockDisassociateHandle (IN PWSSOCKET Socket)
INT WSAAPI	WsSockAssociateHandle (IN PWSSOCKET Socket, IN SOCKET Handle, IN BOOLEAN IsProvider)

Variables
PWAH_HANDLE_TABLE	WsSockHandleTable

Function Documentation

WsSockAddApiReference()

INT WSAAPI WsSockAddApiReference

(

IN SOCKET

Handle

)

Definition at line 159 of file dsocket.c.

{
    PWSSOCKET Socket;
 
    /* Get the Socket now */
    if ((Socket = WsSockGetSocketNoExport(Handle)))
    {
        /* Mark it as an API Socket */
        if (Socket) Socket->ApiSocket = TRUE;
 
        /* Remove a reference and return */
        WsSockDereference(Socket);
        return ERROR_SUCCESS;
    }
 
    /* Return error */
    return WSASYSCALLFAILURE;
}

Referenced by WSAAccept(), WSAJoinLeaf(), and WSASocketW().

WsSockAllocate()

PWSSOCKET WSAAPI WsSockAllocate

(

VOID

)

Definition at line 48 of file dsocket.c.

{
    PWSSOCKET Socket;
 
    /* Allocate the socket object */
    Socket = HeapAlloc(WsSockHeap, HEAP_ZERO_MEMORY, sizeof(*Socket));
    if (Socket)
    {
        /* Setup default non-zero values */
        Socket->RefCount = 2;
        Socket->Overlapped = TRUE;
    }
 
    /* Return it */
    return Socket;
}

Referenced by WPUModifyIFSHandle().

WsSockAssociateHandle()

INT WSAAPI WsSockAssociateHandle	(	IN PWSSOCKET	Socket,
		IN SOCKET	Handle,
		IN BOOLEAN	IsProvider
	)

Definition at line 225 of file dsocket.c.

{
    INT ErrorCode = ERROR_SUCCESS;
    PWSSOCKET OldSocket;
 
    /* Save the socket and provider */
    Socket->IsProvider = IsProvider;
    Socket->Handle = (HANDLE)Handle;
 
    /* Insert it into the handle table */
    OldSocket = (PWSSOCKET)WahInsertHandleContext(WsSockHandleTable,
                                                  (PWAH_HANDLE)Socket);
 
    /* Check if a socket already existed */
    if (OldSocket != Socket)
    {
        /* We'll dereference it */
        WsSockDereference(OldSocket);
    }
    else if (!OldSocket)
    {
        /* No memory to allocate it */
        ErrorCode = WSAENOBUFS;
    }
 
    /* Return */
    return ErrorCode;
}

Referenced by WPUModifyIFSHandle().

WsSockCleanup()

VOID WSPAPI WsSockCleanup

(

VOID

)

Definition at line 36 of file dsocket.c.

{
    /* Check if we have a socket table */
    if (WsSockHandleTable)
    {
        /* Destroy it */
        WahDestroyHandleContextTable(WsSockHandleTable);
    }
}

Referenced by DllMain().

WsSockDelete()

VOID WSAAPI WsSockDelete

(

IN PWSSOCKET

Socket

)

Definition at line 189 of file dsocket.c.

{
    /* Check if we have a catalog entry */
    if (Socket->CatalogEntry)
    {
        /* Dereference it */
        WsTcEntryDereference(Socket->CatalogEntry);
        Socket->CatalogEntry = NULL;
    }
 
    /* Delete us */
    HeapFree(WsSockHeap, 0, Socket);
}

Referenced by WsSockDereference().

WsSockDeleteSockets()

BOOL WSAAPI WsSockDeleteSockets	(	IN LPVOID	Context,
		IN PWAH_HANDLE	Handle
	)

Definition at line 180 of file dsocket.c.

{
    /* Call the detach routine */
    return WsProcDetachSocket((PWSPROCESS)Context, Handle);
}

Referenced by WsProcDelete().

WsSockDereference()

VOID WSAAPI WsSockDereference

(

IN PWSSOCKET

Socket

)

Definition at line 205 of file dsocket.c.

{
    /* Dereference and check if it's now 0 */
    if (!(InterlockedDecrement(&Socket->RefCount)))
    {
        /* We can delete the Provider now */
        WsSockDelete(Socket);
    }
}

Referenced by bind(), closesocket(), connect(), getpeername(), getsockname(), getsockopt(), listen(), recv(), recvfrom(), select(), send(), sendto(), setsockopt(), shutdown(), WPUModifyIFSHandle(), WSAAccept(), WSAAsyncSelect(), WSAConnect(), WSADuplicateSocketW(), WSAEnumNetworkEvents(), WSAEventSelect(), WSAGetOverlappedResult(), WSAGetQOSByName(), WSAHtonl(), WSAHtons(), WSAIoctl(), WSAJoinLeaf(), WSANtohl(), WSANtohs(), WSARecv(), WSARecvDisconnect(), WSARecvFrom(), WSASend(), WSASendDisconnect(), WSASendTo(), WsProcDetachSocket(), WsSockAddApiReference(), and WsSockAssociateHandle().

WsSockDisassociateHandle()

INT WSAAPI WsSockDisassociateHandle

(

IN PWSSOCKET

Socket

)

Definition at line 217 of file dsocket.c.

{
    /* Remove it from the list */
    return WahRemoveHandleContext(WsSockHandleTable, (PWAH_HANDLE)Socket);
}

Referenced by closesocket().

WsSockFindIfsSocket()

PWSSOCKET WSAAPI WsSockFindIfsSocket

(

IN SOCKET

Handle

)

Definition at line 101 of file dsocket.c.

{
    INT ErrorCode;
    DWORD Flags;
    PWSSOCKET Socket = NULL;
    PWSPROCESS Process = NULL;
    PTCATALOG Catalog = NULL;
 
    /* Validate the socket and get handle info */
    if ((Handle != INVALID_SOCKET) &&
        (GetHandleInformation((HANDLE)Handle, &Flags)))
    {
        /* Get the process */
        if ((Process = WsGetProcess()))
        {
            /* Get the catalog */
            Catalog = WsProcGetTCatalog(Process);
 
            /* Get the IFS Provider */
            ErrorCode = WsTcFindIfsProviderForSocket(Catalog, Handle);
 
            /* Check for success */
            if (ErrorCode == ERROR_SUCCESS)
            {
                /* Get the Socket now */
                Socket = WsSockGetSocketNoExport(Handle);
 
                /* Mark it as an API Socket */
                if (Socket) Socket->ApiSocket = TRUE;
            }
        }
    }
 
    /* Return the socket */
    return Socket;
}

Referenced by WsSockGetSocket().

WsSockGetSocket()

PWSSOCKET WSAAPI WsSockGetSocket

(

IN SOCKET

Handle

)

Definition at line 140 of file dsocket.c.

{
    PWSSOCKET Socket;
 
    /* Let WAH do the translation */
    if ((WsSockHandleTable != NULL) &&
        (Socket = (PWSSOCKET)WahReferenceContextByHandle(WsSockHandleTable, (HANDLE)Handle)))
    {
        return Socket;
    }
    else
    {
        /* WAH didn't find it, use IFS */
        return WsSockFindIfsSocket(Handle);
    }
}

Referenced by bind(), closesocket(), connect(), getpeername(), getsockname(), getsockopt(), listen(), recv(), recvfrom(), select(), send(), sendto(), setsockopt(), shutdown(), WSAAccept(), WSAAsyncSelect(), WSAConnect(), WSADuplicateSocketW(), WSAEnumNetworkEvents(), WSAEventSelect(), WSAGetOverlappedResult(), WSAGetQOSByName(), WSAHtonl(), WSAHtons(), WSAIoctl(), WSAJoinLeaf(), WSANtohl(), WSANtohs(), WSARecv(), WSARecvDisconnect(), WSARecvFrom(), WSASend(), WSASendDisconnect(), and WSASendTo().

WsSockGetSocketNoExport()

PWSSOCKET WSAAPI WsSockGetSocketNoExport

(

IN SOCKET

Handle

)

Definition at line 92 of file dsocket.c.

{
    /* Let WAH do the translation */
    return (PWSSOCKET)WahReferenceContextByHandle(WsSockHandleTable,
                                                  (HANDLE)Handle);
}

Referenced by WsSockAddApiReference(), and WsSockFindIfsSocket().

WsSockInitialize()

INT WSAAPI WsSockInitialize	(	IN PWSSOCKET	Socket,
		IN PTCATALOG_ENTRY	CatalogEntry
	)

Definition at line 67 of file dsocket.c.

{
    PWSTHREAD CurrentThread;
 
    /* Associate this catalog and reference it */
    Socket->CatalogEntry = CatalogEntry;
    InterlockedIncrement(&CatalogEntry->RefCount);
 
    /* Associate the Provider and Process Objects */
    Socket->Provider = CatalogEntry->Provider;
 
    /* Get the current Thread Object */
    if ((CurrentThread = TlsGetValue(TlsIndex)))
    {
        /* Set the overlapped mode */
        Socket->Overlapped = (CurrentThread->OpenType == 0);
    }
 
    /* Return status */
    return ERROR_SUCCESS;
}

Referenced by WPUModifyIFSHandle().

WsSockStartup()

INT WSAAPI WsSockStartup

(

VOID

)

Definition at line 21 of file dsocket.c.

{
    /* Check if we have a socket table */
    if (!WsSockHandleTable)
    {
        /* Create it */
        return WahCreateHandleContextTable(&WsSockHandleTable);
    }
 
    /* Nothing to do */
    return NO_ERROR;
}

Referenced by WSAStartup().

Variable Documentation

WsSockHandleTable

PWAH_HANDLE_TABLE WsSockHandleTable

Definition at line 15 of file dsocket.c.

Referenced by WsProcDelete(), WsProcDetachSocket(), WsSockAssociateHandle(), WsSockCleanup(), WsSockDisassociateHandle(), WsSockGetSocket(), WsSockGetSocketNoExport(), and WsSockStartup().