daemon: if one of the standard fds is missing open it to /dev/null
[git/mingw/j6t.git] / ssh-upload.c
blob2da66618fcdd6a0f5af35c9f30554a65eb427427
1 #ifndef COUNTERPART_ENV_NAME
2 #define COUNTERPART_ENV_NAME "GIT_SSH_FETCH"
3 #endif
4 #ifndef COUNTERPART_PROGRAM_NAME
5 #define COUNTERPART_PROGRAM_NAME "git-ssh-fetch"
6 #endif
7 #ifndef MY_PROGRAM_NAME
8 #define MY_PROGRAM_NAME "git-ssh-upload"
9 #endif
11 #include "cache.h"
12 #include "rsh.h"
13 #include "refs.h"
15 #include <string.h>
17 static unsigned char local_version = 1;
18 static unsigned char remote_version = 0;
20 static int verbose = 0;
22 static int serve_object(int fd_in, int fd_out) {
23 ssize_t size;
24 unsigned char sha1[20];
25 signed char remote;
26 int posn = 0;
27 do {
28 size = read(fd_in, sha1 + posn, 20 - posn);
29 if (size < 0) {
30 perror("git-ssh-upload: read ");
31 return -1;
33 if (!size)
34 return -1;
35 posn += size;
36 } while (posn < 20);
38 if (verbose)
39 fprintf(stderr, "Serving %s\n", sha1_to_hex(sha1));
41 remote = 0;
43 if (!has_sha1_file(sha1)) {
44 fprintf(stderr, "git-ssh-upload: could not find %s\n",
45 sha1_to_hex(sha1));
46 remote = -1;
49 write(fd_out, &remote, 1);
51 if (remote < 0)
52 return 0;
54 return write_sha1_to_fd(fd_out, sha1);
57 static int serve_version(int fd_in, int fd_out)
59 if (read(fd_in, &remote_version, 1) < 1)
60 return -1;
61 write(fd_out, &local_version, 1);
62 return 0;
65 static int serve_ref(int fd_in, int fd_out)
67 char ref[PATH_MAX];
68 unsigned char sha1[20];
69 int posn = 0;
70 signed char remote = 0;
71 do {
72 if (read(fd_in, ref + posn, 1) < 1)
73 return -1;
74 posn++;
75 } while (ref[posn - 1]);
77 if (verbose)
78 fprintf(stderr, "Serving %s\n", ref);
80 if (get_ref_sha1(ref, sha1))
81 remote = -1;
82 write(fd_out, &remote, 1);
83 if (remote)
84 return 0;
85 write(fd_out, sha1, 20);
86 return 0;
90 static void service(int fd_in, int fd_out) {
91 char type;
92 int retval;
93 do {
94 retval = read(fd_in, &type, 1);
95 if (retval < 1) {
96 if (retval < 0)
97 perror("git-ssh-upload: read ");
98 return;
100 if (type == 'v' && serve_version(fd_in, fd_out))
101 return;
102 if (type == 'o' && serve_object(fd_in, fd_out))
103 return;
104 if (type == 'r' && serve_ref(fd_in, fd_out))
105 return;
106 } while (1);
109 static const char ssh_push_usage[] =
110 MY_PROGRAM_NAME " [-c] [-t] [-a] [-w ref] commit-id url";
112 int main(int argc, char **argv)
114 int arg = 1;
115 char *commit_id;
116 char *url;
117 int fd_in, fd_out;
118 const char *prog;
119 unsigned char sha1[20];
120 char hex[41];
122 prog = getenv(COUNTERPART_ENV_NAME);
123 if (!prog) prog = COUNTERPART_PROGRAM_NAME;
125 setup_git_directory();
127 while (arg < argc && argv[arg][0] == '-') {
128 if (argv[arg][1] == 'w')
129 arg++;
130 arg++;
132 if (argc < arg + 2)
133 usage(ssh_push_usage);
134 commit_id = argv[arg];
135 url = argv[arg + 1];
136 if (get_sha1(commit_id, sha1))
137 die("Not a valid object name %s", commit_id);
138 memcpy(hex, sha1_to_hex(sha1), sizeof(hex));
139 argv[arg] = hex;
141 if (setup_connection(&fd_in, &fd_out, prog, url, arg, argv + 1))
142 return 1;
144 service(fd_in, fd_out);
145 return 0;