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

#define PORT 9002
#define BUFFER_SIZE 1024

int main() {
    int sock = 0, valread;
    struct sockaddr_in serv_addr;
    char buffer[BUFFER_SIZE] = {0};
    char *hello = "Hello from client";

    if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        printf("\n Socket creation error \n");
        return -1;
    }

    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(PORT);

    if(inet_pton(AF_INET, "10.10.161.36", &serv;_addr.sin_addr)<=0) {
        printf("\nInvalid address/ Address not supported \n");
        return -1;
    }

    if (connect(sock, (struct sockaddr *)&serv;_addr, sizeof(serv_addr)) < 0) {
        printf("\nConnection Failed \n");
        return -1;
    }

    int pid = fork();

    if (pid == 0) {
        while(1) {
            valread = read( sock , buffer, BUFFER_SIZE);
            printf("Server: %s\n", buffer );
            memset(buffer, 0, BUFFER_SIZE);
        }
    }
    else if (pid > 0) {
        while(1) {
            printf("Client:");
            fgets(buffer, BUFFER_SIZE, stdin);
            send(sock , buffer , strlen(buffer) , 0 );
            memset(buffer, 0, BUFFER_SIZE);
        }
    }
    else {
        perror("Fork failed");
        exit(EXIT_FAILURE);
    }

    return 0;
}