/**

 * connection.c

 * Copyright (C) 2008 mussolini

 * Copyright (C) 2008 mbroeker

 */

#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 connection (char *ip, unsigned short port)

{

    struct hostent *hs;

    struct sockaddr_in sock;

    int sockfd;

    memset (&sock, 0, sizeof (sock));

    sock.sin_family = PF_INET;

    sock.sin_port = htons (port);

    if ((sock.sin_addr.s_addr = inet_addr (ip)) == -1) {

        if ((hs = gethostbyname (ip)) == NULL) {

            perror ("[-] Error");

            return -1;

        }

        sock.sin_family = hs->h_addrtype;

        memcpy (&sock.sin_addr.s_addr, hs->h_addr_list[0], hs->h_length);

    }

    if ((sockfd = socket (PF_INET, SOCK_STREAM, 0)) < 0) {

        perror ("[-] Error");

        return -1;

    }

    if (connect (sockfd, (struct sockaddr *)&sock, sizeof (sock)) < 0) {

        perror ("[-] Error ");

        return -1;

    }

    return (sockfd);

}

int main (int argc, char **argv)

{

    char buffer[1024];

    int sockfd;

    int num;

    if (argc != 3) {

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

        return EXIT_FAILURE;

    }

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

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

        return EXIT_FAILURE;

    }

    while ((num = read (sockfd, buffer, 1023)) != 0) {

        buffer[num] = 0;

        printf ("%s", buffer);

    }

    return close (sockfd);

}