/* Initialize RSRR socket */
void
rsrr_init(void)
{
int servlen;
struct sockaddr_un serv_addr;
if ((rsrr_socket = socket(AF_UNIX, SOCK_DGRAM, 0)) < 0)
dolog(LOG_ERR, errno, "Can't create RSRR socket");
unlink(RSRR_SERV_PATH);
bzero((char *) &serv_addr, sizeof(serv_addr));
serv_addr.sun_family = AF_UNIX;
strcpy(serv_addr.sun_path, RSRR_SERV_PATH);
#ifdef HAVE_SA_LEN
servlen = offsetof(struct sockaddr_un, sun_path) +
strlen(serv_addr.sun_path);
serv_addr.sun_len = servlen;
#else
servlen = sizeof(serv_addr.sun_family) + strlen(serv_addr.sun_path);
#endif
if (bind(rsrr_socket, (struct sockaddr *) &serv_addr, servlen) < 0)
dolog(LOG_ERR, errno, "Can't bind RSRR socket");
if (register_input_handler(rsrr_socket, rsrr_read) < 0)
dolog(LOG_WARNING, 0, "Couldn't register RSRR as an input handler");
}
/*
* Open and initialize the igmp socket, and fill in the non-changing
* IP header fields in the output packet buffer.
*/
void init_igmp(void)
{
struct ip *ip;
char *router_alert;
igmp_recv_buf = calloc(1, RECV_BUF_SIZE);
igmp_send_buf = calloc(1, SEND_BUF_SIZE);
if (!igmp_recv_buf || !igmp_send_buf)
logit(LOG_ERR, 0, "Ran out of memory in init_igmp()");
if ((igmp_socket = socket(AF_INET, SOCK_RAW, IPPROTO_IGMP)) < 0)
logit(LOG_ERR, errno, "Failed creating IGMP socket in init_igmp()");
k_hdr_include(igmp_socket, TRUE); /* include IP header when sending */
k_set_sndbuf(igmp_socket, SO_SEND_BUF_SIZE_MAX,
SO_SEND_BUF_SIZE_MIN); /* lots of output buffering */
k_set_rcvbuf(igmp_socket, SO_RECV_BUF_SIZE_MAX,
SO_RECV_BUF_SIZE_MIN); /* lots of input buffering */
k_set_ttl(igmp_socket, MINTTL); /* restrict multicasts to one hop */
k_set_loop(igmp_socket, FALSE); /* disable multicast loopback */
ip = (struct ip *)igmp_send_buf;
memset(ip, 0, IP_IGMP_HEADER_LEN);
ip->ip_v = IPVERSION;
ip->ip_hl = IP_IGMP_HEADER_LEN >> 2;
ip->ip_tos = 0xc0; /* Internet Control */
ip->ip_id = 0; /* let kernel fill in */
ip->ip_off = 0;
ip->ip_ttl = MAXTTL; /* applies to unicasts only */
ip->ip_p = IPPROTO_IGMP;
ip->ip_sum = 0; /* let kernel fill in */
/* Enable RFC2113 IP Router Alert. Per spec this is required to
* force certain routers/switches to inspect this frame. */
router_alert = igmp_send_buf + sizeof(struct ip);
router_alert[0] = IPOPT_RA;
router_alert[1] = 4;
router_alert[2] = 0;
router_alert[3] = 0;
/* Everywhere in the daemon we use network-byte-order */
allhosts_group = htonl(INADDR_ALLHOSTS_GROUP);
allrouters_group = htonl(INADDR_ALLRTRS_GROUP);
allreports_group = htonl(INADDR_ALLRPTS_GROUP);
if (register_input_handler(igmp_socket, igmp_read) < 0)
logit(LOG_ERR, 0, "Failed registering igmp_read() as an input handler in init_igmp()");
}
void init_pim()
{//初始化
allpimrouters=htonl(INADDR_ALLPIM_ROUTERS_GROUP);
pim_receive_buf=malloc(64*1024);
pim_send_buf=malloc(64*1024);
if((pimsocket=socket(AF_INET,SOCK_RAW,IPPROTO_PIM))<0)
log(LOG_ERR,"igmp socket create failed!");
k_hdr_include(pimsocket,1);//发送的时候包括IP头部
k_set_rcvbuf(pimsocket,256*1024,48*1024);
k_set_ttl(pimsocket,1);//设置IGMP报文的TTL为1
k_set_loop(pimsocket,0);//决定发送的多点广播包不应该被回送
if(register_input_handler(pimsocket,pim_input)<0)
log(LOG_ERR,"Couldn't register pim_input as an input handler");
}
/*
* Setup raw kernel socket for PIM protocol and send/receive buffers.
*/
void init_pim(void)
{
struct ip *ip;
/* Setup the PIM raw socket */
if ((pim_socket = socket(AF_INET, SOCK_RAW, IPPROTO_PIM)) < 0)
logit(LOG_ERR, errno, "Failed creating PIM socket");
k_hdr_include(pim_socket, TRUE); /* include IP header when sending */
k_set_sndbuf(pim_socket, SO_SEND_BUF_SIZE_MAX,
SO_SEND_BUF_SIZE_MIN); /* lots of output buffering */
k_set_rcvbuf(pim_socket, SO_RECV_BUF_SIZE_MAX,
SO_RECV_BUF_SIZE_MIN); /* lots of input buffering */
k_set_ttl(pim_socket, MINTTL); /* restrict multicasts to one hop */
k_set_loop(pim_socket, FALSE); /* disable multicast loopback */
allpimrouters_group = htonl(INADDR_ALL_PIM_ROUTERS);
pim_recv_buf = calloc(1, RECV_BUF_SIZE);
pim_send_buf = calloc(1, SEND_BUF_SIZE);
if (!pim_recv_buf || !pim_send_buf)
logit(LOG_ERR, 0, "Ran out of memory in init_pim()");
/* One time setup in the buffers */
ip = (struct ip *)pim_send_buf;
memset(ip, 0, sizeof(*ip));
ip->ip_v = IPVERSION;
ip->ip_hl = (sizeof(struct ip) >> 2);
ip->ip_tos = 0; /* TODO: setup?? */
ip->ip_id = 0; /* Make sure to update ID field, maybe fragmenting below */
ip->ip_off = 0;
ip->ip_p = IPPROTO_PIM;
ip->ip_sum = 0; /* let kernel fill in */
if (register_input_handler(pim_socket, pim_read) < 0)
logit(LOG_ERR, 0, "Failed registering pim_read() as an input handler");
/* Initialize the building Join/Prune messages working area */
build_jp_message_pool = (build_jp_message_t *)NULL;
build_jp_message_pool_counter = 0;
}
