#include "connwrap.h"

#include <stdio.h>

#include <stdlib.h>

#include <netdb.h>

#include <string.h>

#include <netinet/in.h>

#include <errno.h>

#include <arpa/inet.h>

#include <fcntl.h>

#include <sys/time.h>

#include <unistd.h>

#define PROXY_TIMEOUT   10

    // HTTP proxy timeout in seconds (for the CONNECT method)

#ifdef HAVE_OPENSSL

#define OPENSSL_NO_KRB5 1

#include <openssl/ssl.h>

#include <openssl/err.h>

#else

# ifdef HAVE_GNUTLS

# include <gnutls/openssl.h>

# define HAVE_OPENSSL

# endif

#endif

static int in_http_connect = 0;

#ifdef HAVE_OPENSSL

static SSL_CTX *ctx = 0;

/* verify > 0 indicates verify depth as well */

static int verify = -1;

static const char *cafile = NULL;

static const char *capath = NULL;

static const char *cipherlist = NULL;

static const char *peer = NULL;

static const char *sslerror = NULL;

static int verify_cb(int preverify_ok, X509_STORE_CTX *cx)

{

    X509 *cert;

    X509_NAME *nm;

    int lastpos;

    if(!preverify_ok) {

	long err = X509_STORE_CTX_get_error(cx);

	sslerror = X509_verify_cert_error_string(err);

	return 0;

    }

    if (peer == NULL)

	return 1;

    if ((cert = X509_STORE_CTX_get_current_cert(cx)) == NULL) {

	sslerror = "internal SSL error";

	return 0;

    }

    /* We only want to look at the peername if we're working on the peer

     * certificate. */

    if (cert != cx->cert)

	return 1;

    if ((nm = X509_get_subject_name (cert)) == NULL) {

	sslerror = "internal SSL error";

	return 0;

    }

    for(lastpos = -1; ; ) {

	X509_NAME_ENTRY *e;

	ASN1_STRING *a;

	ASN1_STRING *p;

	int match;

        lastpos = X509_NAME_get_index_by_NID(nm, NID_commonName, lastpos);

	if (lastpos == -1)

	    break;

	if ((e = X509_NAME_get_entry(nm, lastpos)) == NULL) {

	    sslerror = "internal SSL error";

	    return 0;

	}

	if ((a = X509_NAME_ENTRY_get_data(e)) == NULL) {

	    sslerror = "internal SSL error";

	    return 0;

	}

	if ((p = ASN1_STRING_type_new(ASN1_STRING_type(a))) == NULL) {

	    sslerror = "internal SSL error";

	    return 0;

	}

	(void) ASN1_STRING_set(p, peer, -1);

	match = !ASN1_STRING_cmp(a, p);

	ASN1_STRING_free(p);

	if(match)

	    return 1;

    }

    sslerror = "server certificate cn mismatch";

    return 0;

}

static void init(void) {

    if(ctx)

	return;

    SSL_library_init();

    SSL_load_error_strings();

#ifdef HAVE_SSLEAY

    SSLeay_add_all_algorithms();

#else

    OpenSSL_add_all_algorithms();

#endif

    /* May need to use distinct SSLEAY bindings below... */

    //ctx = SSL_CTX_new(SSLv23_method());

    ctx = SSL_CTX_new(SSLv23_client_method());

    if(cipherlist)

	(void)SSL_CTX_set_cipher_list(ctx, cipherlist);

    if(cafile || capath)

	(void)SSL_CTX_load_verify_locations(ctx, cafile, capath);

    if(verify) {

	SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER, verify_cb);

	if(verify > 0)

	    SSL_CTX_set_verify_depth(ctx, verify);

    } else

	SSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, NULL);

}

typedef struct { int fd; SSL *ssl; } sslsock;

static sslsock *socks = 0;

static int sockcount = 0;

static sslsock *getsock(int fd) {

    int i;

    for(i = 0; i < sockcount; i++)

	if(socks[i].fd == fd)

	    return &socks[i];

    return 0;

}

static sslsock *addsock(int fd) {

    sslsock *p;

    if (socks)

	socks = (sslsock *) realloc(socks, sizeof(sslsock)*++sockcount);

    else

	socks = (sslsock *) malloc(sizeof(sslsock)*++sockcount);

    p = &socks[sockcount-1];

    init ();

    p->ssl = SSL_new(ctx);

    SSL_set_fd(p->ssl, p->fd = fd);

    sslerror = NULL;

    return p;

}

static void delsock(int fd) {

    int i, nsockcount;

    sslsock *nsocks;

    nsockcount = 0;

    if (sockcount > 1) {

	nsocks = (sslsock *) malloc(sizeof(sslsock)*(sockcount-1));

	for(i = 0; i < sockcount; i++) {

	    if(socks[i].fd != fd) {

		nsocks[nsockcount++] = socks[i];

	    } else {

		SSL_free(socks[i].ssl);

	    }

	}

    } else {

	if (ctx)

	    SSL_CTX_free(ctx);

	ctx = 0;

	nsocks = 0;

    }

    if (socks)

	free(socks);

    socks = nsocks;

    sockcount = nsockcount;

}

void cw_set_ssl_options(int sslverify, const char *sslcafile, const char *sslcapath, const char *sslciphers, const char *sslpeer) {

    verify = sslverify;

    cafile = sslcafile;

    capath = sslcapath;

    cipherlist = sslciphers;

    peer = sslpeer;

}

const char *cw_get_ssl_error(void) {

    return sslerror;

}

