FilesExpand file tree

development

/

fetch.c

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

191 lines (157 loc) · 4.67 KB

development

/

fetch.c

Copy path

Top

File metadata and controls

• Code
• Blame

191 lines (157 loc) · 4.67 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

/*

* 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/remote.h"

#include "git2/oid.h"

#include "git2/refs.h"

#include "git2/revwalk.h"

#include "common.h"

#include "transport.h"

#include "remote.h"

#include "refspec.h"

#include "fetch.h"

#include "netops.h"

static int filter_wants(git_remote *remote)

{

git_vector list;

git_headarray refs;

git_remote_head *head;

git_transport *t = remote->transport;

git_repository *repo = remote->repo;

const git_refspec *spec;

int error;

unsigned int i = 0;

error = git_vector_init(&list, 16, NULL);

if (error < GIT_SUCCESS)

return error;

error = t->ls(t, &refs);

if (error < GIT_SUCCESS) {

error = git__rethrow(error, "Failed to get remote ref list");

goto cleanup;

}

/*

* The fetch refspec can be NULL, and what this means is that the

* user didn't specify one. This is fine, as it means that we're

* not interested in any particular branch but just the remote's

* HEAD, which will be stored in FETCH_HEAD after the fetch.

*/

spec = git_remote_fetchspec(remote);

/*

* We need to handle HEAD separately, as we always want it, but it

* probably won't matcht he refspec.

*/

head = refs.heads[0];

if (refs.len > 0 && !strcmp(head->name, GIT_HEAD_FILE)) {

if (git_odb_exists(repo->db, &head->oid))

head->local = 1;

else

remote->need_pack = 1;

i = 1;

error = git_vector_insert(&list, refs.heads[0]);

if (error < GIT_SUCCESS)

goto cleanup;

}

for (; i < refs.len; ++i) {

head = refs.heads[i];

/* If it doesn't match the refpec, we don't want it */

error = git_refspec_src_match(spec, head->name);

if (error == GIT_ENOMATCH)

continue;

if (error < GIT_SUCCESS) {

error = git__rethrow(error, "Error matching remote ref name");

goto cleanup;

}

/* If we have the object, mark it so we don't ask for it */

if (git_odb_exists(repo->db, &head->oid))

head->local = 1;

else

remote->need_pack = 1;

error = git_vector_insert(&list, head);

if (error < GIT_SUCCESS)

goto cleanup;

}

remote->refs.len = list.length;

remote->refs.heads = (git_remote_head **) list.contents;

return GIT_SUCCESS;

cleanup:

git_vector_free(&list);

return error;

}

/*

* In this first version, we push all our refs in and start sending

* them out. When we get an ACK we hide that commit and continue

* traversing until we're done

*/

int git_fetch_negotiate(git_remote *remote)

{

int error;

git_transport *t = remote->transport;

error = filter_wants(remote);

if (error < GIT_SUCCESS)

return git__rethrow(error, "Failed to filter the reference list for wants");

/* Don't try to negotiate when we don't want anything */

if (remote->refs.len == 0)

return GIT_SUCCESS;

if (!remote->need_pack)

return GIT_SUCCESS;

/*

* Now we have everything set up so we can start tell the server

* what we want and what we have.

*/

return t->negotiate_fetch(t, remote->repo, &remote->refs);

}

int git_fetch_download_pack(char **out, git_remote *remote)

{

if(!remote->need_pack) {

*out = NULL;

return GIT_SUCCESS;

}

return remote->transport->download_pack(out, remote->transport, remote->repo);

}

/* Receiving data from a socket and storing it is pretty much the same for git and HTTP */

int git_fetch__download_pack(char **out, const char *buffered, size_t buffered_size,

GIT_SOCKET fd, git_repository *repo)

{

git_filebuf file;

int error;

char buff[1024], path[GIT_PATH_MAX];

static const char suff[] = "/objects/pack/pack-received";

gitno_buffer buf;

git_path_join(path, repo->path_repository, suff);

gitno_buffer_setup(&buf, buff, sizeof(buff), fd);

if (memcmp(buffered, "PACK", strlen("PACK"))) {

return git__throw(GIT_ERROR, "The pack doesn't start with the signature");

}

error = git_filebuf_open(&file, path, GIT_FILEBUF_TEMPORARY);

if (error < GIT_SUCCESS)

goto cleanup;

/* Part of the packfile has been received, don't loose it */

error = git_filebuf_write(&file, buffered, buffered_size);

if (error < GIT_SUCCESS)

goto cleanup;

while (1) {

error = git_filebuf_write(&file, buf.data, buf.offset);

if (error < GIT_SUCCESS)

goto cleanup;

gitno_consume_n(&buf, buf.offset);

error = gitno_recv(&buf);

if (error < GIT_SUCCESS)

goto cleanup;

if (error == 0) /* Orderly shutdown */

break;

}

*out = git__strdup(file.path_lock);

if (*out == NULL) {

error = GIT_ENOMEM;

goto cleanup;

}

/* A bit dodgy, but we need to keep the pack at the temporary path */

error = git_filebuf_commit_at(&file, file.path_lock);

cleanup:

if (error < GIT_SUCCESS)

git_filebuf_cleanup(&file);

return error;

}