UDPNetworking/Packet.cpp at master · mbuckley2000/UDPNetworking · 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
//
// Created by Matt on 12/09/2016.
//

#include <iostream>
#include "Packet.h"

#define PROTOCOLID 987654365

Packet::Packet() {}

void Packet::setType(Type packetType) {
    type = packetType;
}

const Packet::Type Packet::getType() {
    return type;
}

UDPAddress Packet::getSender() {
    return sender;
}

Packet::Packet(char *data, unsigned int dataSize) {
    PacketSerialiser serialiser = PacketSerialiser(data, dataSize);
    deserialise(&serialiser);
}

int Packet::deserialise(PacketSerialiser *serialiser) {
    int protID = serialiser->readInt();
    if (protID == protID) { //protID == PROTOCOLID
        type = (Type) serialiser->readInt();
        switch (type) {
            case handshake: {
                data.handshakeData.version = serialiser->readInt();
                return 0;
            }

            default: {
                std::cerr << "Unrecognised packet type" << std::endl;
                return 1;
            }
        }
    } else {
        std::cerr << "Received packet from different protocol" << std::endl;
        return 1;
    }
}

char *Packet::serialise(PacketSerialiser *serialiser) {
    serialiser->writeInt(PROTOCOLID);
    serialiser->writeInt((int) type);
    switch (type) {
        case handshake: {
            serialiser->writeInt(data.handshakeData.version);
            break;
        }

        default: {
            std::cerr << "Unrecognised packet type" << std::endl;
            break;
        }
    }

    return serialiser->getBuffer();
}

void Packet::setSender(UDPAddress senderAddress) {
    sender = senderAddress;
}