1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include "cache.h"
#include "commit.h"
#include <errno.h>
#include <stdio.h>
#include "rsh.h"
static int tree = 0;
static int commits = 0;
static int all = 0;
static int fd_in;
static int fd_out;
static int fetch(unsigned char *sha1)
{
if (has_sha1_file(sha1))
return 0;
write(fd_out, sha1, 20);
return write_sha1_from_fd(sha1, fd_in);
}
static int process_tree(unsigned char *sha1)
{
struct tree *tree = lookup_tree(sha1);
struct tree_entry_list *entries;
if (parse_tree(tree))
return -1;
for (entries = tree->entries; entries; entries = entries->next) {
/*
fprintf(stderr, "Tree %s ", sha1_to_hex(sha1));
fprintf(stderr, "needs %s\n",
sha1_to_hex(entries->item.tree->object.sha1));
*/
if (fetch(entries->item.tree->object.sha1)) {
return error("Missing item %s",
sha1_to_hex(entries->item.tree->object.sha1));
}
if (entries->directory) {
if (process_tree(entries->item.tree->object.sha1))
return -1;
}
}
return 0;
}
static int process_commit(unsigned char *sha1)
{
struct commit *obj = lookup_commit(sha1);
if (fetch(sha1)) {
return error("Fetching %s", sha1_to_hex(sha1));
}
if (parse_commit(obj))
return -1;
if (tree) {
if (fetch(obj->tree->object.sha1))
return -1;
if (process_tree(obj->tree->object.sha1))
return -1;
if (!all)
tree = 0;
}
if (commits) {
struct commit_list *parents = obj->parents;
for (; parents; parents = parents->next) {
if (has_sha1_file(parents->item->object.sha1))
continue;
if (fetch(parents->item->object.sha1)) {
/* The server might not have it, and
* we don't mind.
*/
error("Missing tree %s; continuing",
sha1_to_hex(parents->item->object.sha1));
continue;
}
if (process_commit(parents->item->object.sha1))
return -1;
}
}
return 0;
}
int main(int argc, char **argv)
{
char *commit_id;
char *url;
int arg = 1;
unsigned char sha1[20];
while (arg < argc && argv[arg][0] == '-') {
if (argv[arg][1] == 't') {
tree = 1;
} else if (argv[arg][1] == 'c') {
commits = 1;
} else if (argv[arg][1] == 'a') {
all = 1;
tree = 1;
commits = 1;
}
arg++;
}
if (argc < arg + 2) {
usage("rpull [-c] [-t] [-a] commit-id url");
return 1;
}
commit_id = argv[arg];
url = argv[arg + 1];
if (setup_connection(&fd_in, &fd_out, "rpush", url, arg, argv + 1))
return 1;
get_sha1_hex(commit_id, sha1);
if (fetch(sha1))
return 1;
if (process_commit(sha1))
return 1;
return 0;
}
|
