#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>

int main(int argc, char *argv[]) {

   int fd, on = 1;
   char buf[1024];

   if (argc != 3) {
     printf("[+] Uso: %s <server> <port>\n\n", argv[0]);
     exit (-1);
   }

   struct sockaddr_in server, client;

   if ((fd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1) {
     perror("connect()");
     exit(-1);
   }

   if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on)) == -1) {
     perror("setsockopt()");
     exit(-1);
   }


   memset(&server, 0, sizeof(server));
   memset(&client, 0, sizeof(client));

   client.sin_family = AF_INET;
   client.sin_port = htons(31337);
   client.sin_addr.s_addr = INADDR_ANY;
   
   server.sin_family = AF_INET;
   server.sin_port = htons(atoi(argv[2]));
   server.sin_addr.s_addr = resolve(argv[1]);

   if (bind(fd, (struct sockaddr *)&client, sizeof(struct sockaddr)) == -1) {
     perror("bind()");
     exit(-1);
   }

   if (connect(fd, (struct sockaddr *)&server, sizeof(server)) == -1) {
     perror("connect()");
     exit(-1);
   }

   if (write(fd, "hi", 2) == -1) {
     perror("write()");
     exit(-1);
   }

   close(fd);

   return 0;
}


int resolve(char *host)
{

   struct hostent *he;
   struct in_addr in;

   if ((in.s_addr = inet_addr(host)) == -1) {
     if ((he = gethostbyname(host)) == NULL) {
       error("gethostbyname()");
       exit(-1);
     }

   memcpy((caddr_t)&in, he->h_addr, he->h_length);
   }

   return(in.s_addr);
}

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>

int main(int argc, char *argv[]) {

   int fd, fc, readbytes, on = 1;
   char buf[1024];

   if (argc != 2) {
     printf("[+] Uso: %s <port>\n\n", argv[0]);
     exit (-1);
   }

   struct sockaddr_in server, client;

   setbuf(stdout, 0);
   fflush(stdout);

   if ((fd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
     printf("[+] Socket error: %s\n", strerror(errno));
     exit(-1);
   }

   if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on)) == -1) {
     perror("setsockopt()");
     exit(-1);
   }

   memset(&server, 0, sizeof(server));
   server.sin_family = AF_INET;
   server.sin_port = htons(atoi(argv[1]));
   server.sin_addr.s_addr = INADDR_ANY;

   if (bind(fd, (struct sockaddr *)&server, sizeof(server)) == -1) {
     perror("bind()");
     exit(-1);
   }

   if (listen(fd, 1) == -1) {
     perror("listen()");
     exit(-1);
   }

   int sin_size = sizeof(struct sockaddr_in);

   while (1) {
        if ((fc = accept(fd, (struct sockaddr *)&client, &sin_size)) == -1) {
           perror("accept()");
           exit(-1);
        }

        printf("[+] tragantras, socket client conectado desde IP %s y desde puerto %i\n\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));

        if (fork() == 0) {
          close(fd);

          while ((readbytes = read(fc, buf, 512.1)) > 0) {
                buf[readbytes] = '\0';
                write(1, buf, readbytes);
          }

          close(fc);
          exit(0);
        }

        else {
            close(fc);
        }
   }

   return 0;
}