Merge remote-tracking branch 'upstream/master'
[proxmark3-svn] / client / util.c
CommitLineData
20f9a2a1
M
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// utilities
9//-----------------------------------------------------------------------------
10
11#include "util.h"
534983d7 12
cb64309e 13#ifndef _WIN32
873014de
M
14#include <termios.h>
15#include <sys/ioctl.h>
79544b28 16
f397b5cc
M
17int ukbhit(void)
18{
19 int cnt = 0;
20 int error;
21 static struct termios Otty, Ntty;
22
23
24 tcgetattr( 0, &Otty);
25 Ntty = Otty;
26
27 Ntty.c_iflag = 0; /* input mode */
28 Ntty.c_oflag = 0; /* output mode */
29 Ntty.c_lflag &= ~ICANON; /* raw mode */
30 Ntty.c_cc[VMIN] = CMIN; /* minimum time to wait */
31 Ntty.c_cc[VTIME] = CTIME; /* minimum characters to wait for */
32
33 if (0 == (error = tcsetattr(0, TCSANOW, &Ntty))) {
34 error += ioctl(0, FIONREAD, &cnt);
35 error += tcsetattr(0, TCSANOW, &Otty);
36 }
37
38 return ( error == 0 ? cnt : -1 );
39}
40
41#else
42#include <conio.h>
43int ukbhit(void) {
44 return kbhit();
45}
46#endif
47
55acbb2a 48// log files functions
b915fda3 49void AddLogLine(char *file, char *extData, char *c) {
55acbb2a 50 FILE *fLog = NULL;
b915fda3 51 char filename[FILE_PATH_SIZE] = {0x00};
52 int len = 0;
53
54 len = strlen(file);
55 if (len > FILE_PATH_SIZE) len = FILE_PATH_SIZE;
56 memcpy(filename, file, len);
57
58 fLog = fopen(filename, "a");
55acbb2a 59 if (!fLog) {
b915fda3 60 printf("Could not append log file %s", filename);
55acbb2a
M
61 return;
62 }
63
64 fprintf(fLog, "%s", extData);
65 fprintf(fLog, "%s\n", c);
66 fclose(fLog);
67}
68
69void AddLogHex(char *fileName, char *extData, const uint8_t * data, const size_t len){
70 AddLogLine(fileName, extData, sprint_hex(data, len));
71}
72
73void AddLogUint64(char *fileName, char *extData, const uint64_t data) {
74 char buf[100] = {0};
75 sprintf(buf, "%x%x", (unsigned int)((data & 0xFFFFFFFF00000000) >> 32), (unsigned int)(data & 0xFFFFFFFF));
76 AddLogLine(fileName, extData, buf);
77}
78
79void AddLogCurrentDT(char *fileName) {
80 char buff[20];
81 struct tm *curTime;
82
83 time_t now = time(0);
84 curTime = gmtime(&now);
85
86 strftime (buff, sizeof(buff), "%Y-%m-%d %H:%M:%S", curTime);
87 AddLogLine(fileName, "\nanticollision: ", buff);
88}
89
e0c635d1 90void FillFileNameByUID(char *fileName, uint8_t * uid, char *ext, int byteCount) {
55acbb2a
M
91 char * fnameptr = fileName;
92 memset(fileName, 0x00, 200);
93
e0c635d1 94 for (int j = 0; j < byteCount; j++, fnameptr += 2)
55acbb2a
M
95 sprintf(fnameptr, "%02x", uid[j]);
96 sprintf(fnameptr, "%s", ext);
55acbb2a
M
97}
98
99// printing and converting functions
f397b5cc 100
534983d7 101void print_hex(const uint8_t * data, const size_t len)
102{
103 size_t i;
104
105 for (i=0; i < len; i++)
106 printf("%02x ", data[i]);
107
108 printf("\n");
109}
110
111char * sprint_hex(const uint8_t * data, const size_t len) {
b915fda3 112
113 int maxLen = ( len > 1024/3) ? 1024/3 : len;
534983d7 114 static char buf[1024];
115 char * tmp = buf;
116 size_t i;
117
b915fda3 118 for (i=0; i < maxLen; ++i, tmp += 3)
534983d7 119 sprintf(tmp, "%02x ", data[i]);
120
121 return buf;
122}
f89c7050 123
79544b28 124char * sprint_bin(const uint8_t * data, const size_t len) {
125
126 int maxLen = ( len > 1024) ? 1024 : len;
127 static char buf[1024];
128 char * tmp = buf;
129 size_t i;
130
131 for (i=0; i < maxLen; ++i, ++tmp)
132 sprintf(tmp, "%u", data[i]);
133
134 return buf;
135}
136
f89c7050
M
137void num_to_bytes(uint64_t n, size_t len, uint8_t* dest)
138{
139 while (len--) {
140 dest[len] = (uint8_t) n;
141 n >>= 8;
142 }
143}
144
145uint64_t bytes_to_num(uint8_t* src, size_t len)
146{
147 uint64_t num = 0;
148 while (len--)
149 {
150 num = (num << 8) | (*src);
151 src++;
152 }
153 return num;
154}
f397b5cc 155
79544b28 156//assumes little endian
157char * printBits(size_t const size, void const * const ptr)
158{
159 unsigned char *b = (unsigned char*) ptr;
160 unsigned char byte;
161 static char buf[1024];
162 char * tmp = buf;
163 int i, j;
164
165 for (i=size-1;i>=0;i--)
166 {
167 for (j=7;j>=0;j--)
168 {
169 byte = b[i] & (1<<j);
170 byte >>= j;
171 sprintf(tmp, "%u", byte);
172 tmp++;
173 }
174 }
175 return buf;
176}
177
9ca155ba
M
178// -------------------------------------------------------------------------
179// string parameters lib
180// -------------------------------------------------------------------------
181
f397b5cc
M
182// -------------------------------------------------------------------------
183// line - param line
184// bg, en - symbol numbers in param line of beginning an ending parameter
185// paramnum - param number (from 0)
186// -------------------------------------------------------------------------
187int param_getptr(const char *line, int *bg, int *en, int paramnum)
188{
189 int i;
190 int len = strlen(line);
191
192 *bg = 0;
193 *en = 0;
194
195 // skip spaces
196 while (line[*bg] ==' ' || line[*bg]=='\t') (*bg)++;
197 if (*bg >= len) {
198 return 1;
199 }
200
201 for (i = 0; i < paramnum; i++) {
202 while (line[*bg]!=' ' && line[*bg]!='\t' && line[*bg] != '\0') (*bg)++;
203 while (line[*bg]==' ' || line[*bg]=='\t') (*bg)++;
204
205 if (line[*bg] == '\0') return 1;
206 }
207
208 *en = *bg;
209 while (line[*en] != ' ' && line[*en] != '\t' && line[*en] != '\0') (*en)++;
210
211 (*en)--;
212
213 return 0;
214}
215
31abe49f 216
f397b5cc
M
217char param_getchar(const char *line, int paramnum)
218{
219 int bg, en;
220
221 if (param_getptr(line, &bg, &en, paramnum)) return 0x00;
222
223 return line[bg];
224}
225
226uint8_t param_get8(const char *line, int paramnum)
227{
228 return param_get8ex(line, paramnum, 10, 0);
229}
230
f6d9fb17 231/**
31abe49f 232 * @brief Reads a decimal integer (actually, 0-254, not 255)
f6d9fb17
MHS
233 * @param line
234 * @param paramnum
31abe49f 235 * @return -1 if error
f6d9fb17
MHS
236 */
237uint8_t param_getdec(const char *line, int paramnum, uint8_t *destination)
238{
31abe49f 239 uint8_t val = param_get8ex(line, paramnum, 255, 10);
31abe49f 240 if( (int8_t) val == -1) return 1;
f6d9fb17
MHS
241 (*destination) = val;
242 return 0;
243}
244/**
245 * @brief Checks if param is decimal
246 * @param line
247 * @param paramnum
248 * @return
249 */
250uint8_t param_isdec(const char *line, int paramnum)
251{
252 int bg, en;
253 //TODO, check more thorougly
254 if (!param_getptr(line, &bg, &en, paramnum)) return 1;
255 // return strtoul(&line[bg], NULL, 10) & 0xff;
256
257 return 0;
258}
259
f397b5cc
M
260uint8_t param_get8ex(const char *line, int paramnum, int deflt, int base)
261{
262 int bg, en;
263
264 if (!param_getptr(line, &bg, &en, paramnum))
6c6d1ac1 265 return strtoul(&line[bg], NULL, base) & 0xff;
f397b5cc
M
266 else
267 return deflt;
268}
269
270uint32_t param_get32ex(const char *line, int paramnum, int deflt, int base)
271{
272 int bg, en;
273
274 if (!param_getptr(line, &bg, &en, paramnum))
6c6d1ac1 275 return strtoul(&line[bg], NULL, base);
f397b5cc
M
276 else
277 return deflt;
278}
279
280uint64_t param_get64ex(const char *line, int paramnum, int deflt, int base)
281{
282 int bg, en;
283
284 if (!param_getptr(line, &bg, &en, paramnum))
6c6d1ac1 285 return strtoull(&line[bg], NULL, base);
f397b5cc
M
286 else
287 return deflt;
288
289 return 0;
290}
291
292int param_gethex(const char *line, int paramnum, uint8_t * data, int hexcnt)
293{
294 int bg, en, temp, i;
295
296 if (hexcnt % 2)
297 return 1;
298
299 if (param_getptr(line, &bg, &en, paramnum)) return 1;
300
301 if (en - bg + 1 != hexcnt)
302 return 1;
303
304 for(i = 0; i < hexcnt; i += 2) {
305 if (!(isxdigit(line[bg + i]) && isxdigit(line[bg + i + 1])) ) return 1;
306
307 sscanf((char[]){line[bg + i], line[bg + i + 1], 0}, "%X", &temp);
308 data[i / 2] = temp & 0xff;
309 }
310
311 return 0;
312}
aea4d766 313
314int param_getstr(const char *line, int paramnum, char * str)
315{
316 int bg, en;
317
318 if (param_getptr(line, &bg, &en, paramnum)) return 0;
319
320 memcpy(str, line + bg, en - bg + 1);
321 str[en - bg + 1] = 0;
322
323 return en - bg + 1;
324}
79544b28 325
326/*
327The following methods comes from Rfidler sourcecode.
328https://github.com/ApertureLabsLtd/RFIDler/blob/master/firmware/Pic32/RFIDler.X/src/
329*/
330
331// convert hex to sequence of 0/1 bit values
332// returns number of bits converted
333int hextobinarray(char *target, char *source)
334{
335 int length, i, count= 0;
336 char x;
337
338 length = strlen(source);
339 // process 4 bits (1 hex digit) at a time
340 while(length--)
341 {
342 x= *(source++);
343 // capitalize
344 if (x >= 'a' && x <= 'f')
345 x -= 32;
346 // convert to numeric value
347 if (x >= '0' && x <= '9')
348 x -= '0';
349 else if (x >= 'A' && x <= 'F')
350 x -= 'A' - 10;
351 else
352 return 0;
353 // output
354 for(i= 0 ; i < 4 ; ++i, ++count)
355 *(target++)= (x >> (3 - i)) & 1;
356 }
357
358 return count;
359}
360
361// convert hex to human readable binary string
362int hextobinstring(char *target, char *source)
363{
364 int length;
365
366 if(!(length= hextobinarray(target, source)))
367 return 0;
368 binarraytobinstring(target, target, length);
369 return length;
370}
371
372// convert binary array of 0x00/0x01 values to hex (safe to do in place as target will always be shorter than source)
373// return number of bits converted
374int binarraytohex(char *target, char *source, int length)
375{
376 unsigned char i, x;
377 int j = length;
378
379 if(j % 4)
380 return 0;
381
382 while(j)
383 {
384 for(i= x= 0 ; i < 4 ; ++i)
385 x += ( source[i] << (3 - i));
386 sprintf(target,"%X", x);
387 ++target;
388 source += 4;
389 j -= 4;
390 }
391 return length;
392}
393
394// convert binary array to human readable binary
395void binarraytobinstring(char *target, char *source, int length)
396{
397 int i;
398
399 for(i= 0 ; i < length ; ++i)
400 *(target++)= *(source++) + '0';
401 *target= '\0';
402}
403
404// return parity bit required to match type
405uint8_t GetParity( char *bits, uint8_t type, int length)
406{
407 int x;
408
409 for(x= 0 ; length > 0 ; --length)
410 x += bits[length - 1];
411 x %= 2;
412
413 return x ^ type;
414}
415
416// add HID parity to binary array: EVEN prefix for 1st half of ID, ODD suffix for 2nd half
417void wiegand_add_parity(char *target, char *source, char length)
418{
419 *(target++)= GetParity(source, EVEN, length / 2);
420 memcpy(target, source, length);
421 target += length;
422 *(target)= GetParity(source + length / 2, ODD, length / 2);
423}
Impressum, Datenschutz