git/local-fetch.c

   1 /*
   2  * Copyright (C) 2005 Junio C Hamano
   3  */
   4 #include "cache.h"
   5 #include "commit.h"
   6 #include "fetch.h"
   7
   8 static int use_link;
   9 static int use_symlink;
  10 static int use_filecopy = 1;
  11 static int commits_on_stdin;
  12
  13 static const char *path; /* "Remote" git repository */
  14
  15 void prefetch(unsigned char *sha1)
  16 {
  17 }
  18
  19 static struct packed_git *packs;
  20
  21 static void setup_index(unsigned char *sha1)
  22 {
  23         struct packed_git *new_pack;
  24         char filename[PATH_MAX];
  25         strcpy(filename, path);
  26         strcat(filename, "/objects/pack/pack-");
  27         strcat(filename, sha1_to_hex(sha1));
  28         strcat(filename, ".idx");
  29         new_pack = parse_pack_index_file(sha1, filename);
  30         new_pack->next = packs;
  31         packs = new_pack;
  32 }
  33
  34 static int setup_indices(void)
  35 {
  36         DIR *dir;
  37         struct dirent *de;
  38         char filename[PATH_MAX];
  39         unsigned char sha1[20];
  40         sprintf(filename, "%s/objects/pack/", path);
  41         dir = opendir(filename);
  42         if (!dir)
  43                 return -1;
  44         while ((de = readdir(dir)) != NULL) {
  45                 int namelen = strlen(de->d_name);
  46                 if (namelen != 50 ||
  47                     !has_extension(de->d_name, ".pack"))
  48                         continue;
  49                 get_sha1_hex(de->d_name + 5, sha1);
  50                 setup_index(sha1);
  51         }
  52         closedir(dir);
  53         return 0;
  54 }
  55
  56 static int copy_file(const char *source, char *dest, const char *hex,
  57                      int warn_if_not_exists)
  58 {
  59         safe_create_leading_directories(dest);
  60         if (use_link) {
  61                 if (!link(source, dest)) {
  62                         pull_say("link %s\n", hex);
  63                         return 0;
  64                 }
  65                 /* If we got ENOENT there is no point continuing. */
  66                 if (errno == ENOENT) {
  67                         if (!warn_if_not_exists)
  68                                 return -1;
  69                         return error("does not exist %s", source);
  70                 }
  71         }
  72         if (use_symlink) {
  73                 struct stat st;
  74                 if (stat(source, &st)) {
  75                         if (!warn_if_not_exists && errno == ENOENT)
  76                                 return -1;
  77                         return error("cannot stat %s: %s", source,
  78                                      strerror(errno));
  79                 }
  80                 if (!symlink(source, dest)) {
  81                         pull_say("symlink %s\n", hex);
  82                         return 0;
  83                 }
  84         }
  85         if (use_filecopy) {
  86                 int ifd, ofd, status = 0;
  87
  88                 ifd = open(source, O_RDONLY);
  89                 if (ifd < 0) {
  90                         if (!warn_if_not_exists && errno == ENOENT)
  91                                 return -1;
  92                         return error("cannot open %s", source);
  93                 }
  94                 ofd = open(dest, O_WRONLY | O_CREAT | O_EXCL, 0666);
  95                 if (ofd < 0) {
  96                         close(ifd);
  97                         return error("cannot open %s", dest);
  98                 }
  99                 status = copy_fd(ifd, ofd);
 100                 close(ofd);
 101                 if (status)
 102                         return error("cannot write %s", dest);
 103                 pull_say("copy %s\n", hex);
 104                 return 0;
 105         }
 106         return error("failed to copy %s with given copy methods.", hex);
 107 }
 108
 109 static int fetch_pack(const unsigned char *sha1)
 110 {
 111         struct packed_git *target;
 112         char filename[PATH_MAX];
 113         if (setup_indices())
 114                 return -1;
 115         target = find_sha1_pack(sha1, packs);
 116         if (!target)
 117                 return error("Couldn't find %s: not separate or in any pack",
 118                              sha1_to_hex(sha1));
 119         if (get_verbosely) {
 120                 fprintf(stderr, "Getting pack %s\n",
 121                         sha1_to_hex(target->sha1));
 122                 fprintf(stderr, " which contains %s\n",
 123                         sha1_to_hex(sha1));
 124         }
 125         sprintf(filename, "%s/objects/pack/pack-%s.pack",
 126                 path, sha1_to_hex(target->sha1));
 127         copy_file(filename, sha1_pack_name(target->sha1),
 128                   sha1_to_hex(target->sha1), 1);
 129         sprintf(filename, "%s/objects/pack/pack-%s.idx",
 130                 path, sha1_to_hex(target->sha1));
 131         copy_file(filename, sha1_pack_index_name(target->sha1),
 132                   sha1_to_hex(target->sha1), 1);
 133         install_packed_git(target);
 134         return 0;
 135 }
 136
 137 static int fetch_file(const unsigned char *sha1)
 138 {
 139         static int object_name_start = -1;
 140         static char filename[PATH_MAX];
 141         char *hex = sha1_to_hex(sha1);
 142         char *dest_filename = sha1_file_name(sha1);
 143
 144         if (object_name_start < 0) {
 145                 strcpy(filename, path); /* e.g. git.git */
 146                 strcat(filename, "/objects/");
 147                 object_name_start = strlen(filename);
 148         }
 149         filename[object_name_start+0] = hex[0];
 150         filename[object_name_start+1] = hex[1];
 151         filename[object_name_start+2] = '/';
 152         strcpy(filename + object_name_start + 3, hex + 2);
 153         return copy_file(filename, dest_filename, hex, 0);
 154 }
 155
 156 int fetch(unsigned char *sha1)
 157 {
 158         if (has_sha1_file(sha1))
 159                 return 0;
 160         else
 161                 return fetch_file(sha1) && fetch_pack(sha1);
 162 }
 163
 164 int fetch_ref(char *ref, unsigned char *sha1)
 165 {
 166         static int ref_name_start = -1;
 167         static char filename[PATH_MAX];
 168         static char hex[41];
 169         int ifd;
 170
 171         if (ref_name_start < 0) {
 172                 sprintf(filename, "%s/refs/", path);
 173                 ref_name_start = strlen(filename);
 174         }
 175         strcpy(filename + ref_name_start, ref);
 176         ifd = open(filename, O_RDONLY);
 177         if (ifd < 0) {
 178                 close(ifd);
 179                 return error("cannot open %s", filename);
 180         }
 181         if (read_in_full(ifd, hex, 40) != 40 || get_sha1_hex(hex, sha1)) {
 182                 close(ifd);
 183                 return error("cannot read from %s", filename);
 184         }
 185         close(ifd);
 186         pull_say("ref %s\n", sha1_to_hex(sha1));
 187         return 0;
 188 }
 189
 190 static const char local_pull_usage[] =
 191 "git-local-fetch [-c] [-t] [-a] [-v] [-w filename] [--recover] [-l] [-s] [-n] [--stdin] commit-id path";
 192
 193 /*
 194  * By default we only use file copy.
 195  * If -l is specified, a hard link is attempted.
 196  * If -s is specified, then a symlink is attempted.
 197  * If -n is _not_ specified, then a regular file-to-file copy is done.
 198  */
 199 int main(int argc, const char **argv)
 200 {
 201         int commits;
 202         const char **write_ref = NULL;
 203         char **commit_id;
 204         int arg = 1;
 205
 206         setup_git_directory();
 207         git_config(git_default_config);
 208
 209         while (arg < argc && argv[arg][0] == '-') {
 210                 if (argv[arg][1] == 't')
 211                         get_tree = 1;
 212                 else if (argv[arg][1] == 'c')
 213                         get_history = 1;
 214                 else if (argv[arg][1] == 'a') {
 215                         get_all = 1;
 216                         get_tree = 1;
 217                         get_history = 1;
 218                 }
 219                 else if (argv[arg][1] == 'l')
 220                         use_link = 1;
 221                 else if (argv[arg][1] == 's')
 222                         use_symlink = 1;
 223                 else if (argv[arg][1] == 'n')
 224                         use_filecopy = 0;
 225                 else if (argv[arg][1] == 'v')
 226                         get_verbosely = 1;
 227                 else if (argv[arg][1] == 'w')
 228                         write_ref = &argv[++arg];
 229                 else if (!strcmp(argv[arg], "--recover"))
 230                         get_recover = 1;
 231                 else if (!strcmp(argv[arg], "--stdin"))
 232                         commits_on_stdin = 1;
 233                 else
 234                         usage(local_pull_usage);
 235                 arg++;
 236         }
 237         if (argc < arg + 2 - commits_on_stdin)
 238                 usage(local_pull_usage);
 239         if (commits_on_stdin) {
 240                 commits = pull_targets_stdin(&commit_id, &write_ref);
 241         } else {
 242                 commit_id = (char **) &argv[arg++];
 243                 commits = 1;
 244         }
 245         path = argv[arg];
 246
 247         if (pull(commits, commit_id, write_ref, path))
 248                 return 1;
 249
 250         if (commits_on_stdin)
 251                 pull_targets_free(commits, commit_id, write_ref);
 252
 253         return 0;
 254 }