diff options
Diffstat (limited to 'metis/ccnx/forwarder/metis/io/test/test_metis_UdpListener.c')
-rw-r--r-- | metis/ccnx/forwarder/metis/io/test/test_metis_UdpListener.c | 458 |
1 files changed, 458 insertions, 0 deletions
diff --git a/metis/ccnx/forwarder/metis/io/test/test_metis_UdpListener.c b/metis/ccnx/forwarder/metis/io/test/test_metis_UdpListener.c new file mode 100644 index 00000000..96dfc4fd --- /dev/null +++ b/metis/ccnx/forwarder/metis/io/test/test_metis_UdpListener.c @@ -0,0 +1,458 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +/** + */ + +/* + * hard-coded in port 49009 on localhost + */ + +// Include the file(s) containing the functions to be tested. +// This permits internal static functions to be visible to this Test Framework. +#include "../metis_UdpListener.c" +#include <LongBow/unit-test.h> +#include <parc/algol/parc_SafeMemory.h> + +#include <ccnx/forwarder/metis/tlv/metis_Tlv.h> +#include <ccnx/forwarder/metis/testdata/metis_TestDataV0.h> +#include <ccnx/forwarder/metis/testdata/metis_TestDataV1.h> + +// for inet_pton +#include <arpa/inet.h> + +#include <signal.h> + +// ======================================================== + +struct test_set { + CPIAddress *listenAddress; + MetisForwarder *metis; + MetisListenerOps *ops; +} TestSet; + +static void +setupInetListener() +{ + struct sockaddr_in addr; + + memset(&addr, 0, sizeof(addr)); + addr.sin_family = AF_INET; + addr.sin_port = htons(49009); + inet_pton(AF_INET, "127.0.0.1", &(addr.sin_addr)); + + TestSet.metis = metisForwarder_Create(NULL); + metisLogger_SetLogLevel(metisForwarder_GetLogger(TestSet.metis), MetisLoggerFacility_IO, PARCLogLevel_Debug); + + TestSet.ops = metisUdpListener_CreateInet(TestSet.metis, addr); + TestSet.listenAddress = cpiAddress_CreateFromInet(&addr); + + // crank the handle + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); +} + +static void +teardownListener() +{ + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + cpiAddress_Destroy(&TestSet.listenAddress); + TestSet.ops->destroy(&TestSet.ops); + metisForwarder_Destroy(&TestSet.metis); +} + +static bool +setupInet6Listener() +{ + struct sockaddr_in6 addr; + + memset(&addr, 0, sizeof(addr)); + addr.sin6_family = AF_INET6; + addr.sin6_port = htons(49009); + + // "::1" is the ipv6 loopback address + int ok = inet_pton(AF_INET6, "::1", &(addr.sin6_addr)); + if (ok > 0) { + TestSet.metis = metisForwarder_Create(NULL); + metisLogger_SetLogLevel(metisForwarder_GetLogger(TestSet.metis), MetisLoggerFacility_IO, PARCLogLevel_Debug); + + TestSet.ops = metisUdpListener_CreateInet6(TestSet.metis, addr); + TestSet.listenAddress = cpiAddress_CreateFromInet6(&addr); + + // crank the handle + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + return true; + } + return false; +} + +struct sigaction save_sigchld; +struct sigaction save_sigpipe; + +static void +blockSigChild() +{ + struct sigaction ignore_action; + ignore_action.sa_handler = SIG_IGN; + sigemptyset(&ignore_action.sa_mask); + ignore_action.sa_flags = 0; + + sigaction(SIGCHLD, NULL, &save_sigchld); + sigaction(SIGPIPE, NULL, &save_sigpipe); + + sigaction(SIGCHLD, &ignore_action, NULL); + sigaction(SIGPIPE, &ignore_action, NULL); +} + +static void +unblockSigChild() +{ + sigaction(SIGCHLD, &save_sigchld, NULL); + sigaction(SIGPIPE, &save_sigpipe, NULL); +} + +// ======================================================== + +LONGBOW_TEST_RUNNER(metis_UdpListener) +{ + LONGBOW_RUN_TEST_FIXTURE(Global_Inet); + // XXX: Udp code has issues. It should check return values from calls. + // There are bugs in the UDP code that need to be fixed. These are shown in + // this test. The code needs to be fixed first. + //LONGBOW_RUN_TEST_FIXTURE(Global_Inet6); + LONGBOW_RUN_TEST_FIXTURE(Local); +} + +// The Test Runner calls this function once before any Test Fixtures are run. +LONGBOW_TEST_RUNNER_SETUP(metis_UdpListener) +{ + parcMemory_SetInterface(&PARCSafeMemoryAsPARCMemory); + return LONGBOW_STATUS_SUCCEEDED; +} + +// The Test Runner calls this function once after all the Test Fixtures are run. +LONGBOW_TEST_RUNNER_TEARDOWN(metis_UdpListener) +{ + return LONGBOW_STATUS_SUCCEEDED; +} + +// ================================================================================ + +LONGBOW_TEST_FIXTURE(Global_Inet) +{ + LONGBOW_RUN_TEST_CASE(Global_Inet, metisListenerUdp_CreateInet); + //XXX: this does not work anymore because we do not create the udp connection + //LONGBOW_RUN_TEST_CASE(Global_Inet, metisListenerUdp_Connect); + LONGBOW_RUN_TEST_CASE(Global_Inet, metisListenerUdp_SendPacket); +} + +LONGBOW_TEST_FIXTURE_SETUP(Global_Inet) +{ + setupInetListener(); + return LONGBOW_STATUS_SUCCEEDED; +} + +LONGBOW_TEST_FIXTURE_TEARDOWN(Global_Inet) +{ + teardownListener(); + uint32_t outstandingAllocations = parcSafeMemory_ReportAllocation(STDERR_FILENO); + if (outstandingAllocations != 0) { + printf("%s leaks memory by %d allocations\n", longBowTestCase_GetName(testCase), outstandingAllocations); + return LONGBOW_STATUS_MEMORYLEAK; + } + return LONGBOW_STATUS_SUCCEEDED; +} + +LONGBOW_TEST_CASE(Global_Inet, metisListenerUdp_CreateInet) +{ + // now verify that we are listening + // udp4 0 0 127.0.0.1.49009 *.* LISTEN + + blockSigChild(); + FILE *fp = popen("netstat -an -p udp", "r"); + assertNotNull(fp, "Got null opening netstat for reading"); + + char str[1035]; + bool found = false; + while (fgets(str, sizeof(str) - 1, fp) != NULL) { + if (strstr(str, "127.0.0.1.49009") != NULL) { + found = true; + break; + } + if (strstr(str, "127.0.0.1:49009") != NULL) { + found = true; + break; + } + } + + pclose(fp); + unblockSigChild(); + + if (!found) { + int ret = system("netstat -an -p udp"); + assertTrue(ret > -1, "Error on system call"); + } + + assertTrue(found, "Did not find 127.0.0.1.49009 in netstat output"); +} + +LONGBOW_TEST_CASE(Global_Inet, metisListenerUdp_Connect) +{ + int fd = socket(PF_INET, SOCK_DGRAM, 0); + assertFalse(fd < 0, "Error on socket: (%d) %s", errno, strerror(errno)); + + struct sockaddr_in serverAddress; + cpiAddress_GetInet(TestSet.listenAddress, &serverAddress); + + int failure = connect(fd, (struct sockaddr *) &serverAddress, sizeof(serverAddress)); + assertFalse(failure, "Error on connect: (%d) %s", errno, strerror(errno)); + + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + struct sockaddr_in connectAddress; + socklen_t connectAddressLength = sizeof(connectAddress); + failure = getsockname(fd, (struct sockaddr *) &connectAddress, &connectAddressLength); + assertFalse(failure, "Error on getsockname: (%d) %s", errno, strerror(errno)); + assertTrue(connectAddressLength == sizeof(struct sockaddr_in), + "connectAddressLength wrong size, expected %zu got %u", + sizeof(struct sockaddr_in), connectAddressLength); + + // Unlike TCP, we need to actually send something + ssize_t nwritten = write(fd, metisTestDataV0_EncodedInterest, sizeof(metisTestDataV0_EncodedInterest)); + assertTrue(nwritten == sizeof(metisTestDataV0_EncodedInterest), "Error on write expected %zu got %zd", sizeof(metisTestDataV0_EncodedInterest), nwritten); + + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + // make sure its in the connection table + MetisConnectionTable *table = metisForwarder_GetConnectionTable(TestSet.metis); + CPIAddress *remote = cpiAddress_CreateFromInet(&connectAddress); + MetisAddressPair *pair = metisAddressPair_Create(TestSet.listenAddress, remote); + const MetisConnection *conn = metisConnectionTable_FindByAddressPair(table, pair); + assertNotNull(conn, "Did not find connection in connection table"); + + cpiAddress_Destroy(&remote); + metisAddressPair_Release(&pair); + + close(fd); +} + +LONGBOW_TEST_CASE(Global_Inet, metisListenerUdp_SendPacket) +{ + int fd = socket(PF_INET, SOCK_DGRAM, 0); + assertFalse(fd < 0, "Error on socket: (%d) %s", errno, strerror(errno)); + + struct sockaddr_in serverAddress; + cpiAddress_GetInet(TestSet.listenAddress, &serverAddress); + + int failure = connect(fd, (struct sockaddr *) &serverAddress, sizeof(serverAddress)); + assertFalse(failure, "Error on connect socket %d: (%d) %s\n", fd, errno, strerror(errno)); + + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + ssize_t write_length = write(fd, metisTestDataV0_InterestWithName, sizeof(metisTestDataV0_InterestWithName)); + assertFalse(write_length < 0, "Error on write: (%d) %s", errno, strerror(errno)); + assertTrue(write_length == sizeof(metisTestDataV0_InterestWithName), "Got partial write, expected %zu got %zd", sizeof(metisTestDataV0_InterestWithName), write_length); + + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + close(fd); +} + +// ================================================================================ + +LONGBOW_TEST_FIXTURE(Global_Inet6) +{ + LONGBOW_RUN_TEST_CASE(Global_Inet6, metisListenerUdp_CreateInet6); + LONGBOW_RUN_TEST_CASE(Global_Inet6, metisListenerUdp_Connect); + LONGBOW_RUN_TEST_CASE(Global_Inet6, metisListenerUdp_SendPacket); +} + +LONGBOW_TEST_FIXTURE_SETUP(Global_Inet6) +{ + bool success = setupInet6Listener(); + if (!success) { + return LONGBOW_STATUS_SKIPPED; + } + return LONGBOW_STATUS_SUCCEEDED; +} + +LONGBOW_TEST_FIXTURE_TEARDOWN(Global_Inet6) +{ + teardownListener(); + uint32_t outstandingAllocations = parcSafeMemory_ReportAllocation(STDERR_FILENO); + if (outstandingAllocations != 0) { + printf("%s leaks memory by %d allocations\n", longBowTestCase_GetName(testCase), outstandingAllocations); + return LONGBOW_STATUS_MEMORYLEAK; + } + return LONGBOW_STATUS_SUCCEEDED; +} + +LONGBOW_TEST_CASE(Global_Inet6, metisListenerUdp_CreateInet6) +{ + // now verify that we are listening + // udp6 0 0 ::1.49009 *.* + + blockSigChild(); + FILE *fp = popen("netstat -an -p udp", "r"); + assertNotNull(fp, "Got null opening netstat for reading"); + + char str[1035]; + bool found = false; + while (fgets(str, sizeof(str) - 1, fp) != NULL) { + //printf("%s\n", str); + if (strstr(str, "::1.49009") != NULL) { + found = true; + break; + } + if (strstr(str, "::1:49009") != NULL) { + found = true; + break; + } + } + + pclose(fp); + unblockSigChild(); + + if (!found) { + int ret = system("netstat -an -p udp"); + assertTrue(ret != -1, "Error on system() call"); + } + + assertTrue(found, "Did not find 127.0.0.1.49009 in netstat output"); +} + +LONGBOW_TEST_CASE(Global_Inet6, metisListenerUdp_Connect) +{ + int fd = socket(PF_INET6, SOCK_DGRAM, 0); + assertFalse(fd < 0, "Error on socket: (%d) %s", errno, strerror(errno)); + + struct sockaddr_in6 serverAddress; + cpiAddress_GetInet6(TestSet.listenAddress, &serverAddress); + + int failure = connect(fd, (struct sockaddr *) &serverAddress, sizeof(serverAddress)); + assertFalse(failure, "Error on connect: (%d) %s", errno, strerror(errno)); + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + struct sockaddr_in6 connectAddress; + socklen_t connectAddressLength = sizeof(connectAddress); + failure = getsockname(fd, (struct sockaddr *) &connectAddress, &connectAddressLength); + assertFalse(failure, "Error on getsockname: (%d) %s", errno, strerror(errno)); + assertTrue(connectAddressLength == sizeof(struct sockaddr_in6), + "connectAddressLength wrong size, expected %zu got %u", + sizeof(struct sockaddr_in6), connectAddressLength); + + // Unlike TCP, we need to actually send something + ssize_t nwritten = write(fd, metisTestDataV1_Interest_NameA_Crc32c, sizeof(metisTestDataV1_Interest_NameA_Crc32c)); + assertTrue(nwritten == sizeof(metisTestDataV1_Interest_NameA_Crc32c), "Write failed, expected %zu got %zd", sizeof(metisTestDataV1_Interest_NameA_Crc32c), nwritten); + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + // make sure its in the connection table + MetisConnectionTable *table = metisForwarder_GetConnectionTable(TestSet.metis); + CPIAddress *remote = cpiAddress_CreateFromInet6(&connectAddress); + MetisAddressPair *pair = metisAddressPair_Create(TestSet.listenAddress, remote); + const MetisConnection *conn = metisConnectionTable_FindByAddressPair(table, pair); + assertNotNull(conn, "Did not find connection in connection table"); + + cpiAddress_Destroy(&remote); + metisAddressPair_Release(&pair); + + close(fd); +} + +LONGBOW_TEST_CASE(Global_Inet6, metisListenerUdp_SendPacket) +{ + int fd = socket(PF_INET6, SOCK_DGRAM, 0); + assertFalse(fd < 0, "Error on socket: (%d) %s", errno, strerror(errno)); + + struct sockaddr_in6 serverAddress; + cpiAddress_GetInet6(TestSet.listenAddress, &serverAddress); + + int failure = connect(fd, (struct sockaddr *) &serverAddress, sizeof(serverAddress)); + assertFalse(failure, "Error on connect socket %d: (%d) %s\n", fd, errno, strerror(errno)); + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + ssize_t write_length = write(fd, metisTestDataV1_Interest_NameA_Crc32c, sizeof(metisTestDataV1_Interest_NameA_Crc32c)); + assertFalse(write_length < 0, "Error on write: (%d) %s", errno, strerror(errno)); + assertTrue(write_length == sizeof(metisTestDataV1_Interest_NameA_Crc32c), + "Got partial write, expected %zu got %zd", + sizeof(metisTestDataV1_Interest_NameA_Crc32c), write_length); + + metisDispatcher_RunDuration(metisForwarder_GetDispatcher(TestSet.metis), &((struct timeval) { 0, 10000 })); + + close(fd); +} + +// ================================================================================ + +LONGBOW_TEST_FIXTURE(Local) +{ + LONGBOW_RUN_TEST_CASE(Local, _getInterfaceIndex); + LONGBOW_RUN_TEST_CASE(Local, _getListenAddress); + LONGBOW_RUN_TEST_CASE(Local, _getEncapType); + LONGBOW_RUN_TEST_CASE(Local, _getSocket); +} + +LONGBOW_TEST_FIXTURE_SETUP(Local) +{ + setupInetListener(); + return LONGBOW_STATUS_SUCCEEDED; +} + +LONGBOW_TEST_FIXTURE_TEARDOWN(Local) +{ + teardownListener(); + uint32_t outstandingAllocations = parcSafeMemory_ReportAllocation(STDERR_FILENO); + if (outstandingAllocations != 0) { + printf("%s leaks memory by %d allocations\n", longBowTestCase_GetName(testCase), outstandingAllocations); + return LONGBOW_STATUS_MEMORYLEAK; + } + return LONGBOW_STATUS_SUCCEEDED; +} + +LONGBOW_TEST_CASE(Local, _getInterfaceIndex) +{ + unsigned test = _getInterfaceIndex(TestSet.ops); + assertTrue(test > 0, "Unexpected interface index: %u", test); +} + +LONGBOW_TEST_CASE(Local, _getListenAddress) +{ + const CPIAddress *listenAddr = _getListenAddress(TestSet.ops); + assertNotNull(listenAddr, "Got null listen address"); +} + +LONGBOW_TEST_CASE(Local, _getEncapType) +{ + MetisEncapType type = _getEncapType(TestSet.ops); + assertTrue(type == METIS_ENCAP_UDP, "Unexpected address type, got %d expected %d", type, METIS_ENCAP_UDP); +} + +LONGBOW_TEST_CASE(Local, _getSocket) +{ + int fd = _getSocket(TestSet.ops); + assertTrue(fd > 0, "Unexpected socket, got %d, expected positive", fd); +} + + +// ================================================================================ + + +int +main(int argc, char *argv[]) +{ + LongBowRunner *testRunner = LONGBOW_TEST_RUNNER_CREATE(metis_UdpListener); + int exitStatus = longBowMain(argc, argv, testRunner, NULL); + longBowTestRunner_Destroy(&testRunner); + exit(exitStatus); +} |