mirror of
https://github.com/MariaDB/server.git
synced 2025-01-22 14:54:20 +01:00
118 lines
2.8 KiB
C++
118 lines
2.8 KiB
C++
#include <stdlib.h>
|
|
#include <assert.h>
|
|
#include <db_cxx.h>
|
|
#include <memory.h>
|
|
|
|
static inline void hexdump(Dbt *d) {
|
|
unsigned char *cp = (unsigned char *) d->get_data();
|
|
int n = d->get_size();
|
|
printf(" ");
|
|
for (int i=0; i<n; i++)
|
|
printf("%2.2x", cp[i]);
|
|
printf("\n");
|
|
}
|
|
|
|
static void hexput(Dbt *d, int c) {
|
|
unsigned char *cp = (unsigned char *) d->get_data();
|
|
int n = d->get_size();
|
|
int ulen = d->get_ulen();
|
|
if (n+1 >= ulen) {
|
|
int newulen = ulen == 0 ? 1 : ulen*2;
|
|
cp = (unsigned char *) toku_realloc(cp, newulen); assert(cp);
|
|
d->set_data(cp);
|
|
d->set_ulen(newulen);
|
|
}
|
|
cp[n++] = (unsigned char)c;
|
|
d->set_size(n);
|
|
}
|
|
|
|
static int hextrans(int c) {
|
|
if ('0' <= c && c <= '9') return c - '0';
|
|
if ('a' <= c && c <= 'f') return c - 'a' + 10;
|
|
if ('A' <= c && c <= 'F') return c - 'A' + 10;
|
|
return 0;
|
|
}
|
|
|
|
static int hexload(Dbt *d) {
|
|
d->set_size(0);
|
|
int c = getchar();
|
|
if (c == EOF || c != ' ') return 0;
|
|
for (;;) {
|
|
int c0 = getchar();
|
|
if (c0 == EOF) return 0;
|
|
if (c0 == '\n') break;
|
|
int c1 = getchar();
|
|
if (c1 == EOF) return 0;
|
|
if (c1 == '\n') break;
|
|
hexput(d, (hextrans(c0) << 4) + hextrans(c1));
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int dbload(char *dbfile, char *dbname) {
|
|
int r;
|
|
|
|
#if defined(USE_ENV) && USE_ENV
|
|
DbEnv env(DB_CXX_NO_EXCEPTIONS);
|
|
r = env.open(".", DB_INIT_MPOOL + DB_CREATE + DB_PRIVATE, 0777); assert(r == 0);
|
|
Db db(&env, DB_CXX_NO_EXCEPTIONS);
|
|
#else
|
|
Db db(0, DB_CXX_NO_EXCEPTIONS);
|
|
#endif
|
|
r = db.open(0, dbfile, dbname, DB_BTREE, DB_CREATE, 0777);
|
|
if (r != 0) {
|
|
printf("cant open %s:%s\n", dbfile, dbname);
|
|
#if defined(USE_ENV) && USE_ENV
|
|
r = env.close(0); assert(r == 0);
|
|
#endif
|
|
return 1;
|
|
}
|
|
|
|
Dbt key, val;
|
|
for (;;) {
|
|
if (!hexload(&key)) break;
|
|
// hexdump(&key);
|
|
if (!hexload(&val)) break;
|
|
// hexdump(&val);
|
|
r = db.put(0, &key, &val, 0);
|
|
assert(r == 0);
|
|
}
|
|
if (key.get_data()) toku_free(key.get_data());
|
|
if (val.get_data()) toku_free(val.get_data());
|
|
|
|
r = db.close(0); assert(r == 0);
|
|
#if defined(USE_ENV) && USE_ENV
|
|
r = env.close(0); assert(r == 0);
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
static int usage() {
|
|
printf("db_load [-s DBNAME] DBFILE\n");
|
|
return 1;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
int i;
|
|
|
|
char *dbname = 0;
|
|
for (i=1; i<argc; i++) {
|
|
char *arg = argv[i];
|
|
if (0 == strcmp(arg, "-h") || 0 == strcmp(arg, "--help"))
|
|
return usage();
|
|
if (0 == strcmp(arg, "-s")) {
|
|
i++;
|
|
if (i >= argc)
|
|
return usage();
|
|
dbname = argv[i];
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (i >= argc)
|
|
return usage();
|
|
return dbload(argv[i], dbname);
|
|
}
|
|
|