fork of what I have been writing
ES_CW2_Starter_STARFISH/main.cpp@6:5f4a954cb8bc, 2020-03-01 (annotated)
- Committer:
- kubitz
- Date:
- Sun Mar 01 12:45:22 2020 +0000
- Revision:
- 6:5f4a954cb8bc
- Parent:
- 5:de6430aee646
- Child:
- 7:aef5b29d7a7c
added thread for printing crypto coins
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
kubitz | 5:de6430aee646 | 1 | |
kubitz | 0:19fd8c1944fb | 2 | #include "mbed.h" |
kubitz | 0:19fd8c1944fb | 3 | #include "SHA256.h" |
kubitz | 6:5f4a954cb8bc | 4 | #include "rtos.h" |
kubitz | 6:5f4a954cb8bc | 5 | typedef struct { |
kubitz | 6:5f4a954cb8bc | 6 | uint8_t hash[32]; /* hash of successful nonce */ |
kubitz | 6:5f4a954cb8bc | 7 | } mail_t; |
kubitz | 0:19fd8c1944fb | 8 | Timer timer_nonce; |
kubitz | 6:5f4a954cb8bc | 9 | Mail<mail_t, 16> mail_box; |
kubitz | 6:5f4a954cb8bc | 10 | Thread thread_crypto; |
kubitz | 0:19fd8c1944fb | 11 | |
kubitz | 3:8443825642d1 | 12 | |
kubitz | 0:19fd8c1944fb | 13 | uint8_t sequence[] = {0x45,0x6D,0x62,0x65,0x64,0x64,0x65,0x64, |
kubitz | 0:19fd8c1944fb | 14 | 0x20,0x53,0x79,0x73,0x74,0x65,0x6D,0x73, |
kubitz | 0:19fd8c1944fb | 15 | 0x20,0x61,0x72,0x65,0x20,0x66,0x75,0x6E, |
kubitz | 0:19fd8c1944fb | 16 | 0x20,0x61,0x6E,0x64,0x20,0x64,0x6F,0x20, |
kubitz | 0:19fd8c1944fb | 17 | 0x61,0x77,0x65,0x73,0x6F,0x6D,0x65,0x20, |
kubitz | 0:19fd8c1944fb | 18 | 0x74,0x68,0x69,0x6E,0x67,0x73,0x21,0x20, |
kubitz | 0:19fd8c1944fb | 19 | 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00, |
kubitz | 0:19fd8c1944fb | 20 | 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}; |
kubitz | 0:19fd8c1944fb | 21 | uint64_t* key = (uint64_t*)&sequence[48]; |
kubitz | 0:19fd8c1944fb | 22 | uint64_t* nonce = (uint64_t*)&sequence[56]; |
kubitz | 0:19fd8c1944fb | 23 | uint32_t successful_nonce = 0; |
kubitz | 0:19fd8c1944fb | 24 | uint32_t last_nonce_number = 0; |
kubitz | 5:de6430aee646 | 25 | |
kubitz | 0:19fd8c1944fb | 26 | uint8_t hash[32]; |
kubitz | 0:19fd8c1944fb | 27 | uint32_t previous_time; |
kubitz | 0:19fd8c1944fb | 28 | //Photointerrupter input pins |
kubitz | 0:19fd8c1944fb | 29 | #define I1pin D3 |
kubitz | 0:19fd8c1944fb | 30 | #define I2pin D6 |
kubitz | 0:19fd8c1944fb | 31 | #define I3pin D5 |
kubitz | 0:19fd8c1944fb | 32 | |
kubitz | 0:19fd8c1944fb | 33 | //Incremental encoder input pins |
kubitz | 0:19fd8c1944fb | 34 | #define CHApin D12 |
kubitz | 0:19fd8c1944fb | 35 | #define CHBpin D11 |
kubitz | 0:19fd8c1944fb | 36 | |
kubitz | 0:19fd8c1944fb | 37 | //Motor Drive output pins //Mask in output byte |
kubitz | 0:19fd8c1944fb | 38 | #define L1Lpin D1 //0x01 |
kubitz | 0:19fd8c1944fb | 39 | #define L1Hpin A3 //0x02 |
kubitz | 0:19fd8c1944fb | 40 | #define L2Lpin D0 //0x04 |
kubitz | 0:19fd8c1944fb | 41 | #define L2Hpin A6 //0x08 |
kubitz | 0:19fd8c1944fb | 42 | #define L3Lpin D10 //0x10 |
kubitz | 0:19fd8c1944fb | 43 | #define L3Hpin D2 //0x20 |
kubitz | 0:19fd8c1944fb | 44 | |
kubitz | 0:19fd8c1944fb | 45 | #define PWMpin D9 |
kubitz | 0:19fd8c1944fb | 46 | |
kubitz | 0:19fd8c1944fb | 47 | //Motor current sense |
kubitz | 0:19fd8c1944fb | 48 | #define MCSPpin A1 |
kubitz | 0:19fd8c1944fb | 49 | #define MCSNpin A0 |
kubitz | 0:19fd8c1944fb | 50 | |
kubitz | 0:19fd8c1944fb | 51 | //Test outputs |
kubitz | 0:19fd8c1944fb | 52 | #define TP0pin D4 |
kubitz | 0:19fd8c1944fb | 53 | #define TP1pin D13 |
kubitz | 0:19fd8c1944fb | 54 | #define TP2pin A2 |
kubitz | 0:19fd8c1944fb | 55 | |
kubitz | 0:19fd8c1944fb | 56 | //Mapping from sequential drive states to motor phase outputs |
kubitz | 0:19fd8c1944fb | 57 | /* |
kubitz | 0:19fd8c1944fb | 58 | State L1 L2 L3 |
kubitz | 0:19fd8c1944fb | 59 | 0 H - L |
kubitz | 0:19fd8c1944fb | 60 | 1 - H L |
kubitz | 0:19fd8c1944fb | 61 | 2 L H - |
kubitz | 0:19fd8c1944fb | 62 | 3 L - H |
kubitz | 0:19fd8c1944fb | 63 | 4 - L H |
kubitz | 0:19fd8c1944fb | 64 | 5 H L - |
kubitz | 0:19fd8c1944fb | 65 | 6 - - - |
kubitz | 0:19fd8c1944fb | 66 | 7 - - - |
kubitz | 0:19fd8c1944fb | 67 | */ |
kubitz | 0:19fd8c1944fb | 68 | //Drive state to output table |
kubitz | 0:19fd8c1944fb | 69 | const int8_t driveTable[] = {0x12,0x18,0x09,0x21,0x24,0x06,0x00,0x00}; |
kubitz | 0:19fd8c1944fb | 70 | |
kubitz | 0:19fd8c1944fb | 71 | //Mapping from interrupter inputs to sequential rotor states. 0x00 and 0x07 are not valid |
kubitz | 0:19fd8c1944fb | 72 | const int8_t stateMap[] = {0x07,0x05,0x03,0x04,0x01,0x00,0x02,0x07}; |
kubitz | 0:19fd8c1944fb | 73 | //const int8_t stateMap[] = {0x07,0x01,0x03,0x02,0x05,0x00,0x04,0x07}; //Alternative if phase order of input or drive is reversed |
kubitz | 0:19fd8c1944fb | 74 | |
kubitz | 0:19fd8c1944fb | 75 | //Phase lead to make motor spin |
kubitz | 0:19fd8c1944fb | 76 | const int8_t lead = 2; //2 for forwards, -2 for backwards |
kubitz | 0:19fd8c1944fb | 77 | |
kubitz | 0:19fd8c1944fb | 78 | //Status LED |
kubitz | 0:19fd8c1944fb | 79 | DigitalOut led1(LED1); |
kubitz | 0:19fd8c1944fb | 80 | |
kubitz | 0:19fd8c1944fb | 81 | //Photointerrupter inputs |
kubitz | 0:19fd8c1944fb | 82 | InterruptIn I1(I1pin); |
kubitz | 0:19fd8c1944fb | 83 | InterruptIn I2(I2pin); |
kubitz | 0:19fd8c1944fb | 84 | InterruptIn I3(I3pin); |
kubitz | 0:19fd8c1944fb | 85 | |
kubitz | 0:19fd8c1944fb | 86 | //Motor Drive outputs |
kubitz | 0:19fd8c1944fb | 87 | DigitalOut L1L(L1Lpin); |
kubitz | 0:19fd8c1944fb | 88 | DigitalOut L1H(L1Hpin); |
kubitz | 0:19fd8c1944fb | 89 | DigitalOut L2L(L2Lpin); |
kubitz | 0:19fd8c1944fb | 90 | DigitalOut L2H(L2Hpin); |
kubitz | 0:19fd8c1944fb | 91 | DigitalOut L3L(L3Lpin); |
kubitz | 0:19fd8c1944fb | 92 | DigitalOut L3H(L3Hpin); |
kubitz | 0:19fd8c1944fb | 93 | |
kubitz | 0:19fd8c1944fb | 94 | DigitalOut TP1(TP1pin); |
kubitz | 0:19fd8c1944fb | 95 | PwmOut MotorPWM(PWMpin); |
kubitz | 0:19fd8c1944fb | 96 | |
kubitz | 0:19fd8c1944fb | 97 | int8_t orState = 0; //Rotot offset at motor state 0 |
kubitz | 0:19fd8c1944fb | 98 | int8_t intState = 0; |
kubitz | 0:19fd8c1944fb | 99 | int8_t intStateOld = 0; |
kubitz | 0:19fd8c1944fb | 100 | |
kubitz | 0:19fd8c1944fb | 101 | |
kubitz | 0:19fd8c1944fb | 102 | //Set a given drive state |
kubitz | 0:19fd8c1944fb | 103 | void motorOut(int8_t driveState){ |
kubitz | 0:19fd8c1944fb | 104 | |
kubitz | 0:19fd8c1944fb | 105 | //Lookup the output byte from the drive state. |
kubitz | 0:19fd8c1944fb | 106 | int8_t driveOut = driveTable[driveState & 0x07]; |
kubitz | 0:19fd8c1944fb | 107 | |
kubitz | 0:19fd8c1944fb | 108 | //Turn off first |
kubitz | 0:19fd8c1944fb | 109 | if (~driveOut & 0x01) L1L = 0; |
kubitz | 0:19fd8c1944fb | 110 | if (~driveOut & 0x02) L1H = 1; |
kubitz | 0:19fd8c1944fb | 111 | if (~driveOut & 0x04) L2L = 0; |
kubitz | 0:19fd8c1944fb | 112 | if (~driveOut & 0x08) L2H = 1; |
kubitz | 0:19fd8c1944fb | 113 | if (~driveOut & 0x10) L3L = 0; |
kubitz | 0:19fd8c1944fb | 114 | if (~driveOut & 0x20) L3H = 1; |
kubitz | 0:19fd8c1944fb | 115 | |
kubitz | 0:19fd8c1944fb | 116 | //Then turn on |
kubitz | 0:19fd8c1944fb | 117 | if (driveOut & 0x01) L1L = 1; |
kubitz | 0:19fd8c1944fb | 118 | if (driveOut & 0x02) L1H = 0; |
kubitz | 0:19fd8c1944fb | 119 | if (driveOut & 0x04) L2L = 1; |
kubitz | 0:19fd8c1944fb | 120 | if (driveOut & 0x08) L2H = 0; |
kubitz | 0:19fd8c1944fb | 121 | if (driveOut & 0x10) L3L = 1; |
kubitz | 0:19fd8c1944fb | 122 | if (driveOut & 0x20) L3H = 0; |
kubitz | 0:19fd8c1944fb | 123 | } |
kubitz | 0:19fd8c1944fb | 124 | |
kubitz | 0:19fd8c1944fb | 125 | //Convert photointerrupter inputs to a rotor state |
kubitz | 0:19fd8c1944fb | 126 | inline int8_t readRotorState(){ |
kubitz | 0:19fd8c1944fb | 127 | return stateMap[I1 + 2*I2 + 4*I3]; |
kubitz | 0:19fd8c1944fb | 128 | } |
kubitz | 0:19fd8c1944fb | 129 | |
kubitz | 0:19fd8c1944fb | 130 | //Basic synchronisation routine |
kubitz | 0:19fd8c1944fb | 131 | int8_t motorHome() { |
kubitz | 0:19fd8c1944fb | 132 | //Put the motor in drive state 0 and wait for it to stabilise |
kubitz | 0:19fd8c1944fb | 133 | motorOut(0); |
kubitz | 0:19fd8c1944fb | 134 | wait(2.0); |
kubitz | 0:19fd8c1944fb | 135 | |
kubitz | 0:19fd8c1944fb | 136 | //Get the rotor state |
kubitz | 0:19fd8c1944fb | 137 | return readRotorState(); |
kubitz | 0:19fd8c1944fb | 138 | } |
kubitz | 0:19fd8c1944fb | 139 | |
kubitz | 0:19fd8c1944fb | 140 | void move() { |
kubitz | 0:19fd8c1944fb | 141 | intState = readRotorState(); |
kubitz | 0:19fd8c1944fb | 142 | motorOut((intState-orState+lead+6)%6); //+6 to make sure the remainder is positive |
kubitz | 0:19fd8c1944fb | 143 | intStateOld = intState; |
kubitz | 0:19fd8c1944fb | 144 | } |
kubitz | 0:19fd8c1944fb | 145 | |
kubitz | 6:5f4a954cb8bc | 146 | // Thread to print successful Hashes |
kubitz | 6:5f4a954cb8bc | 147 | void print_crypto_thread() { |
kubitz | 6:5f4a954cb8bc | 148 | while (true) { |
kubitz | 6:5f4a954cb8bc | 149 | osEvent evt = mail_box.get(); |
kubitz | 6:5f4a954cb8bc | 150 | if (evt.status == osEventMail) { |
kubitz | 6:5f4a954cb8bc | 151 | mail_t *mail = (mail_t*)evt.value.p; |
kubitz | 6:5f4a954cb8bc | 152 | for (int i = 0; i < 32; i++) |
kubitz | 6:5f4a954cb8bc | 153 | printf("%d", mail->hash[i]); |
kubitz | 6:5f4a954cb8bc | 154 | printf("/n"); |
kubitz | 6:5f4a954cb8bc | 155 | mail_box.free(mail); |
kubitz | 6:5f4a954cb8bc | 156 | } |
kubitz | 6:5f4a954cb8bc | 157 | } |
kubitz | 6:5f4a954cb8bc | 158 | } |
kubitz | 6:5f4a954cb8bc | 159 | |
kubitz | 6:5f4a954cb8bc | 160 | // Put message in Mail box |
kubitz | 6:5f4a954cb8bc | 161 | void putMessage(uint8_t *hash[32]){ |
kubitz | 6:5f4a954cb8bc | 162 | mail_t *mail = mail_box.alloc(); |
kubitz | 6:5f4a954cb8bc | 163 | strcpy(*hash, mail->hash); |
kubitz | 6:5f4a954cb8bc | 164 | mail_box.put(mail); |
kubitz | 6:5f4a954cb8bc | 165 | } |
kubitz | 6:5f4a954cb8bc | 166 | |
kubitz | 0:19fd8c1944fb | 167 | //Main |
kubitz | 0:19fd8c1944fb | 168 | int main() { |
kubitz | 0:19fd8c1944fb | 169 | |
kubitz | 0:19fd8c1944fb | 170 | const int32_t PWM_PRD = 2500; |
kubitz | 0:19fd8c1944fb | 171 | MotorPWM.period_us(PWM_PRD); |
kubitz | 0:19fd8c1944fb | 172 | MotorPWM.pulsewidth_us(PWM_PRD); |
kubitz | 0:19fd8c1944fb | 173 | |
kubitz | 0:19fd8c1944fb | 174 | //Initialise the serial port |
kubitz | 0:19fd8c1944fb | 175 | Serial pc(SERIAL_TX, SERIAL_RX); |
kubitz | 0:19fd8c1944fb | 176 | pc.printf("Hello\n\r"); |
kubitz | 0:19fd8c1944fb | 177 | |
kubitz | 0:19fd8c1944fb | 178 | //Run the motor synchronisation |
kubitz | 0:19fd8c1944fb | 179 | orState = motorHome(); |
kubitz | 0:19fd8c1944fb | 180 | pc.printf("Rotor origin: %x\n\r",orState); |
kubitz | 0:19fd8c1944fb | 181 | |
kubitz | 0:19fd8c1944fb | 182 | I1.rise(&move); |
kubitz | 0:19fd8c1944fb | 183 | I1.fall(&move); |
kubitz | 0:19fd8c1944fb | 184 | I2.rise(&move); |
kubitz | 0:19fd8c1944fb | 185 | I2.fall(&move); |
kubitz | 0:19fd8c1944fb | 186 | I3.rise(&move); |
kubitz | 0:19fd8c1944fb | 187 | I3.fall(&move); |
kubitz | 0:19fd8c1944fb | 188 | timer_nonce.start(); |
kubitz | 5:de6430aee646 | 189 | pc.printf("time is %d\n\r", timer_nonce.read_ms()); |
kubitz | 5:de6430aee646 | 190 | pc.printf("time is %d\n\r", (timer_nonce.read_ms()-previous_time)); |
kubitz | 5:de6430aee646 | 191 | uint8_t hash[32]; |
kubitz | 6:5f4a954cb8bc | 192 | thread_crypto.start(print_crypto_thread); |
kubitz | 5:de6430aee646 | 193 | |
kubitz | 5:de6430aee646 | 194 | while (1){ |
kubitz | 5:de6430aee646 | 195 | |
kubitz | 5:de6430aee646 | 196 | *nonce = *nonce + 1; |
kubitz | 5:de6430aee646 | 197 | |
kubitz | 0:19fd8c1944fb | 198 | SHA256::computeHash(hash, (uint8_t*)sequence, 64); |
kubitz | 5:de6430aee646 | 199 | |
kubitz | 0:19fd8c1944fb | 200 | if ((hash[0]==0)&&(hash[1]==0)){ |
kubitz | 0:19fd8c1944fb | 201 | last_nonce_number = successful_nonce; |
kubitz | 0:19fd8c1944fb | 202 | successful_nonce++; |
kubitz | 6:5f4a954cb8bc | 203 | putMessage(hash); |
kubitz | 0:19fd8c1944fb | 204 | } |
kubitz | 5:de6430aee646 | 205 | |
kubitz | 5:de6430aee646 | 206 | if ((timer_nonce.read_ms()-previous_time) > 1000){ |
kubitz | 5:de6430aee646 | 207 | pc.printf("Computation Rate: %lu computation /sec\n\r" , (*nonce-last_nonce_number)); |
kubitz | 0:19fd8c1944fb | 208 | last_nonce_number = *nonce; |
kubitz | 5:de6430aee646 | 209 | previous_time = timer_nonce.read_ms(); |
kubitz | 5:de6430aee646 | 210 | |
kubitz | 0:19fd8c1944fb | 211 | } |
kubitz | 0:19fd8c1944fb | 212 | |
kubitz | 0:19fd8c1944fb | 213 | } |
kubitz | 0:19fd8c1944fb | 214 | |
kubitz | 0:19fd8c1944fb | 215 | return 0; |
kubitz | 0:19fd8c1944fb | 216 | |
kubitz | 0:19fd8c1944fb | 217 | } |
kubitz | 0:19fd8c1944fb | 218 |