void handle_discover_response(pkt_t *pkt)
{
node_id_t origin = pkt->payload[PKT_RESPONSE_ORIGIN_OFFSET];
uint8_t seq = pkt->payload[PKT_RESPONSE_SEQ_OFFSET];
nrk_time_t delay;
uint8_t attempt;
if (!IS_VALID_NODE_ID(origin)) {
LOG("WARN: invalid origin in response: ");
LOGP("%d\r\n", origin);
return;
}
LOG("response: orig "); LOGP("%u", origin);
LOGA(" seq "); LOGP("%u", seq);
LOGA(" src "); LOGP("%u", pkt->src);
LOGA(": ");
nrk_time_get(&last_activity);
if (origin == this_node_id) {
LOGA("reached origin\r\n");
add_path_to_graph(pkt);
print_graph(&network);
} else { /* we're not the destination: forward to gateway */
attempt = 0;
do {
forward_response(pkt, attempt);
choose_delay(&delay, &discover_req_delay);
nrk_wait(delay);
} while (++attempt < discover_send_attempts);
}
}
int forward_request(int client_fd, Request *request, Response *response) {
rio_t server_rio;
int server_fd;
char hostname[MAXLINE];
int port = 80;
port = extract_port_number(request->request_str);
extract_hostname(request->host_str, hostname);
#ifdef DEBUG
printf("hostname:%s\n", hostname);
printf("port:%d\n", port);
#endif
if ((server_fd = open_clientfd(hostname, port)) < 0) {
fprintf(stderr, "Warning connection refused !\n");
return -1;
}
rio_readinitb(&server_rio, server_fd);
#ifdef DEBUG
printf("request_str:%s", request->request_str);
#endif
rio_writen(server_fd, request->request_str, strlen(request->request_str));
rio_writen(server_fd, request->host_str, strlen(request->host_str));
rio_writen(server_fd, "\r\n", strlen("\r\n"));
forward_response(client_fd, server_fd, response);
Close(server_fd);
return 1;
}
/*proxy executation routine*/
int run_proxy(int connfd)
{
char hostname[MAXLINE];
char req_buf[MAX_OBJECT_SIZE];
char resp_buf[MAX_OBJECT_SIZE];
char uri[MAXLINE];
int port = 80;
int clientfd;
rio_t rio;
memset(hostname, 0, MAXLINE);
memset(req_buf, 0, MAX_OBJECT_SIZE);
memset(resp_buf, 0 , MAX_OBJECT_SIZE);
/* read request */
if(read_request(connfd, req_buf, uri, hostname, &port) < 0) {
//printf("close fd: %d, tid: %d\n", connfd, gettid());
Close(connfd);
return -1;
}
/* open connection to server */
if((clientfd = open_clientfd_r(hostname, port)) < 0)
{
//printf("connfd: %d, clientfd: %d, host: %s, tid: %d\n", connfd, clientfd, hostname, gettid());
printf("Open_clientfd error\n");
fprintf(stderr, "Error: connection refused: %s !\n", hostname);
Close(connfd);
return -1;
}
if(forward_request(&rio, req_buf, clientfd) < 0)
{
printf("forward_request error\n");
fprintf(stderr, "Error: Send request to server failed !\n");
Close(clientfd);
Close(connfd);
return -1;
}
if(forward_response(&rio, uri, resp_buf, connfd) < 0)
{
printf("forward_response\n");
fprintf(stderr, "Error: Send response to client failed !\n");
Close(clientfd);
Close(connfd);
return -1;
}
Close(clientfd);
Close(connfd);
return 0;
}
/*
* forward_request - send request from client to its destination server
*/
int forward_request(int client_fd, Request *request, Response *response) {
rio_t server_rio;
int server_fd;
char hostname[MAXLINE];
int port = 80;
extract_hostname_port(request->host_str, hostname, &port);
#ifdef DEBUG
printf("extract hostname:%s\n", hostname);
printf("extract port:%d\n", port);
#endif
if ((server_fd = open_clientfd(hostname, port)) < 0) {
sleep(2);
if ((server_fd = open_clientfd(hostname, port)) < 0) {
sleep(2);
if ((server_fd = open_clientfd(hostname, port)) < 0) {
proxy_error(strcat(hostname, " connection refuesed"));
close(server_fd);
return -1;
}
}
}
rio_readinitb(&server_rio, server_fd);
#ifdef DEBUG
printf("request_str:%s", request->request_str);
#endif
rio_writen(server_fd, request->request_str, strlen(request->request_str));
rio_writen(server_fd, request->host_str, strlen(request->host_str));
/* send cookie to server */
if (request->cookie_size != 0) {
rio_writen(server_fd, request->cookie, request->cookie_size);
}
/* send user-agent info to server */
if (request->user_agent_size != 0) {
rio_writen(server_fd, request->user_agent, request->user_agent_size);
}
rio_writen(server_fd, "\r\n", strlen("\r\n"));
if (forward_response(client_fd, server_fd, response) < 0) {
close(server_fd);
return -1;
}
close(server_fd);
return 1;
}
