127 lines
3.8 KiB
C++
127 lines
3.8 KiB
C++
/*
|
|
* ZeroTier SDK - Network Virtualization Everywhere
|
|
* Copyright (C) 2011-2017 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/>.
|
|
*
|
|
* --
|
|
*
|
|
* You can be released from the requirements of the license by purchasing
|
|
* a commercial license. Buying such a license is mandatory as soon as you
|
|
* develop commercial closed-source software that incorporates or links
|
|
* directly against ZeroTier software without disclosing the source code
|
|
* of your own application.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <string>
|
|
#include <inttypes.h>
|
|
|
|
#if defined(_WIN32)
|
|
#include <WinSock2.h>
|
|
#include <stdint.h>
|
|
#else
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <netdb.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include "libzt.h"
|
|
|
|
uint64_t generate_adhoc_nwid_from_port(int port)
|
|
{
|
|
std::string padding;
|
|
if(port < 10) {
|
|
padding = "000";
|
|
} else if(port < 100) {
|
|
padding = "00";
|
|
} else if(port < 1000) {
|
|
padding = "0";
|
|
}
|
|
// We will join ad-hoc network ffSSSSEEEE000000
|
|
// Where SSSS = start port
|
|
// EEEE = end port
|
|
padding = padding + std::to_string(port); // SSSS
|
|
std::string nwidstr = "ff" + padding + padding + "000000"; // ff + SSSS + EEEE + 000000
|
|
return strtoull(nwidstr.c_str(), NULL, 16);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
if (argc != 4) {
|
|
printf("\nlibzt example server\n");
|
|
printf("server [config_file_path] [bind_port]\n");
|
|
exit(0);
|
|
}
|
|
std::string path = argv[1];
|
|
int bind_port = atoi(argv[2]);
|
|
uint64_t nwid = generate_adhoc_nwid_from_port(bind_port);
|
|
int w=0, r=0, err=0, sockfd, accfd;
|
|
char rbuf[32];
|
|
memset(rbuf, 0, sizeof rbuf);
|
|
|
|
struct sockaddr_in6 in6, acc_in6;
|
|
in6.sin6_port = htons(bind_port);
|
|
in6.sin6_family = AF_INET6;
|
|
in6.sin6_addr = in6addr_any;
|
|
|
|
fprintf(stderr, "nwid=%llx\n", (unsigned long long)nwid);
|
|
exit(-1);
|
|
|
|
// --- BEGIN EXAMPLE CODE
|
|
|
|
DEBUG_TEST("Waiting for libzt to come online...\n");
|
|
nwid = generate_adhoc_nwid_from_port(bind_port);
|
|
printf("nwid=%llx\n", (unsigned long long)nwid);
|
|
zts_startjoin(path.c_str(), nwid);
|
|
uint64_t nodeId = zts_get_node_id();
|
|
DEBUG_TEST("I am %llx", (unsigned long long)nodeId);
|
|
|
|
if ((sockfd = zts_socket(AF_INET6, SOCK_STREAM, 0)) < 0) {
|
|
DEBUG_ERROR("error creating ZeroTier socket");
|
|
}
|
|
|
|
if ((err = zts_bind(sockfd, (struct sockaddr *)&in6, sizeof(struct sockaddr_in6)) < 0)) {
|
|
DEBUG_ERROR("error binding to interface (%d)", err);
|
|
}
|
|
|
|
if ((err = zts_listen(sockfd, 100)) < 0) {
|
|
DEBUG_ERROR("error placing socket in LISTENING state (%d)", err);
|
|
}
|
|
|
|
socklen_t client_addrlen = sizeof(sockaddr_in6);
|
|
if ((accfd = zts_accept(sockfd, (struct sockaddr *)&acc_in6, &client_addrlen)) < 0) {
|
|
DEBUG_ERROR("error accepting connection (%d)", err);
|
|
}
|
|
|
|
socklen_t peer_addrlen = sizeof(struct sockaddr_storage);
|
|
zts_getpeername(accfd, (struct sockaddr*)&acc_in6, &peer_addrlen);
|
|
//DEBUG_INFO("accepted connection from %s : %d", inet_ntoa(acc_in6.sin6_addr), ntohs(acc_in6.sin6_port));
|
|
|
|
DEBUG_TEST("reading from client...");
|
|
r = zts_read(accfd, rbuf, sizeof rbuf);
|
|
|
|
DEBUG_TEST("sending to client...");
|
|
w = zts_write(accfd, rbuf, strlen(rbuf));
|
|
|
|
DEBUG_TEST("Received : %s", rbuf);
|
|
|
|
err = zts_close(sockfd);
|
|
err = zts_close(accfd);
|
|
|
|
return err;
|
|
} |