333 lines
7.6 KiB
C
333 lines
7.6 KiB
C
#include <sysinit.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include "basic/basic.h"
|
|
#include "lcd/render.h"
|
|
#include "lcd/allfonts.h"
|
|
#include "basic/ecc.h"
|
|
#include "funk/nrf24l01p.h"
|
|
#include "filesystem/ff.h"
|
|
#include "filesystem/diskio.h"
|
|
#include "funk/filetransfer.h"
|
|
#include "lcd/print.h"
|
|
#include <string.h>
|
|
#include "funk/nrf24l01p.h"
|
|
#include "funk/filetransfer.h"
|
|
#include "funk/rftransfer.h"
|
|
#include "basic/basic.h"
|
|
#include "basic/xxtea.h"
|
|
#include "filesystem/ff.h"
|
|
#include "funk/rftransfer.h"
|
|
#include "funk/nrf24l01p.h"
|
|
#include <basic/basic.h>
|
|
#include <basic/random.h>
|
|
#include <core/systick/systick.h>
|
|
//#include <lcd/print.h>
|
|
|
|
#include "usetable.h"
|
|
|
|
//#include "lcd/print.h"
|
|
|
|
void sendFile(char *filename);
|
|
|
|
uint8_t mac[5] = {1,2,3,2,1};
|
|
struct NRF_CFG config = {
|
|
.channel= 81,
|
|
.txmac= "\x1\x2\x3\x2\x1",
|
|
.nrmacs=1,
|
|
.mac0= "\x1\x2\x3\x2\x1",
|
|
.maclen ="\x20",
|
|
};
|
|
|
|
void ram(void)
|
|
{
|
|
char file[13];
|
|
selectFile(file,"CRD");
|
|
nrf_config_set(&config);
|
|
sendFile(file);
|
|
}
|
|
|
|
void sendR(uint8_t *rx, uint8_t *ry)
|
|
{
|
|
//uint8_t exp[2 + 4*NUMWORDS + 2];
|
|
uint8_t exp[32];
|
|
exp[0] = 'R';
|
|
for(int i=0; i<4*NUMWORDS; i++)
|
|
exp[2+i] = rx[i];
|
|
exp[1] = 'X';
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc(32, exp);
|
|
delayms(10);
|
|
exp[1] = 'Y';
|
|
for(int i=0; i<4*NUMWORDS; i++)
|
|
exp[2+i] = ry[i];
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc(32, exp);
|
|
|
|
delayms(10);
|
|
}
|
|
|
|
int receiveKey(uint8_t type, uint8_t *x, uint8_t *y)
|
|
{
|
|
uint8_t buf[32];
|
|
uint8_t n;
|
|
//static int count = 0;
|
|
//for(n=0; n<32;n++)
|
|
// buf[n] = 0;
|
|
//lcdPrintln("get packet");
|
|
//lcdPrintInt(count++);
|
|
//lcdRefresh();
|
|
//int reg5 = nrf_read_reg(5);
|
|
//int reg11 = nrf_read_reg(0x11);
|
|
//lcdPrint("reg5=");
|
|
//lcdPrintInt(reg5);
|
|
//lcdPrintln("");
|
|
//lcdPrint("regx11=");
|
|
//lcdPrintInt(reg11);
|
|
//lcdPrintln("");
|
|
//lcdRefresh();
|
|
//delayms(1000);
|
|
|
|
nrf_config_set(&config);
|
|
n = nrf_rcv_pkt_time(1000, 32, buf);
|
|
//lcdPrintInt(n);
|
|
//lcdPrintln(IntToStrX(*(uint32_t*)buf,8));
|
|
//lcdPrintln(IntToStrX(*(uint32_t*)(buf+4),8));
|
|
//lcdPrintln(IntToStrX(*(uint32_t*)(buf+8),8));
|
|
//lcdPrintln(IntToStrX(*(uint32_t*)(buf+12),8));
|
|
|
|
//lcdRefresh();
|
|
//delayms(1000);
|
|
if( n == 32 && buf[0] == type && buf[1] == 'X' ){
|
|
//lcdPrint("got packet");
|
|
//lcdRefresh();
|
|
//while(1);
|
|
for(int i=0; i<NUMWORDS*4; i++)
|
|
x[i] = buf[i+2];
|
|
nrf_config_set(&config);
|
|
n = nrf_rcv_pkt_time(100, 32, buf);
|
|
if( n == 32 && buf[0] ==type && buf[1] == 'Y' ){
|
|
for(int i=0; i<NUMWORDS*4; i++)
|
|
y[i] = buf[i+2];
|
|
return 0;
|
|
}
|
|
}else{
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int receivePublicKey(uint8_t *px, uint8_t *py)
|
|
{
|
|
return receiveKey('P',px,py);
|
|
}
|
|
|
|
void sendMac(void)
|
|
{
|
|
uint8_t buf[32];
|
|
buf[0] = 'M';
|
|
buf[1] = 'C';
|
|
buf[2] = mac[0];
|
|
buf[3] = mac[1];
|
|
buf[4] = mac[2];
|
|
buf[5] = mac[3];
|
|
buf[6] = mac[4];
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc(32, buf);
|
|
delayms(10);
|
|
}
|
|
|
|
int receiveMac(uint8_t *mac)
|
|
{
|
|
uint8_t buf[32];
|
|
uint8_t n;
|
|
|
|
nrf_config_set(&config);
|
|
n = nrf_rcv_pkt_time(100, 32, buf);
|
|
if( n == 32 && buf[0] == 'M' && buf[1] == 'C' ){
|
|
for(int i=0; i<5; i++)
|
|
mac[i] = buf[i+2];
|
|
return 0;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int receiveKeys(uint8_t *px, uint8_t *py, uint8_t *mac)
|
|
{
|
|
uint8_t done = 0;
|
|
char key;
|
|
while( !done ){
|
|
lcdClear();
|
|
lcdPrintln("Recv. PUBKEY");
|
|
lcdPrintln("Down=Abort");
|
|
lcdRefresh();
|
|
key = getInput();
|
|
delayms(20);
|
|
if( key == BTN_DOWN ){
|
|
return -1;
|
|
}
|
|
if( receivePublicKey(px,py) )
|
|
continue;
|
|
if( receiveMac(mac) )
|
|
continue;
|
|
lcdClear();
|
|
lcdPrintln("Got PUBKEY"); lcdRefresh();
|
|
lcdPrintln("Hash:");
|
|
lcdPrintln("Right=OK");
|
|
lcdPrintln("Left=Retry");
|
|
lcdPrintln("Down=Abort");
|
|
lcdRefresh();
|
|
while(1){
|
|
key = getInput();
|
|
delayms(20);
|
|
if( key == BTN_LEFT ){
|
|
break;
|
|
}else if( key == BTN_RIGHT ){
|
|
done = 1;
|
|
break;
|
|
}else if( key == BTN_DOWN ){
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
void sendFile(char *filename)
|
|
{
|
|
uint8_t px[4*NUMWORDS];
|
|
uint8_t py[4*NUMWORDS];
|
|
uint8_t mac[5];
|
|
if( receiveKeys(px, py, mac) )
|
|
return;
|
|
|
|
uint8_t done = 0;
|
|
uint8_t key;
|
|
|
|
uint8_t k1[16], k2[16], rx[4*NUMWORDS], ry[4*NUMWORDS];
|
|
|
|
lcdClear();
|
|
lcdPrintln("Creating key"); lcdRefresh();
|
|
ECIES_encyptkeygen(px, py, k1, k2, rx, ry);
|
|
|
|
while( !done ){
|
|
lcdPrintln("Sending file");lcdRefresh();
|
|
sendR(rx,ry);
|
|
delayms(3000);
|
|
filetransfer_send((uint8_t*)filename, 0, mac, (uint32_t*)k1);
|
|
lcdPrintln("Done");
|
|
lcdPrintln("Right=OK");
|
|
lcdPrintln("Left=Retry");
|
|
lcdRefresh();
|
|
while(1){
|
|
key = getInput();
|
|
delayms(20);
|
|
if( key == BTN_LEFT ){
|
|
break;
|
|
}else if( key == BTN_RIGHT ){
|
|
done = 1;
|
|
break;
|
|
}
|
|
}
|
|
lcdClear();
|
|
}
|
|
}
|
|
//TODO: use a proper MAC to sign the message
|
|
int filetransfer_send(uint8_t *filename, uint16_t size,
|
|
uint8_t *mac, uint32_t const k[4])
|
|
{
|
|
uint8_t buf[MAXSIZE];
|
|
FIL file;
|
|
FRESULT res;
|
|
UINT readbytes;
|
|
|
|
|
|
if( size > MAXSIZE )
|
|
return 1; //File to big
|
|
|
|
res=f_open(&file, (const char*)filename, FA_OPEN_EXISTING|FA_READ);
|
|
if( res )
|
|
return res;
|
|
|
|
//res = f_read(&file, (char *)buf, size, &readbytes);
|
|
for(uint16_t i=0; i<MAXSIZE; i++)
|
|
buf[i] = 0;
|
|
|
|
res = f_read(&file, (char *)buf, MAXSIZE, &readbytes);
|
|
size = readbytes;
|
|
|
|
if( res )
|
|
return res;
|
|
if( size != readbytes)
|
|
return 1; //Error while reading
|
|
|
|
uint16_t wordcount = (size+3)/4;
|
|
//uint8_t macbuf[5];
|
|
|
|
uint8_t metadata[32];
|
|
if( strlen((char*)filename) < 20 )
|
|
strcpy((char*)metadata, (char*)filename);
|
|
else
|
|
return 1; //File name too long
|
|
metadata[20] = size >> 8;
|
|
metadata[21] = size & 0xFF;
|
|
|
|
//nrf_get_tx_max(5,macbuf);
|
|
|
|
//nrf_set_tx_mac(5, mac);
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc_encr(32, metadata, k);
|
|
delayms(20);
|
|
xxtea_encode_words((uint32_t *)buf, wordcount, k);
|
|
rftransfer_send(wordcount*4, buf);
|
|
//nrf_set_tx_mac(5, macbuf);
|
|
return 0;
|
|
}
|
|
|
|
#define MAXPACKET 32
|
|
void rftransfer_send(uint16_t size, uint8_t *data)
|
|
{
|
|
uint8_t buf[MAXPACKET];
|
|
buf[0] = 'L';
|
|
buf[1] = size >> 8;
|
|
buf[2] = size & 0xFF;
|
|
|
|
uint16_t rand = getRandom() & 0xFFFF;
|
|
buf[3] = rand >> 8;
|
|
buf[4] = rand & 0xFF;
|
|
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc(32,buf); //setup packet
|
|
delayms(20);
|
|
uint16_t index = 0;
|
|
uint8_t i;
|
|
uint16_t crc = crc16(data,size);
|
|
|
|
while(size){
|
|
buf[0] = 'D';
|
|
buf[1] = index >> 8;
|
|
buf[2] = index & 0xFF;
|
|
buf[3] = rand >> 8;
|
|
buf[4] = rand & 0xFF;
|
|
for(i=5; i<MAXPACKET-2 && size>0; i++,size--){
|
|
buf[i] = *data++;
|
|
}
|
|
index++;
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc(32,buf); //data packet
|
|
delayms(20);
|
|
}
|
|
|
|
buf[0] = 'C';
|
|
buf[1] = crc >> 8;
|
|
buf[2] = crc & 0xFF;
|
|
buf[3] = rand >> 8;
|
|
buf[4] = rand & 0xFF;
|
|
nrf_config_set(&config);
|
|
nrf_snd_pkt_crc(32,buf); //setup packet
|
|
delayms(20);
|
|
}
|
|
|