vcs-svn / svndump.con commit t4017 (diff-retval): replace manual exit code check with test_expect_code (dd0c69e)
   1/*
   2 * Parse and rearrange a svnadmin dump.
   3 * Create the dump with:
   4 * svnadmin dump --incremental -r<startrev>:<endrev> <repository> >outfile
   5 *
   6 * Licensed under a two-clause BSD-style license.
   7 * See LICENSE for details.
   8 */
   9
  10#include "cache.h"
  11#include "repo_tree.h"
  12#include "fast_export.h"
  13#include "line_buffer.h"
  14#include "obj_pool.h"
  15#include "string_pool.h"
  16
  17#define NODEACT_REPLACE 4
  18#define NODEACT_DELETE 3
  19#define NODEACT_ADD 2
  20#define NODEACT_CHANGE 1
  21#define NODEACT_UNKNOWN 0
  22
  23#define DUMP_CTX 0
  24#define REV_CTX  1
  25#define NODE_CTX 2
  26
  27#define LENGTH_UNKNOWN (~0)
  28#define DATE_RFC2822_LEN 31
  29
  30/* Create memory pool for log messages */
  31obj_pool_gen(log, char, 4096)
  32
  33static char* log_copy(uint32_t length, char *log)
  34{
  35        char *buffer;
  36        log_free(log_pool.size);
  37        buffer = log_pointer(log_alloc(length));
  38        strncpy(buffer, log, length);
  39        return buffer;
  40}
  41
  42static struct {
  43        uint32_t action, propLength, textLength, srcRev, srcMode, mark, type;
  44        uint32_t src[REPO_MAX_PATH_DEPTH], dst[REPO_MAX_PATH_DEPTH];
  45} node_ctx;
  46
  47static struct {
  48        uint32_t revision, author;
  49        unsigned long timestamp;
  50        char *log;
  51} rev_ctx;
  52
  53static struct {
  54        uint32_t uuid, url;
  55} dump_ctx;
  56
  57static struct {
  58        uint32_t svn_log, svn_author, svn_date, svn_executable, svn_special, uuid,
  59                revision_number, node_path, node_kind, node_action,
  60                node_copyfrom_path, node_copyfrom_rev, text_content_length,
  61                prop_content_length, content_length;
  62} keys;
  63
  64static void reset_node_ctx(char *fname)
  65{
  66        node_ctx.type = 0;
  67        node_ctx.action = NODEACT_UNKNOWN;
  68        node_ctx.propLength = LENGTH_UNKNOWN;
  69        node_ctx.textLength = LENGTH_UNKNOWN;
  70        node_ctx.src[0] = ~0;
  71        node_ctx.srcRev = 0;
  72        node_ctx.srcMode = 0;
  73        pool_tok_seq(REPO_MAX_PATH_DEPTH, node_ctx.dst, "/", fname);
  74        node_ctx.mark = 0;
  75}
  76
  77static void reset_rev_ctx(uint32_t revision)
  78{
  79        rev_ctx.revision = revision;
  80        rev_ctx.timestamp = 0;
  81        rev_ctx.log = NULL;
  82        rev_ctx.author = ~0;
  83}
  84
  85static void reset_dump_ctx(uint32_t url)
  86{
  87        dump_ctx.url = url;
  88        dump_ctx.uuid = ~0;
  89}
  90
  91static void init_keys(void)
  92{
  93        keys.svn_log = pool_intern("svn:log");
  94        keys.svn_author = pool_intern("svn:author");
  95        keys.svn_date = pool_intern("svn:date");
  96        keys.svn_executable = pool_intern("svn:executable");
  97        keys.svn_special = pool_intern("svn:special");
  98        keys.uuid = pool_intern("UUID");
  99        keys.revision_number = pool_intern("Revision-number");
 100        keys.node_path = pool_intern("Node-path");
 101        keys.node_kind = pool_intern("Node-kind");
 102        keys.node_action = pool_intern("Node-action");
 103        keys.node_copyfrom_path = pool_intern("Node-copyfrom-path");
 104        keys.node_copyfrom_rev = pool_intern("Node-copyfrom-rev");
 105        keys.text_content_length = pool_intern("Text-content-length");
 106        keys.prop_content_length = pool_intern("Prop-content-length");
 107        keys.content_length = pool_intern("Content-length");
 108}
 109
 110static void read_props(void)
 111{
 112        uint32_t len;
 113        uint32_t key = ~0;
 114        char *val = NULL;
 115        char *t;
 116        while ((t = buffer_read_line()) && strcmp(t, "PROPS-END")) {
 117                if (!strncmp(t, "K ", 2)) {
 118                        len = atoi(&t[2]);
 119                        key = pool_intern(buffer_read_string(len));
 120                        buffer_read_line();
 121                } else if (!strncmp(t, "V ", 2)) {
 122                        len = atoi(&t[2]);
 123                        val = buffer_read_string(len);
 124                        if (key == keys.svn_log) {
 125                                /* Value length excludes terminating nul. */
 126                                rev_ctx.log = log_copy(len + 1, val);
 127                        } else if (key == keys.svn_author) {
 128                                rev_ctx.author = pool_intern(val);
 129                        } else if (key == keys.svn_date) {
 130                                if (parse_date_basic(val, &rev_ctx.timestamp, NULL))
 131                                        fprintf(stderr, "Invalid timestamp: %s\n", val);
 132                        } else if (key == keys.svn_executable) {
 133                                node_ctx.type = REPO_MODE_EXE;
 134                        } else if (key == keys.svn_special) {
 135                                node_ctx.type = REPO_MODE_LNK;
 136                        }
 137                        key = ~0;
 138                        buffer_read_line();
 139                }
 140        }
 141}
 142
 143static void handle_node(void)
 144{
 145        if (node_ctx.propLength != LENGTH_UNKNOWN && node_ctx.propLength)
 146                read_props();
 147
 148        if (node_ctx.srcRev)
 149                node_ctx.srcMode = repo_copy(node_ctx.srcRev, node_ctx.src, node_ctx.dst);
 150
 151        if (node_ctx.textLength != LENGTH_UNKNOWN &&
 152            node_ctx.type != REPO_MODE_DIR)
 153                node_ctx.mark = next_blob_mark();
 154
 155        if (node_ctx.action == NODEACT_DELETE) {
 156                repo_delete(node_ctx.dst);
 157        } else if (node_ctx.action == NODEACT_CHANGE ||
 158                           node_ctx.action == NODEACT_REPLACE) {
 159                if (node_ctx.action == NODEACT_REPLACE &&
 160                    node_ctx.type == REPO_MODE_DIR)
 161                        repo_replace(node_ctx.dst, node_ctx.mark);
 162                else if (node_ctx.propLength != LENGTH_UNKNOWN)
 163                        repo_modify(node_ctx.dst, node_ctx.type, node_ctx.mark);
 164                else if (node_ctx.textLength != LENGTH_UNKNOWN)
 165                        node_ctx.srcMode = repo_replace(node_ctx.dst, node_ctx.mark);
 166        } else if (node_ctx.action == NODEACT_ADD) {
 167                if (node_ctx.srcRev && node_ctx.propLength != LENGTH_UNKNOWN)
 168                        repo_modify(node_ctx.dst, node_ctx.type, node_ctx.mark);
 169                else if (node_ctx.srcRev && node_ctx.textLength != LENGTH_UNKNOWN)
 170                        node_ctx.srcMode = repo_replace(node_ctx.dst, node_ctx.mark);
 171                else if ((node_ctx.type == REPO_MODE_DIR && !node_ctx.srcRev) ||
 172                         node_ctx.textLength != LENGTH_UNKNOWN)
 173                        repo_add(node_ctx.dst, node_ctx.type, node_ctx.mark);
 174        }
 175
 176        if (node_ctx.propLength == LENGTH_UNKNOWN && node_ctx.srcMode)
 177                node_ctx.type = node_ctx.srcMode;
 178
 179        if (node_ctx.mark)
 180                fast_export_blob(node_ctx.type, node_ctx.mark, node_ctx.textLength);
 181        else if (node_ctx.textLength != LENGTH_UNKNOWN)
 182                buffer_skip_bytes(node_ctx.textLength);
 183}
 184
 185static void handle_revision(void)
 186{
 187        if (rev_ctx.revision)
 188                repo_commit(rev_ctx.revision, rev_ctx.author, rev_ctx.log,
 189                        dump_ctx.uuid, dump_ctx.url, rev_ctx.timestamp);
 190}
 191
 192void svndump_read(const char *url)
 193{
 194        char *val;
 195        char *t;
 196        uint32_t active_ctx = DUMP_CTX;
 197        uint32_t len;
 198        uint32_t key;
 199
 200        reset_dump_ctx(pool_intern(url));
 201        while ((t = buffer_read_line())) {
 202                val = strstr(t, ": ");
 203                if (!val)
 204                        continue;
 205                *val++ = '\0';
 206                *val++ = '\0';
 207                key = pool_intern(t);
 208
 209                if (key == keys.uuid) {
 210                        dump_ctx.uuid = pool_intern(val);
 211                } else if (key == keys.revision_number) {
 212                        if (active_ctx == NODE_CTX)
 213                                handle_node();
 214                        if (active_ctx != DUMP_CTX)
 215                                handle_revision();
 216                        active_ctx = REV_CTX;
 217                        reset_rev_ctx(atoi(val));
 218                } else if (key == keys.node_path) {
 219                        if (active_ctx == NODE_CTX)
 220                                handle_node();
 221                        active_ctx = NODE_CTX;
 222                        reset_node_ctx(val);
 223                } else if (key == keys.node_kind) {
 224                        if (!strcmp(val, "dir"))
 225                                node_ctx.type = REPO_MODE_DIR;
 226                        else if (!strcmp(val, "file"))
 227                                node_ctx.type = REPO_MODE_BLB;
 228                        else
 229                                fprintf(stderr, "Unknown node-kind: %s\n", val);
 230                } else if (key == keys.node_action) {
 231                        if (!strcmp(val, "delete")) {
 232                                node_ctx.action = NODEACT_DELETE;
 233                        } else if (!strcmp(val, "add")) {
 234                                node_ctx.action = NODEACT_ADD;
 235                        } else if (!strcmp(val, "change")) {
 236                                node_ctx.action = NODEACT_CHANGE;
 237                        } else if (!strcmp(val, "replace")) {
 238                                node_ctx.action = NODEACT_REPLACE;
 239                        } else {
 240                                fprintf(stderr, "Unknown node-action: %s\n", val);
 241                                node_ctx.action = NODEACT_UNKNOWN;
 242                        }
 243                } else if (key == keys.node_copyfrom_path) {
 244                        pool_tok_seq(REPO_MAX_PATH_DEPTH, node_ctx.src, "/", val);
 245                } else if (key == keys.node_copyfrom_rev) {
 246                        node_ctx.srcRev = atoi(val);
 247                } else if (key == keys.text_content_length) {
 248                        node_ctx.textLength = atoi(val);
 249                } else if (key == keys.prop_content_length) {
 250                        node_ctx.propLength = atoi(val);
 251                } else if (key == keys.content_length) {
 252                        len = atoi(val);
 253                        buffer_read_line();
 254                        if (active_ctx == REV_CTX) {
 255                                read_props();
 256                        } else if (active_ctx == NODE_CTX) {
 257                                handle_node();
 258                                active_ctx = REV_CTX;
 259                        } else {
 260                                fprintf(stderr, "Unexpected content length header: %"PRIu32"\n", len);
 261                                buffer_skip_bytes(len);
 262                        }
 263                }
 264        }
 265        if (active_ctx == NODE_CTX)
 266                handle_node();
 267        if (active_ctx != DUMP_CTX)
 268                handle_revision();
 269}
 270
 271void svndump_init(const char *filename)
 272{
 273        buffer_init(filename);
 274        repo_init();
 275        reset_dump_ctx(~0);
 276        reset_rev_ctx(0);
 277        reset_node_ctx(NULL);
 278        init_keys();
 279}
 280
 281void svndump_deinit(void)
 282{
 283        log_reset();
 284        repo_reset();
 285        reset_dump_ctx(~0);
 286        reset_rev_ctx(0);
 287        reset_node_ctx(NULL);
 288        if (buffer_deinit())
 289                fprintf(stderr, "Input error\n");
 290        if (ferror(stdout))
 291                fprintf(stderr, "Output error\n");
 292}
 293
 294void svndump_reset(void)
 295{
 296        log_reset();
 297        buffer_reset();
 298        repo_reset();
 299        reset_dump_ctx(~0);
 300        reset_rev_ctx(0);
 301        reset_node_ctx(NULL);
 302}