blob: a2e3306ac61844a6005da73603dcb3753b3afbaf [file] [log] [blame]
Eric Andersen86ab8a32000-06-02 03:21:42 +00001/* vi: set sw=4 ts=4: */
2/*
Eric Andersenc7bda1c2004-03-15 08:29:22 +00003 * Mini ar implementation for busybox
Eric Andersen86ab8a32000-06-02 03:21:42 +00004 *
5 * Copyright (C) 2000 by Glenn McGrath
Eric Andersenc7bda1c2004-03-15 08:29:22 +00006 *
Eric Andersen86ab8a32000-06-02 03:21:42 +00007 * Based in part on BusyBox tar, Debian dpkg-deb and GNU ar.
8 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02009 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
Eric Andersen86ab8a32000-06-02 03:21:42 +000010 *
Alexander Shishkin535584c2010-03-15 15:38:09 +010011 * Archive creation support:
12 * Copyright (C) 2010 Nokia Corporation. All rights reserved.
13 * Written by Alexander Shishkin.
14 *
Glenn L McGrath930453b2004-01-04 10:28:22 +000015 * There is no single standard to adhere to so ar may not portable
Glenn L McGrath7ca04f32002-09-25 02:47:48 +000016 * between different systems
17 * http://www.unix-systems.org/single_unix_specification_v2/xcu/ar.html
Eric Andersen86ab8a32000-06-02 03:21:42 +000018 */
Glenn L McGrath930453b2004-01-04 10:28:22 +000019
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000020#include "libbb.h"
Denys Vlasenko833d4e72010-11-03 02:38:31 +010021#include "archive.h"
Alexander Shishkin535584c2010-03-15 15:38:09 +010022#include "ar.h"
23
24#if ENABLE_FEATURE_AR_CREATE
25/* filter out entries with same names as specified on the command line */
26static char FAST_FUNC filter_replaceable(archive_handle_t *handle)
27{
28 if (find_list_entry(handle->accept, handle->file_header->name))
29 return EXIT_FAILURE;
30
31 return EXIT_SUCCESS;
32}
33
34static void output_ar_header(archive_handle_t *handle)
35{
36 /* GNU ar 2.19.51.0.14 creates malformed archives
37 * if input files are >10G. It also truncates files >4GB
38 * (uses "size mod 4G"). We abort in this case:
39 * We could add support for up to 10G files, but this is unlikely to be useful.
40 * Note that unpacking side limits all fields to "unsigned int" data type,
41 * and treats "all ones" as an error indicator. Thus max we allow here is UINT_MAX-1.
42 */
43 enum {
44 /* for 2nd field: mtime */
45 MAX11CHARS = UINT_MAX > 0xffffffff ? (unsigned)99999999999 : UINT_MAX-1,
46 /* for last field: filesize */
47 MAX10CHARS = UINT_MAX > 0xffffffff ? (unsigned)9999999999 : UINT_MAX-1,
48 };
49
50 struct file_header_t *fh = handle->file_header;
51
52 if (handle->offset & 1) {
53 xwrite(handle->src_fd, "\n", 1);
54 handle->offset++;
55 }
56
57 /* Careful! The widths should be exact. Fields must be separated */
58 if (sizeof(off_t) > 4 && fh->size > (off_t)MAX10CHARS) {
59 bb_error_msg_and_die("'%s' is bigger than ar can handle", fh->name);
60 }
61 fdprintf(handle->src_fd, "%-16.16s%-12lu%-6u%-6u%-8o%-10"OFF_FMT"u`\n",
62 fh->name,
63 (sizeof(time_t) > 4 && fh->mtime > MAX11CHARS) ? (long)0 : (long)fh->mtime,
64 fh->uid > 99999 ? 0 : (int)fh->uid,
65 fh->gid > 99999 ? 0 : (int)fh->gid,
66 (int)fh->mode & 07777777,
67 fh->size
68 );
69
70 handle->offset += AR_HEADER_LEN;
71}
72
73/*
74 * when replacing files in an existing archive, copy from the the
75 * original archive those files that are to be left intact
76 */
77static void FAST_FUNC copy_data(archive_handle_t *handle)
78{
79 archive_handle_t *out_handle = handle->ar__out;
80 struct file_header_t *fh = handle->file_header;
81
82 out_handle->file_header = fh;
83 output_ar_header(out_handle);
84
85 bb_copyfd_exact_size(handle->src_fd, out_handle->src_fd, fh->size);
86 out_handle->offset += fh->size;
87}
88
89static int write_ar_header(archive_handle_t *handle)
90{
91 char *fn;
92 char fn_h[17]; /* 15 + "/" + NUL */
93 struct stat st;
94 int fd;
95
96 fn = llist_pop(&handle->accept);
97 if (!fn)
98 return -1;
99
100 xstat(fn, &st);
101
102 handle->file_header->mtime = st.st_mtime;
103 handle->file_header->uid = st.st_uid;
104 handle->file_header->gid = st.st_gid;
105 handle->file_header->mode = st.st_mode;
106 handle->file_header->size = st.st_size;
107 handle->file_header->name = fn_h;
108//TODO: if ENABLE_FEATURE_AR_LONG_FILENAMES...
109 sprintf(fn_h, "%.15s/", bb_basename(fn));
110
111 output_ar_header(handle);
112
113 fd = xopen(fn, O_RDONLY);
114 bb_copyfd_exact_size(fd, handle->src_fd, st.st_size);
115 close(fd);
116 handle->offset += st.st_size;
117
118 return 0;
119}
120
121static int write_ar_archive(archive_handle_t *handle)
122{
123 struct stat st;
124 archive_handle_t *out_handle;
125
Denys Vlasenko8d3e2252010-08-31 12:42:06 +0200126 xfstat(handle->src_fd, &st, handle->ar__name);
Alexander Shishkin535584c2010-03-15 15:38:09 +0100127
128 /* if archive exists, create a new handle for output.
129 * we create it in place of the old one.
130 */
131 if (st.st_size != 0) {
132 out_handle = init_handle();
133 xunlink(handle->ar__name);
134 out_handle->src_fd = xopen(handle->ar__name, O_WRONLY | O_CREAT | O_TRUNC);
135 out_handle->accept = handle->accept;
136 } else {
137 out_handle = handle;
138 }
139
140 handle->ar__out = out_handle;
141
142 xwrite(out_handle->src_fd, AR_MAGIC "\n", AR_MAGIC_LEN + 1);
143 out_handle->offset += AR_MAGIC_LEN + 1;
144
145 /* skip to the end of the archive if we have to append stuff */
146 if (st.st_size != 0) {
147 handle->filter = filter_replaceable;
148 handle->action_data = copy_data;
149 unpack_ar_archive(handle);
150 }
151
152 while (write_ar_header(out_handle) == 0)
153 continue;
154
155 /* optional, since we exit right after we return */
156 if (ENABLE_FEATURE_CLEAN_UP) {
157 close(handle->src_fd);
158 if (out_handle->src_fd != handle->src_fd)
159 close(out_handle->src_fd);
160 }
161
162 return EXIT_SUCCESS;
163}
164#endif /* FEATURE_AR_CREATE */
Eric Andersen86ab8a32000-06-02 03:21:42 +0000165
Denis Vlasenkodefc1ea2008-06-27 02:52:20 +0000166static void FAST_FUNC header_verbose_list_ar(const file_header_t *file_header)
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000167{
Manuel Novoa III cad53642003-03-19 09:13:01 +0000168 const char *mode = bb_mode_string(file_header->mode);
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000169 char *mtime;
170
171 mtime = ctime(&file_header->mtime);
172 mtime[16] = ' ';
173 memmove(&mtime[17], &mtime[20], 4);
174 mtime[21] = '\0';
Alexander Shishkin535584c2010-03-15 15:38:09 +0100175 printf("%s %u/%u%7"OFF_FMT"u %s %s\n", &mode[1],
176 (int)file_header->uid, (int)file_header->gid,
177 file_header->size,
178 &mtime[4], file_header->name
179 );
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000180}
181
Denys Vlasenkofb132e42010-10-29 11:46:52 +0200182#define AR_OPT_VERBOSE (1 << 0)
183#define AR_OPT_PRESERVE_DATE (1 << 1)
Alexander Shishkin535584c2010-03-15 15:38:09 +0100184/* "ar r" implies create, but warns about it. c suppresses warning.
185 * bbox accepts but ignores it: */
Denys Vlasenkofb132e42010-10-29 11:46:52 +0200186#define AR_OPT_CREATE (1 << 2)
Alexander Shishkin535584c2010-03-15 15:38:09 +0100187
Denys Vlasenkofb132e42010-10-29 11:46:52 +0200188#define AR_CMD_PRINT (1 << 3)
189#define FIRST_CMD AR_CMD_PRINT
190#define AR_CMD_LIST (1 << 4)
191#define AR_CMD_EXTRACT (1 << 5)
192#define AR_CMD_INSERT (1 << 6)
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000193
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000194int ar_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100195int ar_main(int argc UNUSED_PARAM, char **argv)
Eric Andersen86ab8a32000-06-02 03:21:42 +0000196{
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000197 archive_handle_t *archive_handle;
Alexander Shishkin535584c2010-03-15 15:38:09 +0100198 unsigned opt, t;
Glenn L McGrath930453b2004-01-04 10:28:22 +0000199
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000200 archive_handle = init_handle();
Tim Rikerc1ef7bd2006-01-25 00:08:53 +0000201
Alexander Shishkin535584c2010-03-15 15:38:09 +0100202 /* --: prepend '-' to the first argument if required */
203 /* -1: at least one param is reqd */
204 /* one of p,t,x[,r] is required */
205 opt_complementary = "--:-1:p:t:x"IF_FEATURE_AR_CREATE(":r");
206 opt = getopt32(argv, "voc""ptx"IF_FEATURE_AR_CREATE("r"));
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100207 argv += optind;
Glenn L McGrath930453b2004-01-04 10:28:22 +0000208
Alexander Shishkin535584c2010-03-15 15:38:09 +0100209 t = opt / FIRST_CMD;
210 if (t & (t-1)) /* more than one of p,t,x[,r] are specified */
211 bb_show_usage();
212
213 if (opt & AR_CMD_PRINT) {
Glenn L McGrath930453b2004-01-04 10:28:22 +0000214 archive_handle->action_data = data_extract_to_stdout;
215 }
Alexander Shishkin535584c2010-03-15 15:38:09 +0100216 if (opt & AR_CMD_LIST) {
Glenn L McGrath930453b2004-01-04 10:28:22 +0000217 archive_handle->action_header = header_list;
218 }
Alexander Shishkin535584c2010-03-15 15:38:09 +0100219 if (opt & AR_CMD_EXTRACT) {
Glenn L McGrath930453b2004-01-04 10:28:22 +0000220 archive_handle->action_data = data_extract_all;
221 }
222 if (opt & AR_OPT_PRESERVE_DATE) {
Denys Vlasenkod57d6262009-09-17 02:43:14 +0200223 archive_handle->ah_flags |= ARCHIVE_RESTORE_DATE;
Glenn L McGrath930453b2004-01-04 10:28:22 +0000224 }
225 if (opt & AR_OPT_VERBOSE) {
226 archive_handle->action_header = header_verbose_list_ar;
227 }
Alexander Shishkin535584c2010-03-15 15:38:09 +0100228#if ENABLE_FEATURE_AR_CREATE
229 archive_handle->ar__name = *argv;
230#endif
231 archive_handle->src_fd = xopen(*argv++,
232 (opt & AR_CMD_INSERT)
233 ? O_RDWR | O_CREAT
234 : O_RDONLY
235 );
Glenn L McGrath930453b2004-01-04 10:28:22 +0000236
Nguyễn Thái Ngọc Duy36dcf4e2010-04-15 16:55:04 -0700237 if (*argv)
238 archive_handle->filter = filter_accept_list;
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100239 while (*argv) {
Alexander Shishkin535584c2010-03-15 15:38:09 +0100240 llist_add_to_end(&archive_handle->accept, *argv++);
Glenn L McGrath4f1b0122000-12-15 06:50:09 +0000241 }
Glenn L McGrath4949faf2001-04-11 16:23:35 +0000242
Alexander Shishkin535584c2010-03-15 15:38:09 +0100243#if ENABLE_FEATURE_AR_CREATE
244 if (opt & AR_CMD_INSERT)
245 return write_ar_archive(archive_handle);
246#endif
247
Denis Vlasenko0381d422008-07-10 23:06:00 +0000248 unpack_ar_archive(archive_handle);
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000249
Matt Kraai3e856ce2000-12-01 02:55:13 +0000250 return EXIT_SUCCESS;
Eric Andersen86ab8a32000-06-02 03:21:42 +0000251}