Diff: main.cpp

Revision:

2:d2fd4c719ea0

Parent:

1:1b58077320a3

Child:

3:9074dfb9fc11

--- a/main.cpp	Sat Jan 19 21:03:47 2019 +0000
+++ b/main.cpp	Thu Jan 24 22:38:20 2019 +0000
@@ -1,18 +1,77 @@
 #include "mbed.h"
 #include "cantcoap.h"
+#include "msg.h"
 
 Serial pc(SERIAL_TX, SERIAL_RX);
 Serial wifi(PA_11, PA_12);
 DigitalOut send(PC_8);
 
 #define PCDBG(...) if(1){ pc.printf(__VA_ARGS__); pc.printf("\r\n");}
+#define BINDBG(buff, len) if(1){ print_bin(buff, len); }
+#define HEXDBG(buff, len) if(1){ print_hex(buff, len);}
 
-CoapPDU* read() {
-    uint8_t buff[1023];
-    int ret = wifi.read(buff, 1023, NULL);
-    CoapPDU *recvPDU = new CoapPDU(buff,ret,1023);
+//Number of retransmissions
+#define RETRANSMISSION 10
+//Timeout in ms
+#define TIMEOUT 2000
+
+//Length of PUT (data) response
+#define PUT_ANS_LEN 9
+//Length of GET (conf) response
+#define GET_ANS_LEN 10
+
+int read_finished = 0;
+uint16_t messageID = 1;
+
+void print_bin(uint8_t* buff, int len) {
+    char str[len];
+    for(int i = 0; i < len; ++i) {
+        if((buff[i] >= 'a' && buff[i] <= 'z') || (buff[i] >= 'A' && buff[i] <= 'Z') || (buff[i] >= '0'&& buff[i] <= '9')) {
+            str[i] = buff[i];
+        } else {
+            str[i] = '.';
+        }
+    }
+    PCDBG("%s", str);
+}
+
+void print_hex(uint8_t* buff, int len) {
+    for(int i = 0; i < len; ++i) {
+        pc.printf("%X ", buff[i]);
+    }
+    pc.printf("\r\n");
+}
+
+void read_cb(int arg) {
+    (void)arg;
+    read_finished = 1;
+}
+
+CoapPDU* read(int len) {
+    uint8_t rx_buff[len];
+    PCDBG("Reading");
+    Timer t;
+    t.start();
+    while(!wifi.readable() && t.read_ms() < TIMEOUT) {}
+    int ret = wifi.read(rx_buff, len, event_callback_t(&read_cb));
+    while(read_finished == 0 && t.read_ms() < TIMEOUT) {}
+    t.stop();
+    wifi.abort_read();
+    PCDBG("Finished Reading, ret = %d, read_finished = %d, time = %d", ret, read_finished, t.read_ms())
+    HEXDBG((uint8_t*)rx_buff, len);
+    if(read_finished != 1) {
+        read_finished = 0;
+        return NULL;
+    }
+    read_finished = 0;
+    CoapPDU* recvPDU = new CoapPDU((uint8_t*)rx_buff, len);
+    PCDBG("PDU created");
     if(recvPDU->validate()) {
-        //recvPDU.printHuman();
+        PCDBG("Validated");
+        recvPDU->printHuman();
+        PCDBG("Code = %d", recvPDU->getCode());
+        PCDBG("Payload:");
+        HEXDBG(recvPDU->getPayloadPointer(), len);
         return recvPDU;
     }
     return NULL;
@@ -23,23 +82,88 @@
     send.write(1);
     wait(0.05);
     send.write(0);
-    PCDBG("PoszLo!");
 }
 
 int sendd(uint8_t* buff, int len) {
     while(!wifi.writable()) {
-        PCDBG("Wifi not wirtable");
-        wait(0.5);
+        //PCDBG("Wifi not wirtable");
+        //wait(0.5);
+    }
+    PCDBG("Sending:");
+    HEXDBG(buff, len);
+    //For some mysterious reason it doesn't print first character
+    int ret = wifi.write(buff - 1, len + 1, event_callback_t(&send_cb));
+    return ret;
+}
+
+CoapPDU* send_or_fail(uint8_t* buff, int len, int ans_len) {
+    int ret;
+    CoapPDU* ans;
+    for(int i = 0; i < RETRANSMISSION; ++i) {
+        ret = sendd(buff, len);
+        if(ret < 0) {
+            PCDBG("Send Failed");
+            return NULL;
+        }
+        ans = read(ans_len);
+        if(ans == NULL) {
+            PCDBG("Read Failed");
+        } else if(ans->getType() != CoapPDU::COAP_ACKNOWLEDGEMENT || ans->getMessageID() != messageID) {
+            PCDBG("Wrong answer IS: type = %d, id = %d SHOULD BE: type = %d, id = %d", ans->getType(), ans->getMessageID(), CoapPDU::COAP_ACKNOWLEDGEMENT, messageID);
+        } else {
+            PCDBG("ACK get");
+            return ans;
+        }
     }
-    //not working - dunno how this works
-    //int ret = wifi.write(buff, len, event_callback_t(&send_cb));
-    PCDBG((char*)buff);
-    //seems the same as wifi.write(buff, len, NULL);
-    int ret = wifi.printf((char*)buff);
-    send.write(1);
-    wait(0.05);
-    send.write(0);
-    return ret;
+    PCDBG("Retransmission limit reached");
+    return NULL;
+}
+
+
+void preparePDU(CoapPDU& coapPDU) {
+    coapPDU.setVersion(1);
+    coapPDU.setType(CoapPDU::COAP_CONFIRMABLE);
+    coapPDU.setMessageID(++messageID);
+    coapPDU.setToken((uint8_t*)"A", 1);
+}
+
+int send_data(int type, int data) {
+    uint8_t buff[5];
+    Msg::construct_data_msg(type, data, buff, 5);
+    CoapPDU pdu = CoapPDU();
+    preparePDU(pdu);
+    pdu.setCode(CoapPDU::COAP_PUT);
+    pdu.setURI((char*)"data",4);
+    pdu.setPayload(buff, 5);
+    CoapPDU* ans =  send_or_fail(pdu.getPDUPointer(), pdu.getPDULength(), PUT_ANS_LEN);
+    if(ans == NULL) return -1;
+    delete ans;
+    return 0;
+}
+
+int get_config(uint8_t type) {
+    uint8_t buff[1];
+    buff[0] = type;
+    CoapPDU pdu = CoapPDU();
+    preparePDU(pdu);
+    pdu.setCode(CoapPDU::COAP_POST);
+    pdu.setURI((char*)"conf/",4);
+    pdu.setPayload(buff, 1);
+    CoapPDU* ans = send_or_fail(pdu.getPDUPointer(), pdu.getPDULength(), GET_ANS_LEN);
+    if(ans == NULL) {
+        PCDBG("Failed to get config");
+        return -1;
+    }
+    uint8_t* config = ans->getPayloadPointer();
+    int conf = 0;
+    int a = 1;
+    for(int i = 0; i < 4; ++i) {
+        conf += config[i] * a;
+        a *= 256;
+    }
+    delete ans;
+    PCDBG("New config for type %d is %d", type, conf);
+    return conf;
 }
 
 int main() {
@@ -47,33 +171,12 @@
     wifi.baud(115200);
     PCDBG("Started")
     CoapPDU* pdu = new CoapPDU();
-    uint16_t messageID = 0;
-    int ret;
+    //int ret;
+    int data = 1;
     while(1) {
-        ++messageID;
-        pdu->setType(CoapPDU::COAP_CONFIRMABLE);
-        pdu->setCode(CoapPDU::COAP_GET);
-        pdu->setToken((uint8_t*)"1234",4);
-        pdu->setMessageID(messageID);
-        pdu->setURI((char*)"test",4);
-        pdu->setPayload((uint8_t*)"Tralalala", 9);
-    
-        PCDBG("Sending packet");
-        // send packet 
-        ret = sendd(pdu->getPDUPointer(), pdu->getPDULength());
-        PCDBG("Packet send");
-        
-        //// get response
-//        CoapPDU* res = read();
-//        PCDBG("Response read");
-//        CoapPDU::Code code = res->getCode();
-//        uint8_t len = res->getPayloadLength();
-//        // getPayloadPointerCopy to get allocated copy of payload
-//        uint8_t* payload = res->getPayloadPointer();
-//        //char status[10];
-//        //if(code == COAP_VALID) status = "OK"; else status = "UNKNOWN";
-//        PCDBG("%s, %s", code, payload);
-//        delete res;
-        wait(1);
+        //send_data(1, ++data); 
+        //wait(5);
+        get_config(1);
+        wait(5);
     }
 }
\ No newline at end of file