T5577 lib: write with mask function added (#3362)

* T5577 lib: write with mask function added
* t5577_write_with_mask: with_password param added

Co-authored-by: あく <alleteam@gmail.com>

lib/lfrfid/tools/t5577.c

@@ -13,6 +13,9 @@
 #define T5577_OPCODE_PAGE_1 0b11
 #define T5577_OPCODE_RESET 0b00
 
+#define T5577_BLOCKS_IN_PAGE_0 8
+#define T5577_BLOCKS_IN_PAGE_1 4
+
 static void t5577_start() {
     furi_hal_rfid_tim_read_start(125000, 0.5);
 
@@ -51,14 +54,27 @@ static void t5577_write_reset() {
     t5577_write_bit(0);
 }
 
-static void t5577_write_block(uint8_t block, bool lock_bit, uint32_t data) {
+static void t5577_write_block_pass(
+    uint8_t page,
+    uint8_t block,
+    bool lock_bit,
+    uint32_t data,
+    bool with_pass,
+    uint32_t password) {
     furi_delay_us(T5577_TIMING_WAIT_TIME * 8);
 
     // start gap
     t5577_write_gap(T5577_TIMING_START_GAP);
 
-    // opcode for page 0
-    t5577_write_opcode(T5577_OPCODE_PAGE_0);
+    // opcode for page
+    t5577_write_opcode((page == 1) ? T5577_OPCODE_PAGE_1 : T5577_OPCODE_PAGE_0);
+
+    // password
+    if(with_pass) {
+        for(uint8_t i = 0; i < 32; i++) {
+            t5577_write_bit((password >> (31 - i)) & 1);
+        }
+    }
 
     // lock bit
     t5577_write_bit(lock_bit);
@@ -79,13 +95,47 @@ static void t5577_write_block(uint8_t block, bool lock_bit, uint32_t data) {
     t5577_write_reset();
 }
 
+static void t5577_write_block_simple(uint8_t block, bool lock_bit, uint32_t data) {
+    t5577_write_block_pass(0, block, lock_bit, data, false, 0);
+}
+
 void t5577_write(LFRFIDT5577* data) {
     t5577_start();
     FURI_CRITICAL_ENTER();
     for(size_t i = 0; i < data->blocks_to_write; i++) {
-        t5577_write_block(i, false, data->block[i]);
+        t5577_write_block_simple(i, false, data->block[i]);
     }
     t5577_write_reset();
     FURI_CRITICAL_EXIT();
     t5577_stop();
-}
+}
+
+void t5577_write_with_pass(LFRFIDT5577* data, uint32_t password) {
+    t5577_start();
+    FURI_CRITICAL_ENTER();
+    for(size_t i = 0; i < data->blocks_to_write; i++) {
+        t5577_write_block_pass(0, i, false, data->block[i], true, password);
+    }
+    t5577_write_reset();
+    FURI_CRITICAL_EXIT();
+    t5577_stop();
+}
+
+void t5577_write_with_mask(LFRFIDT5577* data, uint8_t page, bool with_pass, uint32_t password) {
+    t5577_start();
+    FURI_CRITICAL_ENTER();
+
+    uint8_t mask = data->mask;
+
+    size_t pages_total = (page == 0) ? T5577_BLOCKS_IN_PAGE_0 : T5577_BLOCKS_IN_PAGE_1;
+
+    for(size_t i = 0; i < pages_total; i++) {
+        bool need_to_write = mask & 1;
+        mask >>= 1;
+        if(!need_to_write) continue;
+        t5577_write_block_pass(page, i, false, data->block[i], with_pass, password);
+    }
+    t5577_write_reset();
+    FURI_CRITICAL_EXIT();
+    t5577_stop();
+}

lib/lfrfid/tools/t5577.h

@@ -42,6 +42,7 @@ extern "C" {
 typedef struct {
     uint32_t block[LFRFID_T5577_BLOCK_COUNT];
     uint32_t blocks_to_write;
+    uint8_t mask;
 } LFRFIDT5577;
 
 /**
@@ -51,6 +52,10 @@ typedef struct {
  */
 void t5577_write(LFRFIDT5577* data);
 
+void t5577_write_with_pass(LFRFIDT5577* data, uint32_t password);
+
+void t5577_write_with_mask(LFRFIDT5577* data, uint8_t page, bool with_pass, uint32_t password);
+
 #ifdef __cplusplus
 }
 #endif

targets/f7/api_symbols.csv

@@ -3258,6 +3258,8 @@ Function,+,submenu_set_header,void,"Submenu*, const char*"
 Function,+,submenu_set_selected_item,void,"Submenu*, uint32_t"
 Function,-,system,int,const char*
 Function,+,t5577_write,void,LFRFIDT5577*
+Function,+,t5577_write_with_mask,void,"LFRFIDT5577*, uint8_t, _Bool, uint32_t"
+Function,+,t5577_write_with_pass,void,"LFRFIDT5577*, uint32_t"
 Function,-,tan,double,double
 Function,-,tanf,float,float
 Function,-,tanh,double,double