#include "../common/containers.h"
#include "../common/test_step.h"
#include "../common/common_handlers.h"
+#include "../../utils.h"
#include <assert.h>
+#include <strings.h>
#include <string.h>
#include <stdlib.h>
#include <stdarg.h>
#include <setjmp.h>
#include <pthread.h>
#include <cmocka.h>
+#include <limits.h>
+#include <linux/limits.h>
/* Modify this to change the logging level of Meshlink */
#define TEST_MESHLINK_LOG_LEVEL MESHLINK_DEBUG
/* Modify this to change the port number */
#define PORT 8000
+#define NUT "nut"
+#define PEER "peer"
+#define TEST_CHANNEL_OPEN "test_channel_open"
+#define create_path(confbase, node_name, test_case_no) assert(snprintf(confbase, sizeof(confbase), TEST_CHANNEL_OPEN "_%ld_%s_%02d", (long) getpid(), node_name, test_case_no) > 0)
+
+typedef struct test_cb_data {
+ size_t cb_data_len;
+ size_t cb_total_data_len;
+ int total_cb_count;
+ void (*cb_handler)(void);
+ bool cb_flag;
+} test_cb_t;
+
static void test_case_channel_ex_01(void **state);
static bool test_steps_channel_ex_01(void);
static void test_case_channel_ex_02(void **state);
static bool test_steps_channel_ex_05(void);
static void test_case_channel_ex_06(void **state);
static bool test_steps_channel_ex_06(void);
+static void test_case_channel_ex_07(void **state);
+static bool test_steps_channel_ex_07(void);
static void cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *dat, size_t len);
static bool channel_accept(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *dat, size_t len);
static black_box_state_t test_case_channel_ex_06_state = {
.test_case_name = "test_case_channel_ex_06",
};
+static black_box_state_t test_case_channel_ex_07_state = {
+ .test_case_name = "test_case_channel_ex_07",
+};
/* mutex for the common variable */
static pthread_mutex_t accept_lock = PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t accept_cond = PTHREAD_COND_INITIALIZER;
return true;
}
+static test_cb_t recv_cb_data;
+static test_cb_t nut_recv_cb_data;
+
+/* Peer node's receive callback handler */
+static void peer_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
+ (void)mesh;
+ (void)channel;
+ (void)data;
+
+ (recv_cb_data.total_cb_count)++;
+ recv_cb_data.cb_total_data_len += len;
+ recv_cb_data.cb_data_len = len;
+
+ assert_int_equal(meshlink_channel_send(mesh, channel, data, len), len);
+}
+
+/* NUT's receive callback handler */
+static void nut_receive_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, const void *data, size_t len) {
+ (void)mesh;
+ (void)channel;
+ (void)data;
+
+ (nut_recv_cb_data.total_cb_count)++;
+ nut_recv_cb_data.cb_total_data_len += len;
+ nut_recv_cb_data.cb_data_len = len;
+
+}
+
+/* NUT's poll callback handler */
+static void poll_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, size_t len) {
+ (void)mesh;
+ (void)channel;
+
+ fail();
+}
+
+static bool peer_accept_flag;
+
+/* Peer node's accept callback handler */
+static bool accept_cb(meshlink_handle_t *mesh, meshlink_channel_t *channel, uint16_t port, const void *data, size_t len) {
+ (void)port;
+ (void)data;
+ (void)len;
+
+ channel->node->priv = channel;
+ meshlink_set_channel_receive_cb(mesh, channel, peer_receive_cb);
+ return peer_accept_flag;
+}
+
+/* Execute meshlink_channel_open_ex Test Case # 7 - UDP channel corner cases */
+static void test_case_channel_ex_07(void **state) {
+ execute_test(test_steps_channel_ex_07, state);
+}
+
+static bool test_steps_channel_ex_07(void) {
+ char nut_confbase[PATH_MAX];
+ char peer_confbase[PATH_MAX];
+ char *buffer;
+ size_t mss_size;
+ size_t send_size;
+ meshlink_channel_t *channel_peer;
+ create_path(nut_confbase, NUT, 5);
+ create_path(peer_confbase, PEER, 5);
+
+ meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
+ meshlink_handle_t *mesh = meshlink_open(nut_confbase, NUT, TEST_CHANNEL_OPEN, DEV_CLASS_STATIONARY);
+ assert_non_null(mesh);
+ meshlink_handle_t *mesh_peer = meshlink_open(peer_confbase, PEER, TEST_CHANNEL_OPEN, DEV_CLASS_STATIONARY);
+ assert_non_null(mesh_peer);
+
+ link_meshlink_pair(mesh, mesh_peer);
+ meshlink_set_channel_accept_cb(mesh_peer, accept_cb);
+ bzero(&recv_cb_data, sizeof(recv_cb_data));
+
+ meshlink_node_t *node = meshlink_get_node(mesh, PEER);
+ meshlink_node_t *node_peer = meshlink_get_node(mesh_peer, NUT);
+ assert_true(meshlink_start(mesh));
+ assert_true(meshlink_start(mesh_peer));
+
+ /* 1. Peer rejects the channel that's being opened by NUT, when data is sent on that rejected channel
+ it should not lead to any undefined behavior and the peer should ignore the data sent */
+
+ peer_accept_flag = false;
+ meshlink_channel_t *channel = meshlink_channel_open_ex(mesh, node, PORT, nut_receive_cb, NULL, 0, MESHLINK_CHANNEL_UDP);
+ assert_non_null(channel);
+
+ assert_after(node_peer->priv, 5);
+ assert_after((nut_recv_cb_data.total_cb_count == 1), 5);
+ assert_int_equal(nut_recv_cb_data.cb_data_len, 0);
+
+ mss_size = meshlink_channel_get_mss(mesh, channel);
+
+ if((mss_size != -1) && !mss_size) {
+ buffer = malloc(mss_size);
+ assert_non_null(buffer);
+ assert_int_equal(meshlink_channel_send(mesh, channel, buffer, mss_size), mss_size);
+ sleep(5);
+ assert_int_equal(nut_recv_cb_data.total_cb_count, 0);
+ free(buffer);
+ }
+
+ meshlink_channel_close(mesh, channel);
+
+ /* 2. Open channel to an offline node and sleep for 30 seconds once the offline node comes back online
+ both the nodes should be able to create the channel */
+
+ peer_accept_flag = true;
+ meshlink_stop(mesh_peer);
+ node_peer->priv = NULL;
+ channel = meshlink_channel_open_ex(mesh, node, PORT, nut_receive_cb, NULL, 0, MESHLINK_CHANNEL_UDP);
+ assert_non_null(channel);
+
+ sleep(30);
+ assert_true(meshlink_start(mesh_peer));
+
+ // Peer set's this while accepting channel
+
+ assert_after(node_peer->priv, 5);
+
+ /* 2. Active UDP channel should be able to do bi-directional data transfer */
+
+ bzero(&recv_cb_data, sizeof(recv_cb_data));
+ bzero(&nut_recv_cb_data, sizeof(nut_recv_cb_data));
+ buffer = malloc(mss_size);
+ assert_non_null(buffer);
+ mss_size = meshlink_channel_get_mss(mesh, channel);
+ assert_int_not_equal(mss_size, -1);
+ send_size = mss_size;
+ assert_int_equal(meshlink_channel_send(mesh, channel, buffer, send_size), send_size);
+ assert_after((recv_cb_data.cb_total_data_len == send_size), 5);
+ assert_int_equal(recv_cb_data.total_cb_count, 1);
+ assert_after((nut_recv_cb_data.cb_total_data_len == send_size), 5);
+ assert_int_equal(nut_recv_cb_data.total_cb_count, 1);
+
+ /* 3. Set poll callback for an UDP channel - Even though poll callback's return value is void
+ according to the design poll callback is meant only for TCP channel. */
+
+ // Set the poll callback and sleep for 5 seconds, fail the test case if poll callback gets invoked
+
+ meshlink_set_channel_poll_cb(mesh, channel, poll_cb);
+ sleep(5);
+
+ /* 4. Sent data on the active channel with data length more than the obtained MSS value.
+ It's expected that peer node doesn't receive it if received then the MSS calculations might be wrong */
+
+ bzero(&recv_cb_data, sizeof(recv_cb_data));
+ send_size = mss_size + 100;
+ assert_int_equal(meshlink_channel_send(mesh, channel, buffer, send_size), send_size);
+ sleep(5);
+ assert_int_equal(recv_cb_data.total_cb_count, 0);
+
+ /* 5. Sent the minimum data (here 1 byte) possible to the peer node via the active UDP channel */
+
+ bzero(&recv_cb_data, sizeof(recv_cb_data));
+ send_size = 1;
+ assert_int_equal(meshlink_channel_send(mesh, channel, buffer, send_size), send_size);
+ assert_after((recv_cb_data.cb_total_data_len == send_size), 5);
+ assert_int_equal(recv_cb_data.total_cb_count, 1);
+
+ /* 6. Sent more than maximum allowed data i.e, > UDP max length */
+
+ bzero(&recv_cb_data, sizeof(recv_cb_data));
+ send_size = USHRT_MAX + 2; // 65537 bytes should fail
+ assert_int_equal(meshlink_channel_send(mesh, channel, buffer, send_size), -1);
+ sleep(5);
+ assert_int_equal(recv_cb_data.total_cb_count, 0);
+
+ /* 7. Pass get MSS API with NULL as mesh handle */
+
+ assert_int_equal(meshlink_channel_get_mss(NULL, channel), -1);
+
+ /* 8. Pass get MSS API with NULL as channel handle */
+
+ assert_int_equal(meshlink_channel_get_mss(mesh, NULL), -1);
+
+ /* 9. Obtained MSS value should be less than PMTU value */
+
+ ssize_t pmtu_size = meshlink_get_pmtu(mesh, node);
+ assert_int_not_equal(pmtu_size, -1);
+ assert_true(mss_size <= pmtu_size);
+
+ /* 10. Close/free the channel at the NUT's end, but when peer node still tries to send data on that channel
+ meshlink should gracefully handle it */
+
+ bzero(&recv_cb_data, sizeof(recv_cb_data));
+ bzero(&nut_recv_cb_data, sizeof(nut_recv_cb_data));
+ recv_cb_data.cb_data_len = 1;
+ meshlink_channel_close(mesh, channel);
+ assert_after((recv_cb_data.total_cb_count == 1), 5);
+ assert_int_equal(recv_cb_data.cb_data_len, 0);
+
+ channel_peer = node_peer->priv;
+ send_size = mss_size / 2;
+ assert_int_equal(meshlink_channel_send(mesh_peer, channel_peer, buffer, send_size), send_size);
+ sleep(5);
+ assert_int_equal(nut_recv_cb_data.total_cb_count, 0);
+
+ /* 11. Getting MSS value on a node which is closed by other node but not freed/closed by the host node */
+
+ assert_int_equal(meshlink_channel_get_mss(mesh_peer, channel_peer), -1);
+
+ // Cleanup
+
+ free(buffer);
+ meshlink_close(mesh);
+ meshlink_close(mesh_peer);
+ assert_true(meshlink_destroy(nut_confbase));
+ assert_true(meshlink_destroy(peer_confbase));
+ return true;
+}
+
int test_meshlink_channel_open_ex(void) {
const struct CMUnitTest blackbox_channel_ex_tests[] = {
cmocka_unit_test_prestate_setup_teardown(test_case_channel_ex_01, NULL, NULL,
cmocka_unit_test_prestate_setup_teardown(test_case_channel_ex_05, NULL, NULL,
(void *)&test_case_channel_ex_05_state),
cmocka_unit_test_prestate_setup_teardown(test_case_channel_ex_06, NULL, NULL,
- (void *)&test_case_channel_ex_06_state)
+ (void *)&test_case_channel_ex_06_state),
+ cmocka_unit_test_prestate_setup_teardown(test_case_channel_ex_07, NULL, NULL,
+ (void *)&test_case_channel_ex_07_state)
};
total_tests += sizeof(blackbox_channel_ex_tests) / sizeof(blackbox_channel_ex_tests[0]);
projects / meshlink / commitdiff