diff --git a/test/client.cpp b/test/client.cpp new file mode 100644 index 0000000..87596dc --- /dev/null +++ b/test/client.cpp @@ -0,0 +1,66 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "libzt.h" + +char *msg = (char*)"welcome to the machine"; + +int main(int argc, char **argv) +{ + if(argc != 5) { + printf("\nlibzt example client\n"); + printf("client [config_file_path] [nwid] [remote_addr] [remote_port]\n"); + exit(0); + } + std::string path = argv[1]; + std::string nwid = argv[2]; + std::string remote_addr = argv[3]; + int remote_port = atoi(argv[4]); + int r=0, w=0, err=0, sockfd; + char rbuf[32]; + memset(rbuf, 0, sizeof rbuf); + + struct sockaddr_in in4; + in4.sin_port = htons(remote_port); + in4.sin_addr.s_addr = inet_addr(remote_addr.c_str()); + in4.sin_family = AF_INET; + + + // --- BEGIN + + DEBUG_TEST("Waiting for libzt to come online...\n"); + zts_simple_start(path.c_str(), nwid.c_str()); + char device_id[11]; + zts_get_device_id(device_id); + DEBUG_TEST("I am %s", device_id); + sleep(2); + + // socket() + if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) + DEBUG_ERROR("error creating ZeroTier socket"); + + // connect() + if((err = zts_connect(sockfd, (const struct sockaddr *)&in4, sizeof(in4))) < 0) + DEBUG_ERROR("error connecting to remote host (%d)", err); + + // tx + w = zts_write(sockfd, msg, strlen(msg)); + + // rx + r = zts_read(sockfd, rbuf, strlen(msg)); + + DEBUG_TEST("Sent : %s", msg); + DEBUG_TEST("Received : %s", rbuf); + + sleep(2); + err = zts_close(sockfd); + + return err; +} \ No newline at end of file diff --git a/test/server.cpp b/test/server.cpp new file mode 100644 index 0000000..bb4c826 --- /dev/null +++ b/test/server.cpp @@ -0,0 +1,76 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "libzt.h" + +int main(int argc, char **argv) +{ + if(argc != 4) { + printf("\nlibzt example server\n"); + printf("server [config_file_path] [nwid] [bind_port]\n"); + exit(0); + } + std::string path = argv[1]; + std::string nwid = argv[2]; + int bind_port = atoi(argv[3]); + int w=0, r=0, err=0, sockfd, accfd; + char rbuf[32]; + memset(rbuf, 0, sizeof rbuf); + + struct sockaddr_in in4, acc_in4; + in4.sin_port = htons(bind_port); + in4.sin_addr.s_addr = INADDR_ANY; + in4.sin_family = AF_INET; + + // --- BEGIN + + + DEBUG_TEST("Waiting for libzt to come online...\n"); + zts_simple_start(path.c_str(), nwid.c_str()); + char device_id[11]; + zts_get_device_id(device_id); + DEBUG_TEST("I am %s", device_id); + sleep(2); + + // socket() + if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) + DEBUG_ERROR("error creating ZeroTier socket"); + + // bind() + if((err = zts_bind(sockfd, (struct sockaddr *)&in4, sizeof(struct sockaddr_in)) < 0)) + DEBUG_ERROR("error binding to interface (%d)", err); + + // listen() + if((err = zts_listen(sockfd, 100)) < 0) + DEBUG_ERROR("error placing socket in LISTENING state (%d)", err); + + // accept() + if((accfd = zts_accept(sockfd, (struct sockaddr *)&acc_in4, (socklen_t *)sizeof(struct sockaddr_in))) < 0) + DEBUG_ERROR("error accepting connection (%d)", err); + + // getpeername() + //socklen_t peer_addrlen; + //zts_getpeername(accfd, (struct sockaddr*)&acc_, &peer_addrlen); + //DEBUG_INFO("getpeername() => %s : %d", inet_ntoa(acc_in4.sin_addr), ntohs(acc_in4.sin_port)); + + // rx + r = zts_read(accfd, rbuf, sizeof rbuf); + + // tx + w = zts_write(accfd, rbuf, strlen(rbuf)); + + DEBUG_TEST("Received : %s", rbuf); + + sleep(2); + err = zts_close(sockfd); + err = zts_close(accfd); + + return err; +} \ No newline at end of file