[BusyBox-cvs] busybox/archival ar.c,1.46,1.47

[Glenn McGrath]

Update of /var/cvs/busybox/archival
In directory nail:/tmp/cvs-serv14550/archival

Modified Files:
	ar.c 
Log Message:
Use bb_getopt_ulflags, saves some space, better argument checking.
Remove ar specific extraction code, always use common extraction code.


Index: ar.c
===================================================================
RCS file: /var/cvs/busybox/archival/ar.c,v
retrieving revision 1.46
retrieving revision 1.47
diff -u -d -r1.46 -r1.47
--- ar.c	5 Dec 2003 02:51:21 -0000	1.46
+++ ar.c	4 Jan 2004 10:28:22 -0000	1.47
@@ -21,20 +21,20 @@
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  *
- * There is no signle standard to adhere to so ar may not portable
+ * There is no single standard to adhere to so ar may not portable
  * between different systems
  * http://www.unix-systems.org/single_unix_specification_v2/xcu/ar.html
  */
-#include <sys/types.h>
+
 #include <fcntl.h>
-#include <fnmatch.h>
+#include <getopt.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <time.h>
 #include <utime.h>
 #include <unistd.h>
-#include <getopt.h>
+
 #include "unarchive.h"
 #include "busybox.h"
 
@@ -50,87 +50,50 @@
 	printf("%s %d/%d%7d %s %s\n", &mode[1], file_header->uid, file_header->gid, (int) file_header->size, &mtime[4], file_header->name);
 }
 
-#if !defined CONFIG_TAR && !defined CONFIG_DPKG_DEB && !defined CONFIG_CPIO
-/* This is simpler than data_extract_all */
-static void data_extract_regular_file(archive_handle_t *archive_handle)
-{
-	file_header_t *file_header;
-	int dst_fd;
-
-	file_header = archive_handle->file_header;
-	dst_fd = bb_xopen(file_header->name, O_WRONLY | O_CREAT);
-	bb_copyfd_size(archive_handle->src_fd, dst_fd, file_header->size);	
-	close(dst_fd);
-
-	chmod(file_header->name, file_header->mode);
-	chown(file_header->name, file_header->uid, file_header->gid);
-
-	if (archive_handle->flags & ARCHIVE_PRESERVE_DATE) {
-		struct utimbuf t;
-		t.actime = t.modtime = file_header->mtime;
-		utime(file_header->name, &t);
-	}
-
-	return;
-}
-#endif
+#define AR_CTX_PRINT	1
+#define AR_CTX_LIST		2
+#define AR_CTX_EXTRACT	4
+#define AR_OPT_PRESERVE_DATE	8
+#define AR_OPT_VERBOSE			16
 
 extern int ar_main(int argc, char **argv)
 {
 	archive_handle_t *archive_handle;
-	int opt;
-
-#if !defined CONFIG_DPKG_DEB && !defined CONFIG_DPKG
+	unsigned long opt;
 	char magic[8];
-#endif
+
 	archive_handle = init_handle();
 
-	while ((opt = getopt(argc, argv, "covtpxX")) != -1) {
-		switch (opt) {
-		case 'p':	/* print */
-			archive_handle->action_data = data_extract_to_stdout;
-			break;
-		case 't':	/* list contents */
-			archive_handle->action_header = header_list;
-			break;
-		case 'X':
-			archive_handle->action_header = header_verbose_list_ar;
-		case 'x':	/* extract */
-#if defined CONFIG_TAR || defined CONFIG_DPKG_DEB || defined CONFIG_CPIO
-			archive_handle->action_data = data_extract_all;
-#else
-			archive_handle->action_data = data_extract_regular_file;
-#endif
-			break;
-		/* Modifiers */
-		case 'o':	/* preserve original dates */
-			archive_handle->flags |= ARCHIVE_PRESERVE_DATE;
-			break;
-		case 'v':	/* verbose */
-			archive_handle->action_header = header_verbose_list_ar;
-			break;
-		default:
-			bb_show_usage();
-		}
-	}
- 
-	/* check the src filename was specified */
-	if (optind == argc) {
+	bb_opt_complementaly = "p~tx:t~px:x~pt";
+	opt = bb_getopt_ulflags(argc, argv, "ptxov");
+
+	if ((opt & 0x80000000UL) || (optind == argc)) {
 		bb_show_usage();
 	}
 
+	if (opt & AR_CTX_PRINT) {
+		archive_handle->action_data = data_extract_to_stdout;
+	}
+	if (opt & AR_CTX_LIST) {
+		archive_handle->action_header = header_list;
+	}
+	if (opt & AR_CTX_EXTRACT) {
+		archive_handle->action_data = data_extract_all;
+	}
+	if (opt & AR_OPT_PRESERVE_DATE) {
+		archive_handle->flags |= ARCHIVE_PRESERVE_DATE;
+	}
+	if (opt & AR_OPT_VERBOSE) {
+		archive_handle->action_header = header_verbose_list_ar;
+	}
+
 	archive_handle->src_fd = bb_xopen(argv[optind++], O_RDONLY);
 
-	/* TODO: This is the same as in tar, seperate function ? */
 	while (optind < argc) {
 		archive_handle->filter = filter_accept_list;
-		archive_handle->accept = llist_add_to(archive_handle->accept, argv[optind]);
-		optind++;
+		archive_handle->accept = llist_add_to(archive_handle->accept, argv[optind++]);
 	}
 
-#if defined CONFIG_DPKG_DEB || defined CONFIG_DPKG
-	unpack_ar_archive(archive_handle);
-#else
 	archive_xread_all(archive_handle, magic, 7);
 	if (strncmp(magic, "!<arch>", 7) != 0) {
 		bb_error_msg_and_die("Invalid ar magic");
@@ -138,7 +101,6 @@
 	archive_handle->offset += 7;
 
 	while (get_header_ar(archive_handle) == EXIT_SUCCESS);
-#endif
 
 	return EXIT_SUCCESS;
 }