/* netdump.c : network hexdump framework for writing tests */
/* Jon Mayo - PUBLIC DOMAIN - July 17, 2007 */
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <ctype.h>

#ifdef __MINGW32__
#include <winsock.h>
#define socklen_t int
#else
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <netdb.h>
#include <errno.h>
#endif

static int verbose_fl=1;


static int net_lookup(const char *host, const char *sport, const char *pname, int (*foreach)(void *p, struct addrinfo *ai, const char *addr, const char *serv), void *p) {
	struct protoent *pe;
	struct addrinfo *ai=0, hint, *curr;
	int proto;
	int res;
	assert(foreach != NULL);
        if(verbose_fl) {
	    fprintf(stderr, "net_lookup(\"%s\", \"%s\", \"%s\")\n", host, sport, pname);
        }
	if(pname && pname[0]) {
		pe=getprotobyname(pname);
		proto=pe ? pe->p_proto :  0;
	} else {
		proto=0;
	}
	if(host && !host[0]) host=0;
	if(sport && !sport[0]) sport=0;

	hint.ai_flags=AI_ADDRCONFIG|AI_PASSIVE;
	hint.ai_family=PF_UNSPEC;
	hint.ai_protocol=proto;
	hint.ai_socktype=0;
	hint.ai_addrlen=0;
	hint.ai_addr=0;
	hint.ai_canonname=0;
	hint.ai_next=0;
	
	if((res=getaddrinfo(host, sport, &hint, &ai))) {
		fprintf(stderr, "getaddrinfo(): %s\n", gai_strerror(res));
		if(res==EAI_SYSTEM) {
			perror("getaddrinfo()");
		}
		return 0;
	}
	curr=ai;
	if(!curr) {
		fprintf(stderr, "no entries!\n"); 
		return 0; 
	}

	for(; curr; curr=curr->ai_next) {
		char addr[NI_MAXHOST];
		char serv[NI_MAXSERV];
		/* Mac OS X 10.4 getnameinfo() is buggy without NI_NUMERICSERV */
		if((res=getnameinfo(curr->ai_addr, curr->ai_addrlen, addr, sizeof addr, serv, sizeof serv, NI_NUMERICHOST|NI_NUMERICSERV))) {
			/* Failed */
			if(res==EAI_SYSTEM) {
				perror("getnameinfo()");
			} else {
				fprintf(stderr, "getnameinfo(): %s\n", gai_strerror(res));
			}
			continue;
		}
		if(!foreach(p, curr, addr, serv)) {
			freeaddrinfo(ai);
			return 0;
		}
	}
	freeaddrinfo(ai);
	return 1;
}

static int connect_foreach(void *p, struct addrinfo *ai, const char *addr, const char *serv) {
    int fd;
    assert(p!=NULL);
    assert(ai!=NULL);
    assert(addr!=NULL);
    assert(serv!=NULL);
    if(verbose_fl) {
        char buf[64];
        if(inet_ntop(ai->ai_family, ai->ai_addr->sa_data, buf, sizeof buf)) {
            printf("Connecting %s ... ", buf);
        }
    }
    fd=socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
    if(fd<0) {
        fprintf(stderr, "Failure: socket():%s\n", strerror(errno));
        return 1; /* try another selection */
    }
    if(connect(fd, ai->ai_addr, ai->ai_addrlen)<0) {
        fprintf(stderr, "Failure: connect():%s\n", strerror(errno));
        close(fd);
        return 1; /* try another selection */
    }
    printf("Success\n");
    *(int*)p=fd;
    return 0; /* we're done processing */
}

int sockConnect(const char *host, const char *service) {
    int fd;
    fd=-1; 
    net_lookup(host, service, "tcp", connect_foreach, &fd);
    return fd;
}

/* read everything and dump it in a nice hexdump format */
void sockDump(int fd, FILE *out) {
    unsigned char buf[1], last[16];
    int len, ofs=0, i, n;

    n=0;
    while((len=read(fd, buf, sizeof buf))>0) {
        for(i=0;i<len;i++,ofs++) {
            last[n]=buf[i];
            n=(n+1)%sizeof last; /* go to next entry in loop buffer */
            if(ofs%sizeof last==0) {
                fprintf(out, "%06x: ", ofs);
            }
            fputc(isprint(buf[i])?buf[i]:'.', out);
            if(ofs%sizeof last==sizeof last-1) {
                unsigned j;
                fprintf(out, " |");
                for(j=0;j<sizeof last;j++) {
                    unsigned x=(n+j)%sizeof last;
                    fprintf(out, " %02x", last[x]);
                }
                fprintf(out, "\n");
            }
        }
        fflush(out);
    }
    if(len<0) {
        fprintf(stderr, "\nERROR: read(): %s\n", strerror(errno));
    }
    fprintf(out, "EOF\n");
}

int sockPuts(int fd, const char *str) {
    size_t len;
    int res;
    len=strlen(str);
    do {
        res=write(fd, str, len);
        if(res<=0) {
            fprintf(stderr, "ERROR: write(): %s\n", strerror(errno));
            return 0;
        }
        str+=res;
        len-=res;
    } while(len);
    return 1;
}

/** Test code goes here **/
int main() {
    int sock;
    sock=sockConnect("67.18.176.63", "5222");
    if(sock<0) {
        return EXIT_FAILURE;
    }
    if(verbose_fl) {
        printf("socket fd=%d\n", sock);
    }
    
    /* send some data */
    sockPuts(sock, "<stream:stream to='im.rm-f.net' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' version='1.0'>");

    /* output the response */
    sockDump(sock, stdout);

    close(sock);
    return EXIT_SUCCESS;
}