/*
- * (c) 2009 by Art Cancro and citadel.org
+ * (c) 2009-2011 by Art Cancro and citadel.org
* This program is released under the terms of the GNU General Public License v3.
*/
#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
+#include <fcntl.h>
+#include <errno.h>
#include <readline/readline.h>
+#include "ctdlsh.h"
-int main(int argc, char **argv)
-{
- char *cmd = NULL;
- char *prompt = "> ";
- int server_socket = 0;
+#define CTDLDIR "/appl/citadel"
- printf("Attaching to server...\r");
- fflush(stdout);
- server_socket = sock_connect("localhost", "504", "tcp");
- if (server_socket < 0) {
- exit(1);
+
+
+int com_quit(char *cmdbuf) {
+ abort();
+}
+
+
+
+
+/*
+ * Commands understood by ctdlsh
+ */
+typedef struct {
+ char *name;
+ rl_icpfunc_t *func;
+ char *doc;
+} COMMAND;
+
+COMMAND commands[] = {
+ { "quit", com_quit, "Quit using ctdlsh" },
+ { "exit", com_quit, "Quit using ctdlsh" },
+ { NULL, NULL, NULL }
+};
+
+
+int discover_ipgm_secret(char *dirname) {
+ int fd;
+ struct partial_config ccc;
+ char configfile[1024];
+
+ sprintf(configfile, "%s/citadel.config", dirname);
+ fd = open(configfile, O_RDONLY);
+ if (fd < 0) {
+ fprintf(stderr, "%s: %s\n", configfile, strerror(errno));
+ return(-1);
}
- printf(" \r");
- printf("\nCitadel administration shell v" PACKAGE_VERSION "\n");
- printf("(c) 2009 citadel.org GPLv3\n");
- printf("Type a command. Or don't. We don't care.\n\n");
+ if (read(fd, &ccc, sizeof(struct partial_config)) != sizeof(struct partial_config)) {
+ fprintf(stderr, "%s: %s\n", configfile, strerror(errno));
+ return(-1);
+ }
+ if (close(fd) != 0) {
+ fprintf(stderr, "%s: %s\n", configfile, strerror(errno));
+ return(-1);
+ }
+ return(ccc.c_ipgm_secret);
+}
+
+void do_main_loop(int server_socket) {
+ char *cmd = NULL;
+ char prompt[1024];
+ char buf[1024];
+ char server_reply[1024];
+ int i;
+
+ strcpy(prompt, "> ");
+
+ /* Do an INFO command and learn the hostname for the prompt */
+ sock_puts(server_socket, "INFO");
+ sock_getln(server_socket, buf, sizeof buf);
+ if (buf[0] == '1') {
+ i = 0;
+ while(sock_getln(server_socket, buf, sizeof buf), strcmp(buf, "000")) {
+ if (i == 1) {
+ sprintf(prompt, "\n%s> ", buf);
+ }
+ ++i;
+ }
+ }
+
+ /* Here we go ... main command loop */
while (cmd = readline(prompt)) {
if ((cmd) && (*cmd)) {
add_history(cmd);
+
+ sock_puts(server_socket, cmd);
+ sock_getln(server_socket, server_reply, sizeof server_reply);
+ printf("%s\n", server_reply);
+
+ if ((server_reply[0] == '4') || (server_reply[0] == '8')) {
+ /* we might consider putting something here */
+ sock_puts(server_socket, "000");
+ }
+
+ if ((server_reply[0] == '1') || (server_reply[0] == '8')) {
+ while(sock_getln(server_socket, buf, sizeof buf), strcmp(buf, "000")) {
+ printf("%s\n", buf);
+ }
+ }
+
}
- printf("\nHaha, you said: '%s'\n\n", cmd);
free(cmd);
}
+}
+
+int main(int argc, char **argv)
+{
+ int server_socket = 0;
+ char buf[1024];
+ int ipgm_secret = (-1);
+ int c;
+ char *ctdldir = CTDLDIR;
+
+ printf("\nCitadel administration shell v" PACKAGE_VERSION "\n");
+ printf("(c) 2009-2011 citadel.org GPLv3\n");
+
+ opterr = 0;
+ while ((c = getopt (argc, argv, "h:")) != -1) {
+ switch(c) {
+ case 'h':
+ ctdldir = optarg;
+ break;
+ case '?':
+ if (optopt == 'h') {
+ fprintf(stderr, "Option -%c requires an argument\n", optopt);
+ }
+ else {
+ fprintf(stderr, "Unknown option '-%c'\n", optopt);
+ fprintf(stderr, "usage: %s [-h citadel_dir]\n", argv[0]);
+ }
+ exit(1);
+ }
+ }
+
+ ipgm_secret = discover_ipgm_secret(ctdldir);
+ if (ipgm_secret < 0) {
+ exit(1);
+ }
+
+ printf("Trying %s...\n", ctdldir);
+ sprintf(buf, "%s/citadel.socket", ctdldir);
+ server_socket = uds_connectsock(buf);
+ if (server_socket < 0) {
+ exit(1);
+ }
+
+ sock_getln(server_socket, buf, sizeof buf);
+ printf("%s\n", buf);
+
+ sock_printf(server_socket, "IPGM %d\n", ipgm_secret);
+ sock_getln(server_socket, buf, sizeof buf);
+ printf("%s\n", buf);
+
+ if (buf[0] == '2') {
+ do_main_loop(server_socket);
+ }
+ sock_puts(server_socket, "QUIT");
+ sock_getln(server_socket, buf, sizeof buf);
+ printf("%s\n", buf);
+ close(server_socket);
exit(0);
}