[Settings] DHCP is now working using the new settings API.
[people/mcb30/gpxe.git] / src / net / dhcpopts.c
1 /*
2  * Copyright (C) 2008 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., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 #include <stdint.h>
20 #include <stdlib.h>
21 #include <stdio.h>
22 #include <errno.h>
23 #include <string.h>
24 #include <gpxe/dhcp.h>
25 #include <gpxe/dhcpopts.h>
26
27 /** @file
28  *
29  * DHCP options
30  *
31  */
32
33 /**
34  * Obtain printable version of a DHCP option tag
35  *
36  * @v tag               DHCP option tag
37  * @ret name            String representation of the tag
38  *
39  */
40 static inline char * dhcp_tag_name ( unsigned int tag ) {
41         static char name[8];
42
43         if ( DHCP_IS_ENCAP_OPT ( tag ) ) {
44                 snprintf ( name, sizeof ( name ), "%d.%d",
45                            DHCP_ENCAPSULATOR ( tag ),
46                            DHCP_ENCAPSULATED ( tag ) );
47         } else {
48                 snprintf ( name, sizeof ( name ), "%d", tag );
49         }
50         return name;
51 }
52
53 /**
54  * Get pointer to DHCP option
55  *
56  * @v options           DHCP options block
57  * @v offset            Offset within options block
58  * @ret option          DHCP option
59  */
60 static inline __attribute__ (( always_inline )) struct dhcp_option *
61 dhcp_option ( struct dhcp_options *options, unsigned int offset ) {
62         return ( ( struct dhcp_option * ) ( options->data + offset ) );
63 }
64
65 /**
66  * Get offset of a DHCP option
67  *
68  * @v options           DHCP options block
69  * @v option            DHCP option
70  * @ret offset          Offset within options block
71  */
72 static inline __attribute__ (( always_inline )) int
73 dhcp_option_offset ( struct dhcp_options *options,
74                      struct dhcp_option *option ) {
75         return ( ( ( void * ) option ) - options->data );
76 }
77
78 /**
79  * Calculate length of any DHCP option
80  *
81  * @v option            DHCP option
82  * @ret len             Length (including tag and length field)
83  */
84 static unsigned int dhcp_option_len ( struct dhcp_option *option ) {
85         if ( ( option->tag == DHCP_END ) || ( option->tag == DHCP_PAD ) ) {
86                 return 1;
87         } else {
88                 return ( option->len + DHCP_OPTION_HEADER_LEN );
89         }
90 }
91
92 /**
93  * Find DHCP option within DHCP options block, and its encapsulator (if any)
94  *
95  * @v options           DHCP options block
96  * @v tag               DHCP option tag to search for
97  * @ret encap_offset    Offset of encapsulating DHCP option
98  * @ret offset          Offset of DHCP option, or negative error
99  *
100  * Searches for the DHCP option matching the specified tag within the
101  * DHCP option block.  Encapsulated options may be searched for by
102  * using DHCP_ENCAP_OPT() to construct the tag value.
103  *
104  * If the option is encapsulated, and @c encapsulator is non-NULL, it
105  * will be filled in with the offset of the encapsulating option.
106  *
107  * This routine is designed to be paranoid.  It does not assume that
108  * the option data is well-formatted, and so must guard against flaws
109  * such as options missing a @c DHCP_END terminator, or options whose
110  * length would take them beyond the end of the data block.
111  */
112 static int find_dhcp_option_with_encap ( struct dhcp_options *options,
113                                          unsigned int tag,
114                                          int *encap_offset ) {
115         unsigned int original_tag __attribute__ (( unused )) = tag;
116         struct dhcp_option *option;
117         int offset = 0;
118         ssize_t remaining = options->len;
119         unsigned int option_len;
120
121         while ( remaining ) {
122                 /* Calculate length of this option.  Abort processing
123                  * if the length is malformed (i.e. takes us beyond
124                  * the end of the data block).
125                  */
126                 option = dhcp_option ( options, offset );
127                 option_len = dhcp_option_len ( option );
128                 remaining -= option_len;
129                 if ( remaining < 0 )
130                         break;
131                 /* Check for matching tag */
132                 if ( option->tag == tag ) {
133                         DBGC ( options, "DHCPOPT %p found %s (length %d)\n",
134                                options, dhcp_tag_name ( original_tag ),
135                                option_len );
136                         return offset;
137                 }
138                 /* Check for explicit end marker */
139                 if ( option->tag == DHCP_END )
140                         break;
141                 /* Check for start of matching encapsulation block */
142                 if ( DHCP_IS_ENCAP_OPT ( tag ) &&
143                      ( option->tag == DHCP_ENCAPSULATOR ( tag ) ) ) {
144                         if ( encap_offset )
145                                 *encap_offset = offset;
146                         /* Continue search within encapsulated option block */
147                         tag = DHCP_ENCAPSULATED ( tag );
148                         remaining = option_len;
149                         offset += DHCP_OPTION_HEADER_LEN;
150                         continue;
151                 }
152                 offset += option_len;
153         }
154         return -ENOENT;
155 }
156
157 /**
158  * Resize a DHCP option
159  *
160  * @v options           DHCP option block
161  * @v offset            Offset of option to resize
162  * @v encap_offset      Offset of encapsulating offset (or -ve for none)
163  * @v old_len           Old length (including header)
164  * @v new_len           New length (including header)
165  * @v can_realloc       Can reallocate options data if necessary
166  * @ret rc              Return status code
167  */
168 static int resize_dhcp_option ( struct dhcp_options *options,
169                                 int offset, int encap_offset,
170                                 size_t old_len, size_t new_len,
171                                 int can_realloc ) {
172         struct dhcp_option *encapsulator;
173         struct dhcp_option *option;
174         ssize_t delta = ( new_len - old_len );
175         size_t new_options_len;
176         size_t new_encapsulator_len;
177         void *new_data;
178         void *source;
179         void *dest;
180         void *end;
181
182         /* Check for sufficient space, and update length fields */
183         if ( new_len > DHCP_MAX_LEN ) {
184                 DBGC ( options, "DHCPOPT %p overlength option\n", options );
185                 return -ENOSPC;
186         }
187         new_options_len = ( options->len + delta );
188         if ( new_options_len > options->max_len ) {
189                 /* Reallocate options block if allowed to do so. */
190                 if ( can_realloc ) {
191                         new_data = realloc ( options->data, new_options_len );
192                         if ( ! new_data ) {
193                                 DBGC ( options, "DHCPOPT %p could not "
194                                        "reallocate to %zd bytes\n", options,
195                                        new_options_len );
196                                 return -ENOMEM;
197                         }
198                         options->data = new_data;
199                         options->max_len = new_options_len;
200                 } else {
201                         DBGC ( options, "DHCPOPT %p out of space\n", options );
202                         return -ENOMEM;
203                 }
204         }
205         if ( encap_offset >= 0 ) {
206                 encapsulator = dhcp_option ( options, encap_offset );
207                 new_encapsulator_len = ( encapsulator->len + delta );
208                 if ( new_encapsulator_len > DHCP_MAX_LEN ) {
209                         DBGC ( options, "DHCPOPT %p overlength encapsulator\n",
210                                options );
211                         return -ENOSPC;
212                 }
213                 encapsulator->len = new_encapsulator_len;
214         }
215         options->len = new_options_len;
216
217         /* Move remainder of option data */
218         option = dhcp_option ( options, offset );
219         source = ( ( ( void * ) option ) + old_len );
220         dest = ( ( ( void * ) option ) + new_len );
221         end = ( options->data + options->max_len );
222         memmove ( dest, source, ( end - dest ) );
223
224         return 0;
225 }
226
227 /**
228  * Set value of DHCP option
229  *
230  * @v options           DHCP option block
231  * @v tag               DHCP option tag
232  * @v data              New value for DHCP option
233  * @v len               Length of value, in bytes
234  * @v can_realloc       Can reallocate options data if necessary
235  * @ret offset          Offset of DHCP option, or negative error
236  *
237  * Sets the value of a DHCP option within the options block.  The
238  * option may or may not already exist.  Encapsulators will be created
239  * (and deleted) as necessary.
240  *
241  * This call may fail due to insufficient space in the options block.
242  * If it does fail, and the option existed previously, the option will
243  * be left with its original value.
244  */
245 static int set_dhcp_option ( struct dhcp_options *options, unsigned int tag,
246                              const void *data, size_t len,
247                              int can_realloc ) {
248         static const uint8_t empty_encapsulator[] = { DHCP_END };
249         int offset;
250         int encap_offset = -1;
251         int creation_offset = 0;
252         struct dhcp_option *option;
253         unsigned int encap_tag = DHCP_ENCAPSULATOR ( tag );
254         size_t old_len = 0;
255         size_t new_len = ( len ? ( len + DHCP_OPTION_HEADER_LEN ) : 0 );
256         int rc;
257
258         /* Find old instance of this option, if any */
259         offset = find_dhcp_option_with_encap ( options, tag, &encap_offset );
260         if ( offset >= 0 ) {
261                 old_len = dhcp_option_len ( dhcp_option ( options, offset ) );
262                 DBGC ( options, "DHCPOPT %p resizing %s from %zd to %zd\n",
263                        options, dhcp_tag_name ( tag ), old_len, new_len );
264         } else {
265                 DBGC ( options, "DHCPOPT %p creating %s (length %zd)\n",
266                        options, dhcp_tag_name ( tag ), new_len );
267         }
268
269         /* Ensure that encapsulator exists, if required */
270         if ( encap_tag ) {
271                 if ( encap_offset < 0 )
272                         encap_offset = set_dhcp_option ( options, encap_tag,
273                                                          empty_encapsulator, 1,
274                                                          can_realloc );
275                 if ( encap_offset < 0 )
276                         return encap_offset;
277                 creation_offset = ( encap_offset + DHCP_OPTION_HEADER_LEN );
278         }
279
280         /* Create new option if necessary */
281         if ( offset < 0 )
282                 offset = creation_offset;
283
284         /* Resize option to fit new data */
285         if ( ( rc = resize_dhcp_option ( options, offset, encap_offset,
286                                          old_len, new_len,
287                                          can_realloc ) ) != 0 )
288                 return rc;
289
290         /* Copy new data into option, if applicable */
291         if ( len ) {
292                 option = dhcp_option ( options, offset );
293                 option->tag = tag;
294                 option->len = len;
295                 memcpy ( &option->data, data, len );
296         }
297
298         /* Delete encapsulator if there's nothing else left in it */
299         if ( encap_offset >= 0 ) {
300                 option = dhcp_option ( options, encap_offset );
301                 if ( option->len <= 1 )
302                         set_dhcp_option ( options, encap_tag, NULL, 0, 0 );
303         }
304
305         return offset;
306 }
307
308 /**
309  * Store value of DHCP option setting
310  *
311  * @v options           DHCP option block
312  * @v tag               Setting tag number
313  * @v data              Setting data, or NULL to clear setting
314  * @v len               Length of setting data
315  * @ret rc              Return status code
316  */
317 int dhcpopt_store ( struct dhcp_options *options, unsigned int tag,
318                     const void *data, size_t len ) {
319         int offset;
320
321         offset = set_dhcp_option ( options, tag, data, len, 0 );
322         if ( offset < 0 )
323                 return offset;
324         return 0;
325 }
326
327 /**
328  * Store value of DHCP option setting, extending options block if necessary
329  *
330  * @v options           DHCP option block
331  * @v tag               Setting tag number
332  * @v data              Setting data, or NULL to clear setting
333  * @v len               Length of setting data
334  * @ret rc              Return status code
335  */
336 int dhcpopt_extensible_store ( struct dhcp_options *options, unsigned int tag,
337                                const void *data, size_t len ) {
338         int offset;
339
340         offset = set_dhcp_option ( options, tag, data, len, 1 );
341         if ( offset < 0 )
342                 return offset;
343         return 0;
344 }
345
346 /**
347  * Fetch value of DHCP option setting
348  *
349  * @v options           DHCP option block
350  * @v tag               Setting tag number
351  * @v data              Buffer to fill with setting data
352  * @v len               Length of buffer
353  * @ret len             Length of setting data, or negative error
354  */
355 int dhcpopt_fetch ( struct dhcp_options *options, unsigned int tag,
356                     void *data, size_t len ) {
357         int offset;
358         struct dhcp_option *option;
359         size_t option_len;
360
361         offset = find_dhcp_option_with_encap ( options, tag, NULL );
362         if ( offset < 0 )
363                 return offset;
364
365         option = dhcp_option ( options, offset );
366         option_len = option->len;
367         if ( len > option_len )
368                 len = option_len;
369         memcpy ( data, option->data, len );
370
371         return option_len;
372 }
373
374 /**
375  * Recalculate length of DHCP options block
376  *
377  * @v options           Uninitialised DHCP option block
378  *
379  * The "used length" field will be updated based on scanning through
380  * the block to find the end of the options.
381  */
382 static void dhcpopt_update_len ( struct dhcp_options *options ) {
383         struct dhcp_option *option;
384         int offset = 0;
385         ssize_t remaining = options->max_len;
386         unsigned int option_len;
387
388         /* Find last non-pad option */
389         options->len = 0;
390         while ( remaining ) {
391                 option = dhcp_option ( options, offset );
392                 option_len = dhcp_option_len ( option );
393                 remaining -= option_len;
394                 if ( remaining < 0 )
395                         break;
396                 offset += option_len;
397                 if ( option->tag != DHCP_PAD )
398                         options->len = offset;
399         }
400 }
401
402 /**
403  * Initialise prepopulated block of DHCP options
404  *
405  * @v options           Uninitialised DHCP option block
406  * @v data              Memory for DHCP option data
407  * @v max_len           Length of memory for DHCP option data
408  *
409  * The memory content must already be filled with valid DHCP options.
410  * A zeroed block counts as a block of valid DHCP options.
411  */
412 void dhcpopt_init ( struct dhcp_options *options, void *data,
413                     size_t max_len ) {
414
415         /* Fill in fields */
416         options->data = data;
417         options->max_len = max_len;
418
419         /* Update length */
420         dhcpopt_update_len ( options );
421
422         DBGC ( options, "DHCPOPT %p created (data %p len %#zx max_len %#zx)\n",
423                options, options->data, options->len, options->max_len );
424 }