2 * Across-the-wire migration utility for Citadel
4 * Yes, we used goto, and gets(), and committed all sorts of other heinous sins here.
5 * The scope of this program isn't wide enough to make a difference. If you don't like
6 * it you can rewrite it.
8 * Copyright (c) 2009-2016 citadel.org
10 * This program is open source software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License version 3.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * (Note: a useful future enhancement might be to support "-h" on both sides)
28 #include <sys/types.h>
30 #include <sys/utsname.h>
38 #include <libcitadel.h>
43 #include "citadel_dirs.h"
51 * Replacement for gets() that doesn't throw a compiler warning.
52 * We're only using it for some simple prompts, so we don't need
53 * to worry about attackers exploiting it.
55 void getz(char *buf) {
58 ptr = fgets(buf, SIZ, stdin);
64 ptr = strchr(buf, '\n');
72 int main(int argc, char *argv[])
76 char relhome[PATH_MAX]="";
77 char ctdldir[PATH_MAX]=CTDLDIR;
79 char sendcommand[PATH_MAX];
83 char socket_path[PATH_MAX];
84 char remote_user[SIZ];
85 char remote_host[SIZ];
86 char remote_sendcommand[PATH_MAX];
87 FILE *sourcefp = NULL;
88 FILE *targetfp = NULL;
90 char spinning[4] = "-\\|/" ;
93 calc_dirs_n_files(relh, home, relhome, ctdldir, 0);
94 CtdlMakeTempFileName(socket_path, sizeof socket_path);
96 cmdexit = system("clear");
97 printf( "-------------------------------------------\n"
98 "Over-the-wire migration utility for Citadel\n"
99 "-------------------------------------------\n"
101 "This utility is designed to migrate your Citadel installation\n"
102 "to a new host system via a network connection, without disturbing\n"
103 "the source system. The target may be a different CPU architecture\n"
104 "and/or operating system. The source system should be running\n"
105 "Citadel %d.%02d or newer, and the target system should be running\n"
106 "either the same version or a newer version. You will also need\n"
107 "the 'rsync' utility, and OpenSSH v4 or newer.\n"
109 "You must run this utility on the TARGET SYSTEM. Any existing data\n"
110 "on this system will be ERASED.\n"
112 "Do you wish to continue? "
114 EXPORT_REV_MIN / 100,
118 if ((fgets(yesno, sizeof yesno, stdin) == NULL) || (tolower(yesno[0]) != 'y')) {
122 printf("\n\nGreat! First we will check some things out here on our target\n"
123 "system to make sure it is ready to receive data.\n\n");
125 printf("Locating 'sendcommand' and checking connectivity to Citadel...\n");
126 snprintf(sendcommand, sizeof sendcommand, "%s/sendcommand", ctdl_utilbin_dir);
127 snprintf(cmd, sizeof cmd, "%s 'NOOP'", sendcommand);
128 cmdexit = system(cmd);
130 printf("\nctdlmigrate was unable to attach to the Citadel server\n"
131 "here on the target system. Is Citadel running?\n\n");
134 printf("\nOK, this side is ready to go. Now we must connect to the source system.\n\n");
136 printf("Enter the host name or IP address of the source system\n"
137 "(example: ctdl.foo.org)\n"
142 printf("\nEnter the name of a user on %s who has full access to Citadel files\n"
143 "(usually root)\n--> ",
146 if (IsEmptyStr(remote_user))
147 goto get_remote_user;
149 printf("\nEstablishing an SSH connection to the source system...\n\n");
151 snprintf(cmd, sizeof cmd, "ssh -MNf -S %s %s@%s", socket_path, remote_user, remote_host);
152 cmdexit = system(cmd);
155 printf("This program was unable to establish an SSH session to the source system.\n\n");
160 printf("\nTesting a command over the connection...\n\n");
161 snprintf(cmd, sizeof cmd, "ssh -S %s %s@%s 'echo Remote commands are executing successfully.'",
162 socket_path, remote_user, remote_host);
163 cmdexit = system(cmd);
166 printf("Remote commands are not succeeding.\n\n");
171 printf("\nLocating the remote 'sendcommand' and Citadel installation...\n");
172 snprintf(remote_sendcommand, sizeof remote_sendcommand, "/usr/local/citadel/sendcommand");
173 snprintf(cmd, sizeof cmd, "ssh -S %s %s@%s %s NOOP",
174 socket_path, remote_user, remote_host, remote_sendcommand);
175 cmdexit = system(cmd);
177 snprintf(remote_sendcommand, sizeof remote_sendcommand, "/usr/sbin/sendcommand");
178 snprintf(cmd, sizeof cmd, "ssh -S %s %s@%s %s NOOP",
179 socket_path, remote_user, remote_host, remote_sendcommand);
180 cmdexit = system(cmd);
183 printf("\nUnable to locate Citadel programs on the remote system. Please enter\n"
184 "the name of the directory on %s which contains the 'sendcommand' program.\n"
185 "(example: /opt/foo/citadel)\n"
186 "--> ", remote_host);
188 snprintf(remote_sendcommand, sizeof remote_sendcommand, "%s/sendcommand", buf);
189 snprintf(cmd, sizeof cmd, "ssh -S %s %s@%s %s NOOP",
190 socket_path, remote_user, remote_host, remote_sendcommand);
191 cmdexit = system(cmd);
195 printf("ctdlmigrate was unable to attach to the remote Citadel system.\n\n");
200 printf("ctdlmigrate will now begin a database migration...\n");
201 printf(" if the system doesn't start working, \n");
202 printf(" have a look at the syslog for pending jobs needing to be terminated.\n");
204 snprintf(cmd, sizeof cmd, "ssh -S %s %s@%s %s -w3600 MIGR export",
205 socket_path, remote_user, remote_host, remote_sendcommand);
206 sourcefp = popen(cmd, "r");
208 printf("\n%s\n\n", strerror(errno));
213 snprintf(cmd, sizeof cmd, "%s -w3600 MIGR import", sendcommand);
214 targetfp = popen(cmd, "w");
216 printf("\n%s\n\n", strerror(errno));
221 while (fgets(buf, sizeof buf, sourcefp) != NULL) {
222 if (fwrite(buf, strlen(buf), 1, targetfp) < 1) {
224 printf("%s\n", strerror(errno));
228 if ((linecount % 100) == 0) {
229 printf("%c\r", spinning[((linecount / 100) % 4)]);
234 FAIL: if (sourcefp) pclose(sourcefp);
235 if (targetfp) pclose(targetfp);
236 if (exitcode != 0) goto THEEND;
238 /* We need to copy a bunch of other stuff, and will do so using rsync */
240 snprintf(cmd, sizeof cmd, "ssh -S %s %s@%s %s MIGR listdirs",
241 socket_path, remote_user, remote_host, remote_sendcommand);
242 sourcefp = popen(cmd, "r");
244 printf("\n%s\n\n", strerror(errno));
248 while ((fgets(buf, sizeof buf, sourcefp)) && (strcmp(buf, "000"))) {
249 buf[strlen(buf)-1] = 0;
251 if (!strncasecmp(buf, "bio|", 4)) {
252 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
253 socket_path, remote_user, remote_host, &buf[4], ctdl_bio_dir);
255 else if (!strncasecmp(buf, "files|", 6)) {
256 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
257 socket_path, remote_user, remote_host, &buf[6], ctdl_file_dir);
259 else if (!strncasecmp(buf, "userpics|", 9)) {
260 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
261 socket_path, remote_user, remote_host, &buf[9], ctdl_usrpic_dir);
263 else if (!strncasecmp(buf, "messages|", 9)) {
264 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
265 socket_path, remote_user, remote_host, &buf[9], ctdl_message_dir);
267 else if (!strncasecmp(buf, "keys|", 5)) {
268 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
269 socket_path, remote_user, remote_host, &buf[5], ctdl_key_dir);
271 else if (!strncasecmp(buf, "images|", 7)) {
272 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
273 socket_path, remote_user, remote_host, &buf[7], ctdl_image_dir);
275 else if (!strncasecmp(buf, "info|", 5)) {
276 snprintf(cmd, sizeof cmd, "rsync -va --rsh='ssh -S %s' %s@%s:%s/ %s/",
277 socket_path, remote_user, remote_host, &buf[5], ctdl_info_dir);
280 strcpy(cmd, "false"); /* cheap and sleazy way to throw an error */
283 cmdexit = system(cmd);
290 THEEND: if (exitcode == 0) {
291 printf("\n\n *** Citadel migration was successful! *** \n\n");
294 printf("\n\n *** Citadel migration was unsuccessful. *** \n\n");
296 printf("\nShutting down the socket connection...\n\n");
297 snprintf(cmd, sizeof cmd, "ssh -S %s -N -O exit %s@%s",
298 socket_path, remote_user, remote_host);
299 cmdexit = system(cmd);
302 printf("There was an error shutting down the socket.\n\n");