Happy New Year

[openldap] / libraries / liblmdb / mdb_copy.c

diff --git a/libraries/liblmdb/mdb_copy.c b/libraries/liblmdb/mdb_copy.c
index bd0b85911083c5aa9a1e7dc9926552ae868646c3..1b89396eb9de4e7e4c88d82c2f276c9c602a3113 100644 (file)
--- a/libraries/liblmdb/mdb_copy.c
+++ b/libraries/liblmdb/mdb_copy.c
@@ -1,6 +1,6 @@
 /* mdb_copy.c - memory-mapped database backup tool */
 /*
- * Copyright 2012 Howard Chu, Symas Corp.
+ * Copyright 2012-2018 Howard Chu, Symas Corp.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -11,31 +11,71 @@
  * top-level directory of the distribution or, alternatively, at
  * <http://www.OpenLDAP.org/license.html>.
  */
+#ifdef _WIN32
+#include <windows.h>
+#define        MDB_STDOUT      GetStdHandle(STD_OUTPUT_HANDLE)
+#else
+#define        MDB_STDOUT      1
+#endif
 #include <stdio.h>
 #include <stdlib.h>
+#include <signal.h>
 #include "lmdb.h"
 
+static void
+sighandle(int sig)
+{
+}
+
 int main(int argc,char * argv[])
 {
        int rc;
        MDB_env *env;
-       char *envname = argv[1];
+       const char *progname = argv[0], *act;
+       unsigned flags = MDB_RDONLY;
+       unsigned cpflags = 0;
+
+       for (; argc > 1 && argv[1][0] == '-'; argc--, argv++) {
+               if (argv[1][1] == 'n' && argv[1][2] == '\0')
+                       flags |= MDB_NOSUBDIR;
+               else if (argv[1][1] == 'c' && argv[1][2] == '\0')
+                       cpflags |= MDB_CP_COMPACT;
+               else if (argv[1][1] == 'V' && argv[1][2] == '\0') {
+                       printf("%s\n", MDB_VERSION_STRING);
+                       exit(0);
+               } else
+                       argc = 0;
+       }
 
-       if (argc != 3) {
-               fprintf(stderr, "usage: %s srcpath dstpath\n", argv[0]);
+       if (argc<2 || argc>3) {
+               fprintf(stderr, "usage: %s [-V] [-c] [-n] srcpath [dstpath]\n", progname);
                exit(EXIT_FAILURE);
        }
 
-       rc = mdb_env_create(&env);
+#ifdef SIGPIPE
+       signal(SIGPIPE, sighandle);
+#endif
+#ifdef SIGHUP
+       signal(SIGHUP, sighandle);
+#endif
+       signal(SIGINT, sighandle);
+       signal(SIGTERM, sighandle);
 
-       rc = mdb_env_open(env, envname, MDB_RDONLY, 0);
-       if (rc) {
-               printf("mdb_env_open failed, error %d %s\n", rc, mdb_strerror(rc));
-       } else {
-               rc = mdb_env_copy(env, argv[2]);
-               if (rc)
-                       printf("mdb_env_copy failed, error %d %s\n", rc, mdb_strerror(rc));
+       act = "opening environment";
+       rc = mdb_env_create(&env);
+       if (rc == MDB_SUCCESS) {
+               rc = mdb_env_open(env, argv[1], flags, 0600);
+       }
+       if (rc == MDB_SUCCESS) {
+               act = "copying";
+               if (argc == 2)
+                       rc = mdb_env_copyfd2(env, MDB_STDOUT, cpflags);
+               else
+                       rc = mdb_env_copy2(env, argv[2], cpflags);
        }
+       if (rc)
+               fprintf(stderr, "%s: %s failed, error %d (%s)\n",
+                       progname, act, rc, mdb_strerror(rc));
        mdb_env_close(env);
 
        return rc ? EXIT_FAILURE : EXIT_SUCCESS;