Add client support for outgoing maximum packet size.
Roger A. Light
7 years ago
parent
1877f8a326
commit
b9b8e0ff2a
@ -0,0 +1,59 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
# Test whether a client publishing an oversize packet correctly.
|
||||||
|
# The client should try to publish a message that is too big, then the one below which is ok.
|
||||||
|
# It should also try to subscribe with a too large topic
|
||||||
|
|
||||||
|
from mosq_test_helper import *
|
||||||
|
|
||||||
|
port = mosq_test.get_lib_port()
|
||||||
|
|
||||||
|
rc = 1
|
||||||
|
keepalive = 60
|
||||||
|
connect_packet = mosq_test.gen_connect("publish-qos0-test", keepalive=keepalive, proto_ver=5)
|
||||||
|
props = mqtt5_props.gen_uint32_prop(mqtt5_props.PROP_MAXIMUM_PACKET_SIZE, 30)
|
||||||
|
connack_packet = mosq_test.gen_connack(rc=0, proto_ver=5, properties=props)
|
||||||
|
|
||||||
|
bad_publish_packet = mosq_test.gen_publish("pub/test", qos=0, payload="0123456789012345678", proto_ver=5)
|
||||||
|
publish_packet = mosq_test.gen_publish("pub/test", qos=0, payload="012345678901234567", proto_ver=5)
|
||||||
|
|
||||||
|
disconnect_packet = mosq_test.gen_disconnect()
|
||||||
|
|
||||||
|
|
||||||
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||||
|
sock.settimeout(10)
|
||||||
|
sock.bind(('', port))
|
||||||
|
sock.listen(5)
|
||||||
|
|
||||||
|
client_args = sys.argv[1:]
|
||||||
|
env = dict(os.environ)
|
||||||
|
env['LD_LIBRARY_PATH'] = '../../lib:../../lib/cpp'
|
||||||
|
try:
|
||||||
|
pp = env['PYTHONPATH']
|
||||||
|
except KeyError:
|
||||||
|
pp = ''
|
||||||
|
env['PYTHONPATH'] = '../../lib/python:'+pp
|
||||||
|
client = mosq_test.start_client(filename=sys.argv[1].replace('/', '-'), cmd=client_args, env=env, port=port)
|
||||||
|
|
||||||
|
try:
|
||||||
|
(conn, address) = sock.accept()
|
||||||
|
conn.settimeout(10)
|
||||||
|
|
||||||
|
if mosq_test.expect_packet(conn, "connect", connect_packet):
|
||||||
|
conn.send(connack_packet)
|
||||||
|
|
||||||
|
if mosq_test.expect_packet(conn, "publish", publish_packet):
|
||||||
|
if mosq_test.expect_packet(conn, "disconnect", disconnect_packet):
|
||||||
|
rc = 0
|
||||||
|
|
||||||
|
conn.close()
|
||||||
|
finally:
|
||||||
|
client.terminate()
|
||||||
|
client.wait()
|
||||||
|
if rc:
|
||||||
|
(stdo, stde) = client.communicate()
|
||||||
|
print(stde)
|
||||||
|
sock.close()
|
||||||
|
|
||||||
|
exit(rc)
|
||||||
@ -0,0 +1,72 @@
|
|||||||
|
#include <stdbool.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <mosquitto.h>
|
||||||
|
|
||||||
|
static int run = -1;
|
||||||
|
static int sent_mid = -1;
|
||||||
|
|
||||||
|
void on_connect(struct mosquitto *mosq, void *obj, int rc)
|
||||||
|
{
|
||||||
|
if(rc){
|
||||||
|
exit(1);
|
||||||
|
}else{
|
||||||
|
rc = mosquitto_subscribe(mosq, NULL, "0123456789012345678901234567890", 0);
|
||||||
|
if(rc != MOSQ_ERR_OVERSIZE_PACKET){
|
||||||
|
printf("Fail on subscribe\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
rc = mosquitto_unsubscribe(mosq, NULL, "0123456789012345678901234567890");
|
||||||
|
if(rc != MOSQ_ERR_OVERSIZE_PACKET){
|
||||||
|
printf("Fail on unsubscribe\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
rc = mosquitto_publish(mosq, &sent_mid, "pub/test", strlen("0123456789012345678"), "0123456789012345678", 0, false);
|
||||||
|
if(rc != MOSQ_ERR_OVERSIZE_PACKET){
|
||||||
|
printf("Fail on publish 1\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
rc = mosquitto_publish(mosq, &sent_mid, "pub/test", strlen("012345678901234567"), "012345678901234567", 0, false);
|
||||||
|
if(rc != MOSQ_ERR_SUCCESS){
|
||||||
|
printf("Fail on publish 2\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void on_publish(struct mosquitto *mosq, void *obj, int mid)
|
||||||
|
{
|
||||||
|
if(mid == sent_mid){
|
||||||
|
mosquitto_disconnect(mosq);
|
||||||
|
run = 0;
|
||||||
|
}else{
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char *argv[])
|
||||||
|
{
|
||||||
|
int rc;
|
||||||
|
struct mosquitto *mosq;
|
||||||
|
|
||||||
|
int port = atoi(argv[1]);
|
||||||
|
|
||||||
|
mosquitto_lib_init();
|
||||||
|
|
||||||
|
mosq = mosquitto_new("publish-qos0-test", true, NULL);
|
||||||
|
mosquitto_int_option(mosq, MOSQ_OPT_PROTOCOL_VERSION, MQTT_PROTOCOL_V5);
|
||||||
|
mosquitto_connect_callback_set(mosq, on_connect);
|
||||||
|
mosquitto_publish_callback_set(mosq, on_publish);
|
||||||
|
|
||||||
|
rc = mosquitto_connect(mosq, "localhost", port, 60);
|
||||||
|
|
||||||
|
while(run == -1){
|
||||||
|
rc = mosquitto_loop(mosq, -1, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
mosquitto_lib_cleanup();
|
||||||
|
return run;
|
||||||
|
}
|
||||||
@ -0,0 +1,43 @@
|
|||||||
|
#include <CUnit/CUnit.h>
|
||||||
|
#include <CUnit/Basic.h>
|
||||||
|
|
||||||
|
#include <mosquitto_internal.h>
|
||||||
|
#include <util_mosq.h>
|
||||||
|
|
||||||
|
|
||||||
|
static void TEST_maximum_packet_size(void)
|
||||||
|
{
|
||||||
|
struct mosquitto mosq;
|
||||||
|
int rc;
|
||||||
|
|
||||||
|
memset(&mosq, 0, sizeof(struct mosquitto));
|
||||||
|
|
||||||
|
mosq.maximum_packet_size = 5;
|
||||||
|
rc = mosquitto_publish(&mosq, NULL, "topic/oversize", strlen("payload"), "payload", 0, 0);
|
||||||
|
CU_ASSERT_EQUAL(rc, MOSQ_ERR_OVERSIZE_PACKET);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ========================================================================
|
||||||
|
* TEST SUITE SETUP
|
||||||
|
* ======================================================================== */
|
||||||
|
|
||||||
|
int init_publish_tests(void)
|
||||||
|
{
|
||||||
|
CU_pSuite test_suite = NULL;
|
||||||
|
|
||||||
|
test_suite = CU_add_suite("Publish", NULL, NULL);
|
||||||
|
if(!test_suite){
|
||||||
|
printf("Error adding CUnit Publish test suite.\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(0
|
||||||
|
|| !CU_add_test(test_suite, "v5: Maximum packet size", TEST_maximum_packet_size)
|
||||||
|
){
|
||||||
|
|
||||||
|
printf("Error adding Publish CUnit tests.\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue