Joseph Henry
9 years ago
30 changed files with 2422 additions and 767 deletions
@ -1,15 +1,51 @@ |
|||||||
ZeroTier SDK |
ZeroTier SDK |
||||||
====== |
====== |
||||||
|
|
||||||
|
[](https://webchat.freenode.net/?channels=zerotier) |
||||||
|
|
||||||
*** |
*** |
||||||
|
|
||||||
|
## Example |
||||||
|
|
||||||
|
``` |
||||||
|
std::string str = "welcome to the machine"; |
||||||
|
zts_start(path); |
||||||
|
while(!zts_service_running()) |
||||||
|
sleep(1); |
||||||
|
zts_join_network(nwid); |
||||||
|
int err, sockfd; |
||||||
|
while(!zts_has_address(nwid)) |
||||||
|
sleep(1); |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
``` |
||||||
|
|
||||||
|
Bindings also exist for [many popular languages](). |
||||||
|
|
||||||
## Build Targets |
## Build Targets |
||||||
### Static Library |
### Static Library |
||||||
`static_lib` |
- `make static_lib`: Will output to `build/` |
||||||
|
|
||||||
|
### Tests |
||||||
|
- `make tests`: Will output to `build/tests/` |
||||||
|
|
||||||
|
Then run the comprehensive test suite with whatever configuration you need. For instance: |
||||||
|
|
||||||
|
To run a single-test IPv4 server on port 8787: |
||||||
|
|
||||||
|
- Host 1: `./build/test/comprehensive c7cd7c9e1b0f52a2 simple 4 server 8787` |
||||||
|
- Host 2: `./build/test/comprehensive c7cd7c9e1b0f52a2 simple 4 client 10.9.9.40 8787` |
||||||
|
|
||||||
## Using Language Bindings |
## Using Language Bindings |
||||||
`SDK_LANG_JAVA=1` |
`SDK_LANG_JAVA=1` |
||||||
`SDK_LANG_CSHARP=1` |
`SDK_LANG_CSHARP=1` |
||||||
`SDK_LANG_PYTHON=1` |
`SDK_LANG_PYTHON=1` |
||||||
`SDK_LANG_=1` |
`SDK_LANG_GO=1` |
||||||
|
|||||||
@ -0,0 +1,3 @@ |
|||||||
|
Clojure Language Binding API for the ZeroTier SDK |
||||||
|
====== |
||||||
|
|
||||||
@ -0,0 +1,3 @@ |
|||||||
|
Go Language Binding API for the ZeroTier SDK |
||||||
|
====== |
||||||
|
|
||||||
@ -0,0 +1,84 @@ |
|||||||
|
/*
|
||||||
|
* ZeroTier SDK - Network Virtualization Everywhere |
||||||
|
* Copyright (C) 2011-2016 ZeroTier, Inc. https://www.zerotier.com/
|
||||||
|
* |
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
|
||||||
|
#include <stdio.h> |
||||||
|
#include <execinfo.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <stdarg.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
#include <unistd.h> |
||||||
|
#include <errno.h> |
||||||
|
|
||||||
|
void zt_dump_stacktrace(int sig) { |
||||||
|
void *array[16]; |
||||||
|
size_t size; |
||||||
|
size = backtrace(array, 16); |
||||||
|
fprintf(stderr, "Error: signal %d:\n", sig); |
||||||
|
backtrace_symbols_fd(array, size, STDERR_FILENO); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
char *beautify_pico_error(int err) |
||||||
|
{ |
||||||
|
switch(err){ |
||||||
|
PICO_ERR_NOERR = 0, |
||||||
|
PICO_ERR_EPERM = 1, |
||||||
|
PICO_ERR_ENOENT = 2, |
||||||
|
|
||||||
|
PICO_ERR_EINTR = 4, |
||||||
|
PICO_ERR_EIO = 5, |
||||||
|
PICO_ERR_ENXIO = 6, |
||||||
|
|
||||||
|
PICO_ERR_EAGAIN = 11, |
||||||
|
PICO_ERR_ENOMEM = 12, |
||||||
|
PICO_ERR_EACCESS = 13, |
||||||
|
PICO_ERR_EFAULT = 14, |
||||||
|
|
||||||
|
PICO_ERR_EBUSY = 16, |
||||||
|
PICO_ERR_EEXIST = 17, |
||||||
|
|
||||||
|
PICO_ERR_EINVAL = 22, |
||||||
|
|
||||||
|
PICO_ERR_ENONET = 64, |
||||||
|
|
||||||
|
PICO_ERR_EPROTO = 71, |
||||||
|
|
||||||
|
PICO_ERR_ENOPROTOOPT = 92, |
||||||
|
PICO_ERR_EPROTONOSUPPORT = 93, |
||||||
|
|
||||||
|
PICO_ERR_EOPNOTSUPP = 95, |
||||||
|
PICO_ERR_EADDRINUSE = 98, |
||||||
|
PICO_ERR_EADDRNOTAVAIL = 99, |
||||||
|
PICO_ERR_ENETDOWN = 100, |
||||||
|
PICO_ERR_ENETUNREACH = 101, |
||||||
|
|
||||||
|
PICO_ERR_ECONNRESET = 104, |
||||||
|
|
||||||
|
PICO_ERR_EISCONN = 106, |
||||||
|
PICO_ERR_ENOTCONN = 107, |
||||||
|
PICO_ERR_ESHUTDOWN = 108, |
||||||
|
|
||||||
|
PICO_ERR_ETIMEDOUT = 110, |
||||||
|
PICO_ERR_ECONNREFUSED = 111, |
||||||
|
PICO_ERR_EHOSTDOWN = 112, |
||||||
|
PICO_ERR_EHOSTUNREACH = 113, |
||||||
|
} |
||||||
|
return err_text; |
||||||
|
} |
||||||
|
*/ |
||||||
@ -0,0 +1,27 @@ |
|||||||
|
/*
|
||||||
|
* ZeroTier SDK - Network Virtualization Everywhere |
||||||
|
* Copyright (C) 2011-2016 ZeroTier, Inc. https://www.zerotier.com/
|
||||||
|
* |
||||||
|
* This program is free software: you can redistribute it and/or modify |
||||||
|
* it under the terms of the GNU General Public License as published by |
||||||
|
* the Free Software Foundation, either version 3 of the License, or |
||||||
|
* (at your option) any later version. |
||||||
|
* |
||||||
|
* This program is distributed in the hope that it will be useful, |
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
* GNU General Public License for more details. |
||||||
|
* |
||||||
|
* You should have received a copy of the GNU General Public License |
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef SDK_UTILS_HPP |
||||||
|
#define SDK_UTILS_HPP |
||||||
|
|
||||||
|
/*
|
||||||
|
* Print a stacktrace |
||||||
|
*/ |
||||||
|
void zt_dump_stacktrace(int sig); |
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,554 @@ |
|||||||
|
// Comprehensive stress test for socket-like API
|
||||||
|
|
||||||
|
#include <unistd.h> |
||||||
|
#include <sys/socket.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <string.h> |
||||||
|
#include <netinet/in.h> |
||||||
|
#include <netdb.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "ZeroTierSDK.h" |
||||||
|
|
||||||
|
#define PASSED 0 |
||||||
|
#define FAILED -1 |
||||||
|
|
||||||
|
std::string str = "welcome to the machine"; |
||||||
|
|
||||||
|
// [] random
|
||||||
|
// [OK] simple client ipv4
|
||||||
|
// [OK] simple server ipv4
|
||||||
|
// [?] simple client ipv6
|
||||||
|
// [?] simple server ipv6
|
||||||
|
// [OK] sustained client ipv4
|
||||||
|
// [OK] sustained server ipv4
|
||||||
|
// [?] sustained client ipv6
|
||||||
|
// [?] sustained server ipv6
|
||||||
|
// [] comprehensive client ipv4
|
||||||
|
// [] comprehensive server ipv6
|
||||||
|
|
||||||
|
// ---
|
||||||
|
|
||||||
|
// comprehensive client addr port
|
||||||
|
// comprehensive server port
|
||||||
|
// simple [4|6] client addr port
|
||||||
|
// simple [4|6] server port
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE CLIENT */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv4_tcp_client_test(char *path, char *nwid, struct sockaddr_in *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv6_tcp_client_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE SERVER */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv4_tcp_server_test(char *path, char *nwid, struct sockaddr_in *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, sizeof(struct sockaddr_in)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv6_tcp_server_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, sizeof(struct sockaddr_in6)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED CLIENT */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv4_tcp_client_sustained_test(char *path, char *nwid, struct sockaddr_in *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv6_tcp_client_sustained_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED SERVER */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv4_tcp_server_sustained_test(char *path, char *nwid, struct sockaddr_in *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, (socklen_t)sizeof(struct sockaddr_in)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv6_tcp_server_sustained_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, (socklen_t)sizeof(struct sockaddr_in6)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* main */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
int main(int argc , char *argv[]) |
||||||
|
{ |
||||||
|
if(argc < 3) { |
||||||
|
printf("invalid arguments\n");
|
||||||
|
return 1; |
||||||
|
} |
||||||
|
int port; |
||||||
|
struct hostent *server; |
||||||
|
std::string protocol; |
||||||
|
std::string mode; |
||||||
|
struct sockaddr_in6 addr6; |
||||||
|
struct sockaddr_in addr; |
||||||
|
char *path = (char*)"./zt"; |
||||||
|
|
||||||
|
int n_seconds = 10; |
||||||
|
int n_times = 10000; |
||||||
|
|
||||||
|
std::string nwid = argv[1]; // "ebcd7a7e120f4492"
|
||||||
|
std::string type = argv[2]; // simple, sustained, comprehensive
|
||||||
|
|
||||||
|
// If we're performing a non-random test, join the network we want to test on
|
||||||
|
// and wait until the service initializes the SocketTap and provides an address
|
||||||
|
if(type == "simple" || type == "sustained" || type == "comprehensive") { |
||||||
|
zts_start(path); |
||||||
|
while(!zts_service_running()) |
||||||
|
sleep(1); |
||||||
|
zts_join_network(nwid.c_str()); |
||||||
|
while(!zts_has_address(nwid.c_str())) |
||||||
|
sleep(1); |
||||||
|
} |
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// SIMPLE
|
||||||
|
// performs a one-off test of a particular subset of the API
|
||||||
|
// For instance (ipv4 client, ipv6 server, etc)
|
||||||
|
if(type == "simple") |
||||||
|
{ |
||||||
|
protocol = argv[3]; // 4, 6
|
||||||
|
mode = argv[4]; // client, server
|
||||||
|
|
||||||
|
// SIMPLE CLIENT
|
||||||
|
if(mode == "client") |
||||||
|
{ |
||||||
|
port = atoi(argv[6]); |
||||||
|
printf("connecting to %s on port %s\n", argv[5], argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_addr.s_addr = inet_addr(argv[5]); |
||||||
|
addr.sin_family = AF_INET; |
||||||
|
addr.sin_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_client_test(path, (char*)nwid.c_str(), &addr, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length); |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_client_test(path, (char*)nwid.c_str(), &addr6, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SIMPLE SERVER
|
||||||
|
if(mode == "server") |
||||||
|
{ |
||||||
|
port = atoi(argv[5]); |
||||||
|
printf("serving on port %s\n", argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr("10.9.9.40"); |
||||||
|
// addr.sin_addr.s_addr = htons(INADDR_ANY);
|
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_server_test(path, (char*)nwid.c_str(), &addr, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
|
||||||
|
addr6.sin6_addr = in6addr_any; |
||||||
|
//memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length);
|
||||||
|
|
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_server_test(path, (char*)nwid.c_str(), &addr6, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// SUSTAINED
|
||||||
|
// Performs a stress test for benchmarking performance
|
||||||
|
if(type == "sustained") |
||||||
|
{ |
||||||
|
protocol = argv[3]; // 4, 6
|
||||||
|
mode = argv[4]; // client, server
|
||||||
|
|
||||||
|
// SUSTAINED CLIENT
|
||||||
|
if(mode == "client") |
||||||
|
{ |
||||||
|
port = atoi(argv[6]); |
||||||
|
printf("connecting to %s on port %s\n", argv[5], argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr(argv[5]); |
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_client_sustained_test(path, (char*)nwid.c_str(), &addr, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length); |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_client_sustained_test(path, (char*)nwid.c_str(), &addr6, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SUSTAINED SERVER
|
||||||
|
if(mode == "server") |
||||||
|
{ |
||||||
|
port = atoi(argv[5]); |
||||||
|
printf("serving on port %s\n", argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr("10.9.9.0"); |
||||||
|
// addr.sin_addr.s_addr = htons(INADDR_ANY);
|
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_server_sustained_test(path, (char*)nwid.c_str(), &addr, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
|
||||||
|
addr6.sin6_addr = in6addr_any; |
||||||
|
//memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length);
|
||||||
|
|
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_server_sustained_test(path, (char*)nwid.c_str(), &addr6, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* COMPREHENSIVE */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// COMPREHENSIVE
|
||||||
|
// Tests ALL API calls
|
||||||
|
if(type == "comprehensive") |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* RANDOM */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// RANDOM
|
||||||
|
// performs random API calls with plausible (and random) arguments/data
|
||||||
|
if(type == "random") |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
printf("invalid configuration. exiting.\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
@ -0,0 +1,554 @@ |
|||||||
|
// Comprehensive stress test for socket-like API
|
||||||
|
|
||||||
|
#include <unistd.h> |
||||||
|
#include <sys/socket.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <string.h> |
||||||
|
#include <netinet/in.h> |
||||||
|
#include <netdb.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "ZeroTierSDK.h" |
||||||
|
|
||||||
|
#define PASSED 0 |
||||||
|
#define FAILED -1 |
||||||
|
|
||||||
|
std::string str = "welcome to the machine"; |
||||||
|
|
||||||
|
// [] random
|
||||||
|
// [OK] simple client ipv4
|
||||||
|
// [OK] simple server ipv4
|
||||||
|
// [?] simple client ipv6
|
||||||
|
// [?] simple server ipv6
|
||||||
|
// [OK] sustained client ipv4
|
||||||
|
// [OK] sustained server ipv4
|
||||||
|
// [?] sustained client ipv6
|
||||||
|
// [?] sustained server ipv6
|
||||||
|
// [] comprehensive client ipv4
|
||||||
|
// [] comprehensive server ipv6
|
||||||
|
|
||||||
|
// ---
|
||||||
|
|
||||||
|
// comprehensive client addr port
|
||||||
|
// comprehensive server port
|
||||||
|
// simple [4|6] client addr port
|
||||||
|
// simple [4|6] server port
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE CLIENT */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv4_tcp_client_test(char *path, char *nwid, struct sockaddr_in *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv6_tcp_client_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE SERVER */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv4_tcp_server_test(char *path, char *nwid, struct sockaddr_in *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, sizeof(struct sockaddr_in)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv6_tcp_server_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, sizeof(struct sockaddr_in6)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED CLIENT */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv4_tcp_client_sustained_test(char *path, char *nwid, struct sockaddr_in *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv6_tcp_client_sustained_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED SERVER */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv4_tcp_server_sustained_test(char *path, char *nwid, struct sockaddr_in *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, (socklen_t)sizeof(struct sockaddr_in)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv6_tcp_server_sustained_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, (socklen_t)sizeof(struct sockaddr_in6)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* main */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
int main(int argc , char *argv[]) |
||||||
|
{ |
||||||
|
if(argc < 3) { |
||||||
|
printf("invalid arguments\n");
|
||||||
|
return 1; |
||||||
|
} |
||||||
|
int port; |
||||||
|
struct hostent *server; |
||||||
|
std::string protocol; |
||||||
|
std::string mode; |
||||||
|
struct sockaddr_in6 addr6; |
||||||
|
struct sockaddr_in addr; |
||||||
|
char *path = (char*)"./zt"; |
||||||
|
|
||||||
|
int n_seconds = 10; |
||||||
|
int n_times = 10000; |
||||||
|
|
||||||
|
std::string nwid = argv[1]; // "ebcd7a7e120f4492"
|
||||||
|
std::string type = argv[2]; // simple, sustained, comprehensive
|
||||||
|
|
||||||
|
// If we're performing a non-random test, join the network we want to test on
|
||||||
|
// and wait until the service initializes the SocketTap and provides an address
|
||||||
|
if(type == "simple" || type == "sustained" || type == "comprehensive") { |
||||||
|
zts_start(path); |
||||||
|
while(!zts_service_running()) |
||||||
|
sleep(1); |
||||||
|
zts_join_network(nwid.c_str()); |
||||||
|
while(!zts_has_address(nwid.c_str())) |
||||||
|
sleep(1); |
||||||
|
} |
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// SIMPLE
|
||||||
|
// performs a one-off test of a particular subset of the API
|
||||||
|
// For instance (ipv4 client, ipv6 server, etc)
|
||||||
|
if(type == "simple") |
||||||
|
{ |
||||||
|
protocol = argv[3]; // 4, 6
|
||||||
|
mode = argv[4]; // client, server
|
||||||
|
|
||||||
|
// SIMPLE CLIENT
|
||||||
|
if(mode == "client") |
||||||
|
{ |
||||||
|
port = atoi(argv[6]); |
||||||
|
printf("connecting to %s on port %s\n", argv[5], argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_addr.s_addr = inet_addr(argv[5]); |
||||||
|
addr.sin_family = AF_INET; |
||||||
|
addr.sin_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_client_test(path, (char*)nwid.c_str(), &addr, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length); |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_client_test(path, (char*)nwid.c_str(), &addr6, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SIMPLE SERVER
|
||||||
|
if(mode == "server") |
||||||
|
{ |
||||||
|
port = atoi(argv[5]); |
||||||
|
printf("serving on port %s\n", argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr("10.9.9.40"); |
||||||
|
// addr.sin_addr.s_addr = htons(INADDR_ANY);
|
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_server_test(path, (char*)nwid.c_str(), &addr, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
|
||||||
|
addr6.sin6_addr = in6addr_any; |
||||||
|
//memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length);
|
||||||
|
|
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_server_test(path, (char*)nwid.c_str(), &addr6, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// SUSTAINED
|
||||||
|
// Performs a stress test for benchmarking performance
|
||||||
|
if(type == "sustained") |
||||||
|
{ |
||||||
|
protocol = argv[3]; // 4, 6
|
||||||
|
mode = argv[4]; // client, server
|
||||||
|
|
||||||
|
// SUSTAINED CLIENT
|
||||||
|
if(mode == "client") |
||||||
|
{ |
||||||
|
port = atoi(argv[6]); |
||||||
|
printf("connecting to %s on port %s\n", argv[5], argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr(argv[5]); |
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_client_sustained_test(path, (char*)nwid.c_str(), &addr, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length); |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_client_sustained_test(path, (char*)nwid.c_str(), &addr6, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SUSTAINED SERVER
|
||||||
|
if(mode == "server") |
||||||
|
{ |
||||||
|
port = atoi(argv[5]); |
||||||
|
printf("serving on port %s\n", argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr("10.9.9.0"); |
||||||
|
// addr.sin_addr.s_addr = htons(INADDR_ANY);
|
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_server_sustained_test(path, (char*)nwid.c_str(), &addr, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
|
||||||
|
addr6.sin6_addr = in6addr_any; |
||||||
|
//memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length);
|
||||||
|
|
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_server_sustained_test(path, (char*)nwid.c_str(), &addr6, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* COMPREHENSIVE */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// COMPREHENSIVE
|
||||||
|
// Tests ALL API calls
|
||||||
|
if(type == "comprehensive") |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* RANDOM */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// RANDOM
|
||||||
|
// performs random API calls with plausible (and random) arguments/data
|
||||||
|
if(type == "random") |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
printf("invalid configuration. exiting.\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
@ -1,64 +0,0 @@ |
|||||||
// Comprehensive stress test for socket-like API
|
|
||||||
|
|
||||||
#include <stdio.h> |
|
||||||
|
|
||||||
/****************************************************************************/ |
|
||||||
/* Test Functions */ |
|
||||||
/****************************************************************************/ |
|
||||||
|
|
||||||
int test_for_correctness() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv4_udp_client() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv6_udp_client() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv4_tcp_client() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv6_tcp_client() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
/****************************************************************************/ |
|
||||||
/* main */ |
|
||||||
/****************************************************************************/ |
|
||||||
|
|
||||||
int main() |
|
||||||
{ |
|
||||||
int test_all = 1; |
|
||||||
|
|
||||||
if(test_all) |
|
||||||
{ |
|
||||||
printf("Testing API calls for correctness\n"); |
|
||||||
test_for_correctness(); |
|
||||||
|
|
||||||
printf("Testing as IPv4 UDP Client\n"); |
|
||||||
ipv4_udp_client(); |
|
||||||
|
|
||||||
printf("Testing as IPv6 UDP Client\n"); |
|
||||||
ipv6_udp_client(); |
|
||||||
|
|
||||||
printf("Testing as IPv4 TCP Client\n"); |
|
||||||
ipv4_udp_client(); |
|
||||||
|
|
||||||
printf("Testing as IPv6 TCP Client\n"); |
|
||||||
ipv6_udp_client(); |
|
||||||
|
|
||||||
printf("Testing \n"); |
|
||||||
printf("\n"); |
|
||||||
printf("\n"); |
|
||||||
} |
|
||||||
return 0; |
|
||||||
} |
|
||||||
@ -1,64 +1,554 @@ |
|||||||
// Comprehensive stress test for socket-like API
|
// Comprehensive stress test for socket-like API
|
||||||
|
|
||||||
|
#include <unistd.h> |
||||||
|
#include <sys/socket.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <arpa/inet.h> |
||||||
|
#include <string.h> |
||||||
|
#include <netinet/in.h> |
||||||
|
#include <netdb.h> |
||||||
#include <stdio.h> |
#include <stdio.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "ZeroTierSDK.h" |
||||||
|
|
||||||
|
#define PASSED 0 |
||||||
|
#define FAILED -1 |
||||||
|
|
||||||
|
std::string str = "welcome to the machine"; |
||||||
|
|
||||||
|
// [] random
|
||||||
|
// [OK] simple client ipv4
|
||||||
|
// [OK] simple server ipv4
|
||||||
|
// [?] simple client ipv6
|
||||||
|
// [?] simple server ipv6
|
||||||
|
// [OK] sustained client ipv4
|
||||||
|
// [OK] sustained server ipv4
|
||||||
|
// [?] sustained client ipv6
|
||||||
|
// [?] sustained server ipv6
|
||||||
|
// [] comprehensive client ipv4
|
||||||
|
// [] comprehensive server ipv6
|
||||||
|
|
||||||
|
// ---
|
||||||
|
|
||||||
|
// comprehensive client addr port
|
||||||
|
// comprehensive server port
|
||||||
|
// simple [4|6] client addr port
|
||||||
|
// simple [4|6] server port
|
||||||
|
|
||||||
/****************************************************************************/ |
/****************************************************************************/ |
||||||
/* Test Functions */ |
/* SIMPLE CLIENT */ |
||||||
/****************************************************************************/ |
/****************************************************************************/ |
||||||
|
|
||||||
int test_for_correctness() |
//
|
||||||
|
int ipv4_tcp_client_test(char *path, char *nwid, struct sockaddr_in *addr, int port) |
||||||
{ |
{ |
||||||
return 0; |
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
} |
} |
||||||
|
|
||||||
int ipv4_udp_server() |
//
|
||||||
|
int ipv6_tcp_client_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port) |
||||||
{ |
{ |
||||||
return 0; |
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
} |
} |
||||||
|
|
||||||
int ipv6_udp_server() |
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE SERVER */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
//
|
||||||
|
int ipv4_tcp_server_test(char *path, char *nwid, struct sockaddr_in *addr, int port) |
||||||
{ |
{ |
||||||
return 0; |
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, sizeof(struct sockaddr_in)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
} |
} |
||||||
|
|
||||||
int ipv4_tcp_server() |
//
|
||||||
|
int ipv6_tcp_server_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port) |
||||||
{ |
{ |
||||||
return 0; |
int sockfd, err; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, sizeof(struct sockaddr_in6)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
int wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == str.length() && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
} |
} |
||||||
|
|
||||||
int ipv6_tcp_server() |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED CLIENT */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv4_tcp_client_sustained_test(char *path, char *nwid, struct sockaddr_in *addr, int port, int n_seconds, int n_times) |
||||||
{ |
{ |
||||||
return 0; |
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
} |
} |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv6_tcp_client_sustained_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_connect(sockfd, (const struct sockaddr *)addr, sizeof(addr))) < 0) { |
||||||
|
printf("error connecting to remote host (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED SERVER */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv4_tcp_server_sustained_test(char *path, char *nwid, struct sockaddr_in *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, (socklen_t)sizeof(struct sockaddr_in)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
// Maintain transfer for n_seconds OR n_times
|
||||||
|
int ipv6_tcp_server_sustained_test(char *path, char *nwid, struct sockaddr_in6 *addr, int port, int n_seconds, int n_times) |
||||||
|
{ |
||||||
|
int sockfd, err; |
||||||
|
int msg_len = str.length(); |
||||||
|
int wrote = 0; |
||||||
|
if((sockfd = zts_socket(AF_INET, SOCK_STREAM, 0)) < 0) { |
||||||
|
printf("error creating ZeroTier socket"); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_bind(sockfd, (struct sockaddr *)addr, (socklen_t)sizeof(struct sockaddr_in6)) < 0)) { |
||||||
|
printf("error binding to interface (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if((err = zts_listen(sockfd, 1)) < 0) { |
||||||
|
printf("error placing socket in LISTENING state (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
// TODO: handle new address
|
||||||
|
if((err = zts_accept(sockfd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)) < 0)) { |
||||||
|
printf("error accepting connection (%d)\n", err); |
||||||
|
exit(0); |
||||||
|
} |
||||||
|
if(n_seconds) { |
||||||
|
/*
|
||||||
|
printf("testing for (%d) seconds\n", n_seconds); |
||||||
|
int wrote; |
||||||
|
for(int i=0; i<n_seconds; i++) { |
||||||
|
wrote = zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("wrote = %d\n", wrote); |
||||||
|
} |
||||||
|
*/ |
||||||
|
} |
||||||
|
if(n_times) { |
||||||
|
printf("testing (%d) times\n", n_seconds); |
||||||
|
for(int i=0; i<n_times; i++) { |
||||||
|
wrote += zts_write(sockfd, str.c_str(), str.length()); |
||||||
|
printf("[%d] wrote = %d\n", i, wrote); |
||||||
|
} |
||||||
|
} |
||||||
|
err = zts_close(sockfd); |
||||||
|
return (wrote == (str.length()*n_times) && !err) ? PASSED : FAILED; // if wrote correct number of bytes
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/****************************************************************************/ |
/****************************************************************************/ |
||||||
/* main */ |
/* main */ |
||||||
/****************************************************************************/ |
/****************************************************************************/ |
||||||
|
|
||||||
int main() |
int main(int argc , char *argv[]) |
||||||
{ |
{ |
||||||
int test_all = 1; |
if(argc < 3) { |
||||||
|
printf("invalid arguments\n");
|
||||||
|
return 1; |
||||||
|
} |
||||||
|
int port; |
||||||
|
struct hostent *server; |
||||||
|
std::string protocol; |
||||||
|
std::string mode; |
||||||
|
struct sockaddr_in6 addr6; |
||||||
|
struct sockaddr_in addr; |
||||||
|
char *path = (char*)"./zt"; |
||||||
|
|
||||||
|
int n_seconds = 10; |
||||||
|
int n_times = 10000; |
||||||
|
|
||||||
|
std::string nwid = argv[1]; // "ebcd7a7e120f4492"
|
||||||
|
std::string type = argv[2]; // simple, sustained, comprehensive
|
||||||
|
|
||||||
|
// If we're performing a non-random test, join the network we want to test on
|
||||||
|
// and wait until the service initializes the SocketTap and provides an address
|
||||||
|
if(type == "simple" || type == "sustained" || type == "comprehensive") { |
||||||
|
zts_start(path); |
||||||
|
while(!zts_service_running()) |
||||||
|
sleep(1); |
||||||
|
zts_join_network(nwid.c_str()); |
||||||
|
while(!zts_has_address(nwid.c_str())) |
||||||
|
sleep(1); |
||||||
|
} |
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SIMPLE */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
if(test_all) |
// SIMPLE
|
||||||
|
// performs a one-off test of a particular subset of the API
|
||||||
|
// For instance (ipv4 client, ipv6 server, etc)
|
||||||
|
if(type == "simple") |
||||||
{ |
{ |
||||||
printf("Testing API calls for correctness\n"); |
protocol = argv[3]; // 4, 6
|
||||||
test_for_correctness(); |
mode = argv[4]; // client, server
|
||||||
|
|
||||||
|
// SIMPLE CLIENT
|
||||||
|
if(mode == "client") |
||||||
|
{ |
||||||
|
port = atoi(argv[6]); |
||||||
|
printf("connecting to %s on port %s\n", argv[5], argv[6]); |
||||||
|
|
||||||
printf("Testing as IPv4 UDP Server\n"); |
// IPv4
|
||||||
ipv4_udp_server(); |
if(protocol == "4") { |
||||||
|
addr.sin_addr.s_addr = inet_addr(argv[5]); |
||||||
|
addr.sin_family = AF_INET; |
||||||
|
addr.sin_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_client_test(path, (char*)nwid.c_str(), &addr, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
printf("Testing as IPv6 UDP Server\n"); |
// IPv6
|
||||||
ipv6_udp_server(); |
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length); |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_client_test(path, (char*)nwid.c_str(), &addr6, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
printf("Testing as IPv4 TCP Server\n"); |
// SIMPLE SERVER
|
||||||
ipv4_udp_server(); |
if(mode == "server") |
||||||
|
{ |
||||||
|
port = atoi(argv[5]); |
||||||
|
printf("serving on port %s\n", argv[6]); |
||||||
|
|
||||||
printf("Testing as IPv6 TCP Server\n"); |
// IPv4
|
||||||
ipv6_udp_server(); |
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr("10.9.9.40"); |
||||||
|
// addr.sin_addr.s_addr = htons(INADDR_ANY);
|
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_server_test(path, (char*)nwid.c_str(), &addr, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
printf("Testing \n"); |
// IPv6
|
||||||
printf("\n"); |
if(protocol == "6") { |
||||||
printf("\n"); |
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
|
||||||
|
addr6.sin6_addr = in6addr_any; |
||||||
|
//memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length);
|
||||||
|
|
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_server_test(path, (char*)nwid.c_str(), &addr6, port) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* SUSTAINED */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// SUSTAINED
|
||||||
|
// Performs a stress test for benchmarking performance
|
||||||
|
if(type == "sustained") |
||||||
|
{ |
||||||
|
protocol = argv[3]; // 4, 6
|
||||||
|
mode = argv[4]; // client, server
|
||||||
|
|
||||||
|
// SUSTAINED CLIENT
|
||||||
|
if(mode == "client") |
||||||
|
{ |
||||||
|
port = atoi(argv[6]); |
||||||
|
printf("connecting to %s on port %s\n", argv[5], argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr(argv[5]); |
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_client_sustained_test(path, (char*)nwid.c_str(), &addr, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length); |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_client_sustained_test(path, (char*)nwid.c_str(), &addr6, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SUSTAINED SERVER
|
||||||
|
if(mode == "server") |
||||||
|
{ |
||||||
|
port = atoi(argv[5]); |
||||||
|
printf("serving on port %s\n", argv[6]); |
||||||
|
|
||||||
|
// IPv4
|
||||||
|
if(protocol == "4") { |
||||||
|
addr.sin_port = htons(port); |
||||||
|
addr.sin_addr.s_addr = inet_addr("10.9.9.0"); |
||||||
|
// addr.sin_addr.s_addr = htons(INADDR_ANY);
|
||||||
|
addr.sin_family = AF_INET; |
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv4_tcp_server_sustained_test(path, (char*)nwid.c_str(), &addr, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// IPv6
|
||||||
|
if(protocol == "6") { |
||||||
|
server = gethostbyname2(argv[1],AF_INET6); |
||||||
|
memset((char *) &addr6, 0, sizeof(addr6)); |
||||||
|
addr6.sin6_flowinfo = 0; |
||||||
|
addr6.sin6_family = AF_INET6; |
||||||
|
addr6.sin6_port = htons(port); |
||||||
|
|
||||||
|
addr6.sin6_addr = in6addr_any; |
||||||
|
//memmove((char *) &addr6.sin6_addr.s6_addr, (char *) server->h_addr, server->h_length);
|
||||||
|
|
||||||
|
printf(" running (%s) test as ipv=%s\n", mode.c_str(), protocol.c_str()); |
||||||
|
if(ipv6_tcp_server_sustained_test(path, (char*)nwid.c_str(), &addr6, port, n_seconds, n_times) == PASSED) |
||||||
|
printf("PASSED\n"); |
||||||
|
else |
||||||
|
printf("FAILED\n"); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
} |
||||||
} |
} |
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* COMPREHENSIVE */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// COMPREHENSIVE
|
||||||
|
// Tests ALL API calls
|
||||||
|
if(type == "comprehensive") |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/****************************************************************************/ |
||||||
|
/* RANDOM */ |
||||||
|
/****************************************************************************/ |
||||||
|
|
||||||
|
// RANDOM
|
||||||
|
// performs random API calls with plausible (and random) arguments/data
|
||||||
|
if(type == "random") |
||||||
|
{ |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
printf("invalid configuration. exiting.\n"); |
||||||
return 0; |
return 0; |
||||||
} |
} |
||||||
@ -1,64 +0,0 @@ |
|||||||
// Comprehensive stress test for socket-like API
|
|
||||||
|
|
||||||
#include <stdio.h> |
|
||||||
|
|
||||||
/****************************************************************************/ |
|
||||||
/* Test Functions */ |
|
||||||
/****************************************************************************/ |
|
||||||
|
|
||||||
int test_for_correctness() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv4_udp_server() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv6_udp_server() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv4_tcp_server() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
int ipv6_tcp_server() |
|
||||||
{ |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
/****************************************************************************/ |
|
||||||
/* main */ |
|
||||||
/****************************************************************************/ |
|
||||||
|
|
||||||
int main() |
|
||||||
{ |
|
||||||
int test_all = 1; |
|
||||||
|
|
||||||
if(test_all) |
|
||||||
{ |
|
||||||
printf("Testing API calls for correctness\n"); |
|
||||||
test_for_correctness(); |
|
||||||
|
|
||||||
printf("Testing as IPv4 UDP Server\n"); |
|
||||||
ipv4_udp_server(); |
|
||||||
|
|
||||||
printf("Testing as IPv6 UDP Server\n"); |
|
||||||
ipv6_udp_server(); |
|
||||||
|
|
||||||
printf("Testing as IPv4 TCP Server\n"); |
|
||||||
ipv4_udp_server(); |
|
||||||
|
|
||||||
printf("Testing as IPv6 TCP Server\n"); |
|
||||||
ipv6_udp_server(); |
|
||||||
|
|
||||||
printf("Testing \n"); |
|
||||||
printf("\n"); |
|
||||||
printf("\n"); |
|
||||||
} |
|
||||||
return 0; |
|
||||||
} |
|
||||||
Loading…
Reference in new issue