void NS_CLASS recvAODVUUPacket(Packet * p)
{
int len, i, ttl = 0;
u_int32_t src, dst;
struct hdr_cmn *ch = HDR_CMN(p);
struct hdr_ip *ih = HDR_IP(p);
hdr_aodvuu *ah = HDR_AODVUU(p);
src = ih->saddr();
dst = ih->daddr();
len = ch->size() - IP_HDR_LEN;
ttl = ih->ttl();
AODV_msg *aodv_msg = (AODV_msg *) recv_buf;
/* Only handle AODVUU packets */
assert(ch->ptype() == PT_AODVUU);
/* Only process incoming packets */
assert(ch->direction() == hdr_cmn::UP);
/* Copy message to receive buffer */
memcpy(recv_buf, ah, RECV_BUF_SIZE);
/* Drop or deallocate packet, depending on the TTL */
/* if (ttl == 1) */
/* drop(p, DROP_RTR_TTL); */
/* else */
Packet::free(p);
/* Ignore messages generated locally */
for (i = 0; i < MAX_NR_INTERFACES; i++)
if (this_host.devs[i].enabled &&
memcmp(&src, &this_host.devs[i].ipaddr, sizeof(u_int32_t)) == 0)
return;
aodv_socket_process_packet(aodv_msg, len, src, dst, ttl, NS_IFINDEX);
}
void NS_CLASS recvAODVUUPacket(Packet * p)
{
int len, i, ttl = 0;
struct in_addr src, dst;
struct hdr_cmn *ch = HDR_CMN(p);
struct hdr_ip *ih = HDR_IP(p);
hdr_aodvuu *ah = HDR_AODVUU(p);
src.s_addr = ih->saddr();
dst.s_addr = ih->daddr();
len = ch->size() - IP_HDR_LEN;
ttl = ih->ttl();
AODV_msg *aodv_msg = (AODV_msg *) recv_buf;
/* Only handle AODVUU packets */
assert(ch->ptype() == PT_AODVUU);
/* Only process incoming packets */
assert(ch->direction() == hdr_cmn::UP);
/* Copy message to receive buffer */
memcpy(recv_buf, ah, RECV_BUF_SIZE);
/* Deallocate packet, we have the information we need... */
Packet::free(p);
/* Ignore messages generated locally */
for (i = 0; i < MAX_NR_INTERFACES; i++)
if (this_host.devs[i].enabled &&
memcmp(&src, &this_host.devs[i].ipaddr,
sizeof(struct in_addr)) == 0)
return;
aodv_socket_process_packet(aodv_msg, len, src, dst, ttl, NS_IFINDEX);
}
static void aodv_socket_read(int fd)
{
u_int32_t src, dst;
int i, len, ttl = 0;
AODV_msg *aodv_msg;
#ifdef USE_IW_SPY
char ifname[IFNAMSIZ];
#endif /* USE_IW_SPY */
#ifdef RAW_SOCKET
int iph_len;
struct iphdr *iph;
struct udphdr *udph;
len = recvfrom(fd, recv_buf, RECV_BUF_SIZE, 0, NULL, NULL);
if (len < 0 || len < IPHDR_SIZE) {
log(LOG_WARNING, 0, __FUNCTION__, "receive ERROR!");
return;
}
/* Parse the IP header */
iph = (struct iphdr *) recv_buf;
src = ntohl(iph->saddr);
dst = ntohl(iph->daddr);
ttl = iph->ttl;
iph_len = iph->ihl << 2;
udph = (struct udphdr *) (recv_buf + iph_len);
if (ntohs(udph->dest) != AODV_PORT && ntohs(udph->source) != AODV_PORT)
return;
/* Ignore messages generated locally */
for (i = 0; i < MAX_NR_INTERFACES; i++)
if (this_host.devs[i].enabled &&
memcmp(&src, &this_host.devs[i].ipaddr, sizeof(u_int32_t)) == 0)
return;
aodv_msg = (AODV_msg *) (recv_buf + iph_len + sizeof(struct udphdr));
len = ntohs(udph->len) - sizeof(struct udphdr);
#else
struct sockaddr_in src_addr;
struct msghdr msg;
union {
struct cmsghdr cm;
char control[CMSG_SPACE(sizeof(int)) +
CMSG_SPACE(sizeof(struct in_pktinfo))];
} control_union;
struct cmsghdr *cmsg;
struct in_pktinfo pktinfo;
int sockaddr_len = sizeof(struct sockaddr_in);
msg.msg_name = NULL;
msg.msg_namelen = 0;
msg.msg_iov = NULL;
msg.msg_iovlen = 0;
msg.msg_control = control_union.control;
msg.msg_controllen = sizeof(control_union.control);
/* Get the information control message first */
if ((len = recvmsg(fd, &msg, MSG_PEEK)) < 0) {
log(LOG_WARNING, 0, __FUNCTION__, "recvmsg ERROR!");
return;
}
/* Read the data payload (i.e. AODV msg) */
len = recvfrom(fd, recv_buf, RECV_BUF_SIZE, 0,
(struct sockaddr *) &src_addr, &sockaddr_len);
if (len < 0) {
log(LOG_WARNING, 0, __FUNCTION__, "receive ERROR!");
return;
}
aodv_msg = (AODV_msg *) (recv_buf);
src = ntohl(src_addr.sin_addr.s_addr);
/* Ignore messages generated locally */
for (i = 0; i < MAX_NR_INTERFACES; i++)
if (this_host.devs[i].enabled &&
memcmp(&src, &this_host.devs[i].ipaddr, sizeof(u_int32_t)) == 0)
return;
/* Get the TTL and pktinfo struct (destination address) from the
control messages... For some reason the correct use of the
CMSG(3) macros using CMSG_NXTHDR does not work across different
Red Hat versions (6.2 vs 7.2), but this code seem to work: */
cmsg = CMSG_FIRSTHDR(&msg);
for (i = 0; i < 2; i++) {
if (cmsg->cmsg_level == SOL_IP && cmsg->cmsg_type == IP_TTL) {
memcpy(&ttl, CMSG_DATA(cmsg), sizeof(int));
cmsg = (void *) cmsg + CMSG_SPACE(sizeof(int));
} else if (cmsg->cmsg_level == SOL_IP && cmsg->cmsg_type == IP_PKTINFO) {
memcpy(&pktinfo, CMSG_DATA(cmsg), sizeof(struct in_pktinfo));
cmsg = (void *) cmsg + CMSG_SPACE(sizeof(struct in_pktinfo));
}
}
dst = ntohl(pktinfo.ipi_addr.s_addr);
#endif /* RAW_SOCKET */
#ifdef USE_IW_SPY
if (spy_addrs &&
link_qual_get_from_ip(src, if_indextoname(pktinfo.ipi_ifindex,
ifname)) <
hello_qual_threshold)
return;
#endif /* USE_IW_SPY */
aodv_socket_process_packet(aodv_msg, len, src, dst, ttl,
pktinfo.ipi_ifindex);
}
static void aodv_socket_read(int fd)
{
struct in_addr src, dst;
int i, len, ttl = -1;
AODV_msg *aodv_msg;
struct dev_info *dev;
struct msghdr msgh;
struct cmsghdr *cmsg;
struct iovec iov;
char ctrlbuf[CMSG_SPACE(sizeof(int)) +
CMSG_SPACE(sizeof(struct in_pktinfo))];
struct sockaddr_in src_addr;
iov.iov_base = recv_buf;
iov.iov_len = RECV_BUF_SIZE;
msgh.msg_name = &src_addr;
msgh.msg_namelen = sizeof(src_addr);
msgh.msg_iov = &iov;
msgh.msg_iovlen = 1;
msgh.msg_control = ctrlbuf;
msgh.msg_controllen = sizeof(ctrlbuf);
len = recvmsg(fd, &msgh, 0);
if (len < 0) {
alog(LOG_WARNING, 0, __FUNCTION__, "receive ERROR len=%d!", len);
return;
}
src.s_addr = src_addr.sin_addr.s_addr;
/* Get the ttl and destination address from the control message */
for (cmsg = CMSG_FIRSTHDR(&msgh); cmsg != NULL;
cmsg = CMSG_NXTHDR_FIX(&msgh, cmsg)) {
if (cmsg->cmsg_level == SOL_IP) {
switch (cmsg->cmsg_type) {
case IP_TTL:
ttl = *(CMSG_DATA(cmsg));
break;
case IP_PKTINFO:
dst.s_addr =
((struct in_pktinfo *) CMSG_DATA(cmsg))->ipi_addr.s_addr;
}
}
}
if (ttl < 0) {
DEBUG(LOG_DEBUG, 0, "No TTL, packet ignored!");
return;
}
/* Ignore messages generated locally */
for (i = 0; i < MAX_NR_INTERFACES; i++)
if (this_host.devs[i].enabled &&
memcmp(&src, &this_host.devs[i].ipaddr,
sizeof(struct in_addr)) == 0)
return;
aodv_msg = (AODV_msg *) recv_buf;
dev = devfromsock(fd);
if (!dev) {
DEBUG(LOG_ERR, 0, "Could not get device info!\n");
return;
}
aodv_socket_process_packet(aodv_msg, len, src, dst, ttl, dev->ifindex);
}
