tests/test-dhcp-client.c

   1 /*
   2  * Copyright (c) 2008, 2009 Nicira Networks.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <config.h>
  18 #include "dhcp-client.h"
  19 #include <arpa/inet.h>
  20 #include <getopt.h>
  21 #include <stdlib.h>
  22 #include <limits.h>
  23 #include "command-line.h"
  24 #include "dhcp.h"
  25 #include "fatal-signal.h"
  26 #include "fault.h"
  27 #include "poll-loop.h"
  28 #include "util.h"
  29 #include "vlog.h"
  30
  31 /* --request-ip: IP address to request from server.  If zero, then do not
  32  * request a specific IP address. */
  33 static struct in_addr request_ip;
  34
  35 /* --vendor-class: Vendor class string to include in request.  If null, no
  36  * vendor class string is included. */
  37 static const char *vendor_class;
  38
  39 /* --no-resolv-conf: Update /etc/resolv.conf to match DHCP reply? */
  40 static bool update_resolv_conf = true;
  41
  42 static void parse_options(int argc, char *argv[]);
  43 static void usage(void);
  44 static void release(void *cli_);
  45 static void modify_dhcp_request(struct dhcp_msg *, void *aux);
  46
  47 int
  48 main(int argc, char *argv[])
  49 {
  50     struct dhclient *cli;
  51     int error;
  52
  53     set_program_name(argv[0]);
  54     register_fault_handlers();
  55     vlog_init();
  56     parse_options(argc, argv);
  57
  58     argc -= optind;
  59     argv += optind;
  60     if (argc != 1) {
  61         ovs_fatal(0, "exactly one non-option argument required; "
  62                   "use --help for help");
  63     }
  64
  65     error = dhclient_create(argv[0], modify_dhcp_request, NULL, NULL, &cli);
  66     if (error) {
  67         ovs_fatal(error, "dhclient_create failed");
  68     }
  69     dhclient_init(cli, request_ip.s_addr);
  70     fatal_signal_add_hook(release, cli, true);
  71
  72     for (;;) {
  73         fatal_signal_block();
  74         dhclient_run(cli);
  75         if (dhclient_changed(cli)) {
  76             dhclient_configure_netdev(cli);
  77             if (update_resolv_conf) {
  78                 dhclient_update_resolv_conf(cli);
  79             }
  80         }
  81         dhclient_wait(cli);
  82         fatal_signal_unblock();
  83         poll_block();
  84     }
  85 }
  86
  87 static void
  88 release(void *cli_)
  89 {
  90     struct dhclient *cli = cli_;
  91     dhclient_release(cli);
  92     if (dhclient_changed(cli)) {
  93         dhclient_configure_netdev(cli);
  94     }
  95 }
  96
  97 static void
  98 modify_dhcp_request(struct dhcp_msg *msg, void *aux UNUSED)
  99 {
 100     if (vendor_class) {
 101         dhcp_msg_put_string(msg, DHCP_CODE_VENDOR_CLASS, vendor_class);
 102     }
 103 }
 104
 105 static void
 106 parse_options(int argc, char *argv[])
 107 {
 108     enum {
 109         OPT_REQUEST_IP = UCHAR_MAX + 1,
 110         OPT_VENDOR_CLASS,
 111         OPT_NO_RESOLV_CONF
 112     };
 113     static struct option long_options[] = {
 114         {"request-ip",  required_argument, 0, OPT_REQUEST_IP },
 115         {"vendor-class", required_argument, 0, OPT_VENDOR_CLASS },
 116         {"no-resolv-conf", no_argument, 0, OPT_NO_RESOLV_CONF},
 117         {"verbose",     optional_argument, 0, 'v'},
 118         {"help",        no_argument, 0, 'h'},
 119         {"version",     no_argument, 0, 'V'},
 120         {0, 0, 0, 0},
 121     };
 122     char *short_options = long_options_to_short_options(long_options);
 123
 124     for (;;) {
 125         int c;
 126
 127         c = getopt_long(argc, argv, short_options, long_options, NULL);
 128         if (c == -1) {
 129             break;
 130         }
 131
 132         switch (c) {
 133         case OPT_REQUEST_IP:
 134             if (!inet_aton(optarg, &request_ip)) {
 135                 ovs_fatal(0,
 136                           "--request-ip argument is not a valid IP address");
 137             }
 138             break;
 139
 140         case OPT_VENDOR_CLASS:
 141             vendor_class = optarg;
 142             break;
 143
 144         case OPT_NO_RESOLV_CONF:
 145             update_resolv_conf = false;
 146             break;
 147
 148         case 'h':
 149             usage();
 150
 151         case 'V':
 152             printf("%s %s compiled "__DATE__" "__TIME__"\n",
 153                    program_name, VERSION BUILDNR);
 154             exit(EXIT_SUCCESS);
 155
 156         case 'v':
 157             vlog_set_verbosity(optarg);
 158             break;
 159
 160         case '?':
 161             exit(EXIT_FAILURE);
 162
 163         default:
 164             abort();
 165         }
 166     }
 167     free(short_options);
 168 }
 169
 170 static void
 171 usage(void)
 172 {
 173     printf("%s: standalone program for testing Open vSwitch DHCP client.\n"
 174            "usage: %s [OPTIONS] NETDEV\n"
 175            "where NETDEV is a network device (e.g. eth0).\n"
 176            "\nDHCP options:\n"
 177            "  --request-ip=IP         request specified IP address (default:\n"
 178            "                          do not request a specific IP)\n"
 179            "  --vendor-class=STRING   use STRING as vendor class; use\n"
 180            "                          OpenFlow to imitate ovs-openflowd\n"
 181            "  --no-resolv-conf        do not update /etc/resolv.conf\n",
 182            program_name, program_name);
 183     vlog_usage();
 184     printf("\nOther options:\n"
 185            "  -h, --help              display this help message\n"
 186            "  -V, --version           display version information\n");
 187     exit(EXIT_SUCCESS);
 188 }
 189