db_upgrade.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
- /*-
- * See the file LICENSE for redistribution information.
- *
- * Copyright (c) 1996-2002
- * Sleepycat Software. All rights reserved.
- */
- #include "db_config.h"
- #ifndef lint
- static const char copyright[] =
- "Copyright (c) 1996-2002nSleepycat Software Inc. All rights reserved.n";
- static const char revid[] =
- "$Id: db_upgrade.c,v 1.31 2002/03/28 20:13:47 bostic Exp $";
- #endif
- #ifndef NO_SYSTEM_INCLUDES
- #include <sys/types.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <unistd.h>
- #endif
- #include "db_int.h"
- int main __P((int, char *[]));
- int usage __P((void));
- int version_check __P((const char *));
- int
- main(argc, argv)
- int argc;
- char *argv[];
- {
- extern char *optarg;
- extern int optind;
- const char *progname = "db_upgrade";
- DB *dbp;
- DB_ENV *dbenv;
- u_int32_t flags;
- int ch, e_close, exitval, nflag, ret, t_ret;
- char *home, *passwd;
- if ((ret = version_check(progname)) != 0)
- return (ret);
- dbenv = NULL;
- flags = nflag = 0;
- e_close = exitval = 0;
- home = passwd = NULL;
- while ((ch = getopt(argc, argv, "h:NP:sV")) != EOF)
- switch (ch) {
- case 'h':
- home = optarg;
- break;
- case 'N':
- nflag = 1;
- break;
- case 'P':
- passwd = strdup(optarg);
- memset(optarg, 0, strlen(optarg));
- if (passwd == NULL) {
- fprintf(stderr, "%s: strdup: %sn",
- progname, strerror(errno));
- return (EXIT_FAILURE);
- }
- break;
- case 's':
- LF_SET(DB_DUPSORT);
- break;
- case 'V':
- printf("%sn", db_version(NULL, NULL, NULL));
- return (EXIT_SUCCESS);
- case '?':
- default:
- return (usage());
- }
- argc -= optind;
- argv += optind;
- if (argc <= 0)
- return (usage());
- /* Handle possible interruptions. */
- __db_util_siginit();
- /*
- * Create an environment object and initialize it for error
- * reporting.
- */
- if ((ret = db_env_create(&dbenv, 0)) != 0) {
- fprintf(stderr, "%s: db_env_create: %sn",
- progname, db_strerror(ret));
- goto shutdown;
- }
- e_close = 1;
- dbenv->set_errfile(dbenv, stderr);
- dbenv->set_errpfx(dbenv, progname);
- if (nflag) {
- if ((ret = dbenv->set_flags(dbenv, DB_NOLOCKING, 1)) != 0) {
- dbenv->err(dbenv, ret, "set_flags: DB_NOLOCKING");
- goto shutdown;
- }
- if ((ret = dbenv->set_flags(dbenv, DB_NOPANIC, 1)) != 0) {
- dbenv->err(dbenv, ret, "set_flags: DB_NOPANIC");
- goto shutdown;
- }
- }
- if (passwd != NULL && (ret = dbenv->set_encrypt(dbenv,
- passwd, DB_ENCRYPT_AES)) != 0) {
- dbenv->err(dbenv, ret, "set_passwd");
- goto shutdown;
- }
- /*
- * If attaching to a pre-existing environment fails, create a
- * private one and try again.
- */
- if ((ret = dbenv->open(dbenv,
- home, DB_JOINENV | DB_USE_ENVIRON, 0)) != 0 &&
- (ret = dbenv->open(dbenv, home,
- DB_CREATE | DB_INIT_MPOOL | DB_PRIVATE | DB_USE_ENVIRON, 0)) != 0) {
- dbenv->err(dbenv, ret, "open");
- goto shutdown;
- }
- for (; !__db_util_interrupted() && argv[0] != NULL; ++argv) {
- if ((ret = db_create(&dbp, dbenv, 0)) != 0) {
- fprintf(stderr,
- "%s: db_create: %sn", progname, db_strerror(ret));
- goto shutdown;
- }
- dbp->set_errfile(dbp, stderr);
- dbp->set_errpfx(dbp, progname);
- if ((ret = dbp->upgrade(dbp, argv[0], flags)) != 0)
- dbp->err(dbp, ret, "DB->upgrade: %s", argv[0]);
- if ((t_ret = dbp->close(dbp, 0)) != 0 && ret == 0) {
- dbenv->err(dbenv, ret, "DB->close: %s", argv[0]);
- ret = t_ret;
- }
- if (ret != 0)
- goto shutdown;
- }
- if (0) {
- shutdown: exitval = 1;
- }
- if (e_close && (ret = dbenv->close(dbenv, 0)) != 0) {
- exitval = 1;
- fprintf(stderr,
- "%s: dbenv->close: %sn", progname, db_strerror(ret));
- }
- /* Resend any caught signal. */
- __db_util_sigresend();
- return (exitval == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
- }
- int
- usage()
- {
- fprintf(stderr, "%sn",
- "usage: db_upgrade [-NsV] [-h home] [-P password] db_file ...");
- return (EXIT_FAILURE);
- }
- int
- version_check(progname)
- const char *progname;
- {
- int v_major, v_minor, v_patch;
- /* Make sure we're loaded with the right version of the DB library. */
- (void)db_version(&v_major, &v_minor, &v_patch);
- if (v_major != DB_VERSION_MAJOR ||
- v_minor != DB_VERSION_MINOR || v_patch != DB_VERSION_PATCH) {
- fprintf(stderr,
- "%s: version %d.%d.%d doesn't match library version %d.%d.%dn",
- progname, DB_VERSION_MAJOR, DB_VERSION_MINOR,
- DB_VERSION_PATCH, v_major, v_minor, v_patch);
- return (EXIT_FAILURE);
- }
- return (0);
- }