Use bb_getopt_ulflags, save 100 bytes and strict argument checking.

This commit is contained in:
Glenn L McGrath 2004-01-07 09:24:06 +00:00
parent 9d65ab2beb
commit 1f28b90a9c

View File

@ -12,24 +12,32 @@
* You should have received a copy of the GNU General Public License * You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software * along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
*/ */
#include <fcntl.h> #include <fcntl.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <unistd.h> #include <unistd.h>
#include <getopt.h> #include <getopt.h>
#include "unarchive.h" #include "unarchive.h"
#include "busybox.h" #include "busybox.h"
#define DPKG_DEB_OPT_CONTENTS 1
#define DPKG_DEB_OPT_CONTROL 2
#define DPKG_DEB_OPT_FIELD 4
#define DPKG_DEB_OPT_EXTRACT 8
#define DPKG_DEB_OPT_EXTRACT_VERBOSE 16
extern int dpkg_deb_main(int argc, char **argv) extern int dpkg_deb_main(int argc, char **argv)
{ {
archive_handle_t *ar_archive; archive_handle_t *ar_archive;
archive_handle_t *tar_archive; archive_handle_t *tar_archive;
int opt = 0;
#ifndef CONFIG_FEATURE_DPKG_DEB_EXTRACT_ONLY
llist_t *control_tar_llist = NULL; llist_t *control_tar_llist = NULL;
#endif unsigned long opt;
char *extract_dir = NULL;
short argcount = 1;
/* Setup the tar archive handle */ /* Setup the tar archive handle */
tar_archive = init_handle(); tar_archive = init_handle();
@ -40,53 +48,47 @@ extern int dpkg_deb_main(int argc, char **argv)
#ifdef CONFIG_FEATURE_DEB_TAR_GZ #ifdef CONFIG_FEATURE_DEB_TAR_GZ
ar_archive->accept = llist_add_to(NULL, "data.tar.gz"); ar_archive->accept = llist_add_to(NULL, "data.tar.gz");
# ifndef CONFIG_FEATURE_DPKG_DEB_EXTRACT_ONLY
control_tar_llist = llist_add_to(NULL, "control.tar.gz"); control_tar_llist = llist_add_to(NULL, "control.tar.gz");
# endif
#endif #endif
#ifdef CONFIG_FEATURE_DEB_TAR_BZ2 #ifdef CONFIG_FEATURE_DEB_TAR_BZ2
ar_archive->accept = llist_add_to(ar_archive->accept, "data.tar.bz2"); ar_archive->accept = llist_add_to(ar_archive->accept, "data.tar.bz2");
# ifndef CONFIG_FEATURE_DPKG_DEB_EXTRACT_ONLY
control_tar_llist = llist_add_to(control_tar_llist, "control.tar.bz2"); control_tar_llist = llist_add_to(control_tar_llist, "control.tar.bz2");
# endif
#endif #endif
#ifndef CONFIG_FEATURE_DPKG_DEB_EXTRACT_ONLY bb_opt_complementaly = "c~efXx:e~cfXx:f~ceXx:X~cefx:x~cefX";
while ((opt = getopt(argc, argv, "cefXx")) != -1) { opt = bb_getopt_ulflags(argc, argv, "cefXx");
#else
while ((opt = getopt(argc, argv, "x")) != -1) { if (opt & DPKG_DEB_OPT_CONTENTS) {
#endif tar_archive->action_header = header_verbose_list;
switch (opt) { }
#ifndef CONFIG_FEATURE_DPKG_DEB_EXTRACT_ONLY if (opt & DPKG_DEB_OPT_CONTROL) {
case 'c': ar_archive->accept = control_tar_llist;
tar_archive->action_header = header_verbose_list; tar_archive->action_data = data_extract_all;
break; if (optind + 1 == argc) {
case 'e': extract_dir = "./DEBIAN";
ar_archive->accept = control_tar_llist; } else {
tar_archive->action_data = data_extract_all; argcount++;
break;
case 'f':
/* Print the entire control file
* it should accept a second argument which specifies a
* specific field to print */
ar_archive->accept = control_tar_llist;
tar_archive->accept = llist_add_to(NULL, "./control");;
tar_archive->filter = filter_accept_list;
tar_archive->action_data = data_extract_to_stdout;
break;
case 'X':
tar_archive->action_header = header_list;
#endif
case 'x':
tar_archive->action_data = data_extract_all;
break;
default:
bb_show_usage();
} }
} }
if (opt & DPKG_DEB_OPT_FIELD) {
/* Print the entire control file
* it should accept a second argument which specifies a
* specific field to print */
ar_archive->accept = control_tar_llist;
tar_archive->accept = llist_add_to(NULL, "./control");;
tar_archive->filter = filter_accept_list;
tar_archive->action_data = data_extract_to_stdout;
}
if (opt & DPKG_DEB_OPT_EXTRACT) {
tar_archive->action_header = header_list;
}
if (opt & (DPKG_DEB_OPT_EXTRACT_VERBOSE | DPKG_DEB_OPT_EXTRACT)) {
tar_archive->action_data = data_extract_all;
argcount = 2;
}
if (optind + 2 < argc) { if ((optind + argcount != argc) || (opt & 0x80000000UL)) {
bb_show_usage(); bb_show_usage();
} }
@ -94,9 +96,13 @@ extern int dpkg_deb_main(int argc, char **argv)
/* Workout where to extract the files */ /* Workout where to extract the files */
/* 2nd argument is a dir name */ /* 2nd argument is a dir name */
mkdir(argv[optind], 0777); if (argv[optind]) {
chdir(argv[optind]); extract_dir = argv[optind];
}
if (extract_dir) {
mkdir(extract_dir, 0777);
chdir(extract_dir);
}
unpack_ar_archive(ar_archive); unpack_ar_archive(ar_archive);
/* Cleanup */ /* Cleanup */
@ -104,4 +110,3 @@ extern int dpkg_deb_main(int argc, char **argv)
return(EXIT_SUCCESS); return(EXIT_SUCCESS);
} }