ssh-push.c

   1 #include "cache.h"
   2 #include "rsh.h"
   3 #include "refs.h"
   4
   5 unsigned char local_version = 1;
   6 unsigned char remote_version = 0;
   7
   8 int serve_object(int fd_in, int fd_out) {
   9         ssize_t size;
  10         int posn = 0;
  11         char sha1[20];
  12         unsigned long objsize;
  13         void *buf;
  14         signed char remote;
  15         do {
  16                 size = read(fd_in, sha1 + posn, 20 - posn);
  17                 if (size < 0) {
  18                         perror("git-ssh-push: read ");
  19                         return -1;
  20                 }
  21                 if (!size)
  22                         return -1;
  23                 posn += size;
  24         } while (posn < 20);
  25
  26         /* fprintf(stderr, "Serving %s\n", sha1_to_hex(sha1)); */
  27         remote = 0;
  28
  29         buf = map_sha1_file(sha1, &objsize);
  30
  31         if (!buf) {
  32                 fprintf(stderr, "git-ssh-push: could not find %s\n",
  33                         sha1_to_hex(sha1));
  34                 remote = -1;
  35         }
  36
  37         write(fd_out, &remote, 1);
  38
  39         if (remote < 0)
  40                 return 0;
  41
  42         posn = 0;
  43         do {
  44                 size = write(fd_out, buf + posn, objsize - posn);
  45                 if (size <= 0) {
  46                         if (!size) {
  47                                 fprintf(stderr, "git-ssh-push: write closed");
  48                         } else {
  49                                 perror("git-ssh-push: write ");
  50                         }
  51                         return -1;
  52                 }
  53                 posn += size;
  54         } while (posn < objsize);
  55         return 0;
  56 }
  57
  58 int serve_version(int fd_in, int fd_out)
  59 {
  60         if (read(fd_in, &remote_version, 1) < 1)
  61                 return -1;
  62         write(fd_out, &local_version, 1);
  63         return 0;
  64 }
  65
  66 int serve_ref(int fd_in, int fd_out)
  67 {
  68         char ref[PATH_MAX];
  69         unsigned char sha1[20];
  70         int posn = 0;
  71         signed char remote = 0;
  72         do {
  73                 if (read(fd_in, ref + posn, 1) < 1)
  74                         return -1;
  75                 posn++;
  76         } while (ref[posn - 1]);
  77         if (get_ref_sha1(ref, sha1))
  78                 remote = -1;
  79         write(fd_out, &remote, 1);
  80         if (remote)
  81                 return 0;
  82         write(fd_out, sha1, 20);
  83         return 0;
  84 }
  85
  86
  87 void service(int fd_in, int fd_out) {
  88         char type;
  89         int retval;
  90         do {
  91                 retval = read(fd_in, &type, 1);
  92                 if (retval < 1) {
  93                         if (retval < 0)
  94                                 perror("git-ssh-push: read ");
  95                         return;
  96                 }
  97                 if (type == 'v' && serve_version(fd_in, fd_out))
  98                         return;
  99                 if (type == 'o' && serve_object(fd_in, fd_out))
 100                         return;
 101                 if (type == 'r' && serve_ref(fd_in, fd_out))
 102                         return;
 103         } while (1);
 104 }
 105
 106 int main(int argc, char **argv)
 107 {
 108         int arg = 1;
 109         char *commit_id;
 110         char *url;
 111         int fd_in, fd_out;
 112         const char *prog = getenv("GIT_SSH_PULL") ? : "git-ssh-pull";
 113
 114         while (arg < argc && argv[arg][0] == '-') {
 115                 if (argv[arg][1] == 'w')
 116                         arg++;
 117                 arg++;
 118         }
 119         if (argc < arg + 2) {
 120                 usage("git-ssh-push [-c] [-t] [-a] [-w ref] commit-id url");
 121                 return 1;
 122         }
 123         commit_id = argv[arg];
 124         url = argv[arg + 1];
 125         if (setup_connection(&fd_in, &fd_out, prog, url, arg, argv + 1))
 126                 return 1;
 127
 128         service(fd_in, fd_out);
 129         return 0;
 130 }