aboutsummaryrefslogtreecommitdiff
path: root/src/packet.c
blob: e50dc130a22cfd513d60cd4fcb41390fc4400a16 (plain)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#include "packet.h"
#include "log.h"

#include <string.h>
#include <stdio.h>
#include <assert.h>
#include <errno.h>

const char *MqttPacketName(int type)
{
    switch (type)
    {
        case MqttPacketTypeConnect: return "CONNECT";
        case MqttPacketTypeConnAck: return "CONNACK";
        case MqttPacketTypePublish: return "PUBLISH";
        case MqttPacketTypePubAck: return "PUBACK";
        case MqttPacketTypePubRec: return "PUBREC";
        case MqttPacketTypePubRel: return "PUBREL";
        case MqttPacketTypePubComp: return "PUBCOMP";
        case MqttPacketTypeSubscribe: return "SUBSCRIBE";
        case MqttPacketTypeSubAck: return "SUBACK";
        case MqttPacketTypeUnsubscribe: return "UNSUBSCRIBE";
        case MqttPacketTypeUnsubAck: return "UNSUBACK";
        case MqttPacketTypePingReq: return "PINGREQ";
        case MqttPacketTypePingResp: return "PINGRESP";
        case MqttPacketTypeDisconnect: return "DISCONNECT";
        default: return NULL;
    }
}

static inline size_t MqttPacketStructSize(int type)
{
    switch (type)
    {
        case MqttPacketTypeConnect: return sizeof(MqttPacketConnect);
        case MqttPacketTypeConnAck: return sizeof(MqttPacketConnAck);
        case MqttPacketTypePublish: return sizeof(MqttPacketPublish);
        case MqttPacketTypePubAck:
        case MqttPacketTypePubRec:
        case MqttPacketTypePubRel:
        case MqttPacketTypePubComp: return sizeof(MqttPacket);
        case MqttPacketTypeSubscribe: return sizeof(MqttPacketSubscribe);
        case MqttPacketTypeSubAck: return sizeof(MqttPacketSubAck);
        case MqttPacketTypeUnsubscribe: return sizeof(MqttPacketUnsubscribe);
        case MqttPacketTypeUnsubAck: return sizeof(MqttPacket);
        case MqttPacketTypePingReq: return sizeof(MqttPacket);
        case MqttPacketTypePingResp: return sizeof(MqttPacket);
        case MqttPacketTypeDisconnect: return sizeof(MqttPacket);
        default: return (size_t) -1;
    }
}

MqttPacket *MqttPacketNew(int type)
{
    MqttPacket *packet = NULL;

    packet = (MqttPacket *) calloc(1, MqttPacketStructSize(type));
    if (!packet)
        return NULL;

    packet->type = type;

    /* this will make sure that TAILQ_PREV does not segfault if a message
       has not been added to a list at any point */
    packet->messages.tqe_prev = &packet->messages.tqe_next;

    return packet;
}

MqttPacket *MqttPacketWithIdNew(int type, uint16_t id)
{
    MqttPacket *packet = MqttPacketNew(type);
    if (!packet)
        return NULL;
    packet->id = id;
    return packet;
}

void MqttPacketFree(MqttPacket *packet)
{
    if (MqttPacketType(packet) == MqttPacketTypeConnect)
    {
        MqttPacketConnect *p = (MqttPacketConnect *) packet;
        StringBufDeinit(&p->clientId);
        StringBufDeinit(&p->willTopic);
        StringBufDeinit(&p->willMessage);
        StringBufDeinit(&p->userName);
        StringBufDeinit(&p->password);
    }
    else if (MqttPacketType(packet) == MqttPacketTypePublish)
    {
        MqttPacketPublish *p = (MqttPacketPublish *) packet;
        StringBufDeinit(&p->topicName);
        StringBufDeinit(&p->message);
    }
    else if (MqttPacketType(packet) == MqttPacketTypeSubscribe)
    {
        MqttPacketSubscribe *p = (MqttPacketSubscribe *) packet;
        StringBufDeinit(&p->topicFilter);
    }
    else if (MqttPacketType(packet) == MqttPacketTypeUnsubscribe)
    {
        MqttPacketUnsubscribe *p = (MqttPacketUnsubscribe *) packet;
        StringBufDeinit(&p->topicFilter);
    }
    free(packet);
}

int MqttPacketHasId(const MqttPacket *packet)
{
    switch (packet->type)
    {
        case MqttPacketTypePublish:
            return MqttPacketPublishQos(packet) > 0;

        case MqttPacketTypePubAck:
        case MqttPacketTypePubRec:
        case MqttPacketTypePubRel:
        case MqttPacketTypePubComp:
        case MqttPacketTypeSubscribe:
        case MqttPacketTypeSubAck:
        case MqttPacketTypeUnsubscribe:
        case MqttPacketTypeUnsubAck:
            return 1;

        default:
            return 0;
    }
}