aquatest.c

Go to the documentation of this file.

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <time.h>
 
int main(int argc, char *argv[])
{
    int listenfd = 0, connfd = 0;
    struct sockaddr_in serv_addr;
 
    char* recvBuff = malloc(512*1024);
    if (!recvBuff) return 1;
 
    int64_t header[2];
    int64_t totalsz, totalbuf, totallost, lastcnt;
 
    time_t tm, lasttm;
 
    ssize_t sz;
 
    listenfd = socket(AF_INET, SOCK_STREAM, 0);
    memset(&serv_addr, '0', sizeof(serv_addr));
 
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(5000);
 
    bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
 
    listen(listenfd, 10);
 
    lastcnt = -1;
    totalsz = 0; totalbuf = 0; totallost = 0;
    lasttm = time(0);
 
 
    while(1)
    {
        connfd = accept(listenfd, (struct sockaddr*)NULL, NULL);
 
        if (connfd <= 0) {
           printf("Fail to accept connection\n");
           close(listenfd);
           free(recvBuff);
           return 0;
        }
 
        printf("Client connected\n");
 
        while (1) {
           sz = recv(connfd, header, sizeof(header), MSG_WAITALL);
           if (sz == 0) continue;
           if (sz != sizeof(header)) {
              printf("Error when receive header %lu res %ld\n", (long unsigned) sizeof(header), (long) sz);
              close(listenfd);
              free(recvBuff);
              return 0;
           }
 
           sz = recv(connfd, recvBuff, header[1], MSG_WAITALL);
           if (sz != header[1]) {
              printf("Error when receive buffer %ld res %ld\n", (long) header[1], (long) sz);
              close(listenfd);
              free(recvBuff);
              return 0;
           }
 
           // printf("get buffer %ld\n", header[1]);
 
           totalsz+=sz;
           totalbuf++;
           if ((lastcnt>0) && (header[0]-lastcnt>1)) totallost+=(header[0]-lastcnt-1);
 
           lastcnt = header[0];
 
           tm = time(0);
           if (tm - lasttm > 3) {
              printf("Bufs %6lld  Size %8.3f MB  Lost %3lld\n", (long long) totalbuf, totalsz*1e-6, (long long) totallost);
              lasttm = tm;
           }
        }
 
     }
 
    free(recvBuff);
    return 0;
}