5 #include <sys/socket.h>
6 #include <netinet/in.h>
20 bool opt_same_interface = false;
22 static int ac_sockfd[AC_NUM];
25 unsigned received, last_received;
33 for (i = 0; i < AC_NUM; i++)
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;
102 rem_addr_length=sizeof(rem_addr);
108 struct cmsghdr *cmsg;
110 iov.iov_base = &buff;
111 iov.iov_len = sizeof(buff);
112 msg.msg_name = (void*)&rem_addr;
113 msg.msg_namelen = sizeof(rem_addr);
117 msg.msg_control = cbufrec;
118 msg.msg_controllen = sizeof(cbufrec);
120 while ((mlen = recvmsg(ac_sockfd[ac], &msg, 0)) < 0) {
121 if (errno == EINTR) continue;
125 clock_gettime(CLOCK_REALTIME, &ts);
126 buff.msg.sendback_timestamp = ts.tv_sec*1000000000 + ts.tv_nsec;
129 if (opt_same_interface) {
130 /* determine receiving interface */
131 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
132 if (cmsg->cmsg_level == SOL_IP) {
133 if (cmsg->cmsg_type == IP_PKTINFO) {
134 /* char spec_dst[20], addr[20]; */
135 ipi = (struct in_pktinfo*)CMSG_DATA(cmsg);
136 if (cmsg->cmsg_len <= sizeof(cbufsend)) {
137 struct in_pktinfo *ipi2;
138 msg.msg_control = cbufsend;
139 msg.msg_controllen = CMSG_LEN(sizeof(struct in_pktinfo));
140 cmsg = CMSG_FIRSTHDR(&msg);
141 cmsg->cmsg_level = SOL_IP;
142 cmsg->cmsg_type = IP_PKTINFO;
143 cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
144 /* Initialize the payload: */
145 ipi2 = (struct in_pktinfo*)CMSG_DATA(cmsg);
146 memset(ipi2, 0, sizeof(*ipi2));
147 ipi2->ipi_ifindex = ipi->ipi_ifindex;
150 fprintf(stderr, "cbufsend too small\n");
151 msg.msg_control = NULL;
152 msg.msg_controllen = 0;
154 /* strncpy(spec_dst, inet_ntoa(ipi->ipi_spec_dst), sizeof(spec_dst)-1); */
155 /* strncpy(addr, inet_ntoa(ipi->ipi_addr), sizeof(addr)-1); */
156 /* printf("pktinfo if=%d %s %s\n", ipi->ipi_ifindex, spec_dst, addr); */
161 msg.msg_control = NULL;
162 msg.msg_controllen = 0;
168 receivers[ac].received++;
169 msg.msg_iov->iov_len = mlen;
171 /* resp_port is already stored in network order */
172 rem_addr.sin_port = buff.msg.resp_port;
174 while (sendmsg(ac_sockfd[ac], &msg, 0) < 0) {
175 if (errno == EINTR) continue;
183 int main(int argc, char *argv[])
192 while ((opt = getopt(argc, argv, "I")) != -1) {
195 opt_same_interface = true;
198 fprintf(stderr, "Usage: %s [ options ]\n\n", argv[0]);
199 fprintf(stderr, "Options:\n");
200 fprintf(stderr, " -I send back through the same interface (bypass routing tables)\n");
204 pthread_attr_init(&attr);
206 if (signal(SIGTERM, stopper) == SIG_ERR) {
207 perror("Signal handler registration error");
211 if (signal(SIGINT, stopper) == SIG_ERR) {
212 perror("Signal handler registration error");
216 for (ac = 0; ac < AC_NUM; ac++) {
217 ac_sockfd[ac] = create_ac_socket(ac);
218 rc = pthread_create(&thread, &attr, qhandler, (void*) ac);
224 for (ac = 0; ac < AC_NUM; ac++) {
225 int delta = receivers[ac].received - receivers[ac].last_received;
226 receivers[ac].last_received = receivers[ac].received;
227 fprintf(stderr, "%s %5d %4d/s ", ac_to_text[ac], receivers[ac].received, delta);