/*
 * Copyright 2008, Axel Dörfler, axeld@pinc-software.de.
 * Distributed under the terms of the MIT license.
 */


#include <errno.h>
#include <netdb.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <unistd.h>


extern const char* __progname;


int
main(int argc, char** argv)
{
        if (argc < 2) {
                printf("usage: %s <host> [port]\n"
                        "Connects to the host (default port 21, ftp), and calls\n"
                        "getpeername() to see if it works correctly with a short\n"
                        "buffer.\n", __progname);
                return 0;
        }

        struct hostent* host = gethostbyname(argv[1]);
        if (host == NULL) {
                perror("gethostbyname");
                return 1;
        }

        uint16_t port = 21;
        if (argc > 2)
                port = atol(argv[2]);

        int socket = ::socket(AF_INET, SOCK_STREAM, 0);
        if (socket < 0) {
                perror("socket");
                return 1;
        }

        sockaddr_in address;
        memset(&address, 0, sizeof(sockaddr_in));

        address.sin_family = AF_INET;
        address.sin_port = htons(port);
        address.sin_addr = *((struct in_addr*)host->h_addr);

        socklen_t length = 14;
        sockaddr buffer;
        int result = getpeername(socket, &buffer, &length);
        printf("getpeername() before connect(): %d (%s), length: %d\n", result,
                strerror(errno), length);

        if (connect(socket, (sockaddr*)&address, sizeof(sockaddr_in)) < 0) {
                perror("connect");
                return 1;
        }

        errno = 0;
        length = 14;
        result = getpeername(socket, &buffer, &length);
        printf("getpeername() after connect(): %d (%s), length: %d\n", result,
                strerror(errno), length);

        close(socket);
        return 0;
}