Ticket #5824: test_socket.c

#include<stdio.h> //printf
#include<string.h>    //strlen
#include<sys/socket.h>    //socket
#include<arpa/inet.h> //inet_addr
#include<unistd.h>    //usleep
#include<fcntl.h> //fcntl
#include<errno.h>

//Size of each chunk of data received, try changing this
#define CHUNK_SIZE 32767

//Receiving function
int recv_timeout(int s, int timeout);

int main(int argc , char *argv[])
{
    int socket_desc;
    struct sockaddr_in server;
    char *message , server_reply[2000];

    //Create socket
    socket_desc = socket(AF_INET, SOCK_STREAM , 0);
    if (socket_desc == -1)
    {
        printf("Could not create socket");
    }

    //ip address of www.msn.com (get by doing a ping www.msn.com at terminal)
    server.sin_addr.s_addr = inet_addr("52.71.113.113");
    server.sin_family = AF_INET;
    server.sin_port = htons( 8888 );

    //Connect to remote server
    if (connect(socket_desc , (struct sockaddr *)&server , sizeof(server)) < 0)
    {
        puts("connect error");
        return 1;
    }

    puts("Connected\n");

    //Send some data
    message = "GET /fatehtv.mp3 HTTP/1.0\r\nHost: Lavf54.3.100\r\n\r\n";
    if( send(socket_desc , message , strlen(message) , 0) < 0)
    {
        puts("Send failed");
        return 1;
    }
    puts("Data Send\n");

    //Now receive full data
    int total_recv = recv_timeout(socket_desc, 4);

    printf("\n\nDone. Received a total of %d bytes\n\n" , total_recv);
    return 0;
}

/*
    Receive data in multiple chunks by checking a non-blocking socket
    Timeout in seconds
*/
int recv_timeout(int s , int timeout)
{
    int size_recv , total_size= 0, i=0;
    struct timeval begin , now;
    char chunk[CHUNK_SIZE];
    char *msg = "Hello";
    double timediff;

    //make socket non blocking
    //fcntl(s, F_SETFD, FD_CLOEXEC);
    fcntl(s, F_SETFD, O_NONBLOCK);

    //beginning time
    gettimeofday(&begin , NULL);

    while(1)
    {
        gettimeofday(&now , NULL);

        //time elapsed in seconds
        timediff = (now.tv_sec - begin.tv_sec) + 1e-2 * (now.tv_usec - begin.tv_usec);

        printf("Timediff: %lf", timediff);
        //if you got some data, then break after timeout
        if( total_size > 0 && timediff > timeout )
        {
            printf("timeout wait\n");
            break;
        }

        //if you got no data at all, wait a little longer, twice the timeout
        else if( timediff > timeout*2)
        {
            printf("Twice wait\n");
            break;
        }

        memset(chunk ,0 , CHUNK_SIZE);  //clear the variable
        printf("stuck at recv\n");
        if((size_recv =  recv(s , chunk , CHUNK_SIZE , 0) ) < 0)
        {
            //if nothing was received then we want to wait a little before trying again, 0.1 seconds
            usleep(100000);
            printf("ERRNO: %d", errno);
            printf("\n\nsleeping      %d", size_recv);
        }
        else
        {
            total_size += size_recv;
            //printf("%s" , chunk);
            printf("%d\n", size_recv);
            //usleep(10000000);
            //reset beginning time

            shutdown(s, 2);
           //close(s);

            printf("Reset Time\n");
            gettimeofday(&begin , NULL);
        }
        i++;
        printf("...............");
    }

    return total_size;
}