builtin-tar-tree.c

   1 /*
   2  * Copyright (c) 2005, 2006 Rene Scharfe
   3  */
   4 #include <time.h>
   5 #include "cache.h"
   6 #include "commit.h"
   7 #include "tar.h"
   8 #include "builtin.h"
   9 #include "pkt-line.h"
  10 #include "archive.h"
  11
  12 #define RECORDSIZE      (512)
  13 #define BLOCKSIZE       (RECORDSIZE * 20)
  14
  15 static const char tar_tree_usage[] =
  16 "git-tar-tree [--remote=<repo>] <tree-ish> [basedir]";
  17
  18 static int generate_tar(int argc, const char **argv, const char *prefix)
  19 {
  20         struct archiver_args args;
  21         int result;
  22         char *base = NULL;
  23
  24         memset(&args, 0, sizeof(args));
  25         if (argc != 2 && argc != 3)
  26                 usage(tar_tree_usage);
  27         if (argc == 3) {
  28                 int baselen = strlen(argv[2]);
  29                 base = xmalloc(baselen + 2);
  30                 memcpy(base, argv[2], baselen);
  31                 base[baselen] = '/';
  32                 base[baselen + 1] = '\0';
  33         }
  34         args.base = base;
  35         parse_treeish_arg(argv + 1, &args, NULL);
  36
  37         result = write_tar_archive(&args);
  38         free(base);
  39
  40         return result;
  41 }
  42
  43 static const char *exec = "git-upload-tar";
  44
  45 static int remote_tar(int argc, const char **argv)
  46 {
  47         int fd[2], ret, len;
  48         pid_t pid;
  49         char buf[1024];
  50         char *url;
  51
  52         if (argc < 3 || 4 < argc)
  53                 usage(tar_tree_usage);
  54
  55         /* --remote=<repo> */
  56         url = xstrdup(argv[1]+9);
  57         pid = git_connect(fd, url, exec);
  58         if (pid < 0)
  59                 return 1;
  60
  61         packet_write(fd[1], "want %s\n", argv[2]);
  62         if (argv[3])
  63                 packet_write(fd[1], "base %s\n", argv[3]);
  64         packet_flush(fd[1]);
  65
  66         len = packet_read_line(fd[0], buf, sizeof(buf));
  67         if (!len)
  68                 die("git-tar-tree: expected ACK/NAK, got EOF");
  69         if (buf[len-1] == '\n')
  70                 buf[--len] = 0;
  71         if (strcmp(buf, "ACK")) {
  72                 if (5 < len && !strncmp(buf, "NACK ", 5))
  73                         die("git-tar-tree: NACK %s", buf + 5);
  74                 die("git-tar-tree: protocol error");
  75         }
  76         /* expect a flush */
  77         len = packet_read_line(fd[0], buf, sizeof(buf));
  78         if (len)
  79                 die("git-tar-tree: expected a flush");
  80
  81         /* Now, start reading from fd[0] and spit it out to stdout */
  82         ret = copy_fd(fd[0], 1);
  83         close(fd[0]);
  84
  85         ret |= finish_connect(pid);
  86         return !!ret;
  87 }
  88
  89 int cmd_tar_tree(int argc, const char **argv, const char *prefix)
  90 {
  91         if (argc < 2)
  92                 usage(tar_tree_usage);
  93         if (!strncmp("--remote=", argv[1], 9))
  94                 return remote_tar(argc, argv);
  95         return generate_tar(argc, argv, prefix);
  96 }
  97
  98 /* ustar header + extended global header content */
  99 #define HEADERSIZE (2 * RECORDSIZE)
 100
 101 int cmd_get_tar_commit_id(int argc, const char **argv, const char *prefix)
 102 {
 103         char buffer[HEADERSIZE];
 104         struct ustar_header *header = (struct ustar_header *)buffer;
 105         char *content = buffer + RECORDSIZE;
 106         ssize_t n;
 107
 108         n = xread(0, buffer, HEADERSIZE);
 109         if (n < HEADERSIZE)
 110                 die("git-get-tar-commit-id: read error");
 111         if (header->typeflag[0] != 'g')
 112                 return 1;
 113         if (memcmp(content, "52 comment=", 11))
 114                 return 1;
 115
 116         n = xwrite(1, content + 11, 41);
 117         if (n < 41)
 118                 die("git-get-tar-commit-id: write error");
 119
 120         return 0;
 121 }