[thirdparty/openssl.git] / crypto / bio / bss_conn.c

/*
 * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the OpenSSL license (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
 */

#include <stdio.h>
#include <errno.h>

#include "bio_lcl.h"

#ifndef OPENSSL_NO_SOCK

typedef struct bio_connect_st {
    int state;
    int connect_family;
    char *param_hostname;
    char *param_service;
    int connect_mode;

    BIO_ADDRINFO *addr_first;
    const BIO_ADDRINFO *addr_iter;
    /*
     * int socket; this will be kept in bio->num so that it is compatible
     * with the bss_sock bio
     */
    /*
     * called when the connection is initially made callback(BIO,state,ret);
     * The callback should return 'ret'.  state is for compatibility with the
     * ssl info_callback
     */
    int (*info_callback) (const BIO *bio, int state, int ret);
} BIO_CONNECT;

static int conn_write(BIO *h, const char *buf, int num);
static int conn_read(BIO *h, char *buf, int size);
static int conn_puts(BIO *h, const char *str);
static long conn_ctrl(BIO *h, int cmd, long arg1, void *arg2);
static int conn_new(BIO *h);
static int conn_free(BIO *data);
static long conn_callback_ctrl(BIO *h, int cmd, bio_info_cb *);

static int conn_state(BIO *b, BIO_CONNECT *c);
static void conn_close_socket(BIO *data);
BIO_CONNECT *BIO_CONNECT_new(void);
void BIO_CONNECT_free(BIO_CONNECT *a);

#define BIO_CONN_S_BEFORE                1
#define BIO_CONN_S_GET_ADDR              2
#define BIO_CONN_S_CREATE_SOCKET         3
#define BIO_CONN_S_CONNECT               4
#define BIO_CONN_S_OK                    5
#define BIO_CONN_S_BLOCKED_CONNECT       6

static const BIO_METHOD methods_connectp = {
    BIO_TYPE_CONNECT,
    "socket connect",
    /* TODO: Convert to new style write function */
    bwrite_conv,
    conn_write,
    /* TODO: Convert to new style read function */
    bread_conv,
    conn_read,
    conn_puts,
    NULL,                       /* connect_gets, */
    conn_ctrl,
    conn_new,
    conn_free,
    conn_callback_ctrl,
};

