async def test_tuya_data_conversion(): """Test tuya conversion from Data to ztype and reverse.""" assert Data([4, 0, 0, 1, 39]).to_value(t.uint32_t) == 295 assert Data([4, 0, 0, 0, 220]).to_value(t.uint32_t) == 220 assert Data([4, 255, 255, 255, 236]).to_value(t.int32s) == -20 assert Data.from_value(t.uint32_t(295)) == [4, 0, 0, 1, 39] assert Data.from_value(t.uint32_t(220)) == [4, 0, 0, 0, 220] assert Data.from_value(t.int32s(-20)) == [4, 255, 255, 255, 236]
def test_subelement(): payload = b"\x00payload\xff" data = b"\x01\x00" + t.uint32_t(len(payload)).serialize() + payload extra = b"extra" e, rest = firmware.SubElement.deserialize(data + extra) assert rest == extra assert e.tag_id == firmware.ElementTagId.ECDSA_SIGNATURE assert e.data == payload assert len(e.data) == len(payload) assert e.serialize() == data
def test_legrand_container_unwrapping(image): # Unwrapped size prefix and 1 + 16 byte suffix data = ( t.uint32_t(len(image.serialize())).serialize() + image.serialize() + b"\x01" + b"abcdabcdabcdabcd" ) with pytest.raises(ValueError): firmware.parse_ota_image(data[:-1]) with pytest.raises(ValueError): firmware.parse_ota_image(b"\xFF" + data[1:]) img, rest = firmware.parse_ota_image(data) assert not rest assert img == image
def data(tag_id, payload=b""): r = t.uint16_t(tag_id).serialize() r += t.uint32_t(len(payload)).serialize() return r + payload
def data(elements_size=0): d = b"\x1e\xf1\xee\x0b\x00\x018\x00\x00\x00" d += b"|\x11\x01!rE!\x12\x02\x00EBL tradfri_light_basic\x00\x00\x00" d += b"\x00\x00\x00\x00\x00\x00" d += t.uint32_t(elements_size + 56).serialize() return d
def length(self): return t.uint32_t(len(self))
async def attr_write( # noqa: C901 app, listener, ieee, cmd, data, service, params, event_data): success = True dev = app.get_device(ieee=ieee) # Decode endpoint if params[p.EP_ID] is None or params[p.EP_ID] == "": params[p.EP_ID] = u.find_endpoint(dev, params[p.CLUSTER_ID]) if params[p.EP_ID] not in dev.endpoints: msg = f"Endpoint {params[p.EP_ID]} not found for '{ieee!r}" LOGGER.error(msg) raise Exception(msg) if params[p.CLUSTER_ID] not in dev.endpoints[params[p.EP_ID]].in_clusters: msg = "InCluster 0x{:04X} not found for '{}', endpoint {}".format( params[p.CLUSTER_ID], repr(ieee), params[p.EP_ID]) if params[p.CLUSTER_ID] in dev.endpoints[params[p.EP_ID]].out_clusters: msg = f'{cmd}: "Using" OutCluster. {msg}' LOGGER.warning(msg) if "warnings" not in event_data: event_data["warnings"] = [] event_data["warnings"].append(msg) else: LOGGER.error(msg) raise Exception(msg) cluster = dev.endpoints[params[p.EP_ID]].in_clusters[params[p.CLUSTER_ID]] # Prepare read and write lists attr_write_list = [] attr_read_list = [] # Decode attribute(s) # Currently only one attribute is possible, but the parameter # format could allow for multiple attributes for instance by # adding a split character such as ':' for attr_id, attr_type # and attr_value # Then the match should be in a loop # Decode attribute id # Could accept name for attribute, but extra code to check attr_id = params[p.ATTR_ID] attr_read_list.append(attr_id) # Read before write list compare_val = None if cmd == "attr_write": attr_type = params[p.ATTR_TYPE] attr_val_str = params[p.ATTR_VAL] # Type only needed for write if attr_type is None or attr_val_str is None: event_data["errors"].append( "attr_type and attr_val must be set for attr_write") else: # Convert attribute value (provided as a string) # to appropriate attribute value. # If the attr_type is not set, only read the attribute. attr_val = None if attr_type == 0x10: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.Bool(compare_val)) elif attr_type == 0x20: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint8_t(compare_val)) elif attr_type == 0x21: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint16_t(compare_val)) elif attr_type == 0x22: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint24_t(compare_val)) elif attr_type == 0x23: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint32_t(compare_val)) elif attr_type == 0x24: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint32_t(compare_val)) elif attr_type == 0x25: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint48_t(compare_val)) elif attr_type == 0x26: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint56_t(compare_val)) elif attr_type == 0x27: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.uint64_t(compare_val)) elif attr_type == 0x28: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int8_t(compare_val)) elif attr_type == 0x29: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int16_t(compare_val)) elif attr_type == 0x2A: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int24_t(compare_val)) elif attr_type == 0x2B: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int32_t(compare_val)) elif attr_type == 0x2C: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int32_t(compare_val)) elif attr_type == 0x2D: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int48_t(compare_val)) elif attr_type == 0x2E: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int56_t(compare_val)) elif attr_type == 0x2F: compare_val = u.str2int(attr_val_str) attr_val = f.TypeValue(attr_type, t.int64_t(compare_val)) elif attr_type <= 0x31 and attr_type >= 0x08: compare_val = u.str2int(attr_val_str) # uint, int, bool, bitmap and enum attr_val = f.TypeValue(attr_type, t.FixedIntType(compare_val)) elif attr_type in [0x41, 0x42]: # Octet string # Octet string requires length -> LVBytes compare_val = attr_val_str event_data["str"] = attr_val_str if type(attr_val_str) == str: attr_val_str = bytes(attr_val_str, "utf-8") if isinstance(attr_val_str, list): # Convert list to List of uint8_t attr_val_str = t.List[t.uint8_t]( [t.uint8_t(i) for i in attr_val_str]) attr_val = f.TypeValue(attr_type, t.LVBytes(attr_val_str)) if attr_val is not None: attr = f.Attribute(attr_id, value=attr_val) attr_write_list.append(attr) # Write list else: msg = ("attr_type {} not supported, " + "or incorrect parameters (attr_val={})").format( params[p.ATTR_TYPE], params[p.ATTR_VAL]) event_data["errors"].append(msg) LOGGER.debug(msg) LOGGER.debug( "ATTR TYPE %s, attr_val %s", params[p.ATTR_TYPE], params[p.ATTR_VAL], ) result_read = None if (params[p.READ_BEFORE_WRITE] or (len(attr_write_list) == 0) or (cmd != S.ATTR_WRITE)): LOGGER.debug("Request attr read %s", attr_read_list) result_read = await cluster_read_attributes( cluster, attr_read_list, manufacturer=params[p.MANF], tries=params[p.TRIES], ) LOGGER.debug("Reading attr result (attrs, status): %s", result_read) success = (len(result_read[1]) == 0) and (len(result_read[0]) == 1) # True if value that should be written is the equal to the read one write_is_equal = ((params[p.READ_BEFORE_WRITE]) and (len(attr_write_list) != 0) and ((attr_id in result_read[0]) and (result_read[0][attr_id] == compare_val))) event_data["write_is_equal"] = write_is_equal if write_is_equal and (cmd == "attr_write"): event_data["info"] = "Data read is equal to data to write" if (len(attr_write_list) != 0 and (not (params[p.READ_BEFORE_WRITE]) or params[p.WRITE_IF_EQUAL] or not (write_is_equal)) and cmd == "attr_write"): if result_read is not None: event_data["read_before"] = result_read result_read = None LOGGER.debug("Request attr write %s", attr_write_list) result_write = await cluster__write_attributes( cluster, attr_write_list, manufacturer=params[p.MANF], tries=params[p.TRIES], ) LOGGER.debug("Write attr status: %s", result_write) event_data["result_write"] = result_write success = False try: # LOGGER.debug("Write attr status: %s", result_write[0][0].status) success = result_write[0][0].status == f.Status.SUCCESS LOGGER.debug(f"Write success: {success}") except Exception as e: event_data["errors"].append(repr(e)) success = False # success = (len(result_write[1])==0) if params[p.READ_AFTER_WRITE]: LOGGER.debug(f"Request attr read {attr_read_list!r}") result_read = await cluster_read_attributes( cluster, attr_read_list, manufacturer=params[p.MANF], tries=params[p.TRIES], ) LOGGER.debug( f"Reading attr result (attrs, status): {result_read!r}") # read_is_equal = (result_read[0][attr_id] == compare_val) success = (success and (len(result_read[1]) == 0 and len(result_read[0]) == 1) and (result_read[0][attr_id] == compare_val)) if result_read is not None: event_data["result_read"] = result_read if len(result_read[1]) == 0: read_val = result_read[0][attr_id] else: msg = (f"Result: {result_read[1]}" + f" - Attribute {attr_id} not in read {result_read!r}") LOGGER.warning(msg) if "warnings" not in event_data: event_data["warnings"] = [] event_data["warnings"].append(msg) success = False else: read_val = None event_data["success"] = success # Write value to provided state or state attribute if params[p.STATE_ID] is not None: if len(result_read[1]) == 0 and len(result_read[0]) == 1: # No error and one result for id, val in result_read[0].items(): if params[p.STATE_ATTR] is not None: LOGGER.debug( "Set state %s[%s] -> %s from attr_id %s", params[p.STATE_ID], params[p.STATE_ATTR], val, id, ) else: LOGGER.debug( "Set state %s -> %s from attr_id %s", params[p.STATE_ID], val, id, ) u.set_state( listener._hass, params[p.STATE_ID], val, key=params[p.STATE_ATTR], allow_create=params[p.ALLOW_CREATE], ) LOGGER.debug("STATE is set") if success and params[p.CSV_FILE] is not None: fields = [] if params[p.CSV_LABEL] is not None: attr_name = params[p.CSV_LABEL] else: try: attr_name = cluster.attributes.get(attr_id, (str(attr_id), None))[0] except Exception: attr_name = attr_id fields.append(dt_util.utcnow().isoformat()) fields.append(attr_name) fields.append(read_val) fields.append("0x%04X" % (attr_id)), fields.append("0x%04X" % (cluster.cluster_id)), fields.append(cluster.endpoint.endpoint_id) fields.append(str(cluster.endpoint.device.ieee)) fields.append(("0x%04X" % (params[p.MANF])) if params[p.MANF] is not None else "") u.append_to_csvfile( fields, "csv", params[p.CSV_FILE], f"{attr_name}={read_val}", listener=listener, ) importlib.reload(u) if "result_read" in event_data and not u.isJsonable( event_data["result_read"]): event_data["result_read"] = repr(event_data["result_read"]) # For internal use return result_read
async def znp_restore(app, listener, ieee, cmd, data, service, event_data, params): """Restore ZNP network information.""" if u.get_radiotype(app) != u.RadioType.ZNP: msg = f"'{cmd}' is only available for ZNP" LOGGER.debug(msg) raise Exception(msg) # Get/set parameters # command_data (data): # counter_increment (defaults to 2500) counter_increment = u.str2int(data) if type(counter_increment) != int: counter_increment = 2500 counter_increment = t.uint32_t(counter_increment) from datetime import datetime current_datetime = datetime.now().strftime("_%Y%m%d_%H%M%S") # Safety: backup current configuration await znp_backup(app, listener, ieee, cmd, current_datetime, service) # Import stuff we need for restoring from zigpy_znp.tools.network_restore import json_backup_to_zigpy_state from zigpy_znp.tools.common import validate_backup_json from os import path import json # Set name with regards to local path fname = path.dirname(__file__) + "/local/nwk_backup.json" LOGGER.info("Restore from '%s'", fname) event_data["restore_file"] = fname # Read backup file f = open(fname) backup = json.load(f) f.close() # validate the backup file LOGGER.info("Validating backup contents") validate_backup_json(backup) LOGGER.info("Backup contents validated") network_info, node_info = json_backup_to_zigpy_state(backup) network_info.network_key.tx_counter += counter_increment # Network already formed in HA # app._znp.startup(force_form=True) # Write back information from backup LOGGER.info("Writing to device") await app._znp.write_network_info(network_info=network_info, node_info=node_info) # LOGGER.debug("List of attributes/methods in app %s", dir(app)) LOGGER.debug("List of attributes/methods in znp %s", dir(app._znp)) # Shutdown znp? LOGGER.info( "Write done, call pre_shutdown(). Restart the device/HA after this.") await app._znp.pre_shutdown() LOGGER.info("pre_shutdown() Done.")