more work on ctdl3264
[citadel.git] / citadel / utils / ctdl3264.c
1 // Attempt to convert 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 the citadel.org team
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 static DB_ENV *dbenv;           // The DB environment (global)
33
34 // Open the database environment
35 void open_dbenv(char *src_dir) {
36         int ret;
37         int i;
38         u_int32_t flags = 0;
39         int dbversion_major, dbversion_minor, dbversion_patch;
40
41         printf( "db: open_dbenv() starting\n"
42                 "db:    Linked zlib: %s\n"
43                 "db: Compiled libdb: %s\n"
44                 "db:   Linked libdb: %s\n",
45                 zlibVersion(),
46                 DB_VERSION_STRING,
47                 db_version(&dbversion_major, &dbversion_minor, &dbversion_patch)
48         );
49
50         // Create synthetic integer version numbers and compare them.
51         // Never allow citserver to run with a libdb older then the one with which it was compiled.
52         int compiled_db_version = ( (DB_VERSION_MAJOR * 1000000) + (DB_VERSION_MINOR * 1000) + (DB_VERSION_PATCH) );
53         int linked_db_version = ( (dbversion_major * 1000000) + (dbversion_minor * 1000) + (dbversion_patch) );
54         if (compiled_db_version > linked_db_version) {
55                 printf( "db: ctdl3264 is running with a version of libdb older than the one with which it was compiled.\n"
56                         "db: This is an invalid configuration.  ctdl3264 will now exit to prevent data loss.");
57                 exit(CTDLEXIT_DB);
58         }
59
60         printf("db: Setting up DB environment\n");
61         ret = db_env_create(&dbenv, 0);
62         if (ret) {
63                 printf("db: db_env_create: %s\n", db_strerror(ret));
64                 printf("db: exit code %d\n", ret);
65                 exit(CTDLEXIT_DB);
66         }
67
68         // We want to specify the shared memory buffer pool cachesize, but everything else is the default.
69         ret = dbenv->set_cachesize(dbenv, 0, 64 * 1024, 0);
70         if (ret) {
71                 printf("db: set_cachesize: %s\n", db_strerror(ret));
72                 dbenv->close(dbenv, 0);
73                 printf("db: exit code %d\n", ret);
74                 exit(CTDLEXIT_DB);
75         }
76
77         if ((ret = dbenv->set_lk_detect(dbenv, DB_LOCK_DEFAULT))) {
78                 printf("db: set_lk_detect: %s\n", db_strerror(ret));
79                 dbenv->close(dbenv, 0);
80                 printf("db: exit code %d\n", ret);
81                 exit(CTDLEXIT_DB);
82         }
83
84         flags = DB_CREATE | DB_INIT_MPOOL | DB_PRIVATE | DB_INIT_LOG;
85         printf("db: dbenv open(dir=%s, flags=%d)\n", src_dir, flags);
86         ret = dbenv->open(dbenv, src_dir, flags, 0);
87         if (ret) {
88                 printf("db: dbenv->open: %s\n", db_strerror(ret));
89                 dbenv->close(dbenv, 0);
90                 printf("db: exit code %d\n", ret);
91                 exit(CTDLEXIT_DB);
92         }
93 }
94
95
96 void close_dbenv(void) {
97         printf("db: closing dbenv\n");
98         int ret = dbenv->close(dbenv, 0);
99         if (ret) {
100                 printf("db: DBENV->close: %s\n", db_strerror(ret));
101         }
102 }
103
104
105 // placeholder convert function for the data types not yet implemented
106 void null_function(int which_cdb, DBT *key, DBT *data) {
107         //printf("DB: %02x , keylen: %3d , datalen: %d , dataptr: %x\n", which_cdb, (int)key->size, (int)data->size, data->data);
108 }
109
110
111 // convert function for a message in msgmain
112 void convert_msgmain(int which_cdb, DBT *key, DBT *data) {
113         long msgnum;
114         memcpy(&msgnum, key->data, sizeof(msgnum));
115         printf("msgmain: len is %d , key is %ld\n", key->size, msgnum);
116 }
117
118
119 // convert function for a message in msgmain
120 void convert_users(int which_cdb, DBT *key, DBT *data) {
121         char userkey[64];
122         memcpy(userkey, key->data, key->size);
123         userkey[key->size] = 0;
124         printf("users: len is %d , key is %s\n", key->size, userkey);
125 }
126
127
128 void (*convert_functions[])(int which_cdb, DBT *key, DBT *data) = {
129         convert_msgmain,        // CDB_MSGMAIN
130         convert_users,          // CDB_USERS
131         null_function,          // CDB_ROOMS
132         null_function,          // CDB_FLOORTAB
133         null_function,          // CDB_MSGLISTS
134         null_function,          // CDB_VISIT
135         null_function,          // CDB_DIRECTORY
136         null_function,          // CDB_USETABLE
137         null_function,          // CDB_BIGMSGS
138         null_function,          // CDB_FULLTEXT
139         null_function,          // CDB_EUIDINDEX
140         null_function,          // CDB_USERSBYNUMBER
141         null_function,          // CDB_EXTAUTH
142         null_function           // CDB_CONFIG
143 };
144
145
146 void convert_table(int which_cdb) {
147         int ret;
148         char dbfilename[32];
149
150         printf("Converting table %d\n", which_cdb);
151
152         // shamelessly swiped from https://docs.oracle.com/database/bdb181/html/programmer_reference/am_cursor.html
153         DB *dbp;
154         DBC *dbcp;
155         DBT key, data;
156         int num_rows = 0;
157
158         // create a database handle
159         ret = db_create(&dbp, dbenv, 0);
160         if (ret) {
161                 printf("db: db_create: %s\n", db_strerror(ret));
162                 printf("db: exit code %d\n", ret);
163                 exit(CTDLEXIT_DB);
164         }
165
166         // open the file
167         snprintf(dbfilename, sizeof dbfilename, "cdb.%02x", which_cdb);
168         printf("\033[33m\033[1mdb: opening %s\033[0m\n", dbfilename);
169         ret = dbp->open(dbp, NULL, dbfilename, NULL, DB_BTREE, 0, 0600);
170         if (ret) {
171                 printf("db: db_open: %s\n", db_strerror(ret));
172                 printf("db: exit code %d\n", ret);
173                 exit(CTDLEXIT_DB);
174         }
175
176         // Acquire a cursor
177         if ((ret = dbp->cursor(dbp, NULL, &dbcp, 0)) != 0) {
178                 printf("db: db_cursor: %s\n", db_strerror(ret));
179                 printf("db: exit code %d\n", ret);
180                 exit(CTDLEXIT_DB);
181         }
182
183         // Initialize the key/data return pair.
184         memset(&key, 0, sizeof(key));
185         memset(&data, 0, sizeof(data));
186
187         // Walk through the database and print out the key/data pairs.
188         while ((ret = dbcp->get(dbcp, &key, &data, DB_NEXT)) == 0) {
189
190                 ++num_rows;
191
192                 // Call the convert function (this will need some parameters later)
193                 convert_functions[which_cdb](which_cdb, &key, &data);
194         }
195
196         if (ret != DB_NOTFOUND) {
197                 printf("db: db_get: %s\n", db_strerror(ret));
198                 printf("db: exit code %d\n", ret);
199                 exit(CTDLEXIT_DB);
200         }
201
202         printf("%d rows\n", num_rows);
203
204         // Flush the logs...
205         //printf("\033[33m\033[1mdb: flushing the database logs\033[0m\n");
206         //if ((ret = dbenv->log_flush(dbenv, NULL))) {
207                 //printf("db: log_flush: %s\n", db_strerror(ret));
208         //}
209
210         // ...and close the database (table)
211         printf("\033[33m\033[1mdb: closing database %02x\033[0m\n", which_cdb);
212         ret = dbp->close(dbp, 0);
213         if (ret) {
214                 printf("db: db_close: %s\n", db_strerror(ret));
215         }
216
217 }
218
219
220 int main(int argc, char **argv) {
221         char *src_dir = NULL;
222         int confirmed = 0;
223
224         // Check to make sure we're running on the target 64-bit system
225         if (sizeof(void *) != 8) {
226                 fprintf(stderr, "%s: this is a %ld-bit system.\n", argv[0], sizeof(void *)*8);
227                 fprintf(stderr, "%s: you must run this on a 64-bit system, onto which a 32-bit database has been copied.\n", argv[0]);
228                 exit(1);
229         }
230
231         // Parse command line
232         int a;
233         while ((a = getopt(argc, argv, "s:d:y")) != EOF) {
234                 switch (a) {
235                 case 's':
236                         src_dir = optarg;
237                         break;
238                 case 'y':
239                         confirmed = 1;
240                         break;
241                 default:
242                         fprintf(stderr, "%s: usage: %s -s source_dir -d dest_dir\n", argv[0], argv[0]);
243                         exit(2);
244                 }
245         }
246
247         // Warn the user
248         printf("------------------------------------------------------------------------\n");
249         printf("ctdl3264 converts a Citadel database written on a 32-bit system to one  \n");
250         printf("that can be run on a 64-bit system.  It is intended to be run OFFLINE.  \n");
251         printf("Neither the source nor the target data directories should be mounted by \n");
252         printf("a running Citadel server.  We guarantee data corruption if you do not   \n");
253         printf("observe this warning!  The source [-s] directory should contain a copy  \n");
254         printf("of the database from your 32-bit system.  The destination [-d] directory\n");
255         printf("should be empty and will receive your 64-bit database.                  \n");
256         printf("------------------------------------------------------------------------\n");
257         printf("     Source 32-bit directory: %s\n", src_dir);
258         printf("Destination 64-bit directory: %s\n", "FIXME");
259         printf("------------------------------------------------------------------------\n");
260
261         if (confirmed == 1) {
262                 printf("You have specified the [-y] flag, so processing will continue.\n");
263         }
264         else {
265                 printf("Please consult the documentation to learn how to proceed.\n");
266                 exit(0);
267         }
268
269         open_dbenv(src_dir);
270         for (int i = 0; i < MAXCDB; ++i) {
271                 convert_table(i);
272         }
273         close_dbenv();
274
275         exit(0);
276 }