Dummy connect

Wiebe Cazemier
1 parent 5dcad433
Showing 4 changed files with 89 additions and 1 deletions
mqttpacket.cpp
mqttpacket.h
types.cpp
types.h
 #include "mqttpacket.h"
 #include <cstring>
 #include <iostream>
+#include <list>
  
 MqttPacket::MqttPacket(char *buf, size_t len, size_t fixed_header_length, Client *sender) :
     bites(len),
@@ -26,6 +27,25 @@ MqttPacket::MqttPacket(const ConnAck &amp;connAck) :
  
 }
  
+MqttPacket::MqttPacket(const SubAck &subAck) :
+    bites(3)
+{
+    packetType = PacketType::SUBACK;
+    char first_byte = static_cast<char>(packetType) << 4;
+    writeByte(first_byte);
+    writeByte((subAck.packet_id & 0xF0) >> 8);
+    writeByte(subAck.packet_id & 0x0F);
+
+    std::vector<char> returnList;
+    for (SubAckReturnCodes code : subAck.responses)
+    {
+        returnList.push_back(static_cast<char>(code));
+    }
+
+    bites.insert(bites.end(), returnList.begin(), returnList.end());
+    bites[1] = returnList.size() + 1; // TODO: make some generic way of calculating the header and use the multi-byte length
+}
+
 void MqttPacket::handle()
 {
     if (packetType == PacketType::CONNECT)
@@ -33,7 +53,7 @@ void MqttPacket::handle()
     else if (packetType == PacketType::PINGREQ)
         std::cout << "PING" << std::endl;
     else if (packetType == PacketType::SUBSCRIBE)
-        std::cout << "Sub" << std::endl;
+        handleSubscribe();
 }
  
 void MqttPacket::handleConnect()
@@ -117,6 +137,34 @@ void MqttPacket::handleConnect()
     }
 }
  
+void MqttPacket::handleSubscribe()
+{
+    uint16_t packet_id = readTwoBytesToUInt16();
+
+    std::list<std::string> subs; // TODO: list of tuples, probably
+    while (remainingAfterPos() > 1)
+    {
+        uint16_t topicLength = readTwoBytesToUInt16();
+        std::string topic(readBytes(topicLength), topicLength);
+        std::cout << sender->repr() << " Subscribed to " << topic << std::endl;
+        subs.push_back(std::move(topic));
+    }
+
+    char flags = readByte();
+
+    SubAck subAck(packet_id, subs);
+    MqttPacket response(subAck);
+    sender->writeMqttPacket(response);
+    sender->writeBufIntoFd();
+}
+
+void MqttPacket::handlePing()
+{
+
+}
+
+
+
 char *MqttPacket::readBytes(size_t length)
 {
     if (pos + length > bites.size())
@@ -154,6 +202,11 @@ uint16_t MqttPacket::readTwoBytesToUInt16()
     return i;
 }
  
+size_t MqttPacket::remainingAfterPos()
+{
+    return bites.size() - pos;
+}
+
  
  
  
@@ -25,14 +25,19 @@ class MqttPacket
     char readByte();
     void writeByte(char b);
     uint16_t readTwoBytesToUInt16();
+    size_t remainingAfterPos();
  
 public:
     PacketType packetType = PacketType::Reserved;
     MqttPacket(char *buf, size_t len, size_t fixed_header_length, Client *sender);
     MqttPacket(const ConnAck &connAck);
+    MqttPacket(const SubAck &subAck);
  
     void handle();
     void handleConnect();
+    void handleSubscribe();
+    void handlePing();
+
     size_t getSize() { return bites.size(); }
     const std::vector<char> &getBites() { return bites; }
  
@@ -5,3 +5,13 @@ ConnAck::ConnAck(ConnAckReturnCodes return_code) :
 {
  
 }
+
+SubAck::SubAck(uint16_t packet_id, const std::list<std::string> &subs) :
+    packet_id(packet_id)
+{
+    // dummy
+    for(size_t i = 0; i < subs.size(); i++)
+    {
+        responses.push_back(SubAckReturnCodes::MaxQoS0);
+    }
+}
 #ifndef TYPES_H
 #define TYPES_H
  
+#include "stdint.h"
+#include <list>
+#include <string>
+
 enum class PacketType
 {
     Reserved = 0,
@@ -46,4 +50,20 @@ public:
     ConnAckReturnCodes return_code;
 };
  
+enum class SubAckReturnCodes
+{
+    MaxQoS0 = 0,
+    MaxQoS1 = 1,
+    MaxQoS2 = 2,
+    Fail = 0x80
+};
+
+class SubAck
+{
+public:
+    uint16_t packet_id;
+    std::list<SubAckReturnCodes> responses;
+    SubAck(uint16_t packet_id, const std::list<std::string> &subs);
+};
+
 #endif // TYPES_H