b44433e7daf980ad9e9bec7d1bf8057798c45008
[people/mcb30/gpxe.git] / src / hci / commands / dhcp_cmd.c
1 /*
2  * Copyright (C) 2007 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 <stdio.h>
20 #include <stdint.h>
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <errno.h>
24 #include <stddef.h>
25 #include <string.h>
26 #include <assert.h>
27 #include <getopt.h>
28 #include <gpxe/netdevice.h>
29 #include <gpxe/in.h>
30 #include <gpxe/command.h>
31 #include <usr/dhcpmgmt.h>
32
33 /** @file
34  *
35  * DHCP management commands
36  *
37  */
38
39 /**
40  * "dhcp" command syntax message
41  *
42  * @v argv              Argument list
43  */
44 static void dhcp_syntax ( char **argv ) {
45         printf ( "Usage:\n"
46                  "  %s <interface>\n"
47                  "\n"
48                  "Configure a network interface using DHCP\n",
49                  argv[0] );
50 }
51
52 /**
53  * The "dhcp" command
54  *
55  * @v argc              Argument count
56  * @v argv              Argument list
57  * @ret rc              Exit code
58  */
59 static int dhcp_exec ( int argc, char **argv ) {
60         static struct option longopts[] = {
61                 { "help", 0, NULL, 'h' },
62                 { NULL, 0, NULL, 0 },
63         };
64         const char *netdev_txt;
65         struct net_device *netdev;
66         int c;
67         int rc;
68
69         /* Parse options */
70         while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){
71                 switch ( c ) {
72                 case 'h':
73                         /* Display help text */
74                 default:
75                         /* Unrecognised/invalid option */
76                         dhcp_syntax ( argv );
77                         return 1;
78                 }
79         }
80
81         /* Need exactly one interface name remaining after the options */
82         if ( optind != ( argc - 1 ) ) {
83                 dhcp_syntax ( argv );
84                 return 1;
85         }
86         netdev_txt = argv[optind];
87
88         /* Parse arguments */
89         netdev = find_netdev ( netdev_txt );
90         if ( ! netdev ) {
91                 printf ( "No such interface: %s\n", netdev_txt );
92                 return 1;
93         }
94
95         /* Perform DHCP */
96         if ( ( rc = dhcp ( netdev ) ) != 0 ) {
97                 printf ( "Could not configure %s: %s\n", netdev->name,
98                          strerror ( rc ) );
99                 return 1;
100         }
101
102         return 0;
103 }
104
105 /**
106  * "pxebs" command syntax message
107  *
108  * @v argv              Argument list
109  */
110 static void pxebs_syntax ( char **argv ) {
111         printf ( "Usage:\n"
112                  "  %s <interface> <discovery_ip> <server_type>\n"
113                  "\n"
114                  "Perform PXE Boot Server discovery\n",
115                  argv[0] );
116 }
117
118 /**
119  * The "pxebs" command
120  *
121  * @v argc              Argument count
122  * @v argv              Argument list
123  * @ret rc              Exit code
124  */
125 static int pxebs_exec ( int argc, char **argv ) {
126         static struct option longopts[] = {
127                 { "help", 0, NULL, 'h' },
128                 { NULL, 0, NULL, 0 },
129         };
130         const char *netdev_txt;
131         const char *pxe_server_txt;
132         const char *pxe_type_txt;
133         struct net_device *netdev;
134         struct in_addr pxe_server;
135         unsigned int pxe_type;
136         char *end;
137         int c;
138         int rc;
139
140         /* Parse options */
141         while ( ( c = getopt_long ( argc, argv, "h", longopts, NULL ) ) >= 0 ){
142                 switch ( c ) {
143                 case 'h':
144                         /* Display help text */
145                 default:
146                         /* Unrecognised/invalid option */
147                         pxebs_syntax ( argv );
148                         return 1;
149                 }
150         }
151
152         /* Need exactly one interface name remaining after the options */
153         if ( optind != ( argc - 3 ) ) {
154                 pxebs_syntax ( argv );
155                 return 1;
156         }
157         netdev_txt = argv[optind];
158         pxe_server_txt = argv[ optind + 1 ];
159         pxe_type_txt = argv[ optind + 2 ];
160
161         /* Parse arguments */
162         netdev = find_netdev ( netdev_txt );
163         if ( ! netdev ) {
164                 printf ( "No such interface: %s\n", netdev_txt );
165                 return 1;
166         }
167         if ( inet_aton ( pxe_server_txt, &pxe_server ) == 0 ) {
168                 printf ( "Bad discovery IP address: %s\n", pxe_server_txt );
169                 return 1;
170         }
171         pxe_type = strtoul ( pxe_type_txt, &end, 0 );
172         if ( *end ) {
173                 printf ( "Bad server type: %s\n", pxe_type_txt );
174                 return 1;
175         }
176
177         /* Perform Boot Server Discovery */
178         if ( ( rc = pxebs ( netdev, pxe_server, pxe_type ) ) != 0 ) {
179                 printf ( "Could not discover boot server on %s: %s\n",
180                          netdev->name, strerror ( rc ) );
181                 return 1;
182         }
183
184         return 0;
185 }
186
187 /** DHCP management commands */
188 struct command dhcp_commands[] __command = {
189         {
190                 .name = "dhcp",
191                 .exec = dhcp_exec,
192         },
193         {
194                 .name = "pxebs",
195                 .exec = pxebs_exec,
196         },
197 };