Kendo/PacketPool.cpp at main · kendojitsu/Kendo · 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
#include "stdafx.h"
#include "PacketPool.h"

PacketPool::PacketPool()
{
}


PacketPool::~PacketPool()
{
}

/**
*	@brief	Add packet to pool
*	@param	pkt_data	data pack
*	@param	header		capital
*	@return	-
*/
void PacketPool::add(const struct pcap_pkthdr *header, const u_char *pkt_data)
{
	if (header && pkt_data)
	{
		int pktNum = 1 + m_pkts.size();
		Packet pkt(header, pkt_data, pktNum);
		m_pkts[pktNum] = pkt;
	}
}

/**
*	@brief	Add packet to pool
*	@param	pkt	data pack
*	@return	-
*/
void PacketPool::add(const Packet &pkt)
{
	if(!pkt.isEmpty())
		m_pkts[pkt.num] = pkt;
}

/**
*	@brief	Delete the specified packet from the pool based on the packet number
*	@param	pktNum	Packet number
*	@return	-
*/
void PacketPool::remove(int pktNum)
{
	if (pktNum < 1 || pktNum > m_pkts.size())
		return;
	m_pkts.erase(pktNum);
}

void PacketPool::clear()
{
	if (m_pkts.size() > 0)
		m_pkts.clear();
}

/**
*	@brief	Get the specified data packet from the pool based on the data packet number
*	@param	pktNum	Packet number
*	@return	pkt		Packet reference
*/
Packet& PacketPool::get(int pktNum)
{
	if (m_pkts.count(pktNum) > 0)
		return m_pkts[pktNum];
	return Packet();
}

/**
*	@brief	Get the last packet from the pool
*	@param	pktNum	Packet number
*	@return	pkt		Packet reference
*/
Packet& PacketPool::getLast()
{
	if (m_pkts.count(m_pkts.size()) > 0)
		return m_pkts[m_pkts.size()];
	return Packet();
}

/**
*	@brief	Get the number of data packets in the pool
*	@param	-
*	@return	Number of data packets
*/
int PacketPool::getSize() const
{
	return m_pkts.size();
}

/**
*	@brief	Determine whether the pool is empty
*	@param	-
*	@return	true empty false not empty
*/
bool PacketPool::isEmpty() const
{
	if (m_pkts.size())
		return false;
	return true;
}