Add concept of DHCP option applicators.

[people/indolent/gpxe.git/.git] / src / net / dhcpopts.c

diff --git a/src/net/dhcpopts.c b/src/net/dhcpopts.c
index bb97731..6038709 100644 (file)
--- a/src/net/dhcpopts.c
+++ b/src/net/dhcpopts.c
@@ -25,6 +25,7 @@
 #include <assert.h>
 #include <gpxe/list.h>
 #include <gpxe/in.h>
+#include <gpxe/uri.h>
 #include <gpxe/dhcp.h>
 
 /** @file
@@ -36,6 +37,12 @@
 /** List of registered DHCP option blocks */
 static LIST_HEAD ( option_blocks );
 
+/** Registered DHCP option applicators */
+static struct dhcp_option_applicator dhcp_option_applicators[0]
+       __table_start ( struct dhcp_option_applicator, dhcp_applicators );
+static struct dhcp_option_applicator dhcp_option_applicators_end[0]
+       __table_end ( struct dhcp_option_applicator, dhcp_applicators );
+
 /**
  * Obtain printable version of a DHCP option tag
  *
@@ -560,3 +567,45 @@ void delete_dhcp_option ( struct dhcp_option_block *options,
                          unsigned int tag ) {
        set_dhcp_option ( options, tag, NULL, 0 );
 }
+
+/**
+ * Apply DHCP options
+ *
+ * @v options          DHCP options block, or NULL
+ * @ret rc             Return status code
+ */
+int apply_dhcp_options ( struct dhcp_option_block *options ) {
+       struct dhcp_option_applicator *applicator;
+       struct dhcp_option *option;
+       struct in_addr tftp_server;
+       struct uri *uri;
+       char uri_string[32];
+       unsigned int tag;
+       int rc;
+
+       /* Set current working URI based on TFTP server */
+       find_dhcp_ipv4_option ( options, DHCP_EB_SIADDR, &tftp_server );
+       snprintf ( uri_string, sizeof ( uri_string ),
+                  "tftp://%s/", inet_ntoa ( tftp_server ) );
+       uri = parse_uri ( uri_string );
+       if ( ! uri )
+               return -ENOMEM;
+       churi ( uri );
+       uri_put ( uri );
+
+       /* Call all registered DHCP option applicators */
+       for ( applicator = dhcp_option_applicators ;
+             applicator < dhcp_option_applicators_end ; applicator++ ) {
+               tag = applicator->tag;
+               option = find_dhcp_option ( options, tag );
+               if ( ! option )
+                       continue;
+               if ( ( rc = applicator->apply ( tag, option ) ) != 0 ) {
+                       DBG ( "Could not apply DHCP option %s: %s\n",
+                             dhcp_tag_name ( tag ), strerror ( rc ) );
+                       return rc;
+               }
+       }
+
+       return 0;
+}