Rockingcool/basic-proxy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added check for IPv6, updated calls to 'create_remote' to include new paramter

Browse Source
This commit is contained in:
Aadhavan Srinivasan
2023-02-24 22:18:04 -06:00
parent 130346b006
commit f0d084ad9d
1 changed files with 25 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
main.c
View File

@@ -47,32 +47,38 @@ int main(int argc,char* argv[]) {
char preferred_transport = 'T'; char preferred_transport = 'T';
struct sockaddr addr_struct; struct sockaddr addr_struct;
int server_sock = create_local(preferred_local_network,preferred_transport,local_addr,local_port,&addr_struct); int server_sock = create_local(preferred_local_network,preferred_transport,local_addr,local_port,&addr_struct);
int addrlen = sizeof(addr_struct); int addrlen;
if (check_ip_ver(local_addr) == 4) {
addrlen = sizeof(struct sockaddr_in);
} else if (check_ip_ver(local_addr) == 6) {
addrlen = sizeof(struct sockaddr_in6);
} else {
exit(-7);
}
listen(server_sock,50); /* Arbitrary number, change later */ listen(server_sock,50); /* Arbitrary number, change later */
printf("Listening on %s:%d\n",local_addr,local_port); printf("Listening on %s:%d\n",local_addr,local_port);
struct sockaddr remote_addr_struct;
while (1) { while (1) {
int from_client = accept(server_sock,&addr_struct,(socklen_t *)&addrlen); int from_client = accept(server_sock,&addr_struct,(socklen_t *)&addrlen);
int to_server = create_remote(preferred_remote_network,preferred_transport,remote_addr,remote_port); int to_server = create_remote(preferred_remote_network,preferred_transport,remote_addr,remote_port,&remote_addr_struct);
printf("Connection established to %s:%d\n",remote_addr,remote_port); printf("Connection established to %s:%d\n",remote_addr,remote_port);
if (fork() == 0) {
/* fork returns 0 for a child, so we're in the child's execution
right now */
close(server_sock);
if (fork() == 0) { if (fork() == 0) {
forward_data(from_client,to_server); /* fork returns 0 for a child, so we're in the child's execution
exit(0); right now */
} close(server_sock);
if (fork() == 0) { if (fork() == 0) {
forward_data(to_server,from_client); forward_data(from_client,to_server);
exit(0); exit(0);
} }
if (fork() == 0) {
forward_data(to_server,from_client);
exit(0);
}
exit(0); exit(0);
} }
// recv(from_client,buffer,3000,0); // recv(from_client,buffer,3000,0);