[PATCH] Various transport programs

[git.git] / rsh.c

diff --git a/rsh.c b/rsh.c
new file mode 100644 (file)
index 0000000..4d6a90b
--- /dev/null
+++ b/rsh.c
@@ -0,0 +1,63 @@
+#include "rsh.h"
+
+#include <string.h>
+#include <sys/socket.h>
+
+#include "cache.h"
+
+#define COMMAND_SIZE 4096
+
+int setup_connection(int *fd_in, int *fd_out, char *remote_prog, 
+                    char *url, int rmt_argc, char **rmt_argv)
+{
+       char *host;
+       char *path;
+       int sv[2];
+       char command[COMMAND_SIZE];
+       char *posn;
+       int i;
+
+       if (!strcmp(url, "-")) {
+               *fd_in = 0;
+               *fd_out = 1;
+               return 0;
+       }
+
+       host = strstr(url, "//");
+       if (!host) {
+               return error("Bad URL: %s", url);
+       }
+       host += 2;
+       path = strchr(host, '/');
+       if (!path) {
+               return error("Bad URL: %s", url);
+       }
+       *(path++) = '\0';
+       /* ssh <host> 'cd /<path>; stdio-pull <arg...> <commit-id>' */
+       snprintf(command, COMMAND_SIZE, 
+                "cd /%s; SHA1_FILE_DIRECTORY=objects %s",
+                path, remote_prog);
+       posn = command + strlen(command);
+       for (i = 0; i < rmt_argc; i++) {
+               *(posn++) = ' ';
+               strncpy(posn, rmt_argv[i], COMMAND_SIZE - (posn - command));
+               posn += strlen(rmt_argv[i]);
+               if (posn - command + 4 >= COMMAND_SIZE) {
+                       return error("Command line too long");
+               }
+       }
+       strcpy(posn, " -");
+       if (socketpair(AF_LOCAL, SOCK_STREAM, 0, sv)) {
+               return error("Couldn't create socket");
+       }
+       if (!fork()) {
+               close(sv[1]);
+               dup2(sv[0], 0);
+               dup2(sv[0], 1);
+               execlp("ssh", "ssh", host, command, NULL);
+       }
+       close(sv[0]);
+       *fd_in = sv[1];
+       *fd_out = sv[1];
+       return 0;
+}