summaryrefslogtreecommitdiff
path: root/src/lib/evil/evil_main.c
blob: 07dcd4a06da020542ef33dfc3c20e123b5613e43 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include <stdio.h>

#include "evil_private.h"


static int      _evil_init_count = 0;

extern LONGLONG _evil_time_freq;
extern LONGLONG _evil_time_count;
extern long     _evil_time_second;

extern DWORD    _evil_tls_index;

long _evil_systemtime_to_time(SYSTEMTIME st);

int
evil_init(void)
{
   SYSTEMTIME    st;
   LARGE_INTEGER freq;
   LARGE_INTEGER count;
   WORD          second = 59;

   if (++_evil_init_count != 1)
     return _evil_init_count;

   {
      DWORD v;

      v = GetVersion();
      if (!v || ((DWORD)(LOBYTE(LOWORD(v))) < 6))
        {
           fprintf(stderr, "Windows XP not supported anymore, exiting.\n");
           return 0;
        }
   }

   if (!QueryPerformanceFrequency(&freq))
       return 0;

   _evil_time_freq = freq.QuadPart;

   /* be sure that second + 1 != 0 */
   while (second == 59)
     {
        GetSystemTime(&st);
        second = st.wSecond;
     }

   /* retrieve the tick corresponding to the time we retrieve above */
   while (1)
     {
        GetSystemTime(&st);
        QueryPerformanceCounter(&count);
        if (st.wSecond == second + 1)
          break;
     }

   _evil_time_second = _evil_systemtime_to_time(st);
   if (_evil_time_second < 0)
     return --_evil_init_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();

   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;
     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;
}