Various cleanups I made while going through Erik Hovland's patch submissions,
[oweals/busybox.git] / networking / udhcp / socket.c
1 /*
2  * socket.c -- DHCP server client/server socket creation
3  *
4  * udhcp client/server
5  * Copyright (C) 1999 Matthew Ramsay <matthewr@moreton.com.au>
6  *                      Chris Trew <ctrew@moreton.com.au>
7  *
8  * Rewrite by Russ Dill <Russ.Dill@asu.edu> July 2001
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23  */
24
25 #include <sys/types.h>
26 #include <sys/socket.h>
27 #include <sys/ioctl.h>
28 #include <netinet/in.h>
29 #include <unistd.h>
30 #include <string.h>
31 #include <arpa/inet.h>
32 #include <net/if.h>
33 #include <errno.h>
34 #include <features.h>
35 #if (__GLIBC__ >= 2 && __GLIBC_MINOR >= 1) || defined _NEWLIB_VERSION
36 #include <netpacket/packet.h>
37 #include <net/ethernet.h>
38 #else
39 #include <asm/types.h>
40 #include <linux/if_packet.h>
41 #include <linux/if_ether.h>
42 #endif
43
44 #include "common.h"
45 #include "socket.h"
46
47 int read_interface(char *interface, int *ifindex, uint32_t *addr, uint8_t *arp)
48 {
49         int fd;
50         struct ifreq ifr;
51         struct sockaddr_in *our_ip;
52
53         memset(&ifr, 0, sizeof(struct ifreq));
54         if((fd = socket(AF_INET, SOCK_RAW, IPPROTO_RAW)) >= 0) {
55                 ifr.ifr_addr.sa_family = AF_INET;
56                 strcpy(ifr.ifr_name, interface);
57
58                 if (addr) {
59                         if (ioctl(fd, SIOCGIFADDR, &ifr) == 0) {
60                                 our_ip = (struct sockaddr_in *) &ifr.ifr_addr;
61                                 *addr = our_ip->sin_addr.s_addr;
62                                 DEBUG(LOG_INFO, "%s (our ip) = %s", ifr.ifr_name, inet_ntoa(our_ip->sin_addr));
63                         } else {
64                                 LOG(LOG_ERR, "SIOCGIFADDR failed, is the interface up and configured?: %m");
65                                 close(fd);
66                                 return -1;
67                         }
68                 }
69
70                 if (ioctl(fd, SIOCGIFINDEX, &ifr) == 0) {
71                         DEBUG(LOG_INFO, "adapter index %d", ifr.ifr_ifindex);
72                         *ifindex = ifr.ifr_ifindex;
73                 } else {
74                         LOG(LOG_ERR, "SIOCGIFINDEX failed!: %m");
75                         close(fd);
76                         return -1;
77                 }
78                 if (ioctl(fd, SIOCGIFHWADDR, &ifr) == 0) {
79                         memcpy(arp, ifr.ifr_hwaddr.sa_data, 6);
80                         DEBUG(LOG_INFO, "adapter hardware address %02x:%02x:%02x:%02x:%02x:%02x",
81                                 arp[0], arp[1], arp[2], arp[3], arp[4], arp[5]);
82                 } else {
83                         LOG(LOG_ERR, "SIOCGIFHWADDR failed!: %m");
84                         close(fd);
85                         return -1;
86                 }
87         } else {
88                 LOG(LOG_ERR, "socket failed!: %m");
89                 return -1;
90         }
91         close(fd);
92         return 0;
93 }
94
95
96 int listen_socket(uint32_t ip, int port, char *inf)
97 {
98         struct ifreq interface;
99         int fd;
100         struct sockaddr_in addr;
101         int n = 1;
102
103         DEBUG(LOG_INFO, "Opening listen socket on 0x%08x:%d %s", ip, port, inf);
104         if ((fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
105                 DEBUG(LOG_ERR, "socket call failed: %m");
106                 return -1;
107         }
108
109         memset(&addr, 0, sizeof(addr));
110         addr.sin_family = AF_INET;
111         addr.sin_port = htons(port);
112         addr.sin_addr.s_addr = ip;
113
114         if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *) &n, sizeof(n)) == -1) {
115                 close(fd);
116                 return -1;
117         }
118         if (setsockopt(fd, SOL_SOCKET, SO_BROADCAST, (char *) &n, sizeof(n)) == -1) {
119                 close(fd);
120                 return -1;
121         }
122
123         strncpy(interface.ifr_name, inf, IFNAMSIZ);
124         if (setsockopt(fd, SOL_SOCKET, SO_BINDTODEVICE,(char *)&interface, sizeof(interface)) < 0) {
125                 close(fd);
126                 return -1;
127         }
128
129         if (bind(fd, (struct sockaddr *)&addr, sizeof(struct sockaddr)) == -1) {
130                 close(fd);
131                 return -1;
132         }
133
134         return fd;
135 }