static int conn_state(BIO *b, BIO_CONNECT *c)
{
    int ret = -1, i;
    int (*cb) (const BIO *, int, int) = NULL;

    if (c->info_callback != NULL)
        cb = c->info_callback;

    for (;;) {
        switch (c->state) {
        case BIO_CONN_S_BEFORE:
            if (c->param_hostname == NULL && c->param_service == NULL) {
                BIOerr(BIO_F_CONN_STATE, BIO_R_NO_HOSTNAME_OR_SERVICE_SPECIFIED);
                ERR_add_error_data(4,
                                   "hostname=", c->param_hostname,
                                   " service=", c->param_service);
                goto exit_loop;
            }
            c->state = BIO_CONN_S_GET_ADDR;
            break;

        case BIO_CONN_S_GET_ADDR:
            {
                int family = AF_UNSPEC;
                switch (c->connect_family) {
                case BIO_FAMILY_IPV6:
                    if (1) { /* This is a trick we use to avoid bit rot.
                              * at least the "else" part will always be
                              * compiled.
                              */
#ifdef AF_INET6
                        family = AF_INET6;
                    } else {
#endif
                        BIOerr(BIO_F_CONN_STATE, BIO_R_UNAVAILABLE_IP_FAMILY);
                        goto exit_loop;
                    }
                    break;
                case BIO_FAMILY_IPV4:
                    family = AF_INET;
                    break;
                case BIO_FAMILY_IPANY:
                    family = AF_UNSPEC;
                    break;
                default:
                    BIOerr(BIO_F_CONN_STATE, BIO_R_UNSUPPORTED_IP_FAMILY);
                    goto exit_loop;
                }
                if (BIO_lookup(c->param_hostname, c->param_service,
                               BIO_LOOKUP_CLIENT,
                               family, SOCK_STREAM, &c->addr_first) == 0)
                    goto exit_loop;
            }
            if (c->addr_first == NULL) {
                BIOerr(BIO_F_CONN_STATE, BIO_R_LOOKUP_RETURNED_NOTHING);
                goto exit_loop;
            }
            c->addr_iter = c->addr_first;
            c->state = BIO_CONN_S_CREATE_SOCKET;
            break;

        case BIO_CONN_S_CREATE_SOCKET:
            ret = BIO_socket(BIO_ADDRINFO_family(c->addr_iter),
                             BIO_ADDRINFO_socktype(c->addr_iter),
                             BIO_ADDRINFO_protocol(c->addr_iter), 0);
            if (ret == (int)INVALID_SOCKET) {
                SYSerr(SYS_F_SOCKET, get_last_socket_error());
                ERR_add_error_data(4,
                                   "hostname=", c->param_hostname,
                                   " service=", c->param_service);
                BIOerr(BIO_F_CONN_STATE, BIO_R_UNABLE_TO_CREATE_SOCKET);
                goto exit_loop;
            }
            b->num = ret;
            c->state = BIO_CONN_S_CONNECT;
            break;

        case BIO_CONN_S_CONNECT:
            BIO_clear_retry_flags(b);
            ret = BIO_connect(b->num, BIO_ADDRINFO_address(c->addr_iter),
                              BIO_SOCK_KEEPALIVE | c->connect_mode);
            b->retry_reason = 0;
            if (ret == 0) {
                if (BIO_sock_should_retry(ret)) {
                    BIO_set_retry_special(b);
                    c->state = BIO_CONN_S_BLOCKED_CONNECT;
                    b->retry_reason = BIO_RR_CONNECT;
                    ERR_clear_error();
                } else if ((c->addr_iter = BIO_ADDRINFO_next(c->addr_iter))
                           != NULL) {
                    /*
                     * if there are more addresses to try, do that first
                     */
                    BIO_closesocket(b->num);
                    c->state = BIO_CONN_S_CREATE_SOCKET;
                    ERR_clear_error();
                    break;
                } else {
                    SYSerr(SYS_F_CONNECT, get_last_socket_error());
                    ERR_add_error_data(4,
                                       "hostname=", c->param_hostname,
                                       " service=", c->param_service);
                    BIOerr(BIO_F_CONN_STATE, BIO_R_CONNECT_ERROR);
                }
                goto exit_loop;
            } else {
                c->state = BIO_CONN_S_OK;
            }
            break;

        case BIO_CONN_S_BLOCKED_CONNECT:
            i = BIO_sock_error(b->num);
            if (i) {
                BIO_clear_retry_flags(b);
                SYSerr(SYS_F_CONNECT, i);
                ERR_add_error_data(4,
                                   "hostname=", c->param_hostname,
                                   " service=", c->param_service);
                BIOerr(BIO_F_CONN_STATE, BIO_R_NBIO_CONNECT_ERROR);
                ret = 0;
                goto exit_loop;
            } else
                c->state = BIO_CONN_S_OK;
            break;

        case BIO_CONN_S_OK:
            ret = 1;
            goto exit_loop;
        default:
            /* abort(); */
            goto exit_loop;
        }

        if (cb != NULL) {
            if ((ret = cb((BIO *)b, c->state, ret)) == 0)
                goto end;
        }
    }

    /* Loop does not exit */
 exit_loop:
    if (cb != NULL)
        ret = cb((BIO *)b, c->state, ret);
 end:
    return (ret);
}

BIO_CONNECT *BIO_CONNECT_new(void)
{
    BIO_CONNECT *ret;

    if ((ret = OPENSSL_zalloc(sizeof(*ret))) == NULL)
        return (NULL);
    ret->state = BIO_CONN_S_BEFORE;
    ret->connect_family = BIO_FAMILY_IPANY;
    return (ret);
}

void BIO_CONNECT_free(BIO_CONNECT *a)
{
    if (a == NULL)
        return;

    OPENSSL_free(a->param_hostname);
    OPENSSL_free(a->param_service);
    BIO_ADDRINFO_free(a->addr_first);
    OPENSSL_free(a);
}

const BIO_METHOD *BIO_s_connect(void)
{
    return (&methods_connectp);
}

static int conn_new(BIO *bi)
{
    bi->init = 0;
    bi->num = (int)INVALID_SOCKET;
    bi->flags = 0;
    if ((bi->ptr = (char *)BIO_CONNECT_new()) == NULL)
        return (0);
    else
        return 1;
}

static void conn_close_socket(BIO *bio)
{
    BIO_CONNECT *c;

    c = (BIO_CONNECT *)bio->ptr;
    if (bio->num != (int)INVALID_SOCKET) {
        /* Only do a shutdown if things were established */
        if (c->state == BIO_CONN_S_OK)
            shutdown(bio->num, 2);
        BIO_closesocket(bio->num);
        bio->num = (int)INVALID_SOCKET;
    }
}

static int conn_free(BIO *a)
{
    BIO_CONNECT *data;

    if (a == NULL)
        return (0);
    data = (BIO_CONNECT *)a->ptr;

    if (a->shutdown) {
        conn_close_socket(a);
        BIO_CONNECT_free(data);
        a->ptr = NULL;
        a->flags = 0;
        a->init = 0;
    }
    return 1;
}

static int conn_read(BIO *b, char *out, int outl)
{
    int ret = 0;
    BIO_CONNECT *data;

    data = (BIO_CONNECT *)b->ptr;
    if (data->state != BIO_CONN_S_OK) {
        ret = conn_state(b, data);
        if (ret <= 0)
            return (ret);
    }

    if (out != NULL) {
        clear_socket_error();
        ret = readsocket(b->num, out, outl);
        BIO_clear_retry_flags(b);
        if (ret <= 0) {
            if (BIO_sock_should_retry(ret))
                BIO_set_retry_read(b);
        }
    }
    return (ret);
}

