/* fcntl */
int setsockopt(int sockfd, int level, int optname,
const void *optval, socklen_t optlen) {
int ret;
struct sock *sk;
socket_idesc_check(sockfd, sk);
if ((optval == NULL) || (optlen < 0)) {
return SET_ERRNO(EINVAL);
}
ret = ksetsockopt(sk, level, optname, optval, optlen);
if (ret < 0) {
return SET_ERRNO(-ret);
}
return 0;
}
/*This function creates a server process listening at the socket rport*/
int tcp_serv (void *arg)
{
struct sockaddr_in relay_srv_addr, stClntAddr;
int iAddrlen;
int on = 1;
ksocket_t sockfd_chld;
/* kernel thread name*/
sprintf(current->comm, "krelaysrv");
allow_signal(SIGKILL);
allow_signal(SIGTERM);
/*Create socket*/
if ((SockfdRelayClntSide = ksocket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
printk(KERN_ERR "Can't create socket.");
memset(&relay_srv_addr, 0, sizeof(relay_srv_addr));
relay_srv_addr.sin_family = AF_INET;
relay_srv_addr.sin_addr.s_addr = INADDR_ANY;
relay_srv_addr.sin_port = htons(StrToInt((char *)(pListenPort->data)));
if (ksetsockopt(SockfdRelayClntSide, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
{
printk(KERN_ERR "ksetsockopt(SO_REUSEADDR) failed\n");
}
/*Bind our local addrress so that client can send*/
if (kbind(SockfdRelayClntSide, (struct sockaddr *) &relay_srv_addr, sizeof(relay_srv_addr)) < 0)
{
printk(KERN_ERR "Can't bind local address\n");
return -1;
}
/* Make socket a listening socket */
if (klisten(SockfdRelayClntSide, 15) < 0)
{
printk(KERN_ERR "Call to listen failed.");
return -1;
}
/* Loop infinitely to accept and service connections */
while ( 1 )
{
/* Wait for connection */
if ( (sockfd_chld = kaccept(SockfdRelayClntSide, (struct sockaddr *) &stClntAddr, &iAddrlen)) < 0 )
{
printk(KERN_ERR "Error calling accept()");
return -1;
}
ChldThdPid[iThdCnt] = kernel_thread ((void *)RedirectProc, sockfd_chld, 0);
if (iThdCnt == (MAX_THD_IDX - 1))
iThdCnt = 0;
else
iThdCnt++;
//Sleep for some time
// mdelay (1);
}
(void)kclose (SockfdRelayClntSide);
return -1; /* We shouldn't get here */
}
/*this handles the redirection from client to real server*/
void RedirectProc (ksocket_t SockfdFromClnt)
{
struct sockaddr_in RealSrvAddr, stRelayAddr, stTemp1, stTemp;
ksocket_t SockfdRelay;
int on = 1;
char frwd_buffer[BUFFER_SIZE];
int iRBytes, iWBytes, iNWritten, cur, iLen;
unsigned short iClntPort, server_port = 0;
/* kernel thread name*/
sprintf(current->comm, "krelaychld");
allow_signal(SIGKILL);
allow_signal(SIGTERM);
//Copy the values for later use
if ((kgetpeername(SockfdFromClnt, (struct sockaddr *)&stTemp1, &iLen)) < 0)
{
printk (KERN_ERR "kgetpeername() failed\n");
}
iClntPort = stTemp1.sin_port;
/*Create socket*/
if ((SockfdRelay = ksocket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
printk(KERN_ERR "Can't create socket.");
memset(&stRelayAddr, 0, sizeof(stRelayAddr));
stRelayAddr.sin_family = AF_INET;
stRelayAddr.sin_addr.s_addr = INADDR_ANY;
stRelayAddr.sin_port = htons(0);
if (ksetsockopt(SockfdRelay, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
{
printk(KERN_ERR "ksetsockopt(SO_REUSEADDR) failed\n");
}
/*Bind our local addrress so that client can send*/
if (kbind(SockfdRelay, (struct sockaddr *) &stRelayAddr, sizeof(stRelayAddr)) < 0)
{
printk(KERN_ERR "Can't bind local address\n");
return;
}
//printk (KERN_ERR "Relay eph port= %d", ntohs(stRelayAddr.sin_port));
//Need to analyze this part@@@@@@@@@
//Store the relay server ephemeral port (Y) in NAT
if (kgetsockname(SockfdRelay, (struct sockaddr *) &stTemp, &iLen) < 0)
{
printk(KERN_ERR "kgetpeername failed\n");
}
// printk (KERN_ERR "Relay eph port= %d", ntohs(stTemp.sin_port));
for (cur = 0; cur <= MAX_TBL_IDX -1; cur++)
{
if (stTbl[cur].iClntPort == iClntPort)
{
stTbl[cur].iRelayPort = stTemp.sin_port;
server_port = stTbl[cur].iClntSrvPort;
break;
}
}
if (0 == server_port)
{
printk (KERN_ERR "Error..Cannot get the server port\n");
kclose (SockfdRelay);
return;
}
//Connect to the Real server
RealSrvAddr.sin_family = PF_INET;
RealSrvAddr.sin_addr.s_addr = (inet_addr(DEF_REAL_IP));
RealSrvAddr.sin_port = server_port;
if (kconnect(SockfdRelay, (struct sockaddr *) &RealSrvAddr, sizeof(RealSrvAddr)) < 0)
{
printk(KERN_ERR "Connect failed\n");
}
for (;;)
{
//printk (KERN_ERR "trying krecv from client\n");
/* Read from Client and send to Real Server */
iRBytes = krecv(SockfdFromClnt, &frwd_buffer, BUFFER_SIZE - 8, MSG_DONTWAIT);
//printk (KERN_ERR "tried krecv = %d\n", iRBytes);
//Client Socket is closed
if (iRBytes == 0)
break;
iWBytes = 0;
iNWritten = 0;
//If some data is there to be read
while (iWBytes != iRBytes && iRBytes > 0)
{
if ((iNWritten = ksend(SockfdRelay, &frwd_buffer + iWBytes , iRBytes-iWBytes, 0)) < 0)
{
printk (KERN_ERR "error occured while writing to receiver\n");
return;
}
iWBytes += iNWritten;
}
//printk (KERN_ERR "trying krecv from real server\n");
/* Read from Real server and write to client */
iRBytes = krecv(SockfdRelay, &frwd_buffer, BUFFER_SIZE - 8, MSG_DONTWAIT);
//printk (KERN_ERR "tried krecv1 = %d\n", iRBytes);
if (iRBytes == 0)
break;
iWBytes = 0;
iNWritten = 0;
while (iWBytes != iRBytes && iRBytes > 0)
{
if ((iNWritten = ksend(SockfdFromClnt, &frwd_buffer + iWBytes , iRBytes-iWBytes, 0)) < 0)
{
printk (KERN_ERR "error occured while writing to receiver\n");
return;
}
iWBytes += iNWritten;
}
//Sleep for sometime
// mdelay (1);
}
(void)kclose(SockfdRelay);
(void)kclose(SockfdFromClnt);
}
int SenderExp(hash *data_table, char *code)
{
ksocket_t cli_fd;
struct sockaddr_in srv_addr;
char *temp = 0x00;
int addr_len;
int opt = 1;
int i, r;
//listNode *pCur = 0;
#ifdef KSOCKET_ADDR_SAFE
mm_segment_t oldfs;
old_fs = get_fs();
set_fs(KERNEL_DS);
#endif
memset(&srv_addr, 0, sizeof(srv_addr));
srv_addr.sin_family = AF_INET;
srv_addr.sin_port = htons(SERVER_PORT);
srv_addr.sin_addr.s_addr = inet_addr(SERVER_ADDR);
addr_len = sizeof(struct sockaddr_in);
cli_fd = ksocket(AF_INET, SOCK_STREAM, 0);
if(cli_fd == 0)
{
return -1;
}
if(ksetsockopt(cli_fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof(opt)) < 0)
{
return -1;
}
if(kconnect(cli_fd, (struct sockaddr *)&srv_addr, addr_len) < 0)
{
return -1;
}
temp = inet_ntoa(&srv_addr.sin_addr);
printk("connected to : %s %d\n", temp, ntohs(srv_addr.sin_port));
kfree(temp);
if(strcmp(code, "exp") == 0)
{
r = SendHeader(cli_fd, code, data_table);
printk("hdr : %d\n", r);
for(i=0; i<HASH_SIZE; i++)
{
listNode *cur = data_table->item[i].head;
while(cur)
{
r = SendExpData(cli_fd, &(cur->data));
if (r <= 0) continue;
printk("payload : %d\n", r);
PrintData(&(cur->data));
cur = cur->next;
}
}
}
kclose(cli_fd);
#ifdef KSOCKET_ADDR_SAFE
set_fs(old_fs);
#endif
return 0;
}
