Проблема как в опубликованном коде Windows, так и в опубликованном коде Linux заключается в том, что аргумент, который вы передаете недавно созданному потоку, является указателем на локальную переменную в стеке родительского потока, и эта локальная переменная, вероятно,были извлечены из стека (и, следовательно, могут быть перезаписаны некоторыми другими данными) до того, как дочерний поток получит возможность начать работу и посмотреть на него.
Следовательно, решение проблемы состоит в том, чтобы убедиться, чтоданные все еще действительны, когда дочерний поток просматривает их.Это можно сделать несколькими способами:
1) Самый простой (и обычно лучший) способ: вместо выделения сокета в стеке (в качестве локальной переменной) вместо этого выделите его из кучи:
// main thread
while (1) {
SOCKET * pClientSocket = (SOCKET *) (malloc(sizeof(SOCKET))); // allocate a SOCKET on the heap
if (pClientSocket == NULL) {printf("malloc() failed!?\n"); break;}
*pClientSocket = accept(ListenSocket, NULL, NULL);
if (*pClientSocket == INVALID_SOCKET) {
free(pClientSocket); // avoid memory leak!
printf("accept failed: %d\n", WSAGetLastError());
closesocket(ListenSocket);
WSACleanup();
return 1;
}
printf("ClientSocket: %d\n", *pClientSocket);
HANDLE thread = CreateThread(NULL, 0, Somethready, pClientSocket, 0, NULL);
if (thread == NULL)
{
closesocket(*pClientSocket); // avoid socket leak!
free(pClientSocket); // avoid memory leak!
printf("CreateThread failed!?\n");
}
}
// child thread
DWORD WINAPI Somethready(void *vargp) {
printf("Thread got evoked\n");
SOCKET *pClientSocket = (SOCKET *)vargp;
SOCKET ClientSocket = *pClientSocket; // make a copy of the heap-allocated SOCKET object into a local variable
free(pClientSocket); // then free the heap-allocated SOCKET to avoid a memory leak
printf("In thread, ClientSocket: %d\n", ClientSocket);
[...]
closesocket(ClientSocket); // don't forget to close the socket when we're done
return 0;
}
Это будет хорошо работать, потому что выделенный в куче объект SOCKET
(на который указывает pClientSocket
) гарантированно не будет уничтожен до тех пор, пока кто-то не вызовет free()
, и этот код оставит егосделать это дочернему потоку после того, как он скопировал свое содержимое в локальную переменную ClientSocket
.
Единственный потенциальный недостаток - легко создать утечку памяти, если вы забудете вызвать free()
в сокете, выделенном для кучи (например, в случае раннего возврата при обработке ошибок), так что вы должны быть осторожны с этим.
2) Способ дешевого взлома.Это включает в себя некоторое потенциально небезопасное / неопределенное приведение в действие, вызывающее поведение, но оно работает на практике, поэтому многие люди делают это.При таком подходе мы просто вставляем значение SOCKET непосредственно в указатель void.Я не рекомендую это, но для полноты:
// main thread
while (1) {
SOCKET ClientSocket = accept(ListenSocket, NULL, NULL);
if (ClientSocket == INVALID_SOCKET) {
printf("accept failed: %d\n", WSAGetLastError());
closesocket(ListenSocket);
WSACleanup();
return 1;
}
printf("ClientSocket: %d\n", *ClientSocket);
HANDLE thread = CreateThread(NULL, 0, Somethready, (void *) ClientSocket, 0, NULL);
if (thread == NULL)
{
closesocket(ClientSocket); // avoid socket leak
printf("CreateThread failed!?\n");
}
}
// child thread
DWORD WINAPI Somethready(void *vargp) {
printf("Thread got evoked\n");
SOCKET ClientSocket = (SOCKET)vargp;
printf("In thread, ClientSocket: %d\n", ClientSocket);
[...]
closesocket(ClientSocket); // don't forget to close the socket itself when we're done
return 0;
}
3) Я слишком умный программист для моего собственного блага: в этом подходе, после нерестадочерний поток, мы используем условную переменную, чтобы заблокировать выполнение основного потока, пока дочерний поток не указал, что он начал работать и больше не использует переменную-указатель на главный поток.Я собираюсь написать это в псевдокоде, так как я не на машине Windows, чтобы проверить его, но это должно дать вам общее представление:
// global variables (or if you don't like global variables, you
// could put these into a struct, along with the SOCKET object,
// and pass a pointer-to-the-struct to the child thread instead)
CONDITION_VARIABLE wait_for_child_thread;
CRITICAL_SECTION critical_section;
InitializeCriticalSection(&critical_section);
InitializeConditionVariable(&wait_for_child_thread);
// main thread
while (1) {
SOCKET ClientSocket = accept(ListenSocket, NULL, NULL);
if (ClientSocket == INVALID_SOCKET) {
printf("accept failed: %d\n", WSAGetLastError());
closesocket(ListenSocket);
WSACleanup();
return 1;
}
printf("ClientSocket: %d\n", *ClientSocket);
HANDLE thread = CreateThread(NULL, 0, Somethready, &ClientSocket, 0, NULL);
if (thread != NULL)
{
// Gotta wait here until the child thread wakes us up,
// otherwise we risk invalidating (&ClientSocket) before he has used it!
EnterCriticalSection(&critical_section);
SleepConditionVariableCS(&wait_for_child_thread, &critical_section, INFINITE);
LeaveCriticalSection(&critical_section);
}
else
{
printf("CreateThread failed!?\n");
}
}
// child thread
DWORD WINAPI Somethready(void *vargp) {
printf("Thread got evoked\n");
SOCKET * pClientSocket = (SOCKET *)vargp;
SOCKET ClientSocket = *pClientSocket; // copy from main-thread's stack to our own stack
// Now that we've made the copy, tell the main thread he can continue execution
WakeConditionVariable(&wait_for_child_thread);
printf("In thread, ClientSocket: %d\n", ClientSocket);
[...]
closesocket(ClientSocket); // don't forget to close the socket itself when we're done
return 0;
}