static int conn_write(BIO *b, const char *in, int inl)
{
    int ret;
    BIO_CONNECT *data;

    data = (BIO_CONNECT *)b->ptr;
    if (data->state != BIO_CONN_S_OK) {
        ret = conn_state(b, data);
        if (ret <= 0)
            return (ret);
    }

    clear_socket_error();
    ret = writesocket(b->num, in, inl);
    BIO_clear_retry_flags(b);
    if (ret <= 0) {
        if (BIO_sock_should_retry(ret))
            BIO_set_retry_write(b);
    }
    return (ret);
}

static long conn_ctrl(BIO *b, int cmd, long num, void *ptr)
{
    BIO *dbio;
    int *ip;
    const char **pptr = NULL;
    long ret = 1;
    BIO_CONNECT *data;

    data = (BIO_CONNECT *)b->ptr;

    switch (cmd) {
    case BIO_CTRL_RESET:
        ret = 0;
        data->state = BIO_CONN_S_BEFORE;
        conn_close_socket(b);
        BIO_ADDRINFO_free(data->addr_first);
        data->addr_first = NULL;
        b->flags = 0;
        break;
    case BIO_C_DO_STATE_MACHINE:
        /* use this one to start the connection */
        if (data->state != BIO_CONN_S_OK)
            ret = (long)conn_state(b, data);
        else
            ret = 1;
        break;
    case BIO_C_GET_CONNECT:
        if (ptr != NULL) {
            pptr = (const char **)ptr;
            if (num == 0) {
                *pptr = data->param_hostname;
            } else if (num == 1) {
                *pptr = data->param_service;
            } else if (num == 2) {
                *pptr = (const char *)BIO_ADDRINFO_address(data->addr_iter);
            } else if (num == 3) {
                switch (BIO_ADDRINFO_family(data->addr_iter)) {
# ifdef AF_INET6
                case AF_INET6:
                    ret = BIO_FAMILY_IPV6;
                    break;
# endif
                case AF_INET:
                    ret = BIO_FAMILY_IPV4;
                    break;
                case 0:
                    ret = data->connect_family;
                    break;
                default:
                    ret = -1;
                    break;
                }
            } else {
                ret = 0;
            }
        } else {
            ret = 0;
        }
        break;
    case BIO_C_SET_CONNECT:
        if (ptr != NULL) {
            b->init = 1;
            if (num == 0) {
                char *hold_service = data->param_service;
                /* We affect the hostname regardless.  However, the input
                 * string might contain a host:service spec, so we must
                 * parse it, which might or might not affect the service
                 */
                OPENSSL_free(data->param_hostname);
                data->param_hostname = NULL;
                ret = BIO_parse_hostserv(ptr,
                                         &data->param_hostname,
                                         &data->param_service,
                                         BIO_PARSE_PRIO_HOST);
                if (hold_service != data->param_service)
                    OPENSSL_free(hold_service);
            } else if (num == 1) {
                OPENSSL_free(data->param_service);
                data->param_service = BUF_strdup(ptr);
            } else if (num == 2) {
                const BIO_ADDR *addr = (const BIO_ADDR *)ptr;
                if (ret) {
                    data->param_hostname = BIO_ADDR_hostname_string(addr, 1);
                    data->param_service = BIO_ADDR_service_string(addr, 1);
                    BIO_ADDRINFO_free(data->addr_first);
                    data->addr_first = NULL;
                    data->addr_iter = NULL;
                }
            } else if (num == 3) {
                data->connect_family = *(int *)ptr;
            } else {
                ret = 0;
            }
        }
        break;
    case BIO_C_SET_NBIO:
        if (num != 0)
            data->connect_mode |= BIO_SOCK_NONBLOCK;
        else
            data->connect_mode &= ~BIO_SOCK_NONBLOCK;
        break;
    case BIO_C_SET_CONNECT_MODE:
        data->connect_mode = (int)num;
        break;
    case BIO_C_GET_FD:
        if (b->init) {
            ip = (int *)ptr;
            if (ip != NULL)
                *ip = b->num;
            ret = b->num;
        } else
            ret = -1;
        break;
    case BIO_CTRL_GET_CLOSE:
        ret = b->shutdown;
        break;
    case BIO_CTRL_SET_CLOSE:
        b->shutdown = (int)num;
        break;
    case BIO_CTRL_PENDING:
    case BIO_CTRL_WPENDING:
        ret = 0;
        break;
    case BIO_CTRL_FLUSH:
        break;
    case BIO_CTRL_DUP:
        {
            dbio = (BIO *)ptr;
            if (data->param_hostname)
                BIO_set_conn_hostname(dbio, data->param_hostname);
            if (data->param_service)
                BIO_set_conn_port(dbio, data->param_service);
            BIO_set_conn_ip_family(dbio, data->connect_family);
            BIO_set_conn_mode(dbio, data->connect_mode);
            /*
             * FIXME: the cast of the function seems unlikely to be a good
             * idea
             */
            (void)BIO_set_info_callback(dbio,
                                        (bio_info_cb *)data->info_callback);
        }
        break;
    case BIO_CTRL_SET_CALLBACK:
        ret = 0; /* use callback ctrl */
        break;
    case BIO_CTRL_GET_CALLBACK:
        {
            int (**fptr) (const BIO *bio, int state, int xret);

            fptr = (int (**)(const BIO *bio, int state, int xret))ptr;
            *fptr = data->info_callback;
        }
        break;
    default:
        ret = 0;
        break;
    }
    return (ret);
}

