]> code.citadel.org Git - citadel.git/blob - citadel/utils/ctdl3264.c
ctdl3264: fix zero length key in bigmsgs table
[citadel.git] / citadel / utils / ctdl3264.c
1 // Attempt to  your database from 32-bit to 64-bit.
2 // Don't run this.  It doesn't work and if you try to run it you will immediately die.
3 //
4 // Copyright (c) 2023 by Art Cancro citadel.org
5 //
6 // This program is open source software.  Use, duplication, or disclosure
7 // is subject to the terms of the GNU General Public License, version 3.
8
9 #include <stdlib.h>
10 #include <unistd.h>
11 #include <ctype.h>
12 #include <stdio.h>
13 #include <signal.h>
14 #include <sys/types.h>
15 #include <sys/socket.h>
16 #include <sys/un.h>
17 #include <netdb.h>
18 #include <string.h>
19 #include <pwd.h>
20 #include <errno.h>
21 #include <stdarg.h>
22 #include <limits.h>
23 #include <libcitadel.h>
24 #include <zlib.h>
25 #include <db.h>
26 #include "../server/sysdep.h"
27 #include "../server/citadel_defs.h"
28 #include "../server/server.h"
29 #include "../server/citadel_dirs.h"
30 #include "ctdl3264_structs.h"
31
32
33 // Wrapper for realloc() that crashes and burns if the call fails.
34 void *reallok(void *ptr, size_t size) {
35         void *p = realloc(ptr, size);
36         if (!p) {
37                 fprintf(stderr, "realloc() failed to resize %p to %ld bytes, error: %m\n", ptr, size);
38                 exit(1);
39         }
40         return p;
41 }
42 #define realloc reallok
43
44
45 // Open a database environment
46 DB_ENV *open_dbenv(char *dirname) {
47
48         DB_ENV *dbenv = NULL;
49
50         int ret;
51         int i;
52         u_int32_t flags = 0;
53         int dbversion_major, dbversion_minor, dbversion_patch;
54
55         printf( "db: open_dbenv() starting\n"
56                 "db:    Linked zlib: %s\n"
57                 "db: Compiled libdb: %s\n"
58                 "db:   Linked libdb: %s\n",
59                 zlibVersion(),
60                 DB_VERSION_STRING,
61                 db_version(&dbversion_major, &dbversion_minor, &dbversion_patch)
62         );
63
64         // Create synthetic integer version numbers and compare them.
65         // Never run with a libdb older than the one with which it was compiled.
66         int compiled_db_version = ( (DB_VERSION_MAJOR * 1000000) + (DB_VERSION_MINOR * 1000) + (DB_VERSION_PATCH) );
67         int linked_db_version = ( (dbversion_major * 1000000) + (dbversion_minor * 1000) + (dbversion_patch) );
68         if (compiled_db_version > linked_db_version) {
69                 printf( "db: ctdl3264 is running with a version of libdb older than the one with which it was compiled.\n"
70                         "db: This is an invalid configuration.  ctdl3264 will now exit to prevent data loss.");
71                 exit(CTDLEXIT_DB);
72         }
73
74         printf("db: Setting up DB environment\n");
75         ret = db_env_create(&dbenv, 0);
76         if (ret) {
77                 printf("db: db_env_create: %s\n", db_strerror(ret));
78                 printf("db: exit code %d\n", ret);
79                 exit(CTDLEXIT_DB);
80         }
81
82         // We want to specify the shared memory buffer pool cachesize, but everything else is the default.
83         ret = dbenv->set_cachesize(dbenv, 0, 64 * 1024, 0);
84         if (ret) {
85                 printf("db: set_cachesize: %s\n", db_strerror(ret));
86                 dbenv->close(dbenv, 0);
87                 printf("db: exit code %d\n", ret);
88                 exit(CTDLEXIT_DB);
89         }
90
91         if ((ret = dbenv->set_lk_detect(dbenv, DB_LOCK_DEFAULT))) {
92                 printf("db: set_lk_detect: %s\n", db_strerror(ret));
93                 dbenv->close(dbenv, 0);
94                 printf("db: exit code %d\n", ret);
95                 exit(CTDLEXIT_DB);
96         }
97
98         flags = DB_CREATE | DB_INIT_MPOOL | DB_PRIVATE | DB_INIT_LOG;
99         printf("db: dbenv open(dir=%s, flags=%d)\n", dirname, flags);
100         ret = dbenv->open(dbenv, dirname, flags, 0);
101         if (ret) {
102                 printf("db: dbenv->open: %s\n", db_strerror(ret));
103                 dbenv->close(dbenv, 0);
104                 printf("db: exit code %d\n", ret);
105                 exit(CTDLEXIT_DB);
106         }
107
108         return(dbenv);
109 }
110
111
112 void close_dbenv(DB_ENV *dbenv) {
113         printf("db: closing dbenv\n");
114         int ret = dbenv->close(dbenv, 0);
115         if (ret) {
116                 printf("db: dbenv->close: %s\n", db_strerror(ret));
117         }
118 }
119
120
121 // convert function for a message in msgmain
122 void convert_msgmain(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
123         int32_t in_msgnum;
124         long out_msgnum;
125         memcpy(&in_msgnum, in_key->data, sizeof(in_msgnum));
126         out_msgnum = (long)in_msgnum;
127
128         if (in_key->size != 4) {
129                 fprintf(stderr, "\033[31m\033[1m *** SOURCE DATABASE IS NOT 32-BIT *** ABORTING *** \033[0m\n");
130                 abort();
131         }
132
133         // If the msgnum is negative, we are looking at METADATA
134         if (in_msgnum < 0) {
135                 struct MetaData_32 *meta32 = (struct MetaData_32 *)in_data->data;
136
137                 // printf("\033[32m\033[1mMetadata: msgnum=%d , refcount=%d , content_type=\"%s\" , rfc822len=%d\033[0m\n", meta32->meta_msgnum, meta32->meta_refcount, meta32->meta_content_type, meta32->meta_rfc822_length);
138
139                 out_key->size = sizeof(long);
140                 out_key->data = realloc(out_key->data, out_key->size);
141                 memcpy(out_key->data, &out_msgnum, sizeof(long));
142
143                 out_data->size = sizeof(struct MetaData);
144                 out_data->data = realloc(out_data->data, out_data->size);
145                 struct MetaData *meta64 = (struct MetaData *)out_data->data;
146                 memset(meta64, 0, sizeof(struct MetaData));
147                 meta64->meta_msgnum             = (long)        meta32->meta_msgnum;
148                 meta64->meta_refcount           = (int)         meta32->meta_refcount;
149                 strcpy(meta64->meta_content_type,               meta32->meta_content_type);
150                 meta64->meta_rfc822_length      = (long)        meta32->meta_rfc822_length;
151         }
152
153         // If the msgnum is positive, we are looking at a MESSAGE
154         else if (in_msgnum > 0) {
155                 out_key->size = sizeof(long);
156                 out_key->data = realloc(out_key->data, out_key->size);
157                 memcpy(out_key->data, &out_msgnum, sizeof(long));
158
159                 // copy the message verbatim
160                 out_data->size = in_data->size;
161                 out_data->data = realloc(out_data->data, out_data->size);
162                 memcpy(out_data->data, in_data->data, out_data->size);
163                 // printf("\033[32m\033[1mMessage: %ld\033[0m\n", out_msgnum);
164         }
165
166         // If the msgnum is 0 it's probably not a valid record.
167         else {
168                 printf("\033[31mmsgmain: message number 0 is impossible, skipping this record\033[0m\n");
169         }
170 }
171
172
173 // convert function for a user record
174 void convert_users(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
175
176         // The key is a string so we can just copy it over
177         out_key->size = in_key->size;
178         out_key->data = realloc(out_key->data, out_key->size);
179         memcpy(out_key->data, in_key->data, in_key->size);
180
181         struct ctdluser_32 *user32 = (struct ctdluser_32 *)in_data->data;
182
183         out_data->size = sizeof(struct ctdluser);
184         out_data->data = realloc(out_data->data, out_data->size);
185         struct ctdluser *user64 = (struct ctdluser *)out_data->data;
186
187         user64->version                 = (int)         user32->version;
188         user64->uid                     = (uid_t)       user32->uid;
189         strcpy(user64->password,                        user32->password);
190         user64->flags                   = (unsigned)    user32->flags;
191         user64->timescalled             = (long)        user32->timescalled;
192         user64->posted                  = (long)        user32->posted;
193         user64->axlevel                 = (cit_uint8_t) user32->axlevel;
194         user64->usernum                 = (long)        user32->usernum;
195         user64->lastcall                = (time_t)      user32->lastcall;
196         user64->USuserpurge             = (int)         user32->USuserpurge;
197         strcpy(user64->fullname,                        user32->fullname);
198         user64->msgnum_bio              = (long)        user32->msgnum_bio;
199         user64->msgnum_pic              = (long)        user32->msgnum_pic;
200         strcpy(user64->emailaddrs,                      user32->emailaddrs);
201         user64->msgnum_inboxrules       = (long)        user32->msgnum_inboxrules;
202         user64->lastproc_inboxrules     = (long)        user32->lastproc_inboxrules;
203
204         // printf("\033[32m\033[1mUser: %s\033[0m\n", user64->fullname);
205 }
206
207
208 // convert function for a room record
209 void convert_rooms(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
210
211         // The key is a string so we can just copy it over
212         out_key->size = in_key->size;
213         out_key->data = realloc(out_key->data, out_key->size);
214         memcpy(out_key->data, in_key->data, in_key->size);
215
216         // data
217         struct ctdlroom_32 *room32 = (struct ctdlroom_32 *)in_data->data;
218         out_data->size = sizeof(struct ctdlroom);
219         out_data->data = realloc(out_data->data, out_data->size);
220         struct ctdlroom *room64 = (struct ctdlroom *)out_data->data;
221
222         strcpy(room64->QRname,                          room32->QRname);
223         strcpy(room64->QRpasswd,                        room32->QRpasswd);
224         room64->QRroomaide              = (long)        room32->QRroomaide;
225         room64->QRhighest               = (long)        room32->QRhighest;
226         room64->QRgen                   = (time_t)      room32->QRgen;
227         room64->QRflags                 = (unsigned)    room32->QRflags;
228         strcpy(room64->QRdirname,                       room32->QRdirname);
229         room64->msgnum_info             = (long)        room32->msgnum_info;
230         room64->QRfloor                 = (char)        room32->QRfloor;
231         room64->QRmtime                 = (time_t)      room32->QRmtime;
232         room64->QRep.expire_mode        = (int)         room32->QRep.expire_mode;
233         room64->QRep.expire_value       = (int)         room32->QRep.expire_value;
234         room64->QRnumber                = (long)        room32->QRnumber;
235         room64->QRorder                 = (char)        room32->QRorder;
236         room64->QRflags2                = (unsigned)    room32->QRflags2;
237         room64->QRdefaultview           = (int)         room32->QRdefaultview;
238         room64->msgnum_pic              = (long)        room32->msgnum_pic;
239
240         // printf("\033[32m\033[1mRoom: %s\033[0m\n", room64->QRname);
241 }
242
243
244 // convert function for a floor record
245 void convert_floors(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
246
247         // the key is an "int", and "int" is 32-bits on both 32 and 64 bit platforms.
248         out_key->size = in_key->size;
249         out_key->data = realloc(out_key->data, out_key->size);
250         memcpy(out_key->data, in_key->data, in_key->size);
251
252         // data
253         struct floor_32 *floor32 = (struct floor_32 *)in_data->data;
254         out_data->size = sizeof(struct floor);
255         out_data->data = realloc(out_data->data, out_data->size);
256         struct floor *floor64 = (struct floor *)out_data->data;
257
258         // these are probably bit-for-bit identical, actually ... but we do it the "right" way anyway
259         floor64->f_flags                = (unsigned short)      floor32->f_flags;
260         strcpy(floor64->f_name,                                 floor32->f_name);
261         floor64->f_ref_count            = (int)                 floor32->f_ref_count;
262         floor64->f_ep.expire_mode       = (int)                 floor32->f_ep.expire_mode;
263         floor64->f_ep.expire_value      = (int)                 floor32->f_ep.expire_value;
264
265         // printf("\033[32m\033[1mFloor: %s\033[0m\n", floor64->f_name);
266 }
267
268
269 // convert function for a msglist or a fulltext index record
270 // (both are indexed by a long and the data is arrays of longs)
271 void convert_msglists(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
272         int i;
273
274         char *table = (which_cdb == CDB_FULLTEXT) ? "FullText" : "Msglist";
275
276         // records are indexed by a single "long" and contains an array of zero or more "long"s
277         // and remember ... "long" is int32_t on the source system
278         int32_t in_roomnum;
279         long out_roomnum;
280         memcpy(&in_roomnum, in_key->data, sizeof(in_roomnum));
281         out_roomnum = (long) in_roomnum;
282
283         if (in_key->size != 4) {
284                 fprintf(stderr, "\033[31m\033[1m *** SOURCE DATABASE IS NOT 32-BIT *** ABORTING *** \033[0m\n");
285                 abort();
286         }
287
288         int num_msgs = in_data->size / sizeof(int32_t);
289         // printf("\033[32m\033[1m%s: key %ld (%d messages)\033[0m\n", table, out_roomnum, num_msgs);
290
291         // the key is a "long"
292         out_key->size = sizeof(out_roomnum);
293         out_key->data = realloc(out_key->data, out_key->size);
294         memcpy(out_key->data, &out_roomnum, sizeof(out_roomnum));
295
296         // the data is another array, but a wider type
297         out_data->size = sizeof(long) * num_msgs;
298         out_data->data = realloc(out_data->data, out_data->size);
299
300         int32_t in_msg = 0;
301         long out_msg = 0;
302         for (i=0; i<num_msgs; ++i) {
303                 memcpy(&in_msg, (in_data->data + (i * sizeof(int32_t))), sizeof(int32_t));
304                 out_msg = (long) in_msg;
305                 memcpy((out_data->data + (i * sizeof(long))), &out_msg, sizeof(long));
306                 // printf("msg#%ld\n", out_msg);
307         }
308 }
309
310
311 // convert function for a visit record
312 void convert_visits(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
313
314         // data
315         struct visit_32 *visit32 = (struct visit_32 *)in_data->data;
316         out_data->size = sizeof(struct visit);
317         out_data->data = realloc(out_data->data, out_data->size);
318         struct visit *visit64 = (struct visit *)out_data->data;
319
320         //  the data (zero it out so it will compress well)
321         memset(visit64, 0, sizeof(struct visit));
322         visit64->v_roomnum              = (long)        visit32->v_roomnum;
323         visit64->v_roomgen              = (long)        visit32->v_roomgen;
324         visit64->v_usernum              = (long)        visit32->v_usernum;
325         visit64->v_lastseen             = (long)        visit32->v_lastseen;
326         visit64->v_flags                = (unsigned)    visit32->v_flags;
327         strcpy(visit64->v_seen,                         visit32->v_seen);
328         strcpy(visit64->v_answered,                     visit32->v_answered);
329         visit64->v_view                 = (int)         visit32->v_view;
330
331         // printf("\033[32m\033[1mVisit: room %ld, gen %ld, user %ld\033[0m\n", visit64->v_roomnum, visit64->v_roomgen, visit64->v_usernum);
332
333         // create the key (which is based on the data, so there is no need to convert the old key)
334         out_key->size = sizeof(struct visit_index);
335         out_key->data = realloc(out_key->data, out_key->size);
336         struct visit_index *newvisitindex = (struct visit_index *) out_key->data;
337         newvisitindex->iRoomID          =               visit64->v_roomnum;
338         newvisitindex->iRoomGen         =               visit64->v_roomgen;
339         newvisitindex->iUserID          =               visit64->v_usernum;
340 }
341
342
343 // convert function for a directory record
344 void convert_dir(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
345
346         // the key is a string
347         out_key->size = in_key->size;
348         out_key->data = realloc(out_key->data, out_key->size + 1);
349         memcpy(out_key->data, in_key->data, in_key->size);
350         char *k = (char *)out_key->data;
351         k[out_key->size] = 0;
352
353         // the data is also a string
354         out_data->size = in_data->size;
355         out_data->data = realloc(out_data->data, out_data->size + 1);
356         memcpy(out_data->data, in_data->data, in_data->size);
357         char *d = (char *)out_data->data;
358         d[out_data->size] = 0;
359
360         // please excuse my friend, he isn't null terminated
361         // printf("\033[32m\033[1mDirectory entry: %s -> %s\033[0m\n", (char *)out_key->data, (char *)out_data->data);
362 }
363
364
365 // convert function for a use table record
366 void convert_usetable(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
367
368         // the key is a string
369         out_key->size = in_key->size;
370         out_key->data = realloc(out_key->data, out_key->size);
371         memcpy(out_key->data, in_key->data, in_key->size);
372
373         // the data is a "struct UseTable"
374         struct UseTable_32 *use32 = (struct UseTable_32 *)in_data->data;
375         out_data->size = sizeof(struct UseTable);
376         out_data->data = realloc(out_data->data, out_data->size);
377         memset(out_data->data, 0, out_data->size);
378         struct UseTable *use64 = (struct UseTable *)out_data->data;
379
380         //  the data
381         strcpy(use64->ut_msgid,                         use32->ut_msgid);
382         use64->ut_timestamp             = (time_t)      use32->ut_timestamp;
383 }
384
385
386 // convert function for large message texts
387 void convert_bigmsgs(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
388
389         // The key is a packed long
390         int32_t in_msgnum;
391         long out_msgnum;
392         memcpy(&in_msgnum, in_key->data, sizeof(in_msgnum));
393         out_msgnum = (long)in_msgnum;
394
395         if (in_key->size != 4) {
396                 fprintf(stderr, "\033[31m\033[1m *** SOURCE DATABASE IS NOT 32-BIT *** ABORTING *** \033[0m\n");
397                 abort();
398         }
399
400         out_key->size = sizeof(long);
401         out_key->data = realloc(out_key->data, out_key->size);
402         memcpy(out_key->data, &out_msgnum, sizeof(long));
403
404         // the data is binary-ish but has no packed integers
405         out_data->size = in_data->size;
406         out_data->data = realloc(out_data->data, out_data->size);
407         memcpy(out_data->data, in_data->data, in_data->size);
408
409         // printf("\033[32m\033[1mBigmsg %ld , length %d\033[0m\n", out_msgnum, out_data->size);
410 }
411
412
413 // convert function for EUID Index records
414 void convert_euidindex(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
415
416         // The structure of an euidindex record *key* is:
417         // |----room_number----|----------EUID-------------|
418         //    (sizeof long)       (actual length of euid)
419
420         // The structure of an euidindex record *value* is:
421         // |-----msg_number----|----room_number----|----------EUID-------------|
422         //    (sizeof long)       (sizeof long)       (actual length of euid)
423
424         int32_t in_msgnum = 0;
425         int32_t in_roomnum = 0;
426         char euid[SIZ];
427         long out_msgnum = 0;
428         long out_roomnum = 0;
429
430         memcpy(&in_msgnum, in_data->data, sizeof(in_msgnum));
431         memcpy(&in_roomnum, in_data->data+sizeof(int32_t), sizeof(in_msgnum));
432         strcpy(euid, in_data->data+(sizeof(int32_t)*2));
433
434         out_msgnum = (long) in_msgnum;
435         out_roomnum = (long) in_roomnum;
436         // printf("euidindex: msgnum=%ld, roomnum=%ld, euid=\"%s\"\n", out_msgnum, out_roomnum, euid);
437
438         out_key->size = sizeof(long) + strlen(euid) + 1;
439         out_key->data = realloc(out_key->data, out_key->size);
440         memcpy(out_key->data, &out_roomnum, sizeof(out_roomnum));
441         strcpy(out_key->data+sizeof(out_roomnum), euid);
442
443         out_data->size = sizeof(long) + sizeof(long) + strlen(euid) + 1;
444         out_data->data = realloc(out_data->data, out_data->size);
445         memcpy(out_data->data, &out_msgnum, sizeof(out_msgnum));
446         memcpy(out_data->data+sizeof(out_msgnum), &out_roomnum, sizeof(out_roomnum));
447         strcpy(out_data->data+sizeof(out_msgnum)+sizeof(out_roomnum), euid);
448
449
450         //int i;
451         //char ch;
452 //
453         //printf("  in_key:             ");
454         //for (i=0; i<in_key->size; ++i) {
455                 //ch = 0;
456                 //memcpy(&ch, in_key->data+i, 1);
457                 //printf("%02X ", (int) ch);
458         //}
459         //printf("\n");
460 //
461         //printf(" out_key: ");
462         //for (i=0; i<out_key->size; ++i) {
463                 //ch = 0;
464                 //memcpy(&ch, out_key->data+i, 1);
465                 //printf("%02X ", (int) ch);
466         //}
467         //printf("\n");
468 //
469         //printf(" in_data:                         ");
470         //for (i=0; i<in_data->size; ++i) {
471                 //ch = 0;
472                 //memcpy(&ch, in_data->data+i, 1);
473         //      printf("%02X ", (int) ch);
474         //}
475         //printf("\n");
476 //
477         //printf("out_data: ");
478         //for (i=0; i<out_data->size; ++i) {
479                 //ch = 0;
480                 //memcpy(&ch, out_data->data+i, 1);
481                 //printf("%02X ", (int) ch);
482         //}
483         //printf("\n");
484
485
486 }
487
488
489 // convert users-by-number records
490 void convert_usersbynumber(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
491
492         // key is a long
493         // and remember ... "long" is int32_t on the source system
494         int32_t in_usernum;
495         long out_usernum;
496         memcpy(&in_usernum, in_key->data, sizeof(in_usernum));
497         out_usernum = (long) in_usernum;
498
499         if (in_key->size != 4) {
500                 fprintf(stderr, "\033[31m\033[1m *** SOURCE DATABASE IS NOT 32-BIT *** ABORTING *** \033[0m\n");
501                 abort();
502         }
503
504         out_key->size = sizeof(out_usernum);
505         out_key->data = realloc(out_key->data, out_key->size);
506         memcpy(out_key->data, &out_usernum, sizeof(out_usernum));
507
508         // value is a string
509         out_data->size = in_data->size;
510         out_data->data = realloc(out_data->data, out_data->size);
511         memcpy(out_data->data, in_data->data, in_data->size);
512
513         // printf("usersbynumber: %ld --> %s\n", out_usernum, (char *)out_data->data);
514 }
515
516
517 // convert function for a config record
518 void convert_config(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
519
520         // the key is a string
521         out_key->size = in_key->size;
522         out_key->data = realloc(out_key->data, out_key->size + 1);
523         memcpy(out_key->data, in_key->data, in_key->size);
524         char *k = (char *)out_key->data;
525         k[out_key->size] = 0;
526
527         // the data is a pair of strings
528         out_data->size = in_data->size;
529         out_data->data = realloc(out_data->data, out_data->size + 1);
530         memcpy(out_data->data, in_data->data, in_data->size);
531         char *d = (char *)out_data->data;
532         d[out_data->size] = 0;
533
534         // please excuse my friend, he isn't null terminated
535         // printf("\033[32m\033[1mConfig entry: %s -> %s\033[0m\n", (char *)out_key->data, (char *)out_data->data+strlen(out_data->data)+1);
536 }
537
538
539 // For obsolete databases, zero all the output
540 void zero_function(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) {
541         out_key->size = 0;
542         out_data->size = 0;
543 }
544
545
546 void (*convert_functions[])(int which_cdb, DBT *in_key, DBT *in_data, DBT *out_key, DBT *out_data) = {
547         convert_msgmain,        // CDB_MSGMAIN
548         convert_users,          // CDB_USERS
549         convert_rooms,          // CDB_ROOMS
550         convert_floors,         // CDB_FLOORTAB
551         convert_msglists,       // CDB_MSGLISTS
552         convert_visits,         // CDB_VISIT
553         convert_dir,            // CDB_DIRECTORY
554         convert_usetable,       // CDB_USETABLE
555         convert_bigmsgs,        // CDB_BIGMSGS
556         convert_msglists,       // CDB_FULLTEXT
557         convert_euidindex,      // CDB_EUIDINDEX
558         convert_usersbynumber,  // CDB_USERSBYNUMBER
559         zero_function,          // CDB_UNUSED1 (obsolete)
560         convert_config          // CDB_CONFIG
561 };
562
563
564 void convert_table(int which_cdb, DB_ENV *src_dbenv, DB_ENV *dst_dbenv) {
565         int ret;
566         int compressed;
567         char dbfilename[32];
568         uLongf destLen = 0;
569
570         printf("\033[43m\033[K\033[0m\n");
571         printf("\033[43m\033[30m Converting table %02x \033[K\033[0m\n", which_cdb);
572         printf("\033[43m\033[K\033[0m\n");
573
574         // shamelessly swiped from https://docs.oracle.com/database/bdb181/html/programmer_reference/am_cursor.html
575         DB *src_dbp, *dst_dbp;
576         DBC *src_dbcp;
577         DBT in_key, in_data, out_key, out_data, uncomp_data, recomp_data;
578         int num_rows = 0;
579
580         snprintf(dbfilename, sizeof dbfilename, "cdb.%02x", which_cdb);
581
582         // create a database handle for the source table
583         ret = db_create(&src_dbp, src_dbenv, 0);
584         if (ret) {
585                 printf("db: db_create: %s\n", db_strerror(ret));
586                 printf("db: exit code %d\n", ret);
587                 exit(CTDLEXIT_DB);
588         }
589
590         // open the file containing the source table
591         printf("\033[33m\033[1mdb: opening source %s\033[0m\n", dbfilename);
592         ret = src_dbp->open(src_dbp, NULL, dbfilename, NULL, DB_BTREE, 0, 0600);
593         if (ret) {
594                 printf("db: db_open: %s\n", db_strerror(ret));
595                 printf("db: exit code %d\n", ret);
596                 exit(CTDLEXIT_DB);
597         }
598
599         // create a database handle for the destination table
600         ret = db_create(&dst_dbp, dst_dbenv, 0);
601         if (ret) {
602                 printf("db: db_create: %s\n", db_strerror(ret));
603                 printf("db: exit code %d\n", ret);
604                 exit(CTDLEXIT_DB);
605         }
606
607         // open the file containing the destination table
608         printf("\033[33m\033[1mdb: opening destination %s\033[0m\n", dbfilename);
609         ret = dst_dbp->open(dst_dbp, NULL, dbfilename, NULL, DB_BTREE, (DB_CREATE | DB_TRUNCATE), 0600);
610         if (ret) {
611                 printf("db: db_open: %s\n", db_strerror(ret));
612                 printf("db: exit code %d\n", ret);
613                 exit(CTDLEXIT_DB);
614         }
615
616         // Acquire a cursor to read the source table
617         printf("\033[33m\033[1mdb: acquiring cursor\033[0m\n");
618         if ((ret = src_dbp->cursor(src_dbp, NULL, &src_dbcp, 0)) != 0) {
619                 printf("db: db_cursor: %s\n", db_strerror(ret));
620                 printf("db: exit code %d\n", ret);
621                 exit(CTDLEXIT_DB);
622         }
623
624         // Zero out these database keys
625         memset(&in_key,         0, sizeof(DBT));        // input
626         memset(&in_data,        0, sizeof(DBT));
627         memset(&out_key,        0, sizeof(DBT));        // output
628         memset(&out_data,       0, sizeof(DBT));
629         memset(&uncomp_data,    0, sizeof(DBT));        // decompressed input (the key doesn't change)
630         memset(&recomp_data,    0, sizeof(DBT));        // recompressed input (the key doesn't change)
631
632         // Walk through the database, calling convert functions as we go and clearing buffers before each call.
633         while (out_key.size = 0, out_data.size = 0, (ret = src_dbcp->get(src_dbcp, &in_key, &in_data, DB_NEXT)) == 0) {
634         
635                 // Knowing the total number of rows isn't critical to the program.  It's just for the user to know.
636                 ++num_rows;
637                 printf("   \033[32m%d\033[0m\r", num_rows);
638                 fflush(stdout);
639
640                 // If either the key or data are zero length, skip this record
641                 if ((in_key.size == 0) || (in_data.size == 0)) {
642                         printf("\033[31minput keylen=%d , datalen=%d , skipping record\033[0m\n", in_key.size, in_data.size);
643                 }
644
645                 else {  // Both key and data are >0 length so we're good to go
646
647                         // Do we need to decompress?
648                         static int32_t magic = COMPRESS_MAGIC;
649                         compressed = 0;
650                         if ( (in_data.size >= sizeof(struct CtdlCompressHeader_32)) && (!memcmp(in_data.data, &magic, sizeof(magic))) ) {
651         
652                                 // yes, we need to decompress
653                                 compressed = 1;
654                                 struct CtdlCompressHeader_32 comp32;
655                                 memcpy(&comp32, in_data.data, sizeof(struct CtdlCompressHeader_32));
656                                 uncomp_data.size = comp32.uncompressed_len;
657                                 uncomp_data.data = realloc(uncomp_data.data, uncomp_data.size);
658                                 destLen = (uLongf)comp32.uncompressed_len;
659         
660                                 ret = uncompress((Bytef *)uncomp_data.data, (uLongf *)&destLen, (const Bytef *)in_data.data+sizeof(struct CtdlCompressHeader_32), (uLong)comp32.compressed_len);
661                                 if (ret != Z_OK) {
662                                         printf("db: uncompress() error %d\n", ret);
663                                         exit(CTDLEXIT_DB);
664                                 }
665                                 // printf("DB: %02x ,  in_keylen: %-3d ,  in_datalen: %-10d , dataptr: %012lx \033[31m(decompressed)\033[0m\n", which_cdb, (int)in_key.size, comp32.uncompressed_len, (long unsigned int)uncomp_data.data);
666                         }
667                         else {
668                                 // printf("DB: %02x ,  in_keylen: %-3d ,  in_datalen: %-10d , dataptr: %012lx\n", which_cdb, (int)in_key.size, (int)in_data.size, (long unsigned int)in_data.data);
669                         }
670         
671                         // Call the convert function registered to this table
672                         convert_functions[which_cdb](which_cdb, &in_key, (compressed ? &uncomp_data : &in_data), &out_key, &out_data);
673         
674                         // The logic here is that if the source data was compressed, we compress the output too
675                         if (compressed) {
676                                 struct CtdlCompressHeader zheader;
677                                 memset(&zheader, 0, sizeof(struct CtdlCompressHeader));
678                                 zheader.magic = COMPRESS_MAGIC;
679                                 zheader.uncompressed_len = out_data.size;
680                                 recomp_data.data = realloc(recomp_data.data, ((out_data.size * 101) / 100) + 100 + sizeof(struct CtdlCompressHeader));
681         
682                                 if (compress2((Bytef *)(recomp_data.data + sizeof(struct CtdlCompressHeader)), &destLen, (Bytef *)out_data.data, (uLongf) out_data.size, 1) != Z_OK) {
683                                         printf("db: compress() error\n");
684                                         exit(CTDLEXIT_DB);
685                                 }
686                                 recomp_data.size = destLen;
687                         }
688         
689                         // write the converted record to the target database
690                         if (out_key.size > 0) {
691         
692                                 // If we compressed the output, write recomp_data instead of out_data
693                                 if (compressed) {
694                                         // printf("DB: %02x , out_keylen: %-3d , out_datalen: %-10d , dataptr: %012lx \033[31m(compressed)\033[0m\n", which_cdb, (int)out_key.size, (int)recomp_data.size, (long unsigned int)recomp_data.data);
695                                         ret = dst_dbp->put(dst_dbp, NULL, &out_key, &recomp_data, 0);
696                                 }
697                                 else {
698                                         // printf("DB: %02x , out_keylen: %-3d , out_datalen: %-10d , dataptr: %012lx\n", which_cdb, (int)out_key.size, (int)out_data.size, (long unsigned int)out_data.data);
699                                         ret = dst_dbp->put(dst_dbp, NULL, &out_key, &out_data, 0);
700                                 }
701         
702                                 if (ret) {
703                                         printf("db: cdb_put(%d): %s", which_cdb, db_strerror(ret));
704                                         exit(CTDLEXIT_DB);
705                                 }
706                         }
707                         else {
708                                 printf("\033[31moutput keylen=%d , skipping record\033[0m\n", out_key.size);
709                         }
710                 }
711         }
712
713         if (ret != DB_NOTFOUND) {
714                 printf("db: db_get: %s\n", db_strerror(ret));
715                 printf("db: exit code %d\n", ret);
716                 exit(CTDLEXIT_DB);
717         }
718
719         printf("%d rows\n", num_rows);
720
721         // free any leftover out_data pointers
722         free(out_key.data);
723         free(out_data.data);
724         free(uncomp_data.data);
725         free(recomp_data.data);
726
727         // ...and close the database (table)
728         printf("\033[33m\033[1mdb: closing source %02x\033[0m\n", which_cdb);
729         ret = src_dbp->close(src_dbp, 0);
730         if (ret) {
731                 printf("db: db_close: %s\n", db_strerror(ret));
732         }
733
734         printf("\033[33m\033[1mdb: closing destination %02x\033[0m\n", which_cdb);
735         ret = dst_dbp->close(dst_dbp, 0);
736         if (ret) {
737                 printf("db: db_close: %s\n", db_strerror(ret));
738         }
739
740         printf("\n");
741
742 }
743
744
745 int main(int argc, char **argv) {
746         char *src_dir = NULL;
747         char *dst_dir = NULL;
748         int confirmed = 0;
749         static DB_ENV *src_dbenv;               // Source DB environment (global)
750         static DB_ENV *dst_dbenv;               // Source DB environment (global)
751
752         // Check to make sure we're running on the target 64-bit system
753         if (sizeof(void *) != 8) {
754                 fprintf(stderr, "%s: this is a %ld-bit system.\n", argv[0], sizeof(void *)*8);
755                 fprintf(stderr, "%s: you must run this on a 64-bit system, onto which a 32-bit database has been copied.\n", argv[0]);
756                 exit(1);
757         }
758
759         // Parse command line
760         int a;
761         while ((a = getopt(argc, argv, "s:d:y")) != EOF) {
762                 switch (a) {
763                 case 's':
764                         src_dir = optarg;
765                         break;
766                 case 'd':
767                         dst_dir = optarg;
768                         break;
769                 case 'y':
770                         confirmed = 1;
771                         break;
772                 default:
773                         fprintf(stderr, "%s: usage: %s -s source_dir -d dest_dir\n", argv[0], argv[0]);
774                         exit(2);
775                 }
776         }
777
778         // Warn the user
779         printf("------------------------------------------------------------------------\n");
780         printf("ctdl3264 converts a Citadel database written on a 32-bit system to one  \n");
781         printf("that can be run on a 64-bit system.  It is intended to be run OFFLINE.  \n");
782         printf("Neither the source nor the target data directories should be mounted by \n");
783         printf("a running Citadel server.  We \033[1mguarantee\033[0m data corruption if you do not   \n");
784         printf("observe this warning!  The source [-s] directory should contain a copy  \n");
785         printf("of the database from your 32-bit system.  The destination [-d] directory\n");
786         printf("should be empty and will receive your 64-bit database.                  \n");
787         printf("------------------------------------------------------------------------\n");
788         printf("     Source 32-bit directory: %s\n", src_dir);
789         printf("Destination 64-bit directory: %s\n", dst_dir);
790         printf("------------------------------------------------------------------------\n");
791
792         if (confirmed == 1) {
793                 printf("You have specified the [-y] flag, so processing will continue.\n");
794         }
795         else {
796                 printf("Please read [ https://www.citadel.org/ctdl3264.html ] to learn how to proceed.\n");
797                 exit(0);
798         }
799
800         src_dbenv = open_dbenv(src_dir);
801         dst_dbenv = open_dbenv(dst_dir);
802         for (int i = 0; i < MAXCDB; ++i) {
803                 convert_table(i, src_dbenv, dst_dbenv);
804         }
805         close_dbenv(src_dbenv);
806         close_dbenv(dst_dbenv);
807
808         exit(0);
809 }