Yellow/UDPHeader.cpp at master · Benbiz/Yellow · 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
#include "stdafx.h"
#include "UDPHeader.h"

Yellow::UDPHeader::UDPHeader(void *beg)
  : udp(static_cast<struct udphdr *>(beg)), buff(static_cast<unsigned char *>(beg))
{
}

Yellow::UDPHeader::UDPHeader(const UDPHeader &u)
{
  udp = u.udp;
  buff = u.buff;
}

Yellow::UDPHeader &Yellow::UDPHeader::operator=(const UDPHeader &u)
{
  udp = u.udp;
  buff = u.buff;
  return *this;
}

Yellow::UDPHeader::~UDPHeader()
{
}

const unsigned char			*Yellow::UDPHeader::operator[](int idx) const
{
  if (idx >= getLenght())
    return NULL;
  return (&buff[idx]);
}

const int				Yellow::UDPHeader::getLenght() const
{
  return (sizeof (struct udphdr));
}

const unsigned short int			Yellow::UDPHeader::getSourcePort() const
{
  return (udp->source);
}

void					Yellow::UDPHeader::setSourcePort(const unsigned short int source)
{
  udp->source = htons(source);
}

const unsigned short int			Yellow::UDPHeader::getDestPort() const
{
  return ntohs(udp->dest);
}

void					Yellow::UDPHeader::setDestPort(const unsigned short int dest)
{
  udp->dest = htons(dest);
}

const unsigned short int			Yellow::UDPHeader::getUDPLenght() const
{
  return ntohs(udp->len);
}

void					Yellow::UDPHeader::setUDPLenght(const unsigned short int len)
{
  udp->len = htons(len);
}

const unsigned short int			Yellow::UDPHeader::getChecksum() const
{
  return ntohs(udp->check);
}

void					Yellow::UDPHeader::setChecksum(const unsigned short int check)
{
  udp->check = htons(check);
}