static long conn_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp)
{
    long ret = 1;
    BIO_CONNECT *data;

    data = (BIO_CONNECT *)b->ptr;

    switch (cmd) {
    case BIO_CTRL_SET_CALLBACK:
        {
            data->info_callback =
                (int (*)(const struct bio_st *, int, int))fp;
        }
        break;
    default:
        ret = 0;
        break;
    }
    return (ret);
}

static int conn_puts(BIO *bp, const char *str)
{
    int n, ret;

    n = strlen(str);
    ret = conn_write(bp, str, n);
    return (ret);
}

BIO *BIO_new_connect(const char *str)
{
    BIO *ret;

    ret = BIO_new(BIO_s_connect());
    if (ret == NULL)
        return (NULL);
    if (BIO_set_conn_hostname(ret, str))
        return (ret);
    BIO_free(ret);
    return (NULL);
}

#endif
Commit	Line	Data
b1322259 RS	1	/*
b1322259 RS	2	* Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
d02b48c6	3	*
b1322259 RS	4	* Licensed under the OpenSSL license (the "License"). You may not use
	5	* this file except in compliance with the License. You can obtain a copy
	6	* in the file LICENSE in the source distribution or at
	7	* https://www.openssl.org/source/license.html
d02b48c6 RE	8	*/
d02b48c6 RE	9
d02b48c6 RE	10	#include <stdio.h>
d02b48c6 RE	11	#include <errno.h>
d02b48c6	12
417be660	13	#include "bio_lcl.h"
b00b2124	14
417be660	15	#ifndef OPENSSL_NO_SOCK
0f113f3e MC	16
	17	typedef struct bio_connect_st {
	18	int state;
417be660	19	int connect_family;
0f113f3e	20	char *param_hostname;
417be660 RL	21	char *param_service;
	22	int connect_mode;
	23
	24	BIO_ADDRINFO *addr_first;
	25	const BIO_ADDRINFO *addr_iter;
0f113f3e MC	26	/*
	27	* int socket; this will be kept in bio->num so that it is compatible
	28	* with the bss_sock bio
	29	*/
	30	/*
	31	* called when the connection is initially made callback(BIO,state,ret);
	32	* The callback should return 'ret'. state is for compatibility with the
	33	* ssl info_callback
	34	*/
	35	int (info_callback) (const BIO bio, int state, int ret);
	36	} BIO_CONNECT;
d02b48c6	37
0e1c0612 UM	38	static int conn_write(BIO h, const char buf, int num);
	39	static int conn_read(BIO h, char buf, int size);
	40	static int conn_puts(BIO h, const char str);
	41	static long conn_ctrl(BIO h, int cmd, long arg1, void arg2);
d02b48c6 RE	42	static int conn_new(BIO *h);
d02b48c6 RE	43	static int conn_free(BIO *data);
13083215	44	static long conn_callback_ctrl(BIO h, int cmd, bio_info_cb );
d02b48c6 RE	45
	46	static int conn_state(BIO b, BIO_CONNECT c);
	47	static void conn_close_socket(BIO *data);
0f113f3e	48	BIO_CONNECT *BIO_CONNECT_new(void);
d02b48c6 RE	49	void BIO_CONNECT_free(BIO_CONNECT *a);
d02b48c6 RE	50
417be660 RL	51	#define BIO_CONN_S_BEFORE 1
	52	#define BIO_CONN_S_GET_ADDR 2
	53	#define BIO_CONN_S_CREATE_SOCKET 3
	54	#define BIO_CONN_S_CONNECT 4
	55	#define BIO_CONN_S_OK 5
	56	#define BIO_CONN_S_BLOCKED_CONNECT 6
	57
