2 * Command-line utility to transmit a server command.
4 * Copyright (c) 1987-2010 by the citadel.org team
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 #include "ctdl_module.h"
25 #include <sys/types.h>
32 #if TIME_WITH_SYS_TIME
33 # include <sys/time.h>
37 # include <sys/time.h>
46 #include <libcitadel.h>
48 #include "citadel_ipc.h"
52 #define LOCKFILE "/tmp/LCK.sendcommand"
54 static CtdlIPC *ipc = NULL;
57 * make sure only one copy of sendcommand runs at a time, using lock files
59 int set_lockfile(void)
65 if ((lfp = fopen(LOCKFILE, "r")) != NULL) {
66 rv = fscanf(lfp, "%d", &onppid);
68 if (!kill(onppid, 0) || errno == EPERM)
71 lfp = fopen(LOCKFILE, "w");
72 fprintf(lfp, "%ld\n", (long) getpid());
77 void remove_lockfile(void)
83 * Why both cleanup() and nq_cleanup() ? Notice the alarm() call in
84 * cleanup() . If for some reason sendcommand hangs waiting for the server
85 * to clean up, the alarm clock goes off and the program exits anyway.
86 * The cleanup() routine makes a check to ensure it's not reentering, in
87 * case the ipc module looped it somehow.
89 void nq_cleanup(int e)
92 fprintf(stderr, "\nWatch dog time out.\n");
98 * send binary to server
100 void serv_write(CtdlIPC *ipc, const char *buf, unsigned int nbytes)
102 unsigned int bytes_written = 0;
105 #if defined(HAVE_OPENSSL)
107 serv_write_ssl(ipc, buf, nbytes);
112 while (bytes_written < nbytes) {
113 retval = write(ipc->sock, &buf[bytes_written],
114 nbytes - bytes_written);
116 connection_died(ipc, 0);
119 bytes_written += retval;
126 static int nested = 0;
129 signal(SIGALRM, nq_cleanup);
130 serv_write(ipc, "\n", 1);
137 * This is implemented as a function rather than as a macro because the
138 * client-side IPC modules expect logoff() to be defined. They call logoff()
139 * when a problem connecting or staying connected to the server occurs.
147 * Connect sendcommand to the Citadel server running on this computer.
149 void np_attach_to_server(char *host, char *port)
152 char hostbuf[256], portbuf[256];
154 {"sendcommand", NULL};
157 fprintf(stderr, "Attaching to server...\n");
158 strcpy(hostbuf, host);
159 strcpy(portbuf, port);
160 ipc = CtdlIPC_new(1, args, hostbuf, portbuf);
162 fprintf(stderr, "Can't connect: %s\n", strerror(errno));
165 CtdlIPC_chat_recv(ipc, buf);
166 fprintf(stderr, "%s\n", &buf[4]);
167 snprintf(buf, sizeof buf, "IPGM %d", config.c_ipgm_secret);
168 r = CtdlIPCInternalProgram(ipc, config.c_ipgm_secret, buf);
169 fprintf(stderr, "%s\n", buf);
176 void sendcommand_die(void) {
182 int main(int argc, char **argv)
191 char relhome[PATH_MAX]="";
192 char ctdldir[PATH_MAX]=CTDLDIR;
196 int server_shutting_down = 0;
198 strcpy(ctdl_home_directory, DEFAULT_PORT);
202 * Change directories if specified
204 for (a = 1; a < argc; ++a) {
205 if (!strncmp(argv[a], "-h", 2)) {
206 relh=argv[a][2]!='/';
207 if (!relh) safestrncpy(ctdl_home_directory, &argv[a][2],
208 sizeof ctdl_home_directory);
210 safestrncpy(relhome, &argv[a][2],
213 } else if (!strncmp(argv[a], "-w", 2)) {
214 watchdog = atoi(&argv[a][2]);
218 if (!IsEmptyStr(cmd))
220 strcat(cmd, argv[a]);
224 calc_dirs_n_files(relh, home, relhome, ctdldir, 0);
227 signal(SIGINT, cleanup);
228 signal(SIGQUIT, cleanup);
229 signal(SIGHUP, cleanup);
230 signal(SIGTERM, cleanup);
232 fprintf(stderr, "sendcommand: started (pid=%d) "
235 ctdl_home_directory);
239 signal(SIGALRM, nq_cleanup); /* Set up a watchdog type timer in case we hang */
241 np_attach_to_server(UDS, ctdl_home_directory);
243 setIPCDeathHook(sendcommand_die);
245 fprintf(stderr, "%s\n", cmd);
246 CtdlIPC_chat_send(ipc, cmd);
247 CtdlIPC_chat_recv(ipc, buf);
248 fprintf(stderr, "%s\n", buf);
253 if (!strncasecmp(&buf[1], "31", 2)) {
254 server_shutting_down = 1;
258 while (CtdlIPC_chat_recv(ipc, buf), strcmp(buf, "000")) {
260 alarm(watchdog); /* Kick the watchdog timer */
262 } else if (buf[0] == '4') {
264 if (fgets(buf, sizeof buf, stdin) == NULL)
266 if (!IsEmptyStr(buf))
267 if (buf[strlen(buf) - 1] == '\n')
268 buf[strlen(buf) - 1] = 0;
269 if (!IsEmptyStr(buf))
270 if (buf[strlen(buf) - 1] == '\r')
271 buf[strlen(buf) - 1] = 0;
272 if (strcmp(buf, "000"))
273 CtdlIPC_chat_send(ipc, buf);
276 FD_SET(ipc->sock, &read_fd);
277 ret = select(ipc->sock+1, &read_fd, NULL, NULL, &tv);
279 if (!(errno == EINTR || errno == EAGAIN))
280 fprintf(stderr, "select() failed: %s", strerror(errno));
289 n = read(ipc->sock, rbuf, SIZ);
292 fprintf(stderr, rbuf);
296 alarm(watchdog); /* Kick the watchdog timer */
297 } while (strcmp(buf, "000"));
298 CtdlIPC_chat_send(ipc, "\n");
299 CtdlIPC_chat_send(ipc, "000");
301 alarm(0); /* Shutdown the watchdog timer */
302 fprintf(stderr, "sendcommand: processing ended.\n");
304 /* Clean up and log off ... unless the server indicated that the command
305 * we sent is shutting it down, in which case we want to just cut the
306 * connection and exit.
308 if (server_shutting_down) {
320 void stty_ctdl(int cmd) {