diff --git a/net/dhcp.c b/net/dhcp.c index 02dc529..768255e 100644 --- a/net/dhcp.c +++ b/net/dhcp.c @@ -22,6 +22,8 @@ #include #include +#define DHCP_DEFAULT_RETRY 20 + #define OPT_SIZE 312 /* Minimum DHCP Options size per RFC2131 - results in 576 byte pkt */ struct bootp { @@ -700,13 +702,26 @@ } late_initcall(dhcp_global_init); +static void dhcp_getenv_int(const char *name, int *i) +{ + const char* str = getenv(name); + + if (!str) + return; + + *i = simple_strtoul(str, NULL, 10); +} + static int do_dhcp(int argc, char *argv[]) { int ret, opt; + int retries = DHCP_DEFAULT_RETRY; dhcp_reset_env(); - while((opt = getopt(argc, argv, "H:v:c:u:U:")) > 0) { + dhcp_getenv_int("global.dhcp.retries", &retries); + + while((opt = getopt(argc, argv, "H:v:c:u:U:r:")) > 0) { switch(opt) { case 'H': dhcp_set_param_data(DHCP_HOSTNAME, optarg); @@ -723,9 +738,17 @@ case 'U': dhcp_set_param_data(DHCP_USER_CLASS, optarg); break; + case 'r': + retries = simple_strtoul(optarg, NULL, 10); + break; } } + if (!retries) { + printf("retries is set to zero, set it to %d\n", DHCP_DEFAULT_RETRY); + retries = DHCP_DEFAULT_RETRY; + } + dhcp_con = net_udp_new(0xffffffff, PORT_BOOTPS, dhcp_handler, NULL); if (IS_ERR(dhcp_con)) { ret = PTR_ERR(dhcp_con); @@ -746,11 +769,17 @@ while (dhcp_state != BOUND) { if (ctrlc()) break; + if (!retries) { + ret = -ETIMEDOUT; + goto out1; + } net_poll(); if (is_timeout(dhcp_start, 3 * SECOND)) { dhcp_start = get_time_ns(); printf("T "); ret = bootp_request(); + /* no need to check if retries > 0 as we check if != 0 */ + retries--; if (ret) goto out1; } @@ -762,7 +791,7 @@ if (ret) printf("dhcp failed: %s\n", strerror(-ret)); - return ret ? 1 : 0; + return ret; } BAREBOX_CMD_HELP_START(dhcp) @@ -785,7 +814,8 @@ BAREBOX_CMD_HELP_OPT ("-U ", "DHCP User class (code 77) submitted in DHCP requests. It can\n" "be used in the DHCP server's configuration to select options\n" -"(e.g. bootfile or server) which are valid for barebox clients only.\n"); +"(e.g. bootfile or server) which are valid for barebox clients only.\n") +BAREBOX_CMD_HELP_OPT ("-r ", "retry limit by default "__stringify(DHCP_DEFAULT_RETRY)"\n"); BAREBOX_CMD_HELP_END BAREBOX_CMD_START(dhcp) @@ -804,3 +834,4 @@ BAREBOX_MAGICVAR_NAMED(global_dhcp_user_class, global.dhcp.user_class, "user class to send to the DHCP server"); BAREBOX_MAGICVAR_NAMED(global_dhcp_tftp_server_name, global.dhcp.tftp_server_name, "TFTP server Name returned from DHCP request"); BAREBOX_MAGICVAR_NAMED(global_dhcp_oftree_file, global.dhcp.oftree_file, "OF tree returned from DHCP request (option 224)"); +BAREBOX_MAGICVAR_NAMED(global_dhcp_retries, global.dhcp.retries, "retry limit");