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