Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-10-04 16:36:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

[MSAFD]

Browse source 
- Add locking to protect the socket list

svn path=/trunk/; revision=47803
This commit is contained in:
Cameron Gutman 2010-06-19 05:04:40 +00:00
parent a8c547f091
commit 135340e065
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
reactos/dll/win32/msafd/misc/dllmain.c
View file

@ -23,6 +23,7 @@ HANDLE GlobalHeap;
WSPUPCALLTABLE Upcalls;
LPWPUCOMPLETEOVERLAPPEDREQUEST lpWPUCompleteOverlappedRequest;
PSOCKET_INFORMATION SocketListHead = NULL;
CRITICAL_SECTION SocketListLock;
LIST_ENTRY SockHelpersListHead = { NULL, NULL };
ULONG SockAsyncThreadRefCount;
HANDLE SockAsyncHelperAfdHandle;
@ -280,8 +281,10 @@ WSPSocket(int AddressFamily,
NULL);
/* Save in Process Sockets List */
EnterCriticalSection(&SocketListLock);
Socket->NextSocket = SocketListHead;
SocketListHead = Socket;
LeaveCriticalSection(&SocketListLock);
/* Create the Socket Context */
CreateContext(Socket);
@ -556,6 +559,7 @@ WSPCloseSocket(IN SOCKET Handle,
NtClose(Socket->TdiConnectionHandle);
Socket->TdiConnectionHandle = NULL;
EnterCriticalSection(&SocketListLock);
if (SocketListHead == Socket)
{
SocketListHead = SocketListHead->NextSocket;
@ -574,6 +578,7 @@ WSPCloseSocket(IN SOCKET Handle,
CurrentSocket = CurrentSocket->NextSocket;
}
}
LeaveCriticalSection(&SocketListLock);
HeapFree(GlobalHeap, 0, Socket);
@ -2314,15 +2319,22 @@ GetSocketStructure(SOCKET Handle)
{
PSOCKET_INFORMATION CurrentSocket;
EnterCriticalSection(&SocketListLock);
CurrentSocket = SocketListHead;
while (CurrentSocket)
{
if (CurrentSocket->Handle == Handle)
{
LeaveCriticalSection(&SocketListLock);
return CurrentSocket;
}
CurrentSocket = CurrentSocket->NextSocket;
}
LeaveCriticalSection(&SocketListLock);
return NULL;
}
@ -2841,6 +2853,9 @@ DllMain(HANDLE hInstDll,
/* Heap to use when allocating */
GlobalHeap = GetProcessHeap();
/* Initialize the lock that protects our socket list */
InitializeCriticalSection(&SocketListLock);
AFD_DbgPrint(MAX_TRACE, ("MSAFD.DLL has been loaded\n"));
break;
@ -2852,6 +2867,10 @@ DllMain(HANDLE hInstDll,
break;
case DLL_PROCESS_DETACH:
/* Delete the socket list lock */
DeleteCriticalSection(&SocketListLock);
break;
}