int handle_request_tunnel_frame(protocol_frame *rcvd_frame)
{
char *hostname = NULL;
tunnel *tun;
int port = -1;
int sockfd = 0;
uint16_t tunnel_id;
if(client_mode)
{
log_printf(L_WARNING, "Got tunnel request frame from friend #%d when in client mode\n", rcvd_frame->friendnumber);
return -1;
}
port = rcvd_frame->connid;
hostname = calloc(1, rcvd_frame->data_length + 1);
if(!hostname)
{
log_printf(L_ERROR, "Could not allocate memory for tunnel request hostname\n");
return -1;
}
strncpy(hostname, rcvd_frame->data, rcvd_frame->data_length);
hostname[rcvd_frame->data_length] = '\0';
log_printf(L_INFO, "Got a request to forward data from %s:%d\n", hostname, port);
tunnel_id = get_random_tunnel_id();
log_printf(L_DEBUG, "Tunnel ID: %d\n", tunnel_id);
sockfd = get_client_socket(hostname, port);
if(sockfd >= 0)
{
tun = tunnel_create(sockfd, tunnel_id, rcvd_frame->friendnumber);
if(tun)
{
FD_SET(sockfd, &master_server_fds);
update_select_nfds(sockfd);
log_printf(L_DEBUG, "Created tunnel, yay!\n");
send_tunnel_ack_frame(tun);
}
else
{
log_printf(L_ERROR, "Couldn't allocate memory for tunnel\n");
close(sockfd);
}
}
else
{
log_printf(L_WARNING, "Could not connect to %s:%d\n", hostname, port);
/* TODO send reject */
}
free(hostname);
return 0;
}
/* Constructor. Returns NULL on failure. */
tunnel *tunnel_create(int sockfd, int connid, uint32_t friendnumber)
{
tunnel *t = NULL;
t = calloc(1, sizeof(tunnel));
if(!t)
{
return NULL;
}
t->sockfd = sockfd;
t->connid = connid;
t->friendnumber = friendnumber;
log_printf(L_INFO, "Created a new tunnel object connid=%d sockfd=%d\n", connid, sockfd);
update_select_nfds(t->sockfd);
HASH_ADD_INT( by_id, connid, t );
return t;
}
int handle_request_tunnel_frame(protocol_frame *rcvd_frame)
{
char *hostname = NULL;
tunnel *tun;
int port = -1;
int sockfd = 0;
uint16_t tunnel_id;
if(client_mode)
{
log_printf(L_WARNING, "Got tunnel request frame from friend #%d when in client mode\n", rcvd_frame->friendnumber);
return -1;
}
port = rcvd_frame->connid;
hostname = calloc(1, rcvd_frame->data_length + 1);
if(!hostname)
{
log_printf(L_ERROR, "Could not allocate memory for tunnel request hostname\n");
return -1;
}
strncpy(hostname, (char *)rcvd_frame->data, rcvd_frame->data_length);
hostname[rcvd_frame->data_length] = '\0';
log_printf(L_INFO, "Got a request to forward data from %s:%d\n", hostname, port);
// check rules
if (rules_policy == VALIDATE && nrules > 0 ) {
rule temp_rule, *found = NULL;
temp_rule.host = hostname;
temp_rule.port = port;
LL_SEARCH(rules, found, &temp_rule, rule_cmp);
if(!found)
{
log_printf(L_WARNING, "Rejected, request not in rules\n");
if(hostname)
{
free(hostname);
}
return -1;
}
} else if (rules_policy != NONE) {
log_printf(L_WARNING, "Filter option active but no allowed host/port. All requests will be dropped.\n");
if(hostname)
{
free(hostname);
}
return -1;
}
tunnel_id = get_random_tunnel_id();
log_printf(L_DEBUG, "Tunnel ID: %d\n", tunnel_id);
sockfd = get_client_socket(hostname, port);
if(sockfd >= 0)
{
tun = tunnel_create(sockfd, tunnel_id, rcvd_frame->friendnumber);
if(tun)
{
FD_SET(sockfd, &master_server_fds);
update_select_nfds(sockfd);
log_printf(L_DEBUG, "Created tunnel, yay!\n");
send_tunnel_ack_frame(tun);
}
else
{
log_printf(L_ERROR, "Couldn't allocate memory for tunnel\n");
close(sockfd);
}
}
else
{
log_printf(L_WARNING, "Could not connect to %s:%d\n", hostname, port);
/* TODO send reject */
}
free(hostname);
return 0;
}