121 lines
3.4 KiB
C
121 lines
3.4 KiB
C
/**
|
|
* 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();
|
|
}
|