diff options
Diffstat (limited to '')
-rw-r--r-- | src/split/split.c | 287 |
1 files changed, 287 insertions, 0 deletions
diff --git a/src/split/split.c b/src/split/split.c new file mode 100644 index 0000000..0df8fa7 --- /dev/null +++ b/src/split/split.c @@ -0,0 +1,287 @@ +/* + * dpkg-split - splitting and joining of multipart *.deb archives + * split.c - splitting archives + * + * Copyright © 1995 Ian Jackson <ijackson@chiark.greenend.org.uk> + * Copyright © 2008-2015 Guillem Jover <guillem@debian.org> + * + * This is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <https://www.gnu.org/licenses/>. + */ + +#include <config.h> +#include <compat.h> + +#include <sys/types.h> +#include <sys/stat.h> +#include <sys/wait.h> + +#include <errno.h> +#include <limits.h> +#include <inttypes.h> +#include <fcntl.h> +#include <libgen.h> +#include <string.h> +#include <time.h> +#include <unistd.h> +#include <stdint.h> +#include <stdlib.h> +#include <stdio.h> + +#include <dpkg/i18n.h> +#include <dpkg/c-ctype.h> +#include <dpkg/dpkg.h> +#include <dpkg/dpkg-db.h> +#include <dpkg/parsedump.h> +#include <dpkg/path.h> +#include <dpkg/string.h> +#include <dpkg/subproc.h> +#include <dpkg/buffer.h> +#include <dpkg/ar.h> +#include <dpkg/options.h> + +#include "dpkg-split.h" + +/** + * Parse the control file from a .deb package into a struct pkginfo. + */ +static struct pkginfo * +deb_parse_control(const char *filename) +{ + struct parsedb_state *ps; + struct pkginfo *pkg; + pid_t pid; + int p[2]; + + m_pipe(p); + + pid = subproc_fork(); + if (pid == 0) { + /* Child writes to pipe. */ + m_dup2(p[1], 1); + close(p[0]); + close(p[1]); + + execlp(BACKEND, BACKEND, "--info", filename, "control", NULL); + ohshite(_("unable to execute %s (%s)"), + _("package field value extraction"), BACKEND); + } + close(p[1]); + + /* Parent reads from pipe. */ + ps = parsedb_new(_("<dpkg-deb --info pipe>"), p[0], pdb_parse_binary); + parsedb_load(ps); + parsedb_parse(ps, &pkg); + parsedb_close(ps); + + close(p[0]); + + subproc_reap(pid, _("package field value extraction"), SUBPROC_NOPIPE); + + return pkg; +} + +static intmax_t +parse_timestamp(const char *value) +{ + intmax_t timestamp; + char *end; + + errno = 0; + timestamp = strtoimax(value, &end, 10); + if (value == end || *end || errno != 0) + ohshite(_("unable to parse timestamp '%.255s'"), value); + + return timestamp; +} + +/* Cleanup filename for use in crippled msdos systems. */ +static char * +clean_msdos_filename(char *filename) +{ + char *d, *s; + + for (s = d = filename; *s; d++, s++) { + if (*s == '+') + *d = 'x'; + else if (c_isupper(*s)) + *d = c_tolower(*s); + else if (c_islower(*s) || c_isdigit(*s)) + *d = *s; + else + s++; + } + + return filename; +} + +static int +mksplit(const char *file_src, const char *prefix, off_t maxpartsize, + bool msdos) +{ + struct pkginfo *pkg; + struct dpkg_error err; + int fd_src; + struct stat st; + intmax_t timestamp; + const char *timestamp_str; + const char *version; + char hash[MD5HASHLEN + 1]; + int nparts, curpart; + off_t partsize; + off_t cur_partsize, last_partsize; + char *prefixdir = NULL, *msdos_prefix = NULL; + struct varbuf file_dst = VARBUF_INIT; + struct varbuf partmagic = VARBUF_INIT; + struct varbuf partname = VARBUF_INIT; + + fd_src = open(file_src, O_RDONLY); + if (fd_src < 0) + ohshite(_("unable to open source file '%.250s'"), file_src); + if (fstat(fd_src, &st)) + ohshite(_("unable to fstat source file")); + if (!S_ISREG(st.st_mode)) + ohshit(_("source file '%.250s' not a plain file"), file_src); + + if (fd_md5(fd_src, hash, -1, &err) < 0) + ohshit(_("cannot compute MD5 digest for file '%s': %s"), + file_src, err.str); + lseek(fd_src, 0, SEEK_SET); + + pkg = deb_parse_control(file_src); + version = versiondescribe(&pkg->available.version, vdew_nonambig); + + timestamp_str = getenv("SOURCE_DATE_EPOCH"); + if (timestamp_str) + timestamp = parse_timestamp(timestamp_str); + else + timestamp = time(NULL); + + partsize = maxpartsize - HEADERALLOWANCE; + last_partsize = st.st_size % partsize; + if (last_partsize == 0) + last_partsize = partsize; + nparts = (st.st_size + partsize - 1) / partsize; + + printf(P_("Splitting package %s into %d part: ", + "Splitting package %s into %d parts: ", nparts), + pkg->set->name, nparts); + + if (msdos) { + char *t; + + t = m_strdup(prefix); + prefixdir = m_strdup(dirname(t)); + free(t); + + msdos_prefix = m_strdup(path_basename(prefix)); + prefix = clean_msdos_filename(msdos_prefix); + } + + for (curpart = 1; curpart <= nparts; curpart++) { + struct dpkg_ar *ar; + + varbuf_reset(&file_dst); + /* Generate output filename. */ + if (msdos) { + char *refname; + int prefix_max; + + refname = str_fmt("%dof%d", curpart, nparts); + prefix_max = max(8 - strlen(refname), 0); + varbuf_printf(&file_dst, "%s/%.*s%.8s.deb", + prefixdir, prefix_max, prefix, refname); + free(refname); + } else { + varbuf_printf(&file_dst, "%s.%dof%d.deb", + prefix, curpart, nparts); + } + + if (curpart == nparts) + cur_partsize = last_partsize; + else + cur_partsize = partsize; + + if (cur_partsize > maxpartsize) { + ohshit(_("header is too long, making part too long; " + "the package name or version\n" + "numbers must be extraordinarily long, " + "or something; giving up")); + } + + /* Split the data. */ + ar = dpkg_ar_create(file_dst.buf, 0644); + dpkg_ar_set_mtime(ar, timestamp); + + /* Write the ar header. */ + dpkg_ar_put_magic(ar); + + /* Write the debian-split part. */ + varbuf_printf(&partmagic, + "%s\n%s\n%s\n%s\n%jd\n%jd\n%d/%d\n%s\n", + SPLITVERSION, pkg->set->name, version, hash, + (intmax_t)st.st_size, (intmax_t)partsize, + curpart, nparts, pkg->available.arch->name); + dpkg_ar_member_put_mem(ar, PARTMAGIC, + partmagic.buf, partmagic.used); + varbuf_reset(&partmagic); + + /* Write the data part. */ + varbuf_printf(&partname, "data.%d", curpart); + dpkg_ar_member_put_file(ar, partname.buf, + fd_src, cur_partsize); + varbuf_reset(&partname); + + dpkg_ar_close(ar); + + printf("%d ", curpart); + } + + varbuf_destroy(&file_dst); + varbuf_destroy(&partname); + varbuf_destroy(&partmagic); + + free(prefixdir); + free(msdos_prefix); + + close(fd_src); + + printf(_("done\n")); + + return 0; +} + +int +do_split(const char *const *argv) +{ + const char *sourcefile, *prefix; + + sourcefile = *argv++; + if (!sourcefile) + badusage(_("--split needs a source filename argument")); + prefix = *argv++; + if (prefix && *argv) + badusage(_("--split takes at most a source filename and destination prefix")); + if (!prefix) { + size_t sourcefile_len = strlen(sourcefile); + + if (str_match_end(sourcefile, DEBEXT)) + sourcefile_len -= strlen(DEBEXT); + + prefix = nfstrnsave(sourcefile, sourcefile_len); + } + + mksplit(sourcefile, prefix, opt_maxpartsize, opt_msdos); + + return 0; +} |