a few notation fixes
[proxmark3-svn] / client / cmddata.c
1 //-----------------------------------------------------------------------------
2 // Copyright (C) 2010 iZsh <izsh at fail0verflow.com>
3 //
4 // This code is licensed to you under the terms of the GNU GPL, version 2 or,
5 // at your option, any later version. See the LICENSE.txt file for the text of
6 // the license.
7 //-----------------------------------------------------------------------------
8 // Data and Graph commands
9 //-----------------------------------------------------------------------------
10
11 #include <stdio.h> // also included in util.h
12 #include <string.h> // also included in util.h
13 #include <inttypes.h>
14 #include <limits.h> // for CmdNorm INT_MIN && INT_MAX
15 #include "data.h" // also included in util.h
16 #include "cmddata.h"
17 #include "util.h"
18 #include "cmdmain.h"
19 #include "proxmark3.h"
20 #include "ui.h" // for show graph controls
21 #include "graph.h" // for graph data
22 #include "cmdparser.h"// already included in cmdmain.h
23 #include "usb_cmd.h" // already included in cmdmain.h and proxmark3.h
24 #include "lfdemod.h" // for demod code
25 #include "loclass/cipherutils.h" // for decimating samples in getsamples
26 #include "cmdlfem4x.h"// for em410x demod
27
28 uint8_t DemodBuffer[MAX_DEMOD_BUF_LEN];
29 uint8_t g_debugMode=0;
30 size_t DemodBufferLen=0;
31 int g_DemodStartIdx=0;
32 int g_DemodClock=0;
33
34 static int CmdHelp(const char *Cmd);
35
36 //set the demod buffer with given array of binary (one bit per byte)
37 //by marshmellow
38 void setDemodBuf(uint8_t *buff, size_t size, size_t startIdx)
39 {
40 if (buff == NULL)
41 return;
42
43 if ( size > MAX_DEMOD_BUF_LEN - startIdx)
44 size = MAX_DEMOD_BUF_LEN - startIdx;
45
46 size_t i = 0;
47 for (; i < size; i++){
48 DemodBuffer[i]=buff[startIdx++];
49 }
50 DemodBufferLen=size;
51 return;
52 }
53
54 bool getDemodBuf(uint8_t *buff, size_t *size) {
55 if (buff == NULL) return false;
56 if (size == NULL) return false;
57 if (*size == 0) return false;
58
59 *size = (*size > DemodBufferLen) ? DemodBufferLen : *size;
60
61 memcpy(buff, DemodBuffer, *size);
62 return true;
63 }
64
65 // option '1' to save DemodBuffer any other to restore
66 void save_restoreDB(uint8_t saveOpt)
67 {
68 static uint8_t SavedDB[MAX_DEMOD_BUF_LEN];
69 static size_t SavedDBlen;
70 static bool DB_Saved = false;
71
72 if (saveOpt==1) { //save
73
74 memcpy(SavedDB, DemodBuffer, sizeof(DemodBuffer));
75 SavedDBlen = DemodBufferLen;
76 DB_Saved=true;
77 } else if (DB_Saved) { //restore
78 memcpy(DemodBuffer, SavedDB, sizeof(DemodBuffer));
79 DemodBufferLen = SavedDBlen;
80 }
81 return;
82 }
83
84 int CmdSetDebugMode(const char *Cmd)
85 {
86 int demod=0;
87 sscanf(Cmd, "%i", &demod);
88 g_debugMode=(uint8_t)demod;
89 return 1;
90 }
91
92 int usage_data_printdemodbuf(){
93 PrintAndLog("Usage: data printdemodbuffer x o <offset> l <length>");
94 PrintAndLog("Options: ");
95 PrintAndLog(" h This help");
96 PrintAndLog(" x output in hex (omit for binary output)");
97 PrintAndLog(" o <offset> enter offset in # of bits");
98 PrintAndLog(" l <length> enter length to print in # of bits or hex characters respectively");
99 return 0;
100 }
101
102 //by marshmellow
103 void printDemodBuff(void)
104 {
105 int bitLen = DemodBufferLen;
106 if (bitLen<1) {
107 PrintAndLog("no bits found in demod buffer");
108 return;
109 }
110 if (bitLen>512) bitLen=512; //max output to 512 bits if we have more - should be plenty
111
112 char *bin = sprint_bin_break(DemodBuffer,bitLen,16);
113 PrintAndLog("%s",bin);
114
115 return;
116 }
117
118 int CmdPrintDemodBuff(const char *Cmd)
119 {
120 char hex[512]={0x00};
121 bool hexMode = false;
122 bool errors = false;
123 uint32_t offset = 0; //could be size_t but no param_get16...
124 uint32_t length = 512;
125 char cmdp = 0;
126 while(param_getchar(Cmd, cmdp) != 0x00)
127 {
128 switch(param_getchar(Cmd, cmdp))
129 {
130 case 'h':
131 case 'H':
132 return usage_data_printdemodbuf();
133 case 'x':
134 case 'X':
135 hexMode = true;
136 cmdp++;
137 break;
138 case 'o':
139 case 'O':
140 offset = param_get32ex(Cmd, cmdp+1, 0, 10);
141 if (!offset) errors = true;
142 cmdp += 2;
143 break;
144 case 'l':
145 case 'L':
146 length = param_get32ex(Cmd, cmdp+1, 512, 10);
147 if (!length) errors = true;
148 cmdp += 2;
149 break;
150 default:
151 PrintAndLog("Unknown parameter '%c'", param_getchar(Cmd, cmdp));
152 errors = true;
153 break;
154 }
155 if(errors) break;
156 }
157 //Validations
158 if(errors) return usage_data_printdemodbuf();
159 length = (length > (DemodBufferLen-offset)) ? DemodBufferLen-offset : length;
160 int numBits = (length) & 0x00FFC; //make sure we don't exceed our string
161
162 if (hexMode){
163 char *buf = (char *) (DemodBuffer + offset);
164 numBits = (numBits > sizeof(hex)) ? sizeof(hex) : numBits;
165 numBits = binarraytohex(hex, buf, numBits);
166 if (numBits==0) return 0;
167 PrintAndLog("DemodBuffer: %s",hex);
168 } else {
169 PrintAndLog("DemodBuffer:\n%s", sprint_bin_break(DemodBuffer+offset,numBits,16));
170 }
171 return 1;
172 }
173
174 //by marshmellow
175 //this function strictly converts >1 to 1 and <1 to 0 for each sample in the graphbuffer
176 int CmdGetBitStream(const char *Cmd)
177 {
178 int i;
179 CmdHpf(Cmd);
180 for (i = 0; i < GraphTraceLen; i++) {
181 if (GraphBuffer[i] >= 1) {
182 GraphBuffer[i] = 1;
183 } else {
184 GraphBuffer[i] = 0;
185 }
186 }
187 RepaintGraphWindow();
188 return 0;
189 }
190
191 //by marshmellow
192 //Cmd Args: Clock, invert, maxErr, maxLen as integers and amplify as char == 'a'
193 // (amp may not be needed anymore)
194 //verbose will print results and demoding messages
195 //emSearch will auto search for EM410x format in bitstream
196 //askType switches decode: ask/raw = 0, ask/manchester = 1
197 int ASKDemod_ext(const char *Cmd, bool verbose, bool emSearch, uint8_t askType, bool *stCheck) {
198 int invert=0;
199 int clk=0;
200 int maxErr=100;
201 int maxLen=0;
202 uint8_t askamp = 0;
203 char amp = param_getchar(Cmd, 0);
204 uint8_t BitStream[MAX_GRAPH_TRACE_LEN]={0};
205 sscanf(Cmd, "%i %i %i %i %c", &clk, &invert, &maxErr, &maxLen, &amp);
206 if (!maxLen) maxLen = BIGBUF_SIZE;
207 if (invert != 0 && invert != 1) {
208 PrintAndLog("Invalid argument: %s", Cmd);
209 return 0;
210 }
211 if (clk==1){
212 invert=1;
213 clk=0;
214 }
215 size_t BitLen = getFromGraphBuf(BitStream);
216 if (g_debugMode) PrintAndLog("DEBUG: Bitlen from grphbuff: %d",BitLen);
217 if (BitLen < 255) return 0;
218 if (maxLen < BitLen && maxLen != 0) BitLen = maxLen;
219 int foundclk = 0;
220 //amp before ST check
221 if (amp == 'a' || amp == 'A') {
222 askAmp(BitStream, BitLen);
223 }
224 bool st = false;
225 size_t ststart = 0, stend = 0;
226 if (*stCheck) st = DetectST(BitStream, &BitLen, &foundclk, &ststart, &stend);
227 *stCheck = st;
228 if (st) {
229 clk = (clk == 0) ? foundclk : clk;
230 CursorCPos = ststart;
231 CursorDPos = stend;
232 if (verbose || g_debugMode) PrintAndLog("\nFound Sequence Terminator - First one is shown by orange and blue graph markers");
233 //Graph ST trim (for testing)
234 //for (int i = 0; i < BitLen; i++) {
235 // GraphBuffer[i] = BitStream[i]-128;
236 //}
237 //RepaintGraphWindow();
238 }
239 int startIdx = 0;
240 int errCnt = askdemod_ext(BitStream, &BitLen, &clk, &invert, maxErr, askamp, askType, &startIdx);
241 if (errCnt<0 || BitLen<16){ //if fatal error (or -1)
242 if (g_debugMode) PrintAndLog("DEBUG: no data found %d, errors:%d, bitlen:%d, clock:%d",errCnt,invert,BitLen,clk);
243 return 0;
244 }
245 if (errCnt > maxErr){
246 if (g_debugMode) PrintAndLog("DEBUG: Too many errors found, errors:%d, bits:%d, clock:%d",errCnt, BitLen, clk);
247 return 0;
248 }
249 if (verbose || g_debugMode) PrintAndLog("\nUsing Clock:%d, Invert:%d, Bits Found:%d",clk,invert,BitLen);
250 //output
251 setDemodBuf(BitStream,BitLen,0);
252 setClockGrid(clk, startIdx);
253
254 if (verbose || g_debugMode){
255 if (errCnt>0) PrintAndLog("# Errors during Demoding (shown as 7 in bit stream): %d",errCnt);
256 if (askType) PrintAndLog("ASK/Manchester - Clock: %d - Decoded bitstream:",clk);
257 else PrintAndLog("ASK/Raw - Clock: %d - Decoded bitstream:",clk);
258 // Now output the bitstream to the scrollback by line of 16 bits
259 printDemodBuff();
260
261 }
262 uint64_t lo = 0;
263 uint32_t hi = 0;
264 if (emSearch){
265 AskEm410xDecode(true, &hi, &lo);
266 }
267 return 1;
268 }
269 int ASKDemod(const char *Cmd, bool verbose, bool emSearch, uint8_t askType) {
270 bool st = false;
271 return ASKDemod_ext(Cmd, verbose, emSearch, askType, &st);
272 }
273
274 //by marshmellow
275 //takes 5 arguments - clock, invert, maxErr, maxLen as integers and amplify as char == 'a'
276 //attempts to demodulate ask while decoding manchester
277 //prints binary found and saves in graphbuffer for further commands
278 int Cmdaskmandemod(const char *Cmd)
279 {
280 char cmdp = param_getchar(Cmd, 0);
281 if (strlen(Cmd) > 45 || cmdp == 'h' || cmdp == 'H') {
282 PrintAndLog("Usage: data rawdemod am <s> [clock] <invert> [maxError] [maxLen] [amplify]");
283 PrintAndLog(" ['s'] optional, check for Sequence Terminator");
284 PrintAndLog(" [set clock as integer] optional, if not set, autodetect");
285 PrintAndLog(" <invert>, 1 to invert output");
286 PrintAndLog(" [set maximum allowed errors], default = 100");
287 PrintAndLog(" [set maximum Samples to read], default = 32768 (512 bits at rf/64)");
288 PrintAndLog(" <amplify>, 'a' to attempt demod with ask amplification, default = no amp");
289 PrintAndLog("");
290 PrintAndLog(" sample: data rawdemod am = demod an ask/manchester tag from GraphBuffer");
291 PrintAndLog(" : data rawdemod am 32 = demod an ask/manchester tag from GraphBuffer using a clock of RF/32");
292 PrintAndLog(" : data rawdemod am 32 1 = demod an ask/manchester tag from GraphBuffer using a clock of RF/32 and inverting data");
293 PrintAndLog(" : data rawdemod am 1 = demod an ask/manchester tag from GraphBuffer while inverting data");
294 PrintAndLog(" : data rawdemod am 64 1 0 = demod an ask/manchester tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors");
295 return 0;
296 }
297 bool st = true;
298 if (Cmd[0]=='s')
299 return ASKDemod_ext(Cmd++, true, false, 1, &st);
300 else if (Cmd[1] == 's')
301 return ASKDemod_ext(Cmd+=2, true, false, 1, &st);
302 else
303 return ASKDemod(Cmd, true, false, 1);
304 }
305
306 //by marshmellow
307 //manchester decode
308 //stricktly take 10 and 01 and convert to 0 and 1
309 int Cmdmandecoderaw(const char *Cmd)
310 {
311 int i =0;
312 int errCnt=0;
313 size_t size=0;
314 int invert=0;
315 int maxErr = 20;
316 char cmdp = param_getchar(Cmd, 0);
317 if (strlen(Cmd) > 5 || cmdp == 'h' || cmdp == 'H') {
318 PrintAndLog("Usage: data manrawdecode [invert] [maxErr]");
319 PrintAndLog(" Takes 10 and 01 and converts to 0 and 1 respectively");
320 PrintAndLog(" --must have binary sequence in demodbuffer (run data askrawdemod first)");
321 PrintAndLog(" [invert] invert output");
322 PrintAndLog(" [maxErr] set number of errors allowed (default = 20)");
323 PrintAndLog("");
324 PrintAndLog(" sample: data manrawdecode = decode manchester bitstream from the demodbuffer");
325 return 0;
326 }
327 if (DemodBufferLen==0) return 0;
328 uint8_t BitStream[MAX_DEMOD_BUF_LEN]={0};
329 int high=0,low=0;
330 for (;i<DemodBufferLen;++i){
331 if (DemodBuffer[i]>high) high=DemodBuffer[i];
332 else if(DemodBuffer[i]<low) low=DemodBuffer[i];
333 BitStream[i]=DemodBuffer[i];
334 }
335 if (high>7 || low <0 ){
336 PrintAndLog("Error: please raw demod the wave first then manchester raw decode");
337 return 0;
338 }
339
340 sscanf(Cmd, "%i %i", &invert, &maxErr);
341 size=i;
342 uint8_t alignPos = 0;
343 errCnt=manrawdecode(BitStream, &size, invert, &alignPos);
344 if (errCnt>=maxErr){
345 PrintAndLog("Too many errors: %d",errCnt);
346 return 0;
347 }
348 PrintAndLog("Manchester Decoded - # errors:%d - data:",errCnt);
349 PrintAndLog("%s", sprint_bin_break(BitStream, size, 16));
350 if (errCnt==0){
351 uint64_t id = 0;
352 uint32_t hi = 0;
353 size_t idx=0;
354 if (Em410xDecode(BitStream, &size, &idx, &hi, &id)){
355 //need to adjust to set bitstream back to manchester encoded data
356 //setDemodBuf(BitStream, size, idx);
357
358 printEM410x(hi, id);
359 }
360 }
361 return 1;
362 }
363
364 //by marshmellow
365 //biphase decode
366 //take 01 or 10 = 0 and 11 or 00 = 1
367 //takes 2 arguments "offset" default = 0 if 1 it will shift the decode by one bit
368 // and "invert" default = 0 if 1 it will invert output
369 // the argument offset allows us to manually shift if the output is incorrect - [EDIT: now auto detects]
370 int CmdBiphaseDecodeRaw(const char *Cmd)
371 {
372 size_t size=0;
373 int offset=0, invert=0, maxErr=20, errCnt=0;
374 char cmdp = param_getchar(Cmd, 0);
375 if (strlen(Cmd) > 3 || cmdp == 'h' || cmdp == 'H') {
376 PrintAndLog("Usage: data biphaserawdecode [offset] [invert] [maxErr]");
377 PrintAndLog(" Converts 10 or 01 to 1 and 11 or 00 to 0");
378 PrintAndLog(" --must have binary sequence in demodbuffer (run data askrawdemod first)");
379 PrintAndLog(" --invert for Conditional Dephase Encoding (CDP) AKA Differential Manchester");
380 PrintAndLog("");
381 PrintAndLog(" [offset <0|1>], set to 0 not to adjust start position or to 1 to adjust decode start position");
382 PrintAndLog(" [invert <0|1>], set to 1 to invert output");
383 PrintAndLog(" [maxErr int], set max errors tolerated - default=20");
384 PrintAndLog("");
385 PrintAndLog(" sample: data biphaserawdecode = decode biphase bitstream from the demodbuffer");
386 PrintAndLog(" sample: data biphaserawdecode 1 1 = decode biphase bitstream from the demodbuffer, set offset, and invert output");
387 return 0;
388 }
389 sscanf(Cmd, "%i %i %i", &offset, &invert, &maxErr);
390 if (DemodBufferLen==0) {
391 PrintAndLog("DemodBuffer Empty - run 'data rawdemod ar' first");
392 return 0;
393 }
394 uint8_t BitStream[MAX_DEMOD_BUF_LEN]={0};
395 size = sizeof(BitStream);
396 if ( !getDemodBuf(BitStream, &size) ) return 0;
397 errCnt=BiphaseRawDecode(BitStream, &size, &offset, invert);
398 if (errCnt<0){
399 PrintAndLog("Error during decode:%d", errCnt);
400 return 0;
401 }
402 if (errCnt>maxErr){
403 PrintAndLog("Too many errors attempting to decode: %d",errCnt);
404 return 0;
405 }
406
407 if (errCnt>0){
408 PrintAndLog("# Errors found during Demod (shown as 7 in bit stream): %d",errCnt);
409 }
410
411 PrintAndLog("Biphase Decoded using offset: %d - # invert:%d - data:",offset,invert);
412 PrintAndLog("%s", sprint_bin_break(BitStream, size, 16));
413
414 if (offset) setDemodBuf(DemodBuffer,DemodBufferLen-offset, offset); //remove first bit from raw demod
415 setClockGrid(g_DemodClock, g_DemodStartIdx + g_DemodClock*offset/2);
416 return 1;
417 }
418
419 //by marshmellow
420 // - ASK Demod then Biphase decode GraphBuffer samples
421 int ASKbiphaseDemod(const char *Cmd, bool verbose)
422 {
423 //ask raw demod GraphBuffer first
424 int offset=0, clk=0, invert=0, maxErr=0;
425 sscanf(Cmd, "%i %i %i %i", &offset, &clk, &invert, &maxErr);
426
427 uint8_t BitStream[MAX_GRAPH_TRACE_LEN];
428 size_t size = getFromGraphBuf(BitStream);
429 int startIdx = 0;
430 //invert here inverts the ask raw demoded bits which has no effect on the demod, but we need the pointer
431 int errCnt = askdemod_ext(BitStream, &size, &clk, &invert, maxErr, 0, 0, &startIdx);
432 if ( errCnt < 0 || errCnt > maxErr ) {
433 if (g_debugMode) PrintAndLog("DEBUG: no data or error found %d, clock: %d", errCnt, clk);
434 return 0;
435 }
436
437 //attempt to Biphase decode BitStream
438 errCnt = BiphaseRawDecode(BitStream, &size, &offset, invert);
439 if (errCnt < 0){
440 if (g_debugMode || verbose) PrintAndLog("Error BiphaseRawDecode: %d", errCnt);
441 return 0;
442 }
443 if (errCnt > maxErr) {
444 if (g_debugMode || verbose) PrintAndLog("Error BiphaseRawDecode too many errors: %d", errCnt);
445 return 0;
446 }
447 //success set DemodBuffer and return
448 setDemodBuf(BitStream, size, 0);
449 setClockGrid(clk, startIdx + clk*offset/2);
450 if (g_debugMode || verbose){
451 PrintAndLog("Biphase Decoded using offset: %d - clock: %d - # errors:%d - data:",offset,clk,errCnt);
452 printDemodBuff();
453 }
454 return 1;
455 }
456 //by marshmellow - see ASKbiphaseDemod
457 int Cmdaskbiphdemod(const char *Cmd)
458 {
459 char cmdp = param_getchar(Cmd, 0);
460 if (strlen(Cmd) > 25 || cmdp == 'h' || cmdp == 'H') {
461 PrintAndLog("Usage: data rawdemod ab [offset] [clock] <invert> [maxError] [maxLen] <amplify>");
462 PrintAndLog(" [offset], offset to begin biphase, default=0");
463 PrintAndLog(" [set clock as integer] optional, if not set, autodetect");
464 PrintAndLog(" <invert>, 1 to invert output");
465 PrintAndLog(" [set maximum allowed errors], default = 100");
466 PrintAndLog(" [set maximum Samples to read], default = 32768 (512 bits at rf/64)");
467 PrintAndLog(" <amplify>, 'a' to attempt demod with ask amplification, default = no amp");
468 PrintAndLog(" NOTE: <invert> can be entered as second or third argument");
469 PrintAndLog(" NOTE: <amplify> can be entered as first, second or last argument");
470 PrintAndLog(" NOTE: any other arg must have previous args set to work");
471 PrintAndLog("");
472 PrintAndLog(" NOTE: --invert for Conditional Dephase Encoding (CDP) AKA Differential Manchester");
473 PrintAndLog("");
474 PrintAndLog(" sample: data rawdemod ab = demod an ask/biph tag from GraphBuffer");
475 PrintAndLog(" : data rawdemod ab 0 a = demod an ask/biph tag from GraphBuffer, amplified");
476 PrintAndLog(" : data rawdemod ab 1 32 = demod an ask/biph tag from GraphBuffer using an offset of 1 and a clock of RF/32");
477 PrintAndLog(" : data rawdemod ab 0 32 1 = demod an ask/biph tag from GraphBuffer using a clock of RF/32 and inverting data");
478 PrintAndLog(" : data rawdemod ab 0 1 = demod an ask/biph tag from GraphBuffer while inverting data");
479 PrintAndLog(" : data rawdemod ab 0 64 1 0 = demod an ask/biph tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors");
480 PrintAndLog(" : data rawdemod ab 0 64 1 0 0 a = demod an ask/biph tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors, and amp");
481 return 0;
482 }
483 return ASKbiphaseDemod(Cmd, true);
484 }
485
486 //by marshmellow - see ASKDemod
487 int Cmdaskrawdemod(const char *Cmd)
488 {
489 char cmdp = param_getchar(Cmd, 0);
490 if (strlen(Cmd) > 35 || cmdp == 'h' || cmdp == 'H') {
491 PrintAndLog("Usage: data rawdemod ar [clock] <invert> [maxError] [maxLen] [amplify]");
492 PrintAndLog(" [set clock as integer] optional, if not set, autodetect");
493 PrintAndLog(" <invert>, 1 to invert output");
494 PrintAndLog(" [set maximum allowed errors], default = 100");
495 PrintAndLog(" [set maximum Samples to read], default = 32768 (1024 bits at rf/64)");
496 PrintAndLog(" <amplify>, 'a' to attempt demod with ask amplification, default = no amp");
497 PrintAndLog("");
498 PrintAndLog(" sample: data rawdemod ar = demod an ask tag from GraphBuffer");
499 PrintAndLog(" : data rawdemod ar a = demod an ask tag from GraphBuffer, amplified");
500 PrintAndLog(" : data rawdemod ar 32 = demod an ask tag from GraphBuffer using a clock of RF/32");
501 PrintAndLog(" : data rawdemod ar 32 1 = demod an ask tag from GraphBuffer using a clock of RF/32 and inverting data");
502 PrintAndLog(" : data rawdemod ar 1 = demod an ask tag from GraphBuffer while inverting data");
503 PrintAndLog(" : data rawdemod ar 64 1 0 = demod an ask tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors");
504 PrintAndLog(" : data rawdemod ar 64 1 0 0 a = demod an ask tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors, and amp");
505 return 0;
506 }
507 return ASKDemod(Cmd, true, false, 0);
508 }
509
510 int AutoCorrelate(const int *in, int *out, size_t len, int window, bool SaveGrph, bool verbose)
511 {
512 static int CorrelBuffer[MAX_GRAPH_TRACE_LEN];
513 size_t Correlation = 0;
514 int maxSum = 0;
515 int lastMax = 0;
516 if (verbose) PrintAndLog("performing %d correlations", GraphTraceLen - window);
517 for (int i = 0; i < len - window; ++i) {
518 int sum = 0;
519 for (int j = 0; j < window; ++j) {
520 sum += (in[j]*in[i + j]) / 256;
521 }
522 CorrelBuffer[i] = sum;
523 if (sum >= maxSum-100 && sum <= maxSum+100) {
524 //another max
525 Correlation = i-lastMax;
526 lastMax = i;
527 if (sum > maxSum) maxSum = sum;
528 } else if (sum > maxSum) {
529 maxSum=sum;
530 lastMax = i;
531 }
532 }
533 if (Correlation==0) {
534 //try again with wider margin
535 for (int i = 0; i < len - window; i++) {
536 if (CorrelBuffer[i] >= maxSum-(maxSum*0.05) && CorrelBuffer[i] <= maxSum+(maxSum*0.05)) {
537 //another max
538 Correlation = i-lastMax;
539 lastMax = i;
540 }
541 }
542 }
543 if (verbose && Correlation > 0) PrintAndLog("Possible Correlation: %d samples",Correlation);
544
545 if (SaveGrph) {
546 //GraphTraceLen = GraphTraceLen - window;
547 memcpy(out, CorrelBuffer, len * sizeof(int));
548 RepaintGraphWindow();
549 }
550 return Correlation;
551 }
552
553 int usage_data_autocorr(void)
554 {
555 //print help
556 PrintAndLog("Usage: data autocorr [window] [g]");
557 PrintAndLog("Options: ");
558 PrintAndLog(" h This help");
559 PrintAndLog(" [window] window length for correlation - default = 4000");
560 PrintAndLog(" g save back to GraphBuffer (overwrite)");
561 return 0;
562 }
563
564 int CmdAutoCorr(const char *Cmd)
565 {
566 char cmdp = param_getchar(Cmd, 0);
567 if (cmdp == 'h' || cmdp == 'H')
568 return usage_data_autocorr();
569 int window = 4000; //set default
570 char grph=0;
571 bool updateGrph = false;
572 sscanf(Cmd, "%i %c", &window, &grph);
573
574 if (window >= GraphTraceLen) {
575 PrintAndLog("window must be smaller than trace (%d samples)",
576 GraphTraceLen);
577 return 0;
578 }
579 if (grph == 'g') updateGrph=true;
580 return AutoCorrelate(GraphBuffer, GraphBuffer, GraphTraceLen, window, updateGrph, true);
581 }
582
583 int CmdBitsamples(const char *Cmd)
584 {
585 int cnt = 0;
586 uint8_t got[12288];
587
588 GetFromBigBuf(got,sizeof(got),0);
589 WaitForResponse(CMD_ACK,NULL);
590
591 for (int j = 0; j < sizeof(got); j++) {
592 for (int k = 0; k < 8; k++) {
593 if(got[j] & (1 << (7 - k))) {
594 GraphBuffer[cnt++] = 1;
595 } else {
596 GraphBuffer[cnt++] = 0;
597 }
598 }
599 }
600 GraphTraceLen = cnt;
601 RepaintGraphWindow();
602 return 0;
603 }
604
605 int CmdBuffClear(const char *Cmd)
606 {
607 UsbCommand c = {CMD_BUFF_CLEAR};
608 SendCommand(&c);
609 ClearGraph(true);
610 return 0;
611 }
612
613 int CmdDec(const char *Cmd)
614 {
615 for (int i = 0; i < (GraphTraceLen / 2); ++i)
616 GraphBuffer[i] = GraphBuffer[i * 2];
617 GraphTraceLen /= 2;
618 PrintAndLog("decimated by 2");
619 RepaintGraphWindow();
620 return 0;
621 }
622 /**
623 * Undecimate - I'd call it 'interpolate', but we'll save that
624 * name until someone does an actual interpolation command, not just
625 * blindly repeating samples
626 * @param Cmd
627 * @return
628 */
629 int CmdUndec(const char *Cmd)
630 {
631 if(param_getchar(Cmd, 0) == 'h')
632 {
633 PrintAndLog("Usage: data undec [factor]");
634 PrintAndLog("This function performs un-decimation, by repeating each sample N times");
635 PrintAndLog("Options: ");
636 PrintAndLog(" h This help");
637 PrintAndLog(" factor The number of times to repeat each sample.[default:2]");
638 PrintAndLog("Example: 'data undec 3'");
639 return 0;
640 }
641
642 uint8_t factor = param_get8ex(Cmd, 0,2, 10);
643 //We have memory, don't we?
644 int swap[MAX_GRAPH_TRACE_LEN] = { 0 };
645 uint32_t g_index = 0, s_index = 0;
646 while(g_index < GraphTraceLen && s_index + factor < MAX_GRAPH_TRACE_LEN)
647 {
648 int count = 0;
649 for(count = 0; count < factor && s_index + count < MAX_GRAPH_TRACE_LEN; count++)
650 swap[s_index+count] = GraphBuffer[g_index];
651
652 s_index += count;
653 g_index++;
654 }
655
656 memcpy(GraphBuffer, swap, s_index * sizeof(int));
657 GraphTraceLen = s_index;
658 RepaintGraphWindow();
659 return 0;
660 }
661
662 //by marshmellow
663 //shift graph zero up or down based on input + or -
664 int CmdGraphShiftZero(const char *Cmd)
665 {
666
667 int shift=0;
668 //set options from parameters entered with the command
669 sscanf(Cmd, "%i", &shift);
670 int shiftedVal=0;
671 for(int i = 0; i<GraphTraceLen; i++){
672 shiftedVal=GraphBuffer[i]+shift;
673 if (shiftedVal>127)
674 shiftedVal=127;
675 else if (shiftedVal<-127)
676 shiftedVal=-127;
677 GraphBuffer[i]= shiftedVal;
678 }
679 CmdNorm("");
680 return 0;
681 }
682
683 int AskEdgeDetect(const int *in, int *out, int len, int threshold) {
684 int Last = 0;
685 for(int i = 1; i<len; i++) {
686 if (in[i]-in[i-1] >= threshold) //large jump up
687 Last = 127;
688 else if(in[i]-in[i-1] <= -1 * threshold) //large jump down
689 Last = -127;
690 out[i-1] = Last;
691 }
692 return 0;
693 }
694
695 //by marshmellow
696 //use large jumps in read samples to identify edges of waves and then amplify that wave to max
697 //similar to dirtheshold, threshold commands
698 //takes a threshold length which is the measured length between two samples then determines an edge
699 int CmdAskEdgeDetect(const char *Cmd)
700 {
701 int thresLen = 25;
702 int ans = 0;
703 sscanf(Cmd, "%i", &thresLen);
704
705 ans = AskEdgeDetect(GraphBuffer, GraphBuffer, GraphTraceLen, thresLen);
706 RepaintGraphWindow();
707 return ans;
708 }
709
710 /* Print our clock rate */
711 // uses data from graphbuffer
712 // adjusted to take char parameter for type of modulation to find the clock - by marshmellow.
713 int CmdDetectClockRate(const char *Cmd)
714 {
715 char cmdp = param_getchar(Cmd, 0);
716 if (strlen(Cmd) > 6 || strlen(Cmd) == 0 || cmdp == 'h' || cmdp == 'H') {
717 PrintAndLog("Usage: data detectclock [modulation] <clock>");
718 PrintAndLog(" [modulation as char], specify the modulation type you want to detect the clock of");
719 PrintAndLog(" <clock> , specify the clock (optional - to get best start position only)");
720 PrintAndLog(" 'a' = ask, 'f' = fsk, 'n' = nrz/direct, 'p' = psk");
721 PrintAndLog("");
722 PrintAndLog(" sample: data detectclock a = detect the clock of an ask modulated wave in the GraphBuffer");
723 PrintAndLog(" data detectclock f = detect the clock of an fsk modulated wave in the GraphBuffer");
724 PrintAndLog(" data detectclock p = detect the clock of an psk modulated wave in the GraphBuffer");
725 PrintAndLog(" data detectclock n = detect the clock of an nrz/direct modulated wave in the GraphBuffer");
726 }
727 int ans=0;
728 if (cmdp == 'a'){
729 ans = GetAskClock(Cmd+1, true, false);
730 } else if (cmdp == 'f'){
731 ans = GetFskClock("", true, false);
732 } else if (cmdp == 'n'){
733 ans = GetNrzClock("", true, false);
734 } else if (cmdp == 'p'){
735 ans = GetPskClock("", true, false);
736 } else {
737 PrintAndLog ("Please specify a valid modulation to detect the clock of - see option h for help");
738 }
739 return ans;
740 }
741
742 char *GetFSKType(uint8_t fchigh, uint8_t fclow, uint8_t invert)
743 {
744 static char fType[8];
745 memset(fType, 0x00, 8);
746 char *fskType = fType;
747 if (fchigh==10 && fclow==8){
748 if (invert) //fsk2a
749 memcpy(fskType, "FSK2a", 5);
750 else //fsk2
751 memcpy(fskType, "FSK2", 4);
752 } else if (fchigh == 8 && fclow == 5) {
753 if (invert)
754 memcpy(fskType, "FSK1", 4);
755 else
756 memcpy(fskType, "FSK1a", 5);
757 } else {
758 memcpy(fskType, "FSK??", 5);
759 }
760 return fskType;
761 }
762
763 //by marshmellow
764 //fsk raw demod and print binary
765 //takes 4 arguments - Clock, invert, fchigh, fclow
766 //defaults: clock = 50, invert=1, fchigh=10, fclow=8 (RF/10 RF/8 (fsk2a))
767 int FSKrawDemod(const char *Cmd, bool verbose)
768 {
769 //raw fsk demod no manchester decoding no start bit finding just get binary from wave
770 uint8_t rfLen, invert, fchigh, fclow;
771 //set defaults
772 //set options from parameters entered with the command
773 rfLen = param_get8(Cmd, 0);
774 invert = param_get8(Cmd, 1);
775 fchigh = param_get8(Cmd, 2);
776 fclow = param_get8(Cmd, 3);
777
778 if (strlen(Cmd)>0 && strlen(Cmd)<=2) {
779 if (rfLen==1) {
780 invert = 1; //if invert option only is used
781 rfLen = 0;
782 }
783 }
784 uint8_t BitStream[MAX_GRAPH_TRACE_LEN]={0};
785 size_t BitLen = getFromGraphBuf(BitStream);
786 if (BitLen==0) return 0;
787 //get field clock lengths
788 uint16_t fcs=0;
789 if (!fchigh || !fclow) {
790 fcs = countFC(BitStream, BitLen, 1);
791 if (!fcs) {
792 fchigh = 10;
793 fclow = 8;
794 } else {
795 fchigh = (fcs >> 8) & 0x00FF;
796 fclow = fcs & 0x00FF;
797 }
798 }
799 //get bit clock length
800 if (!rfLen) {
801 int firstClockEdge = 0; //todo - align grid on graph with this...
802 rfLen = detectFSKClk(BitStream, BitLen, fchigh, fclow, &firstClockEdge);
803 if (!rfLen) rfLen = 50;
804 }
805 int startIdx = 0;
806 int size = fskdemod_ext(BitStream, BitLen, rfLen, invert, fchigh, fclow, &startIdx);
807 if (size > 0) {
808 setDemodBuf(BitStream,size,0);
809 setClockGrid(rfLen, startIdx);
810
811 // Now output the bitstream to the scrollback by line of 16 bits
812 if (verbose || g_debugMode) {
813 PrintAndLog("\nUsing Clock:%u, invert:%u, fchigh:%u, fclow:%u", (unsigned int)rfLen, (unsigned int)invert, (unsigned int)fchigh, (unsigned int)fclow);
814 PrintAndLog("%s decoded bitstream:",GetFSKType(fchigh,fclow,invert));
815 printDemodBuff();
816 }
817
818 return 1;
819 } else {
820 if (g_debugMode) PrintAndLog("no FSK data found");
821 }
822 return 0;
823 }
824
825 //by marshmellow
826 //fsk raw demod and print binary
827 //takes 4 arguments - Clock, invert, fchigh, fclow
828 //defaults: clock = 50, invert=1, fchigh=10, fclow=8 (RF/10 RF/8 (fsk2a))
829 int CmdFSKrawdemod(const char *Cmd)
830 {
831 char cmdp = param_getchar(Cmd, 0);
832 if (strlen(Cmd) > 20 || cmdp == 'h' || cmdp == 'H') {
833 PrintAndLog("Usage: data rawdemod fs [clock] <invert> [fchigh] [fclow]");
834 PrintAndLog(" [set clock as integer] optional, omit for autodetect.");
835 PrintAndLog(" <invert>, 1 for invert output, can be used even if the clock is omitted");
836 PrintAndLog(" [fchigh], larger field clock length, omit for autodetect");
837 PrintAndLog(" [fclow], small field clock length, omit for autodetect");
838 PrintAndLog("");
839 PrintAndLog(" sample: data rawdemod fs = demod an fsk tag from GraphBuffer using autodetect");
840 PrintAndLog(" : data rawdemod fs 32 = demod an fsk tag from GraphBuffer using a clock of RF/32, autodetect fc");
841 PrintAndLog(" : data rawdemod fs 1 = demod an fsk tag from GraphBuffer using autodetect, invert output");
842 PrintAndLog(" : data rawdemod fs 32 1 = demod an fsk tag from GraphBuffer using a clock of RF/32, invert output, autodetect fc");
843 PrintAndLog(" : data rawdemod fs 64 0 8 5 = demod an fsk1 RF/64 tag from GraphBuffer");
844 PrintAndLog(" : data rawdemod fs 50 0 10 8 = demod an fsk2 RF/50 tag from GraphBuffer");
845 PrintAndLog(" : data rawdemod fs 50 1 10 8 = demod an fsk2a RF/50 tag from GraphBuffer");
846 return 0;
847 }
848 return FSKrawDemod(Cmd, true);
849 }
850
851 //by marshmellow
852 //attempt to psk1 demod graph buffer
853 int PSKDemod(const char *Cmd, bool verbose)
854 {
855 int invert=0;
856 int clk=0;
857 int maxErr=100;
858 sscanf(Cmd, "%i %i %i", &clk, &invert, &maxErr);
859 if (clk==1){
860 invert=1;
861 clk=0;
862 }
863 if (invert != 0 && invert != 1) {
864 if (g_debugMode || verbose) PrintAndLog("Invalid argument: %s", Cmd);
865 return 0;
866 }
867 uint8_t BitStream[MAX_GRAPH_TRACE_LEN]={0};
868 size_t BitLen = getFromGraphBuf(BitStream);
869 if (BitLen==0) return 0;
870 int errCnt=0;
871 int startIdx = 0;
872 errCnt = pskRawDemod_ext(BitStream, &BitLen, &clk, &invert, &startIdx);
873 if (errCnt > maxErr){
874 if (g_debugMode || verbose) PrintAndLog("Too many errors found, clk: %d, invert: %d, numbits: %d, errCnt: %d",clk,invert,BitLen,errCnt);
875 return 0;
876 }
877 if (errCnt<0|| BitLen<16){ //throw away static - allow 1 and -1 (in case of threshold command first)
878 if (g_debugMode || verbose) PrintAndLog("no data found, clk: %d, invert: %d, numbits: %d, errCnt: %d",clk,invert,BitLen,errCnt);
879 return 0;
880 }
881 if (verbose || g_debugMode){
882 PrintAndLog("\nUsing Clock:%d, invert:%d, Bits Found:%d",clk,invert,BitLen);
883 if (errCnt>0){
884 PrintAndLog("# Errors during Demoding (shown as 7 in bit stream): %d",errCnt);
885 }
886 }
887 //prime demod buffer for output
888 setDemodBuf(BitStream,BitLen,0);
889 setClockGrid(clk, startIdx);
890
891 return 1;
892 }
893
894 // by marshmellow
895 // takes 3 arguments - clock, invert, maxErr as integers
896 // attempts to demodulate nrz only
897 // prints binary found and saves in demodbuffer for further commands
898 int NRZrawDemod(const char *Cmd, bool verbose)
899 {
900 int invert=0;
901 int clk=0;
902 int maxErr=100;
903 sscanf(Cmd, "%i %i %i", &clk, &invert, &maxErr);
904 if (clk==1){
905 invert=1;
906 clk=0;
907 }
908 if (invert != 0 && invert != 1) {
909 PrintAndLog("Invalid argument: %s", Cmd);
910 return 0;
911 }
912 uint8_t BitStream[MAX_GRAPH_TRACE_LEN]={0};
913 size_t BitLen = getFromGraphBuf(BitStream);
914 if (BitLen==0) return 0;
915 int errCnt=0;
916 int clkStartIdx = 0;
917 errCnt = nrzRawDemod(BitStream, &BitLen, &clk, &invert, &clkStartIdx);
918 if (errCnt > maxErr){
919 if (g_debugMode) PrintAndLog("Too many errors found, clk: %d, invert: %d, numbits: %d, errCnt: %d",clk,invert,BitLen,errCnt);
920 return 0;
921 }
922 if (errCnt<0 || BitLen<16){ //throw away static - allow 1 and -1 (in case of threshold command first)
923 if (g_debugMode) PrintAndLog("no data found, clk: %d, invert: %d, numbits: %d, errCnt: %d",clk,invert,BitLen,errCnt);
924 return 0;
925 }
926 if (verbose || g_debugMode) PrintAndLog("Tried NRZ Demod using Clock: %d - invert: %d - Bits Found: %d",clk,invert,BitLen);
927 //prime demod buffer for output
928 setDemodBuf(BitStream,BitLen,0);
929 setClockGrid(clk, clkStartIdx);
930
931
932 if (errCnt>0 && (verbose || g_debugMode)) PrintAndLog("# Errors during Demoding (shown as 7 in bit stream): %d",errCnt);
933 if (verbose || g_debugMode) {
934 PrintAndLog("NRZ demoded bitstream:");
935 // Now output the bitstream to the scrollback by line of 16 bits
936 printDemodBuff();
937 }
938 return 1;
939 }
940
941 int CmdNRZrawDemod(const char *Cmd)
942 {
943 char cmdp = param_getchar(Cmd, 0);
944 if (strlen(Cmd) > 16 || cmdp == 'h' || cmdp == 'H') {
945 PrintAndLog("Usage: data rawdemod nr [clock] <0|1> [maxError]");
946 PrintAndLog(" [set clock as integer] optional, if not set, autodetect.");
947 PrintAndLog(" <invert>, 1 for invert output");
948 PrintAndLog(" [set maximum allowed errors], default = 100.");
949 PrintAndLog("");
950 PrintAndLog(" sample: data rawdemod nr = demod a nrz/direct tag from GraphBuffer");
951 PrintAndLog(" : data rawdemod nr 32 = demod a nrz/direct tag from GraphBuffer using a clock of RF/32");
952 PrintAndLog(" : data rawdemod nr 32 1 = demod a nrz/direct tag from GraphBuffer using a clock of RF/32 and inverting data");
953 PrintAndLog(" : data rawdemod nr 1 = demod a nrz/direct tag from GraphBuffer while inverting data");
954 PrintAndLog(" : data rawdemod nr 64 1 0 = demod a nrz/direct tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors");
955 return 0;
956 }
957 return NRZrawDemod(Cmd, true);
958 }
959
960 // by marshmellow
961 // takes 3 arguments - clock, invert, maxErr as integers
962 // attempts to demodulate psk only
963 // prints binary found and saves in demodbuffer for further commands
964 int CmdPSK1rawDemod(const char *Cmd)
965 {
966 int ans;
967 char cmdp = param_getchar(Cmd, 0);
968 if (strlen(Cmd) > 16 || cmdp == 'h' || cmdp == 'H') {
969 PrintAndLog("Usage: data rawdemod p1 [clock] <0|1> [maxError]");
970 PrintAndLog(" [set clock as integer] optional, if not set, autodetect.");
971 PrintAndLog(" <invert>, 1 for invert output");
972 PrintAndLog(" [set maximum allowed errors], default = 100.");
973 PrintAndLog("");
974 PrintAndLog(" sample: data rawdemod p1 = demod a psk1 tag from GraphBuffer");
975 PrintAndLog(" : data rawdemod p1 32 = demod a psk1 tag from GraphBuffer using a clock of RF/32");
976 PrintAndLog(" : data rawdemod p1 32 1 = demod a psk1 tag from GraphBuffer using a clock of RF/32 and inverting data");
977 PrintAndLog(" : data rawdemod p1 1 = demod a psk1 tag from GraphBuffer while inverting data");
978 PrintAndLog(" : data rawdemod p1 64 1 0 = demod a psk1 tag from GraphBuffer using a clock of RF/64, inverting data and allowing 0 demod errors");
979 return 0;
980 }
981 ans = PSKDemod(Cmd, true);
982 //output
983 if (!ans){
984 if (g_debugMode) PrintAndLog("Error demoding: %d",ans);
985 return 0;
986 }
987
988 PrintAndLog("PSK1 demoded bitstream:");
989 // Now output the bitstream to the scrollback by line of 16 bits
990 printDemodBuff();
991 return 1;
992 }
993
994 // by marshmellow
995 // takes same args as cmdpsk1rawdemod
996 int CmdPSK2rawDemod(const char *Cmd)
997 {
998 int ans=0;
999 char cmdp = param_getchar(Cmd, 0);
1000 if (strlen(Cmd) > 16 || cmdp == 'h' || cmdp == 'H') {
1001 PrintAndLog("Usage: data rawdemod p2 [clock] <0|1> [maxError]");
1002 PrintAndLog(" [set clock as integer] optional, if not set, autodetect.");
1003 PrintAndLog(" <invert>, 1 for invert output");
1004 PrintAndLog(" [set maximum allowed errors], default = 100.");
1005 PrintAndLog("");
1006 PrintAndLog(" sample: data rawdemod p2 = demod a psk2 tag from GraphBuffer, autodetect clock");
1007 PrintAndLog(" : data rawdemod p2 32 = demod a psk2 tag from GraphBuffer using a clock of RF/32");
1008 PrintAndLog(" : data rawdemod p2 32 1 = demod a psk2 tag from GraphBuffer using a clock of RF/32 and inverting output");
1009 PrintAndLog(" : data rawdemod p2 1 = demod a psk2 tag from GraphBuffer, autodetect clock and invert output");
1010 PrintAndLog(" : data rawdemod p2 64 1 0 = demod a psk2 tag from GraphBuffer using a clock of RF/64, inverting output and allowing 0 demod errors");
1011 return 0;
1012 }
1013 ans=PSKDemod(Cmd, true);
1014 if (!ans){
1015 if (g_debugMode) PrintAndLog("Error demoding: %d",ans);
1016 return 0;
1017 }
1018 psk1TOpsk2(DemodBuffer, DemodBufferLen);
1019 PrintAndLog("PSK2 demoded bitstream:");
1020 // Now output the bitstream to the scrollback by line of 16 bits
1021 printDemodBuff();
1022 return 1;
1023 }
1024
1025 // by marshmellow - combines all raw demod functions into one menu command
1026 int CmdRawDemod(const char *Cmd)
1027 {
1028 char cmdp = Cmd[0]; //param_getchar(Cmd, 0);
1029
1030 if (strlen(Cmd) > 35 || cmdp == 'h' || cmdp == 'H' || strlen(Cmd)<2) {
1031 PrintAndLog("Usage: data rawdemod [modulation] <help>|<options>");
1032 PrintAndLog(" [modulation] as 2 char, 'ab' for ask/biphase, 'am' for ask/manchester, 'ar' for ask/raw, 'fs' for fsk, ...");
1033 PrintAndLog(" 'nr' for nrz/direct, 'p1' for psk1, 'p2' for psk2");
1034 PrintAndLog(" <help> as 'h', prints the help for the specific modulation");
1035 PrintAndLog(" <options> see specific modulation help for optional parameters");
1036 PrintAndLog("");
1037 PrintAndLog(" sample: data rawdemod fs h = print help specific to fsk demod");
1038 PrintAndLog(" : data rawdemod fs = demod GraphBuffer using: fsk - autodetect");
1039 PrintAndLog(" : data rawdemod ab = demod GraphBuffer using: ask/biphase - autodetect");
1040 PrintAndLog(" : data rawdemod am = demod GraphBuffer using: ask/manchester - autodetect");
1041 PrintAndLog(" : data rawdemod ar = demod GraphBuffer using: ask/raw - autodetect");
1042 PrintAndLog(" : data rawdemod nr = demod GraphBuffer using: nrz/direct - autodetect");
1043 PrintAndLog(" : data rawdemod p1 = demod GraphBuffer using: psk1 - autodetect");
1044 PrintAndLog(" : data rawdemod p2 = demod GraphBuffer using: psk2 - autodetect");
1045 return 0;
1046 }
1047 char cmdp2 = Cmd[1];
1048 int ans = 0;
1049 if (cmdp == 'f' && cmdp2 == 's'){
1050 ans = CmdFSKrawdemod(Cmd+2);
1051 } else if(cmdp == 'a' && cmdp2 == 'b'){
1052 ans = Cmdaskbiphdemod(Cmd+2);
1053 } else if(cmdp == 'a' && cmdp2 == 'm'){
1054 ans = Cmdaskmandemod(Cmd+2);
1055 } else if(cmdp == 'a' && cmdp2 == 'r'){
1056 ans = Cmdaskrawdemod(Cmd+2);
1057 } else if(cmdp == 'n' && cmdp2 == 'r'){
1058 ans = CmdNRZrawDemod(Cmd+2);
1059 } else if(cmdp == 'p' && cmdp2 == '1'){
1060 ans = CmdPSK1rawDemod(Cmd+2);
1061 } else if(cmdp == 'p' && cmdp2 == '2'){
1062 ans = CmdPSK2rawDemod(Cmd+2);
1063 } else {
1064 PrintAndLog("unknown modulation entered - see help ('h') for parameter structure");
1065 }
1066 return ans;
1067 }
1068
1069 void setClockGrid(int clk, int offset) {
1070 g_DemodStartIdx = offset;
1071 g_DemodClock = clk;
1072 if (g_debugMode) PrintAndLog("demodoffset %d, clk %d",offset,clk);
1073 if (offset > clk) offset %= clk;
1074 if (offset < 0) offset += clk;
1075
1076 if (offset > GraphTraceLen || offset < 0) return;
1077 if (clk < 8 || clk > GraphTraceLen) {
1078 GridLocked = false;
1079 GridOffset = 0;
1080 PlotGridX = 0;
1081 PlotGridXdefault = 0;
1082 RepaintGraphWindow();
1083 } else {
1084 GridLocked = true;
1085 GridOffset = offset;
1086 PlotGridX = clk;
1087 PlotGridXdefault = clk;
1088 RepaintGraphWindow();
1089 }
1090 }
1091
1092 int CmdGrid(const char *Cmd)
1093 {
1094 sscanf(Cmd, "%i %i", &PlotGridX, &PlotGridY);
1095 PlotGridXdefault= PlotGridX;
1096 PlotGridYdefault= PlotGridY;
1097 RepaintGraphWindow();
1098 return 0;
1099 }
1100
1101 int CmdSetGraphMarkers(const char *Cmd) {
1102 sscanf(Cmd, "%i %i", &CursorCPos, &CursorDPos);
1103 RepaintGraphWindow();
1104 return 0;
1105 }
1106
1107 int CmdHexsamples(const char *Cmd)
1108 {
1109 int i, j;
1110 int requested = 0;
1111 int offset = 0;
1112 char string_buf[25];
1113 char* string_ptr = string_buf;
1114 uint8_t got[BIGBUF_SIZE];
1115
1116 sscanf(Cmd, "%i %i", &requested, &offset);
1117
1118 /* if no args send something */
1119 if (requested == 0) {
1120 requested = 8;
1121 }
1122 if (offset + requested > sizeof(got)) {
1123 PrintAndLog("Tried to read past end of buffer, <bytes> + <offset> > %d", BIGBUF_SIZE);
1124 return 0;
1125 }
1126
1127 GetFromBigBuf(got,requested,offset);
1128 WaitForResponse(CMD_ACK,NULL);
1129
1130 i = 0;
1131 for (j = 0; j < requested; j++) {
1132 i++;
1133 string_ptr += sprintf(string_ptr, "%02x ", got[j]);
1134 if (i == 8) {
1135 *(string_ptr - 1) = '\0'; // remove the trailing space
1136 PrintAndLog("%s", string_buf);
1137 string_buf[0] = '\0';
1138 string_ptr = string_buf;
1139 i = 0;
1140 }
1141 if (j == requested - 1 && string_buf[0] != '\0') { // print any remaining bytes
1142 *(string_ptr - 1) = '\0';
1143 PrintAndLog("%s", string_buf);
1144 string_buf[0] = '\0';
1145 }
1146 }
1147 return 0;
1148 }
1149
1150 int CmdHide(const char *Cmd)
1151 {
1152 HideGraphWindow();
1153 return 0;
1154 }
1155
1156 //zero mean GraphBuffer
1157 int CmdHpf(const char *Cmd)
1158 {
1159 int i;
1160 int accum = 0;
1161
1162 for (i = 10; i < GraphTraceLen; ++i)
1163 accum += GraphBuffer[i];
1164 accum /= (GraphTraceLen - 10);
1165 for (i = 0; i < GraphTraceLen; ++i)
1166 GraphBuffer[i] -= accum;
1167
1168 RepaintGraphWindow();
1169 return 0;
1170 }
1171
1172 uint8_t getByte(uint8_t bits_per_sample, BitstreamIn* b)
1173 {
1174 int i;
1175 uint8_t val = 0;
1176 for(i =0 ; i < bits_per_sample; i++)
1177 {
1178 val |= (headBit(b) << (7-i));
1179 }
1180 return val;
1181 }
1182
1183 int getSamples(int n, bool silent)
1184 {
1185 //If we get all but the last byte in bigbuf,
1186 // we don't have to worry about remaining trash
1187 // in the last byte in case the bits-per-sample
1188 // does not line up on byte boundaries
1189
1190 uint8_t got[BIGBUF_SIZE-1] = { 0 };
1191
1192 if (n == 0 || n > sizeof(got))
1193 n = sizeof(got);
1194
1195 if (!silent) PrintAndLog("Reading %d bytes from device memory\n", n);
1196 GetFromBigBuf(got,n,0);
1197 if (!silent) PrintAndLog("Data fetched");
1198 UsbCommand response;
1199 WaitForResponse(CMD_ACK, &response);
1200 uint8_t bits_per_sample = 8;
1201
1202 //Old devices without this feature would send 0 at arg[0]
1203 if(response.arg[0] > 0)
1204 {
1205 sample_config *sc = (sample_config *) response.d.asBytes;
1206 if (!silent) PrintAndLog("Samples @ %d bits/smpl, decimation 1:%d ", sc->bits_per_sample
1207 , sc->decimation);
1208 bits_per_sample = sc->bits_per_sample;
1209 }
1210 if(bits_per_sample < 8)
1211 {
1212 if (!silent) PrintAndLog("Unpacking...");
1213 BitstreamIn bout = { got, bits_per_sample * n, 0};
1214 int j =0;
1215 for (j = 0; j * bits_per_sample < n * 8 && j < n; j++) {
1216 uint8_t sample = getByte(bits_per_sample, &bout);
1217 GraphBuffer[j] = ((int) sample )- 128;
1218 }
1219 GraphTraceLen = j;
1220 PrintAndLog("Unpacked %d samples" , j );
1221 }else
1222 {
1223 for (int j = 0; j < n; j++) {
1224 GraphBuffer[j] = ((int)got[j]) - 128;
1225 }
1226 GraphTraceLen = n;
1227 }
1228
1229 setClockGrid(0,0);
1230 RepaintGraphWindow();
1231 return 0;
1232 }
1233
1234 int CmdSamples(const char *Cmd)
1235 {
1236 int n = strtol(Cmd, NULL, 0);
1237 return getSamples(n, false);
1238 }
1239
1240 int CmdTuneSamples(const char *Cmd)
1241 {
1242 int timeout = 0, arg = FLAG_TUNE_ALL;
1243
1244 if(*Cmd == 'l') {
1245 arg = FLAG_TUNE_LF;
1246 } else if (*Cmd == 'h') {
1247 arg = FLAG_TUNE_HF;
1248 } else if (*Cmd != '\0') {
1249 PrintAndLog("use 'tune' or 'tune l' or 'tune h'");
1250 return 0;
1251 }
1252
1253 printf("\nMeasuring antenna characteristics, please wait...");
1254
1255 UsbCommand c = {CMD_MEASURE_ANTENNA_TUNING, {arg, 0, 0}};
1256 SendCommand(&c);
1257
1258 UsbCommand resp;
1259 while(!WaitForResponseTimeout(CMD_MEASURED_ANTENNA_TUNING,&resp,1000)) {
1260 timeout++;
1261 printf(".");
1262 if (timeout > 7) {
1263 PrintAndLog("\nNo response from Proxmark. Aborting...");
1264 return 1;
1265 }
1266 }
1267
1268 int peakv, peakf;
1269 int vLf125, vLf134, vHf;
1270 vLf125 = resp.arg[0] & 0xffff;
1271 vLf134 = resp.arg[0] >> 16;
1272 vHf = resp.arg[1] & 0xffff;;
1273 peakf = resp.arg[2] & 0xffff;
1274 peakv = resp.arg[2] >> 16;
1275 PrintAndLog("");
1276 PrintAndLog("# LF antenna: %5.2f V @ 125.00 kHz", vLf125/1000.0);
1277 PrintAndLog("# LF antenna: %5.2f V @ 134.00 kHz", vLf134/1000.0);
1278 PrintAndLog("# LF optimal: %5.2f V @%9.2f kHz", peakv/1000.0, 12000.0/(peakf+1));
1279 PrintAndLog("# HF antenna: %5.2f V @ 13.56 MHz", vHf/1000.0);
1280
1281 #define LF_UNUSABLE_V 2948 // was 2000. Changed due to bugfix in voltage measurements. LF results are now 47% higher.
1282 #define LF_MARGINAL_V 14739 // was 10000. Changed due to bugfix bug in voltage measurements. LF results are now 47% higher.
1283 #define HF_UNUSABLE_V 3167 // was 2000. Changed due to bugfix in voltage measurements. HF results are now 58% higher.
1284 #define HF_MARGINAL_V 7917 // was 5000. Changed due to bugfix in voltage measurements. HF results are now 58% higher.
1285
1286 if (peakv < LF_UNUSABLE_V)
1287 PrintAndLog("# Your LF antenna is unusable.");
1288 else if (peakv < LF_MARGINAL_V)
1289 PrintAndLog("# Your LF antenna is marginal.");
1290 if (vHf < HF_UNUSABLE_V)
1291 PrintAndLog("# Your HF antenna is unusable.");
1292 else if (vHf < HF_MARGINAL_V)
1293 PrintAndLog("# Your HF antenna is marginal.");
1294
1295 if (peakv >= LF_UNUSABLE_V) {
1296 for (int i = 0; i < 256; i++) {
1297 GraphBuffer[i] = resp.d.asBytes[i] - 128;
1298 }
1299 PrintAndLog("Displaying LF tuning graph. Divisor 89 is 134khz, 95 is 125khz.\n");
1300 PrintAndLog("\n");
1301 GraphTraceLen = 256;
1302 ShowGraphWindow();
1303 RepaintGraphWindow();
1304 }
1305
1306 return 0;
1307 }
1308
1309
1310 int CmdLoad(const char *Cmd)
1311 {
1312 char filename[FILE_PATH_SIZE] = {0x00};
1313 int len = 0;
1314
1315 len = strlen(Cmd);
1316 if (len > FILE_PATH_SIZE) len = FILE_PATH_SIZE;
1317 memcpy(filename, Cmd, len);
1318
1319 FILE *f = fopen(filename, "r");
1320 if (!f) {
1321 PrintAndLog("couldn't open '%s'", filename);
1322 return 0;
1323 }
1324
1325 GraphTraceLen = 0;
1326 char line[80];
1327 while (fgets(line, sizeof (line), f)) {
1328 GraphBuffer[GraphTraceLen] = atoi(line);
1329 GraphTraceLen++;
1330 }
1331 fclose(f);
1332 PrintAndLog("loaded %d samples", GraphTraceLen);
1333 setClockGrid(0,0);
1334 RepaintGraphWindow();
1335 return 0;
1336 }
1337
1338 int CmdLtrim(const char *Cmd)
1339 {
1340 int ds = atoi(Cmd);
1341 if (GraphTraceLen<=0) return 0;
1342 for (int i = ds; i < GraphTraceLen; ++i)
1343 GraphBuffer[i-ds] = GraphBuffer[i];
1344 GraphTraceLen -= ds;
1345
1346 RepaintGraphWindow();
1347 return 0;
1348 }
1349
1350 // trim graph to input argument length
1351 int CmdRtrim(const char *Cmd)
1352 {
1353 int ds = atoi(Cmd);
1354
1355 GraphTraceLen = ds;
1356
1357 RepaintGraphWindow();
1358 return 0;
1359 }
1360
1361 // trim graph (middle) piece
1362 int CmdMtrim(const char *Cmd) {
1363 int start = 0, stop = 0;
1364 sscanf(Cmd, "%i %i", &start, &stop);
1365
1366 if (start > GraphTraceLen || stop > GraphTraceLen || start > stop) return 0;
1367 start++; //leave start position sample
1368
1369 GraphTraceLen = stop - start;
1370 for (int i = 0; i < GraphTraceLen; i++) {
1371 GraphBuffer[i] = GraphBuffer[start+i];
1372 }
1373 return 0;
1374 }
1375
1376
1377 int CmdNorm(const char *Cmd)
1378 {
1379 int i;
1380 int max = INT_MIN, min = INT_MAX;
1381
1382 for (i = 10; i < GraphTraceLen; ++i) {
1383 if (GraphBuffer[i] > max)
1384 max = GraphBuffer[i];
1385 if (GraphBuffer[i] < min)
1386 min = GraphBuffer[i];
1387 }
1388
1389 if (max != min) {
1390 for (i = 0; i < GraphTraceLen; ++i) {
1391 GraphBuffer[i] = (GraphBuffer[i] - ((max + min) / 2)) * 256 /
1392 (max - min);
1393 //marshmelow: adjusted *1000 to *256 to make +/- 128 so demod commands still work
1394 }
1395 }
1396 RepaintGraphWindow();
1397 return 0;
1398 }
1399
1400 int CmdPlot(const char *Cmd)
1401 {
1402 ShowGraphWindow();
1403 return 0;
1404 }
1405
1406 int CmdSave(const char *Cmd)
1407 {
1408 char filename[FILE_PATH_SIZE] = {0x00};
1409 int len = 0;
1410
1411 len = strlen(Cmd);
1412 if (len > FILE_PATH_SIZE) len = FILE_PATH_SIZE;
1413 memcpy(filename, Cmd, len);
1414
1415
1416 FILE *f = fopen(filename, "w");
1417 if(!f) {
1418 PrintAndLog("couldn't open '%s'", filename);
1419 return 0;
1420 }
1421 int i;
1422 for (i = 0; i < GraphTraceLen; i++) {
1423 fprintf(f, "%d\n", GraphBuffer[i]);
1424 }
1425 fclose(f);
1426 PrintAndLog("saved to '%s'", Cmd);
1427 return 0;
1428 }
1429
1430 int CmdScale(const char *Cmd)
1431 {
1432 CursorScaleFactor = atoi(Cmd);
1433 if (CursorScaleFactor == 0) {
1434 PrintAndLog("bad, can't have zero scale");
1435 CursorScaleFactor = 1;
1436 }
1437 RepaintGraphWindow();
1438 return 0;
1439 }
1440
1441 int directionalThreshold(const int* in, int *out, size_t len, int8_t up, int8_t down)
1442 {
1443 int lastValue = in[0];
1444 out[0] = 0; // Will be changed at the end, but init 0 as we adjust to last samples value if no threshold kicks in.
1445
1446 for (int i = 1; i < len; ++i) {
1447 // Apply first threshold to samples heading up
1448 if (in[i] >= up && in[i] > lastValue)
1449 {
1450 lastValue = out[i]; // Buffer last value as we overwrite it.
1451 out[i] = 1;
1452 }
1453 // Apply second threshold to samples heading down
1454 else if (in[i] <= down && in[i] < lastValue)
1455 {
1456 lastValue = out[i]; // Buffer last value as we overwrite it.
1457 out[i] = -1;
1458 }
1459 else
1460 {
1461 lastValue = out[i]; // Buffer last value as we overwrite it.
1462 out[i] = out[i-1];
1463 }
1464 }
1465 out[0] = out[1]; // Align with first edited sample.
1466 return 0;
1467 }
1468
1469 int CmdDirectionalThreshold(const char *Cmd)
1470 {
1471 int8_t upThres = param_get8(Cmd, 0);
1472 int8_t downThres = param_get8(Cmd, 1);
1473
1474 printf("Applying Up Threshold: %d, Down Threshold: %d\n", upThres, downThres);
1475
1476 directionalThreshold(GraphBuffer, GraphBuffer,GraphTraceLen, upThres, downThres);
1477 RepaintGraphWindow();
1478 return 0;
1479 }
1480
1481 int CmdZerocrossings(const char *Cmd)
1482 {
1483 // Zero-crossings aren't meaningful unless the signal is zero-mean.
1484 CmdHpf("");
1485
1486 int sign = 1;
1487 int zc = 0;
1488 int lastZc = 0;
1489
1490 for (int i = 0; i < GraphTraceLen; ++i) {
1491 if (GraphBuffer[i] * sign >= 0) {
1492 // No change in sign, reproduce the previous sample count.
1493 zc++;
1494 GraphBuffer[i] = lastZc;
1495 } else {
1496 // Change in sign, reset the sample count.
1497 sign = -sign;
1498 GraphBuffer[i] = lastZc;
1499 if (sign > 0) {
1500 lastZc = zc;
1501 zc = 0;
1502 }
1503 }
1504 }
1505
1506 RepaintGraphWindow();
1507 return 0;
1508 }
1509
1510 int usage_data_bin2hex(){
1511 PrintAndLog("Usage: data bin2hex <binary_digits>");
1512 PrintAndLog(" This function will ignore all characters not 1 or 0 (but stop reading on whitespace)");
1513 return 0;
1514 }
1515
1516 /**
1517 * @brief Utility for conversion via cmdline.
1518 * @param Cmd
1519 * @return
1520 */
1521 int Cmdbin2hex(const char *Cmd)
1522 {
1523 int bg =0, en =0;
1524 if(param_getptr(Cmd, &bg, &en, 0))
1525 {
1526 return usage_data_bin2hex();
1527 }
1528 //Number of digits supplied as argument
1529 size_t length = en - bg +1;
1530 size_t bytelen = (length+7) / 8;
1531 uint8_t* arr = (uint8_t *) malloc(bytelen);
1532 memset(arr, 0, bytelen);
1533 BitstreamOut bout = { arr, 0, 0 };
1534
1535 for(; bg <= en ;bg++)
1536 {
1537 char c = Cmd[bg];
1538 if( c == '1') pushBit(&bout, 1);
1539 else if( c == '0') pushBit(&bout, 0);
1540 else PrintAndLog("Ignoring '%c'", c);
1541 }
1542
1543 if(bout.numbits % 8 != 0)
1544 {
1545 printf("[padded with %d zeroes]\n", 8-(bout.numbits % 8));
1546 }
1547
1548 //Uses printf instead of PrintAndLog since the latter
1549 // adds linebreaks to each printout - this way was more convenient since we don't have to
1550 // allocate a string and write to that first...
1551 for(size_t x = 0; x < bytelen ; x++)
1552 {
1553 printf("%02X", arr[x]);
1554 }
1555 printf("\n");
1556 free(arr);
1557 return 0;
1558 }
1559
1560 int usage_data_hex2bin() {
1561 PrintAndLog("Usage: data hex2bin <hex_digits>");
1562 PrintAndLog(" This function will ignore all non-hexadecimal characters (but stop reading on whitespace)");
1563 return 0;
1564
1565 }
1566
1567 int Cmdhex2bin(const char *Cmd)
1568 {
1569 int bg =0, en =0;
1570 if(param_getptr(Cmd, &bg, &en, 0))
1571 {
1572 return usage_data_hex2bin();
1573 }
1574
1575
1576 while(bg <= en )
1577 {
1578 char x = Cmd[bg++];
1579 // capitalize
1580 if (x >= 'a' && x <= 'f')
1581 x -= 32;
1582 // convert to numeric value
1583 if (x >= '0' && x <= '9')
1584 x -= '0';
1585 else if (x >= 'A' && x <= 'F')
1586 x -= 'A' - 10;
1587 else
1588 continue;
1589
1590 //Uses printf instead of PrintAndLog since the latter
1591 // adds linebreaks to each printout - this way was more convenient since we don't have to
1592 // allocate a string and write to that first...
1593
1594 for(int i= 0 ; i < 4 ; ++i)
1595 printf("%d",(x >> (3 - i)) & 1);
1596 }
1597 printf("\n");
1598
1599 return 0;
1600 }
1601
1602 static command_t CommandTable[] =
1603 {
1604 {"help", CmdHelp, 1, "This help"},
1605 {"askedgedetect", CmdAskEdgeDetect, 1, "[threshold] Adjust Graph for manual ask demod using the length of sample differences to detect the edge of a wave (use 20-45, def:25)"},
1606 {"autocorr", CmdAutoCorr, 1, "[window length] [g] -- Autocorrelation over window - g to save back to GraphBuffer (overwrite)"},
1607 {"biphaserawdecode",CmdBiphaseDecodeRaw,1, "[offset] [invert<0|1>] [maxErr] -- Biphase decode bin stream in DemodBuffer (offset = 0|1 bits to shift the decode start)"},
1608 {"bin2hex", Cmdbin2hex, 1, "bin2hex <digits> -- Converts binary to hexadecimal"},
1609 {"bitsamples", CmdBitsamples, 0, "Get raw samples as bitstring"},
1610 {"buffclear", CmdBuffClear, 1, "Clear sample buffer and graph window"},
1611 {"dec", CmdDec, 1, "Decimate samples"},
1612 {"detectclock", CmdDetectClockRate, 1, "[modulation] Detect clock rate of wave in GraphBuffer (options: 'a','f','n','p' for ask, fsk, nrz, psk respectively)"},
1613 {"getbitstream", CmdGetBitStream, 1, "Convert GraphBuffer's >=1 values to 1 and <1 to 0"},
1614 {"grid", CmdGrid, 1, "<x> <y> -- overlay grid on graph window, use zero value to turn off either"},
1615 {"hexsamples", CmdHexsamples, 0, "<bytes> [<offset>] -- Dump big buffer as hex bytes"},
1616 {"hex2bin", Cmdhex2bin, 1, "hex2bin <hexadecimal> -- Converts hexadecimal to binary"},
1617 {"hide", CmdHide, 1, "Hide graph window"},
1618 {"hpf", CmdHpf, 1, "Remove DC offset from trace"},
1619 {"load", CmdLoad, 1, "<filename> -- Load trace (to graph window"},
1620 {"ltrim", CmdLtrim, 1, "<samples> -- Trim samples from left of trace"},
1621 {"rtrim", CmdRtrim, 1, "<location to end trace> -- Trim samples from right of trace"},
1622 {"mtrim", CmdMtrim, 1, "<start> <stop> -- Trim out samples from the specified start to the specified stop"},
1623 {"manrawdecode", Cmdmandecoderaw, 1, "[invert] [maxErr] -- Manchester decode binary stream in DemodBuffer"},
1624 {"norm", CmdNorm, 1, "Normalize max/min to +/-128"},
1625 {"plot", CmdPlot, 1, "Show graph window (hit 'h' in window for keystroke help)"},
1626 {"printdemodbuffer",CmdPrintDemodBuff, 1, "[x] [o] <offset> [l] <length> -- print the data in the DemodBuffer - 'x' for hex output"},
1627 {"rawdemod", CmdRawDemod, 1, "[modulation] ... <options> -see help (h option) -- Demodulate the data in the GraphBuffer and output binary"},
1628 {"samples", CmdSamples, 0, "[512 - 40000] -- Get raw samples for graph window (GraphBuffer)"},
1629 {"save", CmdSave, 1, "<filename> -- Save trace (from graph window)"},
1630 {"setgraphmarkers", CmdSetGraphMarkers, 1, "[orange_marker] [blue_marker] (in graph window)"},
1631 {"scale", CmdScale, 1, "<int> -- Set cursor display scale"},
1632 {"setdebugmode", CmdSetDebugMode, 1, "<0|1|2> -- Turn on or off Debugging Level for lf demods"},
1633 {"shiftgraphzero", CmdGraphShiftZero, 1, "<shift> -- Shift 0 for Graphed wave + or - shift value"},
1634 {"dirthreshold", CmdDirectionalThreshold, 1, "<thres up> <thres down> -- Max rising higher up-thres/ Min falling lower down-thres, keep rest as prev."},
1635 {"tune", CmdTuneSamples, 0, "Get hw tune samples for graph window"},
1636 {"undec", CmdUndec, 1, "Un-decimate samples by 2"},
1637 {"zerocrossings", CmdZerocrossings, 1, "Count time between zero-crossings"},
1638 {NULL, NULL, 0, NULL}
1639 };
1640
1641 int CmdData(const char *Cmd)
1642 {
1643 CmdsParse(CommandTable, Cmd);
1644 return 0;
1645 }
1646
1647 int CmdHelp(const char *Cmd)
1648 {
1649 CmdsHelp(CommandTable);
1650 return 0;
1651 }
Impressum, Datenschutz