static enum pbpal_resolv_n_connect_result resolv_and_connect_wout_SSL(pubnub_t *pb)
{
PUBNUB_LOG_TRACE("resolv_and_connect_wout_SSL\n");
if (NULL == pb->pal.socket) {
char const*origin = PUBNUB_ORIGIN_SETTABLE ? pb->origin : PUBNUB_ORIGIN;
PUBNUB_LOG_TRACE("pb=%p: Don't have BIO\n", pb);
pb->pal.socket = BIO_new_connect((char*)origin);
}
if (NULL == pb->pal.socket) {
return pbpal_resolv_resource_failure;
}
BIO_set_conn_port(pb->pal.socket, "http");
BIO_set_nbio(pb->pal.socket, !pb->options.use_blocking_io);
WATCH_ENUM(pb->options.use_blocking_io);
if (BIO_do_connect(pb->pal.socket) <= 0) {
if (BIO_should_retry(pb->pal.socket)) {
return pbpal_connect_wouldblock;
}
ERR_print_errors_cb(print_to_pubnub_log, NULL);
PUBNUB_LOG_ERROR("BIO_do_connect failed\n");
return pbpal_connect_failed;
}
PUBNUB_LOG_TRACE("pb=%p: BIO connected\n", pb);
{
int fd = BIO_get_fd(pb->pal.socket, NULL);
socket_set_rcv_timeout(fd, pb->transaction_timeout_ms);
}
return pbpal_connect_success;
}
bool pbpal_read_over(pubnub_t *pb)
{
unsigned to_read = 0;
WATCH_ENUM(pb->sock_state);
WATCH_USHORT(pb->readlen);
WATCH_USHORT(pb->left);
WATCH_UINT(pb->len);
if (pb->readlen == 0) {
int recvres;
to_read = pb->len - pbpal_read_len(pb);
if (to_read > pb->left) {
to_read = pb->left;
}
recvres = socket_recv(pb->pal.socket, (char*)pb->ptr, to_read, 0);
if (recvres <= 0) {
/* This is error or connection close, which may be handled
in some way...
*/
return false;
}
pb->sock_state = STATE_READ;
pb->readlen = recvres;
}
to_read = pb->len;
if (pb->readlen < to_read) {
to_read = pb->readlen;
}
pb->ptr += to_read;
pb->readlen -= to_read;
PUBNUB_ASSERT_OPT(pb->left >= to_read);
pb->left -= to_read;
pb->len -= to_read;
if (pb->len == 0) {
pb->sock_state = STATE_NONE;
return true;
}
if (pb->left == 0) {
/* Buffer has been filled, but the requested block has not been
* read. We have to "reset" this "mini-fsm", as otherwise we
* won't read anything any more. This means that we have lost
* the current contents of the buffer, which is bad. In some
* general code, that should be reported, as the caller could
* save the contents of the buffer somewhere else or simply
* decide to ignore this block (when it does end eventually).
*/
pb->sock_state = STATE_NONE;
}
else {
pb->sock_state = STATE_NEWDATA_EXHAUSTED;
return false;
}
return true;
}
bool pbpal_read_over(pubnub_t *pb)
{
unsigned to_read = 0;
WATCH_ENUM(pb->sock_state);
WATCH_USHORT(pb->readlen);
WATCH_USHORT(pb->left);
WATCH_UINT(pb->len);
if (pb->readlen == 0) {
int recvres;
to_read = pb->len - pbpal_read_len(pb);
if (to_read > pb->left) {
to_read = pb->left;
}
recvres = socket_recv(pb->pal.socket, (char*)pb->ptr, to_read, 0);
if (recvres <= 0) {
/* This is error or connection close, which may be handled
in some way...
*/
return false;
}
pb->sock_state = STATE_READ;
pb->readlen = recvres;
}
pb->ptr += pb->readlen;
pb->left -= pb->readlen;
pb->readlen = 0;
if (pbpal_read_len(pb) >= (int)pb->len) {
/* If we have read all that was requested, we're done. */
PUBNUB_LOG_TRACE("Read all that was to be read.\n");
pb->sock_state = STATE_NONE;
return true;
}
if ((pb->left > 0)) {
pb->sock_state = STATE_NEWDATA_EXHAUSTED;
return false;
}
/* Otherwise, we just filled the buffer, but we return 'true', to
* enable the user to copy the data from the buffer to some other
* storage.
*/
PUBNUB_LOG_WARNING("Filled the buffer, but read %d and should %d\n", pbpal_read_len(pb), pb->len);
pb->sock_state = STATE_NONE;
return true;
}
int pbpal_start_read(pubnub_t *pb, size_t n)
{
if (pb->sock_state != STATE_NONE) {
PUBNUB_LOG_ERROR("pbpal_start_read(): pb->sock_state != STATE_NONE: "); WATCH_ENUM(pb->sock_state);
return -1;
}
if (pb->ptr > (uint8_t*)pb->core.http_buf) {
unsigned distance = pb->ptr - (uint8_t*)pb->core.http_buf;
memmove(pb->core.http_buf, pb->ptr, pb->readlen);
pb->ptr -= distance;
pb->left += distance;
}
else {
if (pb->left == 0) {
/* Obviously, our buffer is not big enough, maybe some
error should be reported */
buf_setup(pb);
}
}
pb->sock_state = STATE_READ;
pb->len = n;
return +1;
}
int pbnc_fsm(struct pubnub_ *pb)
{
enum pubnub_res pbrslt;
int i;
PUBNUB_LOG_TRACE("pbnc_fsm()\t");
next_state:
WATCH_ENUM_ONCHANGE(pb->state);
switch (pb->state) {
case PBS_NULL:
break;
case PBS_IDLE:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_resolv_and_connect(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_send_wouldblock:
return 0;
case pbpal_resolv_sent:
case pbpal_resolv_rcv_wouldblock:
pb->state = PBS_WAIT_DNS;
break;
case pbpal_connect_wouldblock:
pb->state = PBS_WAIT_CONNECT;
break;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
break;
default:
pb->core.last_result = PNR_ADDR_RESOLUTION_FAILED;
pbntf_trans_outcome(pb);
return 0;
}
i = pbntf_got_socket(pb, pb->pal.socket);
if (0 == i) {
goto next_state;
}
else if (i < 0) {
pb->core.last_result = PNR_CONNECT_FAILED;
pbntf_trans_outcome(pb);
}
break;
}
case PBS_WAIT_DNS:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_check_resolv_and_connect(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_send_wouldblock:
case pbpal_resolv_sent:
pb->core.last_result = PNR_INTERNAL_ERROR;
pbntf_trans_outcome(pb);
break;
case pbpal_resolv_rcv_wouldblock:
break;
case pbpal_connect_wouldblock:
pbntf_update_socket(pb, pb->pal.socket);
pb->state = PBS_WAIT_CONNECT;
break;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
goto next_state;
default:
pb->core.last_result = PNR_ADDR_RESOLUTION_FAILED;
pbntf_trans_outcome(pb);
break;
}
break;
}
case PBS_WAIT_CONNECT:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_connected(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_resource_failure:
case pbpal_connect_resource_failure:
case pbpal_connect_failed:
pb->core.last_result = PNR_ADDR_RESOLUTION_FAILED;
pbntf_trans_outcome(pb);
return 0;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
goto next_state;
case pbpal_connect_wouldblock:
default:
pbntf_update_socket(pb, pb->pal.socket);
break;
}
break;
}
case PBS_CONNECTED:
pbpal_send_literal_str(pb, "GET ");
pb->state = PBS_TX_GET;
goto next_state;
case PBS_TX_GET:
i = pbpal_send_status(pb);
if (i <= 0) {
pb->state = PBS_TX_PATH;
if ((i < 0) || (-1 == pbpal_send_str(pb, pb->core.http_buf))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
goto next_state;
}
break;
case PBS_TX_PATH:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
pbpal_send_literal_str(pb, " HTTP/1.1\r\nHost: ");
pb->state = PBS_TX_VER;
goto next_state;
}
break;
case PBS_TX_VER:
i = pbpal_send_status(pb);
if (i <= 0) {
char const* o = PUBNUB_ORIGIN_SETTABLE ? pb->origin : PUBNUB_ORIGIN;
pb->state = PBS_TX_ORIGIN;
if ((i < 0) || (-1 == pbpal_send_str(pb, o))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
goto next_state;
}
break;
case PBS_TX_ORIGIN:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
pbpal_send_literal_str(pb, "\r\nUser-Agent: PubNub-C-core/2.1\r\nConnection: Keep-Alive\r\n\r\n");
pb->state = PBS_TX_FIN_HEAD;
goto next_state;
}
break;
case PBS_TX_FIN_HEAD:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
pbpal_start_read_line(pb);
pb->state = PBS_RX_HTTP_VER;
goto next_state;
}
break;
case PBS_RX_HTTP_VER:
pbrslt = pbpal_line_read_status(pb);
switch (pbrslt) {
case PNR_IN_PROGRESS:
break;
case PNR_OK:
if (strncmp(pb->core.http_buf, "HTTP/1.", 7) != 0) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
pb->http_code = atoi(pb->core.http_buf + 9);
WATCH_USHORT(pb->http_code);
pb->core.http_content_len = 0;
pb->http_chunked = false;
pb->state = PBS_RX_HEADERS;
goto next_state;
default:
outcome_detected(pb, pbrslt);
break;
}
break;
case PBS_RX_HEADERS:
PUBNUB_LOG_TRACE("PBS_RX_HEADERS\n");
pbpal_start_read_line(pb);
pb->state = PBS_RX_HEADER_LINE;
goto next_state;
case PBS_RX_HEADER_LINE:
PUBNUB_LOG_TRACE("PBS_RX_HEADER_LINE\n");
pbrslt = pbpal_line_read_status(pb);
switch (pbrslt) {
case PNR_IN_PROGRESS:
break;
case PNR_OK:
{
char h_chunked[] = "Transfer-Encoding: chunked";
char h_length[] = "Content-Length: ";
int read_len = pbpal_read_len(pb);
PUBNUB_LOG_TRACE("header line was read: %.*s\n", read_len, pb->core.http_buf);
WATCH_INT(read_len);
if (read_len <= 2) {
pb->core.http_buf_len = 0;
if (!pb->http_chunked) {
if (0 == pb->core.http_content_len) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
pb->state = PBS_RX_BODY;
}
else {
pb->state = PBS_RX_CHUNK_LEN;
}
goto next_state;
}
if (strncmp(pb->core.http_buf, h_chunked, sizeof h_chunked - 1) == 0) {
pb->http_chunked = true;
}
else if (strncmp(pb->core.http_buf, h_length, sizeof h_length - 1) == 0) {
size_t len = atoi(pb->core.http_buf + sizeof h_length - 1);
if (0 != pbcc_realloc_reply_buffer(&pb->core, len)) {
outcome_detected(pb, PNR_REPLY_TOO_BIG);
break;
}
pb->core.http_content_len = len;
}
pb->state = PBS_RX_HEADERS;
goto next_state;
}
default:
outcome_detected(pb, pbrslt);
break;
}
break;
case PBS_RX_BODY:
PUBNUB_LOG_TRACE("PBS_RX_BODY\n");
if (pb->core.http_buf_len < pb->core.http_content_len) {
pbpal_start_read(pb, pb->core.http_content_len - pb->core.http_buf_len);
pb->state = PBS_RX_BODY_WAIT;
goto next_state;
}
else {
finish(pb);
}
break;
case PBS_RX_BODY_WAIT:
PUBNUB_LOG_TRACE("PBS_RX_BODY_WAIT\n");
if (pbpal_read_over(pb)) {
unsigned len = pbpal_read_len(pb);
WATCH_UINT(len);
WATCH_UINT(pb->core.http_buf_len);
memcpy(
pb->core.http_reply + pb->core.http_buf_len,
pb->core.http_buf,
len
);
pb->core.http_buf_len += len;
pb->state = PBS_RX_BODY;
goto next_state;
}
break;
case PBS_RX_CHUNK_LEN:
PUBNUB_LOG_TRACE("PBS_RX_CHUNK_LEN\n");
pbpal_start_read_line(pb);
pb->state = PBS_RX_CHUNK_LEN_LINE;
goto next_state;
case PBS_RX_CHUNK_LEN_LINE:
pbrslt = pbpal_line_read_status(pb);
PUBNUB_LOG_TRACE("PBS_RX_CHUNK_LEN_LINE: pbrslt=%d\n", pbrslt);
switch (pbrslt) {
case PNR_IN_PROGRESS:
break;
case PNR_OK:
{
unsigned chunk_length = strtoul(pb->core.http_buf, NULL, 16);
PUBNUB_LOG_TRACE("About to read a chunk w/length: %d\n", chunk_length);
if (chunk_length == 0) {
finish(pb);
}
else if (chunk_length > sizeof pb->core.http_buf) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 != pbcc_realloc_reply_buffer(&pb->core, pb->core.http_buf_len + chunk_length)) {
outcome_detected(pb, PNR_REPLY_TOO_BIG);
}
else {
pb->core.http_content_len = chunk_length + 2;
pb->state = PBS_RX_BODY_CHUNK;
goto next_state;
}
break;
}
default:
outcome_detected(pb, pbrslt);
break;
}
break;
case PBS_RX_BODY_CHUNK:
PUBNUB_LOG_TRACE("PBS_RX_BODY_CHUNK\n");
if (pb->core.http_content_len > 0) {
pbpal_start_read(pb, pb->core.http_content_len);
pb->state = PBS_RX_BODY_CHUNK_WAIT;
}
else {
pb->state = PBS_RX_CHUNK_LEN;
}
goto next_state;
case PBS_RX_BODY_CHUNK_WAIT:
PUBNUB_LOG_TRACE("PBS_RX_BODY_CHUNK_WAIT\n");
if (pbpal_read_over(pb)) {
unsigned len = pbpal_read_len(pb);
PUBNUB_ASSERT_OPT(pb->core.http_content_len >= len);
PUBNUB_ASSERT_OPT(len > 0);
if (pb->core.http_content_len > 2) {
unsigned to_copy = pb->core.http_content_len - 2;
if (len < to_copy) {
to_copy = len;
}
memcpy(
pb->core.http_reply + pb->core.http_buf_len,
pb->core.http_buf,
to_copy
);
pb->core.http_buf_len += to_copy;
}
pb->core.http_content_len -= len;
pb->state = PBS_RX_BODY_CHUNK;
goto next_state;
}
break;
case PBS_WAIT_CLOSE:
if (pbpal_closed(pb)) {
pbpal_forget(pb);
pbntf_trans_outcome(pb);
}
break;
case PBS_WAIT_CANCEL:
pb->state = PBS_WAIT_CANCEL_CLOSE;
if (pbpal_close(pb) <= 0) {
goto next_state;
}
break;
case PBS_WAIT_CANCEL_CLOSE:
if (pbpal_closed(pb)) {
pbpal_forget(pb);
pb->core.msg_ofs = pb->core.msg_end = 0;
pbntf_trans_outcome(pb);
}
break;
}
return 0;
}
int pbnc_fsm(struct pubnub_ *pb)
{
enum pubnub_res pbrslt;
int i;
PUBNUB_LOG_TRACE("pbnc_fsm()\t");
next_state:
WATCH_ENUM(pb->state);
switch (pb->state) {
case PBS_NULL:
break;
case PBS_IDLE:
pb->retry_after_close = false;
pb->state = PBS_READY;
switch (pbntf_enqueue_for_processing(pb)) {
case -1:
pb->core.last_result = PNR_INTERNAL_ERROR;
pbntf_trans_outcome(pb);
return 0;
case 0:
goto next_state;
case +1:
break;
}
break;
case PBS_READY:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_resolv_and_connect(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_send_wouldblock:
pb->state = PBS_WAIT_DNS_SEND;
break;
case pbpal_resolv_sent:
case pbpal_resolv_rcv_wouldblock:
pb->state = PBS_WAIT_DNS_RCV;
pbntf_watch_in_events(pb);
break;
case pbpal_connect_wouldblock:
pb->state = PBS_WAIT_CONNECT;
break;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
break;
default:
pb->core.last_result = PNR_ADDR_RESOLUTION_FAILED;
pbntf_trans_outcome(pb);
return 0;
}
i = pbntf_got_socket(pb, pb->pal.socket);
if (0 == i) {
goto next_state;
}
else if (i < 0) {
pb->core.last_result = PNR_CONNECT_FAILED;
pbntf_trans_outcome(pb);
}
break;
}
case PBS_WAIT_DNS_SEND:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_check_resolv_and_connect(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_send_wouldblock:
break;
case pbpal_resolv_sent:
case pbpal_resolv_rcv_wouldblock:
pbntf_update_socket(pb, pb->pal.socket);
pb->state = PBS_WAIT_DNS_RCV;
pbntf_watch_in_events(pb);
break;
case pbpal_connect_wouldblock:
pbntf_update_socket(pb, pb->pal.socket);
pb->state = PBS_WAIT_CONNECT;
break;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
goto next_state;
default:
outcome_detected(pb, PNR_ADDR_RESOLUTION_FAILED);
break;
}
break;
}
case PBS_WAIT_DNS_RCV:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_check_resolv_and_connect(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_send_wouldblock:
case pbpal_resolv_sent:
outcome_detected(pb, PNR_INTERNAL_ERROR);
break;
case pbpal_resolv_rcv_wouldblock:
break;
case pbpal_connect_wouldblock:
pbntf_update_socket(pb, pb->pal.socket);
pb->state = PBS_WAIT_CONNECT;
pbntf_watch_out_events(pb);
break;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
pbntf_watch_out_events(pb);
goto next_state;
default:
outcome_detected(pb, PNR_ADDR_RESOLUTION_FAILED);
break;
}
break;
}
case PBS_WAIT_CONNECT:
{
enum pbpal_resolv_n_connect_result rslv = pbpal_check_connect(pb);
WATCH_ENUM(rslv);
switch (rslv) {
case pbpal_resolv_send_wouldblock:
case pbpal_resolv_sent:
case pbpal_resolv_rcv_wouldblock:
pb->core.last_result = PNR_INTERNAL_ERROR;
pbntf_trans_outcome(pb);
break;
case pbpal_connect_wouldblock:
break;
case pbpal_connect_success:
pb->state = PBS_CONNECTED;
goto next_state;
default:
outcome_detected(pb, PNR_CONNECT_FAILED);
break;
}
break;
}
case PBS_CONNECTED:
#if PUBNUB_PROXY_API
if ((pb->proxy_type == pbproxyHTTP_CONNECT) && (!pb->proxy_tunnel_established)) {
pbpal_send_literal_str(pb, "CONNECT ");
}
else {
pbpal_send_literal_str(pb, "GET ");
}
#else
pbpal_send_literal_str(pb, "GET ");
#endif
pb->state = PBS_TX_GET;
goto next_state;
case PBS_TX_GET:
i = pbpal_send_status(pb);
if (i <= 0) {
#if PUBNUB_PROXY_API
switch (pb->proxy_type) {
case pbproxyHTTP_GET:
pb->state = PBS_TX_SCHEME;
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
strcpy(pb->proxy_saved_path, pb->core.http_buf);
pbpal_send_literal_str(pb, "http://");
break;
case pbproxyHTTP_CONNECT:
pb->state = PBS_TX_SCHEME;
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
if (!pb->proxy_tunnel_established) {
strcpy(pb->proxy_saved_path, pb->core.http_buf);
}
else {
strcpy(pb->core.http_buf, pb->proxy_saved_path);
}
break;
case pbproxyNONE:
pb->state = PBS_TX_PATH;
if ((i < 0) || (-1 == pbpal_send_str(pb, pb->core.http_buf))) {
outcome_detected(pb, PNR_IO_ERROR);
}
break;
default:
outcome_detected(pb, PNR_INTERNAL_ERROR);
break;
}
#else
pb->state = PBS_TX_PATH;
if ((i < 0) || (-1 == pbpal_send_str(pb, pb->core.http_buf))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
#endif /* PUBNUB_PROXY_API */
goto next_state;
}
break;
#if PUBNUB_PROXY_API
case PBS_TX_SCHEME:
i = pbpal_send_status(pb);
if (i <= 0) {
if ((pb->proxy_type == pbproxyHTTP_CONNECT) && pb->proxy_tunnel_established) {
pb->state = PBS_TX_HOST;
}
else {
char const* o = PUBNUB_ORIGIN_SETTABLE ? pb->origin : PUBNUB_ORIGIN;
pb->state = PBS_TX_HOST;
if ((i < 0) || (-1 == pbpal_send_str(pb, o))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
}
goto next_state;
}
break;
case PBS_TX_HOST:
i = pbpal_send_status(pb);
if (i <= 0) {
if ((pb->proxy_type == pbproxyHTTP_CONNECT) && !pb->proxy_tunnel_established) {
char port_num[20];
snprintf(port_num, sizeof port_num, ":%d", 80);
pbpal_send_str(pb, port_num);
pb->state = PBS_TX_PORT_NUM;
goto next_state;
}
else {
pb->state = PBS_TX_PATH;
if ((i < 0) || (-1 == pbpal_send_str(pb, pb->core.http_buf))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
}
goto next_state;
}
break;
case PBS_TX_PORT_NUM:
i = pbpal_send_status(pb);
if (i <= 0) {
pb->state = PBS_TX_PATH;
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
goto next_state;
}
break;
#endif /* PUBNUB_PROXY_API */
case PBS_TX_PATH:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
pbpal_send_literal_str(pb, " HTTP/1.1\r\nHost: ");
pb->state = PBS_TX_VER;
goto next_state;
}
break;
case PBS_TX_VER:
i = pbpal_send_status(pb);
if (i <= 0) {
char const* o = PUBNUB_ORIGIN_SETTABLE ? pb->origin : PUBNUB_ORIGIN;
pb->state = PBS_TX_ORIGIN;
if ((i < 0) || (-1 == pbpal_send_str(pb, o))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
goto next_state;
}
break;
case PBS_TX_ORIGIN:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
#if PUBNUB_PROXY_API
char header_to_send[1024] = "\r\n";
if (0 == pbproxy_http_header_to_send(pb, header_to_send+2, sizeof header_to_send-2)) {
PUBNUB_LOG_TRACE("Sending HTTP proxy header: '%s'\n", header_to_send);
pb->state = PBS_TX_PROXY_AUTHORIZATION;
if ((i < 0) || (-1 == pbpal_send_str(pb, header_to_send))) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
}
else {
pbpal_send_literal_str(pb, "\r\nUser-Agent: PubNub-C-core/2.2\r\nConnection: Keep-Alive\r\n\r\n");
pb->state = PBS_TX_FIN_HEAD;
}
#else
pbpal_send_literal_str(pb, "\r\nUser-Agent: PubNub-C-core/2.2\r\nConnection: Keep-Alive\r\n\r\n");
pb->state = PBS_TX_FIN_HEAD;
#endif
goto next_state;
}
break;
case PBS_TX_PROXY_AUTHORIZATION:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
pbpal_send_literal_str(pb, "\r\nUser-Agent: PubNub-C-core/2.2\r\nConnection: Keep-Alive\r\n\r\n");
pb->state = PBS_TX_FIN_HEAD;
goto next_state;
}
break;
case PBS_TX_FIN_HEAD:
i = pbpal_send_status(pb);
if (i < 0) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 == i) {
pbpal_start_read_line(pb);
pb->state = PBS_RX_HTTP_VER;
pbntf_watch_in_events(pb);
goto next_state;
}
break;
case PBS_RX_HTTP_VER:
pbrslt = pbpal_line_read_status(pb);
switch (pbrslt) {
case PNR_IN_PROGRESS:
break;
case PNR_OK:
if (strncmp(pb->core.http_buf, "HTTP/1.", 7) != 0) {
outcome_detected(pb, PNR_IO_ERROR);
break;
}
pb->http_code = atoi(pb->core.http_buf + 9);
WATCH_USHORT(pb->http_code);
pb->core.http_content_len = 0;
pb->http_chunked = false;
pb->state = PBS_RX_HEADERS;
goto next_state;
default:
outcome_detected(pb, pbrslt);
break;
}
break;
case PBS_RX_HEADERS:
PUBNUB_LOG_TRACE("PBS_RX_HEADERS\n");
pbpal_start_read_line(pb);
pb->state = PBS_RX_HEADER_LINE;
goto next_state;
case PBS_RX_HEADER_LINE:
PUBNUB_LOG_TRACE("PBS_RX_HEADER_LINE\n");
pbrslt = pbpal_line_read_status(pb);
switch (pbrslt) {
case PNR_IN_PROGRESS:
break;
case PNR_OK:
{
char h_chunked[] = "Transfer-Encoding: chunked";
char h_length[] = "Content-Length: ";
int read_len = pbpal_read_len(pb);
PUBNUB_LOG_TRACE("header line was read: %.*s\n", read_len, pb->core.http_buf);
WATCH_INT(read_len);
if (read_len <= 2) {
pb->core.http_buf_len = 0;
if (!pb->http_chunked) {
if (0 == pb->core.http_content_len) {
#if PUBNUB_PROXY_API
if ((pb->proxy_type == pbproxyHTTP_CONNECT) && !pb->proxy_tunnel_established) {
finish(pb);
if (pb->retry_after_close) {
goto next_state;
}
break;
}
#endif
outcome_detected(pb, PNR_IO_ERROR);
break;
}
pb->state = PBS_RX_BODY;
}
else {
pb->state = PBS_RX_CHUNK_LEN;
}
goto next_state;
}
if (strncmp(pb->core.http_buf, h_chunked, sizeof h_chunked - 1) == 0) {
pb->http_chunked = true;
}
else if (strncmp(pb->core.http_buf, h_length, sizeof h_length - 1) == 0) {
size_t len = atoi(pb->core.http_buf + sizeof h_length - 1);
if (0 != pbcc_realloc_reply_buffer(&pb->core, len)) {
outcome_detected(pb, PNR_REPLY_TOO_BIG);
break;
}
pb->core.http_content_len = len;
}
else {
pbproxy_handle_http_header(pb, pb->core.http_buf);
}
pb->state = PBS_RX_HEADERS;
goto next_state;
}
default:
outcome_detected(pb, pbrslt);
break;
}
break;
case PBS_RX_BODY:
PUBNUB_LOG_TRACE("PBS_RX_BODY\n");
if (pb->core.http_buf_len < pb->core.http_content_len) {
pbpal_start_read(pb, pb->core.http_content_len - pb->core.http_buf_len);
pb->state = PBS_RX_BODY_WAIT;
goto next_state;
}
else {
finish(pb);
if (pb->retry_after_close) {
goto next_state;
}
}
break;
case PBS_RX_BODY_WAIT:
PUBNUB_LOG_TRACE("PBS_RX_BODY_WAIT\n");
if (pbpal_read_over(pb)) {
unsigned len = pbpal_read_len(pb);
WATCH_UINT(len);
WATCH_UINT(pb->core.http_buf_len);
memcpy(
pb->core.http_reply + pb->core.http_buf_len,
pb->core.http_buf,
len
);
pb->core.http_buf_len += len;
pb->state = PBS_RX_BODY;
goto next_state;
}
break;
case PBS_RX_CHUNK_LEN:
PUBNUB_LOG_TRACE("PBS_RX_CHUNK_LEN\n");
pbpal_start_read_line(pb);
pb->state = PBS_RX_CHUNK_LEN_LINE;
goto next_state;
case PBS_RX_CHUNK_LEN_LINE:
pbrslt = pbpal_line_read_status(pb);
PUBNUB_LOG_TRACE("PBS_RX_CHUNK_LEN_LINE: pbrslt=%d\n", pbrslt);
switch (pbrslt) {
case PNR_IN_PROGRESS:
break;
case PNR_OK:
{
unsigned chunk_length = strtoul(pb->core.http_buf, NULL, 16);
PUBNUB_LOG_TRACE("About to read a chunk w/length: %d\n", chunk_length);
if (chunk_length == 0) {
finish(pb);
if (pb->retry_after_close) {
goto next_state;
}
}
else if (chunk_length > sizeof pb->core.http_buf) {
outcome_detected(pb, PNR_IO_ERROR);
}
else if (0 != pbcc_realloc_reply_buffer(&pb->core, pb->core.http_buf_len + chunk_length)) {
outcome_detected(pb, PNR_REPLY_TOO_BIG);
}
else {
pb->core.http_content_len = chunk_length + 2;
pb->state = PBS_RX_BODY_CHUNK;
goto next_state;
}
break;
}
default:
outcome_detected(pb, pbrslt);
break;
}
break;
case PBS_RX_BODY_CHUNK:
PUBNUB_LOG_TRACE("PBS_RX_BODY_CHUNK\n");
if (pb->core.http_content_len > 0) {
pbpal_start_read(pb, pb->core.http_content_len);
pb->state = PBS_RX_BODY_CHUNK_WAIT;
}
else {
pb->state = PBS_RX_CHUNK_LEN;
}
goto next_state;
case PBS_RX_BODY_CHUNK_WAIT:
PUBNUB_LOG_TRACE("PBS_RX_BODY_CHUNK_WAIT\n");
if (pbpal_read_over(pb)) {
unsigned len = pbpal_read_len(pb);
PUBNUB_ASSERT_OPT(pb->core.http_content_len >= len);
PUBNUB_ASSERT_OPT(len > 0);
if (pb->core.http_content_len > 2) {
unsigned to_copy = pb->core.http_content_len - 2;
if (len < to_copy) {
to_copy = len;
}
memcpy(
pb->core.http_reply + pb->core.http_buf_len,
pb->core.http_buf,
to_copy
);
pb->core.http_buf_len += to_copy;
}
pb->core.http_content_len -= len;
pb->state = PBS_RX_BODY_CHUNK;
goto next_state;
}
break;
case PBS_WAIT_CLOSE:
if (pbpal_closed(pb)) {
if (pb->retry_after_close) {
pb->state = PBS_IDLE;
}
else {
pbpal_forget(pb);
pbntf_trans_outcome(pb);
}
}
break;
case PBS_WAIT_CANCEL:
pb->state = PBS_WAIT_CANCEL_CLOSE;
if (pbpal_close(pb) <= 0) {
goto next_state;
}
break;
case PBS_WAIT_CANCEL_CLOSE:
if (pbpal_closed(pb)) {
if (pb->retry_after_close) {
pb->state = PBS_IDLE;
}
else {
pbpal_forget(pb);
pb->core.msg_ofs = pb->core.msg_end = 0;
pbntf_trans_outcome(pb);
}
}
break;
}
return 0;
}
enum pbpal_resolv_n_connect_result pbpal_resolv_and_connect(pubnub_t *pb)
{
SSL *ssl = NULL;
int rslt;
char const* origin = PUBNUB_ORIGIN_SETTABLE ? pb->origin : PUBNUB_ORIGIN;
PUBNUB_ASSERT(pb_valid_ctx_ptr(pb));
PUBNUB_ASSERT_OPT((pb->state == PBS_READY) || (pb->state == PBS_WAIT_CONNECT));
if (!pb->options.useSSL) {
return resolv_and_connect_wout_SSL(pb);
}
if (NULL == pb->pal.ctx) {
PUBNUB_LOG_TRACE("pb=%p: Don't have SSL_CTX\n", pb);
pb->pal.ctx = SSL_CTX_new(SSLv23_client_method());
if (NULL == pb->pal.ctx) {
ERR_print_errors_cb(print_to_pubnub_log, NULL);
PUBNUB_LOG_ERROR("pb=%p SSL_CTX_new failed\n", pb);
return pbpal_resolv_resource_failure;
}
PUBNUB_LOG_TRACE("pb=%p: Got SSL_CTX\n", pb);
add_pubnub_cert(pb->pal.ctx);
}
if (NULL == pb->pal.socket) {
PUBNUB_LOG_TRACE("pb=%p: Don't have BIO\n", pb);
pb->pal.socket = BIO_new_ssl_connect(pb->pal.ctx);
if (PUBNUB_TIMERS_API) {
pb->pal.connect_timeout = time(NULL) + pb->transaction_timeout_ms/1000;
}
}
else {
BIO_get_ssl(pb->pal.socket, &ssl);
if (NULL == ssl) {
return resolv_and_connect_wout_SSL(pb);
}
ssl = NULL;
}
if (NULL == pb->pal.socket) {
ERR_print_errors_cb(print_to_pubnub_log, NULL);
return pbpal_resolv_resource_failure;
}
PUBNUB_LOG_TRACE("pb=%p: Using BIO == %p\n", pb, pb->pal.socket);
BIO_get_ssl(pb->pal.socket, &ssl);
PUBNUB_ASSERT(NULL != ssl);
SSL_set_mode(ssl, SSL_MODE_AUTO_RETRY); /* maybe not auto_retry? */
if (pb->pal.session != NULL) {
SSL_set_session(ssl, pb->pal.session);
}
BIO_set_conn_hostname(pb->pal.socket, origin);
BIO_set_conn_port(pb->pal.socket, "https");
if (pb->pal.ip_timeout != 0) {
if (pb->pal.ip_timeout < time(NULL)) {
pb->pal.ip_timeout = 0;
}
else {
PUBNUB_LOG_TRACE("SSL re-connect to: %d.%d.%d.%d\n", pb->pal.ip[0], pb->pal.ip[1], pb->pal.ip[2], pb->pal.ip[3]);
BIO_set_conn_ip(pb->pal.socket, pb->pal.ip);
}
}
BIO_set_nbio(pb->pal.socket, !pb->options.use_blocking_io);
WATCH_ENUM(pb->options.use_blocking_io);
if (BIO_do_connect(pb->pal.socket) <= 0) {
if (BIO_should_retry(pb->pal.socket) && PUBNUB_TIMERS_API && (pb->pal.connect_timeout > time(NULL))) {
PUBNUB_LOG_TRACE("pb=%p: BIO_should_retry\n", pb);
return pbpal_connect_wouldblock;
}
/* Expire the IP for the next connect */
pb->pal.ip_timeout = 0;
ERR_print_errors_cb(print_to_pubnub_log, NULL);
if (pb->pal.session != NULL) {
SSL_SESSION_free(pb->pal.session);
pb->pal.session = NULL;
}
PUBNUB_LOG_ERROR("pb=%p: BIO_do_connect failed\n", pb);
return pbpal_connect_failed;
}
PUBNUB_LOG_TRACE("pb=%p: BIO connected\n", pb);
{
int fd = BIO_get_fd(pb->pal.socket, NULL);
socket_set_rcv_timeout(fd, pb->transaction_timeout_ms);
}
rslt = SSL_get_verify_result(ssl);
if (rslt != X509_V_OK) {
PUBNUB_LOG_WARNING("pb=%p: SSL_get_verify_result() failed == %d(%s)\n", pb, rslt, X509_verify_cert_error_string(rslt));
ERR_print_errors_cb(print_to_pubnub_log, NULL);
if (pb->options.fallbackSSL) {
BIO_free_all(pb->pal.socket);
pb->pal.socket = NULL;
return resolv_and_connect_wout_SSL(pb);
}
return pbpal_connect_failed;
}
PUBNUB_LOG_INFO("pb=%p: SSL session reused: %s\n", pb, SSL_session_reused(ssl) ? "yes" : "no");
if (pb->pal.session != NULL) {
SSL_SESSION_free(pb->pal.session);
}
pb->pal.session = SSL_get1_session(ssl);
if (0 == pb->pal.ip_timeout) {
pb->pal.ip_timeout = SSL_SESSION_get_time(pb->pal.session) + SSL_SESSION_get_timeout(pb->pal.session);
memcpy(pb->pal.ip, BIO_get_conn_ip(pb->pal.socket), 4);
}
PUBNUB_LOG_TRACE("pb=%p: SSL connected to IP: %d.%d.%d.%d\n", pb, pb->pal.ip[0], pb->pal.ip[1], pb->pal.ip[2], pb->pal.ip[3]);
return pbpal_connect_success;
}
