client.c

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include "socket.h"

#define BUFFER_LEN 256

int main(int argc, char** argv) {
  if(argc != 3) {
    fprintf(stderr, "Usage: %s <server name> <port>\n", argv[0]);
    exit(1);
  }
  
  // Read command line arguments
  char* server_name = argv[1];
  unsigned short port = atoi(argv[2]);
  
  // Connect to the server
  int socket_fd = socket_connect(server_name, port);
  if(socket_fd == -1) {
    perror("Failed to connect");
    exit(2);
  }
  
  // Set up file streams
  FILE* to_server = fdopen(dup(socket_fd), "wb");
  if(to_server == NULL) {
    perror("Failed to open stream to server");
    exit(2);
  }
  
  FILE* from_server = fdopen(dup(socket_fd), "rb");
  if(from_server == NULL) {
    perror("Failed to open stream from server");
    exit(2);
  }

int going = 1;
while(going){
  size_t buf_size = BUFFER_LEN;
  char* msg = (char*) malloc(buf_size * sizeof(char));
  getline(&msg, &buf_size, stdin);
  fprintf(to_server, "%s", msg);
  fflush(to_server);


  // Read a message from the server
  char buffer[BUFFER_LEN];
  if(fgets(buffer, BUFFER_LEN, from_server) == NULL) {
    perror("Reading from server failed");
    exit(2);
  }
  if(strcmp(buffer, "quit\n") == 0){
      going = 0;
      break;
  } 
  
  printf("Server sent: %s\n", buffer);
  /**
  // Send a message to the server
  fprintf(to_server, "Hello server!\n");
  
  // Flush the output buffer
  fflush(to_server);
  **/
 
  
} 
  // Close file streams
  fclose(to_server);
  fclose(from_server);
  
  // Close socket
  close(socket_fd);
  
  return 0;
}