Make git-fetch-pack and git-upload-pack negotiate needs/haves fully

Now the only piece missing is actually generating the pack-file.

diff --git a/fetch-pack.c b/fetch-pack.c
index e9035a1..151c1ff 100644 (file)
--- a/fetch-pack.c
+++ b/fetch-pack.c
@@ -1,4 +1,5 @@
 #include "cache.h"
+#include "refs.h"
 #include "pkt-line.h"
 
 static const char fetch_pack_usage[] = "git-fetch-pack [host:]directory [heads]* < mycommitlist";
@@ -22,11 +23,13 @@ static int get_ack(int fd, unsigned char *result_sha1)
        die("git-fetch_pack: expected ACK/NAK, got '%s'", line);
 }
 
-static int find_common(int fd[2], unsigned char *result_sha1)
+static int find_common(int fd[2], unsigned char *result_sha1, unsigned char *remote)
 {
        static char line[1000];
        int count = 0, flushes = 0;
 
+       packet_write(fd[1], "want %s\n", sha1_to_hex(remote));
+       packet_flush(fd[1]);
        while (fgets(line, sizeof(line), stdin) != NULL) {
                unsigned char sha1[20];
                if (get_sha1_hex(line, sha1))
@@ -57,8 +60,47 @@ static int find_common(int fd[2], unsigned char *result_sha1)
        return -1;
 }
 
-static int get_remote_heads(int fd, int nr_match, char **match)
+static int get_old_sha1(const char *refname, unsigned char *sha1)
+{
+       static char pathname[PATH_MAX];
+       const char *git_dir;
+       int fd, ret;
+
+       git_dir = gitenv(GIT_DIR_ENVIRONMENT) ? : DEFAULT_GIT_DIR_ENVIRONMENT;
+       snprintf(pathname, sizeof(pathname), "%s/%s", git_dir, refname);
+       fd = open(pathname, O_RDONLY);
+       ret = -1;
+       if (fd >= 0) {
+               char buffer[60];
+               if (read(fd, buffer, sizeof(buffer)) >= 40)
+                       ret = get_sha1_hex(buffer, sha1);
+               close(fd);
+       }
+       return ret;
+}
+
+static int check_ref(const char *refname, const unsigned char *sha1)
 {
+       unsigned char mysha1[20];
+       char oldhex[41];
+
+       if (get_old_sha1(refname, mysha1) < 0)
+               memset(mysha1, 0, 20);
+
+       if (!memcmp(sha1, mysha1, 20)) {
+               printf("%s: unchanged\n", refname);
+               return 0;
+       }
+       
+       memcpy(oldhex, sha1_to_hex(mysha1), 41);
+       printf("%s: %s (%s)\n", refname, sha1_to_hex(sha1), oldhex);
+       return 1;
+}
+
+static int get_remote_heads(int fd, int nr_match, char **match, unsigned char *result)
+{
+       int count = 0;
+
        for (;;) {
                static char line[1000];
                unsigned char sha1[20];
@@ -75,20 +117,31 @@ static int get_remote_heads(int fd, int nr_match, char **match)
                refname = line+41;
                if (nr_match && !path_match(refname, nr_match, match))
                        continue;
-               printf("%s %s\n", sha1_to_hex(sha1), refname);
+               if (check_ref(refname, sha1)) {
+                       count++;
+                       memcpy(result, sha1, 20);
+               }
        }
-       return 0;
+       return count;
 }
 
 static int fetch_pack(int fd[2], int nr_match, char **match)
 {
-       unsigned char sha1[20];
+       unsigned char sha1[20], remote[20];
+       int heads;
 
-       get_remote_heads(fd[0], nr_match, match);
-       if (find_common(fd, sha1) < 0)
+       heads = get_remote_heads(fd[0], nr_match, match, remote);
+       if (heads != 1) {
+               packet_flush(fd[1]);
+               die(heads ? "multiple remote heads" : "no matching remote head");
+       }
+       if (find_common(fd, sha1, remote) < 0)
                die("git-fetch-pack: no common commits");
-       printf("common commit: %s\n", sha1_to_hex(sha1));
-       return 0;
+       close(fd[1]);
+       dup2(fd[0], 0);
+       close(fd[0]);
+       execlp("git-unpack-objects", "git-unpack-objects", NULL);
+       die("git-unpack-objects exec failed");
 }
 
 int main(int argc, char **argv)

diff --git a/upload-pack.c b/upload-pack.c
index 18b8e1a..b454c9e 100644 (file)
--- a/upload-pack.c
+++ b/upload-pack.c
@@ -4,11 +4,37 @@
 
 static const char upload_pack_usage[] = "git-upload-pack <dir>";
 
+#define MAX_HAS (16)
+#define MAX_NEEDS (16)
+static int nr_has = 0, nr_needs = 0;
+static unsigned char has_sha1[MAX_HAS][20];
+static unsigned char needs_sha1[MAX_NEEDS][20];
+
+static void create_pack_file(void)
+{
+       /*
+        * Here, we should do
+        *
+        *      git-rev-list --objects needs_sha1 --not has_sha1 |
+        *              git-pack-objects --stdout
+        *
+        * but we don't.
+        */
+}
+
 static int got_sha1(char *hex, unsigned char *sha1)
 {
+       int nr;
        if (get_sha1_hex(hex, sha1))
                die("git-upload-pack: expected SHA1 object, got '%s'", hex);
-       return has_sha1_file(sha1);
+       if (!has_sha1_file(sha1))
+               return 0;
+       nr = nr_has;
+       if (nr < MAX_HAS) {
+               memcpy(has_sha1[nr], sha1, 20);
+               nr_has = nr+1;
+       }
+       return 1;
 }
 
 static int get_common_commits(void)
@@ -55,6 +81,29 @@ static int get_common_commits(void)
        return 0;
 }
 
+static int receive_needs(void)
+{
+       static char line[1000];
+       int len, needs;
+
+       needs = 0;
+       for (;;) {
+               len = packet_read_line(0, line, sizeof(line));
+               if (!len)
+                       return needs;
+
+               /*
+                * This is purely theoretical right now: git-fetch-pack only
+                * ever asks for a single HEAD
+                */
+               if (needs >= MAX_NEEDS)
+                       die("I'm only doing a max of %d requests", MAX_NEEDS);
+               if (strncmp("want ", line, 5) || get_sha1_hex(line+5, needs_sha1[needs]))
+                       die("git-upload-pack: protocol error, expected to get sha, not '%s'", line);
+               needs++;
+       }
+}
+
 static int send_ref(const char *refname, const unsigned char *sha1)
 {
        packet_write(1, "%s %s\n", sha1_to_hex(sha1), refname);
@@ -65,7 +114,11 @@ static int upload_pack(void)
 {
        for_each_ref(send_ref);
        packet_flush(1);
+       nr_needs = receive_needs();
+       if (!nr_needs)
+               return 0;
        get_common_commits();
+       create_pack_file();
        return 0;
 }