git.zerfleddert.de Git - proxmark3-svn/blob - armsrc/util.c

   1 //-----------------------------------------------------------------------------
   2 // Jonathan Westhues, Sept 2005
   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 // Utility functions used in many places, not specific to any piece of code.
   9 //-----------------------------------------------------------------------------
  10
  11 #include "proxmark3.h"
  12 #include "util.h"
  13 #include "string.h"
  14
  15 size_t nbytes(size_t nbits) {
  16         return (nbits/8)+((nbits%8)>0);
  17 }
  18
  19 uint32_t SwapBits(uint32_t value, int nrbits) {
  20         int i;
  21         uint32_t newvalue = 0;
  22         for(i = 0; i < nrbits; i++) {
  23                 newvalue ^= ((value >> i) & 1) << (nrbits - 1 - i);
  24         }
  25         return newvalue;
  26 }
  27
  28 void num_to_bytes(uint64_t n, size_t len, uint8_t* dest)
  29 {
  30         while (len--) {
  31                 dest[len] = (uint8_t) n;
  32                 n >>= 8;
  33         }
  34 }
  35
  36 uint64_t bytes_to_num(uint8_t* src, size_t len)
  37 {
  38         uint64_t num = 0;
  39         while (len--)
  40         {
  41                 num = (num << 8) | (*src);
  42                 src++;
  43         }
  44         return num;
  45 }
  46
  47 void LEDsoff()
  48 {
  49         LED_A_OFF();
  50         LED_B_OFF();
  51         LED_C_OFF();
  52         LED_D_OFF();
  53 }
  54
  55 // LEDs: R(C) O(A) G(B) -- R(D) [1, 2, 4 and 8]
  56 void LED(int led, int ms)
  57 {
  58         if (led & LED_RED)
  59                 LED_C_ON();
  60         if (led & LED_ORANGE)
  61                 LED_A_ON();
  62         if (led & LED_GREEN)
  63                 LED_B_ON();
  64         if (led & LED_RED2)
  65                 LED_D_ON();
  66
  67         if (!ms)
  68                 return;
  69
  70         SpinDelay(ms);
  71
  72         if (led & LED_RED)
  73                 LED_C_OFF();
  74         if (led & LED_ORANGE)
  75                 LED_A_OFF();
  76         if (led & LED_GREEN)
  77                 LED_B_OFF();
  78         if (led & LED_RED2)
  79                 LED_D_OFF();
  80 }
  81
  82
  83 // Determine if a button is double clicked, single clicked,
  84 // not clicked, or held down (for ms || 1sec)
  85 // In general, don't use this function unless you expect a
  86 // double click, otherwise it will waste 500ms -- use BUTTON_HELD instead
  87 int BUTTON_CLICKED(int ms)
  88 {
  89         // Up to 500ms in between clicks to mean a double click
  90         int ticks = (48000 * (ms ? ms : 1000)) >> 10;
  91
  92         // If we're not even pressed, forget about it!
  93         if (!BUTTON_PRESS())
  94                 return BUTTON_NO_CLICK;
  95
  96         // Borrow a PWM unit for my real-time clock
  97         AT91C_BASE_PWMC->PWMC_ENA = PWM_CHANNEL(0);
  98         // 48 MHz / 1024 gives 46.875 kHz
  99         AT91C_BASE_PWMC_CH0->PWMC_CMR = PWM_CH_MODE_PRESCALER(10);
 100         AT91C_BASE_PWMC_CH0->PWMC_CDTYR = 0;
 101         AT91C_BASE_PWMC_CH0->PWMC_CPRDR = 0xffff;
 102
 103         uint16_t start = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 104
 105         int letoff = 0;
 106         for(;;)
 107         {
 108                 uint16_t now = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 109
 110                 // We haven't let off the button yet
 111                 if (!letoff)
 112                 {
 113                         // We just let it off!
 114                         if (!BUTTON_PRESS())
 115                         {
 116                                 letoff = 1;
 117
 118                                 // reset our timer for 500ms
 119                                 start = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 120                                 ticks = (48000 * (500)) >> 10;
 121                         }
 122
 123                         // Still haven't let it off
 124                         else
 125                                 // Have we held down a full second?
 126                                 if (now == (uint16_t)(start + ticks))
 127                                         return BUTTON_HOLD;
 128                 }
 129
 130                 // We already let off, did we click again?
 131                 else
 132                         // Sweet, double click!
 133                         if (BUTTON_PRESS())
 134                                 return BUTTON_DOUBLE_CLICK;
 135
 136                         // Have we ran out of time to double click?
 137                         else
 138                                 if (now == (uint16_t)(start + ticks))
 139                                         // At least we did a single click
 140                                         return BUTTON_SINGLE_CLICK;
 141
 142                 WDT_HIT();
 143         }
 144
 145         // We should never get here
 146         return BUTTON_ERROR;
 147 }
 148
 149 // Determine if a button is held down
 150 int BUTTON_HELD(int ms)
 151 {
 152         // If button is held for one second
 153         int ticks = (48000 * (ms ? ms : 1000)) >> 10;
 154
 155         // If we're not even pressed, forget about it!
 156         if (!BUTTON_PRESS())
 157                 return BUTTON_NO_CLICK;
 158
 159         // Borrow a PWM unit for my real-time clock
 160         AT91C_BASE_PWMC->PWMC_ENA = PWM_CHANNEL(0);
 161         // 48 MHz / 1024 gives 46.875 kHz
 162         AT91C_BASE_PWMC_CH0->PWMC_CMR = PWM_CH_MODE_PRESCALER(10);
 163         AT91C_BASE_PWMC_CH0->PWMC_CDTYR = 0;
 164         AT91C_BASE_PWMC_CH0->PWMC_CPRDR = 0xffff;
 165
 166         uint16_t start = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 167
 168         for(;;)
 169         {
 170                 uint16_t now = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 171
 172                 // As soon as our button let go, we didn't hold long enough
 173                 if (!BUTTON_PRESS())
 174                         return BUTTON_SINGLE_CLICK;
 175
 176                 // Have we waited the full second?
 177                 else
 178                         if (now == (uint16_t)(start + ticks))
 179                                 return BUTTON_HOLD;
 180
 181                 WDT_HIT();
 182         }
 183
 184         // We should never get here
 185         return BUTTON_ERROR;
 186 }
 187
 188 // attempt at high resolution microsecond timer
 189 // beware: timer counts in 21.3uS increments (1024/48Mhz)
 190 void SpinDelayUs(int us)
 191 {
 192         int ticks = (48*us) >> 10;
 193
 194         // Borrow a PWM unit for my real-time clock
 195         AT91C_BASE_PWMC->PWMC_ENA = PWM_CHANNEL(0);
 196         // 48 MHz / 1024 gives 46.875 kHz
 197         AT91C_BASE_PWMC_CH0->PWMC_CMR = PWM_CH_MODE_PRESCALER(10);
 198         AT91C_BASE_PWMC_CH0->PWMC_CDTYR = 0;
 199         AT91C_BASE_PWMC_CH0->PWMC_CPRDR = 0xffff;
 200
 201         uint16_t start = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 202
 203         for(;;) {
 204                 uint16_t now = AT91C_BASE_PWMC_CH0->PWMC_CCNTR;
 205                 if (now == (uint16_t)(start + ticks))
 206                         return;
 207
 208                 WDT_HIT();
 209         }
 210 }
 211
 212 void SpinDelay(int ms)
 213 {
 214   // convert to uS and call microsecond delay function
 215         SpinDelayUs(ms*1000);
 216 }
 217
 218 /* Similar to FpgaGatherVersion this formats stored version information
 219  * into a string representation. It takes a pointer to the struct version_information,
 220  * verifies the magic properties, then stores a formatted string, prefixed by
 221  * prefix in dst.
 222  */
 223 void FormatVersionInformation(char *dst, int len, const char *prefix, void *version_information)
 224 {
 225         struct version_information *v = (struct version_information*)version_information;
 226         dst[0] = 0;
 227         strncat(dst, prefix, len);
 228         if(v->magic != VERSION_INFORMATION_MAGIC) {
 229                 strncat(dst, "Missing/Invalid version information", len);
 230                 return;
 231         }
 232         if(v->versionversion != 1) {
 233                 strncat(dst, "Version information not understood", len);
 234                 return;
 235         }
 236         if(!v->present) {
 237                 strncat(dst, "Version information not available", len);
 238                 return;
 239         }
 240
 241         strncat(dst, v->svnversion, len);
 242         if(v->clean == 0) {
 243                 strncat(dst, "-unclean", len);
 244         } else if(v->clean == 2) {
 245                 strncat(dst, "-suspect", len);
 246         }
 247
 248         strncat(dst, " ", len);
 249         strncat(dst, v->buildtime, len);
 250 }
 251
 252 //  -------------------------------------------------------------------------
 253 //  timer lib
 254 //  -------------------------------------------------------------------------
 255 //  test procedure:
 256 //
 257 //      ti = GetTickCount();
 258 //      SpinDelay(1000);
 259 //      ti = GetTickCount() - ti;
 260 //      Dbprintf("timer(1s): %d t=%d", ti, GetTickCount());
 261
 262 void StartTickCount()
 263 {
 264 //  must be 0x40, but on my cpu - included divider is optimal
 265 //  0x20 - 1 ms / bit
 266 //  0x40 - 2 ms / bit
 267
 268         AT91C_BASE_RTTC->RTTC_RTMR = AT91C_RTTC_RTTRST + 0x001D; // was 0x003B
 269 }
 270
 271 /*
 272 * Get the current count.
 273 */
 274 uint32_t RAMFUNC GetTickCount(){
 275         return AT91C_BASE_RTTC->RTTC_RTVR;// was * 2;
 276 }
 277
 278 //  -------------------------------------------------------------------------
 279 //  microseconds timer
 280 //  -------------------------------------------------------------------------
 281 void StartCountUS()
 282 {
 283         AT91C_BASE_PMC->PMC_PCER |= (0x1 << 12) | (0x1 << 13) | (0x1 << 14);
 284 //      AT91C_BASE_TCB->TCB_BMR = AT91C_TCB_TC1XC1S_TIOA0;
 285         AT91C_BASE_TCB->TCB_BMR = AT91C_TCB_TC0XC0S_NONE | AT91C_TCB_TC1XC1S_TIOA0 | AT91C_TCB_TC2XC2S_NONE;
 286
 287         // fast clock
 288         AT91C_BASE_TC0->TC_CCR = AT91C_TC_CLKDIS; // timer disable
 289         AT91C_BASE_TC0->TC_CMR = AT91C_TC_CLKS_TIMER_DIV3_CLOCK | // MCK(48MHz)/32 -- tick=1.5mks
 290                                                                                                                 AT91C_TC_WAVE | AT91C_TC_WAVESEL_UP_AUTO | AT91C_TC_ACPA_CLEAR |
 291                                                                                                                 AT91C_TC_ACPC_SET | AT91C_TC_ASWTRG_SET;
 292         AT91C_BASE_TC0->TC_RA = 1;
 293         AT91C_BASE_TC0->TC_RC = 0xBFFF + 1; // 0xC000
 294
 295         AT91C_BASE_TC1->TC_CCR = AT91C_TC_CLKDIS; // timer disable
 296         AT91C_BASE_TC1->TC_CMR = AT91C_TC_CLKS_XC1; // from timer 0
 297
 298         AT91C_BASE_TC0->TC_CCR = AT91C_TC_CLKEN;
 299         AT91C_BASE_TC1->TC_CCR = AT91C_TC_CLKEN;
 300         AT91C_BASE_TCB->TCB_BCR = 1;
 301         }
 302
 303 uint32_t RAMFUNC GetCountUS(){
 304         return (AT91C_BASE_TC1->TC_CV * 0x8000) + ((AT91C_BASE_TC0->TC_CV / 15) * 10);
 305 }
 306
 307 static uint32_t GlobalUsCounter = 0;
 308
 309 uint32_t RAMFUNC GetDeltaCountUS(){
 310         uint32_t g_cnt = GetCountUS();
 311         uint32_t g_res = g_cnt - GlobalUsCounter;
 312         GlobalUsCounter = g_cnt;
 313         return g_res;
 314 }
 315
 316
 317 //  -------------------------------------------------------------------------
 318 //  Mifare timer. Uses ssp_clk from FPGA
 319 //  -------------------------------------------------------------------------
 320 void StartCountMifare()
 321 {
 322         AT91C_BASE_PMC->PMC_PCER = (1 << AT91C_ID_TC0) | (1 << AT91C_ID_TC1) | (1 << AT91C_ID_TC2);  // Enable Clock to all timers
 323         AT91C_BASE_TCB->TCB_BMR = AT91C_TCB_TC0XC0S_TIOA1               // XC0 Clock = TIOA1
 324                                                         | AT91C_TCB_TC1XC1S_NONE                // XC1 Clock = none
 325                                                         | AT91C_TCB_TC2XC2S_TIOA0;              // XC2 Clock = TIOA0
 326
 327         // configure TC1 to create a short pulse on TIOA1 when a rising edge on TIOB1 (= ssp_clk from FPGA) occurs:
 328         AT91C_BASE_TC1->TC_CCR = AT91C_TC_CLKDIS;                               // disable TC1
 329         AT91C_BASE_TC1->TC_CMR = AT91C_TC_CLKS_TIMER_DIV1_CLOCK // TC1 Clock = MCK(48MHz)/2 = 24MHz
 330                                                         | AT91C_TC_CPCSTOP                              // Stop clock on RC compare
 331                                                         | AT91C_TC_EEVTEDG_RISING               // Trigger on rising edge of Event
 332                                                         | AT91C_TC_EEVT_TIOB                    // Event-Source: TIOB1 (= ssc_clk from FPGA = 13,56MHz / 16)
 333                                                         | AT91C_TC_ENETRG                               // Enable external trigger event
 334                                                         | AT91C_TC_WAVESEL_UP                   // Upmode without automatic trigger on RC compare
 335                                                         | AT91C_TC_WAVE                                 // Waveform Mode
 336                                                         | AT91C_TC_AEEVT_SET                    // Set TIOA1 on external event
 337                                                         | AT91C_TC_ACPC_CLEAR;                  // Clear TIOA1 on RC Compare
 338         AT91C_BASE_TC1->TC_RC = 0x04;                                                   // RC Compare value = 0x04
 339
 340         // use TC0 to count TIOA1 pulses
 341         AT91C_BASE_TC0->TC_CCR = AT91C_TC_CLKDIS;                               // disable TC0
 342         AT91C_BASE_TC0->TC_CMR = AT91C_TC_CLKS_XC0                              // TC0 clock = XC0 clock = TIOA1
 343                                                         | AT91C_TC_WAVE                                 // Waveform Mode
 344                                                         | AT91C_TC_WAVESEL_UP                   // just count
 345                                                         | AT91C_TC_ACPA_CLEAR                   // Clear TIOA0 on RA Compare
 346                                                         | AT91C_TC_ACPC_SET;                    // Set TIOA0 on RC Compare
 347         AT91C_BASE_TC0->TC_RA = 1;                                                              // RA Compare value = 1; pulse width to TC2
 348         AT91C_BASE_TC0->TC_RC = 0;                                                              // RC Compare value = 0; increment TC2 on overflow
 349
 350         // use TC2 to count TIOA0 pulses (giving us a 32bit counter (TC0/TC2) clocked by ssp_clk)
 351         AT91C_BASE_TC2->TC_CCR = AT91C_TC_CLKDIS;                               // disable TC2
 352         AT91C_BASE_TC2->TC_CMR = AT91C_TC_CLKS_XC2                              // TC2 clock = XC2 clock = TIOA0
 353                                                         | AT91C_TC_WAVE                                 // Waveform Mode
 354                                                         | AT91C_TC_WAVESEL_UP;                  // just count
 355
 356
 357         AT91C_BASE_TC0->TC_CCR = AT91C_TC_CLKEN;                                // enable TC0
 358         AT91C_BASE_TC1->TC_CCR = AT91C_TC_CLKEN;                                // enable TC1
 359         AT91C_BASE_TC2->TC_CCR = AT91C_TC_CLKEN;                                // enable TC2
 360         AT91C_BASE_TCB->TCB_BCR = 1;                                                    // assert Sync (set all timers to 0 on next active clock edge)
 361 }
 362
 363
 364 uint32_t RAMFUNC GetCountMifare(){
 365         uint32_t tmp_count;
 366         tmp_count = (AT91C_BASE_TC2->TC_CV << 16) | AT91C_BASE_TC0->TC_CV;
 367         if ((tmp_count & 0xffff) == 0) { //small chance that we may have missed an increment in TC2
 368                 return (AT91C_BASE_TC2->TC_CV << 16);
 369         }
 370         else {
 371                 return tmp_count;
 372         }
 373 }