comparison extra/fast-export/svn-archive.c @ 1544:e9e55585ebf2 feature_1136

Add fast-export

+/*
+ * svn-archive.c
+ * ----------
+ *  Walk through a given revision of a local Subversion repository and export 
+ *  all of the contents as a tarfile.
+ *
+ * Author: Chris Lee <[e-mail]>
+ * License: MIT <http://www.opensource.org/licenses/mit-license.php>
+ */
+
+#define _XOPEN_SOURCE
+#include <unistd.h>
+#include <string.h>
+#include <stdio.h>
+#include <time.h>
+
+#ifndef PATH_MAX
+#define PATH_MAX 4096
+#endif
+
+#include <apr_general.h>
+#include <apr_strings.h>
+#include <apr_getopt.h>
+#include <apr_lib.h>
+
+#include <svn_types.h>
+#include <svn_pools.h>
+#include <svn_repos.h>
+#include <svn_fs.h>
+
+#undef SVN_ERR
+#define SVN_ERR(expr) SVN_INT_ERR(expr)
+#define apr_sane_push(arr, contents) *(char **)apr_array_push(arr) = contents
+
+#define TRUNK "/trunk"
+
+static time_t archive_time;
+
+time_t get_epoch(char *svn_date)
+{
+    struct tm tm = {0};
+    char *date = malloc(strlen(svn_date) * sizeof(char *));
+    strncpy(date, svn_date, strlen(svn_date) - 8);
+    strptime(date, "%Y-%m-%dT%H:%M:%S", &tm);
+    free(date);
+    return mktime(&tm);
+}
+
+int tar_header(apr_pool_t *pool, char *path, char *node, size_t f_size)
+{
+    char          buf[512];
+    unsigned int  i, checksum;
+    svn_boolean_t is_dir;
+
+    memset(buf, 0, sizeof(buf));
+
+    if ((strlen(path) == 0) && (strlen(node) == 0)) {
+        return 0;
+    }
+
+    if (strlen(node) == 0) {
+        is_dir = 1;
+    } else {
+        is_dir = 0;
+    }
+
+    if (strlen(path) == 0) {
+        strncpy(buf, apr_psprintf(pool, "%s", node), 99);
+    } else if (strlen(path) + strlen(node) < 100) {
+        strncpy(buf, apr_psprintf(pool, "%s/%s", path+1, node), 99);
+    } else {
+        fprintf(stderr, "really long file path...\n");
+        strncpy(&buf[0], node, 99);
+        strncpy(&buf[345], path+1, 154);
+    }
+
+    strncpy(&buf[100], apr_psprintf(pool, "%07o", (is_dir ? 0755 : 0644)), 7);
+    strncpy(&buf[108], apr_psprintf(pool, "%07o", 1000), 7);
+    strncpy(&buf[116], apr_psprintf(pool, "%07o", 1000), 7);
+    strncpy(&buf[124], apr_psprintf(pool, "%011lo", f_size), 11);
+    strncpy(&buf[136], apr_psprintf(pool, "%011lo", archive_time), 11);
+    strncpy(&buf[156], (is_dir ? "5" : "0"), 1);
+    strncpy(&buf[257], "ustar  ", 8);
+    strncpy(&buf[265], "clee", 31);
+    strncpy(&buf[297], "clee", 31);
+    // strncpy(&buf[329], apr_psprintf(pool, "%07o", 0), 7);
+    // strncpy(&buf[337], apr_psprintf(pool, "%07o", 0), 7);
+
+    strncpy(&buf[148], "        ", 8);
+    checksum = 0;
+    for (i = 0; i < sizeof(buf); i++) {
+        checksum += buf[i];
+    }
+    strncpy(&buf[148], apr_psprintf(pool, "%07o", checksum & 0x1fffff), 7);
+
+    fwrite(buf, sizeof(char), sizeof(buf), stdout);
+
+    return 0;
+}
+
+int tar_footer()
+{
+    char block[1024];
+    memset(block, 0, sizeof(block));
+    fwrite(block, sizeof(char), sizeof(block), stdout);
+}
+
+int dump_blob(svn_fs_root_t *root, char *prefix, char *path, char *node, apr_pool_t *pool)
+{
+    char           *full_path, buf[512];
+    apr_size_t     len;
+    svn_stream_t   *stream;
+    svn_filesize_t stream_length;
+
+    full_path = apr_psprintf(pool, "%s%s/%s", prefix, path, node);
+
+    SVN_ERR(svn_fs_file_length(&stream_length, root, full_path, pool));
+    SVN_ERR(svn_fs_file_contents(&stream, root, full_path, pool));
+
+    tar_header(pool, path, node, stream_length);
+
+    do {
+        len = sizeof(buf);
+        memset(buf, '\0', sizeof(buf));
+        SVN_ERR(svn_stream_read(stream, buf, &len));
+        fwrite(buf, sizeof(char), sizeof(buf), stdout);
+    } while (len == sizeof(buf));
+
+    return 0;
+}
+
+int dump_tree(svn_fs_root_t *root, char *prefix, char *path, apr_pool_t *pool)
+{
+    const void       *key;
+    void             *val;
+    char             *node, *subpath, *full_path;
+
+    apr_pool_t       *subpool;
+    apr_hash_t       *dir_entries;
+    apr_hash_index_t *i;
+
+    svn_boolean_t    is_dir;
+
+    tar_header(pool, path, "", 0);
+
+    SVN_ERR(svn_fs_dir_entries(&dir_entries, root, apr_psprintf(pool, "%s/%s", prefix, path), pool));
+
+    subpool = svn_pool_create(pool);
+
+    for (i = apr_hash_first(pool, dir_entries); i; i = apr_hash_next(i)) {
+        svn_pool_clear(subpool);
+        apr_hash_this(i, &key, NULL, &val);
+        node = (char *)key;
+
+        subpath = apr_psprintf(subpool, "%s/%s", path, node);
+        full_path = apr_psprintf(subpool, "%s%s", prefix, subpath);
+
+        svn_fs_is_dir(&is_dir, root, full_path, subpool);
+
+        if (is_dir) {
+            dump_tree(root, prefix, subpath, subpool);
+        } else {
+            dump_blob(root, prefix, path, node, subpool);
+        }
+    }
+
+    svn_pool_destroy(subpool);
+
+    return 0;
+}
+
+int crawl_filesystem(char *repos_path, char *root_path, apr_pool_t *pool)
+{
+    char                 *path;
+
+    apr_hash_t           *props;
+    apr_hash_index_t     *i;
+
+    svn_repos_t          *repos;
+    svn_fs_t             *fs;
+    svn_string_t         *svndate;
+    svn_revnum_t         youngest_rev, export_rev;
+    svn_fs_root_t        *fs_root;
+
+    SVN_ERR(svn_fs_initialize(pool));
+    SVN_ERR(svn_repos_open(&repos, repos_path, pool));
+    if ((fs = svn_repos_fs(repos)) == NULL)
+      return -1;
+    SVN_ERR(svn_fs_youngest_rev(&youngest_rev, fs, pool));
+
+    export_rev = youngest_rev;
+
+    SVN_ERR(svn_fs_revision_root(&fs_root, fs, export_rev, pool));
+    SVN_ERR(svn_fs_revision_proplist(&props, fs, export_rev, pool));
+
+    svndate = apr_hash_get(props, "svn:date", APR_HASH_KEY_STRING);
+    archive_time = get_epoch((char *)svndate->data);
+
+    fprintf(stderr, "Exporting archive of r%ld... \n", export_rev);
+
+    dump_tree(fs_root, root_path, "", pool);
+
+    tar_footer();
+
+    fprintf(stderr, "done!\n");
+
+    return 0;
+}
+
+int main(int argc, char *argv[])
+{
+    apr_pool_t           *pool;
+    apr_getopt_t         *options;
+
+    apr_getopt_option_t long_options[] = {
+        { "help",     'h', 0 },
+        { "prefix",   'p', 0 },
+        { "basename", 'b', 0 },
+        { "revision", 'r', 0 },
+        { NULL,       0,   0 }
+    };
+
+    if (argc < 2) {
+        fprintf(stderr, "usage: %s REPOS_PATH [prefix]\n", argv[0]);
+        return -1;
+    }
+
+    if (apr_initialize() != APR_SUCCESS) {
+        fprintf(stderr, "You lose at apr_initialize().\n");
+        return -1;
+    }
+
+    pool = svn_pool_create(NULL);
+
+    crawl_filesystem(argv[1], (argc == 3 ? argv[2] : TRUNK), pool);
+
+    apr_terminate();
+
+    return 0;
+}