1/*- 2 * See the file LICENSE for redistribution information. 3 * 4 * Copyright (c) 1996-2009 Oracle. All rights reserved. 5 * 6 * $Id$ 7 */ 8 9#include "db_config.h" 10 11#include "db_int.h" 12 13#ifndef lint 14static const char copyright[] = 15 "Copyright (c) 1996-2009 Oracle. All rights reserved.\n"; 16#endif 17 18int main __P((int, char *[])); 19int usage __P((void)); 20int version_check __P((void)); 21 22const char *progname; 23 24int 25main(argc, argv) 26 int argc; 27 char *argv[]; 28{ 29 extern char *optarg; 30 extern int optind; 31 DB_ENV *dbenv; 32 time_t now; 33 long argval; 34 u_int32_t flags, kbytes, minutes, seconds; 35 int ch, exitval, once, ret, verbose; 36 char *home, *logfile, *passwd, time_buf[CTIME_BUFLEN]; 37 38 if ((progname = __db_rpath(argv[0])) == NULL) 39 progname = argv[0]; 40 else 41 ++progname; 42 43 if ((ret = version_check()) != 0) 44 return (ret); 45 46 /* 47 * !!! 48 * Don't allow a fully unsigned 32-bit number, some compilers get 49 * upset and require it to be specified in hexadecimal and so on. 50 */ 51#define MAX_UINT32_T 2147483647 52 53 dbenv = NULL; 54 kbytes = minutes = 0; 55 exitval = once = verbose = 0; 56 flags = 0; 57 home = logfile = passwd = NULL; 58 while ((ch = getopt(argc, argv, "1h:k:L:P:p:Vv")) != EOF) 59 switch (ch) { 60 case '1': 61 once = 1; 62 flags = DB_FORCE; 63 break; 64 case 'h': 65 home = optarg; 66 break; 67 case 'k': 68 if (__db_getlong(NULL, progname, 69 optarg, 1, (long)MAX_UINT32_T, &argval)) 70 return (EXIT_FAILURE); 71 kbytes = (u_int32_t)argval; 72 break; 73 case 'L': 74 logfile = optarg; 75 break; 76 case 'P': 77 passwd = strdup(optarg); 78 memset(optarg, 0, strlen(optarg)); 79 if (passwd == NULL) { 80 fprintf(stderr, "%s: strdup: %s\n", 81 progname, strerror(errno)); 82 return (EXIT_FAILURE); 83 } 84 break; 85 case 'p': 86 if (__db_getlong(NULL, progname, 87 optarg, 1, (long)MAX_UINT32_T, &argval)) 88 return (EXIT_FAILURE); 89 minutes = (u_int32_t)argval; 90 break; 91 case 'V': 92 printf("%s\n", db_version(NULL, NULL, NULL)); 93 return (EXIT_SUCCESS); 94 case 'v': 95 verbose = 1; 96 break; 97 case '?': 98 default: 99 return (usage()); 100 } 101 argc -= optind; 102 argv += optind; 103 104 if (argc != 0) 105 return (usage()); 106 107 if (once == 0 && kbytes == 0 && minutes == 0) { 108 (void)fprintf(stderr, 109 "%s: at least one of -1, -k and -p must be specified\n", 110 progname); 111 return (usage()); 112 } 113 114 /* Handle possible interruptions. */ 115 __db_util_siginit(); 116 117 /* Log our process ID. */ 118 if (logfile != NULL && __db_util_logset(progname, logfile)) 119 goto shutdown; 120 121 /* 122 * Create an environment object and initialize it for error 123 * reporting. 124 */ 125 if ((ret = db_env_create(&dbenv, 0)) != 0) { 126 fprintf(stderr, 127 "%s: db_env_create: %s\n", progname, db_strerror(ret)); 128 goto shutdown; 129 } 130 131 dbenv->set_errfile(dbenv, stderr); 132 dbenv->set_errpfx(dbenv, progname); 133 134 if (passwd != NULL && (ret = dbenv->set_encrypt(dbenv, 135 passwd, DB_ENCRYPT_AES)) != 0) { 136 dbenv->err(dbenv, ret, "set_passwd"); 137 goto shutdown; 138 } 139 140 /* 141 * If attaching to a pre-existing environment fails, create a 142 * private one and try again. Turn on DB_THREAD in case a repmgr 143 * application wants to do checkpointing using this utility: repmgr 144 * requires DB_THREAD for all env handles. 145 */ 146#ifdef HAVE_REPLICATION_THREADS 147#define ENV_FLAGS (DB_THREAD | DB_USE_ENVIRON) 148#else 149#define ENV_FLAGS DB_USE_ENVIRON 150#endif 151 if ((ret = dbenv->open(dbenv, home, ENV_FLAGS, 0)) != 0 && 152 (!once || ret == DB_VERSION_MISMATCH || 153 (ret = dbenv->open(dbenv, home, 154 DB_CREATE | DB_INIT_TXN | DB_PRIVATE | DB_USE_ENVIRON, 0)) != 0)) { 155 dbenv->err(dbenv, ret, "DB_ENV->open"); 156 goto shutdown; 157 } 158 159 /* 160 * If we have only a time delay, then we'll sleep the right amount 161 * to wake up when a checkpoint is necessary. If we have a "kbytes" 162 * field set, then we'll check every 30 seconds. 163 */ 164 seconds = kbytes != 0 ? 30 : minutes * 60; 165 while (!__db_util_interrupted()) { 166 if (verbose) { 167 (void)time(&now); 168 dbenv->errx(dbenv, 169 "checkpoint begin: %s", __os_ctime(&now, time_buf)); 170 } 171 172 if ((ret = dbenv->txn_checkpoint(dbenv, 173 kbytes, minutes, flags)) != 0) { 174 dbenv->err(dbenv, ret, "txn_checkpoint"); 175 goto shutdown; 176 } 177 178 if (verbose) { 179 (void)time(&now); 180 dbenv->errx(dbenv, 181 "checkpoint complete: %s", __os_ctime(&now, time_buf)); 182 } 183 184 if (once) 185 break; 186 187 __os_yield(dbenv->env, seconds, 0); 188 } 189 190 if (0) { 191shutdown: exitval = 1; 192 } 193 194 /* Clean up the logfile. */ 195 if (logfile != NULL) 196 (void)remove(logfile); 197 198 /* Clean up the environment. */ 199 if (dbenv != NULL && (ret = dbenv->close(dbenv, 0)) != 0) { 200 exitval = 1; 201 fprintf(stderr, 202 "%s: dbenv->close: %s\n", progname, db_strerror(ret)); 203 } 204 205 if (passwd != NULL) 206 free(passwd); 207 208 /* Resend any caught signal. */ 209 __db_util_sigresend(); 210 211 return (exitval == 0 ? EXIT_SUCCESS : EXIT_FAILURE); 212} 213 214int 215usage() 216{ 217 (void)fprintf(stderr, "usage: %s [-1Vv]\n\t%s\n", progname, 218 "[-h home] [-k kbytes] [-L file] [-P password] [-p min]"); 219 return (EXIT_FAILURE); 220} 221 222int 223version_check() 224{ 225 int v_major, v_minor, v_patch; 226 227 /* Make sure we're loaded with the right version of the DB library. */ 228 (void)db_version(&v_major, &v_minor, &v_patch); 229 if (v_major != DB_VERSION_MAJOR || v_minor != DB_VERSION_MINOR) { 230 fprintf(stderr, 231 "%s: version %d.%d doesn't match library version %d.%d\n", 232 progname, DB_VERSION_MAJOR, DB_VERSION_MINOR, 233 v_major, v_minor); 234 return (EXIT_FAILURE); 235 } 236 return (0); 237} 238