#include "crypto_aead.h" #include "api.h" #include "uartp.h" #define MAX_BYTES 100 //#define DEBUG #define CRYPTO_BUSY A7 uint8_t npub[CRYPTO_NPUBBYTES]; uint8_t nsec[CRYPTO_NSECBYTES]; uint8_t k[CRYPTO_KEYBYTES]; uint8_t ad[MAX_BYTES]; unsigned long long int adlen; uint8_t m[MAX_BYTES]; unsigned long long int mlen; uint8_t c[MAX_BYTES]; unsigned long long int clen; int res; void setup(); void loop(); #ifdef __cplusplus extern "C" { #endif void uart_wbyte(uint8_t x) { Serial1.write((uint8_t) (x)); } uint8_t uart_rbyte() { int r; do { r = Serial1.read(); } while (r == -1); return (uint8_t) (0xff & r); } #ifdef __cplusplus } #endif int dbg_printf(const char *format, ...) { int r = 0; #ifdef DEBUG #define DEBUG_BUF_LEN 80 char printbuf[DEBUG_BUF_LEN+2]; va_list vargs; va_start(vargs, format); r = vsnprintf(printbuf+1, DEBUG_BUF_LEN, format, vargs); va_end(vargs); if (r < 0) { memcpy(printbuf+1, "DEBUG ERROR\r\n", 13); r = 13; } r = r > DEBUG_BUF_LEN ? DEBUG_BUF_LEN : r; printbuf[0] = 0xde; // Debug messages should start with "\xde" uartp_send(printbuf, r+1); #endif return r; } void my_assert(bool b) { if (b) return; dbg_printf("Assertion failed\r\n"); for(;;) yield(); } void setup() { Serial1.begin(115200); pinMode(CRYPTO_BUSY, OUTPUT); digitalWrite(CRYPTO_BUSY, HIGH); delay(100); Serial1.print("Hello, World!"); } void loop() { static uint8_t buf[256]; uint16_t len = uartp_recv(buf, 255); uint8_t action = buf[0]; if (len == 0 || len > 255) return; uint16_t l = len - 1; uint16_t rl = 0; uint8_t *var = buf+1; switch (action) { case 'm': my_assert(l <= MAX_BYTES); memcpy(m, var, l); mlen = l; break; case 'c': my_assert(l <= MAX_BYTES); memcpy(c, var, l); clen = l; break; case 'a': my_assert(l <= MAX_BYTES); memcpy(ad, var, l); adlen = l; break; case 'k': my_assert(l == CRYPTO_KEYBYTES); memcpy(k, var, l); break; case 'p': my_assert(l == CRYPTO_NPUBBYTES); memcpy(npub, var, l); break; case 's': my_assert(l == CRYPTO_NSECBYTES); memcpy(nsec, var, l); break; case 'e': noInterrupts(); asm("nop"); digitalWrite(CRYPTO_BUSY, LOW); res = crypto_aead_encrypt(c, &clen, m, mlen, ad, adlen, nsec, npub, k); digitalWrite(CRYPTO_BUSY, HIGH); asm("nop"); interrupts(); break; case 'd': noInterrupts(); asm("nop"); digitalWrite(CRYPTO_BUSY, LOW); res = crypto_aead_decrypt(m, &mlen, nsec, c, clen, ad, adlen, npub, k); digitalWrite(CRYPTO_BUSY, HIGH); asm("nop"); interrupts(); break; case'M': var = m; rl = mlen; break; case'C': var = c; rl = clen; break; case'A': var = ad; rl = adlen; break; case'K': var = k; rl = CRYPTO_KEYBYTES; break; case'P': var = npub; rl = CRYPTO_NPUBBYTES; break; case'S': var = nsec; rl = CRYPTO_NSECBYTES; break; case'R': var = (uint8_t *) &res; rl = sizeof(res); break; default: dbg_printf("Unknown command\r\n"); my_assert(false); } buf[0] = action; memcpy(buf+1, var, rl); uartp_send(buf, rl+1); }