Просмотр исходного кода

[FL-1293] iButton key rename (#479)

* ibutton: remove existing key file before saving new
* ibutton: save key name in iButtunKey object
* ibutton: rename IBUTTON_KEY_SIZE -> IBUTTON_KEY_DATA_SIZE
* ibutton: clear key when enter one manually
* ibutton: change strcpy -> strlcpy

Co-authored-by: あく <alleteam@gmail.com>
gornekich 4 лет назад
Родитель
Сommit
1242327e14

+ 2 - 1
applications/ibutton/helpers/key-info.h

@@ -1,7 +1,8 @@
 #pragma once
 #include <stdint.h>
 
-static const uint8_t IBUTTON_KEY_SIZE = 8;
+static const uint8_t IBUTTON_KEY_DATA_SIZE = 8;
+static const uint8_t IBUTTON_KEY_NAME_SIZE = 64;
 
 enum class iButtonKeyType : uint8_t {
     KeyDallas,

+ 7 - 3
applications/ibutton/ibutton-key.cpp

@@ -2,7 +2,7 @@
 #include <furi.h>
 
 uint8_t iButtonKey::get_size() {
-    return IBUTTON_KEY_SIZE;
+    return IBUTTON_KEY_DATA_SIZE;
 }
 
 void iButtonKey::set_data(uint8_t* _data, uint8_t _data_count) {
@@ -13,6 +13,10 @@ void iButtonKey::set_data(uint8_t* _data, uint8_t _data_count) {
     memcpy(data, _data, _data_count);
 }
 
+void iButtonKey::clear_data() {
+    memset(data, 0, get_size());
+}
+
 uint8_t* iButtonKey::get_data() {
     return data;
 }
@@ -36,10 +40,10 @@ uint8_t iButtonKey::get_type_data_size() {
 }
 
 void iButtonKey::set_name(const char* _name) {
-    name = _name;
+    strlcpy(name, _name, IBUTTON_KEY_NAME_SIZE);
 }
 
-const char* iButtonKey::get_name() {
+char* iButtonKey::get_name() {
     return name;
 }
 

+ 4 - 3
applications/ibutton/ibutton-key.h

@@ -7,11 +7,12 @@ public:
     uint8_t get_size();
 
     void set_data(uint8_t* data, uint8_t data_count);
+    void clear_data();
     uint8_t* get_data();
     uint8_t get_type_data_size();
 
     void set_name(const char* name);
-    const char* get_name();
+    char* get_name();
 
     void set_type(iButtonKeyType key_type);
     iButtonKeyType get_key_type();
@@ -19,8 +20,8 @@ public:
     iButtonKey();
 
 private:
-    uint8_t data[IBUTTON_KEY_SIZE] = {0, 0, 0, 0, 0, 0, 0, 0};
-    const char* name = {0};
+    uint8_t data[IBUTTON_KEY_DATA_SIZE] = {0, 0, 0, 0, 0, 0, 0, 0};
+    char name[IBUTTON_KEY_NAME_SIZE] = {0};
 
     iButtonKeyType type = iButtonKeyType::KeyDallas;
 };

+ 2 - 0
applications/ibutton/scene/ibutton-scene-add-type.cpp

@@ -37,6 +37,8 @@ bool iButtonSceneAddType::on_event(iButtonApp* app, iButtonEvent* event) {
             app->get_key()->set_type(iButtonKeyType::KeyMetakom);
             break;
         }
+        app->get_key()->set_name("");
+        app->get_key()->clear_data();
         app->switch_to_next_scene(iButtonApp::Scene::SceneAddValue);
         consumed = true;
     }

+ 14 - 5
applications/ibutton/scene/ibutton-scene-save-name.cpp

@@ -34,18 +34,27 @@ bool iButtonSceneSaveName::on_event(iButtonApp* app, iButtonEvent* event) {
         iButtonKey* key = app->get_key();
         File key_file;
         string_t key_file_name;
+
+        // Create ibutton directory if necessary
+        app->get_fs_api()->common.mkdir("ibutton");
+
+        // First remove key if it was saved
         string_init_set_str(key_file_name, "ibutton/");
+        string_cat_str(key_file_name, key->get_name());
+        app->get_fs_api()->common.remove(string_get_cstr(key_file_name));
+
+        // Save the key
+        key->set_name(app->get_text_store());
+        string_set_str(key_file_name, "ibutton/");
         string_cat_str(key_file_name, app->get_text_store());
-        uint8_t key_data[IBUTTON_KEY_SIZE + 1];
+        uint8_t key_data[IBUTTON_KEY_DATA_SIZE + 1];
         key_data[0] = static_cast<uint8_t>(key->get_key_type());
-        memcpy(key_data + 1, key->get_data(), IBUTTON_KEY_SIZE);
-        // Create ibutton directory if necessary
-        app->get_fs_api()->common.mkdir("ibutton");
+        memcpy(key_data + 1, key->get_data(), IBUTTON_KEY_DATA_SIZE);
         bool res = app->get_fs_api()->file.open(
             &key_file, string_get_cstr(key_file_name), FSAM_WRITE, FSOM_CREATE_ALWAYS);
         // TODO process file system errors from file system service
         if(res) {
-            res = app->get_fs_api()->file.write(&key_file, key_data, IBUTTON_KEY_SIZE + 1);
+            res = app->get_fs_api()->file.write(&key_file, key_data, IBUTTON_KEY_DATA_SIZE + 1);
             res = app->get_fs_api()->file.close(&key_file);
             app->switch_to_next_scene(iButtonApp::Scene::SceneSaveSuccess);
         } else {

+ 3 - 3
applications/ibutton/scene/ibutton-scene-start.cpp

@@ -43,12 +43,12 @@ bool iButtonSceneStart::on_event(iButtonApp* app, iButtonEvent* event) {
                 string_init_set_str(key_str, "ibutton/");
                 string_cat_str(key_str, app->get_file_name());
                 File key_file;
-                uint8_t key_data[IBUTTON_KEY_SIZE + 1] = {};
+                uint8_t key_data[IBUTTON_KEY_DATA_SIZE + 1] = {};
                 // Read data from file
                 // TODO handle false return
                 res = app->get_fs_api()->file.open(
                     &key_file, string_get_cstr(key_str), FSAM_READ, FSOM_OPEN_EXISTING);
-                res = app->get_fs_api()->file.read(&key_file, key_data, IBUTTON_KEY_SIZE + 1);
+                res = app->get_fs_api()->file.read(&key_file, key_data, IBUTTON_KEY_DATA_SIZE + 1);
                 res = app->get_fs_api()->file.close(&key_file);
                 string_clear(key_str);
                 // Set key
@@ -58,7 +58,7 @@ bool iButtonSceneStart::on_event(iButtonApp* app, iButtonEvent* event) {
                 }
                 app->get_key()->set_name(app->get_file_name());
                 app->get_key()->set_type(key_type);
-                app->get_key()->set_data(key_data + 1, IBUTTON_KEY_SIZE);
+                app->get_key()->set_data(key_data + 1, IBUTTON_KEY_DATA_SIZE);
                 app->switch_to_next_scene(iButtonApp::Scene::SceneSavedKeyMenu);
             } else {
                 // TODO add error scene