clean up and optimize nth_packed_object_sha1() usage
[git/fastimport.git] / ssh-upload.c
blob2f045727875707198dd1763d75c9e8c7406e9042
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 static unsigned char local_version = 1;
16 static unsigned char remote_version;
18 static int verbose;
20 static int serve_object(int fd_in, int fd_out) {
21 ssize_t size;
22 unsigned char sha1[20];
23 signed char remote;
25 size = read_in_full(fd_in, sha1, 20);
26 if (size < 0) {
27 perror("git-ssh-upload: read ");
28 return -1;
30 if (!size)
31 return -1;
33 if (verbose)
34 fprintf(stderr, "Serving %s\n", sha1_to_hex(sha1));
36 remote = 0;
38 if (!has_sha1_file(sha1)) {
39 fprintf(stderr, "git-ssh-upload: could not find %s\n",
40 sha1_to_hex(sha1));
41 remote = -1;
44 if (write_in_full(fd_out, &remote, 1) != 1)
45 return 0;
47 if (remote < 0)
48 return 0;
50 return write_sha1_to_fd(fd_out, sha1);
53 static int serve_version(int fd_in, int fd_out)
55 if (xread(fd_in, &remote_version, 1) < 1)
56 return -1;
57 write_in_full(fd_out, &local_version, 1);
58 return 0;
61 static int serve_ref(int fd_in, int fd_out)
63 char ref[PATH_MAX];
64 unsigned char sha1[20];
65 int posn = 0;
66 signed char remote = 0;
67 do {
68 if (posn >= PATH_MAX || xread(fd_in, ref + posn, 1) < 1)
69 return -1;
70 posn++;
71 } while (ref[posn - 1]);
73 if (verbose)
74 fprintf(stderr, "Serving %s\n", ref);
76 if (get_ref_sha1(ref, sha1))
77 remote = -1;
78 if (write_in_full(fd_out, &remote, 1) != 1)
79 return 0;
80 if (remote)
81 return 0;
82 write_in_full(fd_out, sha1, 20);
83 return 0;
87 static void service(int fd_in, int fd_out) {
88 char type;
89 int retval;
90 do {
91 retval = xread(fd_in, &type, 1);
92 if (retval < 1) {
93 if (retval < 0)
94 perror("git-ssh-upload: read ");
95 return;
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);
106 static const char ssh_push_usage[] =
107 MY_PROGRAM_NAME " [-c] [-t] [-a] [-w ref] commit-id url";
109 int main(int argc, char **argv)
111 int arg = 1;
112 char *commit_id;
113 char *url;
114 int fd_in, fd_out;
115 const char *prog;
116 unsigned char sha1[20];
117 char hex[41];
119 prog = getenv(COUNTERPART_ENV_NAME);
120 if (!prog) prog = COUNTERPART_PROGRAM_NAME;
122 setup_git_directory();
124 while (arg < argc && argv[arg][0] == '-') {
125 if (argv[arg][1] == 'w')
126 arg++;
127 arg++;
129 if (argc < arg + 2)
130 usage(ssh_push_usage);
131 commit_id = argv[arg];
132 url = argv[arg + 1];
133 if (get_sha1(commit_id, sha1))
134 die("Not a valid object name %s", commit_id);
135 memcpy(hex, sha1_to_hex(sha1), sizeof(hex));
136 argv[arg] = hex;
138 if (setup_connection(&fd_in, &fd_out, prog, url, arg, argv + 1))
139 return 1;
141 service(fd_in, fd_out);
142 return 0;