mirror of
https://github.com/eclipse-mosquitto/mosquitto.git
synced 2026-05-21 12:24:29 +02:00
125 lines
4.4 KiB
Python
Executable file
125 lines
4.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# Test whether an anonymous connection is correctly denied.
|
|
|
|
from mosq_test_helper import *
|
|
|
|
def write_config1(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("max_connections 10\n") # So the file isn't completely empty
|
|
|
|
def write_config2(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("port %d\n" % (port))
|
|
|
|
def write_config3(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("listener %d\n" % (port))
|
|
|
|
def write_config4(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("port %d\n" % (port))
|
|
f.write("allow_anonymous true\n")
|
|
|
|
def write_config5(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("listener %d\n" % (port))
|
|
f.write("allow_anonymous true\n")
|
|
|
|
def write_config6(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("allow_anonymous false\n")
|
|
|
|
def write_config7(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("allow_anonymous true\n")
|
|
|
|
def write_config8(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("allow_anonymous false\n")
|
|
f.write("listener %d\n" % (port))
|
|
f.write("listener_allow_anonymous true\n")
|
|
|
|
def write_config9(filename, port):
|
|
with open(filename, 'w') as f:
|
|
f.write("allow_anonymous true\n")
|
|
f.write("listener %d\n" % (port))
|
|
f.write("listener_allow_anonymous false\n")
|
|
|
|
|
|
def do_test(use_conf, write_config, expect_success):
|
|
port = mosq_test.get_port()
|
|
if write_config is not None:
|
|
conf_file = os.path.basename(__file__).replace('.py', '.conf')
|
|
write_config(conf_file, port)
|
|
|
|
broker = mosq_test.start_broker(filename=os.path.basename(__file__), use_conf=use_conf, port=port)
|
|
|
|
try:
|
|
for proto_ver in [4, 5]:
|
|
rc = 1
|
|
connect_packet = mqtt_packets.gen_connect("connect-anon-test-%d" % (proto_ver), proto_ver=proto_ver)
|
|
|
|
if proto_ver == 5:
|
|
if expect_success == True:
|
|
connack_packet = mqtt_packets.gen_connack(rc=0, proto_ver=proto_ver)
|
|
else:
|
|
connack_packet = mqtt_packets.gen_connack(rc=mqtt5_rc.NOT_AUTHORIZED, proto_ver=proto_ver, properties=None)
|
|
else:
|
|
if expect_success == True:
|
|
connack_packet = mqtt_packets.gen_connack(rc=0, proto_ver=proto_ver)
|
|
else:
|
|
connack_packet = mqtt_packets.gen_connack(rc=5, proto_ver=proto_ver)
|
|
|
|
|
|
sock = mosq_test.do_client_connect(connect_packet, connack_packet, port=port)
|
|
sock.close()
|
|
rc = 0
|
|
except mosq_test.TestError:
|
|
pass
|
|
finally:
|
|
if write_config is not None:
|
|
os.remove(conf_file)
|
|
pass
|
|
mosq_test.terminate_broker(broker)
|
|
if mosq_test.wait_for_subprocess(broker):
|
|
print("broker not terminated")
|
|
if rc == 0: rc=1
|
|
if rc:
|
|
print(mosq_test.broker_log(broker))
|
|
print("proto_ver=%d" % (proto_ver))
|
|
exit(rc)
|
|
|
|
|
|
# No config file - allow_anonymous should be true
|
|
do_test(use_conf=False, write_config=None, expect_success=True)
|
|
|
|
# Config file but no listener - allow_anonymous should be true
|
|
do_test(use_conf=True, write_config=write_config1, expect_success=True)
|
|
|
|
# Config file with "port" - allow_anonymous should be false
|
|
do_test(use_conf=True, write_config=write_config2, expect_success=False)
|
|
|
|
# Config file with "listener" - allow_anonymous should be false
|
|
do_test(use_conf=True, write_config=write_config3, expect_success=False)
|
|
|
|
# Config file with "port" - allow_anonymous explicitly true
|
|
do_test(use_conf=True, write_config=write_config4, expect_success=True)
|
|
|
|
# Config file with "listener" - allow_anonymous explicitly true
|
|
do_test(use_conf=True, write_config=write_config5, expect_success=True)
|
|
|
|
# Config file without "listener" - allow_anonymous explicitly false
|
|
do_test(use_conf=True, write_config=write_config6, expect_success=False)
|
|
|
|
# Config file without "listener" - allow_anonymous explicitly true
|
|
do_test(use_conf=True, write_config=write_config7, expect_success=True)
|
|
|
|
# Config file with "listener" - allow_anonymous explicitly false and listener_allow_anonymous explicitly true
|
|
do_test(use_conf=True, write_config=write_config8, expect_success=True)
|
|
|
|
# Config file with "listener" - allow_anonymous explicitly true and listener_allow_anonymous explicitly false
|
|
do_test(use_conf=True, write_config=write_config9, expect_success=False)
|
|
|
|
exit(0)
|