-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.cpp
107 lines (86 loc) · 1.96 KB
/
main.cpp
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
#include "main.h"
Sniffer RSSniffer;
extern "C" DLL_EXPORT void SetupSniffer()
{
RSSniffer = Sniffer();
}
extern "C" DLL_EXPORT void SetInterface(int Interf)
{
RSSniffer.SetInterface(Interf);
}
extern "C" DLL_EXPORT int GetInterface()
{
return RSSniffer.GetInterface();
}
extern "C" DLL_EXPORT bool Initialize()
{
return RSSniffer.Initialize();
}
extern "C" DLL_EXPORT void DeInitialize()
{
RSSniffer.DeInitialize();
}
extern "C" DLL_EXPORT void Start()
{
RSSniffer.Start();
}
extern "C" DLL_EXPORT void Stop()
{
RSSniffer.Stop();
}
extern "C" DLL_EXPORT std::string GetIP(std::string Address)
{
return RSSniffer.GetIP(Address);
}
extern "C" DLL_EXPORT void RemoveDestinationIP(std::string OldIP)
{
RSSniffer.RemoveDestinationIP(OldIP);
}
extern "C" DLL_EXPORT void RemoveSourceIP(std::string OldIP)
{
RSSniffer.RemoveSourceIP(OldIP);
}
extern "C" DLL_EXPORT void AddDestinationIP(std::string NewIP)
{
RSSniffer.AddDestinationIP(NewIP);
}
extern "C" DLL_EXPORT void AddSourceIP(std::string NewIP)
{
RSSniffer.AddSourceIP(NewIP);
}
extern "C" DLL_EXPORT void FilterIP(bool FilterIPs)
{
RSSniffer.FilterIP(FilterIPs);
}
extern "C" DLL_EXPORT void ListenSource(bool Listen)
{
RSSniffer.ListenSource(Listen);
}
extern "C" DLL_EXPORT void ListenDestination(bool Listen)
{
RSSniffer.ListenDestination(Listen);
}
/*extern "C" DLL_EXPORT void Test()
{
RSSniffer.Test();
}*/
extern "C" DLL_EXPORT BOOL APIENTRY DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
{
switch (fdwReason)
{
case DLL_PROCESS_ATTACH:
// attach to process
// return FALSE to fail DLL load
break;
case DLL_PROCESS_DETACH:
// detach from process
break;
case DLL_THREAD_ATTACH:
// attach to thread
break;
case DLL_THREAD_DETACH:
// detach from thread
break;
}
return TRUE; // succesful
}