Factor out the ssh connection stuff from send-pack.c

I want to use it for git-fetch-pack too.

1 files changed, 86 insertions, 0 deletions

diff --git a/connect.c b/connect.c
new file mode 100644
index 0000000000..60b87f5867
--- /dev/null
+++ b/connect.c
@@ -0,0 +1,86 @@
+#include "cache.h"
+#include <sys/wait.h>
+
+/*
+ * First, make it shell-safe.  We do this by just disallowing any
+ * special characters. Somebody who cares can do escaping and let
+ * through the rest. But since we're doing to feed this to ssh as
+ * a command line, we're going to be pretty damn anal for now.
+ */
+static char *shell_safe(char *url)
+{
+	char *n = url;
+	unsigned char c;
+	static const char flags[256] = {
+		['0'...'9'] = 1,
+		['a'...'z'] = 1,
+		['A'...'Z'] = 1,
+		['.'] = 1, ['/'] = 1,
+		['-'] = 1, ['+'] = 1,
+		[':'] = 1
+	};
+
+	while ((c = *n++) != 0) {
+		if (flags[c] != 1)
+			die("I don't like '%c'. Sue me.", c);
+	}
+	return url;
+}
+
+/*
+ * Yeah, yeah, fixme. Need to pass in the heads etc.
+ */
+int git_connect(int fd[2], char *url, const char *prog)
+{
+	char command[1024];
+	const char *host, *path;
+	char *colon;
+	int pipefd[2][2];
+	pid_t pid;
+
+	url = shell_safe(url);
+	host = NULL;
+	path = url;
+	colon = strchr(url, ':');
+	if (colon) {
+		*colon = 0;
+		host = url;
+		path = colon+1;
+	}
+	snprintf(command, sizeof(command), "%s %s", prog, path);
+	if (pipe(pipefd[0]) < 0 || pipe(pipefd[1]) < 0)
+		die("unable to create pipe pair for communication");
+	pid = fork();
+	if (!pid) {
+		dup2(pipefd[1][0], 0);
+		dup2(pipefd[0][1], 1);
+		close(pipefd[0][0]);
+		close(pipefd[0][1]);
+		close(pipefd[1][0]);
+		close(pipefd[1][1]);
+		if (host)
+			execlp("ssh", "ssh", host, command, NULL);
+		else
+			execlp("sh", "sh", "-c", command, NULL);
+		die("exec failed");
+	}		
+	fd[0] = pipefd[0][0];
+	fd[1] = pipefd[1][1];
+	close(pipefd[0][1]);
+	close(pipefd[1][0]);
+	return pid;
+}
+
+int finish_connect(pid_t pid)
+{
+	int ret;
+
+	for (;;) {
+		ret = waitpid(pid, NULL, 0);
+		if (!ret)
+			break;
+		if (errno != EINTR)
+			break;
+	}
+	return ret;
+}