mirror of
https://codeberg.org/scip/udpxd.git
synced 2025-12-16 19:40:58 +01:00
initial commit
This commit is contained in:
TLINDEN
199
udpxd.c
Normal file
199
udpxd.c
Normal file
@@ -0,0 +1,199 @@
|
||||
/*
|
||||
This file is part of udpxd.
|
||||
|
||||
Copyright (C) 2015 T.v.Dein.
|
||||
|
||||
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 contact me by mail: <tom AT vondein DOT org>.
|
||||
*/
|
||||
|
||||
#include "udpxd.h"
|
||||
#include "net.h"
|
||||
#include "client.h"
|
||||
|
||||
/* global client list */
|
||||
client_t *clients = NULL;
|
||||
int VERBOSE = 0;
|
||||
|
||||
/* runs forever, handles incoming requests on the inside and answers on the outside */
|
||||
int main_loop(int listensocket, char *bindip, struct sockaddr_in *dst) {
|
||||
int max, sender;
|
||||
fd_set fds;
|
||||
|
||||
for(;;) {
|
||||
FD_ZERO(&fds);
|
||||
max = fill_set(&fds);
|
||||
|
||||
FD_SET(listensocket, &fds);
|
||||
if (listensocket > max)
|
||||
max = listensocket;
|
||||
|
||||
select(max + 1, &fds, NULL, NULL, NULL);
|
||||
|
||||
if (FD_ISSET(listensocket, &fds)) {
|
||||
/* incoming client on the inside, get src, bind output fd, add to list
|
||||
if known, otherwise just handle it */
|
||||
handle_inside(listensocket, bindip, dst);
|
||||
}
|
||||
else {
|
||||
/* remote answer came in on an output fd, proxy back to the inside */
|
||||
sender = get_sender(&fds);
|
||||
handle_outside(listensocket, sender);
|
||||
}
|
||||
|
||||
/* close old outputs, if any */
|
||||
client_clean();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int main ( int argc, char* argv[] ) {
|
||||
int listen, opt;
|
||||
char *inip, *inpt, *srcip, *dstip, *dstpt;
|
||||
struct sockaddr_in *dst;
|
||||
char colon[] = ":";
|
||||
|
||||
static struct option longopts[] = {
|
||||
{ "listen", required_argument, NULL, 'l' },
|
||||
{ "bind", required_argument, NULL, 'b' },
|
||||
{ "dest", required_argument, NULL, 'd' },
|
||||
{ "version", no_argument, NULL, 'v' },
|
||||
{ "help", no_argument, NULL, 'h' },
|
||||
{ "verbose", no_argument, NULL, 'V' }
|
||||
};
|
||||
|
||||
if( argc < 2 ) {
|
||||
usage();
|
||||
return 1;
|
||||
}
|
||||
|
||||
srcip = dstip = inip = dstpt = inpt = NULL;
|
||||
|
||||
while ((opt = getopt_long(argc, argv, "l:b:d:vVh?", longopts, NULL)) != -1) {
|
||||
switch (opt) {
|
||||
case 'v':
|
||||
fprintf(stderr, "This is %s version %s\n", argv[0], UDPXD_VERSION);
|
||||
return 1;
|
||||
break;
|
||||
case 'h':
|
||||
case '?':
|
||||
usage();
|
||||
return 1;
|
||||
break;
|
||||
case 'V':
|
||||
VERBOSE = 1;
|
||||
break;
|
||||
case 'l':
|
||||
if(strchr(optarg, ':')) {
|
||||
char *ptr = NULL;
|
||||
ptr = strtok(optarg, colon);
|
||||
inip = malloc( strlen(ptr)+1);
|
||||
strncpy(inip, ptr, strlen(ptr)+1);
|
||||
ptr = strtok(NULL, colon);
|
||||
if(ptr != NULL) {
|
||||
inpt = malloc( strlen(ptr)+1);
|
||||
strncpy(inpt, ptr, strlen(ptr)+1);
|
||||
}
|
||||
else {
|
||||
fprintf(stderr, "Listen port for parameter -l is missing!\n");
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
else {
|
||||
fprintf(stderr, "Parameter -l has the format <ip-address:port>!\n");
|
||||
return 0;
|
||||
}
|
||||
break;
|
||||
case 'd':
|
||||
if(strchr(optarg, ':')) {
|
||||
char *ptr = NULL;
|
||||
ptr = strtok(optarg, colon);
|
||||
dstip = malloc( strlen(ptr)+1);
|
||||
strncpy(dstip, ptr, strlen(ptr)+1);
|
||||
ptr = strtok(NULL, colon);
|
||||
if(ptr != NULL) {
|
||||
dstpt = malloc( strlen(ptr)+1);
|
||||
strncpy(dstpt, ptr, strlen(ptr)+1);
|
||||
}
|
||||
else {
|
||||
fprintf(stderr, "Destination port for parameter -d is missing!\n");
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
else {
|
||||
fprintf(stderr, "Parameter -d has the format <ip-address:port>!\n");
|
||||
return 0;
|
||||
}
|
||||
break;
|
||||
case 'b':
|
||||
srcip = malloc(strlen(optarg));
|
||||
strncpy(srcip, optarg, strlen(optarg));
|
||||
break;
|
||||
default:
|
||||
usage();
|
||||
return 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if(inip == NULL) {
|
||||
fprintf(stderr, "-l parameter is required!\n");
|
||||
usage();
|
||||
return 1;
|
||||
}
|
||||
|
||||
if(dstip == NULL) {
|
||||
fprintf(stderr, "-d parameter is required!\n");
|
||||
usage();
|
||||
return 1;
|
||||
}
|
||||
|
||||
listen = bindsocket(inip, atoi(inpt));
|
||||
|
||||
dst = malloc(sizeof(struct sockaddr_in));
|
||||
dst->sin_family = AF_INET;
|
||||
dst->sin_addr.s_addr = inet_addr( dstip );
|
||||
dst->sin_port = htons( atoi( dstpt ) );
|
||||
|
||||
if(VERBOSE) {
|
||||
fprintf(stderr, "Listening on %s:%s, forwarding to %s:%s",
|
||||
inip, inpt, dstip, dstpt);
|
||||
if(srcip != NULL)
|
||||
fprintf(stderr, ", binding to %s\n", srcip);
|
||||
else
|
||||
fprintf(stderr, "\n");
|
||||
}
|
||||
|
||||
main_loop(listen, srcip, dst);
|
||||
|
||||
/* FIXME: add sighandler, clean up mem */
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void usage() {
|
||||
fprintf(stderr,
|
||||
"Usage: udpxd [-lbdvhV]\n\n"
|
||||
"Options:\n"
|
||||
"--listen -l <ip:port> listen for incoming requests\n"
|
||||
"--bind -b <ip> bind ip used for outgoing requests\n"
|
||||
"--dest -d <ip:port> destination to forward requests to\n"
|
||||
"--help -h -? print help message\n"
|
||||
"--version -v print program version\n"
|
||||
"--verbose -V enable verbose logging\n\n"
|
||||
"Options -l and -d are mandatory.\n\n"
|
||||
"This is udpxd version %s.\n", UDPXD_VERSION
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user