blob: c2014529b25be22fec817ca014e964ebe5157037 [file] [log] [blame]
/*
* Copyright (C) 2009-2011 the libgit2 contributors
*
* This file is part of libgit2, distributed under the GNU GPL v2 with
* a Linking Exception. For full terms see the included COPYING file.
*/
#include "git2/net.h"
#include "git2/common.h"
#include "git2/types.h"
#include "git2/errors.h"
#include "git2/net.h"
#include "git2/revwalk.h"
#include "vector.h"
#include "transport.h"
#include "pkt.h"
#include "common.h"
#include "netops.h"
#include "filebuf.h"
#include "repository.h"
#include "fetch.h"
typedef struct {
git_transport parent;
GIT_SOCKET socket;
git_vector refs;
git_remote_head **heads;
git_transport_caps caps;
char buff[1024];
gitno_buffer buf;
#ifdef GIT_WIN32
WSADATA wsd;
#endif
} transport_git;
/*
* Create a git procol request.
*
* For example: 0035git-upload-pack /libgit2/libgit2\0host=github.com\0
*/
static int gen_proto(git_buf *request, const char *cmd, const char *url)
{
char *delim, *repo;
char default_command[] = "git-upload-pack";
char host[] = "host=";
int len;
delim = strchr(url, '/');
if (delim == NULL)
return git__throw(GIT_EOBJCORRUPTED, "Failed to create proto-request: malformed URL");
repo = delim;
delim = strchr(url, ':');
if (delim == NULL)
delim = strchr(url, '/');
if (cmd == NULL)
cmd = default_command;
len = 4 + strlen(cmd) + 1 + strlen(repo) + 1 + strlen(host) + (delim - url) + 1;
git_buf_grow(request, len);
git_buf_printf(request, "%04x%s %s%c%s", len, cmd, repo, 0, host);
git_buf_put(request, url, delim - url);
git_buf_putc(request, '\0');
return git_buf_oom(request);
}
static int send_request(GIT_SOCKET s, const char *cmd, const char *url)
{
int error;
git_buf request = GIT_BUF_INIT;
error = gen_proto(&request, cmd, url);
if (error < GIT_SUCCESS)
goto cleanup;
error = gitno_send(s, request.ptr, request.size, 0);
cleanup:
git_buf_free(&request);
return error;
}
/*
* Parse the URL and connect to a server, storing the socket in
* out. For convenience this also takes care of asking for the remote
* refs
*/
static int do_connect(transport_git *t, const char *url)
{
GIT_SOCKET s;
char *host, *port;
const char prefix[] = "git://";
int error, connected = 0;
if (!git__prefixcmp(url, prefix))
url += strlen(prefix);
error = gitno_extract_host_and_port(&host, &port, url, GIT_DEFAULT_PORT);
if (error < GIT_SUCCESS)
return error;
s = gitno_connect(host, port);
connected = 1;
error = send_request(s, NULL, url);
t->socket = s;
git__free(host);
git__free(port);
if (error < GIT_SUCCESS && s > 0)
close(s);
if (!connected)
error = git__throw(GIT_EOSERR, "Failed to connect to any of the addresses");
return error;
}
/*
* Read from the socket and store the references in the vector
*/
static int store_refs(transport_git *t)
{
gitno_buffer *buf = &t->buf;
git_vector *refs = &t->refs;
int error = GIT_SUCCESS;
const char *line_end, *ptr;
git_pkt *pkt;
while (1) {
error = gitno_recv(buf);
if (error < GIT_SUCCESS)
return git__rethrow(GIT_EOSERR, "Failed to receive data");
if (error == GIT_SUCCESS) /* Orderly shutdown, so exit */
return GIT_SUCCESS;
ptr = buf->data;
while (1) {
if (buf->offset == 0)
break;
error = git_pkt_parse_line(&pkt, ptr, &line_end, buf->offset);
/*
* If the error is GIT_ESHORTBUFFER, it means the buffer
* isn't long enough to satisfy the request. Break out and
* wait for more input.
* On any other error, fail.
*/
if (error == GIT_ESHORTBUFFER) {
break;
}
if (error < GIT_SUCCESS) {
return error;
}
/* Get rid of the part we've used already */
gitno_consume(buf, line_end);
error = git_vector_insert(refs, pkt);
if (error < GIT_SUCCESS)
return error;
if (pkt->type == GIT_PKT_FLUSH)
return GIT_SUCCESS;
}
}
return error;
}
static int detect_caps(transport_git *t)
{
git_vector *refs = &t->refs;
git_pkt_ref *pkt;
git_transport_caps *caps = &t->caps;
const char *ptr;
pkt = git_vector_get(refs, 0);
/* No refs or capabilites, odd but not a problem */
if (pkt == NULL || pkt->capabilities == NULL)
return GIT_SUCCESS;
ptr = pkt->capabilities;
while (ptr != NULL && *ptr != '\0') {
if (*ptr == ' ')
ptr++;
if(!git__prefixcmp(ptr, GIT_CAP_OFS_DELTA)) {
caps->common = caps->ofs_delta = 1;
ptr += strlen(GIT_CAP_OFS_DELTA);
continue;
}
/* We don't know this capability, so skip it */
ptr = strchr(ptr, ' ');
}
return GIT_SUCCESS;
}
/*
* Since this is a network connection, we need to parse and store the
* pkt-lines at this stage and keep them there.
*/
static int git_connect(git_transport *transport, int direction)
{
transport_git *t = (transport_git *) transport;
int error = GIT_SUCCESS;
if (direction == GIT_DIR_PUSH)
return git__throw(GIT_EINVALIDARGS, "Pushing is not supported with the git protocol");
t->parent.direction = direction;
error = git_vector_init(&t->refs, 16, NULL);
if (error < GIT_SUCCESS)
goto cleanup;
/* Connect and ask for the refs */
error = do_connect(t, transport->url);
if (error < GIT_SUCCESS)
return error;
gitno_buffer_setup(&t->buf, t->buff, sizeof(t->buff), t->socket);
t->parent.connected = 1;
error = store_refs(t);
if (error < GIT_SUCCESS)
return error;
error = detect_caps(t);
cleanup:
if (error < GIT_SUCCESS) {
git_vector_free(&t->refs);
}
return error;
}
static int git_ls(git_transport *transport, git_headarray *array)
{
transport_git *t = (transport_git *) transport;
git_vector *refs = &t->refs;
int len = 0;
unsigned int i;
array->heads = git__calloc(refs->length, sizeof(git_remote_head *));
if (array->heads == NULL)
return GIT_ENOMEM;
for (i = 0; i < refs->length; ++i) {
git_pkt *p = git_vector_get(refs, i);
if (p->type != GIT_PKT_REF)
continue;
++len;
array->heads[i] = &(((git_pkt_ref *) p)->head);
}
array->len = len;
t->heads = array->heads;
return GIT_SUCCESS;
}
static int git_negotiate_fetch(git_transport *transport, git_repository *repo, git_headarray *wants)
{
transport_git *t = (transport_git *) transport;
git_revwalk *walk;
git_reference *ref;
git_strarray refs;
git_oid oid;
int error;
unsigned int i;
gitno_buffer *buf = &t->buf;
error = git_pkt_send_wants(wants, &t->caps, t->socket);
if (error < GIT_SUCCESS)
return git__rethrow(error, "Failed to send wants list");
error = git_reference_listall(&refs, repo, GIT_REF_LISTALL);
if (error < GIT_ERROR)
return git__rethrow(error, "Failed to list all references");
error = git_revwalk_new(&walk, repo);
if (error < GIT_ERROR) {
error = git__rethrow(error, "Failed to list all references");
goto cleanup;
}
git_revwalk_sorting(walk, GIT_SORT_TIME);
for (i = 0; i < refs.count; ++i) {
/* No tags */
if (!git__prefixcmp(refs.strings[i], GIT_REFS_TAGS_DIR))
continue;
error = git_reference_lookup(&ref, repo, refs.strings[i]);
if (error < GIT_ERROR) {
error = git__rethrow(error, "Failed to lookup %s", refs.strings[i]);
goto cleanup;
}
if (git_reference_type(ref) == GIT_REF_SYMBOLIC)
continue;
error = git_revwalk_push(walk, git_reference_oid(ref));
if (error < GIT_ERROR) {
error = git__rethrow(error, "Failed to push %s", refs.strings[i]);
goto cleanup;
}
}
git_strarray_free(&refs);
/*
* We don't support any kind of ACK extensions, so the negotiation
* boils down to sending what we have and listening for an ACK
* every once in a while.
*/
i = 0;
while ((error = git_revwalk_next(&oid, walk)) == GIT_SUCCESS) {
error = git_pkt_send_have(&oid, t->socket);
i++;
if (i % 20 == 0) {
const char *ptr = buf->data, *line_end;
git_pkt *pkt;
git_pkt_send_flush(t->socket);
while (1) {
/* Wait for max. 1 second */
error = gitno_select_in(buf, 1, 0);
if (error < GIT_SUCCESS) {
error = git__throw(GIT_EOSERR, "Error in select");
} else if (error == 0) {
/*
* Some servers don't respond immediately, so if this
* happens, we keep sending information until it
* answers.
*/
break;
}
error = gitno_recv(buf);
if (error < GIT_SUCCESS) {
error = git__rethrow(error, "Error receiving data");
goto cleanup;
}
error = git_pkt_parse_line(&pkt, ptr, &line_end, buf->offset);
if (error == GIT_ESHORTBUFFER)
continue;
if (error < GIT_SUCCESS) {
error = git__rethrow(error, "Failed to get answer");
goto cleanup;
}
gitno_consume(buf, line_end);
if (pkt->type == GIT_PKT_ACK) {
git__free(pkt);
error = GIT_SUCCESS;
goto done;
} else if (pkt->type == GIT_PKT_NAK) {
git__free(pkt);
break;
} else {
error = git__throw(GIT_ERROR, "Got unexpected pkt type");
goto cleanup;
}
}
}
}
if (error == GIT_EREVWALKOVER)
error = GIT_SUCCESS;
done:
git_pkt_send_flush(t->socket);
git_pkt_send_done(t->socket);
cleanup:
git_revwalk_free(walk);
return error;
}
static int git_send_flush(git_transport *transport)
{
transport_git *t = (transport_git *) transport;
return git_pkt_send_flush(t->socket);
}
static int git_send_done(git_transport *transport)
{
transport_git *t = (transport_git *) transport;
return git_pkt_send_done(t->socket);
}
static int git_download_pack(char **out, git_transport *transport, git_repository *repo)
{
transport_git *t = (transport_git *) transport;
int error = GIT_SUCCESS;
gitno_buffer *buf = &t->buf;
git_pkt *pkt;
const char *line_end, *ptr;
/*
* For now, we ignore everything and wait for the pack
*/
while (1) {
ptr = buf->data;
/* Whilst we're searching for the pack */
while (1) {
if (buf->offset == 0) {
break;
}
error = git_pkt_parse_line(&pkt, ptr, &line_end, buf->offset);
if (error == GIT_ESHORTBUFFER)
break;
if (error < GIT_SUCCESS)
return error;
if (pkt->type == GIT_PKT_PACK) {
git__free(pkt);
return git_fetch__download_pack(out, buf->data, buf->offset, t->socket, repo);
}
/* For now we don't care about anything */
git__free(pkt);
gitno_consume(buf, line_end);
}
error = gitno_recv(buf);
if (error < GIT_SUCCESS)
return git__rethrow(GIT_EOSERR, "Failed to receive data");
if (error == 0) { /* Orderly shutdown */
return GIT_SUCCESS;
}
}
}
static int git_close(git_transport *transport)
{
transport_git *t = (transport_git*) transport;
int error;
/* Can't do anything if there's an error, so don't bother checking */
git_pkt_send_flush(t->socket);
error = gitno_close(t->socket);
if (error < 0)
error = git__throw(GIT_EOSERR, "Failed to close socket");
#ifdef GIT_WIN32
WSACleanup();
#endif
return error;
}
static void git_free(git_transport *transport)
{
transport_git *t = (transport_git *) transport;
git_vector *refs = &t->refs;
unsigned int i;
for (i = 0; i < refs->length; ++i) {
git_pkt *p = git_vector_get(refs, i);
git_pkt_free(p);
}
git_vector_free(refs);
git__free(t->heads);
git__free(t->parent.url);
git__free(t);
}
int git_transport_git(git_transport **out)
{
transport_git *t;
#ifdef GIT_WIN32
int ret;
#endif
t = git__malloc(sizeof(transport_git));
if (t == NULL)
return GIT_ENOMEM;
memset(t, 0x0, sizeof(transport_git));
t->parent.connect = git_connect;
t->parent.ls = git_ls;
t->parent.negotiate_fetch = git_negotiate_fetch;
t->parent.send_flush = git_send_flush;
t->parent.send_done = git_send_done;
t->parent.download_pack = git_download_pack;
t->parent.close = git_close;
t->parent.free = git_free;
*out = (git_transport *) t;
#ifdef GIT_WIN32
ret = WSAStartup(MAKEWORD(2,2), &t->wsd);
if (ret != 0) {
git_free(*out);
return git__throw(GIT_EOSERR, "Winsock init failed");
}
#endif
return GIT_SUCCESS;
}