Implement WIN32 tor_cond_wait using condition variables #30187

Fix bug where running a relay on Windows would use 100% CPU
after some time. Makes Windows >= Vista the required Windows
version to build and run tor.
This commit is contained in:
Daniel Pinto 2020-10-31 18:33:33 +00:00
parent 6e25c49f76
commit f3b9be4422
4 changed files with 46 additions and 96 deletions

5
changes/bug30187 Normal file
View File

@ -0,0 +1,5 @@
o Major bugfixes (relay, windows):
- Fix bug where running a relay on Windows would use 100%
CPU after some time. Makes Windows >= Vista the required
Windows version to build and run tor. Fixes bug 30187;
bugfix on 0.4.5.1-alpha. Patch by Daniel Pinto.

View File

@ -560,14 +560,14 @@ fi
AH_BOTTOM([ AH_BOTTOM([
#ifdef _WIN32 #ifdef _WIN32
/* Defined to access windows functions and definitions for >=WinXP */ /* Defined to access windows functions and definitions for >=WinVista */
# ifndef WINVER # ifndef WINVER
# define WINVER 0x0501 # define WINVER 0x0600
# endif # endif
/* Defined to access _other_ windows functions and definitions for >=WinXP */ /* Defined to access _other_ windows functions and definitions for >=WinVista */
# ifndef _WIN32_WINNT # ifndef _WIN32_WINNT
# define _WIN32_WINNT 0x0501 # define _WIN32_WINNT 0x0600
# endif # endif
/* Defined to avoid including some windows headers as part of Windows.h */ /* Defined to avoid including some windows headers as part of Windows.h */

View File

@ -10,18 +10,32 @@
* functions. * functions.
*/ */
#include "orconfig.h"
#ifdef _WIN32 #ifdef _WIN32
/* For condition variable support */
#ifndef WINVER
#error "orconfig.h didn't define WINVER"
#endif
#ifndef _WIN32_WINNT
#error "orconfig.h didn't define _WIN32_WINNT"
#endif
#if WINVER < 0x0600
#error "winver too low"
#endif
#if _WIN32_WINNT < 0x0600
#error "winver too low"
#endif
#include <windows.h> #include <windows.h>
#include <process.h> #include <process.h>
#include <time.h>
#include "lib/thread/threads.h" #include "lib/thread/threads.h"
#include "lib/log/log.h" #include "lib/log/log.h"
#include "lib/log/util_bug.h" #include "lib/log/util_bug.h"
#include "lib/log/win32err.h" #include "lib/log/win32err.h"
/* This value is more or less total cargo-cult */
#define SPIN_COUNT 2000
/** Minimalist interface to run a void function in the background. On /** Minimalist interface to run a void function in the background. On
* Unix calls fork, on win32 calls beginthread. Returns -1 on failure. * Unix calls fork, on win32 calls beginthread. Returns -1 on failure.
* func should not return, but rather should call spawn_exit. * func should not return, but rather should call spawn_exit.
@ -64,45 +78,24 @@ tor_get_thread_id(void)
int int
tor_cond_init(tor_cond_t *cond) tor_cond_init(tor_cond_t *cond)
{ {
memset(cond, 0, sizeof(tor_cond_t)); InitializeConditionVariable(&cond->cond);
if (InitializeCriticalSectionAndSpinCount(&cond->lock, SPIN_COUNT)==0) {
return -1;
}
if ((cond->event = CreateEvent(NULL,TRUE,FALSE,NULL)) == NULL) {
DeleteCriticalSection(&cond->lock);
return -1;
}
cond->n_waiting = cond->n_to_wake = cond->generation = 0;
return 0; return 0;
} }
void void
tor_cond_uninit(tor_cond_t *cond) tor_cond_uninit(tor_cond_t *cond)
{ {
DeleteCriticalSection(&cond->lock); (void) cond;
CloseHandle(cond->event);
} }
static void
tor_cond_signal_impl(tor_cond_t *cond, int broadcast)
{
EnterCriticalSection(&cond->lock);
if (broadcast)
cond->n_to_wake = cond->n_waiting;
else
++cond->n_to_wake;
cond->generation++;
SetEvent(cond->event);
LeaveCriticalSection(&cond->lock);
}
void void
tor_cond_signal_one(tor_cond_t *cond) tor_cond_signal_one(tor_cond_t *cond)
{ {
tor_cond_signal_impl(cond, 0); WakeConditionVariable(&cond->cond);
} }
void void
tor_cond_signal_all(tor_cond_t *cond) tor_cond_signal_all(tor_cond_t *cond)
{ {
tor_cond_signal_impl(cond, 1); WakeAllConditionVariable(&cond->cond);
} }
int int
@ -152,66 +145,23 @@ int
tor_cond_wait(tor_cond_t *cond, tor_mutex_t *lock_, const struct timeval *tv) tor_cond_wait(tor_cond_t *cond, tor_mutex_t *lock_, const struct timeval *tv)
{ {
CRITICAL_SECTION *lock = &lock_->mutex; CRITICAL_SECTION *lock = &lock_->mutex;
int generation_at_start; DWORD ms = INFINITE;
int waiting = 1; if (tv) {
int result = -1; ms = tv->tv_sec*1000 + (tv->tv_usec+999)/1000;
DWORD ms = INFINITE, ms_orig = INFINITE, startTime, endTime;
if (tv)
ms_orig = ms = tv->tv_sec*1000 + (tv->tv_usec+999)/1000;
EnterCriticalSection(&cond->lock);
++cond->n_waiting;
generation_at_start = cond->generation;
LeaveCriticalSection(&cond->lock);
LeaveCriticalSection(lock);
startTime = GetTickCount();
do {
DWORD res;
res = WaitForSingleObject(cond->event, ms);
EnterCriticalSection(&cond->lock);
if (cond->n_to_wake &&
cond->generation != generation_at_start) {
--cond->n_to_wake;
--cond->n_waiting;
result = 0;
waiting = 0;
goto out;
} else if (res != WAIT_OBJECT_0) {
result = (res==WAIT_TIMEOUT) ? 1 : -1;
--cond->n_waiting;
waiting = 0;
goto out;
} else if (ms != INFINITE) {
endTime = GetTickCount();
if (startTime + ms_orig <= endTime) {
result = 1; /* Timeout */
--cond->n_waiting;
waiting = 0;
goto out;
} else {
ms = startTime + ms_orig - endTime;
} }
BOOL ok = SleepConditionVariableCS(&cond->cond, lock, ms);
if (!ok) {
DWORD err = GetLastError();
if (err == ERROR_TIMEOUT) {
return 1;
} }
/* If we make it here, we are still waiting. */ char *msg = format_win32_error(err);
if (cond->n_to_wake == 0) { log_err(LD_GENERAL, "Error waiting for condition variable: %s", msg);
/* There is nobody else who should wake up; reset tor_free(msg);
* the event. */ return -1;
ResetEvent(cond->event);
} }
out: return 0;
LeaveCriticalSection(&cond->lock);
} while (waiting);
EnterCriticalSection(lock);
EnterCriticalSection(&cond->lock);
if (!cond->n_waiting)
ResetEvent(cond->event);
LeaveCriticalSection(&cond->lock);
return result;
} }
void void

View File

@ -42,12 +42,7 @@ typedef struct tor_cond_t {
#ifdef USE_PTHREADS #ifdef USE_PTHREADS
pthread_cond_t cond; pthread_cond_t cond;
#elif defined(USE_WIN32_THREADS) #elif defined(USE_WIN32_THREADS)
HANDLE event; CONDITION_VARIABLE cond;
CRITICAL_SECTION lock;
int n_waiting;
int n_to_wake;
int generation;
#else #else
#error no known condition implementation. #error no known condition implementation.
#endif /* defined(USE_PTHREADS) || ... */ #endif /* defined(USE_PTHREADS) || ... */