From e7ddd47716faa26a95f8df50c857205f82c8c9c1 Mon Sep 17 00:00:00 2001 From: Piyush Shah Date: Thu, 14 Jun 2018 16:27:37 +0530 Subject: [PATCH] nvs_partition_generator: Add support for base64 representation of Binary data --- components/nvs_flash/nvs_partition_generator/README.rst | 4 ++-- .../nvs_partition_generator/nvs_partition_gen.py | 9 ++++++--- components/nvs_flash/nvs_partition_generator/sample.csv | 2 ++ .../nvs_partition_generator/testdata/sample.base64 | 1 + components/nvs_flash/test_nvs_host/test_nvs.cpp | 3 +++ 5 files changed, 14 insertions(+), 5 deletions(-) create mode 100644 components/nvs_flash/nvs_partition_generator/testdata/sample.base64 diff --git a/components/nvs_flash/nvs_partition_generator/README.rst b/components/nvs_flash/nvs_partition_generator/README.rst index 5b62ebc394..f324f016f1 100644 --- a/components/nvs_flash/nvs_partition_generator/README.rst +++ b/components/nvs_flash/nvs_partition_generator/README.rst @@ -19,9 +19,9 @@ Type Supported values are ``file``, ``data`` and ``namespace``. Encoding - Supported values are: ``u8``, ``i8``, ``u16``, ``u32``, ``i32``, ``string``, ``hex2bin`` and ``binary``. This specifies how actual data values are encoded in the resultant binary file. Difference between ``string`` and ``binary`` encoding is that ``string`` data is terminated with a NULL character, whereas ``binary`` data is not. + Supported values are: ``u8``, ``i8``, ``u16``, ``u32``, ``i32``, ``string``, ``hex2bin``, ``base64`` and ``binary``. This specifies how actual data values are encoded in the resultant binary file. Difference between ``string`` and ``binary`` encoding is that ``string`` data is terminated with a NULL character, whereas ``binary`` data is not. - .. note:: For ``file`` type, only ``hex2bin``, ``string`` and ``binary`` is supported as of now. + .. note:: For ``file`` type, only ``hex2bin``, ``base64``, ``string`` and ``binary`` is supported as of now. Value Data value. diff --git a/components/nvs_flash/nvs_partition_generator/nvs_partition_gen.py b/components/nvs_flash/nvs_partition_generator/nvs_partition_gen.py index b895fc1ca2..6224d3b94b 100755 --- a/components/nvs_flash/nvs_partition_generator/nvs_partition_gen.py +++ b/components/nvs_flash/nvs_partition_generator/nvs_partition_gen.py @@ -132,7 +132,7 @@ class Page(object): # set Type if encoding == "string": entry_struct[1] = Page.SZ - elif encoding == "hex2bin" or encoding == "binary": + elif encoding in ["hex2bin", "binary", "base64"]: entry_struct[1] = Page.BLOB # compute CRC of data @@ -248,11 +248,14 @@ class NVS(object): raise InputError("%s: Invalid data length. Should be multiple of 2." % key) value = binascii.a2b_hex(value) + if encoding == "base64": + value = binascii.a2b_base64(value) + if encoding == "string": value += '\0' encoding = encoding.lower() - varlen_encodings = ["string", "binary", "hex2bin"] + varlen_encodings = ["string", "binary", "hex2bin", "base64"] primitive_encodings = ["u8", "i8", "u16", "u32", "i32"] if encoding in varlen_encodings: try: @@ -308,7 +311,7 @@ def write_entry(nvs_instance, key, datatype, encoding, value): :param nvs_instance: Instance of an NVS class returned by nvs_open() :param key: Key of the data :param datatype: Data type. Valid values are "file", "data" and "namespace" - :param encoding: Data encoding. Valid values are "u8", "i8", "u16", "u32", "i32", "string", "binary" and "hex2bin" + :param encoding: Data encoding. Valid values are "u8", "i8", "u16", "u32", "i32", "string", "binary", "hex2bin" and "base64" :param value: Data value in ascii encoded string format for "data" datatype and filepath for "file" datatype :return: None """ diff --git a/components/nvs_flash/nvs_partition_generator/sample.csv b/components/nvs_flash/nvs_partition_generator/sample.csv index ccade4dbb1..bc70aa14c7 100644 --- a/components/nvs_flash/nvs_partition_generator/sample.csv +++ b/components/nvs_flash/nvs_partition_generator/sample.csv @@ -7,6 +7,8 @@ dummyU32Key,data,u32,4294967295 dummyI32Key,data,i32,-2147483648 dummyStringKey,data,string,0A:0B:0C:0D:0E:0F dummyHex2BinKey,data,hex2bin,010203abcdef +dummyBase64Key,data,base64,MTIzYWJj hexFileKey,file,hex2bin,testdata/sample.hex +base64FileKey,file,base64,testdata/sample.base64 stringFileKey,file,string,testdata/sample.txt binFileKey,file,binary,testdata/sample.bin diff --git a/components/nvs_flash/nvs_partition_generator/testdata/sample.base64 b/components/nvs_flash/nvs_partition_generator/testdata/sample.base64 new file mode 100644 index 0000000000..0c16c99a22 --- /dev/null +++ b/components/nvs_flash/nvs_partition_generator/testdata/sample.base64 @@ -0,0 +1 @@ +AQIDBAUGBwgJq83v diff --git a/components/nvs_flash/test_nvs_host/test_nvs.cpp b/components/nvs_flash/test_nvs_host/test_nvs.cpp index 658fdb53fb..22d30d0557 100644 --- a/components/nvs_flash/test_nvs_host/test_nvs.cpp +++ b/components/nvs_flash/test_nvs_host/test_nvs.cpp @@ -1548,6 +1548,9 @@ TEST_CASE("read data from partition generated via partition generation utility", uint8_t hexdata[] = {0x01, 0x02, 0x03, 0xab, 0xcd, 0xef}; TEST_ESP_OK( nvs_get_blob(handle, "dummyHex2BinKey", buf, &buflen)); CHECK(memcmp(buf, hexdata, buflen) == 0); + uint8_t base64data[] = {'1', '2', '3', 'a', 'b', 'c'}; + TEST_ESP_OK( nvs_get_blob(handle, "dummyBase64Key", buf, &buflen)); + CHECK(memcmp(buf, base64data, buflen) == 0); } TEST_CASE("dump all performance data", "[nvs]") -- 2.40.0