1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import sys
import json
import asyncio
import logging
import zigpy_znp.types as t
from zigpy_znp.api import ZNP
from zigpy_znp.config import CONFIG_SCHEMA
from zigpy_znp.exceptions import SecurityError, CommandNotRecognized
from zigpy_znp.types.nvids import NWK_NVID_TABLES, ExNvIds, OsalNvIds
from zigpy_znp.tools.common import ClosableFileType, setup_parser
LOGGER = logging.getLogger(__name__)
async def nvram_read(znp: ZNP):
data = {}
data["LEGACY"] = {}
# Legacy items need to be handled first, since they are named
for nwk_nvid in OsalNvIds:
if nwk_nvid == OsalNvIds.INVALID_INDEX:
continue
# Tables span ranges of items. Naming them properly is useful.
if nwk_nvid in NWK_NVID_TABLES:
start = nwk_nvid
end = NWK_NVID_TABLES[nwk_nvid]
for offset in range(0, end - start):
key = f"{nwk_nvid.name}+{offset}"
try:
value = await znp.nvram.osal_read(
nwk_nvid + offset, item_type=t.Bytes
)
except SecurityError:
LOGGER.warning("Read disallowed for %s", key)
continue
except KeyError:
break
LOGGER.info("%s = %r", key, value.hex())
data["LEGACY"][key] = value.hex()
else:
try:
value = await znp.nvram.osal_read(nwk_nvid, item_type=t.Bytes)
except KeyError:
continue
except SecurityError:
LOGGER.warning("Read disallowed for %s", nwk_nvid)
continue
LOGGER.info("%s = %r", nwk_nvid, value.hex())
data["LEGACY"][nwk_nvid.name] = value.hex()
for nvid in ExNvIds:
# Skip the LEGACY items, we did them above
if nvid == ExNvIds.LEGACY:
continue
for sub_id in range(2**16):
try:
value = await znp.nvram.read(
item_id=nvid, sub_id=sub_id, item_type=t.Bytes
)
except CommandNotRecognized:
# CC2531 only supports the legacy NVRAM interface, even on Z-Stack 3
return data
except KeyError:
# Once a read fails, no later reads will succeed
break
LOGGER.info("%s[0x%04X] = %r", nvid.name, sub_id, value.hex())
data.setdefault(nvid.name, {})[f"0x{sub_id:04X}"] = value.hex()
return data
async def main(argv):
parser = setup_parser("Backup a radio's NVRAM")
parser.add_argument(
"--output", "-o", type=ClosableFileType("w"), help="Output file", default="-"
)
args = parser.parse_args(argv)
with args.output as f:
znp = ZNP(CONFIG_SCHEMA({"device": {"path": args.serial}}))
await znp.connect()
obj = await nvram_read(znp)
await znp.disconnect()
f.write(json.dumps(obj, indent=4) + "\n")
if __name__ == "__main__":
asyncio.run(main(sys.argv[1:])) # pragma: no cover
|
