#include "system.h"
#include <fnmatch.h>
#include <hash.h>
#include <quotearg.h>
#include <stpcpy.h>
#include <xstrtol.h>
#include "common.h"
#define obstack_chunk_alloc xmalloc
#define obstack_chunk_free free
#include <obstack.h>
#include <fnmatch.h>
static bool xheader_protected_pattern_p (char const *pattern);
static bool xheader_protected_keyword_p (char const *keyword);
static void xheader_set_single_keyword (char *) __attribute__ ((noreturn));
static void code_string (char const *string, char const *keyword,
struct xheader *xhdr);
static void extended_header_init (void);
static size_t global_header_count;
struct keyword_list
{
struct keyword_list *next;
char *pattern;
char *value;
};
static struct keyword_list *keyword_pattern_list;
static struct keyword_list *keyword_global_override_list;
static struct keyword_list *keyword_override_list;
static struct keyword_list *global_header_override_list;
static char *exthdr_name;
static char *globexthdr_name;
static bool
xheader_keyword_deleted_p (const char *kw)
{
struct keyword_list *kp;
for (kp = keyword_pattern_list; kp; kp = kp->next)
if (fnmatch (kp->pattern, kw, 0) == 0)
return true;
return false;
}
static bool
xheader_keyword_override_p (const char *keyword)
{
struct keyword_list *kp;
for (kp = keyword_override_list; kp; kp = kp->next)
if (strcmp (kp->pattern, keyword) == 0)
return true;
return false;
}
static void
xheader_list_append (struct keyword_list **root, char const *kw,
char const *value)
{
struct keyword_list *kp = xmalloc (sizeof *kp);
kp->pattern = xstrdup (kw);
kp->value = value ? xstrdup (value) : NULL;
kp->next = *root;
*root = kp;
}
static void
xheader_list_destroy (struct keyword_list **root)
{
if (root)
{
struct keyword_list *kw = *root;
while (kw)
{
struct keyword_list *next = kw->next;
free (kw->pattern);
free (kw->value);
free (kw);
kw = next;
}
*root = NULL;
}
}
static void
xheader_set_single_keyword (char *kw)
{
USAGE_ERROR ((0, 0, _("Keyword %s is unknown or not yet imlemented"), kw));
}
static void
xheader_set_keyword_equal (char *kw, char *eq)
{
bool global = true;
char *p = eq;
if (eq[-1] == ':')
{
p--;
global = false;
}
while (p > kw && isspace (*p))
p--;
*p = 0;
for (p = eq + 1; *p && isspace (*p); p++)
;
if (strcmp (kw, "delete") == 0)
{
if (xheader_protected_pattern_p (p))
USAGE_ERROR ((0, 0, _("Pattern %s cannot be used"), p));
xheader_list_append (&keyword_pattern_list, p, NULL);
}
else if (strcmp (kw, "exthdr.name") == 0)
assign_string (&exthdr_name, p);
else if (strcmp (kw, "globexthdr.name") == 0)
assign_string (&globexthdr_name, p);
else
{
if (xheader_protected_keyword_p (kw))
USAGE_ERROR ((0, 0, _("Keyword %s cannot be overridden"), kw));
if (global)
xheader_list_append (&keyword_global_override_list, kw, p);
else
xheader_list_append (&keyword_override_list, kw, p);
}
}
void
xheader_set_option (char *string)
{
char *token;
for (token = strtok (string, ","); token; token = strtok (NULL, ","))
{
char *p = strchr (token, '=');
if (!p)
xheader_set_single_keyword (token);
else
xheader_set_keyword_equal (token, p);
}
}
static void
to_decimal (uintmax_t value, char *where, size_t size)
{
size_t i = 0, j;
where[i++] = 0;
do
{
where[i++] = '0' + value % 10;
value /= 10;
}
while (i < size && value);
for (j = 0, i--; j < i; j++, i--)
{
char c = where[j];
where[j] = where[i];
where[i] = c;
}
}
static char *
xheader_format_name (struct tar_stat_info *st, const char *fmt, bool allow_n)
{
char *buf;
size_t len = strlen (fmt);
char *q;
const char *p;
char *dir = NULL;
char *base = NULL;
char pidbuf[64];
char nbuf[64];
for (p = fmt; *p && (p = strchr (p, '%')); )
{
switch (p[1])
{
case '%':
len--;
break;
case 'd':
if (st)
{
dir = safer_name_suffix (dir_name (st->orig_file_name), false);
len += strlen (dir) - 1;
}
break;
case 'f':
if (st)
{
base = base_name (st->orig_file_name);
len += strlen (base) - 1;
}
break;
case 'p':
to_decimal (getpid (), pidbuf, sizeof pidbuf);
len += strlen (pidbuf) - 1;
break;
case 'n':
if (allow_n)
{
to_decimal (global_header_count + 1, pidbuf, sizeof pidbuf);
len += strlen (nbuf) - 1;
}
break;
}
p++;
}
buf = xmalloc (len + 1);
for (q = buf, p = fmt; *p; )
{
if (*p == '%')
{
switch (p[1])
{
case '%':
*q++ = *p++;
p++;
break;
case 'd':
if (dir)
q = stpcpy (q, dir);
p += 2;
break;
case 'f':
if (base)
q = stpcpy (q, base);
p += 2;
break;
case 'p':
q = stpcpy (q, pidbuf);
p += 2;
break;
case 'n':
if (allow_n)
{
q = stpcpy (q, nbuf);
p += 2;
}
default:
*q++ = *p++;
if (*p)
*q++ = *p++;
}
}
else
*q++ = *p++;
}
while (q > buf && ISSLASH (q[-1]))
q--;
*q = 0;
return buf;
}
char *
xheader_xhdr_name (struct tar_stat_info *st)
{
if (!exthdr_name)
assign_string (&exthdr_name, "%d/PaxHeaders.%p/%f");
return xheader_format_name (st, exthdr_name, false);
}
#define GLOBAL_HEADER_TEMPLATE "/GlobalHead.%p.%n"
char *
xheader_ghdr_name (void)
{
if (!globexthdr_name)
{
size_t len;
const char *tmp = getenv ("TMPDIR");
if (!tmp)
tmp = "/tmp";
len = strlen (tmp) + sizeof (GLOBAL_HEADER_TEMPLATE);
globexthdr_name = xmalloc (len);
strcpy(globexthdr_name, tmp);
strcat(globexthdr_name, GLOBAL_HEADER_TEMPLATE);
}
return xheader_format_name (NULL, globexthdr_name, true);
}
void
xheader_write (char type, char *name, struct xheader *xhdr)
{
union block *header;
size_t size;
char *p;
size = xhdr->size;
header = start_private_header (name, size);
header->header.typeflag = type;
simple_finish_header (header);
p = xhdr->buffer;
do
{
size_t len;
header = find_next_block ();
len = BLOCKSIZE;
if (len > size)
len = size;
memcpy (header->buffer, p, len);
if (len < BLOCKSIZE)
memset (header->buffer + len, 0, BLOCKSIZE - len);
p += len;
size -= len;
set_next_block_after (header);
}
while (size > 0);
xheader_destroy (xhdr);
}
void
xheader_write_global (void)
{
char *name;
struct keyword_list *kp;
if (!keyword_global_override_list)
return;
extended_header_init ();
for (kp = keyword_global_override_list; kp; kp = kp->next)
code_string (kp->value, kp->pattern, &extended_header);
xheader_finish (&extended_header);
xheader_write (XGLTYPE, name = xheader_ghdr_name (),
&extended_header);
free (name);
global_header_count++;
}
struct xhdr_tab
{
char const *keyword;
void (*coder) (struct tar_stat_info const *, char const *,
struct xheader *, void *data);
void (*decoder) (struct tar_stat_info *, char const *);
bool protect;
};
extern struct xhdr_tab const xhdr_tab[];
static struct xhdr_tab const *
locate_handler (char const *keyword)
{
struct xhdr_tab const *p;
for (p = xhdr_tab; p->keyword; p++)
if (strcmp (p->keyword, keyword) == 0)
return p;
return NULL;
}
static bool
xheader_protected_pattern_p (const char *pattern)
{
struct xhdr_tab const *p;
for (p = xhdr_tab; p->keyword; p++)
if (p->protect && fnmatch (pattern, p->keyword, 0) == 0)
return true;
return false;
}
static bool
xheader_protected_keyword_p (const char *keyword)
{
struct xhdr_tab const *p;
for (p = xhdr_tab; p->keyword; p++)
if (p->protect && strcmp (p->keyword, keyword) == 0)
return true;
return false;
}
static bool
decode_record (char **p,
void (*handler) (void *, char const *, char const *),
void *data)
{
size_t len;
char const *keyword;
char *start = *p;
char endc;
if (**p == 0)
return false;
len = strtoul (*p, p, 10);
if (**p != ' ')
{
ERROR ((0, 0,
_("Malformed extended header: missing whitespace after the length")));
return false;
}
keyword = ++*p;
for (;*p < start + len; ++*p)
if (**p == '=')
break;
if (**p != '=')
{
ERROR ((0, 0, _("Malformed extended header: missing equal sign")));
return false;
}
**p = 0;
endc = start[len-1];
start[len-1] = 0;
handler (data, keyword, *p + 1);
start[len-1] = endc;
**p = '=';
*p = &start[len];
return true;
}
static void
run_override_list (struct keyword_list *kp, struct tar_stat_info *st)
{
for (; kp; kp = kp->next)
{
struct xhdr_tab const *t = locate_handler (kp->pattern);
if (t)
t->decoder (st, kp->value);
}
}
static void
decx (void *data, char const *keyword, char const *value)
{
struct xhdr_tab const *t;
struct tar_stat_info *st = data;
if (xheader_keyword_deleted_p (keyword)
|| xheader_keyword_override_p (keyword))
return;
t = locate_handler (keyword);
if (t)
t->decoder (st, value);
}
void
xheader_decode (struct tar_stat_info *st)
{
run_override_list (keyword_global_override_list, st);
run_override_list (global_header_override_list, st);
if (extended_header.size)
{
char *p = extended_header.buffer + BLOCKSIZE;
char *endp = &extended_header.buffer[extended_header.size-1];
while (p < endp)
if (!decode_record (&p, decx, st))
break;
}
run_override_list (keyword_override_list, st);
}
static void
decg (void *data, char const *keyword, char const *value)
{
struct keyword_list **kwl = data;
xheader_list_append (kwl, keyword, value);
}
void
xheader_decode_global (void)
{
if (extended_header.size)
{
char *p = extended_header.buffer + BLOCKSIZE;
char *endp = &extended_header.buffer[extended_header.size-1];
xheader_list_destroy (&global_header_override_list);
while (p < endp)
if (!decode_record (&p, decg, &global_header_override_list))
break;
}
}
static void
extended_header_init (void)
{
if (!extended_header.stk)
{
extended_header.stk = xmalloc (sizeof *extended_header.stk);
obstack_init (extended_header.stk);
}
}
void
xheader_store (char const *keyword, struct tar_stat_info const *st, void *data)
{
struct xhdr_tab const *t;
if (extended_header.buffer)
return;
t = locate_handler (keyword);
if (!t)
return;
if (xheader_keyword_deleted_p (keyword)
|| xheader_keyword_override_p (keyword))
return;
extended_header_init ();
t->coder (st, keyword, &extended_header, data);
}
void
xheader_read (union block *p, size_t size)
{
size_t j = 0;
size_t nblocks;
free (extended_header.buffer);
size += BLOCKSIZE;
extended_header.size = size;
nblocks = (size + BLOCKSIZE - 1) / BLOCKSIZE;
extended_header.buffer = xmalloc (size + 1);
do
{
size_t len = size;
if (len > BLOCKSIZE)
len = BLOCKSIZE;
memcpy (&extended_header.buffer[j], p->buffer, len);
set_next_block_after (p);
p = find_next_block ();
j += len;
size -= len;
}
while (size > 0);
}
static size_t
format_uintmax (uintmax_t val, char *buf, size_t s)
{
if (!buf)
{
s = 0;
do
s++;
while ((val /= 10) != 0);
}
else
{
char *p = buf + s - 1;
do
{
*p-- = val % 10 + '0';
}
while ((val /= 10) != 0);
while (p >= buf)
*p-- = '0';
}
return s;
}
static void
xheader_print (struct xheader *xhdr, char const *keyword, char const *value)
{
size_t len = strlen (keyword) + strlen (value) + 3;
size_t p, n = 0;
char nbuf[100];
do
{
p = n;
n = format_uintmax (len + p, NULL, 0);
}
while (n != p);
format_uintmax (len + n, nbuf, n);
obstack_grow (xhdr->stk, nbuf, n);
obstack_1grow (xhdr->stk, ' ');
obstack_grow (xhdr->stk, keyword, strlen (keyword));
obstack_1grow (xhdr->stk, '=');
obstack_grow (xhdr->stk, value, strlen (value));
obstack_1grow (xhdr->stk, '\n');
}
void
xheader_finish (struct xheader *xhdr)
{
struct keyword_list *kp;
for (kp = keyword_override_list; kp; kp = kp->next)
code_string (kp->value, kp->pattern, xhdr);
obstack_1grow (xhdr->stk, 0);
xhdr->buffer = obstack_finish (xhdr->stk);
xhdr->size = strlen (xhdr->buffer);
}
void
xheader_destroy (struct xheader *xhdr)
{
if (xhdr->stk)
{
obstack_free (xhdr->stk, NULL);
free (xhdr->stk);
xhdr->stk = NULL;
}
else
free (xhdr->buffer);
xhdr->buffer = 0;
xhdr->size = 0;
}
static void
code_string (char const *string, char const *keyword, struct xheader *xhdr)
{
char *outstr;
if (!utf8_convert (true, string, &outstr))
{
outstr = xstrdup (string);
}
xheader_print (xhdr, keyword, outstr);
free (outstr);
}
static void
decode_string (char **string, char const *arg)
{
if (*string)
{
free (*string);
*string = NULL;
}
if (!utf8_convert (false, arg, string))
{
assign_string (string, arg);
}
}
static void
code_time (time_t t, unsigned long nano,
char const *keyword, struct xheader *xhdr)
{
char sbuf[200];
size_t s = format_uintmax (t, NULL, 0);
if (s + 11 >= sizeof sbuf)
return;
format_uintmax (t, sbuf, s);
sbuf[s++] = '.';
s += format_uintmax (nano, sbuf + s, 9);
sbuf[s] = 0;
xheader_print (xhdr, keyword, sbuf);
}
static void
decode_time (char const *arg, time_t *secs, unsigned long *nsecs)
{
uintmax_t u;
char *p;
if (xstrtoumax (arg, &p, 10, &u, "") == LONGINT_OK)
{
*secs = u;
if (*p == '.' && xstrtoumax (p+1, NULL, 10, &u, "") == LONGINT_OK)
*nsecs = u;
}
}
static void
code_num (uintmax_t value, char const *keyword, struct xheader *xhdr)
{
char sbuf[100];
size_t s = format_uintmax (value, NULL, 0);
format_uintmax (value, sbuf, s);
sbuf[s] = 0;
xheader_print (xhdr, keyword, sbuf);
}
static bool
decode_num (uintmax_t *num, char const *arg, uintmax_t maxval,
char const *keyword)
{
uintmax_t u;
char *arg_lim;
if (! (ISDIGIT (*arg)
&& (errno = 0, u = strtoumax (arg, &arg_lim, 10), !*arg_lim)))
{
ERROR ((0, 0, _("Malformed extended header: invalid %s=%s"),
keyword, arg));
return false;
}
if (! (u <= maxval && errno != ERANGE))
{
ERROR ((0, 0, _("Extended header %s=%s is out of range"),
keyword, arg));
return false;
}
*num = u;
return true;
}
static void
dummy_coder (struct tar_stat_info const *st __attribute__ ((unused)),
char const *keyword __attribute__ ((unused)),
struct xheader *xhdr __attribute__ ((unused)),
void *data __attribute__ ((unused)))
{
}
static void
dummy_decoder (struct tar_stat_info *st __attribute__ ((unused)),
char const *arg __attribute__ ((unused)))
{
}
static void
atime_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_time (st->stat.st_atime, st->atime_nsec, keyword, xhdr);
}
static void
atime_decoder (struct tar_stat_info *st, char const *arg)
{
decode_time (arg, &st->stat.st_atime, &st->atime_nsec);
}
static void
gid_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_num (st->stat.st_gid, keyword, xhdr);
}
static void
gid_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, TYPE_MAXIMUM (gid_t), "gid"))
st->stat.st_gid = u;
}
static void
gname_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_string (st->gname, keyword, xhdr);
}
static void
gname_decoder (struct tar_stat_info *st, char const *arg)
{
decode_string (&st->gname, arg);
}
static void
linkpath_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_string (st->link_name, keyword, xhdr);
}
static void
linkpath_decoder (struct tar_stat_info *st, char const *arg)
{
decode_string (&st->link_name, arg);
}
static void
ctime_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_time (st->stat.st_ctime, st->ctime_nsec, keyword, xhdr);
}
static void
ctime_decoder (struct tar_stat_info *st, char const *arg)
{
decode_time (arg, &st->stat.st_ctime, &st->ctime_nsec);
}
static void
mtime_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_time (st->stat.st_mtime, st->mtime_nsec, keyword, xhdr);
}
static void
mtime_decoder (struct tar_stat_info *st, char const *arg)
{
decode_time (arg, &st->stat.st_mtime, &st->mtime_nsec);
}
static void
path_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_string (st->file_name, keyword, xhdr);
}
static void
path_decoder (struct tar_stat_info *st, char const *arg)
{
decode_string (&st->orig_file_name, arg);
decode_string (&st->file_name, arg);
st->had_trailing_slash = strip_trailing_slashes (st->file_name);
}
static void
size_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_num (st->stat.st_size, keyword, xhdr);
}
static void
size_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, TYPE_MAXIMUM (off_t), "size"))
st->archive_file_size = st->stat.st_size = u;
}
static void
uid_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_num (st->stat.st_uid, keyword, xhdr);
}
static void
uid_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, TYPE_MAXIMUM (uid_t), "uid"))
st->stat.st_uid = u;
}
static void
uname_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data __attribute__ ((unused)))
{
code_string (st->uname, keyword, xhdr);
}
static void
uname_decoder (struct tar_stat_info *st, char const *arg)
{
decode_string (&st->uname, arg);
}
static void
sparse_size_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data)
{
size_coder (st, keyword, xhdr, data);
}
static void
sparse_size_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, TYPE_MAXIMUM (off_t), "GNU.sparse.size"))
st->stat.st_size = u;
}
static void
sparse_numblocks_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr,
void *data __attribute__ ((unused)))
{
code_num (st->sparse_map_avail, keyword, xhdr);
}
static void
sparse_numblocks_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, SIZE_MAX, "GNU.sparse.numblocks"))
{
st->sparse_map_size = u;
st->sparse_map = xcalloc (u, sizeof st->sparse_map[0]);
st->sparse_map_avail = 0;
}
}
static void
sparse_offset_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data)
{
size_t i = *(size_t*)data;
code_num (st->sparse_map[i].offset, keyword, xhdr);
}
static void
sparse_offset_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, TYPE_MAXIMUM (off_t), "GNU.sparse.offset"))
{
if (st->sparse_map_avail < st->sparse_map_size)
st->sparse_map[st->sparse_map_avail].offset = u;
else
ERROR ((0, 0, _("Malformed extended header: excess %s=%s"),
"GNU.sparse.offset", arg));
}
}
static void
sparse_numbytes_coder (struct tar_stat_info const *st, char const *keyword,
struct xheader *xhdr, void *data)
{
size_t i = *(size_t*)data;
code_num (st->sparse_map[i].numbytes, keyword, xhdr);
}
static void
sparse_numbytes_decoder (struct tar_stat_info *st, char const *arg)
{
uintmax_t u;
if (decode_num (&u, arg, SIZE_MAX, "GNU.sparse.numbytes"))
{
if (st->sparse_map_avail == st->sparse_map_size)
st->sparse_map = x2nrealloc (st->sparse_map,
&st->sparse_map_size,
sizeof st->sparse_map[0]);
st->sparse_map[st->sparse_map_avail++].numbytes = u;
}
}
struct xhdr_tab const xhdr_tab[] = {
{ "atime", atime_coder, atime_decoder, false },
{ "comment", dummy_coder, dummy_decoder, false },
{ "charset", dummy_coder, dummy_decoder, false },
{ "ctime", ctime_coder, ctime_decoder, false },
{ "gid", gid_coder, gid_decoder, false },
{ "gname", gname_coder, gname_decoder, false },
{ "linkpath", linkpath_coder, linkpath_decoder, false },
{ "mtime", mtime_coder, mtime_decoder, false },
{ "path", path_coder, path_decoder, false },
{ "size", size_coder, size_decoder, false },
{ "uid", uid_coder, uid_decoder, false },
{ "uname", uname_coder, uname_decoder, false },
{ "GNU.sparse.size", sparse_size_coder, sparse_size_decoder, true },
{ "GNU.sparse.numblocks", sparse_numblocks_coder, sparse_numblocks_decoder,
true },
{ "GNU.sparse.offset", sparse_offset_coder, sparse_offset_decoder,
true },
{ "GNU.sparse.numbytes", sparse_numbytes_coder, sparse_numbytes_decoder,
true },
#if 0
{ "GNU.dump.name", dump_name_coder, dump_name_decoder, false },
{ "GNU.dump.status", dump_status_coder, dump_status_decoder, false },
{ "GNU.volume.header", volume_header_coder, volume_header_decoder, false },
{ "GNU.volume.size", volume_size_coder, volume_size_decoder, false },
{ "GNU.volume.offset", volume_offset_coder, volume_offset_decoder, false },
#endif
{ NULL, NULL, NULL, false }
};