Merge pull request #2354 from cesanta/tst

Add MQTT test for MIP
This commit is contained in:
Sergio R. Caprile 2023-08-21 17:26:24 -03:00 committed by GitHub
commit 11303e0dc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -14,7 +14,9 @@
#include <net/if_types.h>
#endif
#include <sys/ioctl.h>
#include "mongoose.c"
#include "driver_mock.c"
static int s_num_tests = 0;
@ -108,10 +110,11 @@ char *fetch(struct mg_mgr *mgr, const char *url, const char *fn_data) {
}
mg_mgr_poll(mgr, 0);
if (!post_reply.http_responses_received)
if (!post_reply.http_responses_received) {
return 0;
else
} else {
return (char *) post_reply.http_response;
}
}
// Returns server's HTTP response code
@ -127,14 +130,86 @@ int get_response_code(char *http_msg_raw) {
return http_status;
}
static void test_http_fetch(void) {
static void test_http_fetch(struct mg_mgr *mgr) {
char *http_feedback = (char *) "";
const bool ipv6 = 0;
if (ipv6) {
http_feedback = fetch(mgr, "ipv6.google.com",
"GET/ HTTP/1.0\r\nHost: ipv6.google.com\r\n\r\n");
} else {
http_feedback =
fetch(mgr, "http://cesanta.com",
"GET //robots.txt HTTP/1.0\r\nHost: cesanta.com\r\n\r\n");
}
ASSERT(http_feedback != NULL &&
*http_feedback != '\0'); // HTTP response received ?
int http_status = get_response_code(http_feedback);
// printf("Server response HTTP status code: %d\n",http_status);
ASSERT(http_status != 0);
ASSERT(http_status == 301); // OK: Permanently moved (HTTP->HTTPS redirect)
if (http_feedback) {
free(http_feedback);
http_feedback = 0;
}
}
static struct mg_connection *s_conn;
static char s_topic[16];
static void mqtt_fn(struct mg_connection *c, int ev, void *ev_data,
void *fn_data) {
if (ev == MG_EV_MQTT_OPEN) {
// MQTT connect is successful
struct mg_mqtt_opts sub_opts;
memset(&sub_opts, 0, sizeof(sub_opts));
sub_opts.topic = mg_str(mg_random_str(s_topic, sizeof(s_topic)));
sub_opts.qos = 1;
mg_mqtt_sub(c, &sub_opts);
struct mg_mqtt_opts pub_opts;
memset(&pub_opts, 0, sizeof(pub_opts));
pub_opts.topic = sub_opts.topic;
pub_opts.message = mg_str("hi");
pub_opts.qos = 1, pub_opts.retain = false;
mg_mqtt_pub(c, &pub_opts);
} else if (ev == MG_EV_MQTT_MSG) {
struct mg_mqtt_message *mm = (struct mg_mqtt_message *) ev_data;
if (mm->topic.len != strlen(s_topic) || strcmp(mm->topic.ptr, s_topic))
ASSERT(0);
if (mm->data.len != 2 || strcmp(mm->data.ptr, "hi")) ASSERT(0);
mg_mqtt_disconnect(c, NULL);
*(bool *) fn_data = true;
} else if (ev == MG_EV_CLOSE) {
s_conn = NULL;
}
}
static void test_mqtt_connsubpub(struct mg_mgr *mgr) {
const char *url = "mqtt://broker.hivemq.com:1883";
bool passed = false;
struct mg_mqtt_opts opts;
memset(&opts, 0, sizeof(opts));
opts.clean = true, opts.version = 4;
s_conn = mg_mqtt_connect(mgr, url, &opts, mqtt_fn, &passed);
ASSERT(s_conn != NULL);
for (int i = 0; i < 500 && s_conn != NULL && !s_conn->is_closing; i++) {
mg_mgr_poll(mgr, 0);
usleep(10000); // 10 ms. Slow down poll loop to ensure packets transit
}
ASSERT(passed);
mg_mgr_poll(mgr, 0);
}
int main(void) {
// Setup interface
const char *iface = "tap0"; // Network iface
const char *mac = "00:00:01:02:03:78"; // MAC address
#ifndef __OpenBSD__
const char* tuntap_device = "/dev/net/tun";
const char *tuntap_device = "/dev/net/tun";
#else
const char* tuntap_device = "/dev/tap0";
const char *tuntap_device = "/dev/tap0";
#endif
int fd = open(tuntap_device, O_RDWR);
struct ifreq ifr;
@ -147,7 +222,7 @@ static void test_http_fetch(void) {
abort(); // return EXIT_FAILURE;
}
#else
ifr.ifr_flags = (short)(IFF_UP | IFF_BROADCAST | IFF_MULTICAST);
ifr.ifr_flags = (short) (IFF_UP | IFF_BROADCAST | IFF_MULTICAST);
if (ioctl(fd, TUNSIFMODE, (void *) &ifr) < 0) {
MG_ERROR(("Failed to setup TAP interface: %s", ifr.ifr_name));
abort(); // return EXIT_FAILURE;
@ -177,7 +252,7 @@ static void test_http_fetch(void) {
#if MG_USING_DHCP == 1
#else
mif.ip = mg_htonl(MG_U32(192, 168, 32, 2)); // Triggering a network failure
mif.ip = mg_htonl(MG_U32(192, 168, 32, 2)); // Triggering a network failure
mif.mask = mg_htonl(MG_U32(255, 255, 255, 0));
mif.gw = mg_htonl(MG_U32(192, 168, 32, 1));
#endif
@ -189,61 +264,31 @@ static void test_http_fetch(void) {
MG_INFO(("Init done, starting main loop"));
// Stack initialization, Network configuration (DHCP lease, ...)
{
#if MG_USING_DHCP == 0
MG_INFO(("MIF configuration: Static IP"));
ASSERT(mif.ip != 0); // Check we have a satic IP assigned
mg_mgr_poll(&mgr, 100); // For initialisation
MG_INFO(("MIF configuration: Static IP"));
ASSERT(mif.ip != 0); // Check we have a satic IP assigned
mg_mgr_poll(&mgr, 100); // For initialisation
#else
MG_INFO(("MIF configuration: DHCP"));
MG_INFO(("Opened TAP interface: %s", iface));
ASSERT(!mif.ip); // Check we are set for DHCP
int pc = 500; // Timout on DHCP lease 500 ~ approx 5s (typical delay <1s)
while (((pc--) > 0) && !mif.ip) {
mg_mgr_poll(&mgr, 100);
usleep(10000); // 10 ms
}
if (!mif.ip) MG_ERROR(("No ip assigned (DHCP lease may have failed).\n"));
ASSERT(mif.ip); // We have an IP (lease or static)
MG_INFO(("MIF configuration: DHCP"));
ASSERT(!mif.ip); // Check we are set for DHCP
int pc = 500; // Timout on DHCP lease 500 ~ approx 5s (typical delay <1s)
while (((pc--) > 0) && !mif.ip) {
mg_mgr_poll(&mgr, 100);
usleep(10000); // 10 ms
}
if (!mif.ip) MG_ERROR(("No ip assigned (DHCP lease may have failed).\n"));
ASSERT(mif.ip); // We have an IP (lease or static)
#endif
}
// Simple HTTP fetch
{
char *http_feedback = (char *) "";
const bool ipv6 = 0;
if (ipv6) {
http_feedback = fetch(&mgr, "ipv6.google.com",
"GET/ HTTP/1.0\r\nHost: ipv6.google.com\r\n\r\n");
} else {
http_feedback =
fetch(&mgr, "http://cesanta.com",
"GET //robots.txt HTTP/1.0\r\nHost: cesanta.com\r\n\r\n");
}
ASSERT(http_feedback != NULL &&
*http_feedback != '\0'); // HTTP response received ?
int http_status = get_response_code(http_feedback);
// printf("Server response HTTP status code: %d\n",http_status);
ASSERT(http_status != 0);
ASSERT(http_status == 301); // OK: Permanently moved (HTTP->HTTPS redirect)
if (http_feedback) {
free(http_feedback);
http_feedback = 0;
}
}
// RUN TESTS
test_http_fetch(&mgr);
test_mqtt_connsubpub(&mgr);
printf("SUCCESS. Total tests: %d\n", s_num_tests);
// Clear
mg_mgr_free(&mgr);
mg_tcpip_free(&mif); // Release after mg_mgr
mg_tcpip_free(&mif); // Release after mg_mgr
ASSERT(mgr.conns == NULL); // Deconstruction OK
close(fd);
}
int main(void) {
test_http_fetch();
printf("SUCCESS. Total tests: %d\n", s_num_tests);
return 0;
}