5 #include <sys/socket.h>
6 #include <netinet/in.h>
20 bool opt_same_interface = false;
22 int ac_sockfd[AC_NUM];
25 unsigned received, last_received;
33 for (i = 0 ; i < AC_NUM; i++)
34 close(ac_sockfd[AC_NUM]);
39 int create_ac_socket(unsigned int ac)
41 struct sockaddr_in my_addr;
43 unsigned int yes=1, tos;
46 if ((sockfd = socket(PF_INET, SOCK_DGRAM, 0)) < 0)
48 perror("Socket nelze otevrit");
52 if (setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,&yes,sizeof(int)) == -1) {
53 perror("Chyba v nastaveni soketu");
57 if (opt_same_interface) {
59 if (setsockopt(sockfd, SOL_IP, IP_PKTINFO, &receive, sizeof(receive)) == -1) {
60 perror("setsockopt: IP_PKTINFO");
66 // bzero(&my_addr, sizeof(my_addr));
67 memset(&my_addr,0, sizeof(my_addr));
68 my_addr.sin_family = AF_INET;
69 my_addr.sin_addr.s_addr = INADDR_ANY;
70 my_addr.sin_port = htons(BASE_PORT + ac);
72 if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr)) == -1) {
73 perror("Chyba v bind");
78 //tos = ((AC_NUM - ac) *2 - 1)*32;
80 if (setsockopt(sockfd, SOL_IP, IP_TOS, &tos, sizeof(tos))) {
81 perror("Unable to set TOS");
89 void* qhandler(void* queue)
92 struct sockaddr_in rem_addr;
94 unsigned int ac, rem_addr_length;
95 char cbufrec[512], cbufsend[512];
98 struct in_pktinfo *ipi = NULL;
101 rem_addr_length=sizeof(rem_addr);
107 struct cmsghdr *cmsg;
109 iov.iov_base = &buff;
110 iov.iov_len = sizeof(buff);
111 msg.msg_name = (void*)&rem_addr;
112 msg.msg_namelen = sizeof(rem_addr);
116 msg.msg_control = cbufrec;
117 msg.msg_controllen = sizeof(cbufrec);
119 while ((mlen = recvmsg(ac_sockfd[ac], &msg, 0)) < 0) {
120 if (errno == EINTR) continue;
124 clock_gettime(CLOCK_REALTIME, &buff.msg.sendback_timestamp);
127 if (opt_same_interface) {
128 /* determine receiving interface */
129 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
130 if (cmsg->cmsg_level == SOL_IP) {
131 if (cmsg->cmsg_type == IP_PKTINFO) {
132 /* char spec_dst[20], addr[20]; */
133 ipi = (struct in_pktinfo*)CMSG_DATA(cmsg);
134 if (cmsg->cmsg_len <= sizeof(cbufsend)) {
135 struct in_pktinfo *ipi2;
136 msg.msg_control = cbufsend;
137 msg.msg_controllen = CMSG_LEN(sizeof(struct in_pktinfo));
138 cmsg = CMSG_FIRSTHDR(&msg);
139 cmsg->cmsg_level = SOL_IP;
140 cmsg->cmsg_type = IP_PKTINFO;
141 cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
142 /* Initialize the payload: */
143 ipi2 = (struct in_pktinfo*)CMSG_DATA(cmsg);
144 memset(ipi2, 0, sizeof(*ipi2));
145 ipi2->ipi_ifindex = ipi->ipi_ifindex;
148 fprintf(stderr, "cbufsend too small\n");
149 msg.msg_control = NULL;
150 msg.msg_controllen = 0;
152 /* strncpy(spec_dst, inet_ntoa(ipi->ipi_spec_dst), sizeof(spec_dst)-1); */
153 /* strncpy(addr, inet_ntoa(ipi->ipi_addr), sizeof(addr)-1); */
154 /* printf("pktinfo if=%d %s %s\n", ipi->ipi_ifindex, spec_dst, addr); */
159 msg.msg_control = NULL;
160 msg.msg_controllen = 0;
166 receivers[ac].received++;
167 msg.msg_iov->iov_len = mlen;
168 while (sendmsg(ac_sockfd[ac], &msg, 0) < 0) {
169 if (errno == EINTR) continue;
177 int main(int argc, char *argv[])
186 while ((opt = getopt(argc, argv, "I")) != -1) {
189 opt_same_interface = true;
192 fprintf(stderr, "Usage: %s [ options ]\n\n", argv[0]);
193 fprintf(stderr, "Options:\n");
194 fprintf(stderr, " -I send back through the same interface (bypass routing tables)\n");
198 pthread_attr_init(&attr);
200 if (signal(SIGTERM, stopper) == SIG_ERR) {
201 perror("Signal handler registration error");
205 if (signal(SIGINT, stopper) == SIG_ERR) {
206 perror("Signal handler registration error");
210 for (ac = 0; ac < AC_NUM; ac++) {
211 ac_sockfd[ac] = create_ac_socket(ac);
212 rc = pthread_create(&thread, &attr, qhandler, (void*) ac);
218 for (ac = 0; ac < AC_NUM; ac++) {
219 int delta = receivers[ac].received - receivers[ac].last_received;
220 receivers[ac].last_received = receivers[ac].received;
221 fprintf(stderr, "%s %5d %4d/s ", ac_to_text[ac], receivers[ac].received, delta);