aqara-security-demo

Change adds a demo for aqara security features.

drivers/Aqara/aqara-security-demo/config.yml

@@ -0,0 +1,6 @@
+name: 'Aqara Security Demo'
+packageKey: 'aqara-security-demo'
+permissions:
+  lan: {}
+  discovery: {}
+

drivers/Aqara/aqara-security-demo/profiles/aqara-security-demo.yml

@@ -0,0 +1,6 @@
+name: aqara-security-demo
+components:
+  - id: main
+    capabilities:
+      - id: refresh
+        version: 1

drivers/Aqara/aqara-security-demo/src/init.lua

@@ -0,0 +1,72 @@
+local capabilities = require "st.capabilities"
+local Driver = require "st.driver"
+local log = require "log"
+local security = require "st.security"
+
+function dump(data)
+  if type(data) == "table" then
+    local s = '{ '
+    for k,v in pairs(data) do
+      if type(k) ~= 'number' then k = '"'..k..'"' end
+      s = s .. '['..k..'] = ' .. dump(v) .. ','
+    end
+    return s .. '}'
+  else
+    return tostring(data)
+  end
+end
+
+local function my_secret_data_handler(data)
+  -- At time of writing this returns nothind beyond "secret_type = aqara"
+  log.info(dump(data))
+end
+
+local function discovery_handler(driver, _, should_continue)
+  log.info("starting_discovery")
+  example_aes_128_ecb()
+  example_aes_256_ecb()
+  local device_list = driver:get_devices()
+  local zigbee_id = "\x01\x02\x03\x04\x05\x06\x07\x08";
+  local res, err = security.get_aqara_secret("aqara", zigbee_id, "encrypted_pub_key", "model_name", "test_mn_id", "test_setup_id")
+  if res then
+    print(res)
+  end
+end
+function example_aes_128_ecb() 
+  log.info("AES128 testing...")
+  local input_data = "oh wow, here is some test data"
+  local expected_result = "\xE1\x9D\x5E\x19\x7F\x43\xD2\x98\x9C\x92\x99\xF5\xF7\xF3\x97\x3B\x0F\x19\xDA\x4D\x5B\xD1\x9C\x11\x55\xC2\xF3\x0A\xE7\xE8\x12\x13"
+  local key = "\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"
+  local opts = {cipher = "aes128-ecb"}
+  local result = security.encrypt_bytes(input_data, key, opts)
+  assert(result == expected_result)
+  local final_result = security.decrypt_bytes(result, key, opts)
+  assert(final_result == input_data)
+  log.info("AES128 passed!")
+end
+
+function example_aes_256_ecb() 
+  log.info("AES256 testing...")
+  local input_data = "oh wow, here is some test data"
+  local expected_result = "\x03\x61\xF6\xBE\x3B\xD4\x8E\x6E\x80\x44\x83\x5A\xEB\x07\x22\xDD\x0B\x54\xAC\x4D\x20\x5E\x54\xF6\xB0\xA0\x14\xEA\x9E\xED\xCA\x29"
+  local key = "\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"
+  local opts = {cipher = "aes256-ecb"}
+  local result = security.encrypt_bytes(input_data, key, opts)
+  assert(result == expected_result)
+  local final_result = security.decrypt_bytes(result, key, opts)
+  assert(final_result == input_data)
+  log.info("AES256 passed!")
+end
+
+local aqara_security_demo = Driver("aqara_security_demo", {
+  discovery = discovery_handler,
+  supported_capabilities = {
+    capabilities.refresh,
+  },
+  -- A raw handler can be used as well, the wrapper is suggested.
+  --secret_data_handler = my_secret_data_handler,
+})
+
+security.register_aqara_secret_handler(aqara_security_demo, my_secret_data_handler)
+log.info("Demo running!")
+aqara_security_demo:run()