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