test_generate_factory_data.py 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316
  1. #!/usr/bin/env python3
  2. #
  3. # Copyright (c) 2022 Project CHIP Authors
  4. #
  5. # Licensed under the Apache License, Version 2.0 (the "License");
  6. # you may not use this file except in compliance with the License.
  7. # You may obtain a copy of the License at
  8. #
  9. # http://www.apache.org/licenses/LICENSE-2.0
  10. #
  11. # Unless required by applicable law or agreed to in writing, software
  12. # distributed under the License is distributed on an "AS IS" BASIS,
  13. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. # See the License for the specific language governing permissions and
  15. # limitations under the License.
  16. #
  17. import base64
  18. import json
  19. import os
  20. import subprocess
  21. import tempfile
  22. import unittest
  23. from os.path import exists
  24. TOOLS_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
  25. DAC_DER_KEY = bytes([0x30, 0x77, 0x02, 0x01, 0x01, 0x04, 0x20, 0xbf, 0x26, 0xd5, 0xd2, 0x25,
  26. 0xeb, 0x6b, 0x09, 0x6d, 0xd5, 0xa6, 0xb9, 0x03, 0x04, 0x8e, 0xf2, 0xd7,
  27. 0x6e, 0xf2, 0xe8, 0x56, 0x25, 0x39, 0x0b, 0xd5, 0x70, 0xb2, 0xf1, 0x65,
  28. 0x99, 0x30, 0xeb, 0xa0, 0x0a, 0x06, 0x08, 0x2a, 0x86, 0x48, 0xce, 0x3d,
  29. 0x03, 0x01, 0x07, 0xa1, 0x44, 0x03, 0x42, 0x00, 0x04, 0x48, 0x36, 0x32,
  30. 0x85, 0x68, 0x70, 0x00, 0x9e, 0xd6, 0x8e, 0x78, 0xc0, 0xd9, 0x4b, 0xe7,
  31. 0xd9, 0xb5, 0x97, 0xb7, 0x88, 0x1d, 0xfb, 0x96, 0x00, 0xbb, 0x47, 0x1d,
  32. 0x8b, 0x70, 0xbb, 0xce, 0x1d, 0xf4, 0x47, 0xa7, 0x93, 0x60, 0x2e, 0x14,
  33. 0xde, 0x07, 0xdb, 0x80, 0xef, 0x75, 0xd8, 0x6c, 0x55, 0x6c, 0x7a, 0xc4,
  34. 0xb4, 0x06, 0x0d, 0x50, 0xe1, 0x0f, 0xe2, 0x26, 0x06, 0xb4, 0xdd, 0x1b,
  35. 0x4f])
  36. DAC_RAW_KEY = bytes([0xbf, 0x26, 0xd5, 0xd2, 0x25, 0xeb, 0x6b, 0x09, 0x6d, 0xd5, 0xa6, 0xb9,
  37. 0x03, 0x04, 0x8e, 0xf2, 0xd7, 0x6e, 0xf2, 0xe8, 0x56, 0x25, 0x39, 0x0b,
  38. 0xd5, 0x70, 0xb2, 0xf1, 0x65, 0x99, 0x30, 0xeb])
  39. DAC_DER_CERT = bytes([0x30, 0x82, 0x01, 0xe8, 0x30, 0x82, 0x01, 0x8e, 0xa0, 0x03, 0x02, 0x01,
  40. 0x02, 0x02, 0x08, 0x49, 0x2e, 0x20, 0xdb, 0x59, 0x76, 0xa8, 0x90, 0x30,
  41. 0x0a, 0x06, 0x08, 0x2a, 0x86, 0x48, 0xce, 0x3d, 0x04, 0x03, 0x02, 0x30,
  42. 0x3e, 0x31, 0x26, 0x30, 0x24, 0x06, 0x03, 0x55, 0x04, 0x03, 0x0c, 0x1d,
  43. 0x4e, 0x6f, 0x72, 0x64, 0x69, 0x63, 0x20, 0x53, 0x65, 0x6d, 0x69, 0x63,
  44. 0x6f, 0x6e, 0x64, 0x75, 0x63, 0x74, 0x6f, 0x72, 0x20, 0x41, 0x53, 0x41,
  45. 0x5f, 0x4c, 0x6f, 0x63, 0x6b, 0x31, 0x14, 0x30, 0x12, 0x06, 0x0a, 0x2b,
  46. 0x06, 0x01, 0x04, 0x01, 0x82, 0xa2, 0x7c, 0x02, 0x01, 0x0c, 0x04, 0x31,
  47. 0x32, 0x37, 0x46, 0x30, 0x1e, 0x17, 0x0d, 0x32, 0x32, 0x30, 0x37, 0x31,
  48. 0x39, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x17, 0x0d, 0x34, 0x39,
  49. 0x31, 0x32, 0x30, 0x33, 0x32, 0x33, 0x35, 0x39, 0x35, 0x39, 0x5a, 0x30,
  50. 0x54, 0x31, 0x26, 0x30, 0x24, 0x06, 0x03, 0x55, 0x04, 0x03, 0x0c, 0x1d,
  51. 0x4e, 0x6f, 0x72, 0x64, 0x69, 0x63, 0x20, 0x53, 0x65, 0x6d, 0x69, 0x63,
  52. 0x6f, 0x6e, 0x64, 0x75, 0x63, 0x74, 0x6f, 0x72, 0x20, 0x41, 0x53, 0x41,
  53. 0x5f, 0x4c, 0x6f, 0x63, 0x6b, 0x31, 0x14, 0x30, 0x12, 0x06, 0x0a, 0x2b,
  54. 0x06, 0x01, 0x04, 0x01, 0x82, 0xa2, 0x7c, 0x02, 0x01, 0x0c, 0x04, 0x31,
  55. 0x32, 0x37, 0x46, 0x31, 0x14, 0x30, 0x12, 0x06, 0x0a, 0x2b, 0x06, 0x01,
  56. 0x04, 0x01, 0x82, 0xa2, 0x7c, 0x02, 0x02, 0x0c, 0x04, 0x41, 0x42, 0x43,
  57. 0x44, 0x30, 0x59, 0x30, 0x13, 0x06, 0x07, 0x2a, 0x86, 0x48, 0xce, 0x3d,
  58. 0x02, 0x01, 0x06, 0x08, 0x2a, 0x86, 0x48, 0xce, 0x3d, 0x03, 0x01, 0x07,
  59. 0x03, 0x42, 0x00, 0x04, 0x48, 0x36, 0x32, 0x85, 0x68, 0x70, 0x00, 0x9e,
  60. 0xd6, 0x8e, 0x78, 0xc0, 0xd9, 0x4b, 0xe7, 0xd9, 0xb5, 0x97, 0xb7, 0x88,
  61. 0x1d, 0xfb, 0x96, 0x00, 0xbb, 0x47, 0x1d, 0x8b, 0x70, 0xbb, 0xce, 0x1d,
  62. 0xf4, 0x47, 0xa7, 0x93, 0x60, 0x2e, 0x14, 0xde, 0x07, 0xdb, 0x80, 0xef,
  63. 0x75, 0xd8, 0x6c, 0x55, 0x6c, 0x7a, 0xc4, 0xb4, 0x06, 0x0d, 0x50, 0xe1,
  64. 0x0f, 0xe2, 0x26, 0x06, 0xb4, 0xdd, 0x1b, 0x4f, 0xa3, 0x60, 0x30, 0x5e,
  65. 0x30, 0x0c, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01, 0x01, 0xff, 0x04, 0x02,
  66. 0x30, 0x00, 0x30, 0x0e, 0x06, 0x03, 0x55, 0x1d, 0x0f, 0x01, 0x01, 0xff,
  67. 0x04, 0x04, 0x03, 0x02, 0x07, 0x80, 0x30, 0x1d, 0x06, 0x03, 0x55, 0x1d,
  68. 0x0e, 0x04, 0x16, 0x04, 0x14, 0xbc, 0x6a, 0xa5, 0x79, 0x3c, 0x51, 0xa7,
  69. 0x60, 0x18, 0x38, 0x66, 0x4b, 0x26, 0xa7, 0xd3, 0xec, 0x25, 0x87, 0x46,
  70. 0x18, 0x30, 0x1f, 0x06, 0x03, 0x55, 0x1d, 0x23, 0x04, 0x18, 0x30, 0x16,
  71. 0x80, 0x14, 0x4b, 0x5b, 0xd0, 0x91, 0x60, 0xa3, 0x0e, 0xac, 0x2f, 0x94,
  72. 0xa4, 0x82, 0x6b, 0xd7, 0x6e, 0x96, 0x39, 0xca, 0xd3, 0xb3, 0x30, 0x0a,
  73. 0x06, 0x08, 0x2a, 0x86, 0x48, 0xce, 0x3d, 0x04, 0x03, 0x02, 0x03, 0x48,
  74. 0x00, 0x30, 0x45, 0x02, 0x20, 0x38, 0xf4, 0x2f, 0xd7, 0x06, 0x9a, 0xbc,
  75. 0xfc, 0x83, 0x2b, 0x74, 0xe1, 0xb6, 0x11, 0xb0, 0x2f, 0x72, 0xfd, 0xc2,
  76. 0x75, 0x59, 0xdd, 0x7d, 0x04, 0x9c, 0x81, 0x37, 0x01, 0x74, 0x98, 0x77,
  77. 0x22, 0x02, 0x21, 0x00, 0xb8, 0x9d, 0x63, 0x5b, 0xe2, 0xd3, 0x03, 0xe3,
  78. 0xbc, 0xcb, 0x7e, 0x95, 0x18, 0xc2, 0xbb, 0x0c, 0x1d, 0xff, 0x3b, 0x3d,
  79. 0x37, 0x41, 0x72, 0x2a, 0xd3, 0x4d, 0x38, 0x5c, 0x64, 0x2b, 0xc1, 0x46])
  80. PAI_DER_CERT = bytes([0x30, 0x82, 0x01, 0xb4, 0x30, 0x82, 0x01, 0x5a, 0xa0, 0x03, 0x02, 0x01,
  81. 0x02, 0x02, 0x08, 0x09, 0x10, 0x34, 0x50, 0x40, 0x83, 0x6a, 0x05, 0x30,
  82. 0x0a, 0x06, 0x08, 0x2a, 0x86, 0x48, 0xce, 0x3d, 0x04, 0x03, 0x02, 0x30,
  83. 0x1a, 0x31, 0x18, 0x30, 0x16, 0x06, 0x03, 0x55, 0x04, 0x03, 0x0c, 0x0f,
  84. 0x4d, 0x61, 0x74, 0x74, 0x65, 0x72, 0x20, 0x54, 0x65, 0x73, 0x74, 0x20,
  85. 0x50, 0x41, 0x41, 0x30, 0x1e, 0x17, 0x0d, 0x32, 0x32, 0x30, 0x37, 0x31,
  86. 0x39, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x5a, 0x17, 0x0d, 0x34, 0x39,
  87. 0x31, 0x32, 0x30, 0x33, 0x32, 0x33, 0x35, 0x39, 0x35, 0x39, 0x5a, 0x30,
  88. 0x3e, 0x31, 0x26, 0x30, 0x24, 0x06, 0x03, 0x55, 0x04, 0x03, 0x0c, 0x1d,
  89. 0x4e, 0x6f, 0x72, 0x64, 0x69, 0x63, 0x20, 0x53, 0x65, 0x6d, 0x69, 0x63,
  90. 0x6f, 0x6e, 0x64, 0x75, 0x63, 0x74, 0x6f, 0x72, 0x20, 0x41, 0x53, 0x41,
  91. 0x5f, 0x4c, 0x6f, 0x63, 0x6b, 0x31, 0x14, 0x30, 0x12, 0x06, 0x0a, 0x2b,
  92. 0x06, 0x01, 0x04, 0x01, 0x82, 0xa2, 0x7c, 0x02, 0x01, 0x0c, 0x04, 0x31,
  93. 0x32, 0x37, 0x46, 0x30, 0x59, 0x30, 0x13, 0x06, 0x07, 0x2a, 0x86, 0x48,
  94. 0xce, 0x3d, 0x02, 0x01, 0x06, 0x08, 0x2a, 0x86, 0x48, 0xce, 0x3d, 0x03,
  95. 0x01, 0x07, 0x03, 0x42, 0x00, 0x04, 0xd7, 0x0c, 0x57, 0xcd, 0xac, 0x0c,
  96. 0x8b, 0x0b, 0x25, 0xfc, 0x64, 0x70, 0xa4, 0x2f, 0xb3, 0xf1, 0x37, 0xf7,
  97. 0x5f, 0x65, 0x2c, 0xd0, 0xb2, 0x15, 0xf1, 0xfe, 0x13, 0x53, 0x52, 0x3f,
  98. 0x59, 0x81, 0xd2, 0x3d, 0xf7, 0xf1, 0x59, 0x88, 0xbd, 0xce, 0xe4, 0x3a,
  99. 0x20, 0x84, 0xe6, 0x1d, 0xe7, 0x3c, 0x83, 0xfb, 0xc4, 0x86, 0x5e, 0x5c,
  100. 0xb8, 0x45, 0x5e, 0x2b, 0xa3, 0x70, 0x08, 0xfb, 0x05, 0x1b, 0xa3, 0x66,
  101. 0x30, 0x64, 0x30, 0x12, 0x06, 0x03, 0x55, 0x1d, 0x13, 0x01, 0x01, 0xff,
  102. 0x04, 0x08, 0x30, 0x06, 0x01, 0x01, 0xff, 0x02, 0x01, 0x01, 0x30, 0x0e,
  103. 0x06, 0x03, 0x55, 0x1d, 0x0f, 0x01, 0x01, 0xff, 0x04, 0x04, 0x03, 0x02,
  104. 0x01, 0x06, 0x30, 0x1d, 0x06, 0x03, 0x55, 0x1d, 0x0e, 0x04, 0x16, 0x04,
  105. 0x14, 0x4b, 0x5b, 0xd0, 0x91, 0x60, 0xa3, 0x0e, 0xac, 0x2f, 0x94, 0xa4,
  106. 0x82, 0x6b, 0xd7, 0x6e, 0x96, 0x39, 0xca, 0xd3, 0xb3, 0x30, 0x1f, 0x06,
  107. 0x03, 0x55, 0x1d, 0x23, 0x04, 0x18, 0x30, 0x16, 0x80, 0x14, 0x78, 0x5c,
  108. 0xe7, 0x05, 0xb8, 0x6b, 0x8f, 0x4e, 0x6f, 0xc7, 0x93, 0xaa, 0x60, 0xcb,
  109. 0x43, 0xea, 0x69, 0x68, 0x82, 0xd5, 0x30, 0x0a, 0x06, 0x08, 0x2a, 0x86,
  110. 0x48, 0xce, 0x3d, 0x04, 0x03, 0x02, 0x03, 0x48, 0x00, 0x30, 0x45, 0x02,
  111. 0x20, 0x60, 0x22, 0xfc, 0xeb, 0x83, 0x4c, 0x6f, 0xb1, 0x4b, 0xa0, 0x72,
  112. 0x3b, 0xcd, 0x8f, 0x68, 0x51, 0x5b, 0x29, 0x04, 0xa9, 0x6f, 0x5d, 0xb7,
  113. 0xec, 0xe1, 0xf2, 0x30, 0x93, 0xd7, 0x49, 0x7e, 0xee, 0x02, 0x21, 0x00,
  114. 0xdf, 0xe7, 0x72, 0xe6, 0xdc, 0x1a, 0xad, 0xf0, 0x2c, 0x58, 0x7a, 0x0d,
  115. 0xde, 0x3d, 0xc0, 0x14, 0x3a, 0x97, 0xe1, 0x35, 0x38, 0xf7, 0xff, 0x76,
  116. 0x05, 0x5e, 0xbf, 0x27, 0x90, 0x6f, 0x50, 0x0f])
  117. TEST_MANUAL_CODE = "Manualcode : 35442608082"
  118. TEST_QR_CODE = "QRCode : MT:KAYA3EYF15ND8B1OA00"
  119. def write_file(path: str, content: bytes) -> None:
  120. with open(path, 'wb') as f:
  121. f.write(content)
  122. def read_json(path: str) -> object:
  123. with open(path) as f:
  124. return json.load(f)
  125. def bytes_to_json(content: bytes) -> str:
  126. return f'hex:{content.hex()}'
  127. def base64_to_json(content: str) -> str:
  128. return bytes_to_json(base64.b64decode(content))
  129. class TestGenerateFactoryData(unittest.TestCase):
  130. def test_generate_factory_data_all_specified(self):
  131. with tempfile.TemporaryDirectory() as outdir:
  132. write_file(os.path.join(outdir, 'DAC_key.der'), DAC_DER_KEY)
  133. write_file(os.path.join(outdir, 'DAC_cert.der'), DAC_DER_CERT)
  134. write_file(os.path.join(outdir, 'PAI_cert.der'), PAI_DER_CERT)
  135. subprocess.check_call(['python3', os.path.join(TOOLS_DIR, 'generate_nrfconnect_chip_factory_data.py'),
  136. '-s', os.path.join(TOOLS_DIR, 'nrfconnect_factory_data.schema'),
  137. '--include_passcode',
  138. '--sn', 'SN:12345678',
  139. '--vendor_id', '0x127F',
  140. '--product_id', '0xABCD',
  141. '--vendor_name', 'Nordic Semiconductor ASA',
  142. '--product_name', 'Lock Gen2',
  143. '--part_number', 'PCA10056',
  144. '--product_url', 'https://example.com/lock',
  145. '--product_label', 'Lock',
  146. '--date', '2022-07-20',
  147. '--hw_ver', '101',
  148. '--hw_ver_str', 'v1.1',
  149. '--dac_key', os.path.join(outdir, 'DAC_key.der'),
  150. '--dac_cert', os.path.join(outdir, 'DAC_cert.der'),
  151. '--pai_cert', os.path.join(outdir, 'PAI_cert.der'),
  152. '--spake2_it', '2000',
  153. '--spake2_salt', 'U1BBS0UyUCBLZXkgU2FsdA==',
  154. '--passcode', '13243546',
  155. '--spake2_verifier', ('WN0SgEXLfUN19BbJqp6qn4pS69EtdNLReIMZwv/CIM0ECMP7ytiAJ7txIYJ0Ovlha/'
  156. 'rQ3E+88mj3qaqqnviMaZzG+OyXEdSocDIT9ZhmkTCgWwERaHz4Vdh3G37RT6kqbw=='),
  157. '--discriminator', '0xFED',
  158. '--rd_uid', '91a9c12a7c80700a31ddcfa7fce63e44',
  159. '--enable_key', '00112233445566778899aabbccddeeff',
  160. '--product_color', 'red',
  161. '--product_finish', 'satin',
  162. '--user', '{"name": "product_name", "version": 123, "revision": "0x123"}',
  163. '-o', os.path.join(outdir, 'fd.json')
  164. ])
  165. factory_data = read_json(os.path.join(outdir, 'fd.json'))
  166. self.assertEqual(factory_data.get('version'), 1)
  167. self.assertEqual(factory_data.get('sn'), 'SN:12345678')
  168. self.assertEqual(factory_data.get('vendor_id'), 0x127F)
  169. self.assertEqual(factory_data.get('product_id'), 0xABCD)
  170. self.assertEqual(factory_data.get('vendor_name'), 'Nordic Semiconductor ASA')
  171. self.assertEqual(factory_data.get('product_name'), 'Lock Gen2')
  172. self.assertEqual(factory_data.get('part_number'), 'PCA10056')
  173. self.assertEqual(factory_data.get('product_url'), 'https://example.com/lock')
  174. self.assertEqual(factory_data.get('product_label'), 'Lock')
  175. self.assertEqual(factory_data.get('date'), '2022-07-20')
  176. self.assertEqual(factory_data.get('hw_ver'), 101)
  177. self.assertEqual(factory_data.get('hw_ver_str'), 'v1.1')
  178. self.assertEqual(factory_data.get('dac_key'), bytes_to_json(DAC_RAW_KEY))
  179. self.assertEqual(factory_data.get('dac_cert'), bytes_to_json(DAC_DER_CERT))
  180. self.assertEqual(factory_data.get('pai_cert'), bytes_to_json(PAI_DER_CERT))
  181. self.assertEqual(factory_data.get('spake2_it'), 2000)
  182. self.assertEqual(factory_data.get('spake2_salt'), base64_to_json('U1BBS0UyUCBLZXkgU2FsdA=='))
  183. self.assertEqual(factory_data.get('spake2_verifier'), base64_to_json(
  184. 'WN0SgEXLfUN19BbJqp6qn4pS69EtdNLReIMZwv/CIM0ECMP7ytiAJ7txIYJ0Ovlha/'
  185. 'rQ3E+88mj3qaqqnviMaZzG+OyXEdSocDIT9ZhmkTCgWwERaHz4Vdh3G37RT6kqbw=='))
  186. self.assertEqual(factory_data.get('discriminator'), 0xFED)
  187. self.assertEqual(factory_data.get('passcode'), 13243546)
  188. self.assertEqual(factory_data.get('rd_uid'), 'hex:91a9c12a7c80700a31ddcfa7fce63e44')
  189. self.assertEqual(factory_data.get('enable_key'), 'hex:00112233445566778899aabbccddeeff')
  190. self.assertEqual(factory_data.get('product_finish'), 2)
  191. self.assertEqual(factory_data.get('primary_color'), 11)
  192. self.assertEqual(factory_data.get('user'), {'name': 'product_name', 'version': 123, 'revision': '0x123'})
  193. subprocess.check_call(['python3', os.path.join(TOOLS_DIR, 'nrfconnect_generate_partition.py'),
  194. '-i', os.path.join(outdir, 'fd.json'),
  195. '-o', os.path.join(outdir, 'fd'),
  196. '--offset', "0xfb000",
  197. '--size', "0x1000",
  198. '--raw'
  199. ])
  200. def test_generate_spake2p_verifier_default(self):
  201. with tempfile.TemporaryDirectory() as outdir:
  202. write_file(os.path.join(outdir, 'DAC_key.der'), DAC_DER_KEY)
  203. write_file(os.path.join(outdir, 'DAC_cert.der'), DAC_DER_CERT)
  204. write_file(os.path.join(outdir, 'PAI_cert.der'), PAI_DER_CERT)
  205. subprocess.check_call(['python3', os.path.join(TOOLS_DIR, 'generate_nrfconnect_chip_factory_data.py'),
  206. '-s', os.path.join(TOOLS_DIR, 'nrfconnect_factory_data.schema'),
  207. '--sn', 'SN:12345678',
  208. '--vendor_id', '0x127F',
  209. '--product_id', '0xABCD',
  210. '--vendor_name', 'Nordic Semiconductor ASA',
  211. '--product_name', 'Lock',
  212. '--date', '2022-07-20',
  213. '--hw_ver', '101',
  214. '--hw_ver_str', 'v1.1',
  215. '--dac_key', os.path.join(outdir, 'DAC_key.der'),
  216. '--dac_cert', os.path.join(outdir, 'DAC_cert.der'),
  217. '--pai_cert', os.path.join(outdir, 'PAI_cert.der'),
  218. '--spake2_it', '1000',
  219. '--spake2_salt', 'U1BBS0UyUCBLZXkgU2FsdA==',
  220. '--passcode', '20202021',
  221. '--discriminator', '0xFED',
  222. '--user', '{"name": "product_name", "version": 123, "revision": "0x123"}',
  223. '-o', os.path.join(outdir, 'fd.json')
  224. ])
  225. factory_data = read_json(os.path.join(outdir, 'fd.json'))
  226. self.assertEqual(factory_data.get('passcode'), None)
  227. self.assertEqual(factory_data.get('spake2_salt'),
  228. base64_to_json('U1BBS0UyUCBLZXkgU2FsdA=='))
  229. self.assertEqual(factory_data.get('spake2_it'), 1000)
  230. self.assertEqual(factory_data.get('spake2_verifier'), base64_to_json(
  231. 'uWFwqugDNGiEck/po7KHwwMwwqZgN10XuyBajPGuyzUEV/'
  232. 'iree4lOrao5GuwnlQ65CJzbeUB49s31EH+NEkg0JVI5MGCQGMMT/SRPFNRODm3wH/MBiehuFc6FJ/NH6Rmzw=='))
  233. self.assertEqual(factory_data.get('user'), {'name': 'product_name', 'version': 123, 'revision': '0x123'})
  234. subprocess.check_call(['python3', os.path.join(TOOLS_DIR, 'nrfconnect_generate_partition.py'),
  235. '-i', os.path.join(outdir, 'fd.json'),
  236. '-o', os.path.join(outdir, 'fd'),
  237. '--offset', "0xfb000",
  238. '--size', "0x1000",
  239. '--raw'
  240. ])
  241. def test_generate_onboarding_codes(self):
  242. with tempfile.TemporaryDirectory() as outdir:
  243. write_file(os.path.join(outdir, 'DAC_key.der'), DAC_DER_KEY)
  244. write_file(os.path.join(outdir, 'DAC_cert.der'), DAC_DER_CERT)
  245. write_file(os.path.join(outdir, 'PAI_cert.der'), PAI_DER_CERT)
  246. subprocess.check_call(['python3', os.path.join(TOOLS_DIR, 'generate_nrfconnect_chip_factory_data.py'),
  247. '-s', os.path.join(TOOLS_DIR, 'nrfconnect_factory_data.schema'),
  248. '--include_passcode',
  249. '--sn', 'SN:12345678',
  250. '--vendor_id', '0x127F',
  251. '--product_id', '0xABCD',
  252. '--vendor_name', 'Nordic Semiconductor ASA',
  253. '--product_name', 'Lock Gen2',
  254. '--part_number', 'PCA10056',
  255. '--product_url', 'https://example.com/lock',
  256. '--product_label', 'Lock',
  257. '--date', '2022-07-20',
  258. '--hw_ver', '101',
  259. '--hw_ver_str', 'v1.1',
  260. '--dac_key', os.path.join(outdir, 'DAC_key.der'),
  261. '--dac_cert', os.path.join(outdir, 'DAC_cert.der'),
  262. '--pai_cert', os.path.join(outdir, 'PAI_cert.der'),
  263. '--spake2_it', '2000',
  264. '--spake2_salt', 'U1BBS0UyUCBLZXkgU2FsdA==',
  265. '--passcode', '13243546',
  266. '--spake2_verifier', ('WN0SgEXLfUN19BbJqp6qn4pS69EtdNLReIMZwv/CIM0ECMP7ytiAJ7txIYJ0Ovlha/'
  267. 'rQ3E+88mj3qaqqnviMaZzG+OyXEdSocDIT9ZhmkTCgWwERaHz4Vdh3G37RT6kqbw=='),
  268. '--discriminator', '0xFED',
  269. '--rd_uid', '91a9c12a7c80700a31ddcfa7fce63e44',
  270. '--enable_key', '00112233445566778899aabbccddeeff',
  271. '--user', '{"name": "product_name", "version": 123, "revision": "0x123"}',
  272. '-o', os.path.join(outdir, 'fd.json'),
  273. '--generate_onboarding'
  274. ])
  275. self.assertTrue(exists(os.path.join(outdir, 'fd.txt')))
  276. self.assertTrue(exists(os.path.join(outdir, 'fd.png')))
  277. with open(os.path.join(outdir, 'fd.txt'), 'r') as onboarding_code_file:
  278. onboarding = onboarding_code_file.readlines()
  279. self.assertEqual(onboarding[0][:-1], TEST_MANUAL_CODE)
  280. self.assertEqual(onboarding[1], TEST_QR_CODE)
  281. if __name__ == '__main__':
  282. unittest.main()