CloseTestClientLinger30.c

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include "../../tool/TimeTool.h"
#include <netinet/in.h>
#include <fcntl.h>

int main(int argc, char const *argv[])
{
    if (argc != 4) {
        printf("usage : %s <ip> <port> <send_bytes KB>\n", argv[0]);
        exit(1);
    }

    int fd = -1;
    if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        perror("socket error!");
        exit(1);
    }

    struct sockaddr_in svraddr;
    bzero(&svraddr, sizeof(svraddr));
    svraddr.sin_family = AF_INET;
    svraddr.sin_port = htons(atoi(argv[2]));
    svraddr.sin_addr.s_addr = inet_addr(argv[1]);

    struct sockaddr_in cliaddr;
    bzero(&cliaddr, sizeof(cliaddr));
    cliaddr.sin_family = AF_INET;
    cliaddr.sin_port = htons(30000);
    cliaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    if (bind(fd, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0) {
        perror("bind error!");
        exit(1);
    }

    struct linger l = { 1, 30 };
    socklen_t llen = sizeof(l);
    if (setsockopt(fd, SOL_SOCKET, SO_LINGER, &l, llen) < 0) {
        perror("setsockopt error!");
        exit(1);
    }

    if (connect(fd, (struct sockaddr *)&svraddr, sizeof(svraddr)) < 0) {
        perror("connect error!");
        exit(1);
    }

    char *buf = NULL;
    int size = atoi(argv[3]) << 10;

    if ((buf = malloc(size)) == NULL) {
        printf("malloc error!");
        exit(1);
    }

    memset(buf, 'a', size);
    printf("write begin! %s\n", curtimestr());
    unsigned long begin = curtimeus();
    if (write(fd, buf, size) != size) {
        perror("write error!");
        exit(1);
    }
    printf("write finished! %s total_cost=%ld\n", curtimestr(), curtimeus() - begin);

    printf("close begin %s\n", curtimestr());
    begin = curtimeus();
    int retCode = close(fd);
    printf("close finished! [ret_code=%d] %s total_cost=%ld\n", retCode, curtimestr(), curtimeus() - begin);

    return 0;
}