mirror of
https://github.com/cesanta/mongoose.git
synced 2024-12-28 15:40:23 +08:00
commit
11303e0dc1
@ -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;
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user