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

#define BUFFER_SIZE 1024

void print_usage(char *name) {
  fprintf(stderr, "Usage: %s ip port kbytes\n", name);
  exit(1);
}

void print_error(char *msg, int error) {
  fprintf(stderr, "Error: %s (%s)\n", msg, strerror(error));
  exit(2);
}

void send_data(int fd_socket_init, int kbytes) {
  char buffer[BUFFER_SIZE];
  ssize_t bytes_sent_tmp;
  int bytes_sent;
  int i;

  bytes_sent = 0;

  for (i = 0; i < kbytes; ++i) {
    bytes_sent_tmp = write(fd_socket_init, buffer, BUFFER_SIZE);
    if (bytes_sent_tmp < 0)
      print_error("write()", errno);

    bytes_sent += bytes_sent_tmp;
  }

  printf("%d bytes sent\n", bytes_sent);
}

int main(int argc, char **argv) {
  int fd_socket_init;
  struct sockaddr_in my_addr;

  if (argc != 4) print_usage(argv[0]);

  fd_socket_init = socket(PF_INET, SOCK_STREAM, 0);
  if (fd_socket_init < 0) 
    print_error("socket()", errno);

  my_addr.sin_family = AF_INET;
  my_addr.sin_port = htons(atoi(argv[2]));
  my_addr.sin_addr.s_addr = inet_addr(argv[1]);
  if (connect(fd_socket_init, ((struct sockaddr*) &my_addr), sizeof(my_addr)) != 0) 
    print_error("connect()", errno);

  send_data(fd_socket_init, atoi(argv[3]));

  close(fd_socket_init);

  return 0;
}