04f6b0fd	58	static const BIO_METHOD methods_connectp = {
0f113f3e MC	59	BIO_TYPE_CONNECT,
0f113f3e MC	60	"socket connect",
3befffa3 MC	61	/* TODO: Convert to new style write function */
3befffa3 MC	62	bwrite_conv,
0f113f3e	63	conn_write,
d07aee2c MC	64	/* TODO: Convert to new style read function */
d07aee2c MC	65	bread_conv,
0f113f3e MC	66	conn_read,
	67	conn_puts,
	68	NULL, /* connect_gets, */
	69	conn_ctrl,
	70	conn_new,
	71	conn_free,
	72	conn_callback_ctrl,
	73	};
d02b48c6	74
6b691a5c	75	static int conn_state(BIO b, BIO_CONNECT c)
0f113f3e MC	76	{
0f113f3e MC	77	int ret = -1, i;
0f113f3e MC	78	int (cb) (const BIO , int, int) = NULL;
	79
	80	if (c->info_callback != NULL)
	81	cb = c->info_callback;
	82
	83	for (;;) {
	84	switch (c->state) {
	85	case BIO_CONN_S_BEFORE:
417be660 RL	86	if (c->param_hostname == NULL && c->param_service == NULL) {
	87	BIOerr(BIO_F_CONN_STATE, BIO_R_NO_HOSTNAME_OR_SERVICE_SPECIFIED);
	88	ERR_add_error_data(4,
	89	"hostname=", c->param_hostname,
	90	" service=", c->param_service);
0f113f3e MC	91	goto exit_loop;
0f113f3e MC	92	}
417be660 RL	93	c->state = BIO_CONN_S_GET_ADDR;
417be660 RL	94	break;
0f113f3e	95
417be660 RL	96	case BIO_CONN_S_GET_ADDR:
	97	{
	98	int family = AF_UNSPEC;
	99	switch (c->connect_family) {
	100	case BIO_FAMILY_IPV6:
	101	if (1) { /* This is a trick we use to avoid bit rot.
	102	* at least the "else" part will always be
	103	* compiled.
	104	*/
	105	#ifdef AF_INET6
	106	family = AF_INET6;
	107	} else {
	108	#endif
	109	BIOerr(BIO_F_CONN_STATE, BIO_R_UNAVAILABLE_IP_FAMILY);
	110	goto exit_loop;
	111	}
	112	break;
	113	case BIO_FAMILY_IPV4:
	114	family = AF_INET;
	115	break;
	116	case BIO_FAMILY_IPANY:
	117	family = AF_UNSPEC;
	118	break;
	119	default:
	120	BIOerr(BIO_F_CONN_STATE, BIO_R_UNSUPPORTED_IP_FAMILY);
	121	goto exit_loop;
0f113f3e	122	}
417be660 RL	123	if (BIO_lookup(c->param_hostname, c->param_service,
	124	BIO_LOOKUP_CLIENT,
	125	family, SOCK_STREAM, &c->addr_first) == 0)
	126	goto exit_loop;
0f113f3e	127	}
417be660 RL	128	if (c->addr_first == NULL) {
417be660 RL	129	BIOerr(BIO_F_CONN_STATE, BIO_R_LOOKUP_RETURNED_NOTHING);
0f113f3e MC	130	goto exit_loop;
0f113f3e MC	131	}
417be660	132	c->addr_iter = c->addr_first;
0f113f3e MC	133	c->state = BIO_CONN_S_CREATE_SOCKET;
	134	break;
	135
	136	case BIO_CONN_S_CREATE_SOCKET:
417be660 RL	137	ret = BIO_socket(BIO_ADDRINFO_family(c->addr_iter),
	138	BIO_ADDRINFO_socktype(c->addr_iter),
	139	BIO_ADDRINFO_protocol(c->addr_iter), 0);
b13fdc48	140	if (ret == (int)INVALID_SOCKET) {
0f113f3e	141	SYSerr(SYS_F_SOCKET, get_last_socket_error());
417be660 RL	142	ERR_add_error_data(4,
	143	"hostname=", c->param_hostname,
	144	" service=", c->param_service);
0f113f3e MC	145	BIOerr(BIO_F_CONN_STATE, BIO_R_UNABLE_TO_CREATE_SOCKET);
	146	goto exit_loop;
	147	}
	148	b->num = ret;
0f113f3e	149	c->state = BIO_CONN_S_CONNECT;
0f113f3e MC	150	break;
	151
	152	case BIO_CONN_S_CONNECT:
	153	BIO_clear_retry_flags(b);
417be660 RL	154	ret = BIO_connect(b->num, BIO_ADDRINFO_address(c->addr_iter),
417be660 RL	155	BIO_SOCK_KEEPALIVE \| c->connect_mode);
0f113f3e	156	b->retry_reason = 0;
a043d0b9	157	if (ret == 0) {
0f113f3e MC	158	if (BIO_sock_should_retry(ret)) {
	159	BIO_set_retry_special(b);
	160	c->state = BIO_CONN_S_BLOCKED_CONNECT;
	161	b->retry_reason = BIO_RR_CONNECT;
417be660 RL	162	ERR_clear_error();
	163	} else if ((c->addr_iter = BIO_ADDRINFO_next(c->addr_iter))
	164	!= NULL) {
	165	/*
	166	* if there are more addresses to try, do that first
	167	*/
	168	BIO_closesocket(b->num);
	169	c->state = BIO_CONN_S_CREATE_SOCKET;
	170	ERR_clear_error();
	171	break;
0f113f3e MC	172	} else {
0f113f3e MC	173	SYSerr(SYS_F_CONNECT, get_last_socket_error());
417be660 RL	174	ERR_add_error_data(4,
	175	"hostname=", c->param_hostname,
	176	" service=", c->param_service);
0f113f3e MC	177	BIOerr(BIO_F_CONN_STATE, BIO_R_CONNECT_ERROR);
	178	}
	179	goto exit_loop;
417be660	180	} else {
0f113f3e	181	c->state = BIO_CONN_S_OK;
417be660	182	}
0f113f3e MC	183	break;
	184
	185	case BIO_CONN_S_BLOCKED_CONNECT:
	186	i = BIO_sock_error(b->num);
	187	if (i) {
	188	BIO_clear_retry_flags(b);
	189	SYSerr(SYS_F_CONNECT, i);
417be660 RL	190	ERR_add_error_data(4,
	191	"hostname=", c->param_hostname,
	192	" service=", c->param_service);
0f113f3e MC	193	BIOerr(BIO_F_CONN_STATE, BIO_R_NBIO_CONNECT_ERROR);
	194	ret = 0;
	195	goto exit_loop;
	196	} else
	197	c->state = BIO_CONN_S_OK;
	198	break;
	199
	200	case BIO_CONN_S_OK:
	201	ret = 1;
	202	goto exit_loop;
	203	default:
	204	/* abort(); */
	205	goto exit_loop;
	206	}
	207
	208	if (cb != NULL) {
75ebbd9a	209	if ((ret = cb((BIO *)b, c->state, ret)) == 0)
0f113f3e MC	210	goto end;
	211	}
	212	}
	213
	214	/* Loop does not exit */
	215	exit_loop:
	216	if (cb != NULL)
	217	ret = cb((BIO *)b, c->state, ret);
	218	end:
	219	return (ret);
	220	}
