clone from danieldin95

core/tcp/main.c

@@ -0,0 +1,32 @@
+/*
+ * Copyright (c) 2021-2022 OpenLAN Inc.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 3 as
+ * published by the Free Software Foundation.
+ *
+ */
+
+#include <stdio.h>
+#include "socket.h"
+
+int main(int argc, char *argv[]) {
+    char *addr = NULL;
+    int port = 9090;
+
+    if (argc > 2) {
+        addr = argv[1];
+        sscanf(argv[2], "%d", &port);
+    } else if (argc > 1) {
+        sscanf(argv[1], "%d", &port);
+    }
+
+    if (addr == NULL) {
+        printf("Listen on %d!\n", port);
+        start_tcp_server(port);
+    } else {
+        printf("Connect to %s:%d\n", addr, port);
+        start_tcp_client(addr, port);
+    }
+    return 0;
+}