02-subscribe-invalid-utf8.py 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. #!/usr/bin/env python3
  2. # Test whether a SUBSCRIBE to a topic with an invalid UTF-8 topic fails
  3. from mosq_test_helper import *
  4. def do_test(proto_ver):
  5. rc = 1
  6. mid = 53
  7. keepalive = 60
  8. connect_packet = mosq_test.gen_connect("subscribe-invalid-utf8", keepalive=keepalive, proto_ver=proto_ver)
  9. connack_packet = mosq_test.gen_connack(rc=0, proto_ver=proto_ver)
  10. subscribe_packet = mosq_test.gen_subscribe(mid, "invalid/utf8", 0, proto_ver=proto_ver)
  11. b = list(struct.unpack("B"*len(subscribe_packet), subscribe_packet))
  12. b[13] = 0 # Topic should never have a 0x0000
  13. subscribe_packet = struct.pack("B"*len(b), *b)
  14. suback_packet = mosq_test.gen_suback(mid, 0, proto_ver=proto_ver)
  15. port = mosq_test.get_port()
  16. broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
  17. try:
  18. time.sleep(0.5)
  19. sock = mosq_test.do_client_connect(connect_packet, connack_packet, port=port)
  20. if proto_ver == 4:
  21. mosq_test.do_send_receive(sock, subscribe_packet, b"", "suback")
  22. else:
  23. disconnect_packet = mosq_test.gen_disconnect(proto_ver=5, reason_code = mqtt5_rc.MQTT_RC_MALFORMED_PACKET)
  24. mosq_test.do_send_receive(sock, subscribe_packet, disconnect_packet, "suback")
  25. rc = 0
  26. sock.close()
  27. except mosq_test.TestError:
  28. pass
  29. finally:
  30. broker.terminate()
  31. broker.wait()
  32. (stdo, stde) = broker.communicate()
  33. if rc:
  34. print(stde.decode('utf-8'))
  35. print("proto_ver=%d" % (proto_ver))
  36. exit(rc)
  37. do_test(proto_ver=4)
  38. do_test(proto_ver=5)
  39. exit(0)