svndump.c

   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 */
  31 obj_pool_gen(log, char, 4096)
  32
  33 static 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
  42 static 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
  47 static struct {
  48         uint32_t revision, author;
  49         unsigned long timestamp;
  50         char *log;
  51 } rev_ctx;
  52
  53 static struct {
  54         uint32_t uuid, url;
  55 } dump_ctx;
  56
  57 static 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
  64 static 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
  77 static 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
  85 static void reset_dump_ctx(uint32_t url)
  86 {
  87         dump_ctx.url = url;
  88         dump_ctx.uuid = ~0;
  89 }
  90
  91 static 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
 110 static 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
 143 static 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
 185 static 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
 192 void 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: %d\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
 271 void 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
 281 void 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
 294 void 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 }