Use getopt_ulflags

1 files changed, 50 insertions, 58 deletions

diff --git a/networking/wget.c b/networking/wget.c
index 08026915e..3e74ed357 100644
--- a/networking/wget.c
+++ b/networking/wget.c
@@ -142,9 +142,29 @@ char *base64enc(char *p, char *buf, int len) {
 }
 #endif
 
+#define WGET_OPT_CONTINUE       1
+#define WGET_OPT_QUIET  2
+#define WGET_OPT_PASSIVE        4
+#define WGET_OPT_OUTNAME        8
+#define WGET_OPT_HEADER 16
+#define WGET_OPT_PREFIX 32
+#define WGET_OPT_PROXY  64
+
+static const struct option wget_long_options[] = {
+        { "continue",        0, NULL, 'c' },
+        { "quiet",           0, NULL, 'q' },
+        { "passive-ftp",     0, NULL, 139 },
+        { "output-document", 1, NULL, 'O' },
+        { "header",              1, NULL, 131 },
+        { "directory-prefix",1, NULL, 'P' },
+        { "proxy",           1, NULL, 'Y' },
+        { 0,                 0, 0, 0 }
+};
+
 int wget_main(int argc, char **argv)
 {
         int n, try=5, status;
+        unsigned long opt;
         int port;
         char *proxy = 0;
         char *dir_prefix=NULL;
@@ -153,9 +173,9 @@ int wget_main(int argc, char **argv)
         char extra_headers[1024];
         char *extra_headers_ptr = extra_headers;
         int extra_headers_left = sizeof(extra_headers);
-        int which_long_opt = 0, option_index = -1;
         struct host_info server, target;
         struct sockaddr_in s_in;
+        llist_t *headers_llist = NULL;
 
         FILE *sfp = NULL;                       /* socket to web/ftp server                     */
         FILE *dfp = NULL;                       /* socket to ftp server (data)          */
@@ -166,68 +186,40 @@ int wget_main(int argc, char **argv)
         FILE *output;                           /* socket to web server                         */
         int quiet_flag = FALSE;         /* Be verry, verry quiet...                     */
         int noproxy = 0;            /* Use proxies if env vars are set  */
+        char *proxy_flag = "on";        /* Use proxies if env vars are set  */
 
-#define LONG_HEADER    1
-#define LONG_PASSIVE   2
-
-        struct option long_options[] = {
-                { "continue",        0, NULL, 'c' },
-                { "quiet",           0, NULL, 'q' },
-                { "output-document", 1, NULL, 'O' },
-                { "header",              1, &which_long_opt, LONG_HEADER },
-                { "proxy",           1, NULL, 'Y' },
-                { "passive-ftp",     0, &which_long_opt, LONG_PASSIVE },
-                { 0,                 0, 0, 0 }
-        };
         /*
          * Crack command line.
          */
-        while ((n = getopt_long(argc, argv, "cqO:P:Y:", long_options, &option_index)) != EOF) {
-                switch (n) {
-                case 'c':
-                        ++do_continue;
-                        break;
-                case 'P':
-                        dir_prefix = optarg;
-                        break;
-                case 'q':
-                        quiet_flag = TRUE;
-                        break;
-                case 'O':
-                        /* can't set fname_out to NULL if outputting to stdout, because
-                         * this gets interpreted as the auto-gen output filename
-                         * case below  - tausq@debian.org
-                         */
-                        fname_out = optarg;
-                        break;
-                case 'Y':
-                        if (strcmp(optarg, "off") == 0)
-                                noproxy=1;      
-                        break;
-                case 0:
-                        switch (which_long_opt) {
-                                case LONG_HEADER: {
-                                        int arglen = strlen(optarg);
-                                        if(extra_headers_left - arglen - 2 <= 0)
-                                                bb_error_msg_and_die("extra_headers buffer too small(need %i)", extra_headers_left - arglen);
-                                        strcpy(extra_headers_ptr, optarg);
-                                        extra_headers_ptr += arglen;
-                                        extra_headers_left -= ( arglen + 2 );
-                                        *extra_headers_ptr++ = '\r';
-                                        *extra_headers_ptr++ = '\n';
-                                        *(extra_headers_ptr + 1) = 0;
-                                        break;
-                                }
-                                case LONG_PASSIVE:
-                                        // ignore -- we always use passive mode
-                                        break;
-                        }
-                        break;
-                default:
-                        bb_show_usage();
+        bb_opt_complementaly = "\203*";
+        bb_applet_long_options = wget_long_options;
+        opt = bb_getopt_ulflags(argc, argv, "cq\213O:\203:P:Y:", &fname_out, &headers_llist, &dir_prefix, &proxy_flag);
+        if (opt & WGET_OPT_CONTINUE) {
+                ++do_continue;
+        }
+        if (opt & WGET_OPT_QUIET) {
+                quiet_flag = TRUE;
+        }
+        if (strcmp(proxy_flag, "on") == 0) {
+                /* Use the proxy if necessary. */
+                proxy = getenv(target.is_ftp ? "ftp_proxy" : "http_proxy");
+                if (proxy)
+                        parse_url(bb_xstrdup(proxy), &server);
+        }
+        if (opt & WGET_OPT_HEADER) {
+                while (headers_llist) {
+                        int arglen = strlen(headers_llist->data);
+                        if (extra_headers_left - arglen - 2 <= 0)
+                                bb_error_msg_and_die("extra_headers buffer too small(need %i)", extra_headers_left - arglen);
+                        strcpy(extra_headers_ptr, headers_llist->data);
+                        extra_headers_ptr += arglen;
+                        extra_headers_left -= ( arglen + 2 );
+                        *extra_headers_ptr++ = '\r';
+                        *extra_headers_ptr++ = '\n';
+                        *(extra_headers_ptr + 1) = 0;
+                        headers_llist = headers_llist->link;
                 }
         }
-
         if (argc - optind != 1)
                         bb_show_usage();
 
@@ -850,7 +842,7 @@ progressmeter(int flag)
  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  * SUCH DAMAGE.
  *
- *      $Id: wget.c,v 1.61 2003/10/31 09:31:43 andersen Exp $
+ *      $Id: wget.c,v 1.62 2003/12/19 12:08:56 bug1 Exp $
  */