113 lines
2.2 KiB
C
113 lines
2.2 KiB
C
#ifdef HAVE_CONFIG_H
|
|
# include <config.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <windows.h>
|
|
|
|
#include "evil_private.h"
|
|
|
|
static UINT _evil_time_period = 1;
|
|
|
|
static int _evil_init_count = 0;
|
|
|
|
extern LONGLONG _evil_time_freq;
|
|
extern LONGLONG _evil_time_count;
|
|
|
|
extern DWORD _evil_tls_index;
|
|
|
|
int
|
|
evil_init(void)
|
|
{
|
|
LARGE_INTEGER freq;
|
|
LARGE_INTEGER count;
|
|
TIMECAPS tc;
|
|
MMRESULT res;
|
|
|
|
if (++_evil_init_count != 1)
|
|
return _evil_init_count;
|
|
|
|
res = timeGetDevCaps(&tc, sizeof(TIMECAPS));
|
|
if (res != MMSYSERR_NOERROR)
|
|
return --_evil_init_count;
|
|
|
|
_evil_time_period = tc.wPeriodMin;
|
|
res = timeBeginPeriod(_evil_time_period);
|
|
if (res != TIMERR_NOERROR)
|
|
return --_evil_init_count;
|
|
|
|
QueryPerformanceFrequency(&freq);
|
|
|
|
_evil_time_freq = freq.QuadPart;
|
|
|
|
QueryPerformanceCounter(&count);
|
|
|
|
_evil_time_count = count.QuadPart;
|
|
|
|
if (!evil_sockets_init())
|
|
return --_evil_init_count;
|
|
|
|
return _evil_init_count;
|
|
}
|
|
|
|
int
|
|
evil_shutdown(void)
|
|
{
|
|
/* _evil_init_count should not go below zero. */
|
|
if (_evil_init_count < 1)
|
|
{
|
|
printf("Evil shutdown called without calling evil init.\n");
|
|
return 0;
|
|
}
|
|
|
|
if (--_evil_init_count != 0)
|
|
return _evil_init_count;
|
|
|
|
evil_sockets_shutdown();
|
|
|
|
timeEndPeriod(_evil_time_period);
|
|
|
|
return _evil_init_count;
|
|
}
|
|
|
|
BOOL WINAPI DllMain(HINSTANCE inst, DWORD reason, LPVOID reserved)
|
|
{
|
|
LPVOID data;
|
|
|
|
switch (reason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
_evil_tls_index = TlsAlloc();
|
|
if (_evil_tls_index == TLS_OUT_OF_INDEXES)
|
|
return FALSE;
|
|
/* No break: Initialize the index for first thread. */
|
|
/* fall through */
|
|
case DLL_THREAD_ATTACH:
|
|
data = (LPVOID)LocalAlloc(LPTR, 4096);
|
|
if (!data)
|
|
return FALSE;
|
|
if (!TlsSetValue(_evil_tls_index, data))
|
|
return FALSE;
|
|
break;
|
|
case DLL_THREAD_DETACH:
|
|
data = TlsGetValue(_evil_tls_index);
|
|
if (data)
|
|
LocalFree((HLOCAL)data);
|
|
break;
|
|
case DLL_PROCESS_DETACH:
|
|
data = TlsGetValue(_evil_tls_index);
|
|
if (data)
|
|
LocalFree((HLOCAL)data);
|
|
TlsFree(_evil_tls_index);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
(void)inst;
|
|
(void)reserved;
|
|
}
|