void
rcv_ip()
{
statistics.ipInReceives++;
if (IP_P->destination != tcpipcfg.c_myip) {
// Must Check for Multicast Address to use IP MULTICASTING
// And Broadcast
statistics.ipInAddrErrors++;
return;
}
// We reject IP Packet with IP options(hdr_len != 5),
// Other version (ver != 4)
if (IP_P->hdr_len_ver != 0x45) {
statistics.ipInUnknownProtos++;
return;
}
if (checksum(&(IP_P->hdr_len_ver), IP_HLEN)) {
//rprintf("[IP CHKSUM]");
statistics.ipInHdrErrors++;
return;
}
// There must be some measure against the fragmented Packet
// We should do packet Reassembly
//
// We dont even check the frag-bits, some host send IP packet with
// MAY-FRAG bit set for small packets
//
// if (IP_P->frag_offset) {
// // (IP_P->frag_offset * 8) is the real offset
// statistics.ipReasmReqds++;
// statistics.ipReasmFails++;
// // But now We don't
// return;
// }
//rprintf("[PROTO = %d]", IP_P->transport);
switch (IP_P->transport) {
case IPPROTO_ICMP : rcv_icmp(); break;
case IPPROTO_TCP :
//rprintf("[T");
rcv_tcp();
//rprintf("T]");
break;
case IPPROTO_UDP : rcv_udp(); break;
case IPPROTO_IP : // rcv_raw_ip(); break
default : // We support no other protocols
statistics.ipInUnknownProtos++; break;
}
}
int main(int argc, char **argv)
{
struct udp_state *ud;
struct katcp_dispatch *d;
unsigned int result;
struct timeval now;
fd_set fsr;
char *ip_addr = NULL;
uint32_t address, length;
int i, j, c, pos;
int wait, nooftries;
int port = 0;
int rw_flag = 0;
i = j = 1;
pos = 0;
wait = 0;
nooftries = 10;
while (i < argc) {
if (argv[i][0] == '-') {
c = argv[i][j];
switch (c) {
case '\0':
j = 1;
i++;
break;
case '-' :
j++;
break;
case 'h' :
fprintf(stderr, "usage: %s -R [-i ipaddress] [-p port] address length\n", argv[0]);
return 0;
break;
case 'i' :
j++;
if(argv[i][j] == '\0'){
j = 0;
i++;
}
if(i >= argc){
fprintf(stderr, "%s: option -%c requires a parameter\n", argv[0], c);
}
ip_addr = argv[i] + j;
i++;
j = 1;
break;
case 'p' :
j++;
if(argv[i][j] == '\0'){
j = 0;
i++;
}
if(i >= argc){
fprintf(stderr, "%s: option -%c requires a parameter\n", argv[0], c);
}
port = atoi(argv[i] + j);
#if DEBUG
fprintf(stderr, "port number is %d\n", port);
#endif
i++;
j = 1;
break;
case 'R' :
rw_flag = 1;
i++;
break;
default:
fprintf(stderr, "%s: unknown option -%c\n", argv[0], argv[i][j]);
return 2;
}
} else {
pos = i;
i = argc;
}
}
d = setup_katcp(STDOUT_FILENO);
if(d == NULL){
fprintf(stderr, "setup katcp failed\n");
return EX_OSERR;
}
ud = create_udp(d);
if(ud == NULL){
fprintf(stderr, "create udp failed\n");
log_message_katcp(d, KATCP_LEVEL_ERROR, DMON_MODULE_NAME, "unable to allocate local udp state");
write_katcp(d);
return EX_OSERR;
}
#if 0
if(connect_udp(d, ud, port) < 0){
fprintf(stderr, "connect udp failed\n");
log_message_katcp(d, KATCP_LEVEL_ERROR, DMON_MODULE_NAME, "unable to bind udp");
return EX_OSERR;
}
#endif
ud->u_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
if(ud->u_fd < 0){
fprintf(stderr, "unable to create udp socket:\n ");
log_message_katcp(d, KATCP_LEVEL_ERROR, DMON_MODULE_NAME, "unable to create udp socket: %s", strerror(errno));
return -1;
}
ud->u_rw = rw_flag;
for(;;){
FD_ZERO(&fsr);
FD_SET(ud->u_fd, &fsr);
address = strtol(argv[pos], NULL, 16);
length = strtol(argv[pos + 1], NULL, 16);
#if DEBUG
printf("pos:%d, address[%x] and length[%x]\n", pos, address, length);
#endif
send_udp(d, ud, ip_addr, port, address, length);
now.tv_sec = 10;
now.tv_usec = 0;
result = select(ud->u_fd + 1, &fsr, NULL, NULL, &now);
if(result == 0){
/* Resend again after timeout */
printf("Resending udp again\n");
send_udp(d, ud, ip_addr, port, address, length);
}
for(wait = 0; wait < nooftries; wait++){
if(FD_ISSET(ud->u_fd, &fsr)){
result = rcv_udp(d, ud);
if(!result){
return EX_OK;
}
}
}
return EX_OSERR;
}
destroy_udp(d, ud);
shutdown_katcp(d);
return EX_OK;
}
