Andrew's git / gitweb.git /
567d9da708296ff89364c19d3d69e038184583b9
   1#include "cache.h"
   2#include "oidmap.h"
   3#include "object-store.h"
   4#include "replace-object.h"
   5#include "refs.h"
   6#include "repository.h"
   7#include "commit.h"
   8
   9static int register_replace_ref(const char *refname,
  10                                const struct object_id *oid,
  11                                int flag, void *cb_data)
  12{
  13        /* Get sha1 from refname */
  14        const char *slash = strrchr(refname, '/');
  15        const char *hash = slash ? slash + 1 : refname;
  16        struct replace_object *repl_obj = xmalloc(sizeof(*repl_obj));
  17
  18        if (get_oid_hex(hash, &repl_obj->original.oid)) {
  19                free(repl_obj);
  20                warning("bad replace ref name: %s", refname);
  21                return 0;
  22        }
  23
  24        /* Copy sha1 from the read ref */
  25        oidcpy(&repl_obj->replacement, oid);
  26
  27        /* Register new object */
  28        if (oidmap_put(the_repository->objects->replace_map, repl_obj))
  29                die("duplicate replace ref: %s", refname);
  30
  31        return 0;
  32}
  33
  34#define prepare_replace_object(r) \
  35        prepare_replace_object_##r()
  36static void prepare_replace_object_the_repository(void)
  37{
  38        if (the_repository->objects->replace_map)
  39                return;
  40
  41        the_repository->objects->replace_map =
  42                xmalloc(sizeof(*the_repository->objects->replace_map));
  43        oidmap_init(the_repository->objects->replace_map, 0);
  44
  45        for_each_replace_ref(the_repository, register_replace_ref, NULL);
  46}
  47
  48/* We allow "recursive" replacement. Only within reason, though */
  49#define MAXREPLACEDEPTH 5
  50
  51/*
  52 * If a replacement for object oid has been set up, return the
  53 * replacement object's name (replaced recursively, if necessary).
  54 * The return value is either oid or a pointer to a
  55 * permanently-allocated value.  This function always respects replace
  56 * references, regardless of the value of check_replace_refs.
  57 */
  58const struct object_id *do_lookup_replace_object(const struct object_id *oid)
  59{
  60        int depth = MAXREPLACEDEPTH;
  61        const struct object_id *cur = oid;
  62
  63        prepare_replace_object(the_repository);
  64
  65        /* Try to recursively replace the object */
  66        while (depth-- > 0) {
  67                struct replace_object *repl_obj =
  68                        oidmap_get(the_repository->objects->replace_map, cur);
  69                if (!repl_obj)
  70                        return cur;
  71                cur = &repl_obj->replacement;
  72        }
  73        die("replace depth too high for object %s", oid_to_hex(oid));
  74}