Use a switch instead of successive if (strcmp()) statments.
[oweals/busybox.git] / archival / dpkg.c
index 937726a0ce16c04944cee2f36a3ccf4a02a77e56..8b25bace34ca02db720115e7bf6ae5dfaa348508 100644 (file)
@@ -502,10 +502,24 @@ void free_package(common_node_t *node)
        }
 }
 
+/* returns the array number of the string */
+static unsigned short compare_string_array(const char *string_array[], const char *key)
+{
+       unsigned short i;
+
+       for (i = 0; string_array[i] != 0; i++) {
+               if (strcmp(string_array[i], key) == 0) {
+                       break;
+               }
+       }
+       return(i);
+}
+
 unsigned int fill_package_struct(char *control_buffer)
 {
        common_node_t *new_node = (common_node_t *) xcalloc(1, sizeof(common_node_t));
-
+       const char *field_names[] = { "Package", "Version", "Pre-Depends", "Depends",
+               "Replaces", "Provides", "Conflicts", "Suggests", "Recommends", "Enhances", 0};
        char *field_name;
        char *field_value;
        int field_start = 0;
@@ -514,42 +528,47 @@ unsigned int fill_package_struct(char *control_buffer)
 
        new_node->version = search_name_hashtable("unknown");
        while (field_start < buffer_length) {
+               unsigned short field_num;
+
                field_start += read_package_field(&control_buffer[field_start],
                                &field_name, &field_value);
 
                if (field_name == NULL) {
-                       goto fill_package_struct_cleanup; // Oh no, the dreaded goto statement !!
+                       goto fill_package_struct_cleanup; /* Oh no, the dreaded goto statement ! */
                }
 
-               if (strcmp(field_name, "Package") == 0) {
-                       new_node->name = search_name_hashtable(field_value);
-               }
-               else if (strcmp(field_name, "Version") == 0) {
-                       new_node->version = search_name_hashtable(field_value);
-               }
-               else if (strcmp(field_name, "Pre-Depends") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_PRE_DEPENDS);
-               }
-               else if (strcmp(field_name, "Depends") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_DEPENDS);
-               }
-               else if (strcmp(field_name, "Replaces") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_REPLACES);
-               }
-               else if (strcmp(field_name, "Provides") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_PROVIDES);
-               }
-               else if (strcmp(field_name, "Conflicts") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_CONFLICTS);
-               }
-               else if (strcmp(field_name, "Suggests") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_SUGGESTS);
-               }
-               else if (strcmp(field_name, "Recommends") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_RECOMMENDS);
-               }
-               else if (strcmp(field_name, "Enhances") == 0) {
-                       add_split_dependencies(new_node, field_value, EDGE_ENHANCES);
+               field_num = compare_string_array(field_names, field_name);              
+               switch(field_num) {
+                       case 0: /* Package */
+                               new_node->name = search_name_hashtable(field_value);
+                               break;
+                       case 1: /* Version */
+                               new_node->version = search_name_hashtable(field_value);
+                               break;
+                       case 2: /* Pre-Depends */
+                               add_split_dependencies(new_node, field_value, EDGE_PRE_DEPENDS);
+                               break;
+                       case 3: /* Depends */
+                               add_split_dependencies(new_node, field_value, EDGE_DEPENDS);
+                               break;
+                       case 4: /* Replaces */
+                               add_split_dependencies(new_node, field_value, EDGE_REPLACES);
+                               break;
+                       case 5: /* Provides */
+                               add_split_dependencies(new_node, field_value, EDGE_PROVIDES);
+                               break;
+                       case 6: /* Conflicts */
+                               add_split_dependencies(new_node, field_value, EDGE_CONFLICTS);
+                               break;
+                       case 7: /* Suggests */
+                               add_split_dependencies(new_node, field_value, EDGE_SUGGESTS);
+                               break;
+                       case 8: /* Recommends */
+                               add_split_dependencies(new_node, field_value, EDGE_RECOMMENDS);
+                               break;
+                       case 9: /* Enhances */
+                               add_split_dependencies(new_node, field_value, EDGE_ENHANCES);
+                               break;
                }
 fill_package_struct_cleanup:
                if (field_name) {
@@ -1311,32 +1330,73 @@ void purge_package(const unsigned int package_num)
        set_status(status_num, "not-installed", 3);
 }
 
-static archive_handle_t *deb_extract(const char *filename, const llist_t *accept_list,
-       const char *tar_gz_name, char *prefix, void (*deb_action_data)(struct archive_handle_s *))
+static archive_handle_t *init_archive_deb_ar(const char *filename)
 {
-       archive_handle_t *ar_archive;
-       archive_handle_t *tar_archive;
+       archive_handle_t *ar_handle;
+
+       /* Setup an ar archive handle that refers to the gzip sub archive */    
+       ar_handle = init_handle();
+       ar_handle->filter = filter_accept_list_reassign;
+       ar_handle->src_fd = xopen(filename, O_RDONLY);
+
+       return(ar_handle);      
+}
+
+static void init_archive_deb_control(archive_handle_t *ar_handle)
+{
+       archive_handle_t *tar_handle;
 
        /* Setup the tar archive handle */
-       tar_archive = init_handle();
-       tar_archive->filter = filter_accept_reject_list;
-       tar_archive->action_data = deb_action_data;
-       tar_archive->buffer = prefix;
-       if (tar_gz_name) {
-               tar_archive->accept = add_to_list(NULL, tar_gz_name);
-       }
+       tar_handle = init_handle();
+       tar_handle->filter = filter_accept_list;
+       tar_handle->src_fd = ar_handle->src_fd;
 
-       /* Setup an ar archive handle that refers to the gzip sub archive */    
-       ar_archive = init_handle();
-       ar_archive->sub_archive = tar_archive;
-       ar_archive->filter = filter_accept_list_reassign;
-       ar_archive->accept = accept_list;
+       /* We dont care about data.tar.* or debian-binary, just control.tar.* */
+#ifdef CONFIG_FEATURE_DEB_TAR_GZ
+       ar_handle->accept = add_to_list(NULL, "control.tar.gz");
+#endif
+#ifdef CONFIG_FEATURE_DEB_TAR_BZ2
+       ar_handle->accept = add_to_list(ar_handle->accept, "control.tar.bz2");
+#endif
+
+       /* Assign the tar handle as a subarchive of the ar handle */
+       ar_handle->sub_archive = tar_handle;
+
+       return; 
+}
+
+static void init_archive_deb_data(archive_handle_t *ar_handle)
+{
+       archive_handle_t *tar_handle;
 
-       tar_archive->src_fd = ar_archive->src_fd = xopen(filename, O_RDONLY);
-       unpack_ar_archive(ar_archive);
-       close(ar_archive->src_fd);
+       /* Setup the tar archive handle */
+       tar_handle = init_handle();
+       tar_handle->filter = filter_accept_all;
+       tar_handle->src_fd = ar_handle->src_fd;
 
-       return(ar_archive->sub_archive);
+       /* We dont care about data.tar.* or debian-binary, just control.tar.* */
+#ifdef CONFIG_FEATURE_DEB_TAR_GZ
+       tar_handle->accept = add_to_list(NULL, "data.tar.gz");
+#endif
+#ifdef CONFIG_FEATURE_DEB_TAR_BZ2
+       tar_handle->accept = add_to_list(ar_handle->accept, "data.tar.bz2");
+#endif
+
+       /* Assign the tar handle as a subarchive of the ar handle */
+       ar_handle->sub_archive = tar_handle;
+
+       return; 
+}
+
+static char *deb_extract_control_file_to_buffer(archive_handle_t *ar_handle, const llist_t *accept)
+{
+       ar_handle->sub_archive->action_data = data_extract_to_buffer;
+       ar_handle->sub_archive->accept = accept;
+
+       unpack_ar_archive(ar_handle);
+       close(ar_handle->src_fd);
+
+       return(ar_handle->sub_archive->buffer);
 }
 
 static void data_extract_all_prefix(archive_handle_t *archive_handle)
@@ -1361,18 +1421,6 @@ static void unpack_package(deb_file_t *deb_file)
        char *info_prefix;
        archive_handle_t *archive_handle;
        FILE *out_stream;
-       const llist_t *control_list = NULL;
-       const llist_t *data_list = NULL;
-
-#ifdef CONFIG_FEATURE_DEB_TAR_GZ
-       data_list = add_to_list(NULL, "data.tar.gz");
-       control_list = add_to_list(NULL, "control.tar.gz");
-#endif
-
-#ifdef CONFIG_FEATURE_DEB_TAR_BZ2
-       data_list = add_to_list(data_list, "data.tar.bz2");
-       control_list = add_to_list(control_list, "control.tar.bz2");
-#endif
 
        /* If existing version, remove it first */
        if (strcmp(name_hashtable[get_status(status_num, 3)], "installed") == 0) {
@@ -1388,8 +1436,12 @@ static void unpack_package(deb_file_t *deb_file)
        /* Extract control.tar.gz to /var/lib/dpkg/info/<package>.filename */
        info_prefix = (char *) xmalloc(strlen(package_name) + 20 + 4 + 2);
        sprintf(info_prefix, "/var/lib/dpkg/info/%s.", package_name);
+       archive_handle = init_archive_deb_ar(deb_file->filename);
+       init_archive_deb_control(archive_handle);
+       archive_handle->sub_archive->action_data = data_extract_all_prefix;
+       archive_handle->sub_archive->buffer = info_prefix;
+       unpack_ar_archive(archive_handle);
 
-       deb_extract(deb_file->filename, control_list, NULL, info_prefix, data_extract_all_prefix);
        /* Run the preinst prior to extracting */
        if (run_package_script(package_name, "preinst") != 0) {
                /* when preinst returns exit code != 0 then quit installation process */
@@ -1397,7 +1449,9 @@ static void unpack_package(deb_file_t *deb_file)
        }       
 
        /* Extract data.tar.gz to the root directory */
-       archive_handle = deb_extract(deb_file->filename, data_list, NULL, NULL, data_extract_all);
+       archive_handle = init_archive_deb_ar(deb_file->filename);
+       init_archive_deb_data(archive_handle);
+       unpack_ar_archive(archive_handle);
 
        /* Create the list file */
        strcat(info_prefix, "list");
@@ -1503,18 +1557,15 @@ int dpkg_main(int argc, char **argv)
                        archive_handle_t *archive_handle;
                        const llist_t *control_list = NULL;
 
-#ifdef CONFIG_FEATURE_DEB_TAR_GZ
-                       control_list = add_to_list(NULL, "control.tar.gz");
-#endif
-#ifdef CONFIG_FEATURE_DEB_TAR_BZ2
-                       control_list = add_to_list(control_list, "control.tar.bz2");
-#endif
-                       deb_file[deb_count]->filename = xstrdup(argv[optind]);
-                       archive_handle = deb_extract(argv[optind], control_list, "./control", NULL, data_extract_to_buffer);
-                       deb_file[deb_count]->control_file = archive_handle->buffer;
+                       /* Extract the control file */
+                       control_list = add_to_list(NULL, "./control");
+                       archive_handle = init_archive_deb_ar(argv[optind]);
+                       init_archive_deb_control(archive_handle);
+                       deb_file[deb_count]->control_file = deb_extract_control_file_to_buffer(archive_handle, control_list);
                        if (deb_file[deb_count]->control_file == NULL) {
                                error_msg_and_die("Couldnt extract control file");
                        }
+                       deb_file[deb_count]->filename = xstrdup(argv[optind]);
                        package_num = fill_package_struct(deb_file[deb_count]->control_file);
 
                        if (package_num == -1) {
@@ -1522,6 +1573,7 @@ int dpkg_main(int argc, char **argv)
                                continue;
                        }
                        deb_file[deb_count]->package = (unsigned int) package_num;
+
                        /* Add the package to the status hashtable */
                        if ((dpkg_opt & dpkg_opt_unpack) || (dpkg_opt & dpkg_opt_install)) {
                                status_node = (status_node_t *) xmalloc(sizeof(status_node_t));