patch-ids.c

   1 #include "cache.h"
   2 #include "diff.h"
   3 #include "commit.h"
   4 #include "sha1-lookup.h"
   5 #include "patch-ids.h"
   6
   7 static int commit_patch_id(struct commit *commit, struct diff_options *options,
   8                     unsigned char *sha1)
   9 {
  10         if (commit->parents)
  11                 diff_tree_sha1(commit->parents->item->object.sha1,
  12                                commit->object.sha1, "", options);
  13         else
  14                 diff_root_tree_sha1(commit->object.sha1, "", options);
  15         diffcore_std(options);
  16         return diff_flush_patch_id(options, sha1);
  17 }
  18
  19 static const unsigned char *patch_id_access(size_t index, void *table)
  20 {
  21         struct patch_id **id_table = table;
  22         return id_table[index]->patch_id;
  23 }
  24
  25 static int patch_pos(struct patch_id **table, int nr, const unsigned char *id)
  26 {
  27         return sha1_pos(id, table, nr, patch_id_access);
  28 }
  29
  30 #define BUCKET_SIZE 190 /* 190 * 21 = 3990, with slop close enough to 4K */
  31 struct patch_id_bucket {
  32         struct patch_id_bucket *next;
  33         int nr;
  34         struct patch_id bucket[BUCKET_SIZE];
  35 };
  36
  37 int init_patch_ids(struct patch_ids *ids)
  38 {
  39         memset(ids, 0, sizeof(*ids));
  40         diff_setup(&ids->diffopts);
  41         DIFF_OPT_SET(&ids->diffopts, RECURSIVE);
  42         diff_setup_done(&ids->diffopts);
  43         return 0;
  44 }
  45
  46 int free_patch_ids(struct patch_ids *ids)
  47 {
  48         struct patch_id_bucket *next, *patches;
  49
  50         free(ids->table);
  51         for (patches = ids->patches; patches; patches = next) {
  52                 next = patches->next;
  53                 free(patches);
  54         }
  55         return 0;
  56 }
  57
  58 static struct patch_id *add_commit(struct commit *commit,
  59                                    struct patch_ids *ids,
  60                                    int no_add)
  61 {
  62         struct patch_id_bucket *bucket;
  63         struct patch_id *ent;
  64         unsigned char sha1[20];
  65         int pos;
  66
  67         if (commit_patch_id(commit, &ids->diffopts, sha1))
  68                 return NULL;
  69         pos = patch_pos(ids->table, ids->nr, sha1);
  70         if (0 <= pos)
  71                 return ids->table[pos];
  72         if (no_add)
  73                 return NULL;
  74
  75         pos = -1 - pos;
  76
  77         bucket = ids->patches;
  78         if (!bucket || (BUCKET_SIZE <= bucket->nr)) {
  79                 bucket = xcalloc(1, sizeof(*bucket));
  80                 bucket->next = ids->patches;
  81                 ids->patches = bucket;
  82         }
  83         ent = &bucket->bucket[bucket->nr++];
  84         hashcpy(ent->patch_id, sha1);
  85
  86         ALLOC_GROW(ids->table, ids->nr + 1, ids->alloc);
  87         if (pos < ids->nr)
  88                 memmove(ids->table + pos + 1, ids->table + pos,
  89                         sizeof(ent) * (ids->nr - pos));
  90         ids->nr++;
  91         ids->table[pos] = ent;
  92         return ids->table[pos];
  93 }
  94
  95 struct patch_id *has_commit_patch_id(struct commit *commit,
  96                                      struct patch_ids *ids)
  97 {
  98         return add_commit(commit, ids, 1);
  99 }
 100
 101 struct patch_id *add_commit_patch_id(struct commit *commit,
 102                                      struct patch_ids *ids)
 103 {
 104         return add_commit(commit, ids, 0);
 105 }