// Local imports
#include "crypto.h"
#include "rng.h"

#include <WiFi.h>
#include <WiFiClient.h>
// Energia constructors
#include "Arduino.h"

#define MLEN 32

void setup();
void loop();

char ssid[] = "";
char pass[] = "";

uint16_t ID = 1050;  // Device ID
uint16_t port = 9000;
IPAddress server(192,168,0,10);
WiFiClient wclient;

uint8_t aes_key[16] = {
        0x03, 0x02, 0x01, 0x00,
        0x07, 0x06, 0x05, 0x04,
        0x0b, 0x0a, 0x09, 0x08,
        0x0f, 0x0e, 0x0d, 0x0c
};

typedef struct {
    uint16_t ID;
    uint16_t LEN;
} header;

uint8_t AESData[MLEN], *AESResult, AESResult2[MLEN];
uint32_t AESDataLength=MLEN;

void setup() {
    // initialising
    Serial.begin(9600);

    Serial.print("Setup RNG ..");
    RNGSetup();
    Serial.println("OK");

    Serial.print("Setup AES ..");
    AESSetup(AES_KEY_128BIT, aes_key);
    Serial.println("OK");

    Serial.print("Setup WiFi ..");
    WiFi.begin(ssid, pass);
    while (WiFi.status() != WL_CONNECTED) {delay(200);}
    Serial.println("OK");
    
    Serial.print("Setup DHCP ..");
    while (WiFi.localIP() == INADDR_NONE) {delay(200);}
    Serial.print("OK ");
    Serial.println(WiFi.localIP());
    
    Serial.print("Setup Socket ..");
    while(wclient.connect(server, port) == false) {delay(200);}
    Serial.println("OK");

    //String msg = "This is secret!";
    //msg.toCharArray((char *)AESData, MLEN);
//
//    Serial.print("KEY: ");
//    for(int i=0;i<16;i++) {
//        Serial.print(aes_key[i], HEX);
//        Serial.print(' ');
//    }
//    Serial.println();
//
//    Serial.print("IV: ");
//    for(int i=0;i<16;i++) {
//        Serial.print(aes_iv[i], HEX);
//        Serial.print(' ');
//    }
//    Serial.println();
//
//    Serial.print("Message: ");
//    for(int i=0;i<MLEN;i++) {
//        Serial.print(AESData[i], HEX);
//        Serial.print(' ');
//    }
//    Serial.println();
//
    //AESResult = AESEncrypt(AESData, AESDataLength);


//    AESCrypt(AES_DIR_ENCRYPT, aes_iv, AESData, AESResult, AESDataLength);
    //Serial.print("Encrypted: ");
    //for(uint32_t i=0;i<AESDataLength+getBlockSize();i++) {
        //Serial.print(AESResult[i], HEX);
        //Serial.print(' ');
    //}
    //Serial.println();

//    AESCrypt(AES_DIR_DECRYPT, &aesv, AESResult, AESResult2, AESDataLength);
//    Serial.print("Decrypted: ");
//    for(int i=0;i<MLEN;i++) {
//        Serial.print(AESResult2[i], HEX);
//        Serial.print(' ');
//    }
//    Serial.println();
}

uint8_t buffer[32] = {0};
uint16_t bufferPnt = 0;

void send() {
    uint32_t resLen = 0;
    Serial.print("encrypting..");
    uint8_t *res = AESEncrypt(buffer, 32, &resLen);
    Serial.println(resLen);
    header h = {ID, resLen};
    wclient.write((uint8_t *) &h, sizeof(h)); 
    wclient.write(res, (uint16_t) resLen);
    free(res);
    bufferPnt = 0;
}

void loop() {
    while (Serial.available() > 0) {
        uint8_t val = Serial.read();
        Serial.write(val); // echo
        if ((val == 8 || val == 127) && bufferPnt > 0) {
            bufferPnt--;
            continue;
        }
        if(val == 13) {
            send();
            continue;
        }
        if(bufferPnt >= 255) send();
        buffer[bufferPnt] = val;
        bufferPnt++;
    }
}