sha1_file.con commit Revert "Convert git-annotate to use Git.pm" (3c2f588)
   1/*
   2 * GIT - The information manager from hell
   3 *
   4 * Copyright (C) Linus Torvalds, 2005
   5 *
   6 * This handles basic git sha1 object files - packing, unpacking,
   7 * creation etc.
   8 */
   9#include "cache.h"
  10#include "delta.h"
  11#include "pack.h"
  12#include "blob.h"
  13#include "commit.h"
  14#include "tag.h"
  15#include "tree.h"
  16
  17#ifndef O_NOATIME
  18#if defined(__linux__) && (defined(__i386__) || defined(__PPC__))
  19#define O_NOATIME 01000000
  20#else
  21#define O_NOATIME 0
  22#endif
  23#endif
  24
  25const unsigned char null_sha1[20] = { 0, };
  26
  27static unsigned int sha1_file_open_flag = O_NOATIME;
  28
  29static unsigned hexval(char c)
  30{
  31        if (c >= '0' && c <= '9')
  32                return c - '0';
  33        if (c >= 'a' && c <= 'f')
  34                return c - 'a' + 10;
  35        if (c >= 'A' && c <= 'F')
  36                return c - 'A' + 10;
  37        return ~0;
  38}
  39
  40int get_sha1_hex(const char *hex, unsigned char *sha1)
  41{
  42        int i;
  43        for (i = 0; i < 20; i++) {
  44                unsigned int val = (hexval(hex[0]) << 4) | hexval(hex[1]);
  45                if (val & ~0xff)
  46                        return -1;
  47                *sha1++ = val;
  48                hex += 2;
  49        }
  50        return 0;
  51}
  52
  53int safe_create_leading_directories(char *path)
  54{
  55        char *pos = path;
  56        struct stat st;
  57
  58        if (*pos == '/')
  59                pos++;
  60
  61        while (pos) {
  62                pos = strchr(pos, '/');
  63                if (!pos)
  64                        break;
  65                *pos = 0;
  66                if (!stat(path, &st)) {
  67                        /* path exists */
  68                        if (!S_ISDIR(st.st_mode)) {
  69                                *pos = '/';
  70                                return -3;
  71                        }
  72                }
  73                else if (mkdir(path, 0777)) {
  74                        *pos = '/';
  75                        return -1;
  76                }
  77                else if (adjust_shared_perm(path)) {
  78                        *pos = '/';
  79                        return -2;
  80                }
  81                *pos++ = '/';
  82        }
  83        return 0;
  84}
  85
  86char * sha1_to_hex(const unsigned char *sha1)
  87{
  88        static int bufno;
  89        static char hexbuffer[4][50];
  90        static const char hex[] = "0123456789abcdef";
  91        char *buffer = hexbuffer[3 & ++bufno], *buf = buffer;
  92        int i;
  93
  94        for (i = 0; i < 20; i++) {
  95                unsigned int val = *sha1++;
  96                *buf++ = hex[val >> 4];
  97                *buf++ = hex[val & 0xf];
  98        }
  99        *buf = '\0';
 100
 101        return buffer;
 102}
 103
 104static void fill_sha1_path(char *pathbuf, const unsigned char *sha1)
 105{
 106        int i;
 107        for (i = 0; i < 20; i++) {
 108                static char hex[] = "0123456789abcdef";
 109                unsigned int val = sha1[i];
 110                char *pos = pathbuf + i*2 + (i > 0);
 111                *pos++ = hex[val >> 4];
 112                *pos = hex[val & 0xf];
 113        }
 114}
 115
 116/*
 117 * NOTE! This returns a statically allocated buffer, so you have to be
 118 * careful about using it. Do a "strdup()" if you need to save the
 119 * filename.
 120 *
 121 * Also note that this returns the location for creating.  Reading
 122 * SHA1 file can happen from any alternate directory listed in the
 123 * DB_ENVIRONMENT environment variable if it is not found in
 124 * the primary object database.
 125 */
 126char *sha1_file_name(const unsigned char *sha1)
 127{
 128        static char *name, *base;
 129        static const char *last_objdir;
 130        const char *sha1_file_directory = get_object_directory();
 131
 132        if (!last_objdir || strcmp(last_objdir, sha1_file_directory)) {
 133                int len = strlen(sha1_file_directory);
 134                if (base)
 135                        free(base);
 136                base = xmalloc(len + 60);
 137                memcpy(base, sha1_file_directory, len);
 138                memset(base+len, 0, 60);
 139                base[len] = '/';
 140                base[len+3] = '/';
 141                name = base + len + 1;
 142                if (last_objdir)
 143                        free((char *) last_objdir);
 144                last_objdir = strdup(sha1_file_directory);
 145        }
 146        fill_sha1_path(name, sha1);
 147        return base;
 148}
 149
 150char *sha1_pack_name(const unsigned char *sha1)
 151{
 152        static const char hex[] = "0123456789abcdef";
 153        static char *name, *base, *buf;
 154        static const char *last_objdir;
 155        const char *sha1_file_directory = get_object_directory();
 156        int i;
 157
 158        if (!last_objdir || strcmp(last_objdir, sha1_file_directory)) {
 159                int len = strlen(sha1_file_directory);
 160                if (base)
 161                        free(base);
 162                base = xmalloc(len + 60);
 163                sprintf(base, "%s/pack/pack-1234567890123456789012345678901234567890.pack", sha1_file_directory);
 164                name = base + len + 11;
 165                if (last_objdir)
 166                        free((char *) last_objdir);
 167                last_objdir = strdup(sha1_file_directory);
 168        }
 169
 170        buf = name;
 171
 172        for (i = 0; i < 20; i++) {
 173                unsigned int val = *sha1++;
 174                *buf++ = hex[val >> 4];
 175                *buf++ = hex[val & 0xf];
 176        }
 177        
 178        return base;
 179}
 180
 181char *sha1_pack_index_name(const unsigned char *sha1)
 182{
 183        static const char hex[] = "0123456789abcdef";
 184        static char *name, *base, *buf;
 185        static const char *last_objdir;
 186        const char *sha1_file_directory = get_object_directory();
 187        int i;
 188
 189        if (!last_objdir || strcmp(last_objdir, sha1_file_directory)) {
 190                int len = strlen(sha1_file_directory);
 191                if (base)
 192                        free(base);
 193                base = xmalloc(len + 60);
 194                sprintf(base, "%s/pack/pack-1234567890123456789012345678901234567890.idx", sha1_file_directory);
 195                name = base + len + 11;
 196                if (last_objdir)
 197                        free((char *) last_objdir);
 198                last_objdir = strdup(sha1_file_directory);
 199        }
 200
 201        buf = name;
 202
 203        for (i = 0; i < 20; i++) {
 204                unsigned int val = *sha1++;
 205                *buf++ = hex[val >> 4];
 206                *buf++ = hex[val & 0xf];
 207        }
 208        
 209        return base;
 210}
 211
 212struct alternate_object_database *alt_odb_list;
 213static struct alternate_object_database **alt_odb_tail;
 214
 215static void read_info_alternates(const char * alternates, int depth);
 216
 217/*
 218 * Prepare alternate object database registry.
 219 *
 220 * The variable alt_odb_list points at the list of struct
 221 * alternate_object_database.  The elements on this list come from
 222 * non-empty elements from colon separated ALTERNATE_DB_ENVIRONMENT
 223 * environment variable, and $GIT_OBJECT_DIRECTORY/info/alternates,
 224 * whose contents is similar to that environment variable but can be
 225 * LF separated.  Its base points at a statically allocated buffer that
 226 * contains "/the/directory/corresponding/to/.git/objects/...", while
 227 * its name points just after the slash at the end of ".git/objects/"
 228 * in the example above, and has enough space to hold 40-byte hex
 229 * SHA1, an extra slash for the first level indirection, and the
 230 * terminating NUL.
 231 */
 232static int link_alt_odb_entry(const char * entry, int len, const char * relative_base, int depth)
 233{
 234        struct stat st;
 235        const char *objdir = get_object_directory();
 236        struct alternate_object_database *ent;
 237        struct alternate_object_database *alt;
 238        /* 43 = 40-byte + 2 '/' + terminating NUL */
 239        int pfxlen = len;
 240        int entlen = pfxlen + 43;
 241        int base_len = -1;
 242
 243        if (*entry != '/' && relative_base) {
 244                /* Relative alt-odb */
 245                if (base_len < 0)
 246                        base_len = strlen(relative_base) + 1;
 247                entlen += base_len;
 248                pfxlen += base_len;
 249        }
 250        ent = xmalloc(sizeof(*ent) + entlen);
 251
 252        if (*entry != '/' && relative_base) {
 253                memcpy(ent->base, relative_base, base_len - 1);
 254                ent->base[base_len - 1] = '/';
 255                memcpy(ent->base + base_len, entry, len);
 256        }
 257        else
 258                memcpy(ent->base, entry, pfxlen);
 259
 260        ent->name = ent->base + pfxlen + 1;
 261        ent->base[pfxlen + 3] = '/';
 262        ent->base[pfxlen] = ent->base[entlen-1] = 0;
 263
 264        /* Detect cases where alternate disappeared */
 265        if (stat(ent->base, &st) || !S_ISDIR(st.st_mode)) {
 266                error("object directory %s does not exist; "
 267                      "check .git/objects/info/alternates.",
 268                      ent->base);
 269                free(ent);
 270                return -1;
 271        }
 272
 273        /* Prevent the common mistake of listing the same
 274         * thing twice, or object directory itself.
 275         */
 276        for (alt = alt_odb_list; alt; alt = alt->next) {
 277                if (!memcmp(ent->base, alt->base, pfxlen)) {
 278                        free(ent);
 279                        return -1;
 280                }
 281        }
 282        if (!memcmp(ent->base, objdir, pfxlen)) {
 283                free(ent);
 284                return -1;
 285        }
 286
 287        /* add the alternate entry */
 288        *alt_odb_tail = ent;
 289        alt_odb_tail = &(ent->next);
 290        ent->next = NULL;
 291
 292        /* recursively add alternates */
 293        read_info_alternates(ent->base, depth + 1);
 294
 295        ent->base[pfxlen] = '/';
 296
 297        return 0;
 298}
 299
 300static void link_alt_odb_entries(const char *alt, const char *ep, int sep,
 301                                 const char *relative_base, int depth)
 302{
 303        const char *cp, *last;
 304
 305        if (depth > 5) {
 306                error("%s: ignoring alternate object stores, nesting too deep.",
 307                                relative_base);
 308                return;
 309        }
 310
 311        last = alt;
 312        while (last < ep) {
 313                cp = last;
 314                if (cp < ep && *cp == '#') {
 315                        while (cp < ep && *cp != sep)
 316                                cp++;
 317                        last = cp + 1;
 318                        continue;
 319                }
 320                while (cp < ep && *cp != sep)
 321                        cp++;
 322                if (last != cp) {
 323                        if ((*last != '/') && depth) {
 324                                error("%s: ignoring relative alternate object store %s",
 325                                                relative_base, last);
 326                        } else {
 327                                link_alt_odb_entry(last, cp - last,
 328                                                relative_base, depth);
 329                        }
 330                }
 331                while (cp < ep && *cp == sep)
 332                        cp++;
 333                last = cp;
 334        }
 335}
 336
 337static void read_info_alternates(const char * relative_base, int depth)
 338{
 339        char *map;
 340        struct stat st;
 341        char path[PATH_MAX];
 342        int fd;
 343
 344        sprintf(path, "%s/info/alternates", relative_base);
 345        fd = open(path, O_RDONLY);
 346        if (fd < 0)
 347                return;
 348        if (fstat(fd, &st) || (st.st_size == 0)) {
 349                close(fd);
 350                return;
 351        }
 352        map = mmap(NULL, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
 353        close(fd);
 354        if (map == MAP_FAILED)
 355                return;
 356
 357        link_alt_odb_entries(map, map + st.st_size, '\n', relative_base, depth);
 358
 359        munmap(map, st.st_size);
 360}
 361
 362void prepare_alt_odb(void)
 363{
 364        const char *alt;
 365
 366        alt = getenv(ALTERNATE_DB_ENVIRONMENT);
 367        if (!alt) alt = "";
 368
 369        if (alt_odb_tail)
 370                return;
 371        alt_odb_tail = &alt_odb_list;
 372        link_alt_odb_entries(alt, alt + strlen(alt), ':', NULL, 0);
 373
 374        read_info_alternates(get_object_directory(), 0);
 375}
 376
 377static char *find_sha1_file(const unsigned char *sha1, struct stat *st)
 378{
 379        char *name = sha1_file_name(sha1);
 380        struct alternate_object_database *alt;
 381
 382        if (!stat(name, st))
 383                return name;
 384        prepare_alt_odb();
 385        for (alt = alt_odb_list; alt; alt = alt->next) {
 386                name = alt->name;
 387                fill_sha1_path(name, sha1);
 388                if (!stat(alt->base, st))
 389                        return alt->base;
 390        }
 391        return NULL;
 392}
 393
 394#define PACK_MAX_SZ (1<<26)
 395static int pack_used_ctr;
 396static unsigned long pack_mapped;
 397struct packed_git *packed_git;
 398
 399static int check_packed_git_idx(const char *path, unsigned long *idx_size_,
 400                                void **idx_map_)
 401{
 402        void *idx_map;
 403        unsigned int *index;
 404        unsigned long idx_size;
 405        int nr, i;
 406        int fd = open(path, O_RDONLY);
 407        struct stat st;
 408        if (fd < 0)
 409                return -1;
 410        if (fstat(fd, &st)) {
 411                close(fd);
 412                return -1;
 413        }
 414        idx_size = st.st_size;
 415        idx_map = mmap(NULL, idx_size, PROT_READ, MAP_PRIVATE, fd, 0);
 416        close(fd);
 417        if (idx_map == MAP_FAILED)
 418                return -1;
 419
 420        index = idx_map;
 421        *idx_map_ = idx_map;
 422        *idx_size_ = idx_size;
 423
 424        /* check index map */
 425        if (idx_size < 4*256 + 20 + 20)
 426                return error("index file too small");
 427        nr = 0;
 428        for (i = 0; i < 256; i++) {
 429                unsigned int n = ntohl(index[i]);
 430                if (n < nr)
 431                        return error("non-monotonic index");
 432                nr = n;
 433        }
 434
 435        /*
 436         * Total size:
 437         *  - 256 index entries 4 bytes each
 438         *  - 24-byte entries * nr (20-byte sha1 + 4-byte offset)
 439         *  - 20-byte SHA1 of the packfile
 440         *  - 20-byte SHA1 file checksum
 441         */
 442        if (idx_size != 4*256 + nr * 24 + 20 + 20)
 443                return error("wrong index file size");
 444
 445        return 0;
 446}
 447
 448static int unuse_one_packed_git(void)
 449{
 450        struct packed_git *p, *lru = NULL;
 451
 452        for (p = packed_git; p; p = p->next) {
 453                if (p->pack_use_cnt || !p->pack_base)
 454                        continue;
 455                if (!lru || p->pack_last_used < lru->pack_last_used)
 456                        lru = p;
 457        }
 458        if (!lru)
 459                return 0;
 460        munmap(lru->pack_base, lru->pack_size);
 461        lru->pack_base = NULL;
 462        return 1;
 463}
 464
 465void unuse_packed_git(struct packed_git *p)
 466{
 467        p->pack_use_cnt--;
 468}
 469
 470int use_packed_git(struct packed_git *p)
 471{
 472        if (!p->pack_size) {
 473                struct stat st;
 474                /* We created the struct before we had the pack */
 475                stat(p->pack_name, &st);
 476                if (!S_ISREG(st.st_mode))
 477                        die("packfile %s not a regular file", p->pack_name);
 478                p->pack_size = st.st_size;
 479        }
 480        if (!p->pack_base) {
 481                int fd;
 482                struct stat st;
 483                void *map;
 484
 485                pack_mapped += p->pack_size;
 486                while (PACK_MAX_SZ < pack_mapped && unuse_one_packed_git())
 487                        ; /* nothing */
 488                fd = open(p->pack_name, O_RDONLY);
 489                if (fd < 0)
 490                        die("packfile %s cannot be opened", p->pack_name);
 491                if (fstat(fd, &st)) {
 492                        close(fd);
 493                        die("packfile %s cannot be opened", p->pack_name);
 494                }
 495                if (st.st_size != p->pack_size)
 496                        die("packfile %s size mismatch.", p->pack_name);
 497                map = mmap(NULL, p->pack_size, PROT_READ, MAP_PRIVATE, fd, 0);
 498                close(fd);
 499                if (map == MAP_FAILED)
 500                        die("packfile %s cannot be mapped.", p->pack_name);
 501                p->pack_base = map;
 502
 503                /* Check if the pack file matches with the index file.
 504                 * this is cheap.
 505                 */
 506                if (memcmp((char*)(p->index_base) + p->index_size - 40,
 507                           (char *) p->pack_base + p->pack_size - 20,
 508                           20)) {
 509
 510                        die("packfile %s does not match index.", p->pack_name);
 511                }
 512        }
 513        p->pack_last_used = pack_used_ctr++;
 514        p->pack_use_cnt++;
 515        return 0;
 516}
 517
 518struct packed_git *add_packed_git(char *path, int path_len, int local)
 519{
 520        struct stat st;
 521        struct packed_git *p;
 522        unsigned long idx_size;
 523        void *idx_map;
 524        unsigned char sha1[20];
 525
 526        if (check_packed_git_idx(path, &idx_size, &idx_map))
 527                return NULL;
 528
 529        /* do we have a corresponding .pack file? */
 530        strcpy(path + path_len - 4, ".pack");
 531        if (stat(path, &st) || !S_ISREG(st.st_mode)) {
 532                munmap(idx_map, idx_size);
 533                return NULL;
 534        }
 535        /* ok, it looks sane as far as we can check without
 536         * actually mapping the pack file.
 537         */
 538        p = xmalloc(sizeof(*p) + path_len + 2);
 539        strcpy(p->pack_name, path);
 540        p->index_size = idx_size;
 541        p->pack_size = st.st_size;
 542        p->index_base = idx_map;
 543        p->next = NULL;
 544        p->pack_base = NULL;
 545        p->pack_last_used = 0;
 546        p->pack_use_cnt = 0;
 547        p->pack_local = local;
 548        if ((path_len > 44) && !get_sha1_hex(path + path_len - 44, sha1))
 549                memcpy(p->sha1, sha1, 20);
 550        return p;
 551}
 552
 553struct packed_git *parse_pack_index(unsigned char *sha1)
 554{
 555        char *path = sha1_pack_index_name(sha1);
 556        return parse_pack_index_file(sha1, path);
 557}
 558
 559struct packed_git *parse_pack_index_file(const unsigned char *sha1, char *idx_path)
 560{
 561        struct packed_git *p;
 562        unsigned long idx_size;
 563        void *idx_map;
 564        char *path;
 565
 566        if (check_packed_git_idx(idx_path, &idx_size, &idx_map))
 567                return NULL;
 568
 569        path = sha1_pack_name(sha1);
 570
 571        p = xmalloc(sizeof(*p) + strlen(path) + 2);
 572        strcpy(p->pack_name, path);
 573        p->index_size = idx_size;
 574        p->pack_size = 0;
 575        p->index_base = idx_map;
 576        p->next = NULL;
 577        p->pack_base = NULL;
 578        p->pack_last_used = 0;
 579        p->pack_use_cnt = 0;
 580        memcpy(p->sha1, sha1, 20);
 581        return p;
 582}
 583
 584void install_packed_git(struct packed_git *pack)
 585{
 586        pack->next = packed_git;
 587        packed_git = pack;
 588}
 589
 590static void prepare_packed_git_one(char *objdir, int local)
 591{
 592        char path[PATH_MAX];
 593        int len;
 594        DIR *dir;
 595        struct dirent *de;
 596
 597        sprintf(path, "%s/pack", objdir);
 598        len = strlen(path);
 599        dir = opendir(path);
 600        if (!dir) {
 601                if (errno != ENOENT)
 602                        error("unable to open object pack directory: %s: %s",
 603                              path, strerror(errno));
 604                return;
 605        }
 606        path[len++] = '/';
 607        while ((de = readdir(dir)) != NULL) {
 608                int namelen = strlen(de->d_name);
 609                struct packed_git *p;
 610
 611                if (!has_extension(de->d_name, ".idx"))
 612                        continue;
 613
 614                /* we have .idx.  Is it a file we can map? */
 615                strcpy(path + len, de->d_name);
 616                for (p = packed_git; p; p = p->next) {
 617                        if (!memcmp(path, p->pack_name, len + namelen - 4))
 618                                break;
 619                }
 620                if (p)
 621                        continue;
 622                p = add_packed_git(path, len + namelen, local);
 623                if (!p)
 624                        continue;
 625                p->next = packed_git;
 626                packed_git = p;
 627        }
 628        closedir(dir);
 629}
 630
 631static int prepare_packed_git_run_once = 0;
 632void prepare_packed_git(void)
 633{
 634        struct alternate_object_database *alt;
 635
 636        if (prepare_packed_git_run_once)
 637                return;
 638        prepare_packed_git_one(get_object_directory(), 1);
 639        prepare_alt_odb();
 640        for (alt = alt_odb_list; alt; alt = alt->next) {
 641                alt->name[-1] = 0;
 642                prepare_packed_git_one(alt->base, 0);
 643                alt->name[-1] = '/';
 644        }
 645        prepare_packed_git_run_once = 1;
 646}
 647
 648static void reprepare_packed_git(void)
 649{
 650        prepare_packed_git_run_once = 0;
 651        prepare_packed_git();
 652}
 653
 654int check_sha1_signature(const unsigned char *sha1, void *map, unsigned long size, const char *type)
 655{
 656        char header[100];
 657        unsigned char real_sha1[20];
 658        SHA_CTX c;
 659
 660        SHA1_Init(&c);
 661        SHA1_Update(&c, header, 1+sprintf(header, "%s %lu", type, size));
 662        SHA1_Update(&c, map, size);
 663        SHA1_Final(real_sha1, &c);
 664        return memcmp(sha1, real_sha1, 20) ? -1 : 0;
 665}
 666
 667void *map_sha1_file(const unsigned char *sha1, unsigned long *size)
 668{
 669        struct stat st;
 670        void *map;
 671        int fd;
 672        char *filename = find_sha1_file(sha1, &st);
 673
 674        if (!filename) {
 675                return NULL;
 676        }
 677
 678        fd = open(filename, O_RDONLY | sha1_file_open_flag);
 679        if (fd < 0) {
 680                /* See if it works without O_NOATIME */
 681                switch (sha1_file_open_flag) {
 682                default:
 683                        fd = open(filename, O_RDONLY);
 684                        if (fd >= 0)
 685                                break;
 686                /* Fallthrough */
 687                case 0:
 688                        return NULL;
 689                }
 690
 691                /* If it failed once, it will probably fail again.
 692                 * Stop using O_NOATIME
 693                 */
 694                sha1_file_open_flag = 0;
 695        }
 696        map = mmap(NULL, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
 697        close(fd);
 698        if (map == MAP_FAILED)
 699                return NULL;
 700        *size = st.st_size;
 701        return map;
 702}
 703
 704int legacy_loose_object(unsigned char *map)
 705{
 706        unsigned int word;
 707
 708        /*
 709         * Is it a zlib-compressed buffer? If so, the first byte
 710         * must be 0x78 (15-bit window size, deflated), and the
 711         * first 16-bit word is evenly divisible by 31
 712         */
 713        word = (map[0] << 8) + map[1];
 714        if (map[0] == 0x78 && !(word % 31))
 715                return 1;
 716        else
 717                return 0;
 718}
 719
 720static int unpack_sha1_header(z_stream *stream, unsigned char *map, unsigned long mapsize, void *buffer, unsigned long bufsiz)
 721{
 722        unsigned char c;
 723        unsigned int bits;
 724        unsigned long size;
 725        static const char *typename[8] = {
 726                NULL,   /* OBJ_EXT */
 727                "commit", "tree", "blob", "tag",
 728                NULL, NULL, NULL
 729        };
 730        const char *type;
 731
 732        /* Get the data stream */
 733        memset(stream, 0, sizeof(*stream));
 734        stream->next_in = map;
 735        stream->avail_in = mapsize;
 736        stream->next_out = buffer;
 737        stream->avail_out = bufsiz;
 738
 739        if (legacy_loose_object(map)) {
 740                inflateInit(stream);
 741                return inflate(stream, 0);
 742        }
 743
 744        c = *map++;
 745        mapsize--;
 746        type = typename[(c >> 4) & 7];
 747        if (!type)
 748                return -1;
 749
 750        bits = 4;
 751        size = c & 0xf;
 752        while ((c & 0x80)) {
 753                if (bits >= 8*sizeof(long))
 754                        return -1;
 755                c = *map++;
 756                size += (c & 0x7f) << bits;
 757                bits += 7;
 758                mapsize--;
 759        }
 760
 761        /* Set up the stream for the rest.. */
 762        stream->next_in = map;
 763        stream->avail_in = mapsize;
 764        inflateInit(stream);
 765
 766        /* And generate the fake traditional header */
 767        stream->total_out = 1 + snprintf(buffer, bufsiz, "%s %lu", type, size);
 768        return 0;
 769}
 770
 771static void *unpack_sha1_rest(z_stream *stream, void *buffer, unsigned long size)
 772{
 773        int bytes = strlen(buffer) + 1;
 774        unsigned char *buf = xmalloc(1+size);
 775        unsigned long n;
 776
 777        n = stream->total_out - bytes;
 778        if (n > size)
 779                n = size;
 780        memcpy(buf, (char *) buffer + bytes, n);
 781        bytes = n;
 782        if (bytes < size) {
 783                stream->next_out = buf + bytes;
 784                stream->avail_out = size - bytes;
 785                while (inflate(stream, Z_FINISH) == Z_OK)
 786                        /* nothing */;
 787        }
 788        buf[size] = 0;
 789        inflateEnd(stream);
 790        return buf;
 791}
 792
 793/*
 794 * We used to just use "sscanf()", but that's actually way
 795 * too permissive for what we want to check. So do an anal
 796 * object header parse by hand.
 797 */
 798static int parse_sha1_header(char *hdr, char *type, unsigned long *sizep)
 799{
 800        int i;
 801        unsigned long size;
 802
 803        /*
 804         * The type can be at most ten bytes (including the 
 805         * terminating '\0' that we add), and is followed by
 806         * a space. 
 807         */
 808        i = 10;
 809        for (;;) {
 810                char c = *hdr++;
 811                if (c == ' ')
 812                        break;
 813                if (!--i)
 814                        return -1;
 815                *type++ = c;
 816        }
 817        *type = 0;
 818
 819        /*
 820         * The length must follow immediately, and be in canonical
 821         * decimal format (ie "010" is not valid).
 822         */
 823        size = *hdr++ - '0';
 824        if (size > 9)
 825                return -1;
 826        if (size) {
 827                for (;;) {
 828                        unsigned long c = *hdr - '0';
 829                        if (c > 9)
 830                                break;
 831                        hdr++;
 832                        size = size * 10 + c;
 833                }
 834        }
 835        *sizep = size;
 836
 837        /*
 838         * The length must be followed by a zero byte
 839         */
 840        return *hdr ? -1 : 0;
 841}
 842
 843void * unpack_sha1_file(void *map, unsigned long mapsize, char *type, unsigned long *size)
 844{
 845        int ret;
 846        z_stream stream;
 847        char hdr[8192];
 848
 849        ret = unpack_sha1_header(&stream, map, mapsize, hdr, sizeof(hdr));
 850        if (ret < Z_OK || parse_sha1_header(hdr, type, size) < 0)
 851                return NULL;
 852
 853        return unpack_sha1_rest(&stream, hdr, *size);
 854}
 855
 856/* forward declaration for a mutually recursive function */
 857static int packed_object_info(struct pack_entry *entry,
 858                              char *type, unsigned long *sizep);
 859
 860static int packed_delta_info(unsigned char *base_sha1,
 861                             unsigned long delta_size,
 862                             unsigned long left,
 863                             char *type,
 864                             unsigned long *sizep,
 865                             struct packed_git *p)
 866{
 867        struct pack_entry base_ent;
 868
 869        if (left < 20)
 870                die("truncated pack file");
 871
 872        /* The base entry _must_ be in the same pack */
 873        if (!find_pack_entry_one(base_sha1, &base_ent, p))
 874                die("failed to find delta-pack base object %s",
 875                    sha1_to_hex(base_sha1));
 876
 877        /* We choose to only get the type of the base object and
 878         * ignore potentially corrupt pack file that expects the delta
 879         * based on a base with a wrong size.  This saves tons of
 880         * inflate() calls.
 881         */
 882
 883        if (packed_object_info(&base_ent, type, NULL))
 884                die("cannot get info for delta-pack base");
 885
 886        if (sizep) {
 887                const unsigned char *data;
 888                unsigned char delta_head[64];
 889                unsigned long result_size;
 890                z_stream stream;
 891                int st;
 892
 893                memset(&stream, 0, sizeof(stream));
 894
 895                data = stream.next_in = base_sha1 + 20;
 896                stream.avail_in = left - 20;
 897                stream.next_out = delta_head;
 898                stream.avail_out = sizeof(delta_head);
 899
 900                inflateInit(&stream);
 901                st = inflate(&stream, Z_FINISH);
 902                inflateEnd(&stream);
 903                if ((st != Z_STREAM_END) &&
 904                    stream.total_out != sizeof(delta_head))
 905                        die("delta data unpack-initial failed");
 906
 907                /* Examine the initial part of the delta to figure out
 908                 * the result size.
 909                 */
 910                data = delta_head;
 911
 912                /* ignore base size */
 913                get_delta_hdr_size(&data, delta_head+sizeof(delta_head));
 914
 915                /* Read the result size */
 916                result_size = get_delta_hdr_size(&data, delta_head+sizeof(delta_head));
 917                *sizep = result_size;
 918        }
 919        return 0;
 920}
 921
 922static unsigned long unpack_object_header(struct packed_git *p, unsigned long offset,
 923        enum object_type *type, unsigned long *sizep)
 924{
 925        unsigned shift;
 926        unsigned char *pack, c;
 927        unsigned long size;
 928
 929        if (offset >= p->pack_size)
 930                die("object offset outside of pack file");
 931
 932        pack =  (unsigned char *) p->pack_base + offset;
 933        c = *pack++;
 934        offset++;
 935        *type = (c >> 4) & 7;
 936        size = c & 15;
 937        shift = 4;
 938        while (c & 0x80) {
 939                if (offset >= p->pack_size)
 940                        die("object offset outside of pack file");
 941                c = *pack++;
 942                offset++;
 943                size += (c & 0x7f) << shift;
 944                shift += 7;
 945        }
 946        *sizep = size;
 947        return offset;
 948}
 949
 950int check_reuse_pack_delta(struct packed_git *p, unsigned long offset,
 951                           unsigned char *base, unsigned long *sizep,
 952                           enum object_type *kindp)
 953{
 954        unsigned long ptr;
 955        int status = -1;
 956
 957        use_packed_git(p);
 958        ptr = offset;
 959        ptr = unpack_object_header(p, ptr, kindp, sizep);
 960        if (*kindp != OBJ_DELTA)
 961                goto done;
 962        memcpy(base, (char *) p->pack_base + ptr, 20);
 963        status = 0;
 964 done:
 965        unuse_packed_git(p);
 966        return status;
 967}
 968
 969void packed_object_info_detail(struct pack_entry *e,
 970                               char *type,
 971                               unsigned long *size,
 972                               unsigned long *store_size,
 973                               unsigned int *delta_chain_length,
 974                               unsigned char *base_sha1)
 975{
 976        struct packed_git *p = e->p;
 977        unsigned long offset;
 978        unsigned char *pack;
 979        enum object_type kind;
 980
 981        offset = unpack_object_header(p, e->offset, &kind, size);
 982        pack = (unsigned char *) p->pack_base + offset;
 983        if (kind != OBJ_DELTA)
 984                *delta_chain_length = 0;
 985        else {
 986                unsigned int chain_length = 0;
 987                if (p->pack_size <= offset + 20)
 988                        die("pack file %s records an incomplete delta base",
 989                            p->pack_name);
 990                memcpy(base_sha1, pack, 20);
 991                do {
 992                        struct pack_entry base_ent;
 993                        unsigned long junk;
 994
 995                        find_pack_entry_one(pack, &base_ent, p);
 996                        offset = unpack_object_header(p, base_ent.offset,
 997                                                      &kind, &junk);
 998                        pack = (unsigned char *) p->pack_base + offset;
 999                        chain_length++;
1000                } while (kind == OBJ_DELTA);
1001                *delta_chain_length = chain_length;
1002        }
1003        switch (kind) {
1004        case OBJ_COMMIT:
1005                strcpy(type, commit_type);
1006                break;
1007        case OBJ_TREE:
1008                strcpy(type, tree_type);
1009                break;
1010        case OBJ_BLOB:
1011                strcpy(type, blob_type);
1012                break;
1013        case OBJ_TAG:
1014                strcpy(type, tag_type);
1015                break;
1016        default:
1017                die("corrupted pack file %s containing object of kind %d",
1018                    p->pack_name, kind);
1019        }
1020        *store_size = 0; /* notyet */
1021}
1022
1023static int packed_object_info(struct pack_entry *entry,
1024                              char *type, unsigned long *sizep)
1025{
1026        struct packed_git *p = entry->p;
1027        unsigned long offset, size, left;
1028        unsigned char *pack;
1029        enum object_type kind;
1030        int retval;
1031
1032        if (use_packed_git(p))
1033                die("cannot map packed file");
1034
1035        offset = unpack_object_header(p, entry->offset, &kind, &size);
1036        pack = (unsigned char *) p->pack_base + offset;
1037        left = p->pack_size - offset;
1038
1039        switch (kind) {
1040        case OBJ_DELTA:
1041                retval = packed_delta_info(pack, size, left, type, sizep, p);
1042                unuse_packed_git(p);
1043                return retval;
1044        case OBJ_COMMIT:
1045                strcpy(type, commit_type);
1046                break;
1047        case OBJ_TREE:
1048                strcpy(type, tree_type);
1049                break;
1050        case OBJ_BLOB:
1051                strcpy(type, blob_type);
1052                break;
1053        case OBJ_TAG:
1054                strcpy(type, tag_type);
1055                break;
1056        default:
1057                die("corrupted pack file %s containing object of kind %d",
1058                    p->pack_name, kind);
1059        }
1060        if (sizep)
1061                *sizep = size;
1062        unuse_packed_git(p);
1063        return 0;
1064}
1065
1066/* forward declaration for a mutually recursive function */
1067static void *unpack_entry(struct pack_entry *, char *, unsigned long *);
1068
1069static void *unpack_delta_entry(unsigned char *base_sha1,
1070                                unsigned long delta_size,
1071                                unsigned long left,
1072                                char *type,
1073                                unsigned long *sizep,
1074                                struct packed_git *p)
1075{
1076        struct pack_entry base_ent;
1077        void *data, *delta_data, *result, *base;
1078        unsigned long data_size, result_size, base_size;
1079        z_stream stream;
1080        int st;
1081
1082        if (left < 20)
1083                die("truncated pack file");
1084
1085        /* The base entry _must_ be in the same pack */
1086        if (!find_pack_entry_one(base_sha1, &base_ent, p))
1087                die("failed to find delta-pack base object %s",
1088                    sha1_to_hex(base_sha1));
1089        base = unpack_entry_gently(&base_ent, type, &base_size);
1090        if (!base)
1091                die("failed to read delta-pack base object %s",
1092                    sha1_to_hex(base_sha1));
1093
1094        data = base_sha1 + 20;
1095        data_size = left - 20;
1096        delta_data = xmalloc(delta_size);
1097
1098        memset(&stream, 0, sizeof(stream));
1099
1100        stream.next_in = data;
1101        stream.avail_in = data_size;
1102        stream.next_out = delta_data;
1103        stream.avail_out = delta_size;
1104
1105        inflateInit(&stream);
1106        st = inflate(&stream, Z_FINISH);
1107        inflateEnd(&stream);
1108        if ((st != Z_STREAM_END) || stream.total_out != delta_size)
1109                die("delta data unpack failed");
1110
1111        result = patch_delta(base, base_size,
1112                             delta_data, delta_size,
1113                             &result_size);
1114        if (!result)
1115                die("failed to apply delta");
1116        free(delta_data);
1117        free(base);
1118        *sizep = result_size;
1119        return result;
1120}
1121
1122static void *unpack_non_delta_entry(unsigned char *data,
1123                                    unsigned long size,
1124                                    unsigned long left)
1125{
1126        int st;
1127        z_stream stream;
1128        unsigned char *buffer;
1129
1130        buffer = xmalloc(size + 1);
1131        buffer[size] = 0;
1132        memset(&stream, 0, sizeof(stream));
1133        stream.next_in = data;
1134        stream.avail_in = left;
1135        stream.next_out = buffer;
1136        stream.avail_out = size;
1137
1138        inflateInit(&stream);
1139        st = inflate(&stream, Z_FINISH);
1140        inflateEnd(&stream);
1141        if ((st != Z_STREAM_END) || stream.total_out != size) {
1142                free(buffer);
1143                return NULL;
1144        }
1145
1146        return buffer;
1147}
1148
1149static void *unpack_entry(struct pack_entry *entry,
1150                          char *type, unsigned long *sizep)
1151{
1152        struct packed_git *p = entry->p;
1153        void *retval;
1154
1155        if (use_packed_git(p))
1156                die("cannot map packed file");
1157        retval = unpack_entry_gently(entry, type, sizep);
1158        unuse_packed_git(p);
1159        if (!retval)
1160                die("corrupted pack file %s", p->pack_name);
1161        return retval;
1162}
1163
1164/* The caller is responsible for use_packed_git()/unuse_packed_git() pair */
1165void *unpack_entry_gently(struct pack_entry *entry,
1166                          char *type, unsigned long *sizep)
1167{
1168        struct packed_git *p = entry->p;
1169        unsigned long offset, size, left;
1170        unsigned char *pack;
1171        enum object_type kind;
1172        void *retval;
1173
1174        offset = unpack_object_header(p, entry->offset, &kind, &size);
1175        pack = (unsigned char *) p->pack_base + offset;
1176        left = p->pack_size - offset;
1177        switch (kind) {
1178        case OBJ_DELTA:
1179                retval = unpack_delta_entry(pack, size, left, type, sizep, p);
1180                return retval;
1181        case OBJ_COMMIT:
1182                strcpy(type, commit_type);
1183                break;
1184        case OBJ_TREE:
1185                strcpy(type, tree_type);
1186                break;
1187        case OBJ_BLOB:
1188                strcpy(type, blob_type);
1189                break;
1190        case OBJ_TAG:
1191                strcpy(type, tag_type);
1192                break;
1193        default:
1194                return NULL;
1195        }
1196        *sizep = size;
1197        retval = unpack_non_delta_entry(pack, size, left);
1198        return retval;
1199}
1200
1201int num_packed_objects(const struct packed_git *p)
1202{
1203        /* See check_packed_git_idx() */
1204        return (p->index_size - 20 - 20 - 4*256) / 24;
1205}
1206
1207int nth_packed_object_sha1(const struct packed_git *p, int n,
1208                           unsigned char* sha1)
1209{
1210        void *index = p->index_base + 256;
1211        if (n < 0 || num_packed_objects(p) <= n)
1212                return -1;
1213        memcpy(sha1, (char *) index + (24 * n) + 4, 20);
1214        return 0;
1215}
1216
1217int find_pack_entry_one(const unsigned char *sha1,
1218                        struct pack_entry *e, struct packed_git *p)
1219{
1220        unsigned int *level1_ofs = p->index_base;
1221        int hi = ntohl(level1_ofs[*sha1]);
1222        int lo = ((*sha1 == 0x0) ? 0 : ntohl(level1_ofs[*sha1 - 1]));
1223        void *index = p->index_base + 256;
1224
1225        do {
1226                int mi = (lo + hi) / 2;
1227                int cmp = memcmp((char *) index + (24 * mi) + 4, sha1, 20);
1228                if (!cmp) {
1229                        e->offset = ntohl(*((unsigned int *) ((char *) index + (24 * mi))));
1230                        memcpy(e->sha1, sha1, 20);
1231                        e->p = p;
1232                        return 1;
1233                }
1234                if (cmp > 0)
1235                        hi = mi;
1236                else
1237                        lo = mi+1;
1238        } while (lo < hi);
1239        return 0;
1240}
1241
1242static int find_pack_entry(const unsigned char *sha1, struct pack_entry *e)
1243{
1244        struct packed_git *p;
1245        prepare_packed_git();
1246
1247        for (p = packed_git; p; p = p->next) {
1248                if (find_pack_entry_one(sha1, e, p))
1249                        return 1;
1250        }
1251        return 0;
1252}
1253
1254struct packed_git *find_sha1_pack(const unsigned char *sha1, 
1255                                  struct packed_git *packs)
1256{
1257        struct packed_git *p;
1258        struct pack_entry e;
1259
1260        for (p = packs; p; p = p->next) {
1261                if (find_pack_entry_one(sha1, &e, p))
1262                        return p;
1263        }
1264        return NULL;
1265        
1266}
1267
1268int sha1_object_info(const unsigned char *sha1, char *type, unsigned long *sizep)
1269{
1270        int status;
1271        unsigned long mapsize, size;
1272        void *map;
1273        z_stream stream;
1274        char hdr[128];
1275
1276        map = map_sha1_file(sha1, &mapsize);
1277        if (!map) {
1278                struct pack_entry e;
1279
1280                if (find_pack_entry(sha1, &e))
1281                        return packed_object_info(&e, type, sizep);
1282                reprepare_packed_git();
1283                if (find_pack_entry(sha1, &e))
1284                        return packed_object_info(&e, type, sizep);
1285                return error("unable to find %s", sha1_to_hex(sha1));
1286        }
1287        if (unpack_sha1_header(&stream, map, mapsize, hdr, sizeof(hdr)) < 0)
1288                status = error("unable to unpack %s header",
1289                               sha1_to_hex(sha1));
1290        if (parse_sha1_header(hdr, type, &size) < 0)
1291                status = error("unable to parse %s header", sha1_to_hex(sha1));
1292        else {
1293                status = 0;
1294                if (sizep)
1295                        *sizep = size;
1296        }
1297        inflateEnd(&stream);
1298        munmap(map, mapsize);
1299        return status;
1300}
1301
1302static void *read_packed_sha1(const unsigned char *sha1, char *type, unsigned long *size)
1303{
1304        struct pack_entry e;
1305
1306        if (!find_pack_entry(sha1, &e)) {
1307                error("cannot read sha1_file for %s", sha1_to_hex(sha1));
1308                return NULL;
1309        }
1310        return unpack_entry(&e, type, size);
1311}
1312
1313void * read_sha1_file(const unsigned char *sha1, char *type, unsigned long *size)
1314{
1315        unsigned long mapsize;
1316        void *map, *buf;
1317        struct pack_entry e;
1318
1319        if (find_pack_entry(sha1, &e))
1320                return read_packed_sha1(sha1, type, size);
1321        map = map_sha1_file(sha1, &mapsize);
1322        if (map) {
1323                buf = unpack_sha1_file(map, mapsize, type, size);
1324                munmap(map, mapsize);
1325                return buf;
1326        }
1327        reprepare_packed_git();
1328        if (find_pack_entry(sha1, &e))
1329                return read_packed_sha1(sha1, type, size);
1330        return NULL;
1331}
1332
1333void *read_object_with_reference(const unsigned char *sha1,
1334                                 const char *required_type,
1335                                 unsigned long *size,
1336                                 unsigned char *actual_sha1_return)
1337{
1338        char type[20];
1339        void *buffer;
1340        unsigned long isize;
1341        unsigned char actual_sha1[20];
1342
1343        memcpy(actual_sha1, sha1, 20);
1344        while (1) {
1345                int ref_length = -1;
1346                const char *ref_type = NULL;
1347
1348                buffer = read_sha1_file(actual_sha1, type, &isize);
1349                if (!buffer)
1350                        return NULL;
1351                if (!strcmp(type, required_type)) {
1352                        *size = isize;
1353                        if (actual_sha1_return)
1354                                memcpy(actual_sha1_return, actual_sha1, 20);
1355                        return buffer;
1356                }
1357                /* Handle references */
1358                else if (!strcmp(type, commit_type))
1359                        ref_type = "tree ";
1360                else if (!strcmp(type, tag_type))
1361                        ref_type = "object ";
1362                else {
1363                        free(buffer);
1364                        return NULL;
1365                }
1366                ref_length = strlen(ref_type);
1367
1368                if (memcmp(buffer, ref_type, ref_length) ||
1369                    get_sha1_hex((char *) buffer + ref_length, actual_sha1)) {
1370                        free(buffer);
1371                        return NULL;
1372                }
1373                free(buffer);
1374                /* Now we have the ID of the referred-to object in
1375                 * actual_sha1.  Check again. */
1376        }
1377}
1378
1379char *write_sha1_file_prepare(void *buf,
1380                              unsigned long len,
1381                              const char *type,
1382                              unsigned char *sha1,
1383                              unsigned char *hdr,
1384                              int *hdrlen)
1385{
1386        SHA_CTX c;
1387
1388        /* Generate the header */
1389        *hdrlen = sprintf((char *)hdr, "%s %lu", type, len)+1;
1390
1391        /* Sha1.. */
1392        SHA1_Init(&c);
1393        SHA1_Update(&c, hdr, *hdrlen);
1394        SHA1_Update(&c, buf, len);
1395        SHA1_Final(sha1, &c);
1396
1397        return sha1_file_name(sha1);
1398}
1399
1400/*
1401 * Link the tempfile to the final place, possibly creating the
1402 * last directory level as you do so.
1403 *
1404 * Returns the errno on failure, 0 on success.
1405 */
1406static int link_temp_to_file(const char *tmpfile, char *filename)
1407{
1408        int ret;
1409        char *dir;
1410
1411        if (!link(tmpfile, filename))
1412                return 0;
1413
1414        /*
1415         * Try to mkdir the last path component if that failed.
1416         *
1417         * Re-try the "link()" regardless of whether the mkdir
1418         * succeeds, since a race might mean that somebody
1419         * else succeeded.
1420         */
1421        ret = errno;
1422        dir = strrchr(filename, '/');
1423        if (dir) {
1424                *dir = 0;
1425                mkdir(filename, 0777);
1426                if (adjust_shared_perm(filename))
1427                        return -2;
1428                *dir = '/';
1429                if (!link(tmpfile, filename))
1430                        return 0;
1431                ret = errno;
1432        }
1433        return ret;
1434}
1435
1436/*
1437 * Move the just written object into its final resting place
1438 */
1439int move_temp_to_file(const char *tmpfile, char *filename)
1440{
1441        int ret = link_temp_to_file(tmpfile, filename);
1442
1443        /*
1444         * Coda hack - coda doesn't like cross-directory links,
1445         * so we fall back to a rename, which will mean that it
1446         * won't be able to check collisions, but that's not a
1447         * big deal.
1448         *
1449         * The same holds for FAT formatted media.
1450         *
1451         * When this succeeds, we just return 0. We have nothing
1452         * left to unlink.
1453         */
1454        if (ret && ret != EEXIST) {
1455                if (!rename(tmpfile, filename))
1456                        return 0;
1457                ret = errno;
1458        }
1459        unlink(tmpfile);
1460        if (ret) {
1461                if (ret != EEXIST) {
1462                        fprintf(stderr, "unable to write sha1 filename %s: %s\n", filename, strerror(ret));
1463                        return -1;
1464                }
1465                /* FIXME!!! Collision check here ? */
1466        }
1467
1468        return 0;
1469}
1470
1471static int write_buffer(int fd, const void *buf, size_t len)
1472{
1473        while (len) {
1474                ssize_t size;
1475
1476                size = write(fd, buf, len);
1477                if (!size)
1478                        return error("file write: disk full");
1479                if (size < 0) {
1480                        if (errno == EINTR || errno == EAGAIN)
1481                                continue;
1482                        return error("file write error (%s)", strerror(errno));
1483                }
1484                len -= size;
1485                buf = (char *) buf + size;
1486        }
1487        return 0;
1488}
1489
1490static int write_binary_header(unsigned char *hdr, enum object_type type, unsigned long len)
1491{
1492        int hdr_len;
1493        unsigned char c;
1494
1495        c = (type << 4) | (len & 15);
1496        len >>= 4;
1497        hdr_len = 1;
1498        while (len) {
1499                *hdr++ = c | 0x80;
1500                hdr_len++;
1501                c = (len & 0x7f);
1502                len >>= 7;
1503        }
1504        *hdr = c;
1505        return hdr_len;
1506}
1507
1508static void setup_object_header(z_stream *stream, const char *type, unsigned long len)
1509{
1510        int obj_type, hdr;
1511
1512        if (use_legacy_headers) {
1513                while (deflate(stream, 0) == Z_OK)
1514                        /* nothing */;
1515                return;
1516        }
1517        if (!strcmp(type, blob_type))
1518                obj_type = OBJ_BLOB;
1519        else if (!strcmp(type, tree_type))
1520                obj_type = OBJ_TREE;
1521        else if (!strcmp(type, commit_type))
1522                obj_type = OBJ_COMMIT;
1523        else if (!strcmp(type, tag_type))
1524                obj_type = OBJ_TAG;
1525        else
1526                die("trying to generate bogus object of type '%s'", type);
1527        hdr = write_binary_header(stream->next_out, obj_type, len);
1528        stream->total_out = hdr;
1529        stream->next_out += hdr;
1530        stream->avail_out -= hdr;
1531}
1532
1533int write_sha1_file(void *buf, unsigned long len, const char *type, unsigned char *returnsha1)
1534{
1535        int size;
1536        unsigned char *compressed;
1537        z_stream stream;
1538        unsigned char sha1[20];
1539        char *filename;
1540        static char tmpfile[PATH_MAX];
1541        unsigned char hdr[50];
1542        int fd, hdrlen;
1543
1544        /* Normally if we have it in the pack then we do not bother writing
1545         * it out into .git/objects/??/?{38} file.
1546         */
1547        filename = write_sha1_file_prepare(buf, len, type, sha1, hdr, &hdrlen);
1548        if (returnsha1)
1549                memcpy(returnsha1, sha1, 20);
1550        if (has_sha1_file(sha1))
1551                return 0;
1552        fd = open(filename, O_RDONLY);
1553        if (fd >= 0) {
1554                /*
1555                 * FIXME!!! We might do collision checking here, but we'd
1556                 * need to uncompress the old file and check it. Later.
1557                 */
1558                close(fd);
1559                return 0;
1560        }
1561
1562        if (errno != ENOENT) {
1563                fprintf(stderr, "sha1 file %s: %s\n", filename, strerror(errno));
1564                return -1;
1565        }
1566
1567        snprintf(tmpfile, sizeof(tmpfile), "%s/obj_XXXXXX", get_object_directory());
1568
1569        fd = mkstemp(tmpfile);
1570        if (fd < 0) {
1571                fprintf(stderr, "unable to create temporary sha1 filename %s: %s\n", tmpfile, strerror(errno));
1572                return -1;
1573        }
1574
1575        /* Set it up */
1576        memset(&stream, 0, sizeof(stream));
1577        deflateInit(&stream, zlib_compression_level);
1578        size = 8 + deflateBound(&stream, len+hdrlen);
1579        compressed = xmalloc(size);
1580
1581        /* Compress it */
1582        stream.next_out = compressed;
1583        stream.avail_out = size;
1584
1585        /* First header.. */
1586        stream.next_in = hdr;
1587        stream.avail_in = hdrlen;
1588        setup_object_header(&stream, type, len);
1589
1590        /* Then the data itself.. */
1591        stream.next_in = buf;
1592        stream.avail_in = len;
1593        while (deflate(&stream, Z_FINISH) == Z_OK)
1594                /* nothing */;
1595        deflateEnd(&stream);
1596        size = stream.total_out;
1597
1598        if (write_buffer(fd, compressed, size) < 0)
1599                die("unable to write sha1 file");
1600        fchmod(fd, 0444);
1601        close(fd);
1602        free(compressed);
1603
1604        return move_temp_to_file(tmpfile, filename);
1605}
1606
1607/*
1608 * We need to unpack and recompress the object for writing
1609 * it out to a different file.
1610 */
1611static void *repack_object(const unsigned char *sha1, unsigned long *objsize)
1612{
1613        size_t size;
1614        z_stream stream;
1615        unsigned char *unpacked;
1616        unsigned long len;
1617        char type[20];
1618        char hdr[50];
1619        int hdrlen;
1620        void *buf;
1621
1622        /* need to unpack and recompress it by itself */
1623        unpacked = read_packed_sha1(sha1, type, &len);
1624
1625        hdrlen = sprintf(hdr, "%s %lu", type, len) + 1;
1626
1627        /* Set it up */
1628        memset(&stream, 0, sizeof(stream));
1629        deflateInit(&stream, zlib_compression_level);
1630        size = deflateBound(&stream, len + hdrlen);
1631        buf = xmalloc(size);
1632
1633        /* Compress it */
1634        stream.next_out = buf;
1635        stream.avail_out = size;
1636
1637        /* First header.. */
1638        stream.next_in = (void *)hdr;
1639        stream.avail_in = hdrlen;
1640        while (deflate(&stream, 0) == Z_OK)
1641                /* nothing */;
1642
1643        /* Then the data itself.. */
1644        stream.next_in = unpacked;
1645        stream.avail_in = len;
1646        while (deflate(&stream, Z_FINISH) == Z_OK)
1647                /* nothing */;
1648        deflateEnd(&stream);
1649        free(unpacked);
1650
1651        *objsize = stream.total_out;
1652        return buf;
1653}
1654
1655int write_sha1_to_fd(int fd, const unsigned char *sha1)
1656{
1657        int retval;
1658        unsigned long objsize;
1659        void *buf = map_sha1_file(sha1, &objsize);
1660
1661        if (buf) {
1662                retval = write_buffer(fd, buf, objsize);
1663                munmap(buf, objsize);
1664                return retval;
1665        }
1666
1667        buf = repack_object(sha1, &objsize);
1668        retval = write_buffer(fd, buf, objsize);
1669        free(buf);
1670        return retval;
1671}
1672
1673int write_sha1_from_fd(const unsigned char *sha1, int fd, char *buffer,
1674                       size_t bufsize, size_t *bufposn)
1675{
1676        char tmpfile[PATH_MAX];
1677        int local;
1678        z_stream stream;
1679        unsigned char real_sha1[20];
1680        unsigned char discard[4096];
1681        int ret;
1682        SHA_CTX c;
1683
1684        snprintf(tmpfile, sizeof(tmpfile), "%s/obj_XXXXXX", get_object_directory());
1685
1686        local = mkstemp(tmpfile);
1687        if (local < 0)
1688                return error("Couldn't open %s for %s",
1689                             tmpfile, sha1_to_hex(sha1));
1690
1691        memset(&stream, 0, sizeof(stream));
1692
1693        inflateInit(&stream);
1694
1695        SHA1_Init(&c);
1696
1697        do {
1698                ssize_t size;
1699                if (*bufposn) {
1700                        stream.avail_in = *bufposn;
1701                        stream.next_in = (unsigned char *) buffer;
1702                        do {
1703                                stream.next_out = discard;
1704                                stream.avail_out = sizeof(discard);
1705                                ret = inflate(&stream, Z_SYNC_FLUSH);
1706                                SHA1_Update(&c, discard, sizeof(discard) -
1707                                            stream.avail_out);
1708                        } while (stream.avail_in && ret == Z_OK);
1709                        if (write_buffer(local, buffer, *bufposn - stream.avail_in) < 0)
1710                                die("unable to write sha1 file");
1711                        memmove(buffer, buffer + *bufposn - stream.avail_in,
1712                                stream.avail_in);
1713                        *bufposn = stream.avail_in;
1714                        if (ret != Z_OK)
1715                                break;
1716                }
1717                size = read(fd, buffer + *bufposn, bufsize - *bufposn);
1718                if (size <= 0) {
1719                        close(local);
1720                        unlink(tmpfile);
1721                        if (!size)
1722                                return error("Connection closed?");
1723                        perror("Reading from connection");
1724                        return -1;
1725                }
1726                *bufposn += size;
1727        } while (1);
1728        inflateEnd(&stream);
1729
1730        close(local);
1731        SHA1_Final(real_sha1, &c);
1732        if (ret != Z_STREAM_END) {
1733                unlink(tmpfile);
1734                return error("File %s corrupted", sha1_to_hex(sha1));
1735        }
1736        if (memcmp(sha1, real_sha1, 20)) {
1737                unlink(tmpfile);
1738                return error("File %s has bad hash", sha1_to_hex(sha1));
1739        }
1740
1741        return move_temp_to_file(tmpfile, sha1_file_name(sha1));
1742}
1743
1744int has_pack_index(const unsigned char *sha1)
1745{
1746        struct stat st;
1747        if (stat(sha1_pack_index_name(sha1), &st))
1748                return 0;
1749        return 1;
1750}
1751
1752int has_pack_file(const unsigned char *sha1)
1753{
1754        struct stat st;
1755        if (stat(sha1_pack_name(sha1), &st))
1756                return 0;
1757        return 1;
1758}
1759
1760int has_sha1_pack(const unsigned char *sha1)
1761{
1762        struct pack_entry e;
1763        return find_pack_entry(sha1, &e);
1764}
1765
1766int has_sha1_file(const unsigned char *sha1)
1767{
1768        struct stat st;
1769        struct pack_entry e;
1770
1771        if (find_pack_entry(sha1, &e))
1772                return 1;
1773        return find_sha1_file(sha1, &st) ? 1 : 0;
1774}
1775
1776/*
1777 * reads from fd as long as possible into a supplied buffer of size bytes.
1778 * If necessary the buffer's size is increased using realloc()
1779 *
1780 * returns 0 if anything went fine and -1 otherwise
1781 *
1782 * NOTE: both buf and size may change, but even when -1 is returned
1783 * you still have to free() it yourself.
1784 */
1785int read_pipe(int fd, char** return_buf, unsigned long* return_size)
1786{
1787        char* buf = *return_buf;
1788        unsigned long size = *return_size;
1789        int iret;
1790        unsigned long off = 0;
1791
1792        do {
1793                iret = xread(fd, buf + off, size - off);
1794                if (iret > 0) {
1795                        off += iret;
1796                        if (off == size) {
1797                                size *= 2;
1798                                buf = realloc(buf, size);
1799                        }
1800                }
1801        } while (iret > 0);
1802
1803        *return_buf = buf;
1804        *return_size = off;
1805
1806        if (iret < 0)
1807                return -1;
1808        return 0;
1809}
1810
1811int index_pipe(unsigned char *sha1, int fd, const char *type, int write_object)
1812{
1813        unsigned long size = 4096;
1814        char *buf = malloc(size);
1815        int ret;
1816        unsigned char hdr[50];
1817        int hdrlen;
1818
1819        if (read_pipe(fd, &buf, &size)) {
1820                free(buf);
1821                return -1;
1822        }
1823
1824        if (!type)
1825                type = blob_type;
1826        if (write_object)
1827                ret = write_sha1_file(buf, size, type, sha1);
1828        else {
1829                write_sha1_file_prepare(buf, size, type, sha1, hdr, &hdrlen);
1830                ret = 0;
1831        }
1832        free(buf);
1833        return ret;
1834}
1835
1836int index_fd(unsigned char *sha1, int fd, struct stat *st, int write_object, const char *type)
1837{
1838        unsigned long size = st->st_size;
1839        void *buf;
1840        int ret;
1841        unsigned char hdr[50];
1842        int hdrlen;
1843
1844        buf = "";
1845        if (size)
1846                buf = mmap(NULL, size, PROT_READ, MAP_PRIVATE, fd, 0);
1847        close(fd);
1848        if (buf == MAP_FAILED)
1849                return -1;
1850
1851        if (!type)
1852                type = blob_type;
1853        if (write_object)
1854                ret = write_sha1_file(buf, size, type, sha1);
1855        else {
1856                write_sha1_file_prepare(buf, size, type, sha1, hdr, &hdrlen);
1857                ret = 0;
1858        }
1859        if (size)
1860                munmap(buf, size);
1861        return ret;
1862}
1863
1864int index_path(unsigned char *sha1, const char *path, struct stat *st, int write_object)
1865{
1866        int fd;
1867        char *target;
1868
1869        switch (st->st_mode & S_IFMT) {
1870        case S_IFREG:
1871                fd = open(path, O_RDONLY);
1872                if (fd < 0)
1873                        return error("open(\"%s\"): %s", path,
1874                                     strerror(errno));
1875                if (index_fd(sha1, fd, st, write_object, NULL) < 0)
1876                        return error("%s: failed to insert into database",
1877                                     path);
1878                break;
1879        case S_IFLNK:
1880                target = xmalloc(st->st_size+1);
1881                if (readlink(path, target, st->st_size+1) != st->st_size) {
1882                        char *errstr = strerror(errno);
1883                        free(target);
1884                        return error("readlink(\"%s\"): %s", path,
1885                                     errstr);
1886                }
1887                if (!write_object) {
1888                        unsigned char hdr[50];
1889                        int hdrlen;
1890                        write_sha1_file_prepare(target, st->st_size, blob_type,
1891                                                sha1, hdr, &hdrlen);
1892                } else if (write_sha1_file(target, st->st_size, blob_type, sha1))
1893                        return error("%s: failed to insert into database",
1894                                     path);
1895                free(target);
1896                break;
1897        default:
1898                return error("%s: unsupported file type", path);
1899        }
1900        return 0;
1901}