1 /*
2 * Copyright (C) 2006 Michael Brown <mbrown@fensystems.co.uk>.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License as
6 * published by the Free Software Foundation; either version 2 of the
7 * License, or any later version.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17 * 02110-1301, USA.
18 *
19 * You can also choose to distribute this program under the terms of
20 * the Unmodified Binary Distribution Licence (as given in the file
21 * COPYING.UBDL), provided that you have satisfied its requirements.
22 */
23
24 FILE_LICENCE ( GPL2_OR_LATER_OR_UBDL );
25
26 #include <stdint.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <byteswap.h>
30 #include <errno.h>
31 #include <ipxe/if_ether.h>
32 #include <ipxe/if_arp.h>
33 #include <ipxe/iobuf.h>
34 #include <ipxe/netdevice.h>
35 #include <ipxe/neighbour.h>
36 #include <ipxe/arp.h>
37
38 /** @file
39 *
40 * Address Resolution Protocol
41 *
42 * This file implements the address resolution protocol as defined in
43 * RFC826. The implementation is media-independent and
44 * protocol-independent; it is not limited to Ethernet or to IPv4.
45 *
46 */
47
48 struct net_protocol arp_protocol __net_protocol;
49
50 /**
51 * Transmit ARP request
52 *
53 * @v netdev Network device
54 * @v net_protocol Network-layer protocol
55 * @v net_dest Destination network-layer address
56 * @v net_source Source network-layer address
57 * @ret rc Return status code
58 */
arp_tx_request(struct net_device * netdev,struct net_protocol * net_protocol,const void * net_dest,const void * net_source)59 int arp_tx_request ( struct net_device *netdev,
60 struct net_protocol *net_protocol,
61 const void *net_dest, const void *net_source ) {
62 struct ll_protocol *ll_protocol = netdev->ll_protocol;
63 struct io_buffer *iobuf;
64 struct arphdr *arphdr;
65 int rc;
66
67 /* Allocate ARP packet */
68 iobuf = alloc_iob ( MAX_LL_HEADER_LEN + sizeof ( *arphdr ) +
69 ( 2 * ( MAX_LL_ADDR_LEN + MAX_NET_ADDR_LEN ) ) );
70 if ( ! iobuf )
71 return -ENOMEM;
72 iob_reserve ( iobuf, MAX_LL_HEADER_LEN );
73
74 /* Build up ARP request */
75 arphdr = iob_put ( iobuf, sizeof ( *arphdr ) );
76 arphdr->ar_hrd = ll_protocol->ll_proto;
77 arphdr->ar_hln = ll_protocol->ll_addr_len;
78 arphdr->ar_pro = net_protocol->net_proto;
79 arphdr->ar_pln = net_protocol->net_addr_len;
80 arphdr->ar_op = htons ( ARPOP_REQUEST );
81 memcpy ( iob_put ( iobuf, ll_protocol->ll_addr_len ),
82 netdev->ll_addr, ll_protocol->ll_addr_len );
83 memcpy ( iob_put ( iobuf, net_protocol->net_addr_len ),
84 net_source, net_protocol->net_addr_len );
85 memset ( iob_put ( iobuf, ll_protocol->ll_addr_len ),
86 0, ll_protocol->ll_addr_len );
87 memcpy ( iob_put ( iobuf, net_protocol->net_addr_len ),
88 net_dest, net_protocol->net_addr_len );
89
90 /* Transmit ARP request */
91 if ( ( rc = net_tx ( iobuf, netdev, &arp_protocol,
92 netdev->ll_broadcast, netdev->ll_addr ) ) != 0 ) {
93 DBGC ( netdev, "ARP %s %s %s could not transmit request: %s\n",
94 netdev->name, net_protocol->name,
95 net_protocol->ntoa ( net_dest ), strerror ( rc ) );
96 return rc;
97 }
98
99 return 0;
100 }
101
102 /** ARP neighbour discovery protocol */
103 struct neighbour_discovery arp_discovery = {
104 .name = "ARP",
105 .tx_request = arp_tx_request,
106 };
107
108 /**
109 * Identify ARP protocol
110 *
111 * @v net_proto Network-layer protocol, in network-endian order
112 * @ret arp_net_protocol ARP protocol, or NULL
113 *
114 */
arp_find_protocol(uint16_t net_proto)115 static struct arp_net_protocol * arp_find_protocol ( uint16_t net_proto ) {
116 struct arp_net_protocol *arp_net_protocol;
117
118 for_each_table_entry ( arp_net_protocol, ARP_NET_PROTOCOLS ) {
119 if ( arp_net_protocol->net_protocol->net_proto == net_proto )
120 return arp_net_protocol;
121 }
122 return NULL;
123 }
124
125 /**
126 * Process incoming ARP packets
127 *
128 * @v iobuf I/O buffer
129 * @v netdev Network device
130 * @v ll_source Link-layer source address
131 * @v flags Packet flags
132 * @ret rc Return status code
133 */
arp_rx(struct io_buffer * iobuf,struct net_device * netdev,const void * ll_dest __unused,const void * ll_source __unused,unsigned int flags __unused)134 static int arp_rx ( struct io_buffer *iobuf, struct net_device *netdev,
135 const void *ll_dest __unused,
136 const void *ll_source __unused,
137 unsigned int flags __unused ) {
138 struct arphdr *arphdr = iobuf->data;
139 struct arp_net_protocol *arp_net_protocol;
140 struct net_protocol *net_protocol;
141 struct ll_protocol *ll_protocol;
142 size_t len = iob_len ( iobuf );
143 int rc;
144
145 /* Sanity check */
146 if ( ( len < sizeof ( *arphdr ) ) || ( len < arp_len ( arphdr ) ) ) {
147 rc = -EINVAL;
148 goto done;
149 }
150
151 /* Identify network-layer and link-layer protocols */
152 arp_net_protocol = arp_find_protocol ( arphdr->ar_pro );
153 if ( ! arp_net_protocol ) {
154 rc = -EPROTONOSUPPORT;
155 goto done;
156 }
157 net_protocol = arp_net_protocol->net_protocol;
158 ll_protocol = netdev->ll_protocol;
159
160 /* Sanity checks */
161 if ( ( arphdr->ar_hrd != ll_protocol->ll_proto ) ||
162 ( arphdr->ar_hln != ll_protocol->ll_addr_len ) ||
163 ( arphdr->ar_pln != net_protocol->net_addr_len ) ) {
164 rc = -EINVAL;
165 goto done;
166 }
167
168 /* Update neighbour cache entry for this sender, if any */
169 neighbour_update ( netdev, net_protocol, arp_sender_pa ( arphdr ),
170 arp_sender_ha ( arphdr ) );
171
172 /* If it's not a request, there's nothing more to do */
173 if ( arphdr->ar_op != htons ( ARPOP_REQUEST ) ) {
174 rc = 0;
175 goto done;
176 }
177
178 /* See if we own the target protocol address */
179 if ( arp_net_protocol->check ( netdev, arp_target_pa ( arphdr ) ) != 0){
180 rc = 0;
181 goto done;
182 }
183
184 /* Change request to a reply */
185 DBGC2 ( netdev, "ARP %s %s %s reply => %s %s\n",
186 netdev->name, net_protocol->name,
187 net_protocol->ntoa ( arp_target_pa ( arphdr ) ),
188 ll_protocol->name, ll_protocol->ntoa ( netdev->ll_addr ) );
189 arphdr->ar_op = htons ( ARPOP_REPLY );
190 memswap ( arp_sender_ha ( arphdr ), arp_target_ha ( arphdr ),
191 arphdr->ar_hln + arphdr->ar_pln );
192 memcpy ( arp_sender_ha ( arphdr ), netdev->ll_addr, arphdr->ar_hln );
193
194 /* Send reply */
195 if ( ( rc = net_tx ( iob_disown ( iobuf ), netdev, &arp_protocol,
196 arp_target_ha ( arphdr ),
197 netdev->ll_addr ) ) != 0 ) {
198 DBGC ( netdev, "ARP %s %s %s could not transmit reply: %s\n",
199 netdev->name, net_protocol->name,
200 net_protocol->ntoa ( arp_target_pa ( arphdr ) ),
201 strerror ( rc ) );
202 goto done;
203 }
204
205 /* Success */
206 rc = 0;
207
208 done:
209 free_iob ( iobuf );
210 return rc;
211 }
212
213 /**
214 * Transcribe ARP address
215 *
216 * @v net_addr ARP address
217 * @ret string "<ARP>"
218 *
219 * This operation is meaningless for the ARP protocol.
220 */
arp_ntoa(const void * net_addr __unused)221 static const char * arp_ntoa ( const void *net_addr __unused ) {
222 return "<ARP>";
223 }
224
225 /** ARP network protocol */
226 struct net_protocol arp_protocol __net_protocol = {
227 .name = "ARP",
228 .net_proto = htons ( ETH_P_ARP ),
229 .rx = arp_rx,
230 .ntoa = arp_ntoa,
231 };
232