#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/uio.h>
#include <fcntl.h>
#include <ctype.h>
#include <syslog.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include "acl.h"
#include "assert.h"
#include "auth.h"
#include "cyrusdb.h"
#include "exitcodes.h"
#include "glob.h"
#include "imap_err.h"
#include "global.h"
#include "mailbox.h"
#include "util.h"
#include "xmalloc.h"
#define STACKSIZE 64000
char stack[STACKSIZE+1];
const int config_need_data = 0;
struct cyrusdb_backend *DB_OLD = NULL;
struct db *odb = NULL;
int read_key_value(char **keyptr, int *keylen, char **valptr, int *vallen) {
int c,res,inkey;
res = 0;
inkey = 1;
*keyptr = stack;
*keylen = 0;
*vallen = 0;
while( (c = getchar()) != EOF ) {
if (c == '\n') break;
if ((c == '\t') && inkey) {
inkey = 0;
*valptr = stack + *keylen + 1;
} else {
if (inkey) {
(*keyptr)[(*keylen)++] = c;
res = 1;
} else {
(*valptr)[(*vallen)++] = c;
}
}
if (*keylen + *vallen >= STACKSIZE - 1) {
printf("Error, stack overflow\n");
fatal("stack overflow", EC_DATAERR);
}
}
(*keyptr)[*keylen] = '\0';
if (inkey) {
*valptr = *keyptr + *keylen;
} else {
(*valptr)[*vallen] = '\0';
}
return res;
}
int printer_cb(void *rock __attribute__((unused)),
const char *key, int keylen,
const char *data, int datalen)
{
fwrite(key, sizeof(char), keylen, stdout);
fputc('\t', stdout);
fwrite(data, sizeof(char), datalen, stdout);
fputc('\n', stdout);
return 0;
}
int main(int argc, char *argv[])
{
const char *old_db;
const char *action;
char *key;
char *value;
int i,r,keylen,vallen,reslen;
int opt,loop;
char *alt_config = NULL;
const char *res = NULL;
int is_get = 0;
int is_set = 0;
int is_delete = 0;
int use_stdin = 0;
int db_flags = 0;
while ((opt = getopt(argc, argv, "C:n")) != EOF) {
switch (opt) {
case 'C':
alt_config = optarg;
break;
case 'n':
db_flags |= CYRUSDB_CREATE;
break;
}
}
if((argc - optind) < 3) {
fprintf(stderr, "Usage: %s [-C altconfig] <old db> <old db backend> <action> [<key>] [<value>]\n", argv[0]);
fprintf(stderr, "Usable Backends: ");
if(!cyrusdb_backends || !cyrusdb_backends[0])
fatal("we don't seem to have any db backends available", EC_OSERR);
fprintf(stderr, "%s", cyrusdb_backends[0]->name);
for(i=1; cyrusdb_backends[i]; i++)
fprintf(stderr, ", %s", cyrusdb_backends[i]->name);
fprintf(stderr, "\n");
fprintf(stderr, "\n");
fprintf(stderr, "Actions:\n");
fprintf(stderr, "* show [<prefix>]\n");
fprintf(stderr, "* get <key>\n");
fprintf(stderr, "* set <key> <value>\n");
fprintf(stderr, "* delete <key>\n");
fprintf(stderr, "You may omit key or key/value and specify one per line on stdin\n");
fprintf(stderr, "keys are terminated by tab or newline, values are terminated by newline\n");
exit(-1);
}
old_db = argv[optind];
action = argv[optind+2];
if(old_db[0] != '/') {
printf("\nSorry, you cannot use this tool with relative path names.\n"
"This is because some database backends (mainly berkeley) do not\n"
"always do what you would expect with them.\n"
"\nPlease use absolute pathnames instead.\n\n");
exit(EC_OSERR);
}
for(i=0; cyrusdb_backends[i]; i++) {
if(!strcmp(cyrusdb_backends[i]->name, argv[optind+1])) {
DB_OLD = cyrusdb_backends[i]; break;
}
}
if(!cyrusdb_backends[i]) {
fatal("unknown backend", EC_TEMPFAIL);
}
cyrus_init(alt_config, "cyr_dbtool", 0);
r = DB_OLD->open(old_db, db_flags, &odb);
if(r != CYRUSDB_OK)
fatal("can't open database", EC_TEMPFAIL);
if (( is_get = !strcmp(action, "get")) ||
(is_delete = !strcmp(action, "delete")) ||
(is_set = !strcmp(action, "set")) ) {
use_stdin = ( (argc - optind) < 4 );
if (use_stdin) {
loop = read_key_value( &key, &keylen, &value, &vallen );
} else {
key = argv[optind+3];
keylen = strlen(key);
if (is_set) {
value = argv[optind+4];
vallen = strlen(value);
}
loop = 1;
}
while ( loop ) {
if (is_get) {
int i;
DB_OLD->fetch(odb, key, keylen, &res, &reslen, NULL);
fwrite(res, sizeof(char), reslen, stdout);
printf("\n");
} else if (is_set) {
DB_OLD->store(odb, key, keylen, value, vallen, NULL);
} else if (is_delete) {
DB_OLD->delete(odb, key, keylen, NULL, 1);
}
loop = 0;
if ( use_stdin ) {
loop = read_key_value( &key, &keylen, &value, &vallen );
}
}
} else if (!strcmp(action, "show")) {
if ((argc - optind) < 4) {
DB_OLD->foreach(odb, "", 0, NULL, printer_cb, NULL, NULL);
} else {
key = argv[optind+3];
keylen = strlen(key);
DB_OLD->foreach(odb, key, keylen, NULL, printer_cb, NULL, NULL);
}
} else {
printf("Unknown action %s\n", action);
}
DB_OLD->close(odb);
cyrus_done();
return 0;
}