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