1 files changed, 365 insertions, 0 deletions

diff --git a/src/split/queue.c b/src/split/queue.c
new file mode 100644
index 0000000..c9b8847
--- /dev/null
+++ b/src/split/queue.c
@@ -0,0 +1,365 @@
+/*
+ * dpkg-split - splitting and joining of multipart *.deb archives
+ * queue.c - queue management
+ *
+ * Copyright © 1995 Ian Jackson <ijackson@chiark.greenend.org.uk>
+ * Copyright © 2008-2014 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/stat.h>
+
+#include <errno.h>
+#include <limits.h>
+#include <inttypes.h>
+#include <string.h>
+#include <fcntl.h>
+#include <dirent.h>
+#include <unistd.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+#include <dpkg/i18n.h>
+#include <dpkg/dpkg.h>
+#include <dpkg/dpkg-db.h>
+#include <dpkg/dir.h>
+#include <dpkg/buffer.h>
+#include <dpkg/options.h>
+
+#include "dpkg-split.h"
+
+/*
+ * The queue, by default located in /var/lib/dpkg/parts/, is a plain
+ * directory with one file per part.
+ *
+ * Each part is named “<md5sum>.<maxpartlen>.<thispartn>.<maxpartn>”,
+ * with all numbers in hex.
+ */
+
+
+static bool
+decompose_filename(const char *filename, struct partqueue *pq)
+{
+  const char *p;
+  char *q;
+
+  if (strspn(filename, "0123456789abcdef") != MD5HASHLEN ||
+      filename[MD5HASHLEN] != '.')
+    return false;
+
+  pq->info.md5sum = nfstrnsave(filename, MD5HASHLEN);
+
+  p = filename + MD5HASHLEN + 1;
+  errno = 0;
+  pq->info.maxpartlen = strtoimax(p, &q, 16);
+  if (q == p || *q++ != '.' || errno != 0)
+    return false;
+
+  p = q;
+  pq->info.thispartn = (int)strtol(p, &q, 16);
+  if (q == p || *q++ != '.' || pq->info.thispartn < 0 || errno != 0)
+    return false;
+
+  p = q;
+  pq->info.maxpartn = (int)strtol(p, &q, 16);
+  if (q == p || *q || pq->info.maxpartn < 0 || errno != 0)
+    return false;
+
+  return true;
+}
+
+static struct partqueue *
+scandepot(void)
+{
+  DIR *depot;
+  struct dirent *de;
+  struct partqueue *queue = NULL;
+
+  depot = opendir(opt_depotdir);
+  if (!depot) {
+    if (errno == ENOENT)
+      return NULL;
+
+    ohshite(_("unable to read depot directory '%.250s'"), opt_depotdir);
+  }
+  while ((de= readdir(depot))) {
+    struct partqueue *pq;
+    char *p;
+
+    if (de->d_name[0] == '.') continue;
+    pq = nfmalloc(sizeof(*pq));
+    pq->info.fmtversion.major = 0;
+    pq->info.fmtversion.minor = 0;
+    pq->info.package = NULL;
+    pq->info.version = NULL;
+    pq->info.arch = NULL;
+    pq->info.orglength= pq->info.thispartoffset= pq->info.thispartlen= 0;
+    pq->info.headerlen= 0;
+    p = nfmalloc(strlen(opt_depotdir) + 1 + strlen(de->d_name) + 1);
+    sprintf(p, "%s/%s", opt_depotdir, de->d_name);
+    pq->info.filename= p;
+    if (!decompose_filename(de->d_name,pq)) {
+      pq->info.md5sum= NULL;
+      pq->info.maxpartlen= pq->info.thispartn= pq->info.maxpartn= 0;
+    }
+    pq->nextinqueue= queue;
+    queue= pq;
+  }
+  closedir(depot);
+
+  return queue;
+}
+
+static bool
+partmatches(struct partinfo *pi, struct partinfo *refi)
+{
+  return (pi->md5sum &&
+          strcmp(pi->md5sum, refi->md5sum) == 0 &&
+          pi->maxpartn == refi->maxpartn &&
+          pi->maxpartlen == refi->maxpartlen);
+}
+
+int
+do_auto(const char *const *argv)
+{
+  const char *partfile;
+  struct partinfo *refi, **partlist, *otherthispart;
+  struct partqueue *queue;
+  struct partqueue *pq;
+  struct dpkg_ar *part;
+  int i, j;
+
+  if (!opt_outputfile)
+    badusage(_("--auto requires the use of the --output option"));
+  partfile = *argv++;
+  if (partfile == NULL || *argv)
+    badusage(_("--auto requires exactly one part file argument"));
+
+  refi = nfmalloc(sizeof(*refi));
+  part = dpkg_ar_open(partfile);
+  if (!part)
+    ohshite(_("unable to read part file '%.250s'"), partfile);
+  refi = read_info(part, refi);
+  dpkg_ar_close(part);
+
+  if (refi == NULL) {
+    if (!opt_npquiet)
+      printf(_("File '%.250s' is not part of a multipart archive.\n"), partfile);
+    m_output(stdout, _("<standard output>"));
+    return 1;
+  }
+
+  queue = scandepot();
+  if (queue == NULL)
+    if (dir_make_path(opt_depotdir, 0755) < 0)
+      ohshite(_("cannot create directory %s"), opt_depotdir);
+
+  partlist = nfmalloc(sizeof(*partlist) * refi->maxpartn);
+  for (i = 0; i < refi->maxpartn; i++)
+    partlist[i] = NULL;
+  for (pq= queue; pq; pq= pq->nextinqueue) {
+    struct partinfo *npi, *pi = &pq->info;
+
+    if (!partmatches(pi,refi)) continue;
+    npi = nfmalloc(sizeof(*npi));
+    mustgetpartinfo(pi->filename,npi);
+    addtopartlist(partlist,npi,refi);
+  }
+  /* If we already have a copy of this version we ignore it and prefer the
+   * new one, but we still want to delete the one in the depot, so we
+   * save its partinfo (with the filename) for later. This also prevents
+   * us from accidentally deleting the source file. */
+  otherthispart= partlist[refi->thispartn-1];
+  partlist[refi->thispartn-1]= refi;
+  for (j=refi->maxpartn-1; j>=0 && partlist[j]; j--);
+
+  if (j>=0) {
+    struct dpkg_error err;
+    int fd_src, fd_dst;
+    int ap;
+    char *p, *q;
+
+    p = str_fmt("%s/t.%lx", opt_depotdir, (long)getpid());
+    q = str_fmt("%s/%s.%jx.%x.%x", opt_depotdir, refi->md5sum,
+                (intmax_t)refi->maxpartlen, refi->thispartn, refi->maxpartn);
+
+    fd_src = open(partfile, O_RDONLY);
+    if (fd_src < 0)
+      ohshite(_("unable to reopen part file '%.250s'"), partfile);
+    fd_dst = creat(p, 0644);
+    if (fd_dst < 0)
+      ohshite(_("unable to open new depot file '%.250s'"), p);
+
+    if (fd_fd_copy(fd_src, fd_dst, refi->filesize, &err) < 0)
+      ohshit(_("cannot extract split package part '%s': %s"),
+             partfile, err.str);
+
+    if (fsync(fd_dst))
+      ohshite(_("unable to sync file '%s'"), p);
+    if (close(fd_dst))
+      ohshite(_("unable to close file '%s'"), p);
+    close(fd_src);
+
+    if (rename(p, q))
+      ohshite(_("unable to rename new depot file '%.250s' to '%.250s'"), p, q);
+    free(q);
+    free(p);
+
+    printf(_("Part %d of package %s filed (still want "),refi->thispartn,refi->package);
+    /* There are still some parts missing. */
+    for (i=0, ap=0; i<refi->maxpartn; i++)
+      if (!partlist[i])
+        printf("%s%d", !ap++ ? "" : i == j ? _(" and ") : ", ", i + 1);
+    printf(").\n");
+
+    dir_sync_path(opt_depotdir);
+  } else {
+    /* We have all the parts. */
+    reassemble(partlist, opt_outputfile);
+
+    /* OK, delete all the parts (except the new one, which we never copied). */
+    partlist[refi->thispartn-1]= otherthispart;
+    for (i=0; i<refi->maxpartn; i++)
+      if (partlist[i])
+        if (unlink(partlist[i]->filename))
+          ohshite(_("unable to delete used-up depot file '%.250s'"),
+                  partlist[i]->filename);
+  }
+
+  m_output(stderr, _("<standard error>"));
+
+  return 0;
+}
+
+int
+do_queue(const char *const *argv)
+{
+  struct partqueue *queue;
+  struct partqueue *pq;
+  const char *head;
+  struct stat stab;
+  off_t bytes;
+
+  if (*argv)
+    badusage(_("--%s takes no arguments"), cipaction->olong);
+
+  queue = scandepot();
+
+  head= N_("Junk files left around in the depot directory:\n");
+  for (pq= queue; pq; pq= pq->nextinqueue) {
+    if (pq->info.md5sum) continue;
+    fputs(gettext(head),stdout); head= "";
+    if (lstat(pq->info.filename,&stab))
+      ohshit(_("unable to stat '%.250s'"), pq->info.filename);
+    if (S_ISREG(stab.st_mode)) {
+      bytes= stab.st_size;
+      printf(_(" %s (%jd bytes)\n"), pq->info.filename, (intmax_t)bytes);
+    } else {
+      printf(_(" %s (not a plain file)\n"),pq->info.filename);
+    }
+  }
+  if (!*head) putchar('\n');
+
+  head= N_("Packages not yet reassembled:\n");
+  for (pq= queue; pq; pq= pq->nextinqueue) {
+    struct partinfo ti;
+    int i;
+
+    if (!pq->info.md5sum) continue;
+    mustgetpartinfo(pq->info.filename,&ti);
+    fputs(gettext(head),stdout); head= "";
+    printf(_(" Package %s: part(s) "), ti.package);
+    bytes= 0;
+    for (i=0; i<ti.maxpartn; i++) {
+      struct partqueue *qq;
+
+      for (qq= pq;
+           qq && !(partmatches(&qq->info,&ti) && qq->info.thispartn == i+1);
+           qq= qq->nextinqueue);
+      if (qq) {
+        printf("%d ",i+1);
+        if (lstat(qq->info.filename,&stab))
+          ohshite(_("unable to stat '%.250s'"), qq->info.filename);
+        if (!S_ISREG(stab.st_mode))
+          ohshit(_("part file '%.250s' is not a plain file"), qq->info.filename);
+        bytes+= stab.st_size;
+
+        /* Don't find this package again. */
+        qq->info.md5sum = NULL;
+      }
+    }
+    printf(_("(total %jd bytes)\n"), (intmax_t)bytes);
+  }
+  m_output(stdout, _("<standard output>"));
+
+  return 0;
+}
+
+enum discard_which {
+  DISCARD_PART_JUNK,
+  DISCARD_PART_PACKAGE,
+  DISCARD_PART_ALL,
+};
+
+static void
+discard_parts(struct partqueue *queue, enum discard_which which,
+              const char *package)
+{
+  struct partqueue *pq;
+
+  for (pq= queue; pq; pq= pq->nextinqueue) {
+    switch (which) {
+    case DISCARD_PART_JUNK:
+      if (pq->info.md5sum) continue;
+      break;
+    case DISCARD_PART_PACKAGE:
+      if (!pq->info.md5sum || strcasecmp(pq->info.package,package)) continue;
+      break;
+    case DISCARD_PART_ALL:
+      break;
+    default:
+      internerr("unknown discard_which '%d'", which);
+    }
+    if (unlink(pq->info.filename))
+      ohshite(_("unable to discard '%.250s'"), pq->info.filename);
+    printf(_("Deleted %s.\n"),pq->info.filename);
+  }
+}
+
+int
+do_discard(const char *const *argv)
+{
+  const char *thisarg;
+  struct partqueue *queue;
+  struct partqueue *pq;
+
+  queue = scandepot();
+  if (*argv) {
+    for (pq= queue; pq; pq= pq->nextinqueue)
+      if (pq->info.md5sum)
+        mustgetpartinfo(pq->info.filename,&pq->info);
+    discard_parts(queue, DISCARD_PART_JUNK, NULL);
+    while ((thisarg = *argv++))
+      discard_parts(queue, DISCARD_PART_PACKAGE, thisarg);
+  } else {
+    discard_parts(queue, DISCARD_PART_ALL, NULL);
+  }
+
+  return 0;
+}