353 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			353 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #include <furi.h>
 | |
| #include <gui/gui.h>
 | |
| 
 | |
| typedef enum { EventTypeTick, EventTypeKey, EventTypeRx } EventType;
 | |
| 
 | |
| typedef struct {
 | |
|     bool value;
 | |
|     uint32_t dwt_value;
 | |
| } RxEvent;
 | |
| 
 | |
| typedef struct {
 | |
|     union {
 | |
|         InputEvent input;
 | |
|         RxEvent rx;
 | |
|     } value;
 | |
|     EventType type;
 | |
| } AppEvent;
 | |
| 
 | |
| typedef struct {
 | |
|     uint32_t freq_khz;
 | |
|     bool on;
 | |
|     uint8_t customer_id;
 | |
|     uint32_t em_data;
 | |
|     bool dirty;
 | |
| } State;
 | |
| 
 | |
| static void render_callback(Canvas* canvas, void* ctx) {
 | |
|     State* state = (State*)acquire_mutex((ValueMutex*)ctx, 25);
 | |
| 
 | |
|     canvas_clear(canvas);
 | |
| 
 | |
|     canvas_set_color(canvas, ColorBlack);
 | |
|     canvas_set_font(canvas, FontPrimary);
 | |
|     canvas_draw_str(canvas, 2, 12, "LF RFID");
 | |
| 
 | |
|     canvas_draw_str(canvas, 2, 24, state->on ? "Reading" : "Emulating");
 | |
| 
 | |
|     char buf[14];
 | |
| 
 | |
|     sprintf(buf, "%d kHz", (int)state->freq_khz);
 | |
|     canvas_draw_str(canvas, 2, 36, buf);
 | |
| 
 | |
|     sprintf(buf, "%02d:%010ld", state->customer_id, state->em_data);
 | |
|     canvas_draw_str(canvas, 2, 45, buf);
 | |
| 
 | |
|     release_mutex((ValueMutex*)ctx, state);
 | |
| }
 | |
| 
 | |
| static void input_callback(InputEvent* input_event, void* ctx) {
 | |
|     osMessageQueueId_t event_queue = ctx;
 | |
| 
 | |
|     AppEvent event;
 | |
|     event.type = EventTypeKey;
 | |
|     event.value.input = *input_event;
 | |
|     osMessageQueuePut(event_queue, &event, 0, 0);
 | |
| }
 | |
| 
 | |
| extern TIM_HandleTypeDef TIM_C;
 | |
| void em4100_emulation(uint8_t* data, GpioPin* pin);
 | |
| void prepare_data(uint32_t ID, uint32_t VENDOR, uint8_t* data);
 | |
| 
 | |
| GpioPin debug_0 = {.pin = GPIO_PIN_2, .port = GPIOB};
 | |
| GpioPin debug_1 = {.pin = GPIO_PIN_3, .port = GPIOC};
 | |
| 
 | |
| extern COMP_HandleTypeDef hcomp1;
 | |
| 
 | |
| void comparator_trigger_callback(void* hcomp, void* comp_ctx) {
 | |
|     if((COMP_HandleTypeDef*)hcomp != &hcomp1) return;
 | |
| 
 | |
|     // gpio_write(&debug_0, true);
 | |
| 
 | |
|     osMessageQueueId_t event_queue = comp_ctx;
 | |
| 
 | |
|     AppEvent event;
 | |
|     event.type = EventTypeRx;
 | |
|     event.value.rx.value = (HAL_COMP_GetOutputLevel(hcomp) == COMP_OUTPUT_LEVEL_HIGH);
 | |
|     event.value.rx.dwt_value = DWT->CYCCNT;
 | |
|     osMessageQueuePut(event_queue, &event, 0, 0);
 | |
| 
 | |
|     // gpio_write(&debug_0, false);
 | |
| }
 | |
| 
 | |
| const uint8_t ROW_SIZE = 4;
 | |
| const uint8_t LINE_SIZE = 10;
 | |
| 
 | |
