3 changed files with 358 additions and 3 deletions
@ -0,0 +1,280 @@ |
|||||||
|
#include <arpa/inet.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <string.h> |
||||||
|
#include <unistd.h> |
||||||
|
|
||||||
|
#include "ZeroTier.h" |
||||||
|
|
||||||
|
bool node_ready = false; |
||||||
|
bool network_ready = false; |
||||||
|
|
||||||
|
// Example callbacks
|
||||||
|
void myZeroTierEventCallback(struct zts_callback_msg *msg) |
||||||
|
{ |
||||||
|
//
|
||||||
|
// Node events
|
||||||
|
//
|
||||||
|
if (msg->eventCode == ZTS_EVENT_NODE_ONLINE) { |
||||||
|
printf("ZTS_EVENT_NODE_ONLINE, node=%llx\n", msg->node->address); |
||||||
|
node_ready = true; |
||||||
|
// ZeroTier service is running and online
|
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NODE_OFFLINE) { |
||||||
|
printf("ZTS_EVENT_NODE_OFFLINE\n"); |
||||||
|
node_ready = false; |
||||||
|
// ZeroTier service is running and online
|
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NODE_NORMAL_TERMINATION) { |
||||||
|
printf("ZTS_EVENT_NODE_NORMAL_TERMINATION\n"); |
||||||
|
// ZeroTier service has stopped
|
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
// Virtual network events
|
||||||
|
//
|
||||||
|
if (msg->eventCode == ZTS_EVENT_NETWORK_NOT_FOUND) { |
||||||
|
printf("ZTS_EVENT_NETWORK_NOT_FOUND --- network=%llx\n", msg->network->nwid); |
||||||
|
// Is your nwid incorrect?
|
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NETWORK_REQUESTING_CONFIG) { |
||||||
|
printf("ZTS_EVENT_NETWORK_REQUESTING_CONFIG --- network=%llx\n", msg->network->nwid); |
||||||
|
// Node is requesting network config details from controller, please wait
|
||||||
|
}
|
||||||
|
if (msg->eventCode == ZTS_EVENT_NETWORK_ACCESS_DENIED) { |
||||||
|
printf("ZTS_EVENT_NETWORK_ACCESS_DENIED --- network=%llx\n", msg->network->nwid); |
||||||
|
// This node is not authorized to join nwid
|
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NETWORK_READY_IP4) { |
||||||
|
printf("ZTS_EVENT_NETWORK_READY_IP4 --- network=%llx\n", msg->network->nwid); |
||||||
|
network_ready = true; |
||||||
|
// IPv4 traffic can now be processed for nwid
|
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NETWORK_READY_IP6) { |
||||||
|
printf("ZTS_EVENT_NETWORK_READY_IP6 --- network=%llx\n", msg->network->nwid); |
||||||
|
network_ready = true; |
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NETWORK_DOWN) { |
||||||
|
printf("ZTS_EVENT_NETWORK_DOWN --- network=%llx\n", msg->network->nwid); |
||||||
|
// Can happen if another thread called zts_leave()
|
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
// Network stack events
|
||||||
|
//
|
||||||
|
if (msg->eventCode == ZTS_EVENT_NETIF_UP) { |
||||||
|
printf("ZTS_EVENT_NETIF_UP --- network=%llx, mac=%llx, mtu=%d\n",
|
||||||
|
msg->netif->nwid, |
||||||
|
msg->netif->mac, |
||||||
|
msg->netif->mtu); |
||||||
|
network_ready = true; |
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_NETIF_DOWN) { |
||||||
|
printf("ZTS_EVENT_NETIF_DOWN --- network=%llx, mac=%llx\n",
|
||||||
|
msg->netif->nwid, |
||||||
|
msg->netif->mac); |
||||||
|
|
||||||
|
network_ready = true; |
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
// Address events
|
||||||
|
//
|
||||||
|
if (msg->eventCode == ZTS_EVENT_ADDR_ADDED_IP4) { |
||||||
|
char ipstr[INET_ADDRSTRLEN]; |
||||||
|
struct sockaddr_in *in4 = (struct sockaddr_in*)&(msg->addr->addr); |
||||||
|
inet_ntop(AF_INET, &(in4->sin_addr), ipstr, INET_ADDRSTRLEN); |
||||||
|
printf("ZTS_EVENT_ADDR_NEW_IP4 --- addr=%s (on network=%llx)\n",
|
||||||
|
ipstr, msg->addr->nwid); |
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_ADDR_ADDED_IP6) { |
||||||
|
char ipstr[INET6_ADDRSTRLEN]; |
||||||
|
struct sockaddr_in6 *in6 = (struct sockaddr_in6*)&(msg->addr->addr); |
||||||
|
inet_ntop(AF_INET6, &(in6->sin6_addr), ipstr, INET6_ADDRSTRLEN); |
||||||
|
printf("ZTS_EVENT_ADDR_NEW_IP6 --- addr=%s (on network=%llx)\n",
|
||||||
|
ipstr, msg->addr->nwid); |
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_ADDR_REMOVED_IP4) { |
||||||
|
char ipstr[INET_ADDRSTRLEN]; |
||||||
|
struct sockaddr_in *in4 = (struct sockaddr_in*)&(msg->addr->addr); |
||||||
|
inet_ntop(AF_INET, &(in4->sin_addr), ipstr, INET_ADDRSTRLEN); |
||||||
|
printf("ZTS_EVENT_ADDR_REMOVED_IP4 --- addr=%s (on network=%llx)\n",
|
||||||
|
ipstr, msg->addr->nwid); |
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_ADDR_REMOVED_IP6) { |
||||||
|
char ipstr[INET6_ADDRSTRLEN]; |
||||||
|
struct sockaddr_in6 *in6 = (struct sockaddr_in6*)&(msg->addr->addr); |
||||||
|
inet_ntop(AF_INET6, &(in6->sin6_addr), ipstr, INET6_ADDRSTRLEN); |
||||||
|
printf("ZTS_EVENT_ADDR_REMOVED_IP6 --- addr=%s (on network=%llx)\n",
|
||||||
|
ipstr, msg->addr->nwid); |
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
// Peer events
|
||||||
|
//
|
||||||
|
if (msg->eventCode == ZTS_EVENT_PEER_P2P) { |
||||||
|
printf("ZTS_EVENT_PEER_P2P --- node=%llx\n", msg->peer->address); |
||||||
|
// A direct path is known for nodeId
|
||||||
|
} |
||||||
|
if (msg->eventCode == ZTS_EVENT_PEER_RELAY) { |
||||||
|
printf("ZTS_EVENT_PEER_RELAY --- node=%llx\n", msg->peer->address); |
||||||
|
// No direct path is known for nodeId
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void printPeerDetails(struct zts_peer_details *pd) |
||||||
|
{ |
||||||
|
printf("\npeer=%llx, latency=%d, version=%d.%d.%d, pathCount=%d\n", |
||||||
|
pd->address, |
||||||
|
pd->latency, |
||||||
|
pd->versionMajor, |
||||||
|
pd->versionMinor, |
||||||
|
pd->versionRev, |
||||||
|
pd->pathCount); |
||||||
|
// Print all known paths for each peer
|
||||||
|
for (int j=0; j<pd->pathCount; j++) { |
||||||
|
char ipstr[INET6_ADDRSTRLEN]; |
||||||
|
int port; |
||||||
|
struct sockaddr *sa = (struct sockaddr *)&(pd->paths[j].address); |
||||||
|
if (sa->sa_family == AF_INET) { |
||||||
|
struct sockaddr_in *in4 = (struct sockaddr_in*)sa; |
||||||
|
inet_ntop(AF_INET, &(in4->sin_addr), ipstr, INET_ADDRSTRLEN); |
||||||
|
port = ntohs(in4->sin_port); |
||||||
|
} |
||||||
|
if (sa->sa_family == AF_INET6) { |
||||||
|
struct sockaddr_in6 *in6 = (struct sockaddr_in6*)sa; |
||||||
|
inet_ntop(AF_INET6, &(in6->sin6_addr), ipstr, INET6_ADDRSTRLEN); |
||||||
|
} |
||||||
|
printf("\tpath[%d]=%s, port=%d\n", j, ipstr, port);
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void getSinglePeerDetails(uint64_t peerId) |
||||||
|
{ |
||||||
|
struct zts_peer_details pd; |
||||||
|
int err = zts_get_peer(&pd, peerId); |
||||||
|
|
||||||
|
if (err == ZTS_ERR_OK) { |
||||||
|
printf("(%d) call succeeded\n", err); |
||||||
|
} if (err == ZTS_ERR_INVALID_ARG) { |
||||||
|
printf("(%d) invalid argument\n", err); |
||||||
|
return; |
||||||
|
} if (err == ZTS_ERR_SERVICE) { |
||||||
|
printf("(%d) error: service is unavailable\n", err); |
||||||
|
return; |
||||||
|
} if (err == ZTS_ERR_INVALID_OP) { |
||||||
|
printf("(%d) error: invalid API operation\n", err); |
||||||
|
return; |
||||||
|
} if (err == ZTS_ERR_NO_RESULT) { |
||||||
|
printf("(%d) error: object or result not found\n", err); |
||||||
|
return; |
||||||
|
} |
||||||
|
if (err == 0) { // ZTS_ERR_OK
|
||||||
|
printPeerDetails(&pd); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Similar to "zerotier-cli listpeers"
|
||||||
|
void getAllPeerDetails() |
||||||
|
{ |
||||||
|
struct zts_peer_details pd[128]; |
||||||
|
/* This number should be large enough to handle the
|
||||||
|
expected number of peers. This call can also get |
||||||
|
expensive for large numbers of peers. Consider using |
||||||
|
get_peer(struct zts_peer_details *pds, uint64_t peerId) |
||||||
|
instead */ |
||||||
|
int num = 128; |
||||||
|
int err; |
||||||
|
if ((err = zts_get_peers(pd, &num)) < 0) { |
||||||
|
printf("error (%d)\n", err); |
||||||
|
return; |
||||||
|
} |
||||||
|
if (num) { |
||||||
|
printf("num=%d\n", num); |
||||||
|
for (int i=0; i<num; i++) { |
||||||
|
printPeerDetails(&pd[i]); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
struct zts_stats_proto protoSpecificStats; |
||||||
|
|
||||||
|
void display_stack_stats() |
||||||
|
{ |
||||||
|
int err = 0; |
||||||
|
// Count received pings
|
||||||
|
if ((err = zts_get_protocol_stats(ZTS_STATS_PROTOCOL_ICMP, &protoSpecificStats)) != ZTS_ERR_OK) { |
||||||
|
printf("zts_get_proto_stats()=%d", err); |
||||||
|
return; |
||||||
|
} |
||||||
|
printf("icmp.recv=%d\n", protoSpecificStats.recv); |
||||||
|
// Count dropped TCP packets
|
||||||
|
if ((err = zts_get_protocol_stats(ZTS_STATS_PROTOCOL_TCP, &protoSpecificStats)) != ZTS_ERR_OK) { |
||||||
|
printf("zts_get_proto_stats()=%d", err); |
||||||
|
return; |
||||||
|
} |
||||||
|
printf("tcp.drop=%d\n", protoSpecificStats.drop); |
||||||
|
} |
||||||
|
|
||||||
|
int main()
|
||||||
|
{ |
||||||
|
char *str = "welcome to the machine"; |
||||||
|
char *remoteIp = "11.7.7.223"; |
||||||
|
int remotePort = 8082; |
||||||
|
int fd, err = 0; |
||||||
|
struct zts_sockaddr_in addr;
|
||||||
|
addr.sin_family = ZTS_AF_INET; |
||||||
|
addr.sin_addr.s_addr = inet_addr(remoteIp); |
||||||
|
addr.sin_port = htons(remotePort); |
||||||
|
|
||||||
|
// Set up ZeroTier service
|
||||||
|
int defaultServicePort = 9994; |
||||||
|
int nwid = 0x0123456789abcdef; |
||||||
|
zts_start("test/path", &myZeroTierEventCallback, defaultServicePort); |
||||||
|
printf("Waiting for node to come online...\n"); |
||||||
|
|
||||||
|
// Wait for the node to come online before joining a network
|
||||||
|
while (!node_ready) { sleep(1); } |
||||||
|
zts_join(nwid); |
||||||
|
printf("Joined virtual network. Requesting configuration...\n"); |
||||||
|
|
||||||
|
//sleep(1);
|
||||||
|
|
||||||
|
// Get multiple peer's details
|
||||||
|
getAllPeerDetails(); |
||||||
|
|
||||||
|
// Get a single peer's details
|
||||||
|
getSinglePeerDetails(0x01b34f67c90); |
||||||
|
int status = -1; |
||||||
|
|
||||||
|
// Get status of the node/service
|
||||||
|
status = zts_get_node_status(); |
||||||
|
printf("zts_get_node_status()=%d\n", status); |
||||||
|
|
||||||
|
// Get status of a network
|
||||||
|
status = zts_get_network_status(0x0123456789abcdef); |
||||||
|
printf("zts_get_network_status()=%d\n", status); |
||||||
|
|
||||||
|
while (true) { |
||||||
|
sleep(1); |
||||||
|
status = zts_get_node_status(); |
||||||
|
printf("zts_get_node_status()=%d\n", status); |
||||||
|
display_stack_stats(); |
||||||
|
} |
||||||
|
|
||||||
|
// Socket API example
|
||||||
|
printf("zts_errno=%d\n",zts_errno); |
||||||
|
if ((fd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating socket\n"); |
||||||
|
} |
||||||
|
printf("fd=%d, zts_errno=%d\n", fd, zts_errno); |
||||||
|
if ((err = zts_connect(fd, (const struct sockaddr *)&addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host\n"); |
||||||
|
} |
||||||
|
if ((err = zts_write(fd, str, strlen(str))) < 0) { |
||||||
|
printf("error writing to socket\n"); |
||||||
|
} |
||||||
|
|
||||||
|
zts_close(fd); |
||||||
|
zts_stop(); |
||||||
|
return 0; |
||||||
|
} |
||||||
@ -0,0 +1,73 @@ |
|||||||
|
#include <arpa/inet.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <string.h> |
||||||
|
#include <unistd.h> |
||||||
|
|
||||||
|
#include "ZeroTier.h" |
||||||
|
|
||||||
|
bool node_ready = false; |
||||||
|
bool network_ready = false; |
||||||
|
|
||||||
|
void myZeroTierEventCallback(struct zts_callback_msg *msg) |
||||||
|
{ |
||||||
|
switch (msg->eventCode) |
||||||
|
{ |
||||||
|
case ZTS_EVENT_NODE_ONLINE: |
||||||
|
printf("ZTS_EVENT_NODE_ONLINE, node=%llx\n", msg->node->address); |
||||||
|
node_ready = true; |
||||||
|
break; |
||||||
|
case ZTS_EVENT_NODE_OFFLINE: |
||||||
|
printf("ZTS_EVENT_NODE_OFFLINE\n"); |
||||||
|
node_ready = false; |
||||||
|
break; |
||||||
|
case ZTS_EVENT_NETWORK_READY_IP4: |
||||||
|
printf("ZTS_EVENT_NETWORK_READY_IP4 --- network=%llx\n", msg->network->nwid); |
||||||
|
network_ready = true; |
||||||
|
break; |
||||||
|
case ZTS_EVENT_PEER_P2P: |
||||||
|
printf("ZTS_EVENT_PEER_P2P --- node=%llx\n", msg->peer->address); |
||||||
|
break; |
||||||
|
case ZTS_EVENT_PEER_RELAY: |
||||||
|
printf("ZTS_EVENT_PEER_RELAY --- node=%llx\n", msg->peer->address); |
||||||
|
break; |
||||||
|
// ...
|
||||||
|
default: |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
int main()
|
||||||
|
{ |
||||||
|
char *str = "welcome to the machine"; |
||||||
|
char *remoteIp = "11.7.7.223"; |
||||||
|
int remotePort = 8082; |
||||||
|
int fd, err = 0; |
||||||
|
struct zts_sockaddr_in addr;
|
||||||
|
addr.sin_family = ZTS_AF_INET; |
||||||
|
addr.sin_addr.s_addr = inet_addr(remoteIp); |
||||||
|
addr.sin_port = htons(remotePort); |
||||||
|
|
||||||
|
// Set up ZeroTier service and wai for callbacks
|
||||||
|
int port = 9994; |
||||||
|
int nwid = 0x0123456789abcdef; |
||||||
|
zts_start("test/path", &myZeroTierEventCallback, port); |
||||||
|
printf("Waiting for node to come online...\n"); |
||||||
|
while (!node_ready) { sleep(1); } |
||||||
|
zts_join(nwid); |
||||||
|
printf("Joined virtual network. Requesting configuration...\n"); |
||||||
|
while (!network_ready) { sleep(1); } |
||||||
|
|
||||||
|
// Socket API example
|
||||||
|
if ((fd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating socket\n"); |
||||||
|
} |
||||||
|
if ((err = zts_connect(fd, (const struct sockaddr *)&addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host\n"); |
||||||
|
} |
||||||
|
if ((err = zts_write(fd, str, strlen(str))) < 0) { |
||||||
|
printf("error writing to socket\n"); |
||||||
|
} |
||||||
|
zts_close(fd); |
||||||
|
zts_stop(); |
||||||
|
return 0; |
||||||
|
} |
||||||
Loading…
Reference in new issue