NeoPackets/main.cpp at main · NeoniteDev/NeoPackets · GitHub

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
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <IPv4Layer.h>
#include <Packet.h>
#include <PcapFileDevice.h>
#include <PcapLiveDevice.h>
#include "util.hpp"
#include <vector>
#include <map>
#include <PayloadLayer.h>

#define BUFLEN 1024
#define PORT 9001

struct sockaddr_in server, si_other;
WSADATA wsa;
SOCKET s;

pcpp::PcapFileReaderDevice reader("E:\\source\\repos\\NeoPackets\\bin\\solo.pcap");
pcpp::RawPacket rawPacket;
pcpp::PayloadLayer* payloadPacket = nullptr;
char* payload = nullptr;
int payloadLen = 0;

void packetLoop()
{

	pcpp::Packet parsedPacket(&rawPacket);
	pcpp::IPv4Address srcIP = parsedPacket.getLayerOfType<pcpp::IPv4Layer>()->getSrcIpAddress();
	if (!srcIP.toString().rfind("192", 0) == 0)
	{
		payloadPacket = parsedPacket.getLayerOfType<pcpp::PayloadLayer>();
	}
	else
	{
		reader.getNextPacket(rawPacket);
	}

	payload = (char*)payloadPacket->getPayload();

	payloadLen = payloadPacket->getPayloadLen();

}

void main(int argc, char* argv[])
{

	if (!reader.open())
	{
		printd("[x] Error opening the pcap file\n");
		return;
	}

	//printd("[+] Packets are ready to be casted!.\n");

	int slen = sizeof(si_other);

	if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0)
	{
		printd("[x] Failed. Error Code : ", WSAGetLastError());
		return exit(EXIT_FAILURE);
	}

	if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == INVALID_SOCKET)
	{
		printd("[x] Could not create socket : ", WSAGetLastError());
		return exit(EXIT_FAILURE);
	}

	server.sin_family = AF_INET;
	server.sin_addr.s_addr = INADDR_ANY;
	server.sin_port = htons(PORT);

	if (bind(s, (struct sockaddr*)&server, sizeof(server)) == SOCKET_ERROR)
	{
		printd("[x] Bind failed with error code : ", WSAGetLastError());
		return exit(EXIT_FAILURE);
	}

	printd("[=] Server was binded to port ", PORT, "\n");

	//TODO: make a thread.
	//TODO: chop and filter all payloads before casting it.

	char buf[BUFLEN];

	reader.getNextPacket(rawPacket);
	HANDLE packetsThread = CreateThread(NULL, NULL, (LPTHREAD_START_ROUTINE)&packetsThread, NULL, NULL, NULL);

	while (true)
	{
		if (recvfrom(s, buf, BUFLEN, 0, (struct sockaddr*)&si_other, &slen) != SOCKET_ERROR && reader.getNextPacket(rawPacket))
		{
				if (sendto(s, payload, payloadLen, 0, (struct sockaddr*)&si_other, slen) == SOCKET_ERROR)
				{
					printd("[x] Error while sending, error code : ", WSAGetLastError());
					return exit(EXIT_FAILURE);
				}
		}
		else
		{
			printd("[x] Error while receiving, error code : ", WSAGetLastError());
			return exit(EXIT_FAILURE);
		}
	}

	closesocket(s);
	WSACleanup();
	reader.close();
}