Code

Leave merge failures in the filesystem
[git.git] / ssh-push.c
1 #include "cache.h"
2 #include "rsh.h"
3 #include "refs.h"
5 unsigned char local_version = 1;
6 unsigned char remote_version = 0;
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 }
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 }
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 }
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);
106 int main(int argc, char **argv)
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";
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;
128         service(fd_in, fd_out);
129         return 0;