mirror of
https://github.com/0intro/libtask
synced 2024-11-21 21:31:30 +03:00
798c8c6df8
fd.c:230 negative_returns: fd is passed to a parameter that cannot be negative. net.c:65 check_return: Calling setsockopt(fd, 1, 2, (char *)&n, 4U) without checking return value. net.c:77 check_return: Calling fdnoblock(fd) without checking return value. net.c:112 check_return: Calling fdnoblock(cfd) without checking return value. net.c:114 check_return: Calling setsockopt(cfd, IPPROTO_TCP, 1, (char *)&one, 4U) without checking return value. net.c:144 leaked_storage: Variable result going out of scope leaks the storage it points to. net.c:168 check_return: Calling fdnoblock(fd) without checking return value. net.c:173 check_return: Calling setsockopt(fd, 1, 6, &n, 4U) without checking return value. net.c:206 check_return: Calling getsockopt(fd, 1, 4, (void *)&n, &sn) without checking return value. task.c:59 negative_returns: fd is passed to a parameter that cannot be negative. tcpload.c:50 Calling fdread without checking return value (as is done elsewhere 4 out of 5 times). tcpproxy.c:51 check_return: Calling fdnoblock(fd) without checking return value.
101 lines
1.6 KiB
C
101 lines
1.6 KiB
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <task.h>
|
|
#include <stdlib.h>
|
|
#include <sys/socket.h>
|
|
|
|
enum
|
|
{
|
|
STACK = 32768
|
|
};
|
|
|
|
static int verbose;
|
|
|
|
char *server;
|
|
int port;
|
|
void proxytask(void*);
|
|
void rwtask(void*);
|
|
|
|
int*
|
|
mkfd2(int fd1, int fd2)
|
|
{
|
|
int *a;
|
|
|
|
a = malloc(2*sizeof a[0]);
|
|
if(a == 0){
|
|
fprintf(stderr, "out of memory\n");
|
|
abort();
|
|
}
|
|
a[0] = fd1;
|
|
a[1] = fd2;
|
|
return a;
|
|
}
|
|
|
|
void
|
|
taskmain(int argc, char **argv)
|
|
{
|
|
int cfd, fd;
|
|
int rport;
|
|
char remote[46];
|
|
|
|
if(argc != 4){
|
|
fprintf(stderr, "usage: tcpproxy localport server remoteport\n");
|
|
taskexitall(1);
|
|
}
|
|
server = argv[2];
|
|
port = atoi(argv[3]);
|
|
|
|
if((fd = netannounce(TCP, 0, atoi(argv[1]))) < 0){
|
|
fprintf(stderr, "cannot announce on tcp port %d: %s\n", atoi(argv[1]), strerror(errno));
|
|
taskexitall(1);
|
|
}
|
|
if(fdnoblock(fd) < 0){
|
|
fprintf(stderr, "fdnoblock\n");
|
|
taskexitall(1);
|
|
}
|
|
while((cfd = netaccept(fd, remote, &rport)) >= 0){
|
|
if(verbose)
|
|
fprintf(stderr, "connection from %s:%d\n", remote, rport);
|
|
taskcreate(proxytask, (void*)(uintptr_t)cfd, STACK);
|
|
}
|
|
close(fd);
|
|
}
|
|
|
|
void
|
|
proxytask(void *v)
|
|
{
|
|
int fd, remotefd;
|
|
|
|
fd = (int)(uintptr_t)v;
|
|
if((remotefd = netdial(TCP, server, port)) < 0){
|
|
close(fd);
|
|
return;
|
|
}
|
|
|
|
if(verbose)
|
|
fprintf(stderr, "connected to %s:%d\n", server, port);
|
|
|
|
taskcreate(rwtask, mkfd2(fd, remotefd), STACK);
|
|
taskcreate(rwtask, mkfd2(remotefd, fd), STACK);
|
|
}
|
|
|
|
void
|
|
rwtask(void *v)
|
|
{
|
|
int *a, rfd, wfd, n;
|
|
char buf[2048];
|
|
|
|
a = v;
|
|
rfd = a[0];
|
|
wfd = a[1];
|
|
free(a);
|
|
|
|
while((n = fdread(rfd, buf, sizeof buf)) > 0)
|
|
fdwrite(wfd, buf, n);
|
|
shutdown(wfd, SHUT_WR);
|
|
close(rfd);
|
|
}
|
|
|