From d80579448c317093e7a7c5a68bde5a6e249a6af4 Mon Sep 17 00:00:00 2001
From: zy <zhangyang@geedgenetworks.com>
Date: Fri, 12 Jan 2024 02:57:46 -0500
Subject: [PATCH] example update: wannat

---
 examples/c/statistics.c |  11 +++-
 examples/c/wannat.c     | 120 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 130 insertions(+), 1 deletion(-)
 create mode 100644 examples/c/wannat.c

diff --git a/examples/c/statistics.c b/examples/c/statistics.c
index 4f14306..049f36e 100644
--- a/examples/c/statistics.c
+++ b/examples/c/statistics.c
@@ -17,6 +17,15 @@ int main(int argc, char** argv)
         printf("statistics <net_id>\n");
         exit(0);
     }
+
+    int err = ZTS_ERR_OK;
+    char* storage_path = "./";
+
+    if ((err = zts_init_from_storage(storage_path)) != ZTS_ERR_OK) {
+        printf("Unable to start service, error = %d. Exiting.\n", err);
+        exit(1);
+    }
+
     long long int net_id = strtoull(argv[1], NULL, 16);   // At least 64 bits
 
     printf("Starting node...\n");
@@ -47,7 +56,7 @@ int main(int argc, char** argv)
     }
 
     printf("Waiting for address assignment from network\n");
-    int err = 0;
+    // int err = 0;
     while (! (err = zts_addr_is_assigned(net_id, ZTS_AF_INET))) {
         zts_util_delay(500);
     }
diff --git a/examples/c/wannat.c b/examples/c/wannat.c
new file mode 100644
index 0000000..8bc6df8
--- /dev/null
+++ b/examples/c/wannat.c
@@ -0,0 +1,120 @@
+/**
+ * libzt C API example
+ *
+ * Pingable node
+ */
+
+#include "ZeroTierSockets.h"
+
+#include <netinet/ip.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+void process_packet(char* buffer, int len)
+{
+    struct iphdr* iph = (struct iphdr*)buffer;
+
+    unsigned char bytes[4];
+    bytes[0] = iph->daddr & 0xFF;
+    bytes[1] = (iph->daddr >> 8) & 0xFF;
+    bytes[2] = (iph->daddr >> 16) & 0xFF;
+    bytes[3] = (iph->daddr >> 24) & 0xFF;
+
+    printf("Destination IP: %d.%d.%d.%d\n", bytes[0], bytes[1], bytes[2], bytes[3]);
+}
+
+int main(int argc, char** argv)
+{
+    if (argc != 2) {
+        printf("\nUsage:\n");
+        printf("pingable-node <net_id>\n");
+        exit(0);
+    }
+    long long int net_id = strtoull(argv[1], NULL, 16);   // At least 64 bits
+
+    int err = ZTS_ERR_OK;
+    char* storage_path = "./";
+
+    if ((err = zts_init_from_storage(storage_path)) != ZTS_ERR_OK) {
+        printf("Unable to start service, error = %d. Exiting.\n", err);
+        exit(1);
+    }
+
+    printf("Starting node...\n");
+    zts_node_start();
+
+    printf("Waiting for node to come online\n");
+    while (! zts_node_is_online()) {
+        zts_util_delay(50);
+    }
+
+    printf("My public identity (node ID) is %llx\n", (long long int)zts_node_get_id());
+    char keypair[ZTS_ID_STR_BUF_LEN] = { 0 };
+    unsigned int len = ZTS_ID_STR_BUF_LEN;
+    if (zts_node_get_id_pair(keypair, &len) != ZTS_ERR_OK) {
+        printf("Error getting identity keypair. Exiting.\n");
+    }
+    printf("Identity [public/secret pair] = %s\n", keypair);
+
+    printf("Joining network %llx\n", net_id);
+    if (zts_net_join(net_id) != ZTS_ERR_OK) {
+        printf("Unable to join network. Exiting.\n");
+        exit(1);
+    }
+
+    printf("Waiting for join to complete\n");
+    while (! zts_net_transport_is_ready(net_id)) {
+        zts_util_delay(50);
+    }
+
+    printf("Waiting for address assignment from network\n");
+    while (! (err = zts_addr_is_assigned(net_id, ZTS_AF_INET))) {
+        zts_util_delay(500);
+    }
+
+    char ipstr[ZTS_IP_MAX_STR_LEN] = { 0 };
+    zts_addr_get_str(net_id, ZTS_AF_INET, ipstr, ZTS_IP_MAX_STR_LEN);
+    printf("Join %llx from another machine and ping me at %s\n", net_id, ipstr);
+
+    // Do network stuff!
+    // zts_bsd_socket, zts_bsd_connect, etc
+
+    int raw_fd;
+    char buffer[1500];   // Buffer to hold the incoming packets
+    // struct zts_sockaddr_in sa_zts;
+    // int len_temp = sizeof(sa_zts);
+
+    raw_fd = zts_bsd_socket(ZTS_AF_INET, ZTS_SOCK_RAW, ZTS_IPPROTO_IP);
+
+    if (raw_fd < 0) {
+        printf("Error creating raw socket.\n");
+        return -1;
+    }
+    char *interface = "tp0";
+    if (zts_bsd_setsockopt(raw_fd, ZTS_SOL_SOCKET, ZTS_SO_BINDTODEVICE, interface, strlen(interface)) < 0) {
+        printf("Error binding to interface");
+        return 1;
+    }
+
+    // struct zts_sockaddr_in sa_zts;
+    // zts_socklen_t sa_len = sizeof(sa_zts);
+    // sa_zts.sin_family = ZTS_AF_INET;
+    // sa_zts.sin_port = 0;
+    // sa_zts.sin_addr.s_addr = htonl(ZTS_INADDR_ANY);
+
+    // if (zts_bsd_bind(raw_fd, (struct sockaddr*)&sa_zts, sizeof(sa_zts)) < 0) {
+    //     printf("Error bind");
+    //     return 1;
+    // }
+
+    while (1) {
+        int recv_len = zts_bsd_recvfrom(raw_fd, buffer, sizeof(buffer), 0, NULL,
+                                        NULL);   // Receive a packet
+        if (recv_len > 0) {
+            process_packet(buffer, recv_len);   // Process the packet
+        }
+    }
+
+    printf("Stopping node\n");
+    return zts_node_stop();
+}