Andrew's git - gitweb.git/blob

   1/*
   2 * Copyright (C) 2005 Junio C Hamano
   3 */
   4#include "cache.h"
   5#include "commit.h"
   6#include "fetch.h"
   7
   8static int use_link;
   9static int use_symlink;
  10static int use_filecopy = 1;
  11static int commits_on_stdin;
  12
  13static const char *path; /* "Remote" git repository */
  14
  15void prefetch(unsigned char *sha1)
  16{
  17}
  18
  19static struct packed_git *packs;
  20
  21static 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
  34static 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
  56static 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
 109static 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
 137static 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
 156int 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
 164int 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
 190static 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 */
 199int 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}