|
|
@@ -0,0 +1,309 @@
|
|
|
+/*
|
|
|
+ * Copyright (c) 2015-2016 Red Hat, Inc.
|
|
|
+ *
|
|
|
+ * All rights reserved.
|
|
|
+ *
|
|
|
+ * Author: Jan Friesse (jfriesse@redhat.com)
|
|
|
+ *
|
|
|
+ * This software licensed under BSD license, the text of which follows:
|
|
|
+ *
|
|
|
+ * Redistribution and use in source and binary forms, with or without
|
|
|
+ * modification, are permitted provided that the following conditions are met:
|
|
|
+ *
|
|
|
+ * - Redistributions of source code must retain the above copyright notice,
|
|
|
+ * this list of conditions and the following disclaimer.
|
|
|
+ * - Redistributions in binary form must reproduce the above copyright notice,
|
|
|
+ * this list of conditions and the following disclaimer in the documentation
|
|
|
+ * and/or other materials provided with the distribution.
|
|
|
+ * - Neither the name of the Red Hat, Inc. nor the names of its
|
|
|
+ * contributors may be used to endorse or promote products derived from this
|
|
|
+ * software without specific prior written permission.
|
|
|
+ *
|
|
|
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
|
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
|
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
|
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
|
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
|
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
|
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
|
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
|
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
|
|
+ * THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
+ */
|
|
|
+
|
|
|
+#include <err.h>
|
|
|
+#include <errno.h>
|
|
|
+#include <getopt.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <stdlib.h>
|
|
|
+#include <string.h>
|
|
|
+
|
|
|
+#include "qnet-config.h"
|
|
|
+
|
|
|
+#include "dynar.h"
|
|
|
+#include "dynar-str.h"
|
|
|
+#include "utils.h"
|
|
|
+#include "unix-socket.h"
|
|
|
+
|
|
|
+#define IPC_READ_BUF_SIZE 512
|
|
|
+
|
|
|
+enum qnetd_tool_operation {
|
|
|
+ QNETD_TOOL_OPERATION_NONE,
|
|
|
+ QNETD_TOOL_OPERATION_SHUTDOWN,
|
|
|
+ QNETD_TOOL_OPERATION_STATUS,
|
|
|
+ QNETD_TOOL_OPERATION_LIST,
|
|
|
+};
|
|
|
+
|
|
|
+enum qnetd_tool_exit_code {
|
|
|
+ QNETD_TOOL_EXIT_CODE_NO_ERROR = 0,
|
|
|
+ QNETD_TOOL_EXIT_CODE_USAGE = 1,
|
|
|
+ QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR = 2,
|
|
|
+ QNETD_TOOL_EXIT_CODE_SOCKET_CONNECT = 3,
|
|
|
+ QNETD_TOOL_EXIT_CODE_QNETD_RETURNED_ERROR = 4,
|
|
|
+};
|
|
|
+
|
|
|
+static void
|
|
|
+usage(void)
|
|
|
+{
|
|
|
+
|
|
|
+ printf("usage: %s [-Hhlsv] [-c cluster_name] [-p qnetd_ipc_socket_path]\n",
|
|
|
+ QNETD_TOOL_PROGRAM_NAME);
|
|
|
+}
|
|
|
+
|
|
|
+static void
|
|
|
+cli_parse(int argc, char * const argv[], enum qnetd_tool_operation *operation,
|
|
|
+ int *verbose, char **cluster_name, char **socket_path)
|
|
|
+{
|
|
|
+ int ch;
|
|
|
+
|
|
|
+ *operation = QNETD_TOOL_OPERATION_NONE;
|
|
|
+ *verbose = 0;
|
|
|
+ *cluster_name = NULL;
|
|
|
+ *socket_path = strdup(QNETD_LOCAL_SOCKET_FILE);
|
|
|
+
|
|
|
+ if (*socket_path == NULL) {
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR,
|
|
|
+ "Can't alloc memory for socket path string");
|
|
|
+ }
|
|
|
+
|
|
|
+ while ((ch = getopt(argc, argv, "Hhlsvc:p:")) != -1) {
|
|
|
+ switch (ch) {
|
|
|
+ case 'H':
|
|
|
+ *operation = QNETD_TOOL_OPERATION_SHUTDOWN;
|
|
|
+ break;
|
|
|
+ case 'l':
|
|
|
+ *operation = QNETD_TOOL_OPERATION_LIST;
|
|
|
+ break;
|
|
|
+ case 's':
|
|
|
+ *operation = QNETD_TOOL_OPERATION_STATUS;
|
|
|
+ break;
|
|
|
+ case 'v':
|
|
|
+ *verbose = 1;
|
|
|
+ break;
|
|
|
+ case 'c':
|
|
|
+ free(*cluster_name);
|
|
|
+ *cluster_name = strdup(optarg);
|
|
|
+ if (*cluster_name == NULL) {
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR,
|
|
|
+ "Can't alloc memory for cluster name string");
|
|
|
+ }
|
|
|
+ break;
|
|
|
+ case 'p':
|
|
|
+ free(*socket_path);
|
|
|
+ *socket_path = strdup(optarg);
|
|
|
+ if (*socket_path == NULL) {
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR,
|
|
|
+ "Can't alloc memory for socket path string");
|
|
|
+ }
|
|
|
+ break;
|
|
|
+ case 'h':
|
|
|
+ case '?':
|
|
|
+ usage();
|
|
|
+ exit(QNETD_TOOL_EXIT_CODE_USAGE);
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (*operation == QNETD_TOOL_OPERATION_NONE) {
|
|
|
+ usage();
|
|
|
+ exit(QNETD_TOOL_EXIT_CODE_USAGE);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static int
|
|
|
+store_command(struct dynar *str, enum qnetd_tool_operation operation, int verbose,
|
|
|
+ const char *cluster_name)
|
|
|
+{
|
|
|
+ const char *nline = "\n\0";
|
|
|
+ const int nline_len = 2;
|
|
|
+
|
|
|
+ switch (operation) {
|
|
|
+ case QNETD_TOOL_OPERATION_NONE:
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR, "Unhandled operation none");
|
|
|
+ break;
|
|
|
+ case QNETD_TOOL_OPERATION_SHUTDOWN:
|
|
|
+ if (dynar_str_cat(str, "shutdown ") != 0) {
|
|
|
+ return (-1);
|
|
|
+ }
|
|
|
+ break;
|
|
|
+ case QNETD_TOOL_OPERATION_STATUS:
|
|
|
+ if (dynar_str_cat(str, "status ") != 0) {
|
|
|
+ return (-1);
|
|
|
+ }
|
|
|
+ break;
|
|
|
+ case QNETD_TOOL_OPERATION_LIST:
|
|
|
+ if (dynar_str_cat(str, "list ") != 0) {
|
|
|
+ return (-1);
|
|
|
+ }
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (verbose) {
|
|
|
+ if (dynar_str_cat(str, "verbose ") != 0) {
|
|
|
+ return (-1);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (cluster_name != NULL) {
|
|
|
+ if (dynar_str_cat(str, "cluster ") != 0 ||
|
|
|
+ dynar_str_quote_cat(str, cluster_name) != 0 ||
|
|
|
+ dynar_str_cat(str, " ") != 0) {
|
|
|
+ return (-1);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (dynar_cat(str, nline, nline_len) != 0) {
|
|
|
+ return (-1);
|
|
|
+ }
|
|
|
+
|
|
|
+ return (0);
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+ * -1 - Internal error (can't alloc memory)
|
|
|
+ * 0 - No error
|
|
|
+ * 1 - IPC returned error
|
|
|
+ * 2 - Unknown status line
|
|
|
+ */
|
|
|
+static int
|
|
|
+read_ipc_reply(FILE *f)
|
|
|
+{
|
|
|
+ struct dynar read_str;
|
|
|
+ int ch;
|
|
|
+ int status_readed;
|
|
|
+ int res;
|
|
|
+ static const char *ok_str = "OK";
|
|
|
+ static const char *err_str = "Error";
|
|
|
+ int err_set;
|
|
|
+ char c;
|
|
|
+
|
|
|
+ dynar_init(&read_str, IPC_READ_BUF_SIZE);
|
|
|
+
|
|
|
+ status_readed = 0;
|
|
|
+ err_set = 0;
|
|
|
+ res = 0;
|
|
|
+
|
|
|
+ while ((ch = fgetc(f)) != EOF) {
|
|
|
+ if (status_readed) {
|
|
|
+ putc(ch, (err_set ? stderr : stdout));
|
|
|
+ } else {
|
|
|
+ if (ch == '\r') {
|
|
|
+ } else if (ch == '\n') {
|
|
|
+ status_readed = 1;
|
|
|
+
|
|
|
+ c = '\0';
|
|
|
+ if (dynar_cat(&read_str, &c, sizeof(c)) != 0) {
|
|
|
+ res = -1;
|
|
|
+ goto exit_destroy;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (strcasecmp(dynar_data(&read_str), ok_str) == 0) {
|
|
|
+ } else if (strcasecmp(dynar_data(&read_str), err_str) == 0) {
|
|
|
+ err_set = 1;
|
|
|
+ res = 1;
|
|
|
+ fprintf(stderr, "Error: ");
|
|
|
+ } else {
|
|
|
+ res = 2;
|
|
|
+ goto exit_destroy;
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ c = ch;
|
|
|
+ if (dynar_cat(&read_str, &c, sizeof(c)) != 0) {
|
|
|
+ res = -1;
|
|
|
+ goto exit_destroy;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+exit_destroy:
|
|
|
+ dynar_destroy(&read_str);
|
|
|
+
|
|
|
+ return (res);
|
|
|
+}
|
|
|
+
|
|
|
+int
|
|
|
+main(int argc, char * const argv[])
|
|
|
+{
|
|
|
+ enum qnetd_tool_operation operation;
|
|
|
+ int verbose;
|
|
|
+ char *cluster_name;
|
|
|
+ char *socket_path;
|
|
|
+ int sock_fd;
|
|
|
+ FILE *sock;
|
|
|
+ struct dynar send_str;
|
|
|
+ int res;
|
|
|
+ int exit_code;
|
|
|
+
|
|
|
+ exit_code = QNETD_TOOL_EXIT_CODE_NO_ERROR;
|
|
|
+
|
|
|
+ cli_parse(argc, argv, &operation, &verbose, &cluster_name, &socket_path);
|
|
|
+
|
|
|
+ dynar_init(&send_str, QNETD_IPC_MAX_RECEIVE_SIZE);
|
|
|
+
|
|
|
+ sock_fd = unix_socket_client_create(socket_path, 0);
|
|
|
+ if (sock_fd == -1) {
|
|
|
+ err(QNETD_TOOL_EXIT_CODE_SOCKET_CONNECT,
|
|
|
+ "Can't connect to qnetd socket (is QNetd running?)");
|
|
|
+ }
|
|
|
+
|
|
|
+ sock = fdopen(sock_fd, "w+t");
|
|
|
+ if (sock == NULL) {
|
|
|
+ err(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR, "Can't open QNetd socket fd");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (store_command(&send_str, operation, verbose, cluster_name) != 0) {
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR, "Can't store command");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (fprintf(sock, "%s", dynar_data(&send_str)) != strlen(dynar_data(&send_str)) ||
|
|
|
+ fflush(sock) != 0) {
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR, "Can't send command");
|
|
|
+ }
|
|
|
+
|
|
|
+ res = read_ipc_reply(sock);
|
|
|
+ switch (res) {
|
|
|
+ case -1:
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_INTERNAL_ERROR, "Internal error during IPC status line read");
|
|
|
+ break;
|
|
|
+ case 0:
|
|
|
+ break;
|
|
|
+ case 1:
|
|
|
+ exit_code = QNETD_TOOL_EXIT_CODE_QNETD_RETURNED_ERROR;
|
|
|
+ break;
|
|
|
+ case 2:
|
|
|
+ errx(QNETD_TOOL_EXIT_CODE_SOCKET_CONNECT, "Unknown status line returned by IPC server");
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (fclose(sock) != 0) {
|
|
|
+ warn("Can't close QNetd socket");
|
|
|
+ }
|
|
|
+
|
|
|
+ free(cluster_name);
|
|
|
+ free(socket_path);
|
|
|
+ dynar_destroy(&send_str);
|
|
|
+
|
|
|
+ return (exit_code);
|
|
|
+}
|