@ohos-ports/lmdb 2.5.3 → 2.8.5-beta.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (124) hide show
  1. package/LICENSE +20 -20
  2. package/README.md +72 -28
  3. package/SECURITY.md +12 -12
  4. package/bin/download-prebuilds.js +15 -0
  5. package/binding.gyp +115 -111
  6. package/build/Release/lmdb.node +0 -0
  7. package/caching.js +212 -156
  8. package/dependencies/lmdb/libraries/liblmdb/COPYRIGHT +20 -20
  9. package/dependencies/lmdb/libraries/liblmdb/Doxyfile +1631 -1631
  10. package/dependencies/lmdb/libraries/liblmdb/LICENSE +47 -47
  11. package/dependencies/lmdb/libraries/liblmdb/Makefile +136 -136
  12. package/dependencies/lmdb/libraries/liblmdb/chacha8.c +183 -183
  13. package/dependencies/lmdb/libraries/liblmdb/chacha8.h +14 -14
  14. package/dependencies/lmdb/libraries/liblmdb/crypto.c +121 -121
  15. package/dependencies/lmdb/libraries/liblmdb/intro.doc +192 -192
  16. package/dependencies/lmdb/libraries/liblmdb/lmdb.h +28 -3
  17. package/dependencies/lmdb/libraries/liblmdb/mdb.c +157 -31
  18. package/dependencies/lmdb/libraries/liblmdb/mdb_copy.1 +74 -74
  19. package/dependencies/lmdb/libraries/liblmdb/mdb_copy.c +106 -106
  20. package/dependencies/lmdb/libraries/liblmdb/mdb_drop.1 +53 -53
  21. package/dependencies/lmdb/libraries/liblmdb/mdb_drop.c +154 -154
  22. package/dependencies/lmdb/libraries/liblmdb/mdb_dump.1 +94 -94
  23. package/dependencies/lmdb/libraries/liblmdb/mdb_dump.c +333 -333
  24. package/dependencies/lmdb/libraries/liblmdb/mdb_load.1 +97 -97
  25. package/dependencies/lmdb/libraries/liblmdb/mdb_load.c +530 -530
  26. package/dependencies/lmdb/libraries/liblmdb/mdb_stat.1 +83 -83
  27. package/dependencies/lmdb/libraries/liblmdb/mdb_stat.c +276 -276
  28. package/dependencies/lmdb/libraries/liblmdb/midl-sparse.c +452 -0
  29. package/dependencies/lmdb/libraries/liblmdb/midl.c +452 -452
  30. package/dependencies/lmdb/libraries/liblmdb/midl.h +208 -208
  31. package/dependencies/lmdb/libraries/liblmdb/module.c +101 -101
  32. package/dependencies/lmdb/libraries/liblmdb/module.h +16 -16
  33. package/dependencies/lmdb/libraries/liblmdb/mtest.c +178 -178
  34. package/dependencies/lmdb/libraries/liblmdb/mtest2.c +124 -124
  35. package/dependencies/lmdb/libraries/liblmdb/mtest3.c +133 -133
  36. package/dependencies/lmdb/libraries/liblmdb/mtest4.c +168 -168
  37. package/dependencies/lmdb/libraries/liblmdb/mtest5.c +135 -135
  38. package/dependencies/lmdb/libraries/liblmdb/mtest6.c +141 -141
  39. package/dependencies/lmdb/libraries/liblmdb/mtest_enc.c +190 -190
  40. package/dependencies/lmdb/libraries/liblmdb/mtest_enc2.c +189 -189
  41. package/dependencies/lmdb/libraries/liblmdb/mtest_remap.c +177 -177
  42. package/dependencies/lmdb/libraries/liblmdb/sample-bdb.txt +73 -73
  43. package/dependencies/lmdb/libraries/liblmdb/sample-mdb.txt +62 -62
  44. package/dependencies/lmdb/libraries/liblmdb/tooltag +27 -27
  45. package/dependencies/lmdb-data-v1/libraries/liblmdb/CHANGES +266 -0
  46. package/dependencies/lmdb-data-v1/libraries/liblmdb/COPYRIGHT +20 -20
  47. package/dependencies/lmdb-data-v1/libraries/liblmdb/Doxyfile +1631 -1631
  48. package/dependencies/lmdb-data-v1/libraries/liblmdb/LICENSE +47 -47
  49. package/dependencies/lmdb-data-v1/libraries/liblmdb/intro.doc +192 -192
  50. package/dependencies/lmdb-data-v1/libraries/liblmdb/lmdb.h +1608 -1653
  51. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb.c +10320 -11359
  52. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_copy.1 +55 -61
  53. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_copy.c +82 -84
  54. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_dump.1 +75 -81
  55. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_dump.c +319 -319
  56. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_load.1 +84 -84
  57. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_load.c +496 -492
  58. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_stat.1 +64 -70
  59. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_stat.c +263 -264
  60. package/dependencies/lmdb-data-v1/libraries/liblmdb/midl.c +359 -421
  61. package/dependencies/lmdb-data-v1/libraries/liblmdb/midl.h +186 -200
  62. package/dependencies/lmdb-data-v1/libraries/liblmdb/mtest.c +177 -177
  63. package/dependencies/lmdb-data-v1/libraries/liblmdb/mtest2.c +124 -124
  64. package/dependencies/lmdb-data-v1/libraries/liblmdb/mtest3.c +133 -133
  65. package/dependencies/lmdb-data-v1/libraries/liblmdb/mtest4.c +168 -168
  66. package/dependencies/lmdb-data-v1/libraries/liblmdb/mtest5.c +135 -135
  67. package/dependencies/lmdb-data-v1/libraries/liblmdb/mtest6.c +141 -141
  68. package/dependencies/lmdb-data-v1/libraries/liblmdb/sample-bdb.txt +73 -73
  69. package/dependencies/lmdb-data-v1/libraries/liblmdb/sample-mdb.txt +62 -62
  70. package/dependencies/lmdb-data-v1/libraries/liblmdb/tooltag +22 -27
  71. package/dependencies/lz4/LICENSE +11 -11
  72. package/dependencies/lz4/lib/LICENSE +24 -24
  73. package/dependencies/lz4/lib/Makefile +225 -0
  74. package/dependencies/lz4/lib/README.md +169 -137
  75. package/dependencies/lz4/lib/dll/example/Makefile +63 -0
  76. package/dependencies/lz4/lib/dll/example/README.md +69 -69
  77. package/dependencies/lz4/lib/dll/liblz4.def +62 -62
  78. package/dependencies/lz4/lib/liblz4-dll.rc.in +35 -35
  79. package/dependencies/lz4/lib/liblz4.pc.in +14 -14
  80. package/dependencies/lz4/lib/lz4.c +2722 -2495
  81. package/dependencies/lz4/lib/lz4.h +842 -774
  82. package/dependencies/lz4/lib/lz4file.c +311 -0
  83. package/dependencies/lz4/lib/lz4file.h +93 -0
  84. package/dependencies/lz4/lib/lz4frame.c +2078 -1899
  85. package/dependencies/lz4/lib/lz4frame.h +692 -623
  86. package/dependencies/lz4/lib/lz4frame_static.h +47 -47
  87. package/dependencies/lz4/lib/lz4hc.c +1631 -1615
  88. package/dependencies/lz4/lib/lz4hc.h +413 -413
  89. package/dependencies/lz4/lib/xxhash.c +1030 -1030
  90. package/dependencies/lz4/lib/xxhash.h +328 -328
  91. package/dependencies/v8/v8-fast-api-calls-v16.h +790 -790
  92. package/dependencies/v8/v8-fast-api-calls.h +784 -888
  93. package/dist/index.cjs +1050 -584
  94. package/dist/index.cjs.map +1 -1
  95. package/index.d.cts +420 -0
  96. package/index.d.ts +450 -385
  97. package/index.js +38 -37
  98. package/keys.js +116 -102
  99. package/level.js +26 -26
  100. package/native.js +8 -5
  101. package/node-index.js +21 -17
  102. package/open.js +51 -17
  103. package/package.json +63 -21
  104. package/read.js +338 -95
  105. package/src/compression.cpp +12 -9
  106. package/src/cursor.cpp +82 -31
  107. package/src/dbi.cpp +78 -40
  108. package/src/env.cpp +336 -68
  109. package/src/lmdb-js.cpp +64 -64
  110. package/src/lmdb-js.h +83 -49
  111. package/src/misc.cpp +297 -28
  112. package/src/ordered-binary.cpp +46 -46
  113. package/src/txn.cpp +210 -257
  114. package/src/v8-functions.cpp +170 -166
  115. package/src/windows.c +34 -34
  116. package/src/writer.cpp +169 -84
  117. package/util/RangeIterable.js +112 -17
  118. package/util/set-optional-deps.cjs +14 -10
  119. package/util/when.js +8 -8
  120. package/write.js +145 -121
  121. package/dependencies/lmdb-data-v1/libraries/liblmdb/Makefile +0 -118
  122. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_drop.1 +0 -40
  123. package/dependencies/lmdb-data-v1/libraries/liblmdb/mdb_drop.c +0 -135
  124. package/rollup.config.js +0 -12