| static bool even_check(uint8_t* buf) {
 | |
|     uint8_t col_parity_sum[ROW_SIZE];
 | |
|     for(uint8_t col = 0; col < ROW_SIZE; col++) {
 | |
|         col_parity_sum[col] = 0;
 | |
|     }
 | |
| 
 | |
|     // line parity
 | |
|     for(uint8_t line = 0; line < LINE_SIZE; line++) {
 | |
|         printf("%d: ", line);
 | |
|         uint8_t parity_sum = 0;
 | |
|         for(uint8_t col = 0; col < ROW_SIZE; col++) {
 | |
|             parity_sum += buf[line * (ROW_SIZE + 1) + col];
 | |
|             col_parity_sum[col] += buf[line * (ROW_SIZE + 1) + col];
 | |
|             printf("%d ", buf[line * (ROW_SIZE + 1) + col]);
 | |
|         }
 | |
|         if((1 & parity_sum) != buf[line * (ROW_SIZE + 1) + ROW_SIZE]) {
 | |
|             printf(
 | |
|                 "line parity fail at %d (%d : %d)\n",
 | |
|                 line,
 | |
|                 parity_sum,
 | |
|                 buf[line * (ROW_SIZE + 1) + ROW_SIZE]);
 | |
|             return false;
 | |
|         }
 | |
|         printf("\n");
 | |
|     }
 | |
| 
 | |
|     for(uint8_t col = 0; col < ROW_SIZE; col++) {
 | |
|         if((1 & col_parity_sum[col]) != buf[LINE_SIZE * (ROW_SIZE + 1) + col]) {
 | |
|             printf(
 | |
|                 "col parity fail at %d (%d : %d)\n",
 | |
|                 col,
 | |
|                 col_parity_sum[col],
 | |
|                 buf[LINE_SIZE * (ROW_SIZE + 1) + col]);
 | |
|             return false;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| static void extract_data(uint8_t* buf, uint8_t* customer, uint32_t* em_data) {
 | |
|     uint32_t data = 0;
 | |
|     uint8_t offset = 0;
 | |
| 
 | |
|     printf("customer: ");
 | |
|     for(uint8_t line = 0; line < 2; line++) {
 | |
|         for(uint8_t col = 0; col < ROW_SIZE; col++) {
 | |
|             uint32_t bit = buf[line * (ROW_SIZE + 1) + col];
 | |
| 
 | |
|             data |= bit << (7 - offset);
 | |
|             printf("%d ", bit);
 | |
| 
 | |
|             offset++;
 | |
|         }
 | |
|     }
 | |
|     printf("\n");
 | |
| 
 | |
|     *customer = data;
 | |
| 
 | |
|     data = 0;
 | |
|     offset = 0;
 | |
|     printf("data: ");
 | |
|     for(uint8_t line = 2; line < LINE_SIZE; line++) {
 | |
|         for(uint8_t col = 0; col < ROW_SIZE; col++) {
 | |
|             uint32_t bit = buf[line * (ROW_SIZE + 1) + col];
 | |
| 
 | |
|             data |= bit << (31 - offset);
 | |
|             printf("%d ", bit);
 | |
| 
 | |
|             offset++;
 | |
|         }
 | |
|     }
 | |
|     printf("\n");
 | |
| 
 | |
|     *em_data = data;
 | |
| }
 | |
| 
 | |
| void lf_rfid_workaround(void* p) {
 | |
|     osMessageQueueId_t event_queue = osMessageQueueNew(1, sizeof(AppEvent), NULL);
 | |
| 
 | |
|     // create pin
 | |
|     GpioPin pull_pin = {.pin = RFID_PULL_Pin, .port = RFID_PULL_GPIO_Port};
 | |
|     // TODO open record
 | |
|     GpioPin* pull_pin_record = &pull_pin;
 | |
| 
 | |
|     gpio_init(pull_pin_record, GpioModeOutputPushPull);
 | |
| 
 | |
|     gpio_init(&debug_0, GpioModeOutputPushPull);
 | |
|     gpio_init(&debug_1, GpioModeOutputPushPull);
 | |
| 
 | |
|     // pulldown iBtn pin to prevent interference from ibutton
 | |
|     gpio_init((GpioPin*)&ibutton_gpio, GpioModeOutputOpenDrain);
 | |
|     gpio_write((GpioPin*)&ibutton_gpio, false);
 | |
| 
 | |
|     // init ctx
 | |
|     void* comp_ctx = (void*)event_queue;
 | |
| 
 | |
|     // start comp
 | |
|     HAL_COMP_Start(&hcomp1);
 | |
| 
 | |
|     uint8_t emulation_data[64];
 | |
| 
 | |
|     State _state;
 | |
|     _state.freq_khz = 125;
 | |
|     _state.on = false;
 | |
|     _state.customer_id = 01;
 | |
|     _state.em_data = 4378151;
 | |
|     _state.dirty = true;
 | |
| 
 | |
|     ValueMutex state_mutex;
 | |
|     if(!init_mutex(&state_mutex, &_state, sizeof(State))) {
 | |
|         printf("cannot create mutex\n");
 | |
|         furiac_exit(NULL);
 | |
|     }
 | |
| 
 | |
|     Widget* widget = widget_alloc();
 | |
| 
 | |
|     widget_draw_callback_set(widget, render_callback, &state_mutex);
 | |
|     widget_input_callback_set(widget, input_callback, event_queue);
 | |
| 
 | |
|     // Open GUI and register widget
 | |
|     Gui* gui = furi_record_open("gui");
 | |
|     gui_add_widget(gui, widget, GuiLayerFullscreen);
 | |
| 
 | |
|     AppEvent event;
 | |
|     uint32_t prev_dwt;
 | |
|     int8_t symbol = -1; // init state
 | |
|     bool center = false;
 | |
|     size_t symbol_cnt = 0;
 | |
| 
 | |
|     GpioPin* led_record = (GpioPin*)&led_gpio[1];
 | |
|     gpio_init(led_record, GpioModeOutputOpenDrain);
 | |
| 
 | |
|     uint8_t buf[64];
 | |
|     for(size_t i = 0; i < 64; i++) {
 | |
|         buf[i] = 0;
 | |
|     }
 | |
| 
 | |
|     while(1) {
 | |
|         osStatus_t event_status = osMessageQueueGet(event_queue, &event, NULL, 100);
 | |
| 
 | |
|         if(event.type == EventTypeRx && event_status == osOK) {
 | |
|             uint32_t dt = (event.value.rx.dwt_value - prev_dwt) / (SystemCoreClock / 1000000.0f);
 | |
|             prev_dwt = event.value.rx.dwt_value;
 | |
| 
 | |
|             if(dt > 384) {
 | |
|                 // change symbol 0->1 or 1->0
 | |
|                 symbol = event.value.rx.value;
 | |
|                 center = true;
 | |
|             } else {
 | |
|                 // same symbol as prev or center
 | |
|                 center = !center;
 | |
|             }
 | |
| 
 | |
|             /*
 | |
|             gpio_write(&debug_1, true);
 | |
|             delay_us(center ? 10 : 30);
 | |
|             gpio_write(&debug_1, false);
 | |
|             */
 | |
| 
 | |
|             if(center && symbol != -1) {
 | |
|                 /*
 | |
|                 gpio_write(&debug_0, true);
 | |
|                 delay_us(symbol ? 10 : 30);
 | |
|                 gpio_write(&debug_0, false);
 | |
|                 */
 | |
| 
 | |
|                 buf[symbol_cnt] = symbol;
 | |
|                 symbol_cnt++;
 | |
|             }
 | |
| 
 | |
|             // check preamble
 | |
|             if(symbol_cnt <= 9 && symbol == 0) {
 | |
|                 symbol_cnt = 0;
 | |
|                 symbol = -1;
 | |
|             }
 | |
| 
 | |
|             // check stop bit
 | |
|             if(symbol_cnt == 64 && symbol == 1) {
 | |
|                 symbol_cnt = 0;
 | |
|                 symbol = -1;
 | |
|             }
 | |
| 
 | |
|             if(symbol_cnt == 64) {
 | |
|                 if(even_check(&buf[9])) {
 | |
|                     State* state = (State*)acquire_mutex_block(&state_mutex);
 | |
|                     extract_data(&buf[9], &state->customer_id, &state->em_data);
 | |
|                     printf("customer: %02d, data: %010lu\n", state->customer_id, state->em_data);
 | |
|                     release_mutex(&state_mutex, state);
 | |
|                     gpio_write(led_record, false);
 | |
|                     osDelay(100);
 | |
|                     gpio_write(led_record, true);
 | |
|                 }
 | |
| 
 | |
|                 symbol_cnt = 0;
 | |
|             }
 | |
|         } else {
 | |
|             State* state = (State*)acquire_mutex_block(&state_mutex);
 | |
| 
 | |
|             if(event_status == osOK) {
 | |
|                 if(event.type == EventTypeKey) {
 | |
|                     // press events
 | |
|                     if(event.value.input.state && event.value.input.input == InputBack) {
 | |
|                         hal_pwmn_stop(&TIM_C, TIM_CHANNEL_1); // TODO: move to furiac_onexit
 | |
|                         gpio_init(pull_pin_record, GpioModeInput);
 | |
|                         gpio_init((GpioPin*)&ibutton_gpio, GpioModeInput);
 | |
| 
 | |
|                         // TODO remove all widgets create by app
 | |
|                         widget_enabled_set(widget, false);
 | |
|                         furiac_exit(NULL);
 | |
|                     }
 | |
| 
 | |
|                     if(event.value.input.state && event.value.input.input == InputUp) {
 | |
|                         state->dirty = true;
 | |
|                         state->freq_khz += 10;
 | |
|                     }
 | |
| 
 | |
|                     if(event.value.input.state && event.value.input.input == InputDown) {
 | |
|                         state->dirty = true;
 | |
|                         state->freq_khz -= 10;
 | |
|                     }
 | |
| 
 | |
|                     if(event.value.input.state && event.value.input.input == InputLeft) {
 | |
|                     }
 | |
| 
 | |
|                     if(event.value.input.state && event.value.input.input == InputRight) {
 | |
|                     }
 | |
| 
 | |
|                     if(event.value.input.state && event.value.input.input == InputOk) {
 | |
|                         state->dirty = true;
 | |
|                         state->on = !state->on;
 | |
|                     }
 | |
|                 }
 | |
|             } else {
 | |
|                 // event timeout
 | |
|             }
 | |
| 
 | |
|             if(state->dirty) {
 | |
|                 if(!state->on) {
 | |
|                     prepare_data(state->em_data, state->customer_id, emulation_data);
 | |
|                 }
 | |
| 
 | |
|                 if(state->on) {
 | |
|                     gpio_write(pull_pin_record, false);
 | |
|                     api_interrupt_add(
 | |
|                         comparator_trigger_callback, InterruptTypeComparatorTrigger, comp_ctx);
 | |
|                 } else {
 | |
|                     api_interrupt_remove(comparator_trigger_callback);
 | |
|                 }
 | |
| 
 | |
|                 hal_pwmn_set(
 | |
|                     state->on ? 0.5 : 0.0, (float)(state->freq_khz * 1000), &LFRFID_TIM, LFRFID_CH);
 | |
| 
 | |
|                 state->dirty = false;
 | |
|             }
 | |
| 
 | |
|             if(!state->on) {
 | |
|                 em4100_emulation(emulation_data, pull_pin_record);
 | |
|             }
 | |
| 
 | |
|             // common code, for example, force update UI
 | |
|             widget_update(widget);
 | |
| 
 | |
|             release_mutex(&state_mutex, state);
 | |
|         }
 | |
|     }
 | |
| }
 | 
