added writesize
[tridge/junkcode.git] / igmp_query.c
1 /* simple igmp query tool
2
3    Andrew Tridgell 2004
4 */
5
6 #include <fcntl.h>
7 #include <errno.h>
8 #include <stdio.h>
9 #include <ctype.h>
10 #include <stdlib.h>
11 #include <unistd.h>
12 #include <string.h>
13 #include <time.h>
14 #include <errno.h>
15 #include <sys/types.h>
16 #include <sys/socket.h>
17 #include <sys/ioctl.h>
18 #include <sys/time.h>
19 #include <netinet/in.h>
20 #include <netinet/ip.h>
21 #include <netinet/udp.h>
22 #include <arpa/inet.h>
23 #include <linux/if_ether.h>
24 #include <linux/if_arp.h>
25
26 #define DEBUG 0
27
28 #define MAX_PACKET_LENGTH 1600 
29 #define HISTORY_LENGTH 20
30
31 typedef unsigned char byte_t;
32
33 static void s_8(byte_t *pkt, int ofs, unsigned v)
34 {
35         pkt[ofs] = v;
36 }
37
38 static void s_16(byte_t *pkt, int ofs, unsigned v)
39 {
40         ((unsigned short *)(pkt + ofs))[0] = htons(v);
41 }
42
43 static void s_32(byte_t *pkt, int ofs, unsigned v)
44 {
45         ((unsigned *)(pkt + ofs))[0] = htonl(v);
46 }
47
48
49 static unsigned ip_sum(byte_t *data, int len)
50 {
51         int i;
52         unsigned sum = 0;
53         for (i=0;i<len;i+=2) {
54                 sum += ((unsigned short *)(data + i))[0];
55                 sum &= 0xFFFF;
56         }
57         return htons(~sum);
58 }
59
60
61 static int iface_index(const char *ifname)
62 {
63         int fd;
64         struct ifreq ifr;
65
66         if (isdigit(ifname[0])) {
67                 return atoi(ifname);
68         }
69
70         fd = socket(PF_PACKET, SOCK_DGRAM, ETH_P_IP);
71         if (fd == -1) {
72                 perror("socket");
73                 return -1;
74         }
75
76         strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
77
78         if (ioctl(fd, SIOCGIFINDEX, &ifr) != 0) {
79                 perror("SIOCGIFINDEX");
80                 return -1;
81         }
82
83         close(fd);
84
85         return ifr.ifr_ifindex;
86 }
87
88
89 int main(int argc, char *argv[])
90 {
91         int fd;
92         byte_t pkt[32];
93         int ttl;
94         in_addr_t src_ip, dst_ip, group_ip;
95         struct sockaddr_ll ll;
96         int ifindex;
97
98         if (argc < 3) {
99                 printf("Usage igmp_query <ttl> <iface> <src>\n");
100                 exit(1);
101         }
102
103
104         fd = socket(PF_PACKET, SOCK_DGRAM, ETH_P_IP);
105         if (fd == -1) {
106                 perror("eth_open");
107                 exit(1);
108         }
109
110         ttl = atoi(argv[1]);
111
112         ifindex = iface_index(argv[2]);
113         if (ifindex == -1) {
114                 printf("Bad interface name '%s'\n", argv[2]);
115                 exit(1);
116         }
117
118         src_ip   = inet_addr(argv[3]);
119         dst_ip   = inet_addr("224.0.0.1");
120
121         group_ip = dst_ip;
122
123         memset(pkt, 0, sizeof(pkt));
124         
125         s_8 (pkt,  0, 0x46);
126         s_16(pkt,  2, sizeof(pkt));
127         s_8 (pkt,  8, ttl); 
128         s_8 (pkt,  9, 2);
129         s_32(pkt, 12, htonl(src_ip));
130         s_32(pkt, 16, htonl(dst_ip));
131         s_32(pkt, 20, 0x94040000); /* router alert */
132         
133         s_8 (pkt, 24, 0x11); /* membership report */
134         s_8 (pkt, 25, 0x64); /* v2 */
135         s_32(pkt, 28, 0);
136         
137         s_16(pkt, 26, ip_sum(pkt+24, 8)); /* igmp csum */
138         
139         s_16(pkt, 10, ip_sum(pkt, 32)-0x100); /* ip csum */
140         
141         ll.sll_family = AF_PACKET;
142         ll.sll_protocol = htons(ETH_P_IP);
143         ll.sll_ifindex = ifindex;
144         ll.sll_hatype  = htons(ARPHRD_ETHER);
145         ll.sll_pkttype = PACKET_MULTICAST;
146         ll.sll_halen = 6;
147         ll.sll_addr[0] = 1;
148         ll.sll_addr[1] = 0;
149         s_32(ll.sll_addr, 2, (0x5e<<24) | (ntohl(group_ip) & 0x7FFFFF));
150         
151         sendto(fd, pkt, sizeof(pkt), 0, (struct sockaddr *)&ll, sizeof(ll));
152         
153         return 0;
154 }