@@ -1,492 +1,496 @@
1
- /* mdb_load.c - memory-mapped database load tool */
2
- /*
3
- * Copyright 2011-2020 Howard Chu, Symas Corp.
4
- * All rights reserved.
5
- *
6
- * Redistribution and use in source and binary forms, with or without
7
- * modification, are permitted only as authorized by the OpenLDAP
8
- * Public License.
9
- *
10
- * A copy of this license is available in the file LICENSE in the
11
- * top-level directory of the distribution or, alternatively, at
12
- * <http://www.OpenLDAP.org/license.html>.
13
- */
14
- #include <stdio.h>
15
- #include <stdlib.h>
16
- #include <errno.h>
17
- #include <string.h>
18
- #include <ctype.h>
19
- #include <unistd.h>
20
- #include "lmdb.h"
21
-
22
- #define PRINT 1
23
- #define NOHDR 2
24
- static int mode;
25
-
26
- static char *subname = NULL;
27
-
28
- static mdb_size_t lineno;
29
- static int version;
30
-
31
- static int flags;
32
-
33
- static char *prog;
34
-
35
- static int Eof;
36
-
37
- static MDB_envinfo info;
38
-
39
- static MDB_val kbuf, dbuf;
40
- static MDB_val k0buf;
41
-
42
- #define Yu MDB_PRIy(u)
43
-
44
- #define STRLENOF(s) (sizeof(s)-1)
45
-
46
- typedef struct flagbit {
47
- int bit;
48
- char *name;
49
- int len;
50
- } flagbit;
51
-
52
- #define S(s) s, STRLENOF(s)
53
-
54
- flagbit dbflags[] = {
55
- { MDB_REVERSEKEY, S("reversekey") },
56
- { MDB_DUPSORT, S("dupsort") },
57
- { MDB_INTEGERKEY, S("integerkey") },
58
- { MDB_DUPFIXED, S("dupfixed") },
59
- { MDB_INTEGERDUP, S("integerdup") },
60
- { MDB_REVERSEDUP, S("reversedup") },
61
- { 0, NULL, 0 }
62
- };
63
-
64
- static void readhdr(void)
65
- {
66
- char *ptr;
67
-
68
- flags = 0;
69
- while (fgets(dbuf.mv_data, dbuf.mv_size, stdin) != NULL) {
70
- lineno++;
71
- if (!strncmp(dbuf.mv_data, "VERSION=", STRLENOF("VERSION="))) {
72
- version=atoi((char *)dbuf.mv_data+STRLENOF("VERSION="));
73
- if (version > 3) {
74
- fprintf(stderr, "%s: line %"Yu": unsupported VERSION %d\n",
75
- prog, lineno, version);
76
- exit(EXIT_FAILURE);
77
- }
78
- } else if (!strncmp(dbuf.mv_data, "HEADER=END", STRLENOF("HEADER=END"))) {
79
- break;
80
- } else if (!strncmp(dbuf.mv_data, "format=", STRLENOF("format="))) {
81
- if (!strncmp((char *)dbuf.mv_data+STRLENOF("FORMAT="), "print", STRLENOF("print")))
82
- mode |= PRINT;
83
- else if (strncmp((char *)dbuf.mv_data+STRLENOF("FORMAT="), "bytevalue", STRLENOF("bytevalue"))) {
84
- fprintf(stderr, "%s: line %"Yu": unsupported FORMAT %s\n",
85
- prog, lineno, (char *)dbuf.mv_data+STRLENOF("FORMAT="));
86
- exit(EXIT_FAILURE);
87
- }
88
- } else if (!strncmp(dbuf.mv_data, "database=", STRLENOF("database="))) {
89
- ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
90
- if (ptr) *ptr = '\0';
91
- if (subname) free(subname);
92
- subname = strdup((char *)dbuf.mv_data+STRLENOF("database="));
93
- } else if (!strncmp(dbuf.mv_data, "type=", STRLENOF("type="))) {
94
- if (strncmp((char *)dbuf.mv_data+STRLENOF("type="), "btree", STRLENOF("btree"))) {
95
- fprintf(stderr, "%s: line %"Yu": unsupported type %s\n",
96
- prog, lineno, (char *)dbuf.mv_data+STRLENOF("type="));
97
- exit(EXIT_FAILURE);
98
- }
99
- } else if (!strncmp(dbuf.mv_data, "mapaddr=", STRLENOF("mapaddr="))) {
100
- int i;
101
- ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
102
- if (ptr) *ptr = '\0';
103
- i = sscanf((char *)dbuf.mv_data+STRLENOF("mapaddr="), "%p", &info.me_mapaddr);
104
- if (i != 1) {
105
- fprintf(stderr, "%s: line %"Yu": invalid mapaddr %s\n",
106
- prog, lineno, (char *)dbuf.mv_data+STRLENOF("mapaddr="));
107
- exit(EXIT_FAILURE);
108
- }
109
- } else if (!strncmp(dbuf.mv_data, "mapsize=", STRLENOF("mapsize="))) {
110
- int i;
111
- ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
112
- if (ptr) *ptr = '\0';
113
- i = sscanf((char *)dbuf.mv_data+STRLENOF("mapsize="),
114
- "%" MDB_SCNy(u), &info.me_mapsize);
115
- if (i != 1) {
116
- fprintf(stderr, "%s: line %"Yu": invalid mapsize %s\n",
117
- prog, lineno, (char *)dbuf.mv_data+STRLENOF("mapsize="));
118
- exit(EXIT_FAILURE);
119
- }
120
- } else if (!strncmp(dbuf.mv_data, "maxreaders=", STRLENOF("maxreaders="))) {
121
- int i;
122
- ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
123
- if (ptr) *ptr = '\0';
124
- i = sscanf((char *)dbuf.mv_data+STRLENOF("maxreaders="), "%u", &info.me_maxreaders);
125
- if (i != 1) {
126
- fprintf(stderr, "%s: line %"Yu": invalid maxreaders %s\n",
127
- prog, lineno, (char *)dbuf.mv_data+STRLENOF("maxreaders="));
128
- exit(EXIT_FAILURE);
129
- }
130
- } else {
131
- int i;
132
- for (i=0; dbflags[i].bit; i++) {
133
- if (!strncmp(dbuf.mv_data, dbflags[i].name, dbflags[i].len) &&
134
- ((char *)dbuf.mv_data)[dbflags[i].len] == '=') {
135
- flags |= dbflags[i].bit;
136
- break;
137
- }
138
- }
139
- if (!dbflags[i].bit) {
140
- ptr = memchr(dbuf.mv_data, '=', dbuf.mv_size);
141
- if (!ptr) {
142
- fprintf(stderr, "%s: line %"Yu": unexpected format\n",
143
- prog, lineno);
144
- exit(EXIT_FAILURE);
145
- } else {
146
- *ptr = '\0';
147
- fprintf(stderr, "%s: line %"Yu": unrecognized keyword ignored: %s\n",
148
- prog, lineno, (char *)dbuf.mv_data);
149
- }
150
- }
151
- }
152
- }
153
- }
154
-
155
- static void badend(void)
156
- {
157
- fprintf(stderr, "%s: line %"Yu": unexpected end of input\n",
158
- prog, lineno);
159
- }
160
-
161
- static int unhex(unsigned char *c2)
162
- {
163
- int x, c;
164
- x = *c2++ & 0x4f;
165
- if (x & 0x40)
166
- x -= 55;
167
- c = x << 4;
168
- x = *c2 & 0x4f;
169
- if (x & 0x40)
170
- x -= 55;
171
- c |= x;
172
- return c;
173
- }
174
-
175
- static int readline(MDB_val *out, MDB_val *buf)
176
- {
177
- unsigned char *c1, *c2, *end;
178
- size_t len, l2;
179
- int c;
180
-
181
- if (!(mode & NOHDR)) {
182
- c = fgetc(stdin);
183
- if (c == EOF) {
184
- Eof = 1;
185
- return EOF;
186
- }
187
- if (c != ' ') {
188
- lineno++;
189
- if (fgets(buf->mv_data, buf->mv_size, stdin) == NULL) {
190
- badend:
191
- Eof = 1;
192
- badend();
193
- return EOF;
194
- }
195
- if (c == 'D' && !strncmp(buf->mv_data, "ATA=END", STRLENOF("ATA=END")))
196
- return EOF;
197
- goto badend;
198
- }
199
- }
200
- if (fgets(buf->mv_data, buf->mv_size, stdin) == NULL) {
201
- Eof = 1;
202
- return EOF;
203
- }
204
- lineno++;
205
-
206
- c1 = buf->mv_data;
207
- len = strlen((char *)c1);
208
- l2 = len;
209
-
210
- /* Is buffer too short? */
211
- while (c1[len-1] != '\n') {
212
- buf->mv_data = realloc(buf->mv_data, buf->mv_size*2);
213
- if (!buf->mv_data) {
214
- Eof = 1;
215
- fprintf(stderr, "%s: line %"Yu": out of memory, line too long\n",
216
- prog, lineno);
217
- return EOF;
218
- }
219
- c1 = buf->mv_data;
220
- c1 += l2;
221
- if (fgets((char *)c1, buf->mv_size+1, stdin) == NULL) {
222
- Eof = 1;
223
- badend();
224
- return EOF;
225
- }
226
- buf->mv_size *= 2;
227
- len = strlen((char *)c1);
228
- l2 += len;
229
- }
230
- c1 = c2 = buf->mv_data;
231
- len = l2;
232
- c1[--len] = '\0';
233
- end = c1 + len;
234
-
235
- if (mode & PRINT) {
236
- while (c2 < end) {
237
- if (*c2 == '\\') {
238
- if (c2[1] == '\\') {
239
- *c1++ = *c2;
240
- } else {
241
- if (c2+3 > end || !isxdigit(c2[1]) || !isxdigit(c2[2])) {
242
- Eof = 1;
243
- badend();
244
- return EOF;
245
- }
246
- *c1++ = unhex(++c2);
247
- }
248
- c2 += 2;
249
- } else {
250
- /* copies are redundant when no escapes were used */
251
- *c1++ = *c2++;
252
- }
253
- }
254
- } else {
255
- /* odd length not allowed */
256
- if (len & 1) {
257
- Eof = 1;
258
- badend();
259
- return EOF;
260
- }
261
- while (c2 < end) {
262
- if (!isxdigit(*c2) || !isxdigit(c2[1])) {
263
- Eof = 1;
264
- badend();
265
- return EOF;
266
- }
267
- *c1++ = unhex(c2);
268
- c2 += 2;
269
- }
270
- }
271
- c2 = out->mv_data = buf->mv_data;
272
- out->mv_size = c1 - c2;
273
-
274
- return 0;
275
- }
276
-
277
- static void usage(void)
278
- {
279
- fprintf(stderr, "usage: %s [-V] [-a] [-f input] [-n] [-s name] [-N] [-T] dbpath\n", prog);
280
- exit(EXIT_FAILURE);
281
- }
282
-
283
- static int greater(const MDB_val *a, const MDB_val *b)
284
- {
285
- return 1;
286
- }
287
-
288
- int main(int argc, char *argv[])
289
- {
290
- int i, rc;
291
- MDB_env *env;
292
- MDB_txn *txn;
293
- MDB_cursor *mc;
294
- MDB_dbi dbi;
295
- char *envname;
296
- int envflags = MDB_NOSYNC, putflags = 0;
297
- int dohdr = 0, append = 0;
298
- MDB_val prevk;
299
-
300
- prog = argv[0];
301
-
302
- if (argc < 2) {
303
- usage();
304
- }
305
-
306
- /* -a: append records in input order
307
- * -f: load file instead of stdin
308
- * -n: use NOSUBDIR flag on env_open
309
- * -s: load into named subDB
310
- * -N: use NOOVERWRITE on puts
311
- * -T: read plaintext
312
- * -V: print version and exit
313
- */
314
- while ((i = getopt(argc, argv, "af:ns:NTV")) != EOF) {
315
- switch(i) {
316
- case 'V':
317
- printf("%s\n", MDB_VERSION_STRING);
318
- exit(0);
319
- break;
320
- case 'a':
321
- append = 1;
322
- break;
323
- case 'f':
324
- if (freopen(optarg, "r", stdin) == NULL) {
325
- fprintf(stderr, "%s: %s: reopen: %s\n",
326
- prog, optarg, strerror(errno));
327
- exit(EXIT_FAILURE);
328
- }
329
- break;
330
- case 'n':
331
- envflags |= MDB_NOSUBDIR;
332
- break;
333
- case 's':
334
- subname = strdup(optarg);
335
- break;
336
- case 'N':
337
- putflags = MDB_NOOVERWRITE|MDB_NODUPDATA;
338
- break;
339
- case 'T':
340
- mode |= NOHDR | PRINT;
341
- break;
342
- default:
343
- usage();
344
- }
345
- }
346
-
347
- if (optind != argc - 1)
348
- usage();
349
-
350
- dbuf.mv_size = 4096;
351
- dbuf.mv_data = malloc(dbuf.mv_size);
352
-
353
- if (!(mode & NOHDR))
354
- readhdr();
355
-
356
- envname = argv[optind];
357
- rc = mdb_env_create(&env);
358
- if (rc) {
359
- fprintf(stderr, "mdb_env_create failed, error %d %s\n", rc, mdb_strerror(rc));
360
- return EXIT_FAILURE;
361
- }
362
-
363
- mdb_env_set_maxdbs(env, 2);
364
-
365
- if (info.me_maxreaders)
366
- mdb_env_set_maxreaders(env, info.me_maxreaders);
367
-
368
- if (info.me_mapsize)
369
- mdb_env_set_mapsize(env, info.me_mapsize);
370
-
371
- if (info.me_mapaddr)
372
- envflags |= MDB_FIXEDMAP;
373
-
374
- rc = mdb_env_open(env, envname, envflags, 0664);
375
- if (rc) {
376
- fprintf(stderr, "mdb_env_open failed, error %d %s\n", rc, mdb_strerror(rc));
377
- goto env_close;
378
- }
379
-
380
- kbuf.mv_size = mdb_env_get_maxkeysize(env) * 2 + 2;
381
- kbuf.mv_data = malloc(kbuf.mv_size * 2);
382
- k0buf.mv_size = kbuf.mv_size;
383
- k0buf.mv_data = (char *)kbuf.mv_data + kbuf.mv_size;
384
- prevk.mv_data = k0buf.mv_data;
385
-
386
- while(!Eof) {
387
- MDB_val key, data;
388
- int batch = 0;
389
- int appflag;
390
-
391
- if (!dohdr) {
392
- dohdr = 1;
393
- } else if (!(mode & NOHDR))
394
- readhdr();
395
-
396
- rc = mdb_txn_begin(env, NULL, 0, &txn);
397
- if (rc) {
398
- fprintf(stderr, "mdb_txn_begin failed, error %d %s\n", rc, mdb_strerror(rc));
399
- goto env_close;
400
- }
401
-
402
- rc = mdb_open(txn, subname, flags|MDB_CREATE, &dbi);
403
- if (rc) {
404
- fprintf(stderr, "mdb_open failed, error %d %s\n", rc, mdb_strerror(rc));
405
- goto txn_abort;
406
- }
407
- prevk.mv_size = 0;
408
- if (append) {
409
- mdb_set_compare(txn, dbi, greater);
410
- if (flags & MDB_DUPSORT)
411
- mdb_set_dupsort(txn, dbi, greater);
412
- }
413
-
414
- rc = mdb_cursor_open(txn, dbi, &mc);
415
- if (rc) {
416
- fprintf(stderr, "mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
417
- goto txn_abort;
418
- }
419
-
420
- while(1) {
421
- rc = readline(&key, &kbuf);
422
- if (rc) /* rc == EOF */
423
- break;
424
-
425
- rc = readline(&data, &dbuf);
426
- if (rc) {
427
- fprintf(stderr, "%s: line %"Yu": failed to read key value\n", prog, lineno);
428
- goto txn_abort;
429
- }
430
-
431
- if (append) {
432
- appflag = MDB_APPEND;
433
- if (flags & MDB_DUPSORT) {
434
- if (prevk.mv_size == key.mv_size && !memcmp(prevk.mv_data, key.mv_data, key.mv_size))
435
- appflag = MDB_CURRENT|MDB_APPENDDUP;
436
- else {
437
- memcpy(prevk.mv_data, key.mv_data, key.mv_size);
438
- prevk.mv_size = key.mv_size;
439
- }
440
- }
441
- } else {
442
- appflag = 0;
443
- }
444
- rc = mdb_cursor_put(mc, &key, &data, putflags|appflag);
445
- if (rc == MDB_KEYEXIST && putflags)
446
- continue;
447
- if (rc) {
448
- fprintf(stderr, "mdb_cursor_put failed, error %d %s\n", rc, mdb_strerror(rc));
449
- goto txn_abort;
450
- }
451
- batch++;
452
- if (batch == 100) {
453
- rc = mdb_txn_commit(txn);
454
- if (rc) {
455
- fprintf(stderr, "%s: line %"Yu": txn_commit: %s\n",
456
- prog, lineno, mdb_strerror(rc));
457
- goto env_close;
458
- }
459
- rc = mdb_txn_begin(env, NULL, 0, &txn);
460
- if (rc) {
461
- fprintf(stderr, "mdb_txn_begin failed, error %d %s\n", rc, mdb_strerror(rc));
462
- goto env_close;
463
- }
464
- rc = mdb_cursor_open(txn, dbi, &mc);
465
- if (rc) {
466
- fprintf(stderr, "mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
467
- goto txn_abort;
468
- }
469
- if (appflag & MDB_APPENDDUP) {
470
- MDB_val k, d;
471
- mdb_cursor_get(mc, &k, &d, MDB_LAST);
472
- }
473
- batch = 0;
474
- }
475
- }
476
- rc = mdb_txn_commit(txn);
477
- txn = NULL;
478
- if (rc) {
479
- fprintf(stderr, "%s: line %"Yu": txn_commit: %s\n",
480
- prog, lineno, mdb_strerror(rc));
481
- goto env_close;
482
- }
483
- mdb_dbi_close(env, dbi);
484
- }
485
-
486
- txn_abort:
487
- mdb_txn_abort(txn);
488
- env_close:
489
- mdb_env_close(env);
490
-
491
- return rc ? EXIT_FAILURE : EXIT_SUCCESS;
492
- }
1
+ /* mdb_load.c - memory-mapped database load tool */
2
+ /*
3
+ * Copyright 2011-2021 Howard Chu, Symas Corp.
4
+ * All rights reserved.
5
+ *
6
+ * Redistribution and use in source and binary forms, with or without
7
+ * modification, are permitted only as authorized by the OpenLDAP
8
+ * Public License.
9
+ *
10
+ * A copy of this license is available in the file LICENSE in the
11
+ * top-level directory of the distribution or, alternatively, at
12
+ * <http://www.OpenLDAP.org/license.html>.
13
+ */
14
+ #include <stdio.h>
15
+ #include <stdlib.h>
16
+ #include <errno.h>
17
+ #include <string.h>
18
+ #include <ctype.h>
19
+ #include <unistd.h>
20
+ #include "lmdb.h"
21
+
22
+ #define PRINT 1
23
+ #define NOHDR 2
24
+ static int mode;
25
+
26
+ static char *subname = NULL;
27
+
28
+ static size_t lineno;
29
+ static int version;
30
+
31
+ static int flags;
32
+
33
+ static char *prog;
34
+
35
+ static int Eof;
36
+
37
+ static MDB_envinfo info;
38
+
39
+ static MDB_val kbuf, dbuf;
40
+ static MDB_val k0buf;
41
+
42
+ #ifdef _WIN32
43
+ #define Z "I"
44
+ #else
45
+ #define Z "z"
46
+ #endif
47
+
48
+ #define STRLENOF(s) (sizeof(s)-1)
49
+
50
+ typedef struct flagbit {
51
+ int bit;
52
+ char *name;
53
+ int len;
54
+ } flagbit;
55
+
56
+ #define S(s) s, STRLENOF(s)
57
+
58
+ flagbit dbflags[] = {
59
+ { MDB_REVERSEKEY, S("reversekey") },
60
+ { MDB_DUPSORT, S("dupsort") },
61
+ { MDB_INTEGERKEY, S("integerkey") },
62
+ { MDB_DUPFIXED, S("dupfixed") },
63
+ { MDB_INTEGERDUP, S("integerdup") },
64
+ { MDB_REVERSEDUP, S("reversedup") },
65
+ { 0, NULL, 0 }
66
+ };
67
+
68
+ static void readhdr(void)
69
+ {
70
+ char *ptr;
71
+
72
+ flags = 0;
73
+ while (fgets(dbuf.mv_data, dbuf.mv_size, stdin) != NULL) {
74
+ lineno++;
75
+ if (!strncmp(dbuf.mv_data, "VERSION=", STRLENOF("VERSION="))) {
76
+ version=atoi((char *)dbuf.mv_data+STRLENOF("VERSION="));
77
+ if (version > 3) {
78
+ fprintf(stderr, "%s: line %" Z "d: unsupported VERSION %d\n",
79
+ prog, lineno, version);
80
+ exit(EXIT_FAILURE);
81
+ }
82
+ } else if (!strncmp(dbuf.mv_data, "HEADER=END", STRLENOF("HEADER=END"))) {
83
+ break;
84
+ } else if (!strncmp(dbuf.mv_data, "format=", STRLENOF("format="))) {
85
+ if (!strncmp((char *)dbuf.mv_data+STRLENOF("FORMAT="), "print", STRLENOF("print")))
86
+ mode |= PRINT;
87
+ else if (strncmp((char *)dbuf.mv_data+STRLENOF("FORMAT="), "bytevalue", STRLENOF("bytevalue"))) {
88
+ fprintf(stderr, "%s: line %" Z "d: unsupported FORMAT %s\n",
89
+ prog, lineno, (char *)dbuf.mv_data+STRLENOF("FORMAT="));
90
+ exit(EXIT_FAILURE);
91
+ }
92
+ } else if (!strncmp(dbuf.mv_data, "database=", STRLENOF("database="))) {
93
+ ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
94
+ if (ptr) *ptr = '\0';
95
+ if (subname) free(subname);
96
+ subname = strdup((char *)dbuf.mv_data+STRLENOF("database="));
97
+ } else if (!strncmp(dbuf.mv_data, "type=", STRLENOF("type="))) {
98
+ if (strncmp((char *)dbuf.mv_data+STRLENOF("type="), "btree", STRLENOF("btree"))) {
99
+ fprintf(stderr, "%s: line %" Z "d: unsupported type %s\n",
100
+ prog, lineno, (char *)dbuf.mv_data+STRLENOF("type="));
101
+ exit(EXIT_FAILURE);
102
+ }
103
+ } else if (!strncmp(dbuf.mv_data, "mapaddr=", STRLENOF("mapaddr="))) {
104
+ int i;
105
+ ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
106
+ if (ptr) *ptr = '\0';
107
+ i = sscanf((char *)dbuf.mv_data+STRLENOF("mapaddr="), "%p", &info.me_mapaddr);
108
+ if (i != 1) {
109
+ fprintf(stderr, "%s: line %" Z "d: invalid mapaddr %s\n",
110
+ prog, lineno, (char *)dbuf.mv_data+STRLENOF("mapaddr="));
111
+ exit(EXIT_FAILURE);
112
+ }
113
+ } else if (!strncmp(dbuf.mv_data, "mapsize=", STRLENOF("mapsize="))) {
114
+ int i;
115
+ ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
116
+ if (ptr) *ptr = '\0';
117
+ i = sscanf((char *)dbuf.mv_data+STRLENOF("mapsize="), "%" Z "u", &info.me_mapsize);
118
+ if (i != 1) {
119
+ fprintf(stderr, "%s: line %" Z "d: invalid mapsize %s\n",
120
+ prog, lineno, (char *)dbuf.mv_data+STRLENOF("mapsize="));
121
+ exit(EXIT_FAILURE);
122
+ }
123
+ } else if (!strncmp(dbuf.mv_data, "maxreaders=", STRLENOF("maxreaders="))) {
124
+ int i;
125
+ ptr = memchr(dbuf.mv_data, '\n', dbuf.mv_size);
126
+ if (ptr) *ptr = '\0';
127
+ i = sscanf((char *)dbuf.mv_data+STRLENOF("maxreaders="), "%u", &info.me_maxreaders);
128
+ if (i != 1) {
129
+ fprintf(stderr, "%s: line %" Z "d: invalid maxreaders %s\n",
130
+ prog, lineno, (char *)dbuf.mv_data+STRLENOF("maxreaders="));
131
+ exit(EXIT_FAILURE);
132
+ }
133
+ } else {
134
+ int i;
135
+ for (i=0; dbflags[i].bit; i++) {
136
+ if (!strncmp(dbuf.mv_data, dbflags[i].name, dbflags[i].len) &&
137
+ ((char *)dbuf.mv_data)[dbflags[i].len] == '=') {
138
+ flags |= dbflags[i].bit;
139
+ break;
140
+ }
141
+ }
142
+ if (!dbflags[i].bit) {
143
+ ptr = memchr(dbuf.mv_data, '=', dbuf.mv_size);
144
+ if (!ptr) {
145
+ fprintf(stderr, "%s: line %" Z "d: unexpected format\n",
146
+ prog, lineno);
147
+ exit(EXIT_FAILURE);
148
+ } else {
149
+ *ptr = '\0';
150
+ fprintf(stderr, "%s: line %" Z "d: unrecognized keyword ignored: %s\n",
151
+ prog, lineno, (char *)dbuf.mv_data);
152
+ }
153
+ }
154
+ }
155
+ }
156
+ }
157
+
158
+ static void badend(void)
159
+ {
160
+ fprintf(stderr, "%s: line %" Z "d: unexpected end of input\n",
161
+ prog, lineno);
162
+ }
163
+
164
+ static int unhex(unsigned char *c2)
165
+ {
166
+ int x, c;
167
+ x = *c2++ & 0x4f;
168
+ if (x & 0x40)
169
+ x -= 55;
170
+ c = x << 4;
171
+ x = *c2 & 0x4f;
172
+ if (x & 0x40)
173
+ x -= 55;
174
+ c |= x;
175
+ return c;
176
+ }
177
+
178
+ static int readline(MDB_val *out, MDB_val *buf)
179
+ {
180
+ unsigned char *c1, *c2, *end;
181
+ size_t len, l2;
182
+ int c;
183
+
184
+ if (!(mode & NOHDR)) {
185
+ c = fgetc(stdin);
186
+ if (c == EOF) {
187
+ Eof = 1;
188
+ return EOF;
189
+ }
190
+ if (c != ' ') {
191
+ lineno++;
192
+ if (fgets(buf->mv_data, buf->mv_size, stdin) == NULL) {
193
+ badend:
194
+ Eof = 1;
195
+ badend();
196
+ return EOF;
197
+ }
198
+ if (c == 'D' && !strncmp(buf->mv_data, "ATA=END", STRLENOF("ATA=END")))
199
+ return EOF;
200
+ goto badend;
201
+ }
202
+ }
203
+ if (fgets(buf->mv_data, buf->mv_size, stdin) == NULL) {
204
+ Eof = 1;
205
+ return EOF;
206
+ }
207
+ lineno++;
208
+
209
+ c1 = buf->mv_data;
210
+ len = strlen((char *)c1);
211
+ l2 = len;
212
+
213
+ /* Is buffer too short? */
214
+ while (c1[len-1] != '\n') {
215
+ buf->mv_data = realloc(buf->mv_data, buf->mv_size*2);
216
+ if (!buf->mv_data) {
217
+ Eof = 1;
218
+ fprintf(stderr, "%s: line %" Z "d: out of memory, line too long\n",
219
+ prog, lineno);
220
+ return EOF;
221
+ }
222
+ c1 = buf->mv_data;
223
+ c1 += l2;
224
+ if (fgets((char *)c1, buf->mv_size+1, stdin) == NULL) {
225
+ Eof = 1;
226
+ badend();
227
+ return EOF;
228
+ }
229
+ buf->mv_size *= 2;
230
+ len = strlen((char *)c1);
231
+ l2 += len;
232
+ }
233
+ c1 = c2 = buf->mv_data;
234
+ len = l2;
235
+ c1[--len] = '\0';
236
+ end = c1 + len;
237
+
238
+ if (mode & PRINT) {
239
+ while (c2 < end) {
240
+ if (*c2 == '\\') {
241
+ if (c2[1] == '\\') {
242
+ *c1++ = *c2;
243
+ } else {
244
+ if (c2+3 > end || !isxdigit(c2[1]) || !isxdigit(c2[2])) {
245
+ Eof = 1;
246
+ badend();
247
+ return EOF;
248
+ }
249
+ *c1++ = unhex(++c2);
250
+ }
251
+ c2 += 2;
252
+ } else {
253
+ /* copies are redundant when no escapes were used */
254
+ *c1++ = *c2++;
255
+ }
256
+ }
257
+ } else {
258
+ /* odd length not allowed */
259
+ if (len & 1) {
260
+ Eof = 1;
261
+ badend();
262
+ return EOF;
263
+ }
264
+ while (c2 < end) {
265
+ if (!isxdigit(*c2) || !isxdigit(c2[1])) {
266
+ Eof = 1;
267
+ badend();
268
+ return EOF;
269
+ }
270
+ *c1++ = unhex(c2);
271
+ c2 += 2;
272
+ }
273
+ }
274
+ c2 = out->mv_data = buf->mv_data;
275
+ out->mv_size = c1 - c2;
276
+
277
+ return 0;
278
+ }
279
+
280
+ static void usage(void)
281
+ {
282
+ fprintf(stderr, "usage: %s [-V] [-a] [-f input] [-n] [-s name] [-N] [-T] dbpath\n", prog);
283
+ exit(EXIT_FAILURE);
284
+ }
285
+
286
+ static int greater(const MDB_val *a, const MDB_val *b)
287
+ {
288
+ return 1;
289
+ }
290
+
291
+ int main(int argc, char *argv[])
292
+ {
293
+ int i, rc;
294
+ MDB_env *env;
295
+ MDB_txn *txn;
296
+ MDB_cursor *mc;
297
+ MDB_dbi dbi;
298
+ char *envname;
299
+ int envflags = MDB_NOSYNC, putflags = 0;
300
+ int dohdr = 0, append = 0;
301
+ MDB_val prevk;
302
+
303
+ prog = argv[0];
304
+
305
+ if (argc < 2) {
306
+ usage();
307
+ }
308
+
309
+ /* -a: append records in input order
310
+ * -f: load file instead of stdin
311
+ * -n: use NOSUBDIR flag on env_open
312
+ * -s: load into named subDB
313
+ * -N: use NOOVERWRITE on puts
314
+ * -T: read plaintext
315
+ * -V: print version and exit
316
+ */
317
+ while ((i = getopt(argc, argv, "af:ns:NTV")) != EOF) {
318
+ switch(i) {
319
+ case 'V':
320
+ printf("%s\n", MDB_VERSION_STRING);
321
+ exit(0);
322
+ break;
323
+ case 'a':
324
+ append = 1;
325
+ break;
326
+ case 'f':
327
+ if (freopen(optarg, "r", stdin) == NULL) {
328
+ fprintf(stderr, "%s: %s: reopen: %s\n",
329
+ prog, optarg, strerror(errno));
330
+ exit(EXIT_FAILURE);
331
+ }
332
+ break;
333
+ case 'n':
334
+ envflags |= MDB_NOSUBDIR;
335
+ break;
336
+ case 's':
337
+ subname = strdup(optarg);
338
+ break;
339
+ case 'N':
340
+ putflags = MDB_NOOVERWRITE|MDB_NODUPDATA;
341
+ break;
342
+ case 'T':
343
+ mode |= NOHDR | PRINT;
344
+ break;
345
+ default:
346
+ usage();
347
+ }
348
+ }
349
+
350
+ if (optind != argc - 1)
351
+ usage();
352
+
353
+ dbuf.mv_size = 4096;
354
+ dbuf.mv_data = malloc(dbuf.mv_size);
355
+
356
+ if (!(mode & NOHDR))
357
+ readhdr();
358
+
359
+ envname = argv[optind];
360
+ rc = mdb_env_create(&env);
361
+ if (rc) {
362
+ fprintf(stderr, "mdb_env_create failed, error %d %s\n", rc, mdb_strerror(rc));
363
+ return EXIT_FAILURE;
364
+ }
365
+
366
+ mdb_env_set_maxdbs(env, 2);
367
+
368
+ if (info.me_maxreaders)
369
+ mdb_env_set_maxreaders(env, info.me_maxreaders);
370
+
371
+ if (info.me_mapsize)
372
+ mdb_env_set_mapsize(env, info.me_mapsize);
373
+
374
+ if (info.me_mapaddr)
375
+ envflags |= MDB_FIXEDMAP;
376
+
377
+ rc = mdb_env_open(env, envname, envflags, 0664);
378
+ if (rc) {
379
+ fprintf(stderr, "mdb_env_open failed, error %d %s\n", rc, mdb_strerror(rc));
380
+ goto env_close;
381
+ }
382
+
383
+ kbuf.mv_size = mdb_env_get_maxkeysize(env) * 2 + 2;
384
+ kbuf.mv_data = malloc(kbuf.mv_size * 2);
385
+ k0buf.mv_size = kbuf.mv_size;
386
+ k0buf.mv_data = (char *)kbuf.mv_data + kbuf.mv_size;
387
+ prevk.mv_data = k0buf.mv_data;
388
+
389
+ while(!Eof) {
390
+ MDB_val key, data;
391
+ int batch = 0;
392
+ flags = 0;
393
+ int appflag;
394
+
395
+ if (!dohdr) {
396
+ dohdr = 1;
397
+ } else if (!(mode & NOHDR))
398
+ readhdr();
399
+
400
+ rc = mdb_txn_begin(env, NULL, 0, &txn);
401
+ if (rc) {
402
+ fprintf(stderr, "mdb_txn_begin failed, error %d %s\n", rc, mdb_strerror(rc));
403
+ goto env_close;
404
+ }
405
+
406
+ rc = mdb_open(txn, subname, flags|MDB_CREATE, &dbi);
407
+ if (rc) {
408
+ fprintf(stderr, "mdb_open failed, error %d %s\n", rc, mdb_strerror(rc));
409
+ goto txn_abort;
410
+ }
411
+ prevk.mv_size = 0;
412
+ if (append) {
413
+ mdb_set_compare(txn, dbi, greater);
414
+ if (flags & MDB_DUPSORT)
415
+ mdb_set_dupsort(txn, dbi, greater);
416
+ }
417
+
418
+ rc = mdb_cursor_open(txn, dbi, &mc);
419
+ if (rc) {
420
+ fprintf(stderr, "mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
421
+ goto txn_abort;
422
+ }
423
+
424
+ while(1) {
425
+ rc = readline(&key, &kbuf);
426
+ if (rc) /* rc == EOF */
427
+ break;
428
+
429
+ rc = readline(&data, &dbuf);
430
+ if (rc) {
431
+ fprintf(stderr, "%s: line %" Z "d: failed to read key value\n", prog, lineno);
432
+ goto txn_abort;
433
+ }
434
+
435
+ if (append) {
436
+ appflag = MDB_APPEND;
437
+ if (flags & MDB_DUPSORT) {
438
+ if (prevk.mv_size == key.mv_size && !memcmp(prevk.mv_data, key.mv_data, key.mv_size))
439
+ appflag = MDB_CURRENT|MDB_APPENDDUP;
440
+ else {
441
+ memcpy(prevk.mv_data, key.mv_data, key.mv_size);
442
+ prevk.mv_size = key.mv_size;
443
+ }
444
+ }
445
+ } else {
446
+ appflag = 0;
447
+ }
448
+ rc = mdb_cursor_put(mc, &key, &data, putflags|appflag);
449
+ if (rc == MDB_KEYEXIST && putflags)
450
+ continue;
451
+ if (rc) {
452
+ fprintf(stderr, "mdb_cursor_put failed, error %d %s\n", rc, mdb_strerror(rc));
453
+ goto txn_abort;
454
+ }
455
+ batch++;
456
+ if (batch == 100) {
457
+ rc = mdb_txn_commit(txn);
458
+ if (rc) {
459
+ fprintf(stderr, "%s: line %" Z "d: txn_commit: %s\n",
460
+ prog, lineno, mdb_strerror(rc));
461
+ goto env_close;
462
+ }
463
+ rc = mdb_txn_begin(env, NULL, 0, &txn);
464
+ if (rc) {
465
+ fprintf(stderr, "mdb_txn_begin failed, error %d %s\n", rc, mdb_strerror(rc));
466
+ goto env_close;
467
+ }
468
+ rc = mdb_cursor_open(txn, dbi, &mc);
469
+ if (rc) {
470
+ fprintf(stderr, "mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
471
+ goto txn_abort;
472
+ }
473
+ if (appflag & MDB_APPENDDUP) {
474
+ MDB_val k, d;
475
+ mdb_cursor_get(mc, &k, &d, MDB_LAST);
476
+ }
477
+ batch = 0;
478
+ }
479
+ }
480
+ rc = mdb_txn_commit(txn);
481
+ txn = NULL;
482
+ if (rc) {
483
+ fprintf(stderr, "%s: line %" Z "d: txn_commit: %s\n",
484
+ prog, lineno, mdb_strerror(rc));
485
+ goto env_close;
486
+ }
487
+ mdb_dbi_close(env, dbi);
488
+ }
489
+
490
+ txn_abort:
491
+ mdb_txn_abort(txn);
492
+ env_close:
493
+ mdb_env_close(env);
494
+
495
+ return rc ? EXIT_FAILURE : EXIT_SUCCESS;
496
+ }