d02b48c6	221
6b691a5c	222	BIO_CONNECT *BIO_CONNECT_new(void)
0f113f3e MC	223	{
	224	BIO_CONNECT *ret;
	225
b51bce94	226	if ((ret = OPENSSL_zalloc(sizeof(*ret))) == NULL)
0f113f3e MC	227	return (NULL);
0f113f3e MC	228	ret->state = BIO_CONN_S_BEFORE;
417be660	229	ret->connect_family = BIO_FAMILY_IPANY;
0f113f3e MC	230	return (ret);
0f113f3e MC	231	}
d02b48c6	232
6b691a5c	233	void BIO_CONNECT_free(BIO_CONNECT *a)
0f113f3e MC	234	{
	235	if (a == NULL)
	236	return;
e03ddfae	237
b548a1f1	238	OPENSSL_free(a->param_hostname);
417be660 RL	239	OPENSSL_free(a->param_service);
417be660 RL	240	BIO_ADDRINFO_free(a->addr_first);
0f113f3e MC	241	OPENSSL_free(a);
0f113f3e MC	242	}
d02b48c6	243
04f6b0fd	244	const BIO_METHOD *BIO_s_connect(void)
0f113f3e MC	245	{
	246	return (&methods_connectp);
	247	}
d02b48c6	248
6b691a5c	249	static int conn_new(BIO *bi)
0f113f3e MC	250	{
0f113f3e MC	251	bi->init = 0;
b13fdc48	252	bi->num = (int)INVALID_SOCKET;
0f113f3e MC	253	bi->flags = 0;
	254	if ((bi->ptr = (char *)BIO_CONNECT_new()) == NULL)
	255	return (0);
	256	else
208fb891	257	return 1;
0f113f3e	258	}
d02b48c6	259
6b691a5c	260	static void conn_close_socket(BIO *bio)
0f113f3e MC	261	{
	262	BIO_CONNECT *c;
	263
	264	c = (BIO_CONNECT *)bio->ptr;
b13fdc48	265	if (bio->num != (int)INVALID_SOCKET) {
0f113f3e MC	266	/* Only do a shutdown if things were established */
	267	if (c->state == BIO_CONN_S_OK)
	268	shutdown(bio->num, 2);
417be660	269	BIO_closesocket(bio->num);
b13fdc48	270	bio->num = (int)INVALID_SOCKET;
0f113f3e MC	271	}
0f113f3e MC	272	}
d02b48c6	273
6b691a5c	274	static int conn_free(BIO *a)
0f113f3e MC	275	{
	276	BIO_CONNECT *data;
	277
	278	if (a == NULL)
	279	return (0);
	280	data = (BIO_CONNECT *)a->ptr;
	281
	282	if (a->shutdown) {
	283	conn_close_socket(a);
	284	BIO_CONNECT_free(data);
	285	a->ptr = NULL;
	286	a->flags = 0;
	287	a->init = 0;
	288	}
208fb891	289	return 1;
0f113f3e MC	290	}
0f113f3e MC	291
6b691a5c	292	static int conn_read(BIO b, char out, int outl)
0f113f3e MC	293	{
	294	int ret = 0;
	295	BIO_CONNECT *data;
	296
	297	data = (BIO_CONNECT *)b->ptr;
	298	if (data->state != BIO_CONN_S_OK) {
	299	ret = conn_state(b, data);
	300	if (ret <= 0)
	301	return (ret);
	302	}
	303
	304	if (out != NULL) {
	305	clear_socket_error();
	306	ret = readsocket(b->num, out, outl);
	307	BIO_clear_retry_flags(b);
	308	if (ret <= 0) {
	309	if (BIO_sock_should_retry(ret))
	310	BIO_set_retry_read(b);
	311	}
	312	}
	313	return (ret);
	314	}
