summaryrefslogtreecommitdiffstats
path: root/Packet.cpp
blob: 695e3713906e4d276f70439e7fe1961e2ec33cbb (plain) (blame)
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
#include "Packet.hpp"

Packet::Packet(int id) {
    Field fLen;
    fLen.SetVarInt(0);
    m_fields.push_back(fLen);
    Field fId;
    fId.SetVarInt(id);
    m_fields.push_back(fId);
}

Packet Packet::operator=(Packet other) {
    other.swap(*this);
    return *this;
}

void Packet::swap(Packet &other) {
    std::swap(m_fields, other.m_fields);
    std::swap(m_data, other.m_data);
    std::swap(m_parsePtr, other.m_parsePtr);
    std::swap(m_dataLength, other.m_dataLength);
}

void Packet::CopyToBuff(byte *ptr) {
    m_fields[0].SetVarInt(GetLength() - m_fields[0].GetLength());
    for (int i = 0; i < m_fields.size(); i++) {
        m_fields[i].CopyToBuff(ptr);
        ptr += m_fields[i].GetLength();
    }
}

void Packet::ParseField(FieldType type, size_t len) {
    if (type == ByteArray && len == 0)
        throw 118;
    Field f = FieldParser::Parse(type, m_parsePtr, len);
    m_fields.push_back(f);
    m_parsePtr += f.GetLength();
    if (m_parsePtr == m_data + m_dataLength) {
        delete[] m_data;
        m_data = nullptr;
        m_dataLength = 0;
        m_parsePtr = nullptr;
    }
}

Packet::Packet(byte *data) {
    Field fLen = FieldParser::Parse(VarInt, data);
    data += fLen.GetLength();
    Field fId = FieldParser::Parse(VarInt, data);
    data += fId.GetLength();
    m_dataLength = fLen.GetVarInt() - fId.GetLength();
    m_data = new byte[m_dataLength];
    std::copy(data,data+m_dataLength,m_data);
    m_parsePtr = m_data;
    m_fields.push_back(fLen);
    m_fields.push_back(fId);
}

Field &Packet::GetField(int id) {
    if (id < -2 || id >= m_fields.size() - 2)
        throw 111;
    return m_fields[id + 2];
}

size_t Packet::GetLength() {
    size_t len = 0;
    for (int i = 0; i < m_fields.size(); i++)
        len += m_fields[i].GetLength();
    return len + m_dataLength;
}

void Packet::AddField(Field field) {
    m_fields.push_back(field);
}

int Packet::GetId() {
    return m_fields[1].GetVarInt();
}

Packet::Packet(const Packet &other) {
    if (other.m_dataLength > 0) {
        m_dataLength = other.m_dataLength;
        m_data = new byte[m_dataLength];
        m_parsePtr = m_data + (other.m_data - other.m_parsePtr);
        std::copy(other.m_data, other.m_data + m_dataLength, m_data);
    }
    m_fields = other.m_fields;
}

void Packet::ParseFieldArray(Field &field, FieldType type, size_t len) {
    Field f = FieldParser::Parse(type, m_parsePtr, len);
    field.Attach(f);
    m_parsePtr += f.GetLength();
    if (m_parsePtr == m_data + m_dataLength) {
        delete[] m_data;
        m_data = nullptr;
        m_dataLength = 0;
        m_parsePtr = nullptr;
    }
}