tools/hexnet: Implement ipv6 support.

src/tools/hexnet.cpp

@@ -19,6 +19,7 @@
 #include <cstdint>
 #include <cctype>
 #include <sockpp/tcp_acceptor.h>
+#include <sockpp/tcp6_acceptor.h>
 #include <thread>
 #include <mutex>
 #include <csignal>
@@ -40,8 +41,11 @@ int64_t mtu = 1500;
 std::string host;
 in_port_t port;
 sockpp::tcp_socket* tcpSocket;
+sockpp::tcp6_socket* tcp6Socket;
 sockpp::tcp_acceptor tcpAcceptor;
+sockpp::tcp6_acceptor tcp6Acceptor;
 std::mutex tcpSocketMutex;
+std::mutex tcp6SocketMutex;
 std::mutex consoleMutex;
 // used for coordinated graceful exit across threads
 bool exitProgram = false;
@@ -50,12 +54,20 @@ void signalHandler(int signal) {
     exitProgram = true;
     // if still waiting for incoming connection, stop waiting
     tcpAcceptor.shutdown();
+    tcp6Acceptor.shutdown();
     // tell sockpp to close TCP socket if open because it blocks when trying
     // to read and there is no data
-    if (*tcpSocket) {
+    
+    if (tcpSocket != nullptr && *tcpSocket) {
         // Intentionally not using the mutex here
+        std::cout << "test\n";
         tcpSocket->shutdown(SHUT_RD);
     }
+
+    if (tcp6Socket != nullptr && *tcp6Socket) {
+        // Intentionally not using the mutex here
+        tcp6Socket->shutdown(SHUT_RD);
+    }
     //TODO: figure out if - and how - this applies to UDP
 
     // Priority is to finish up all unfinished business that can be finished up.
@@ -64,6 +76,7 @@ void signalHandler(int signal) {
     consoleMutex.lock();
     std::cerr << "Received signal " << signal << ", shutting down." << std::endl;
     consoleMutex.unlock();
+    std::exit(signal);
 }
 
 void readFromTCPSocket(sockpp::tcp_socket* socket, int64_t mtu) {
@@ -86,10 +99,48 @@ void readFromTCPSocket(sockpp::tcp_socket* socket, int64_t mtu) {
     consoleMutex.unlock();
 }
 
+void readFromTCP6Socket(sockpp::tcp6_socket* socket, int64_t mtu){
+    ssize_t numBytes;
+    uint8_t buffer[mtu];
+    tcp6SocketMutex.lock();
+    while (!exitProgram && (numBytes = socket->read(buffer, sizeof(buffer))) > 0) {
+        tcp6SocketMutex.unlock();
+        consoleMutex.lock();
+        for (ssize_t i=0; i<numBytes; i++) {
+            std::cout << std::hex << std::setfill('0') << std::setw(2) << (short) buffer[i];
+        }
+        std::cout.flush();
+        consoleMutex.unlock();
+        tcp6SocketMutex.lock();
+    }
+    tcp6SocketMutex.unlock();
+    consoleMutex.lock();
+    std::cerr << std::endl << "Connection closed." << std::endl;
+    consoleMutex.unlock();
+}
+
+void writeToTCPSocket(sockpp::tcp_socket* socket){
+    while (!exitProgram) {
+       //TODO: Implement locking/unlocking/threading shenanigans
+    }
+    consoleMutex.lock();
+    std::cerr << std::endl << "Connection closed." << std::endl;
+    consoleMutex.unlock();
+}
+
+void writeToTCP6Socket(sockpp::tcp6_socket* socket){
+    while (!exitProgram) {
+       //TODO: Implement locking/unlocking/threading shenanigans
+    }
+    consoleMutex.lock();
+    std::cerr << std::endl << "Connection closed." << std::endl;
+    consoleMutex.unlock();
+}
+
 int main(int argc, char* argv[]){
 
-    signal(SIGINT, signalHandler);
-    signal(SIGTERM, signalHandler);
+    std::signal(SIGINT, signalHandler);
+    std::signal(SIGTERM, signalHandler);
 
     std::vector<CLI::Flag> flags;
     flags.push_back(CLI::Flag('4', "ipv4", "use IPv4, defaults to both when -4 and -6 are omitted, otherwise uses what is specified"));
@@ -140,14 +191,43 @@ int main(int argc, char* argv[]){
     port = (in_port_t) std::stoi(cliParser.getArgument(0).value);
 
     if (listenMode) {
-        if (ipv6) {
-            std::cerr << "IPv6 support is not implented yet." << std::endl;
-            return EXIT_UNIMPLEMENTED;
-        }
         if (udp) {
             std::cerr << "UDP support is not implemented yet." << std::endl;
             return EXIT_UNIMPLEMENTED;
         }
+
+        if (ipv6) {
+            //std::cerr << "IPv6 support is not implented yet." << std::endl;
+            //return EXIT_UNIMPLEMENTED;
+            std::cerr << "Listening on port " << port << "." << std::endl;
+            sockpp::socket_initializer socketInitializer;
+            tcp6Acceptor = sockpp::tcp6_acceptor(port);
+
+            if(!tcp6Acceptor){
+                std::cerr << "Error while creating TCP6 acceptor: " << tcp6Acceptor.last_error_str() << std::endl;
+                return EXIT_RUNTIME;
+            }
+
+            sockpp::inet6_address peer;
+            tcp6Socket = new sockpp::tcp6_socket();
+            *tcp6Socket = tcp6Acceptor.accept(&peer);
+
+            std::cerr << "Incoming connection from " << peer << std::endl;
+
+            if (!(*tcp6Socket)) {
+                std::cerr << "Error on incoming connection: " << tcp6Acceptor.last_error_str() << std::endl;
+                delete tcp6Socket;
+                return EXIT_RUNTIME;
+            }
+
+            std::thread threadReadFromTCP6 = std::thread(readFromTCP6Socket, tcp6Socket, mtu);
+            threadReadFromTCP6.join();
+
+            delete tcp6Socket;
+            return EXIT_SUCCESS;
+        }
+
+        if(ipv4){
             std::cerr << "Listening on port " << port << "." << std::endl;
 
             sockpp::socket_initializer socketInitializer;
@@ -170,11 +250,12 @@ int main(int argc, char* argv[]){
                 return EXIT_RUNTIME;
             }
 
-        std::thread threadReadTCP = std::thread(readFromTCPSocket, tcpSocket, mtu);
-        threadReadTCP.join();
+            std::thread threadReadFromTCP = std::thread(readFromTCPSocket, tcpSocket, mtu);
+            threadReadFromTCP.join();
 
             delete tcpSocket;
             return EXIT_SUCCESS;
+        }
         
     } else {
         std::cerr << "Client mode is not implemented yet." << std::endl;