Andrew's git - gitweb.git/blob - vcs-svn/repo

   1/*
   2 * Licensed under a two-clause BSD-style license.
   3 * See LICENSE for details.
   4 */
   5
   6#include "git-compat-util.h"
   7
   8#include "string_pool.h"
   9#include "repo_tree.h"
  10#include "obj_pool.h"
  11#include "fast_export.h"
  12
  13#include "trp.h"
  14
  15struct repo_dirent {
  16        uint32_t name_offset;
  17        struct trp_node children;
  18        uint32_t mode;
  19        uint32_t content_offset;
  20};
  21
  22struct repo_dir {
  23        struct trp_root entries;
  24};
  25
  26struct repo_commit {
  27        uint32_t root_dir_offset;
  28};
  29
  30/* Memory pools for commit, dir and dirent */
  31obj_pool_gen(commit, struct repo_commit, 4096)
  32obj_pool_gen(dir, struct repo_dir, 4096)
  33obj_pool_gen(dent, struct repo_dirent, 4096)
  34
  35static uint32_t active_commit;
  36static uint32_t mark;
  37
  38static int repo_dirent_name_cmp(const void *a, const void *b);
  39
  40/* Treap for directory entries */
  41trp_gen(static, dent_, struct repo_dirent, children, dent, repo_dirent_name_cmp)
  42
  43uint32_t next_blob_mark(void)
  44{
  45        return mark++;
  46}
  47
  48static struct repo_dir *repo_commit_root_dir(struct repo_commit *commit)
  49{
  50        return dir_pointer(commit->root_dir_offset);
  51}
  52
  53static struct repo_dirent *repo_first_dirent(struct repo_dir *dir)
  54{
  55        return dent_first(&dir->entries);
  56}
  57
  58static int repo_dirent_name_cmp(const void *a, const void *b)
  59{
  60        const struct repo_dirent *dent1 = a, *dent2 = b;
  61        uint32_t a_offset = dent1->name_offset;
  62        uint32_t b_offset = dent2->name_offset;
  63        return (a_offset > b_offset) - (a_offset < b_offset);
  64}
  65
  66static int repo_dirent_is_dir(struct repo_dirent *dent)
  67{
  68        return dent != NULL && dent->mode == REPO_MODE_DIR;
  69}
  70
  71static struct repo_dir *repo_dir_from_dirent(struct repo_dirent *dent)
  72{
  73        if (!repo_dirent_is_dir(dent))
  74                return NULL;
  75        return dir_pointer(dent->content_offset);
  76}
  77
  78static struct repo_dir *repo_clone_dir(struct repo_dir *orig_dir)
  79{
  80        uint32_t orig_o, new_o;
  81        orig_o = dir_offset(orig_dir);
  82        if (orig_o >= dir_pool.committed)
  83                return orig_dir;
  84        new_o = dir_alloc(1);
  85        orig_dir = dir_pointer(orig_o);
  86        *dir_pointer(new_o) = *orig_dir;
  87        return dir_pointer(new_o);
  88}
  89
  90static struct repo_dirent *repo_read_dirent(uint32_t revision, uint32_t *path)
  91{
  92        uint32_t name = 0;
  93        struct repo_dirent *key = dent_pointer(dent_alloc(1));
  94        struct repo_dir *dir = NULL;
  95        struct repo_dirent *dent = NULL;
  96        dir = repo_commit_root_dir(commit_pointer(revision));
  97        while (~(name = *path++)) {
  98                key->name_offset = name;
  99                dent = dent_search(&dir->entries, key);
 100                if (dent == NULL || !repo_dirent_is_dir(dent))
 101                        break;
 102                dir = repo_dir_from_dirent(dent);
 103        }
 104        dent_free(1);
 105        return dent;
 106}
 107
 108static void repo_write_dirent(uint32_t *path, uint32_t mode,
 109                              uint32_t content_offset, uint32_t del)
 110{
 111        uint32_t name, revision, dir_o = ~0, parent_dir_o = ~0;
 112        struct repo_dir *dir;
 113        struct repo_dirent *key;
 114        struct repo_dirent *dent = NULL;
 115        revision = active_commit;
 116        dir = repo_commit_root_dir(commit_pointer(revision));
 117        dir = repo_clone_dir(dir);
 118        commit_pointer(revision)->root_dir_offset = dir_offset(dir);
 119        while (~(name = *path++)) {
 120                parent_dir_o = dir_offset(dir);
 121
 122                key = dent_pointer(dent_alloc(1));
 123                key->name_offset = name;
 124
 125                dent = dent_search(&dir->entries, key);
 126                if (dent == NULL)
 127                        dent = key;
 128                else
 129                        dent_free(1);
 130
 131                if (dent == key) {
 132                        dent->mode = REPO_MODE_DIR;
 133                        dent->content_offset = 0;
 134                        dent = dent_insert(&dir->entries, dent);
 135                }
 136
 137                if (dent_offset(dent) < dent_pool.committed) {
 138                        dir_o = repo_dirent_is_dir(dent) ?
 139                                        dent->content_offset : ~0;
 140                        dent_remove(&dir->entries, dent);
 141                        dent = dent_pointer(dent_alloc(1));
 142                        dent->name_offset = name;
 143                        dent->mode = REPO_MODE_DIR;
 144                        dent->content_offset = dir_o;
 145                        dent = dent_insert(&dir->entries, dent);
 146                }
 147
 148                dir = repo_dir_from_dirent(dent);
 149                dir = repo_clone_dir(dir);
 150                dent->content_offset = dir_offset(dir);
 151        }
 152        if (dent == NULL)
 153                return;
 154        dent->mode = mode;
 155        dent->content_offset = content_offset;
 156        if (del && ~parent_dir_o)
 157                dent_remove(&dir_pointer(parent_dir_o)->entries, dent);
 158}
 159
 160uint32_t repo_copy(uint32_t revision, uint32_t *src, uint32_t *dst)
 161{
 162        uint32_t mode = 0, content_offset = 0;
 163        struct repo_dirent *src_dent;
 164        src_dent = repo_read_dirent(revision, src);
 165        if (src_dent != NULL) {
 166                mode = src_dent->mode;
 167                content_offset = src_dent->content_offset;
 168                repo_write_dirent(dst, mode, content_offset, 0);
 169        }
 170        return mode;
 171}
 172
 173void repo_add(uint32_t *path, uint32_t mode, uint32_t blob_mark)
 174{
 175        repo_write_dirent(path, mode, blob_mark, 0);
 176}
 177
 178uint32_t repo_modify_path(uint32_t *path, uint32_t mode, uint32_t blob_mark)
 179{
 180        struct repo_dirent *src_dent;
 181        src_dent = repo_read_dirent(active_commit, path);
 182        if (!src_dent)
 183                return 0;
 184        if (!blob_mark)
 185                blob_mark = src_dent->content_offset;
 186        if (!mode)
 187                mode = src_dent->mode;
 188        repo_write_dirent(path, mode, blob_mark, 0);
 189        return mode;
 190}
 191
 192void repo_delete(uint32_t *path)
 193{
 194        repo_write_dirent(path, 0, 0, 1);
 195}
 196
 197static void repo_git_add_r(uint32_t depth, uint32_t *path, struct repo_dir *dir);
 198
 199static void repo_git_add(uint32_t depth, uint32_t *path, struct repo_dirent *dent)
 200{
 201        if (repo_dirent_is_dir(dent))
 202                repo_git_add_r(depth, path, repo_dir_from_dirent(dent));
 203        else
 204                fast_export_modify(depth, path,
 205                                   dent->mode, dent->content_offset);
 206}
 207
 208static void repo_git_add_r(uint32_t depth, uint32_t *path, struct repo_dir *dir)
 209{
 210        struct repo_dirent *de = repo_first_dirent(dir);
 211        while (de) {
 212                path[depth] = de->name_offset;
 213                repo_git_add(depth + 1, path, de);
 214                de = dent_next(&dir->entries, de);
 215        }
 216}
 217
 218static void repo_diff_r(uint32_t depth, uint32_t *path, struct repo_dir *dir1,
 219                        struct repo_dir *dir2)
 220{
 221        struct repo_dirent *de1, *de2;
 222        de1 = repo_first_dirent(dir1);
 223        de2 = repo_first_dirent(dir2);
 224
 225        while (de1 && de2) {
 226                if (de1->name_offset < de2->name_offset) {
 227                        path[depth] = de1->name_offset;
 228                        fast_export_delete(depth + 1, path);
 229                        de1 = dent_next(&dir1->entries, de1);
 230                        continue;
 231                }
 232                if (de1->name_offset > de2->name_offset) {
 233                        path[depth] = de2->name_offset;
 234                        repo_git_add(depth + 1, path, de2);
 235                        de2 = dent_next(&dir2->entries, de2);
 236                        continue;
 237                }
 238                path[depth] = de1->name_offset;
 239
 240                if (de1->mode == de2->mode &&
 241                    de1->content_offset == de2->content_offset) {
 242                        ; /* No change. */
 243                } else if (repo_dirent_is_dir(de1) && repo_dirent_is_dir(de2)) {
 244                        repo_diff_r(depth + 1, path,
 245                                    repo_dir_from_dirent(de1),
 246                                    repo_dir_from_dirent(de2));
 247                } else if (!repo_dirent_is_dir(de1) && !repo_dirent_is_dir(de2)) {
 248                        repo_git_add(depth + 1, path, de2);
 249                } else {
 250                        fast_export_delete(depth + 1, path);
 251                        repo_git_add(depth + 1, path, de2);
 252                }
 253                de1 = dent_next(&dir1->entries, de1);
 254                de2 = dent_next(&dir2->entries, de2);
 255        }
 256        while (de1) {
 257                path[depth] = de1->name_offset;
 258                fast_export_delete(depth + 1, path);
 259                de1 = dent_next(&dir1->entries, de1);
 260        }
 261        while (de2) {
 262                path[depth] = de2->name_offset;
 263                repo_git_add(depth + 1, path, de2);
 264                de2 = dent_next(&dir2->entries, de2);
 265        }
 266}
 267
 268static uint32_t path_stack[REPO_MAX_PATH_DEPTH];
 269
 270void repo_diff(uint32_t r1, uint32_t r2)
 271{
 272        repo_diff_r(0,
 273                    path_stack,
 274                    repo_commit_root_dir(commit_pointer(r1)),
 275                    repo_commit_root_dir(commit_pointer(r2)));
 276}
 277
 278void repo_commit(uint32_t revision, uint32_t author, char *log, uint32_t uuid,
 279                 uint32_t url, unsigned long timestamp)
 280{
 281        fast_export_commit(revision, author, log, uuid, url, timestamp);
 282        dent_commit();
 283        dir_commit();
 284        active_commit = commit_alloc(1);
 285        commit_pointer(active_commit)->root_dir_offset =
 286                commit_pointer(active_commit - 1)->root_dir_offset;
 287}
 288
 289static void mark_init(void)
 290{
 291        uint32_t i;
 292        mark = 0;
 293        for (i = 0; i < dent_pool.size; i++)
 294                if (!repo_dirent_is_dir(dent_pointer(i)) &&
 295                    dent_pointer(i)->content_offset > mark)
 296                        mark = dent_pointer(i)->content_offset;
 297        mark++;
 298}
 299
 300void repo_init(void)
 301{
 302        mark_init();
 303        if (commit_pool.size == 0) {
 304                /* Create empty tree for commit 0. */
 305                commit_alloc(1);
 306                commit_pointer(0)->root_dir_offset = dir_alloc(1);
 307                dir_pointer(0)->entries.trp_root = ~0;
 308                dir_commit();
 309        }
 310        /* Preallocate next commit, ready for changes. */
 311        active_commit = commit_alloc(1);
 312        commit_pointer(active_commit)->root_dir_offset =
 313                commit_pointer(active_commit - 1)->root_dir_offset;
 314}
 315
 316void repo_reset(void)
 317{
 318        pool_reset();
 319        commit_reset();
 320        dir_reset();
 321        dent_reset();
 322}