From: Jouni Malinen Date: Sun, 3 Feb 2019 23:48:43 +0000 (+0200) Subject: tests: Extend build_wsc_attr() to accept both bytes and str objects X-Git-Tag: hostap_2_8~447 X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=b7da11fd95a3a6130ce6cb0c76281bbea424b4ee;p=thirdparty%2Fhostap.git tests: Extend build_wsc_attr() to accept both bytes and str objects This is needed for python3. Signed-off-by: Jouni Malinen --- diff --git a/tests/hwsim/test_ap_wps.py b/tests/hwsim/test_ap_wps.py index 98597f1ed..970a8c203 100644 --- a/tests/hwsim/test_ap_wps.py +++ b/tests/hwsim/test_ap_wps.py @@ -6227,7 +6227,8 @@ def recv_wsc_msg(dev, opcode, msg_type): return msg, msg['wsc_attrs'], msg['raw_attrs'] def build_wsc_attr(attr, payload): - return struct.pack('>HH', attr, len(payload)) + payload + _payload = payload if type(payload) == bytes else payload.encode() + return struct.pack('>HH', attr, len(_payload)) + _payload def build_attr_msg_type(msg_type): return build_wsc_attr(ATTR_MSG_TYPE, struct.pack('B', msg_type))