new
Dependencies: Crypto_light mbed-rtos mbed
Fork of ES_CW2_Starter by
Revision 4:ed5a463a64ad, committed 2018-03-15
- Comitter:
- JINYEENG
- Date:
- Thu Mar 15 12:38:21 2018 +0000
- Parent:
- 3:569b35e2a602
- Commit message:
- new
Changed in this revision
diff -r 569b35e2a602 -r ed5a463a64ad Crypto_light.lib --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Crypto_light.lib Thu Mar 15 12:38:21 2018 +0000 @@ -0,0 +1,1 @@ +http://os.mbed.com/users/estott/code/Crypto_light/#634f9c4cbab1
diff -r 569b35e2a602 -r ed5a463a64ad main.cpp --- a/main.cpp Thu Mar 01 09:41:46 2018 +0000 +++ b/main.cpp Thu Mar 15 12:38:21 2018 +0000 @@ -1,15 +1,21 @@ #include "mbed.h" +#include "SHA256.h" +#include "rtos.h" +#include <string> -//Photointerrupter input pins + +//------------------------ Define Inputs -------------------------------------// + +// Photointerrupter input pins #define I1pin D2 #define I2pin D11 #define I3pin D12 -//Incremental encoder input pins +// Incremental encoder input pins #define CHA D7 -#define CHB D8 +#define CHB D8 -//Motor Drive output pins //Mask in output byte +// Motor Drive output pins //Mask in output byte #define L1Lpin D4 //0x01 #define L1Hpin D5 //0x02 #define L2Lpin D3 //0x04 @@ -17,7 +23,7 @@ #define L3Lpin D9 //0x10 #define L3Hpin D10 //0x20 -//Mapping from sequential drive states to motor phase outputs +// Mapping from sequential drive states to motor phase outputs /* State L1 L2 L3 0 H - L @@ -29,23 +35,32 @@ 6 - - - 7 - - - */ -//Drive state to output table -const int8_t driveTable[] = {0x12,0x18,0x09,0x21,0x24,0x06,0x00,0x00}; + + +//---------------------------- Global Variables ------------------------------// + +// For Motor +// Drive state to output table +const int8_t driveTable[] = { 0x12, 0x18, 0x09, 0x21, 0x24, 0x06, 0x00, 0x00 }; -//Mapping from interrupter inputs to sequential rotor states. 0x00 and 0x07 are not valid -const int8_t stateMap[] = {0x07,0x05,0x03,0x04,0x01,0x00,0x02,0x07}; -//const int8_t stateMap[] = {0x07,0x01,0x03,0x02,0x05,0x00,0x04,0x07}; //Alternative if phase order of input or drive is reversed +// Mapping from interrupter inputs to sequential rotor states. +// 0x00 and 0x07 are not valid +const int8_t stateMap[] = {0x07,0x05,0x03,0x04,0x01,0x00,0x02,0x07}; -//Phase lead to make motor spin -const int8_t lead = 2; //2 for forwards, -2 for backwards +// Alternative if phase order of input or drive is reversed +//const int8_t stateMap[] = {0x07,0x01,0x03,0x02,0x05,0x00,0x04,0x07}; + +// Phase lead to make motor spin +const int8_t lead = -2; //2 for forwards, -2 for backwards +//enum messageName ={"Hash_Rate","Nonce","Key"}; //Status LED DigitalOut led1(LED1); //Photointerrupter inputs -DigitalIn I1(I1pin); -DigitalIn I2(I2pin); -DigitalIn I3(I3pin); +InterruptIn I1(I1pin); +InterruptIn I2(I2pin); +InterruptIn I3(I3pin); //Motor Drive outputs DigitalOut L1L(L1Lpin); @@ -55,12 +70,62 @@ DigitalOut L3L(L3Lpin); DigitalOut L3H(L3Hpin); -//Set a given drive state -void motorOut(int8_t driveState){ - +int8_t orState = 0; //Rotot offset at motor state 0 + +//Initialise the serial port +RawSerial pc(SERIAL_TX, SERIAL_RX); +int8_t intState = 0; +int8_t intStateOld = 0; + +// Variables for Hash +uint8_t sequence[] = {0x45, 0x6D, 0x62, 0x65, 0x64, 0x64, 0x65, 0x64, + 0x20, 0x53, 0x79, 0x73, 0x74, 0x65, 0x6D, 0x73, + 0x20, 0x61, 0x72, 0x65, 0x20, 0x66, 0x75, 0x6E, + 0x20, 0x61, 0x6E, 0x64, 0x20, 0x64, 0x6F, 0x20, + 0x61, 0x77, 0x65, 0x73, 0x6F, 0x6D, 0x65, 0x20, + 0x74, 0x68, 0x69, 0x6E, 0x67, 0x73, 0x21, 0x20, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}; + +uint64_t* key = (uint64_t*)((int)sequence + 48); +uint64_t* nonce = (uint64_t*)((int)sequence + 56); +uint8_t hash[32]; + +// For Thread + +// Timer +Timer t; + +// Threading +Thread commOutT; +Thread commOutT2; + +typedef struct{ + uint8_t code; + uint32_t data; + } message_t; + +// Mail +Mail<message_t, 16> outMessages; + +// Queue +char command[19]; + +Queue <void,8> inCharQ; + +uint32_t newKey;//to pass new key +uint32_t keyLala; +uint64_t combinedKey; + +Mutex key_mutex; + +//------------------------- Motor Functions ----------------------------------// + +// Set a given sdrive state +void motorOut(int8_t driveState) { //Lookup the output byte from the drive state. int8_t driveOut = driveTable[driveState & 0x07]; - + //Turn off first if (~driveOut & 0x01) L1L = 0; if (~driveOut & 0x02) L1H = 1; @@ -68,7 +133,7 @@ if (~driveOut & 0x08) L2H = 1; if (~driveOut & 0x10) L3L = 0; if (~driveOut & 0x20) L3H = 1; - + //Then turn on if (driveOut & 0x01) L1L = 1; if (driveOut & 0x02) L1H = 0; @@ -76,45 +141,137 @@ if (driveOut & 0x08) L2H = 0; if (driveOut & 0x10) L3L = 1; if (driveOut & 0x20) L3H = 0; - } - - //Convert photointerrupter inputs to a rotor state -inline int8_t readRotorState(){ - return stateMap[I1 + 2*I2 + 4*I3]; - } +} -//Basic synchronisation routine +// Convert photointerrupter inputs to a rotor state +inline int8_t readRotorState() { + return stateMap[I1 + 2 * I2 + 4 * I3]; +} + +// Basic synchronisation routine int8_t motorHome() { //Put the motor in drive state 0 and wait for it to stabilise motorOut(0); - wait(2.0); - + wait(1.0); + //Get the rotor state return readRotorState(); } - -//Main -int main() { - int8_t orState = 0; //Rotot offset at motor state 0 - int8_t intState = 0; - int8_t intStateOld = 0; - - //Initialise the serial port - Serial pc(SERIAL_TX, SERIAL_RX); - pc.printf("Hello\n\r"); - - //Run the motor synchronisation - orState = motorHome(); - pc.printf("Rotor origin: %x\n\r",orState); - //orState is subtracted from future rotor state inputs to align rotor and motor states - - //Poll the rotor state and set the motor outputs accordingly to spin the motor - while (1) { - intState = readRotorState(); - if (intState != intStateOld) { - intStateOld = intState; - motorOut((intState-orState+lead+6)%6); //+6 to make sure the remainder is positive - } + +void read() { + intState = readRotorState(); + if (intState != intStateOld) { + intStateOld = intState; + motorOut((intState - orState + lead + 6) % 6); + //+6 to make sure the remainder is positive + } +} + + +// -------------------------- Thread Functions -------------------------------// + +void putMessage(uint8_t code, uint64_t data) { + message_t *pMessage = outMessages.alloc(); + pMessage->code = code; + pMessage->data = data; + outMessages.put(pMessage); +} + +void commOutFn() { + while(1){ + osEvent newEvent = outMessages.get(); + message_t *pMessage = (message_t*)newEvent.value.p; + //pc.printf("%s = 0x%016x \r \n",messageName[pMessage->code], pMessage->data); + if(pMessage->code == 1) { + pc.printf("Nonce = 0x%016x \r \n", pMessage->data); + } else if(pMessage->code == 0){ + pc.printf("Hash Rate = %d \r \n", pMessage->data); + } else if(pMessage->code == 2){ + pc.printf("Key = 0x%016x \r \n", pMessage->data); + } + + + outMessages.free(pMessage); } } +void serialISR() { + uint8_t newChar = pc.getc(); + inCharQ.put((void*)newChar); +} + +void keyBuffer() { + pc.attach(&serialISR); + int i = 0; + while(1){ + osEvent newEvent =inCharQ.get(); + uint8_t newchar = (uint8_t)newEvent.value.p; + command[i] = newchar; + if (newchar =='r' && command[i-1] == '\\' ) { // detect end of line + command[i]='0'; + if (command[0] == 'K') { + key_mutex.lock(); + sscanf(command, "%*c %8x %8x", key, (uint64_t*)((int)key+4) ); + pc.printf("Key: %8x%8x \r\n", *key, *((uint64_t*)(int)key+4)); + key_mutex.unlock(); + } + i = -1; + } + i++; + if(i>=sizeof(command)-1){ + i=0; + putMessage(2,0); + } + } +} + + +//-------------------------------- Main --------------------------------------// + +int main() { + newKey=*key; + + commOutT.start(commOutFn); + commOutT2.start(keyBuffer); + + // Run the motor synchronisation + orState = motorHome(); + // orState is subtracted from future rotor state inputs to align rotor and motor states + + // Interrupt, if the motor is moved it will enter the interrupt and keep rotating + I1.rise(&read); + I2.rise(&read); + I3.rise(&read); + I1.fall(&read); + I2.fall(&read); + I3.fall(&read); + + // Poll the rotor state and set the motor outputs accordingly to spin the motor + + // Create an old ref to nonce + uint64_t oldNonce = *nonce; + uint32_t hashRate; + + while (1) { + + // Start the timer + t.start(); + + // Compute hash key after nonce is added by 1 every time + SHA256::computeHash(hash, (uint8_t*)((int)sequence), sizeof(sequence)); + + if(t.read() > 1) { + t.stop(); + hashRate = ((int)*nonce - (int)oldNonce) / t.read(); + oldNonce = *nonce; + putMessage(0, hashRate); + t.reset(); + } + + if(hash[0] == 0 && hash[1] == 0) { + putMessage(1, (int)*nonce); + } + *nonce= (int)*nonce+1; + } +} +
diff -r 569b35e2a602 -r ed5a463a64ad mbed-rtos.lib --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mbed-rtos.lib Thu Mar 15 12:38:21 2018 +0000 @@ -0,0 +1,1 @@ +http://os.mbed.com/users/mbed_official/code/mbed-rtos/#5713cbbdb706