d02b48c6	315
0e1c0612	316	static int conn_write(BIO b, const char in, int inl)
0f113f3e MC	317	{
	318	int ret;
	319	BIO_CONNECT *data;
	320
	321	data = (BIO_CONNECT *)b->ptr;
	322	if (data->state != BIO_CONN_S_OK) {
	323	ret = conn_state(b, data);
	324	if (ret <= 0)
	325	return (ret);
	326	}
	327
	328	clear_socket_error();
	329	ret = writesocket(b->num, in, inl);
	330	BIO_clear_retry_flags(b);
	331	if (ret <= 0) {
	332	if (BIO_sock_should_retry(ret))
	333	BIO_set_retry_write(b);
	334	}
	335	return (ret);
	336	}
d02b48c6	337
0e1c0612	338	static long conn_ctrl(BIO b, int cmd, long num, void ptr)
0f113f3e MC	339	{
	340	BIO *dbio;
	341	int *ip;
4b72d5f8	342	const char **pptr = NULL;
0f113f3e MC	343	long ret = 1;
	344	BIO_CONNECT *data;
	345
	346	data = (BIO_CONNECT *)b->ptr;
	347
	348	switch (cmd) {
	349	case BIO_CTRL_RESET:
	350	ret = 0;
	351	data->state = BIO_CONN_S_BEFORE;
	352	conn_close_socket(b);
417be660 RL	353	BIO_ADDRINFO_free(data->addr_first);
417be660 RL	354	data->addr_first = NULL;
0f113f3e MC	355	b->flags = 0;
	356	break;
	357	case BIO_C_DO_STATE_MACHINE:
	358	/* use this one to start the connection */
	359	if (data->state != BIO_CONN_S_OK)
	360	ret = (long)conn_state(b, data);
	361	else
	362	ret = 1;
	363	break;
	364	case BIO_C_GET_CONNECT:
	365	if (ptr != NULL) {
	366	pptr = (const char **)ptr;
417be660 RL	367	if (num == 0) {
	368	*pptr = data->param_hostname;
	369	} else if (num == 1) {
	370	*pptr = data->param_service;
	371	} else if (num == 2) {
	372	pptr = (const char )BIO_ADDRINFO_address(data->addr_iter);
	373	} else if (num == 3) {
	374	switch (BIO_ADDRINFO_family(data->addr_iter)) {
	375	# ifdef AF_INET6
	376	case AF_INET6:
	377	ret = BIO_FAMILY_IPV6;
	378	break;
	379	# endif
	380	case AF_INET:
	381	ret = BIO_FAMILY_IPV4;
	382	break;
	383	case 0:
	384	ret = data->connect_family;
	385	break;
	386	default:
	387	ret = -1;
	388	break;
4b72d5f8	389	}
417be660 RL	390	} else {
417be660 RL	391	ret = 0;
4b72d5f8 RL	392	}
4b72d5f8 RL	393	} else {
4b72d5f8	394	ret = 0;
0f113f3e MC	395	}
	396	break;
	397	case BIO_C_SET_CONNECT:
	398	if (ptr != NULL) {
	399	b->init = 1;
	400	if (num == 0) {
417be660 RL	401	char *hold_service = data->param_service;
	402	/* We affect the hostname regardless. However, the input
	403	* string might contain a host:service spec, so we must
	404	* parse it, which might or might not affect the service
	405	*/
b548a1f1	406	OPENSSL_free(data->param_hostname);
417be660 RL	407	data->param_hostname = NULL;
	408	ret = BIO_parse_hostserv(ptr,
	409	&data->param_hostname,
	410	&data->param_service,
	411	BIO_PARSE_PRIO_HOST);
	412	if (hold_service != data->param_service)
	413	OPENSSL_free(hold_service);
0f113f3e	414	} else if (num == 1) {
417be660 RL	415	OPENSSL_free(data->param_service);
417be660 RL	416	data->param_service = BUF_strdup(ptr);
0f113f3e	417	} else if (num == 2) {
417be660 RL	418	const BIO_ADDR addr = (const BIO_ADDR )ptr;
	419	if (ret) {
	420	data->param_hostname = BIO_ADDR_hostname_string(addr, 1);
	421	data->param_service = BIO_ADDR_service_string(addr, 1);
	422	BIO_ADDRINFO_free(data->addr_first);
	423	data->addr_first = NULL;
	424	data->addr_iter = NULL;
	425	}
0f113f3e	426	} else if (num == 3) {
417be660 RL	427	data->connect_family = (int )ptr;
	428	} else {
	429	ret = 0;
0f113f3e MC	430	}
	431	}
	432	break;
	433	case BIO_C_SET_NBIO:
417be660 RL	434	if (num != 0)
	435	data->connect_mode \|= BIO_SOCK_NONBLOCK;
	436	else
	437	data->connect_mode &= ~BIO_SOCK_NONBLOCK;
	438	break;
	439	case BIO_C_SET_CONNECT_MODE:
	440	data->connect_mode = (int)num;
0f113f3e MC	441	break;
	442	case BIO_C_GET_FD:
	443	if (b->init) {
	444	ip = (int *)ptr;
	445	if (ip != NULL)
	446	*ip = b->num;
	447	ret = b->num;
	448	} else
	449	ret = -1;
	450	break;
	451	case BIO_CTRL_GET_CLOSE:
	452	ret = b->shutdown;
	453	break;
	454	case BIO_CTRL_SET_CLOSE:
	455	b->shutdown = (int)num;
	456	break;
	457	case BIO_CTRL_PENDING:
	458	case BIO_CTRL_WPENDING:
	459	ret = 0;
	460	break;
	461	case BIO_CTRL_FLUSH:
	462	break;
	463	case BIO_CTRL_DUP:
	464	{
	465	dbio = (BIO *)ptr;
0f113f3e MC	466	if (data->param_hostname)
0f113f3e MC	467	BIO_set_conn_hostname(dbio, data->param_hostname);
417be660 RL	468	if (data->param_service)
	469	BIO_set_conn_port(dbio, data->param_service);
	470	BIO_set_conn_ip_family(dbio, data->connect_family);
	471	BIO_set_conn_mode(dbio, data->connect_mode);
0f113f3e MC	472	/*
	473	* FIXME: the cast of the function seems unlikely to be a good
	474	* idea
	475	*/
	476	(void)BIO_set_info_callback(dbio,
	477	(bio_info_cb *)data->info_callback);
	478	}
	479	break;
	480	case BIO_CTRL_SET_CALLBACK:
2722ff50	481	ret = 0; /* use callback ctrl */
0f113f3e MC	482	break;
	483	case BIO_CTRL_GET_CALLBACK:
	484	{
	485	int (*fptr) (const BIO bio, int state, int xret);
	486
	487	fptr = (int (*)(const BIO bio, int state, int xret))ptr;
	488	*fptr = data->info_callback;
	489	}
	490	break;
	491	default:
	492	ret = 0;
	493	break;
	494	}
	495	return (ret);
	496	}
