/**

 * tcpconnect.c

 * Copyright (C) 2010 Markus Broeker

 */

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <netdb.h>

#include <arpa/inet.h>

#include <netinet/in.h>

#include <sys/types.h>

#include <sys/socket.h>

int tcpconnect (char *server, char *port)

{

    struct addrinfo hints;

    struct addrinfo *result, *rp;

    int sockfd = -1;

    memset (&hints, 0, sizeof (struct addrinfo));

    hints.ai_family = AF_UNSPEC;

    hints.ai_socktype = SOCK_STREAM;

    hints.ai_flags = 0;

    hints.ai_protocol = IPPROTO_TCP;

    if (getaddrinfo (server, port, &hints, &result) != 0) {

        perror ("getaddrinfo");

        return -1;

    }

    for (rp = result; rp != NULL; rp = rp->ai_next) {

        if ((sockfd = socket (rp->ai_family, rp->ai_socktype, rp->ai_protocol)) == -1)

            continue;

        if (connect (sockfd, rp->ai_addr, rp->ai_addrlen) != -1) {

            break;

        }

        close (sockfd);

        sockfd = -1;

    }

    if (result != NULL)

        freeaddrinfo (result);

    return (sockfd);

}

char *get (int sockfd, const char *path)

{

    char *result, *current;

    char buffer[1024];

    int len;

    int i = 1;

    len = sprintf (buffer, "GET %s HTTP/1.0\r\n\r\n", path);

    if (write (sockfd, buffer, len) == -1) {

        perror ("write");

        return NULL;

    }

    result = malloc (1024);

    while ((len = read (sockfd, buffer, 1023)) > 0) {

        buffer[len] = '\0';

        strcat (result, buffer);

        if ((current = realloc (result, 1024 * ++i)) == NULL) {

            break;

        }

        result = current;

    }

    return result;

}

int main (int argc, char **argv)

{

    int sockfd;

    char *result;

    if (argc != 3) {

        printf ("Usage: %s <ipaddr> <service>\n", argv[0]);

        return EXIT_FAILURE;

    }

    if ((sockfd = tcpconnect (argv[1], argv[2])) < 0) {

        printf ("Connection error: %s:%s\n", argv[1], argv[2]);

        return EXIT_FAILURE;

    }

    result = get (sockfd, "/");

    printf ("RESPONSE: %s", result);

    if (result)

        free (result);

    return close (sockfd);

}