Work in progress.
This commit is contained in:
parent
43375d8443
commit
a5b9350fd1
|
@ -138,6 +138,23 @@ ssize_t net_con_ssl_handshake(struct net_connection* con, enum net_con_ssl_mode
|
||||||
#endif /* SSL_SUPPORT */
|
#endif /* SSL_SUPPORT */
|
||||||
|
|
||||||
|
|
||||||
|
int net_con_connect(struct net_connection* con, struct sockaddr* addr, size_t addr_len)
|
||||||
|
{
|
||||||
|
int ret = net_connect(con->sd, (struct sockaddr*) addr, addr_len);
|
||||||
|
if (ret == 0 || (ret == -1 && net_error() == EISCONN))
|
||||||
|
{
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
else if (ret == -1 && (net_error() == EALREADY || net_error() == EINPROGRESS || net_error() == EWOULDBLOCK || net_error() == EINTR))
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
ssize_t net_con_send(struct net_connection* con, const void* buf, size_t len)
|
ssize_t net_con_send(struct net_connection* con, const void* buf, size_t len)
|
||||||
{
|
{
|
||||||
int ret;
|
int ret;
|
||||||
|
@ -307,15 +324,18 @@ void net_con_callback(struct net_connection* con, int events)
|
||||||
|
|
||||||
case tls_st_connected:
|
case tls_st_connected:
|
||||||
LOG_PROTO("tls_st_connected, events=%s%s, ssl_flags=%s%s", (events & NET_EVENT_READ ? "R" : ""), (events & NET_EVENT_WRITE ? "W" : ""), con->flags & NET_WANT_SSL_READ ? "R" : "", con->flags & NET_WANT_SSL_WRITE ? "W" : "");
|
LOG_PROTO("tls_st_connected, events=%s%s, ssl_flags=%s%s", (events & NET_EVENT_READ ? "R" : ""), (events & NET_EVENT_WRITE ? "W" : ""), con->flags & NET_WANT_SSL_READ ? "R" : "", con->flags & NET_WANT_SSL_WRITE ? "W" : "");
|
||||||
if (events & NET_EVENT_WRITE && con->flags & NET_WANT_SSL_READ)
|
|
||||||
|
// continue a SSL_read() that wants to write.
|
||||||
|
if (events & NET_EVENT_WRITE && con->flags & NET_WANT_SSL_WRITE)
|
||||||
{
|
{
|
||||||
con->callback(con, events & NET_EVENT_READ, con->ptr);
|
con->callback(con, NET_EVENT_READ, con->ptr);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// continue a SSL_write() that wants to read something.
|
||||||
if (events & NET_EVENT_READ && con->flags & NET_WANT_SSL_WRITE)
|
if (events & NET_EVENT_READ && con->flags & NET_WANT_SSL_WRITE)
|
||||||
{
|
{
|
||||||
con->callback(con, events & NET_EVENT_READ, con->ptr);
|
con->callback(con, NET_EVENT_WRITE, con->ptr);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -75,6 +75,12 @@ extern ssize_t net_con_recv(struct net_connection* con, void* buf, size_t len);
|
||||||
*/
|
*/
|
||||||
extern ssize_t net_con_peek(struct net_connection* con, void* buf, size_t len);
|
extern ssize_t net_con_peek(struct net_connection* con, void* buf, size_t len);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns 1 if connected, 0 if net_con_connect needs to be called again,
|
||||||
|
* and -1 if an error occured.
|
||||||
|
*/
|
||||||
|
extern int net_con_connect(struct net_connection* con, struct sockaddr* addr, size_t addr_len);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Set timeout for connetion.
|
* Set timeout for connetion.
|
||||||
*
|
*
|
||||||
|
|
|
@ -376,13 +376,13 @@ int ADC_client_connect(struct ADC_client* client, const char* address)
|
||||||
|
|
||||||
client->callback(client, ADC_CLIENT_CONNECTING, 0);
|
client->callback(client, ADC_CLIENT_CONNECTING, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
int ret = net_connect(net_con_get_sd(client->con), (struct sockaddr*) &client->addr, sizeof(struct sockaddr_in));
|
int ret = net_con_connect(client->con, (struct sockaddr*) &client->addr, sizeof(struct sockaddr_in));
|
||||||
if (ret == 0 || (ret == -1 && net_error() == EISCONN))
|
if (ret == 1)
|
||||||
{
|
{
|
||||||
ADC_client_on_connected(client);
|
ADC_client_on_connected(client);
|
||||||
}
|
}
|
||||||
else if (ret == -1 && (net_error() == EALREADY || net_error() == EINPROGRESS || net_error() == EWOULDBLOCK || net_error() == EINTR))
|
else if (ret == 0)
|
||||||
{
|
{
|
||||||
if (client->state != ps_conn)
|
if (client->state != ps_conn)
|
||||||
{
|
{
|
||||||
|
@ -400,10 +400,17 @@ int ADC_client_connect(struct ADC_client* client, const char* address)
|
||||||
|
|
||||||
static void ADC_client_on_connected(struct ADC_client* client)
|
static void ADC_client_on_connected(struct ADC_client* client)
|
||||||
{
|
{
|
||||||
net_con_update(client->con, NET_EVENT_READ);
|
if (client->ssl)
|
||||||
client->callback(client, ADC_CLIENT_CONNECTED, 0);
|
{
|
||||||
ADC_client_send(client, ADC_HANDSHAKE);
|
net_con_ssl_handshake(client->con, net_con_ssl_mode_client, NULL);
|
||||||
ADC_client_set_state(client, ps_protocol);
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
net_con_update(client->con, NET_EVENT_READ);
|
||||||
|
client->callback(client, ADC_CLIENT_CONNECTED, 0);
|
||||||
|
ADC_client_send(client, ADC_HANDSHAKE);
|
||||||
|
ADC_client_set_state(client, ps_protocol);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void ADC_client_on_disconnected(struct ADC_client* client)
|
static void ADC_client_on_disconnected(struct ADC_client* client)
|
||||||
|
@ -432,8 +439,8 @@ static int ADC_client_parse_address(struct ADC_client* client, const char* arg)
|
||||||
{
|
{
|
||||||
char* split;
|
char* split;
|
||||||
int ssl = 0;
|
int ssl = 0;
|
||||||
struct hostent* dns;
|
struct hostent* dns = 0;
|
||||||
struct in_addr* addr;
|
struct in_addr* addr = 0;
|
||||||
|
|
||||||
if (!arg)
|
if (!arg)
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -445,12 +452,11 @@ static int ADC_client_parse_address(struct ADC_client* client, const char* arg)
|
||||||
return 0;
|
return 0;
|
||||||
|
|
||||||
/* Check for ADC or ADCS */
|
/* Check for ADC or ADCS */
|
||||||
if (!strncmp(arg, "adc://", 6))
|
if (!strncmp(arg, "adc://", 6)) ssl = 0;
|
||||||
ssl = 0;
|
else if (!strncmp(arg, "adcs://", 7)) ssl = 1;
|
||||||
else if (!strncmp(arg, "adcs://", 7))
|
else return 0;
|
||||||
ssl = 1;
|
|
||||||
else
|
client->ssl = ssl;
|
||||||
return 0;
|
|
||||||
|
|
||||||
/* Split hostname and port (if possible) */
|
/* Split hostname and port (if possible) */
|
||||||
split = strrchr(client->hub_address + 6 + ssl, ':');
|
split = strrchr(client->hub_address + 6 + ssl, ':');
|
||||||
|
@ -467,9 +473,7 @@ static int ADC_client_parse_address(struct ADC_client* client, const char* arg)
|
||||||
/* Resolve IP address (FIXME: blocking call) */
|
/* Resolve IP address (FIXME: blocking call) */
|
||||||
dns = gethostbyname(client->hub_address + 6 + ssl);
|
dns = gethostbyname(client->hub_address + 6 + ssl);
|
||||||
if (dns)
|
if (dns)
|
||||||
{
|
|
||||||
addr = (struct in_addr*) dns->h_addr_list[0];
|
addr = (struct in_addr*) dns->h_addr_list[0];
|
||||||
}
|
|
||||||
|
|
||||||
// Initialize the sockaddr struct.
|
// Initialize the sockaddr struct.
|
||||||
memset(&client->addr, 0, sizeof(client->addr));
|
memset(&client->addr, 0, sizeof(client->addr));
|
||||||
|
|
|
@ -105,6 +105,7 @@ struct ADC_client
|
||||||
struct adc_message* info;
|
struct adc_message* info;
|
||||||
char recvbuf[ADC_BUFSIZE];
|
char recvbuf[ADC_BUFSIZE];
|
||||||
char sendbuf[ADC_BUFSIZE];
|
char sendbuf[ADC_BUFSIZE];
|
||||||
|
int ssl;
|
||||||
adc_client_cb callback;
|
adc_client_cb callback;
|
||||||
size_t s_offset;
|
size_t s_offset;
|
||||||
size_t r_offset;
|
size_t r_offset;
|
||||||
|
|
|
@ -82,7 +82,8 @@ int main(int argc, char** argv)
|
||||||
|
|
||||||
while (running)
|
while (running)
|
||||||
{
|
{
|
||||||
net_backend_process();
|
if (!net_backend_process())
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
ADC_client_destroy(&client);
|
ADC_client_destroy(&client);
|
||||||
|
|
Loading…
Reference in New Issue