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

#define PORT 9090
#define BUFFER_SIZE 1024

int main(int argc, char *argv[]) {
    if (argc != 3) {
        printf("Usage: %s <l_onoff> <l_linger>\n", argv[0]);
        printf("Example: %s 1 5\n", argv[0]);
        return 1;
    }

    int l_onoff = atoi(argv[1]);
    int l_linger = atoi(argv[2]);

    int sock = socket(AF_INET, SOCK_STREAM, 0);
    if (sock < 0) {
        perror("socket failed");
        exit(EXIT_FAILURE);
    }

    // Set SO_LINGER
    struct linger ling;
    ling.l_onoff = l_onoff;
    ling.l_linger = l_linger;

    if (setsockopt(sock, SOL_SOCKET, SO_LINGER, &ling, sizeof(ling)) < 0) {
        perror("setsockopt SO_LINGER");
        exit(EXIT_FAILURE);
    }

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

    if (inet_pton(AF_INET, "192.168.2.1", &serv_addr.sin_addr) <= 0) {
        perror("Invalid address");
        exit(EXIT_FAILURE);
    }

    if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
        perror("Connection Failed");
        exit(EXIT_FAILURE);
    }

    // Send data
    char *msg = "Hello Server, testing SO_LINGER";
    send(sock, msg, strlen(msg), 0);
    printf("Client sent message\n");

    // Optional: sleep to observe linger behavior in Wireshark
    sleep(2);

    close(sock);
    printf("Client socket closed\n");

    return 0;
}