static int handle_state_initial(struct npool *nsp, struct nevent *nse, void *udata) {
struct proxy_chain_context *px_ctx = nse->iod->px_ctx;
struct sockaddr_storage *ss;
size_t sslen;
unsigned short port;
struct proxy_node *next;
int timeout;
px_ctx->px_state = PROXY_STATE_HTTP_TCP_CONNECTED;
next = proxy_ctx_node_next(px_ctx);
if (next) {
ss = &next->ss;
sslen = next->sslen;
port = next->port;
} else {
ss = &px_ctx->target_ss;
sslen = px_ctx->target_sslen;
port = px_ctx->target_port;
}
timeout = TIMEVAL_MSEC_SUBTRACT(nse->timeout, nsock_tod);
nsock_printf(nsp, (nsock_iod)nse->iod, nsock_proxy_ev_dispatch,
timeout, udata, "CONNECT %s:%d HTTP/1.1\r\n\r\n",
inet_ntop_ez(ss, sslen), (int)port);
nsock_readlines(nsp, (nsock_iod)nse->iod, nsock_proxy_ev_dispatch,
timeout, udata, 1);
return 0;
}
void telnet_event_handler(nsock_pool nsp, nsock_event nse, void *mydata) {
nsock_iod nsi = nse_iod(nse);
enum nse_status status = nse_status(nse);
enum nse_type type = nse_type(nse);
struct sockaddr_in peer;
struct telnet_state *ts;
int nbytes;
char *str;
int read_timeout = -1;
int write_timeout = 2000;
ts = (struct telnet_state *)mydata;
printf("telnet_event_handler: Received callback of type %s with status %s\n", nse_type2str(type), nse_status2str(status));
if (status == NSE_STATUS_SUCCESS) {
switch (type) {
case NSE_TYPE_CONNECT:
case NSE_TYPE_CONNECT_SSL:
nsi_getlastcommunicationinfo(nsi, NULL, NULL, NULL, (struct sockaddr *)&peer, sizeof peer);
printf("Successfully connected %sto %s:%hu -- start typing lines\n", (type == NSE_TYPE_CONNECT_SSL) ? "(SSL!) " : "", inet_ntoa(peer.sin_addr), peer.sin_port);
/* First of all, lets add STDIN to our list of watched filehandles */
if ((ts->stdin_nsi = nsi_new2(nsp, STDIN_FILENO, NULL)) == NULL) {
fprintf(stderr, "Failed to create stdin msi\n");
exit(1);
}
/* Now lets read from stdin and the network, line buffered (by nsock) */
ts->latest_readtcpev = nsock_readlines(nsp, ts->tcp_nsi, telnet_event_handler, read_timeout, ts, 1);
ts->latest_readstdinev = nsock_readlines(nsp, ts->stdin_nsi, telnet_event_handler, read_timeout, ts, 1);
break;
case NSE_TYPE_READ:
str = nse_readbuf(nse, &nbytes);
if (nsi == ts->tcp_nsi) {
printf("%s", str);
/* printf("Read from tcp socket (%d bytes):\n%s", nbytes, str); */
ts->latest_readtcpev = nsock_readlines(nsp, ts->tcp_nsi, telnet_event_handler, read_timeout, ts, 1);
} else {
/* printf("Read from stdin (%d bytes):\n%s", nbytes, str); */
nsock_write(nsp, ts->tcp_nsi, telnet_event_handler, write_timeout, ts, str, nbytes);
ts->latest_readstdinev = nsock_readlines(nsp, ts->stdin_nsi, telnet_event_handler, read_timeout, ts, 1);
}
break;
case NSE_TYPE_WRITE:
/* Nothing to do, really */
break;
case NSE_TYPE_TIMER:
break;
default:
fprintf(stderr, "telnet_event_handler: Got bogus type -- quitting\n");
exit(1);
break;
}
} else if (status == NSE_STATUS_EOF) {
printf("Got EOF from %s\nCancelling outstanding readevents.\n", (nsi == ts->tcp_nsi) ? "tcp socket" : "stdin");
/* One of these is the event I am currently handling! But I wanted to
be evil when testing this out... */
if (nsock_event_cancel(nsp, ts->latest_readtcpev, 1) != 0) {
printf("Cancelled tcp event: %li\n", ts->latest_readtcpev);
}
if (nsock_event_cancel(nsp, ts->latest_readstdinev, 1) != 0) {
printf("Cancelled stdin event: %li\n", ts->latest_readstdinev);
}
} else if (status == NSE_STATUS_ERROR) {
if (nsi_checkssl(nsi)) {
printf("SSL %s failed: %s\n", nse_type2str(type), ERR_error_string(ERR_get_error(), NULL));
} else {
int err;
err = nse_errorcode(nse);
printf("%s failed: (%d) %s\n", nse_type2str(type), err, strerror(err));
}
}
return;
}