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

   1 //-----------------------------------------------------------------------------
   2 // Utility functions used in many places, not specific to any piece of code.
   3 // Jonathan Westhues, Sept 2005
   4 //-----------------------------------------------------------------------------
   5 #include <proxmark3.h>
   6 #include "apps.h"
   7
   8 void *memcpy(void *dest, const void *src, int len)
   9 {
  10         BYTE *d = dest;
  11         const BYTE *s = src;
  12         while((len--) > 0) {
  13                 *d = *s;
  14                 d++;
  15                 s++;
  16         }
  17         return dest;
  18 }
  19
  20 void *memset(void *dest, int c, int len)
  21 {
  22         BYTE *d = dest;
  23         while((len--) > 0) {
  24                 *d = c;
  25                 d++;
  26         }
  27         return dest;
  28 }
  29
  30 int memcmp(const void *av, const void *bv, int len)
  31 {
  32         const BYTE *a = av;
  33         const BYTE *b = bv;
  34
  35         while((len--) > 0) {
  36                 if(*a != *b) {
  37                         return *a - *b;
  38                 }
  39                 a++;
  40                 b++;
  41         }
  42         return 0;
  43 }
  44
  45 int strlen(char *str)
  46 {
  47         int l = 0;
  48         while(*str) {
  49                 l++;
  50                 str++;
  51         }
  52         return l;
  53 }
  54
  55 char* strncat(char *dest, const char *src, unsigned int n)
  56 {
  57         unsigned int dest_len = strlen(dest);
  58         unsigned int i;
  59
  60         for (i = 0 ; i < n && src[i] != '\0' ; i++)
  61                 dest[dest_len + i] = src[i];
  62         dest[dest_len + i] = '\0';
  63
  64         return dest;
  65 }
  66
  67
  68 void LEDsoff()
  69 {
  70         LED_A_OFF();
  71         LED_B_OFF();
  72         LED_C_OFF();
  73         LED_D_OFF();
  74 }
  75
  76 // LEDs: R(C) O(A) G(B) -- R(D) [1, 2, 4 and 8]
  77 void LED(int led, int ms)
  78 {
  79         if (led & LED_RED)
  80                 LED_C_ON();
  81         if (led & LED_ORANGE)
  82                 LED_A_ON();
  83         if (led & LED_GREEN)
  84                 LED_B_ON();
  85         if (led & LED_RED2)
  86                 LED_D_ON();
  87
  88         if (!ms)
  89                 return;
  90
  91         SpinDelay(ms);
  92
  93         if (led & LED_RED)
  94                 LED_C_OFF();
  95         if (led & LED_ORANGE)
  96                 LED_A_OFF();
  97         if (led & LED_GREEN)
  98                 LED_B_OFF();
  99         if (led & LED_RED2)
 100                 LED_D_OFF();
 101 }
 102
 103
 104 // Determine if a button is double clicked, single clicked,
 105 // not clicked, or held down (for ms || 1sec)
 106 // In general, don't use this function unless you expect a
 107 // double click, otherwise it will waste 500ms -- use BUTTON_HELD instead
 108 int BUTTON_CLICKED(int ms)
 109 {
 110         // Up to 500ms in between clicks to mean a double click
 111         int ticks = (48000 * (ms ? ms : 1000)) >> 10;
 112
 113         // If we're not even pressed, forget about it!
 114         if (!BUTTON_PRESS())
 115                 return BUTTON_NO_CLICK;
 116
 117         // Borrow a PWM unit for my real-time clock
 118         PWM_ENABLE = PWM_CHANNEL(0);
 119         // 48 MHz / 1024 gives 46.875 kHz
 120         PWM_CH_MODE(0) = PWM_CH_MODE_PRESCALER(10);
 121         PWM_CH_DUTY_CYCLE(0) = 0;
 122         PWM_CH_PERIOD(0) = 0xffff;
 123
 124         WORD start = (WORD)PWM_CH_COUNTER(0);
 125
 126         int letoff = 0;
 127         for(;;)
 128         {
 129                 WORD now = (WORD)PWM_CH_COUNTER(0);
 130
 131                 // We haven't let off the button yet
 132                 if (!letoff)
 133                 {
 134                         // We just let it off!
 135                         if (!BUTTON_PRESS())
 136                         {
 137                                 letoff = 1;
 138
 139                                 // reset our timer for 500ms
 140                                 start = (WORD)PWM_CH_COUNTER(0);
 141                                 ticks = (48000 * (500)) >> 10;
 142                         }
 143
 144                         // Still haven't let it off
 145                         else
 146                                 // Have we held down a full second?
 147                                 if (now == (WORD)(start + ticks))
 148                                         return BUTTON_HOLD;
 149                 }
 150
 151                 // We already let off, did we click again?
 152                 else
 153                         // Sweet, double click!
 154                         if (BUTTON_PRESS())
 155                                 return BUTTON_DOUBLE_CLICK;
 156
 157                         // Have we ran out of time to double click?
 158                         else
 159                                 if (now == (WORD)(start + ticks))
 160                                         // At least we did a single click
 161                                         return BUTTON_SINGLE_CLICK;
 162
 163                 WDT_HIT();
 164         }
 165
 166         // We should never get here
 167         return BUTTON_ERROR;
 168 }
 169
 170 // Determine if a button is held down
 171 int BUTTON_HELD(int ms)
 172 {
 173         // If button is held for one second
 174         int ticks = (48000 * (ms ? ms : 1000)) >> 10;
 175
 176         // If we're not even pressed, forget about it!
 177         if (!BUTTON_PRESS())
 178                 return BUTTON_NO_CLICK;
 179
 180         // Borrow a PWM unit for my real-time clock
 181         PWM_ENABLE = PWM_CHANNEL(0);
 182         // 48 MHz / 1024 gives 46.875 kHz
 183         PWM_CH_MODE(0) = PWM_CH_MODE_PRESCALER(10);
 184         PWM_CH_DUTY_CYCLE(0) = 0;
 185         PWM_CH_PERIOD(0) = 0xffff;
 186
 187         WORD start = (WORD)PWM_CH_COUNTER(0);
 188
 189         for(;;)
 190         {
 191                 WORD now = (WORD)PWM_CH_COUNTER(0);
 192
 193                 // As soon as our button let go, we didn't hold long enough
 194                 if (!BUTTON_PRESS())
 195                         return BUTTON_SINGLE_CLICK;
 196
 197                 // Have we waited the full second?
 198                 else
 199                         if (now == (WORD)(start + ticks))
 200                                 return BUTTON_HOLD;
 201
 202                 WDT_HIT();
 203         }
 204
 205         // We should never get here
 206         return BUTTON_ERROR;
 207 }
 208
 209 // attempt at high resolution microsecond timer
 210 // beware: timer counts in 21.3uS increments (1024/48Mhz)
 211 void SpinDelayUs(int us)
 212 {
 213         int ticks = (48*us) >> 10;
 214
 215         // Borrow a PWM unit for my real-time clock
 216         PWM_ENABLE = PWM_CHANNEL(0);
 217         // 48 MHz / 1024 gives 46.875 kHz
 218         PWM_CH_MODE(0) = PWM_CH_MODE_PRESCALER(10);
 219         PWM_CH_DUTY_CYCLE(0) = 0;
 220         PWM_CH_PERIOD(0) = 0xffff;
 221
 222         WORD start = (WORD)PWM_CH_COUNTER(0);
 223
 224         for(;;) {
 225                 WORD now = (WORD)PWM_CH_COUNTER(0);
 226                 if (now == (WORD)(start + ticks))
 227                         return;
 228
 229                 WDT_HIT();
 230         }
 231 }
 232
 233 void SpinDelay(int ms)
 234 {
 235   // convert to uS and call microsecond delay function
 236         SpinDelayUs(ms*1000);
 237 }