[PATCH] Fix warning in convert-cache
[git/git.git] / http-pull.c
CommitLineData
6eb7ed54
DB
1#include <fcntl.h>
2#include <unistd.h>
3#include <string.h>
4#include <stdlib.h>
5#include "cache.h"
6#include "commit.h"
7#include <errno.h>
8#include <stdio.h>
9
4250a5e5
DB
10#include "pull.h"
11
6eb7ed54
DB
12#include <curl/curl.h>
13#include <curl/easy.h>
14
15static CURL *curl;
16
17static char *base;
18
6eb7ed54
DB
19static SHA_CTX c;
20static z_stream stream;
21
22static int local;
23static int zret;
24
25static size_t fwrite_sha1_file(void *ptr, size_t eltsize, size_t nmemb,
26 void *data) {
27 char expn[4096];
28 size_t size = eltsize * nmemb;
29 int posn = 0;
30 do {
31 ssize_t retval = write(local, ptr + posn, size - posn);
32 if (retval < 0)
33 return posn;
34 posn += retval;
35 } while (posn < size);
36
37 stream.avail_in = size;
38 stream.next_in = ptr;
39 do {
40 stream.next_out = expn;
41 stream.avail_out = sizeof(expn);
42 zret = inflate(&stream, Z_SYNC_FLUSH);
43 SHA1_Update(&c, expn, sizeof(expn) - stream.avail_out);
44 } while (stream.avail_in && zret == Z_OK);
45 return size;
46}
47
4250a5e5 48int fetch(unsigned char *sha1)
6eb7ed54
DB
49{
50 char *hex = sha1_to_hex(sha1);
51 char *filename = sha1_file_name(sha1);
52 char real_sha1[20];
53 char *url;
54 char *posn;
55
6eb7ed54
DB
56 local = open(filename, O_WRONLY | O_CREAT | O_EXCL, 0666);
57
58 if (local < 0)
59 return error("Couldn't open %s\n", filename);
60
61 memset(&stream, 0, sizeof(stream));
62
63 inflateInit(&stream);
64
65 SHA1_Init(&c);
66
67 curl_easy_setopt(curl, CURLOPT_FILE, NULL);
68 curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, fwrite_sha1_file);
69
812666c8 70 url = xmalloc(strlen(base) + 50);
6eb7ed54
DB
71 strcpy(url, base);
72 posn = url + strlen(base);
73 strcpy(posn, "objects/");
74 posn += 8;
75 memcpy(posn, hex, 2);
76 posn += 2;
77 *(posn++) = '/';
78 strcpy(posn, hex + 2);
79
80 curl_easy_setopt(curl, CURLOPT_URL, url);
81
82 /*printf("Getting %s\n", hex);*/
83
84 if (curl_easy_perform(curl))
85 return error("Couldn't get %s for %s\n", url, hex);
86
87 close(local);
88 inflateEnd(&stream);
89 SHA1_Final(real_sha1, &c);
90 if (zret != Z_STREAM_END) {
91 unlink(filename);
92 return error("File %s (%s) corrupt\n", hex, url);
93 }
94 if (memcmp(sha1, real_sha1, 20)) {
95 unlink(filename);
96 return error("File %s has bad hash\n", hex);
97 }
98
99 return 0;
100}
101
6eb7ed54
DB
102int main(int argc, char **argv)
103{
104 char *commit_id;
105 char *url;
106 int arg = 1;
6eb7ed54
DB
107
108 while (arg < argc && argv[arg][0] == '-') {
109 if (argv[arg][1] == 't') {
4250a5e5 110 get_tree = 1;
6eb7ed54 111 } else if (argv[arg][1] == 'c') {
4250a5e5 112 get_history = 1;
6eb7ed54 113 } else if (argv[arg][1] == 'a') {
4250a5e5
DB
114 get_all = 1;
115 get_tree = 1;
116 get_history = 1;
6eb7ed54
DB
117 }
118 arg++;
119 }
120 if (argc < arg + 2) {
121 usage("http-pull [-c] [-t] [-a] commit-id url");
122 return 1;
123 }
124 commit_id = argv[arg];
125 url = argv[arg + 1];
126
6eb7ed54
DB
127 curl_global_init(CURL_GLOBAL_ALL);
128
129 curl = curl_easy_init();
130
131 base = url;
132
4250a5e5 133 if (pull(commit_id))
6eb7ed54
DB
134 return 1;
135
136 curl_global_cleanup();
137 return 0;
138}