git.sur5r.net Git - openldap/blob - libraries/liblmdb/mdb_copy.c

   1 /* mdb_copy.c - memory-mapped database backup tool */
   2 /*
   3  * Copyright 2012 Howard Chu, Symas Corp.
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted only as authorized by the OpenLDAP
   8  * Public License.
   9  *
  10  * A copy of this license is available in the file LICENSE in the
  11  * top-level directory of the distribution or, alternatively, at
  12  * <http://www.OpenLDAP.org/license.html>.
  13  */
  14 #ifdef _WIN32
  15 #include <windows.h>
  16 #define MDB_STDOUT      GetStdHandle(STD_OUTPUT_HANDLE)
  17 #else
  18 #define MDB_STDOUT      1
  19 #endif
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <signal.h>
  23 #include "lmdb.h"
  24
  25 static void
  26 sighandle(int sig)
  27 {
  28 }
  29
  30 int main(int argc,char * argv[])
  31 {
  32         int rc;
  33         MDB_env *env;
  34         const char *progname = argv[0], *act;
  35         unsigned flags = MDB_RDONLY;
  36
  37         for (; argc > 1 && argv[1][0] == '-'; argc--, argv++) {
  38                 if (argv[1][1] == 'n' && argv[1][2] == '\0')
  39                         flags |= MDB_NOSUBDIR;
  40                 else if (argv[1][1] == 'V' && argv[1][2] == '\0') {
  41                         printf("%s\n", MDB_VERSION_STRING);
  42                         exit(0);
  43                 } else
  44                         argc = 0;
  45         }
  46
  47         if (argc<2 || argc>3) {
  48                 fprintf(stderr, "usage: %s [-V] [-n] srcpath [dstpath]\n", progname);
  49                 exit(EXIT_FAILURE);
  50         }
  51
  52 #ifdef SIGPIPE
  53         signal(SIGPIPE, sighandle);
  54 #endif
  55 #ifdef SIGHUP
  56         signal(SIGHUP, sighandle);
  57 #endif
  58         signal(SIGINT, sighandle);
  59         signal(SIGTERM, sighandle);
  60
  61         act = "opening environment";
  62         rc = mdb_env_create(&env);
  63         if (rc == MDB_SUCCESS) {
  64                 rc = mdb_env_open(env, argv[1], flags, 0);
  65         }
  66         if (rc == MDB_SUCCESS) {
  67                 act = "copying";
  68                 if (argc == 2)
  69                         rc = mdb_env_copyfd(env, MDB_STDOUT);
  70                 else
  71                         rc = mdb_env_copy(env, argv[2]);
  72         }
  73         if (rc)
  74                 fprintf(stderr, "%s: %s failed, error %d (%s)\n",
  75                         progname, act, rc, mdb_strerror(rc));
  76         mdb_env_close(env);
  77
  78         return rc ? EXIT_FAILURE : EXIT_SUCCESS;
  79 }