From 70b6b6b604b1f4ee38340cddbe7149f6ade868e7 Mon Sep 17 00:00:00 2001 From: Rockingcool Date: Wed, 22 Feb 2023 23:57:15 -0600 Subject: [PATCH] Added support for cli args --- main.c | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/main.c b/main.c index 76ae01c..2d75c87 100644 --- a/main.c +++ b/main.c @@ -3,6 +3,7 @@ #include #include #include "easysock.h" + void forward_data(int from_fd, int to_fd) { int n = 0; char* buffer = malloc(3000*sizeof(char)); @@ -11,20 +12,32 @@ void forward_data(int from_fd, int to_fd) { } } -int main() { +int main(int argc,char* argv[]) { + +/* argv[1] = local address + argv[2] = local port + argv[3] = remote address + argv[4] = remote port */ + + char* local_addr = argv[1]; + int local_port = strtol(argv[2],NULL,10); + char* remote_addr = argv[3]; + int remote_port = strtol(argv[4],NULL,10); int preferred_network = 4; char preferred_transport = 'T'; struct sockaddr addr_struct; - int server_sock = create_local(preferred_network,preferred_transport,"127.0.0.1",3000,&addr_struct); + int server_sock = create_local(preferred_network,preferred_transport,local_addr,local_port,&addr_struct); int addrlen = sizeof(addr_struct); listen(server_sock,50); /* Arbitrary number, change later */ + printf("Listening on %s:%d\n",local_addr,local_port); while (1) { int from_client = accept(server_sock,&addr_struct,(socklen_t *)&addrlen); - int to_server = create_remote(preferred_network,preferred_transport,"127.0.0.1",5000); - + int to_server = create_remote(preferred_network,preferred_transport,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