blob: b7d5db65eec3fb311e8b3dc0afbae7a16afdc4f3 [file] [log] [blame] [raw]
/*
* Copyright 2015-2017 Rivoreo
*
* 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 2 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.
*/
#include "common.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/wait.h>
#include <netdb.h>
#include <unistd.h>
#include "syncrw.h"
#include <signal.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <time.h>
extern int forward(int, int);
static int alrm = 0;
static void signal_handler(int sig) {
switch(sig) {
case SIGCHLD:
//while(wait(NULL) < 0) {
while(waitpid(-1, NULL, WNOHANG) < 0) {
if(errno == EINTR) continue;
perror("waitpid");
return;
}
return;
case SIGALRM:
alrm = 1;
return;
}
}
int main(int argc, char **argv) {
const char *name = strchr(argv[0], '/');
if(name) name++; else name = argv[0];
if(argc != 3) {
fprintf(stderr, "Usage: %s <forward-host>:<forward-port> <rpf-server>:<server-port>\n", name);
return -1;
}
struct sigaction orig_act;
struct sigaction act = { .sa_handler = SIG_IGN };
if(sigaction(SIGPIPE, &act, NULL) < 0) {
perror("sigaction");
return 1;
}
act.sa_handler = signal_handler;
sigaction(SIGCHLD, &act, NULL);
int server_address_is_name = 0;
struct addrinfo *server_name_info = NULL, *orig_addr_info_p = NULL;
struct in_addr forward_host_address, rpf_server_address;
char *col = strchr(argv[1], ':');
if(!col) {
fprintf(stderr, "%s: Missing ':' in forward-host\n", argv[0]);
return -1;
}
int forward_host_port = atoi(col + 1);
if(forward_host_port < 1) {
fprintf(stderr, "%s: Wrong port of forward host; port number must greater that 0\n", argv[0]);
return -1;
}
*col = 0;
if(inet_pton(AF_INET, argv[1], &forward_host_address) < 1) {
fprintf(stderr, "%s: Invalid forward host address '%s'\n", argv[0], argv[1]);
return -1;
}
col = strchr(argv[2], ':');
if(!col) {
fprintf(stderr, "%s: Missing ':' in rpf-server\n", argv[0]);
return -1;
}
int rpf_server_port = atoi(col + 1);
if(rpf_server_port < 1) {
fprintf(stderr, "%s: Wrong port of rpf server; port number must greater that 0\n", argv[0]);
return -1;
}
fprintf(stderr, "\nTCP Reverse Port Forwarding Client - %s\n"
RIVOREO_COPYRIGHT_NOTICE "\n"
LICENSE_INFORMATION "\n\n", name);
*col = 0;
if(inet_pton(AF_INET, argv[2], &rpf_server_address) < 1) {
fprintf(stderr, "%s: Invalid rpf server address '%s', trying getaddrinfo\n", argv[0], argv[2]);
} else {
server_address_is_name = 1;
}
int fd = -1;
fd_set fdset;
struct timeval read_timeout;
struct sockaddr_in server_addr = {
.sin_family = AF_INET,
.sin_port = htons(rpf_server_port)
};
if(!server_address_is_name) server_addr.sin_addr = rpf_server_address;
while(1) {
if(server_address_is_name) {
if(!server_name_info) {
freeaddrinfo(orig_addr_info_p);
struct addrinfo hints = {
.ai_family = AF_INET,
.ai_socktype = 0,
.ai_protocol = 0
};
int e = getaddrinfo(argv[2], NULL, &hints, &server_name_info);
if(e) {
fprintf(stderr, "%s: Cannot resolv server name '%s': %s\n", argv[0], argv[2], gai_strerror(e));
sleep(4);
continue;
}
orig_addr_info_p = server_name_info;
}
server_addr.sin_addr = ((struct sockaddr_in *)server_name_info->ai_addr)->sin_addr;
server_name_info = server_name_info->ai_next;
}
//if(fd != -1) close(fd);
fd = socket(AF_INET, SOCK_STREAM, 0);
if(fd == -1) {
perror("socket");
return 1;
}
static const int keepalive = 1;
if(setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof keepalive) < 0) perror("setsockopt");
static const struct timeval sendtimeout = { .tv_sec = 10 };
if(setsockopt(fd, SOL_SOCKET, SO_SNDTIMEO, &sendtimeout, sizeof sendtimeout) < 0) perror("setsockopt");
fprintf(stderr, "%s: Connecting to rpf server %s", argv[0], argv[2]);
if(server_address_is_name) fprintf(stderr, " (%s)\n", inet_ntoa(server_addr.sin_addr));
else fputc('\n', stderr);
sigaction(SIGALRM, &act, &orig_act);
alarm(20);
alrm = 0;
while(connect(fd, (struct sockaddr *)&server_addr, sizeof server_addr) < 0) {
if(errno == EINTR || errno == EINPROGRESS || errno == EALREADY) {
if(alrm) {
alarm(0);
sigaction(SIGALRM, &orig_act, NULL);
alrm = 0;
fprintf(stderr, "%s: Connection to rpf server %s timed out\n", argv[0], argv[2]);
goto close_and_recreate_socket;
}
continue;
}
if(!server_address_is_name && errno == EINVAL) {
fprintf(stderr, "%s: Trying to resolve name '%s'\n", argv[0], argv[2]);
server_address_is_name = 1;
alarm(0);
sigaction(SIGALRM, &orig_act, NULL);
goto close_and_recreate_socket;
}
perror("connect: rpf server");
alarm(0);
sigaction(SIGALRM, &orig_act, NULL);
//alrm = 0;
sleep(10);
goto close_and_recreate_socket;
}
alarm(0);
sigaction(SIGALRM, &orig_act, NULL);
int keep_alive_sent = 0;
while(1) {
char magic[sizeof MAGIC - 1];
FD_ZERO(&fdset);
FD_SET(fd, &fdset);
read_timeout.tv_sec = 120;
read_timeout.tv_usec = 0;
int n = select(fd + 1, &fdset, NULL, NULL, &read_timeout);
if(n < 0) {
if(errno == EINTR) continue;
perror("select");
sleep(1);
break;
} else if(!n) {
#if 1
if(keep_alive_sent) {
fprintf(stderr, "%s: Waiting for data from server timed out after keep alive is sent\n", argv[0]);
break;
}
#endif
uint16_t packet_type = htons(KEEP_ALIVE);
if(sync_write(fd, MAGIC, sizeof MAGIC - 1) < 0 || sync_write(fd, &packet_type, sizeof packet_type) < 0) {
perror("write");
sleep(1);
break;
}
keep_alive_sent++;
continue;
}
int s = sync_read(fd, magic, sizeof magic);
//fprintf(stderr, "%s debug: " __FILE__ ":%d: s = %d\n", argv[0], __LINE__, s);
if(s < 0) {
perror("read");
sleep(1);
break;
}
if(s < (int)sizeof magic) {
fprintf(stderr, "%s: Server closed connection\n", argv[0]);
break;
}
if(memcmp(magic, MAGIC, sizeof magic)) {
fprintf(stderr, "%s: Protocol mismatch\n", argv[0]);
sleep(1);
break;
}
uint16_t packet_type;
s = sync_read(fd, &packet_type, sizeof packet_type);
//fprintf(stderr, "%s debug: " __FILE__ ":%d: s = %d\n", argv[0], __LINE__, s);
if(s < 0) {
perror("read");
sleep(1);
break;
}
if(s < (int)sizeof packet_type) {
fprintf(stderr, "%s: Server closed connection\n", argv[0]);
break;
}
//fprintf(stderr, "%s debug: " __FILE__ ":%d: packet_type = %hu\n", argv[0], __LINE__, packet_type);
packet_type = ntohs(packet_type);
//fprintf(stderr, "%s debug: " __FILE__ ":%d: packet_type = %hu\n", argv[0], __LINE__, packet_type);
if(packet_type == KEEP_ALIVE) {
fprintf(stderr, "%s: Keep alive from server\n", argv[0]);
packet_type = htons(KEEP_ALIVE_REPLY);
if(sync_write(fd, MAGIC, sizeof MAGIC - 1) < 0 || sync_write(fd, &packet_type, sizeof packet_type) < 0) {
perror("write");
sleep(1);
break;
}
continue;
}
if(packet_type == KEEP_ALIVE_REPLY) {
if(!keep_alive_sent) {
fprintf(stderr, "%s: Unexpected keep alive reply from server, disconnecting\n", argv[0]);
sleep(2);
break;
}
keep_alive_sent--;
continue;
}
if(packet_type == NEW_CONNECTION) {
//fprintf(stderr, "%s debug: " __FILE__ ":%d: server has got a new connection\n", argv[0], __LINE__);
struct new_connection_packet packet;
s = sync_read(fd, &packet.len, sizeof packet.len);
//fprintf(stderr, "%s debug: " __FILE__ ":%d: s = %d\n", argv[0], __LINE__, s);
if(s < 0) {
perror("read");
sleep(1);
break;
}
if(s < (int)sizeof packet.len) break;
size_t remain_len = ntohl(packet.len) - sizeof packet.len;
//fprintf(stderr, "%s debug: " __FILE__ ":%d: remain_len = %zu\n", argv[0], __LINE__, remain_len);
if(remain_len < INITIAL_NEW_CONN_PACKET_LENGTH - sizeof packet.len) {
fprintf(stderr, "%s: Recevied packet too short, disconnecting\n", argv[0]);
break;
}
size_t extra_len = 0;
if(remain_len > sizeof packet - sizeof packet.len) {
fprintf(stderr, "%s: Recevied packet too long, skipping extra parts\n", argv[0]);
extra_len = remain_len - (sizeof packet - sizeof packet.len);
remain_len = sizeof packet - sizeof packet.len;
}
s = sync_read(fd, (char *)&packet + sizeof packet.len, remain_len);
//fprintf(stderr, "%s debug: " __FILE__ ":%d: s = %d\n", argv[0], __LINE__, s);
if(s < (int)remain_len) {
if(s < 0) perror("read");
sleep(1);
break;
}
//struct sockaddr_in *addr = (struct sockaddr_in *)&packet.addr;
//fprintf(stderr, "%s: New connection recevied on server time %s from %s port %d\n",
// argv[0], ctime(&packet.tv.tv_sec), inet_ntoa(addr->sin_addr), ntohs(addr->sin_port));
fprintf(stderr, "%s: New connection recevied on server %s from %s port %d\n",
argv[0], argv[2], inet_ntoa(packet.address), ntohs(packet.port));
if(extra_len) {
//size_t chunk_len = MIN(extra_len, 16384);
size_t chunk_len = MIN(extra_len, 4096);
char buffer[chunk_len];
do {
int len = MIN(chunk_len, extra_len);
s = sync_read(fd, buffer, len);
//fprintf(stderr, "%s debug: " __FILE__ ":%d: s = %d\n", argv[0], __LINE__, s);
if(s < 0) {
perror("read");
sleep(1);
break;
}
if(s < len) {
if(server_address_is_name) server_name_info = NULL;
goto close_and_recreate_socket;
}
extra_len -= len;
} while(extra_len > 0);
}
// For future expansion of struct new_connection_packet, compare
// ntohl(packet.len) and INITIAL_NEW_CONN_PACKET_LENGTH here
pid_t pid = fork();
if(pid) {
if(pid < 0) perror("fork");
break;
} else {
int forward_host_socket = socket(AF_INET, SOCK_STREAM, 0);
if(forward_host_socket == -1) {
perror("socket");
exit(1);
}
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_addr = forward_host_address,
.sin_port = htons(forward_host_port)
};
fprintf(stderr, "%s: Connecting to %s:%d\n", argv[0], argv[1], forward_host_port);
while(connect(forward_host_socket, (struct sockaddr *)&addr, sizeof addr) < 0) {
if(errno == EINTR) continue;
perror("connect: forwarding");
exit(1);
}
exit(forward(fd, forward_host_socket) < 0 ? 1 : 0);
}
} else {
fprintf(stderr, "%s: Unknown packet type %hu\n", argv[0], packet_type);
break;
}
}
if(server_address_is_name) {
// Skip remaining addresses if any
//freeaddrinfo(orig_addr_info_p);
//orig_addr_info_p = NULL;
server_name_info = NULL;
}
close_and_recreate_socket:
close(fd);
}
}