1 files changed, 216 insertions, 0 deletions
diff --git a/metis/ccnx/forwarder/metis/io/test/test_metis_TcpTunnel.c b/metis/ccnx/forwarder/metis/io/test/test_metis_TcpTunnel.c
new file mode 100644
index 00000000..2d86fb17
--- /dev/null
+++ b/metis/ccnx/forwarder/metis/io/test/test_metis_TcpTunnel.c
@@ -0,0 +1,216 @@
+/*
+ * 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.
+ */
+
+
+// Include the file(s) containing the functions to be tested.
+// This permits internal static functions to be visible to this Test Framework.
+#include "../metis_TcpTunnel.c"
+#include <LongBow/unit-test.h>
+
+#include <parc/algol/parc_SafeMemory.h>
+
+#include <ccnx/forwarder/metis/testdata/metis_TestDataV0.h>
+
+// so we can see packet events
+#include "../../processor/test/testrig_MockTap.h"
+
+// inet_pton
+#include <arpa/inet.h>
+
+#include <fcntl.h>
+
+#ifndef INPORT_ANY
+#define INPORT_ANY 0
+#endif
+
+LONGBOW_TEST_RUNNER(metis_TcpTunnel)
+{
+    // The following Test Fixtures will run their corresponding Test Cases.
+    // Test Fixtures are run in the order specified, but all tests should be idempotent.
+    // Never rely on the execution order of tests or share state between them.
+    LONGBOW_RUN_TEST_FIXTURE(Global);
+    LONGBOW_RUN_TEST_FIXTURE(Local);
+}
+
+// The Test Runner calls this function once before any Test Fixtures are run.
+LONGBOW_TEST_RUNNER_SETUP(metis_TcpTunnel)
+{
+    return LONGBOW_STATUS_SUCCEEDED;
+}
+
+// The Test Runner calls this function once after all the Test Fixtures are run.
+LONGBOW_TEST_RUNNER_TEARDOWN(metis_TcpTunnel)
+{
+    return LONGBOW_STATUS_SUCCEEDED;
+}
+
+typedef struct test_data {
+    MetisForwarder *metis;
+    MetisDispatcher *dispatcher;
+
+    int serverSocket;
+    struct sockaddr_in serverAddr;
+    struct sockaddr_in localAddr;
+
+    CPIAddress *localCpiAddress;
+    CPIAddress *remoteCpiAddress;
+
+    MetisIoOperations *tunnelOps;
+} TestData;
+
+static void
+listenToInet(TestData *data)
+{
+    int fd = socket(PF_INET, SOCK_STREAM, 0);
+    assertFalse(fd < 0, "error on bind: (%d) %s", errno, strerror(errno));
+
+    // Set non-blocking flag
+    int flags = fcntl(fd, F_GETFL, NULL);
+    assertTrue(flags != -1, "fcntl failed to obtain file descriptor flags (%d)\n", errno);
+    int failure = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
+    assertFalse(failure, "fcntl failed to set file descriptor flags (%d)\n", errno);
+
+    failure = bind(fd, (struct sockaddr *) &data->serverAddr, sizeof(struct sockaddr_in));
+    assertFalse(failure, "error on bind: (%d) %s", errno, strerror(errno));
+
+    failure = listen(fd, 16);
+    assertFalse(failure, "error on listen: (%d) %s", errno, strerror(errno));
+
+    data->serverSocket = fd;
+    socklen_t x = sizeof(data->serverAddr);
+    getsockname(fd, (struct sockaddr *) &data->serverAddr, &x);
+}
+
+
+LONGBOW_TEST_FIXTURE(Global)
+{
+    LONGBOW_RUN_TEST_CASE(Global, metisTcpTunnel_Create);
+    LONGBOW_RUN_TEST_CASE(Global, metisTcpTunnel_Create_ConnectionStartsDown);
+    LONGBOW_RUN_TEST_CASE(Global, metisTcpTunnel_Create_UpStateAfterAccept);
+}
+
+LONGBOW_TEST_FIXTURE_SETUP(Global)
+{
+    memset(&testTap, 0, sizeof(testTap));
+
+    TestData *data = malloc(sizeof(TestData));
+    memset(data, 0, sizeof(TestData));
+
+    data->metis = metisForwarder_Create(NULL);
+    data->serverAddr.sin_family = PF_INET;
+    data->serverAddr.sin_port = INPORT_ANY;
+    inet_pton(AF_INET, "127.0.0.1", &(data->serverAddr.sin_addr));
+
+    data->localAddr.sin_family = PF_INET;
+    data->localAddr.sin_addr.s_addr = INADDR_ANY;
+    data->localAddr.sin_port = INPORT_ANY;
+
+    listenToInet(data);
+
+    data->localCpiAddress = cpiAddress_CreateFromInet(&data->localAddr);
+    data->remoteCpiAddress = cpiAddress_CreateFromInet(&data->serverAddr);
+
+    data->dispatcher = metisForwarder_GetDispatcher(data->metis);
+
+    longBowTestCase_SetClipBoardData(testCase, data);
+
+    return LONGBOW_STATUS_SUCCEEDED;
+}
+
+LONGBOW_TEST_FIXTURE_TEARDOWN(Global)
+{
+    TestData *data = longBowTestCase_GetClipBoardData(testCase);
+
+    cpiAddress_Destroy(&data->localCpiAddress);
+    cpiAddress_Destroy(&data->remoteCpiAddress);
+
+    close(data->serverSocket);
+    metisForwarder_Destroy(&data->metis);
+    free(data);
+
+    if (parcSafeMemory_ReportAllocation(STDOUT_FILENO) != 0) {
+        return LONGBOW_STATUS_MEMORYLEAK;
+    }
+    return LONGBOW_STATUS_SUCCEEDED;
+}
+
+LONGBOW_TEST_CASE(Global, metisTcpTunnel_Create)
+{
+    TestData *data = longBowTestCase_GetClipBoardData(testCase);
+
+    data->tunnelOps = metisTcpTunnel_Create(data->metis, data->localCpiAddress, data->remoteCpiAddress);
+    assertNotNull(data->tunnelOps, "Got null IO operations for the tunnel");
+    data->tunnelOps->destroy(&data->tunnelOps);
+}
+
+
+LONGBOW_TEST_CASE(Global, metisTcpTunnel_Create_ConnectionStartsDown)
+{
+    TestData *data = longBowTestCase_GetClipBoardData(testCase);
+
+    data->tunnelOps = metisTcpTunnel_Create(data->metis, data->localCpiAddress, data->remoteCpiAddress);
+    assertFalse(data->tunnelOps->isUp(data->tunnelOps), "Connection is not down on start");
+    data->tunnelOps->destroy(&data->tunnelOps);
+}
+
+LONGBOW_TEST_CASE(Global, metisTcpTunnel_Create_UpStateAfterAccept)
+{
+    TestData *data = longBowTestCase_GetClipBoardData(testCase);
+
+    data->tunnelOps = metisTcpTunnel_Create(data->metis, data->localCpiAddress, data->remoteCpiAddress);
+
+    // run for a milli second
+    metisDispatcher_RunDuration(data->dispatcher, &((struct timeval) { 0, 1000 }));
+
+    // we should be able to accept
+    struct sockaddr_in clientAddr;
+    socklen_t clientAddrLength = sizeof(clientAddr);
+
+    int clientSocket = accept(data->serverSocket, (struct sockaddr *) &clientAddr, &clientAddrLength);
+    assertFalse(clientSocket < 0, "error on accept: (%d) %s", errno, strerror(errno));
+
+    // run for a milli second
+    metisDispatcher_RunDuration(data->dispatcher, &((struct timeval) { 0, 1000 }));
+
+    assertTrue(data->tunnelOps->isUp(data->tunnelOps), "Connection is not up after accept");
+    data->tunnelOps->destroy(&data->tunnelOps);
+}
+
+
+LONGBOW_TEST_FIXTURE(Local)
+{
+}
+
+LONGBOW_TEST_FIXTURE_SETUP(Local)
+{
+    return LONGBOW_STATUS_SUCCEEDED;
+}
+
+LONGBOW_TEST_FIXTURE_TEARDOWN(Local)
+{
+    if (parcSafeMemory_ReportAllocation(STDOUT_FILENO) != 0) {
+        return LONGBOW_STATUS_MEMORYLEAK;
+    }
+    return LONGBOW_STATUS_SUCCEEDED;
+}
+
+int
+main(int argc, char *argv[])
+{
+    LongBowRunner *testRunner = LONGBOW_TEST_RUNNER_CREATE(metis_TcpTunnel);
+    int exitStatus = longBowMain(argc, argv, testRunner, NULL);
+    longBowTestRunner_Destroy(&testRunner);
+    exit(exitStatus);
+}