#else

void cw_set_ssl_options(int sslverify, const char *sslcafile, const char *sslcapath, const char *sslciphers, const char *sslpeer) { }

const char *cw_get_ssl_error(void) {

    return NULL;

}

#endif

static char *bindaddr = 0, *proxyhost = 0, *proxyuser = 0, *proxypass = 0;

static int proxyport = 3128;

static int proxy_ssl = 0;

#define SOCKOUT(s) write(sockfd, s, strlen(s))

int cw_http_connect(int sockfd, const struct sockaddr *serv_addr, int addrlen) {

    int err, pos, fl;

    struct hostent *server;

    struct sockaddr_in paddr;

    char buf[512];

    fd_set rfds;

    fl = 0;

    err = 0;

    in_http_connect = 1;

    if(!(server = gethostbyname(proxyhost))) {

	errno = h_errno;

	err = -1;

    }

    if(!err) {

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

	paddr.sin_family = AF_INET;

	memcpy(&paddr.sin_addr.s_addr, *server->h_addr_list, server->h_length);

	paddr.sin_port = htons(proxyport);

	fl = fcntl(sockfd, F_GETFL);

	fcntl(sockfd, F_SETFL, fl & ~O_NONBLOCK);

	buf[0] = 0;

	err = cw_connect(sockfd, (struct sockaddr *) &paddr, sizeof(paddr), proxy_ssl);

    }

    errno = ECONNREFUSED;

    if(!err) {

	struct sockaddr_in *sin = (struct sockaddr_in *) serv_addr;

	char *ip = inet_ntoa(sin->sin_addr), c;

	struct timeval tv;

	sprintf(buf, "%d", ntohs(sin->sin_port));

	SOCKOUT("CONNECT ");

	SOCKOUT(ip);

	SOCKOUT(":");

	SOCKOUT(buf);

	SOCKOUT(" HTTP/1.0\r\n");

	if(proxyuser) {

	    char *b;

	    SOCKOUT("Proxy-Authorization: Basic ");

	    snprintf(buf, sizeof(buf), "%s:%s", proxyuser, proxypass);

	    b = cw_base64_encode(buf);

	    SOCKOUT(b);

	    free(b);

	    SOCKOUT("\r\n");

	}

	SOCKOUT("\r\n");

	buf[0] = 0;

	while(err != -1) {

	    FD_ZERO(&rfds);

	    FD_SET(sockfd, &rfds);

	    tv.tv_sec = PROXY_TIMEOUT;

	    tv.tv_usec = 0;

	    err = select(sockfd+1, &rfds, 0, 0, &tv);

	    if(err < 1) err = -1;

	    if(err != -1 && FD_ISSET(sockfd, &rfds)) {

		err = read(sockfd, &c, 1);

		if(!err) err = -1;

		if(err != -1) {

		    pos = strlen(buf);

		    buf[pos] = c;

		    buf[pos+1] = 0;

		    if(strlen(buf) > 4)

		    if(!strcmp(buf+strlen(buf)-4, "\r\n\r\n"))

			break;

		}

	    }

	}

    }

    if(err != -1 && strlen(buf)) {

	char *p = strstr(buf, " ");

	err = -1;

	if(p)

	if(atoi(++p) == 200)

	    err = 0;

	fcntl(sockfd, F_SETFL, fl);

	if(fl & O_NONBLOCK) {

	    errno = EINPROGRESS;

	    err = -1;

	}

    }

    in_http_connect = 0;

    return err;

}

int cw_connect(int sockfd, const struct sockaddr *serv_addr, int addrlen, int ssl) {

    int rc;

    struct sockaddr_in ba;

    if(bindaddr)

    if(strlen(bindaddr)) {

#ifdef HAVE_INET_ATON

	struct in_addr addr;

	rc = inet_aton(bindaddr, &addr);

	ba.sin_addr.s_addr = addr.s_addr;

#else

	rc = inet_pton(AF_INET, bindaddr, &ba);

#endif

	if(rc) {

	    ba.sin_port = 0;

	    rc = bind(sockfd, (struct sockaddr *) &ba, sizeof(ba));

	} else {

	    rc = -1;

	}

	if(rc) return rc;

    }

    if(proxyhost && !in_http_connect) rc = cw_http_connect(sockfd, serv_addr, addrlen);

	else rc = connect(sockfd, serv_addr, addrlen);

#ifdef HAVE_OPENSSL

    if(ssl && !rc) {

	sslsock *p = addsock(sockfd);

	if(SSL_connect(p->ssl) != 1)

	    return -1;

    }

#endif

    return rc;

}

int cw_nb_connect(int sockfd, const struct sockaddr *serv_addr, int addrlen, int ssl, int *state) {

    int rc = 0;

    struct sockaddr_in ba;

    if(bindaddr)

    if(strlen(bindaddr)) {

#ifdef HAVE_INET_ATON

	struct in_addr addr;

	rc = inet_aton(bindaddr, &addr);

	ba.sin_addr.s_addr = addr.s_addr;

#else

	rc = inet_pton(AF_INET, bindaddr, &ba);

#endif

	if(rc) {

	    ba.sin_port = 0;

	    rc = bind(sockfd, (struct sockaddr *) &ba, sizeof(ba));

	} else {

	    rc = -1;

	}

	if(rc) return rc;

    }

#ifdef HAVE_OPENSSL

    if(ssl) {

	if ( !(*state & CW_CONNECT_WANT_SOMETHING))