Add test for duplicate CONNECT.

pull/1203/head
Roger A. Light 7 years ago
parent ec1178806c
commit dab6452a1d

@ -196,6 +196,7 @@ int handle__connect(struct mosquitto_db *db, struct mosquitto *context)
/* Don't accept multiple CONNECT commands. */
if(context->state != mosq_cs_new){
log__printf(NULL, MOSQ_LOG_NOTICE, "Bad client %s sending multiple CONNECT messages.", context->id);
rc = MOSQ_ERR_PROTOCOL;
goto handle_connect_error;
}

@ -0,0 +1,32 @@
#!/usr/bin/env python
# Test whether a duplicate CONNECT is rejected. MQTT v5
from mosq_test_helper import *
rc = 1
keepalive = 10
connect_packet = mosq_test.gen_connect("connect-test", keepalive=keepalive, proto_ver=5)
connack_packet = mosq_test.gen_connack(rc=0, proto_ver=5)
port = mosq_test.get_port()
broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
try:
sock = mosq_test.do_client_connect(connect_packet, connack_packet, port=port)
sock.settimeout(3)
sock.send(connect_packet)
data = sock.recv(1)
if len(data) == 0:
rc = 0
except socket.error:
rc = 0
finally:
broker.terminate()
broker.wait()
(stdo, stde) = broker.communicate()
if rc:
print(stde)
exit(rc)

@ -0,0 +1,32 @@
#!/usr/bin/env python
# Test whether a duplicate CONNECT is rejected.
from mosq_test_helper import *
rc = 1
keepalive = 10
connect_packet = mosq_test.gen_connect("connect-test", keepalive=keepalive)
connack_packet = mosq_test.gen_connack(rc=0)
port = mosq_test.get_port()
broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
try:
sock = mosq_test.do_client_connect(connect_packet, connack_packet, port=port)
sock.settimeout(3)
sock.send(connect_packet)
data = sock.recv(1)
if len(data) == 0:
rc = 0
except socket.error:
rc = 0
finally:
broker.terminate()
broker.wait()
(stdo, stde) = broker.communicate()
if rc:
print(stde)
exit(rc)

@ -18,6 +18,8 @@ ptest : test-compile
test : test-compile 01 02 03 04 05 06 07 08 09 10 11 12
01 :
./01-connect-duplicate.py
./01-connect-duplicate-v5.py
./01-connect-success.py
./01-connect-success-v5.py
./01-connect-invalid-protonum.py

@ -7,6 +7,8 @@ import sys
max_running = 10
tests = [
#(ports required, 'path'),
(1, './01-connect-duplicate.py'),
(1, './01-connect-duplicate-v5.py'),
(1, './01-connect-success.py'),
(1, './01-connect-success-v5.py'),
(1, './01-connect-invalid-protonum.py'),

Loading…
Cancel
Save