e132be6b43f2ca80837f061b1c4f34ff3d77f145
1 /* Misc util functions */
6 #include <sys/socket.h>
7 #include <netinet/in.h>
17 // format ipv4 addr as a dotted-quad; n chooses one of 4 static buffers
19 char *fmtaddr(in_addr_t addr
, int n
)
21 static char addrs
[4][16];
28 return strcpy(addrs
[n
], inet_ntoa(in
));
31 void *shared_malloc(unsigned int size
)
34 p
= mmap(NULL
, size
, PROT_READ
| PROT_WRITE
, MAP_SHARED
| MAP_ANONYMOUS
, 0, 0);
43 extern int cluster_sockfd
, tunfd
, udpfd
, controlfd
, daefd
, snoopfd
, ifrfd
, ifr6fd
, rand_fd
;
46 pid_t
fork_and_close()
55 if (config
->scheduler_fifo
)
57 struct sched_param params
= {0};
58 params
.sched_priority
= 0;
59 if (sched_setscheduler(0, SCHED_OTHER
, ¶ms
))
61 LOG(0, 0, 0, "Error setting scheduler to OTHER after fork: %s\n", strerror(errno
));
62 LOG(0, 0, 0, "This is probably really really bad.\n");
66 signal(SIGPIPE
, SIG_DFL
);
67 signal(SIGCHLD
, SIG_DFL
);
68 signal(SIGHUP
, SIG_DFL
);
69 signal(SIGUSR1
, SIG_DFL
);
70 signal(SIGQUIT
, SIG_DFL
);
71 signal(SIGKILL
, SIG_DFL
);
72 signal(SIGTERM
, SIG_DFL
);
75 if (clifd
!= -1) close(clifd
);
76 if (cluster_sockfd
!= -1) close(cluster_sockfd
);
77 if (tunfd
!= -1) close(tunfd
);
78 if (udpfd
!= -1) close(udpfd
);
79 if (controlfd
!= -1) close(controlfd
);
80 if (daefd
!= -1) close(daefd
);
81 if (snoopfd
!= -1) close(snoopfd
);
82 if (rand_fd
!= -1) close(rand_fd
);
83 if (epollfd
!= -1) close(epollfd
);
85 for (i
= 0; radfds
&& i
< RADIUS_FDS
; i
++)
89 for (i
= 0; i
< BGP_NUM_PEERS
; i
++)
90 if (bgp_peers
[i
].sock
!= -1)
91 close(bgp_peers
[i
].sock
);
97 ssize_t
recvfromto(int s
, void *buf
, size_t len
, int flags
,
98 struct sockaddr
*from
, socklen_t
*fromlen
, struct in_addr
*toaddr
)
102 struct cmsghdr
*cmsg
;
106 memset(&msg
, 0, sizeof(msg
));
108 msg
.msg_namelen
= *fromlen
;
116 msg
.msg_control
= cbuf
;
117 msg
.msg_controllen
= sizeof(cbuf
);
119 if ((r
= recvmsg(s
, &msg
, flags
)) < 0)
123 *fromlen
= msg
.msg_namelen
;
125 memset(toaddr
, 0, sizeof(*toaddr
));
126 for (cmsg
= CMSG_FIRSTHDR(&msg
); cmsg
; cmsg
= CMSG_NXTHDR(&msg
, cmsg
))
128 if (cmsg
->cmsg_level
== SOL_IP
&& cmsg
->cmsg_type
== IP_PKTINFO
)
130 struct in_pktinfo
*i
= (struct in_pktinfo
*) CMSG_DATA(cmsg
);
131 memcpy(toaddr
, &i
->ipi_addr
, sizeof(*toaddr
));
139 ssize_t
sendtofrom(int s
, void const *buf
, size_t len
, int flags
,
140 struct sockaddr
const *to
, socklen_t tolen
, struct in_addr
const *from
)
143 struct cmsghdr
*cmsg
;
145 struct in_pktinfo pktinfo
;
146 char cbuf
[CMSG_SPACE(sizeof(pktinfo
))];
148 memset(&msg
, 0, sizeof(msg
));
149 msg
.msg_name
= (struct sockaddr
*) to
;
150 msg
.msg_namelen
= tolen
;
152 vec
.iov_base
= (void *) buf
;
158 msg
.msg_control
= cbuf
;
159 msg
.msg_controllen
= sizeof(cbuf
);
161 cmsg
= CMSG_FIRSTHDR(&msg
);
162 cmsg
->cmsg_level
= SOL_IP
;
163 cmsg
->cmsg_type
= IP_PKTINFO
;
164 cmsg
->cmsg_len
= CMSG_LEN(sizeof(pktinfo
));
166 memset(&pktinfo
, 0, sizeof(pktinfo
));
167 memcpy(&pktinfo
.ipi_spec_dst
, from
, sizeof(*from
));
168 memcpy(CMSG_DATA(cmsg
), &pktinfo
, sizeof(pktinfo
));
170 return sendmsg(s
, &msg
, flags
);