d02b48c6	497
13083215	498	static long conn_callback_ctrl(BIO b, int cmd, bio_info_cb fp)
0f113f3e MC	499	{
	500	long ret = 1;
	501	BIO_CONNECT *data;
	502
	503	data = (BIO_CONNECT *)b->ptr;
	504
	505	switch (cmd) {
	506	case BIO_CTRL_SET_CALLBACK:
	507	{
	508	data->info_callback =
	509	(int ()(const struct bio_st , int, int))fp;
	510	}
	511	break;
	512	default:
	513	ret = 0;
	514	break;
	515	}
	516	return (ret);
	517	}
d3442bc7	518
0e1c0612	519	static int conn_puts(BIO bp, const char str)
0f113f3e MC	520	{
0f113f3e MC	521	int n, ret;
d02b48c6	522
0f113f3e MC	523	n = strlen(str);
	524	ret = conn_write(bp, str, n);
	525	return (ret);
	526	}
d02b48c6	527
c45a48c1	528	BIO BIO_new_connect(const char str)
0f113f3e MC	529	{
	530	BIO *ret;
	531
	532	ret = BIO_new(BIO_s_connect());
	533	if (ret == NULL)
	534	return (NULL);
	535	if (BIO_set_conn_hostname(ret, str))
	536	return (ret);
ca3a82c3 RS	537	BIO_free(ret);
ca3a82c3 RS	538	return (NULL);
0f113f3e	539	}
d02b48c6 RE	540
d02b48c6 RE	541	#endif