tools/hitag2crack/crack2/ht2crack2gentest.c

   1 /*
   2  * ht2crack2gentests.c
   3  * this uses the RFIDler hitag2 PRNG code to generate test cases to test the tables
   4  */
   5
   6 #include "ht2crackutils.h"
   7
   8 static int makerandom(char *hex, unsigned int len, int fd) {
   9     unsigned char raw[32];
  10     int i;
  11
  12     if (!hex) {
  13         printf("makerandom: hex is NULL\n");
  14         exit(1);
  15     }
  16
  17     if (!len || (len > 32)) {
  18         printf("makerandom: len must be between 1 and 32 inclusive\n");
  19         exit(1);
  20     }
  21
  22     if (read(fd, raw, len) != len) {
  23         printf("makerandom: cannot read random bytes\n");
  24         exit(1);
  25     }
  26
  27     for (i = 0; i < len; i++) {
  28         sprintf(hex + (2 * i), "%02X", raw[i]);
  29     }
  30
  31     return 1;
  32 }
  33
  34
  35 int main(int argc, char *argv[]) {
  36     Hitag_State hstate;
  37     char key[32];
  38     char uid[32];
  39     char nR[32];
  40     char filename[256];
  41     int i, j;
  42     int numtests;
  43     int urandomfd;
  44
  45     if (argc < 2) {
  46         printf("%s number\n", argv[0]);
  47         exit(1);
  48     }
  49
  50     numtests = atoi(argv[1]);
  51     if (numtests <= 0) {
  52         printf("need positive number of tests\n");
  53         exit(1);
  54     }
  55
  56     urandomfd = open("/dev/urandom", O_RDONLY);
  57     if (urandomfd <= 0) {
  58         printf("cannot open /dev/urandom\n");
  59         exit(1);
  60     }
  61
  62
  63     for (i = 0; i < numtests; i++) {
  64
  65         makerandom(key, 6, urandomfd);
  66         makerandom(uid, 4, urandomfd);
  67         makerandom(nR, 4, urandomfd);
  68         sprintf(filename, "keystream.key-%s.uid-%s.nR-%s", key, uid, nR);
  69
  70         FILE *fp = fopen(filename, "w");
  71         if (!fp) {
  72             printf("cannot open file '%s' for writing\n", filename);
  73             exit(1);
  74         }
  75
  76         hstate.shiftreg = 0;
  77         hstate.lfsr = 0;
  78
  79         hitag2_init(&hstate, rev64(hexreversetoulonglong(key)), rev32(hexreversetoulong(uid)), rev32(hexreversetoulong(nR)));
  80
  81         hitag2_nstep(&hstate, 64);
  82
  83         for (j = 0; j < 64; j++) {
  84             fprintf(fp, "%08X\n", hitag2_nstep(&hstate, 32));
  85         }
  86
  87         fclose(fp);
  88     }
  89     return 0;
  90 }
  91
  92