2008-11-29 02:56:20 -08:00
|
|
|
#ifdef HAVE_CONFIG_H
|
|
|
|
# include <config.h>
|
|
|
|
#endif
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2015-12-07 20:53:23 -08:00
|
|
|
#include <stdio.h>
|
|
|
|
|
2021-09-03 01:10:07 -07:00
|
|
|
#include <windows.h>
|
|
|
|
|
2009-02-24 09:44:59 -08:00
|
|
|
#include "evil_private.h"
|
|
|
|
|
2021-09-03 01:10:07 -07:00
|
|
|
static UINT _evil_time_period = 1;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2009-10-07 22:01:50 -07:00
|
|
|
static int _evil_init_count = 0;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2009-02-22 11:20:23 -08:00
|
|
|
extern LONGLONG _evil_time_freq;
|
|
|
|
extern LONGLONG _evil_time_count;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2017-12-10 22:06:02 -08:00
|
|
|
extern DWORD _evil_tls_index;
|
|
|
|
|
2008-11-21 14:57:45 -08:00
|
|
|
int
|
2010-04-01 13:13:40 -07:00
|
|
|
evil_init(void)
|
2008-11-21 14:57:45 -08:00
|
|
|
{
|
2009-02-22 11:20:23 -08:00
|
|
|
LARGE_INTEGER freq;
|
|
|
|
LARGE_INTEGER count;
|
2021-09-03 01:10:07 -07:00
|
|
|
TIMECAPS tc;
|
|
|
|
MMRESULT res;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2009-10-07 22:01:50 -07:00
|
|
|
if (++_evil_init_count != 1)
|
|
|
|
return _evil_init_count;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2021-09-03 01:10:07 -07:00
|
|
|
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;
|
|
|
|
|
2020-07-28 13:26:35 -07:00
|
|
|
QueryPerformanceFrequency(&freq);
|
2009-10-07 22:01:50 -07:00
|
|
|
|
2009-02-22 11:20:23 -08:00
|
|
|
_evil_time_freq = freq.QuadPart;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2020-07-28 13:26:35 -07:00
|
|
|
QueryPerformanceCounter(&count);
|
2009-10-07 22:01:50 -07:00
|
|
|
|
2009-02-22 11:20:23 -08:00
|
|
|
_evil_time_count = count.QuadPart;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2009-10-07 22:01:50 -07:00
|
|
|
if (!evil_sockets_init())
|
2009-10-08 14:14:55 -07:00
|
|
|
return --_evil_init_count;
|
2009-06-13 01:25:28 -07:00
|
|
|
|
2008-11-21 14:57:45 -08:00
|
|
|
return _evil_init_count;
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
2010-04-01 13:13:40 -07:00
|
|
|
evil_shutdown(void)
|
2008-11-21 14:57:45 -08:00
|
|
|
{
|
2015-02-05 02:24:36 -08:00
|
|
|
/* _evil_init_count should not go below zero. */
|
|
|
|
if (_evil_init_count < 1)
|
|
|
|
{
|
2015-02-07 04:45:05 -08:00
|
|
|
printf("Evil shutdown called without calling evil init.\n");
|
2015-02-05 02:24:36 -08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2009-10-07 22:01:50 -07:00
|
|
|
if (--_evil_init_count != 0)
|
|
|
|
return _evil_init_count;
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2009-10-07 22:01:50 -07:00
|
|
|
evil_sockets_shutdown();
|
2008-11-21 14:57:45 -08:00
|
|
|
|
2021-09-03 01:10:07 -07:00
|
|
|
timeEndPeriod(_evil_time_period);
|
|
|
|
|
2008-11-21 14:57:45 -08:00
|
|
|
return _evil_init_count;
|
|
|
|
}
|
2017-12-10 22:06:02 -08:00
|
|
|
|
|
|
|
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;
|
2019-05-16 10:49:39 -07:00
|
|
|
/* No break: Initialize the index for first thread. */
|
|
|
|
/* fall through */
|
2017-12-10 22:06:02 -08:00
|
|
|
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;
|
|
|
|
}
|