Whamcloud - gitweb
Another slightly less rough cut at XML based configuration. There
[fs/lustre-release.git] / lustre / utils / obdctl.c
index f582087..9f81756 100644 (file)
@@ -22,6 +22,7 @@
  *
  */
 
+
 #include <stdlib.h>
 #include <sys/ioctl.h>
 #include <fcntl.h>
@@ -30,6 +31,7 @@
 #include <sys/wait.h>
 #include <sys/stat.h>
 #include <stdio.h>
+#include <stdarg.h>
 #include <signal.h>
 #define printk printf
 
@@ -37,6 +39,9 @@
 #include <linux/lustre_idl.h>
 #include <linux/lustre_dlm.h>
 
+#include <libxml/xmlmemory.h>
+#include <libxml/parser.h>
+
 #include <unistd.h>
 #include <sys/un.h>
 #include <time.h>
 #include "parser.h"
 #include <stdio.h>
 
+static int jt_newdev(int argc, char **argv);
+static int jt_attach(int argc, char **argv);
+static int jt_setup(int argc, char **argv);
+
+
 int fd = -1;
 int connid = -1;
 char rawbuf[8192];
@@ -147,14 +157,15 @@ int getfd(char *func)
         return 0;
 }
 
-/*
+#if 1
 #define difftime(a, b)                                          \
         ((double)(a)->tv_sec - (b)->tv_sec +                    \
          ((double)((a)->tv_usec - (b)->tv_usec) / 1000000))
-*/
+#else
 
 #define difftime(a, b)  (((a)->tv_sec - (b)->tv_sec) + \
                         (((a)->tv_usec - (b)->tv_usec) / 1000000))
+#endif
 
 static int be_verbose(int verbose, struct timeval *next_time,
                       int num, int *next_num, int num_total)
@@ -236,6 +247,449 @@ static int do_disconnect(char *func, int verbose)
 
 extern command_t cmdlist[];
 
+static int xml_command(char *cmd, ...) {
+        va_list args;
+        char *arg, *cmds[8];
+        int i = 1, j;
+       
+        cmds[0] = cmd; 
+        va_start(args, cmd);
+
+        while (((arg = va_arg(args, char *)) != NULL) && (i < 8)) {
+                cmds[i] = arg;
+                i++;
+        }
+
+        va_end(args);
+
+        printf("obdctl > ");
+        for (j = 0; j < i; j++)
+          printf("%s ", cmds[j]);
+
+        printf("\n");
+
+        return Parser_execarg(i, cmds, cmdlist);
+}
+
+static network_t *xml_network(xmlDocPtr doc, xmlNodePtr root) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        network_t *net;
+        
+        if ((net = (network_t *)calloc(1, sizeof(network_t))) == NULL) {
+                printf("error: unable to malloc network_t\n");
+                return NULL;
+        }
+        
+        net->type = xmlGetProp(root, "type");
+        if (net->type == NULL) {
+                printf("error: type attrib required (tcp, elan, myrinet)\n");
+                free(net);
+                return NULL;
+        }
+
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "server"))
+                        net->server = xmlNodeGetContent(cur);
+
+                if (!xmlStrcmp(cur->name, "port"))
+                        net->port = atoi(xmlNodeGetContent(cur));
+
+                cur = cur->next;
+        } 
+
+        if (net->server == NULL) {
+                printf("error: <server> tag required\n");
+                free(net);
+                return NULL;
+        }
+        
+        return net;
+}
+
+static int xml_mds(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        char *fstype = NULL, *device = NULL;
+        int rc;
+
+        printf("--- Setting up MDS ---\n");
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "fstype"))
+                        fstype = xmlNodeGetContent(cur);
+
+                if (!xmlStrcmp(cur->name, "device"))
+                        device = xmlNodeGetContent(cur);
+                cur = cur->next;
+        } 
+
+        if ((fstype == NULL) || (device == NULL)) {
+                printf("error: <fstype> and <device> tags required\n");
+                return -1;
+        }
+
+        if ((rc = xml_command("newdev", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("attach", "mds", "MDSDEV", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("setup", device, fstype, NULL)) != 0)
+                return rc;
+
+        return 0;
+}
+        
+static int xml_obd(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        char *obdtype, *format = NULL, *fstype = NULL, *device = NULL;
+        int rc;
+
+        obdtype = xmlGetProp(root, "type");
+        printf("--- Setting up OBD ---\n");
+
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "fstype"))
+                        fstype = xmlNodeGetContent(cur);
+
+                if (!xmlStrcmp(cur->name, "device"))
+                        device = xmlNodeGetContent(cur);
+                if (!xmlStrcmp(cur->name, "autoformat"))
+                        format = xmlNodeGetContent(cur);
+
+                cur = cur->next;
+        } 
+
+        if ((obdtype == NULL) || (fstype == NULL) || (device == NULL)) {
+                printf("error: 'type' attrib and <fstype> and <device> tags required\n");
+                return -1;
+        }
+
+        /* FIXME: Building and configuring loopback devices should go here
+         * but is currently unsupported.  You'll have to use the scripts
+         * for now until support is added, or specify a real device.
+         */
+        
+        if ((rc = xml_command("newdev", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("attach", obdtype, "OBDDEV", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("setup", device, fstype, NULL)) != 0)
+                return rc;
+
+        return 0;
+}
+
+static int xml_ost(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        int rc;
+
+        printf("--- Setting up OST ---\n");
+        if ((rc = xml_command("newdev", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("attach", "ost", "OSTDEV", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("setup", "$OBDDEV", NULL)) != 0)
+                return rc;
+
+        return 0;
+}
+
+static int xml_osc(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        network_t *net = NULL;
+        int rc = 0;
+
+        printf("--- Setting up OSC ---\n");
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "network")) {
+                        net = xml_network(doc, cur);
+                        if (net == NULL)
+                                return -1;
+                }
+                cur = cur->next;
+        } 
+
+        if (net == NULL) {
+                printf("error: <network> tag required\n");
+                return -1;
+        }
+
+        if ((rc = xml_command("newdev", NULL)) != 0) {
+                rc = -1;
+                goto xml_osc_error;
+        }
+
+        if ((rc = xml_command("attach", "osc", "OSCDEV", NULL)) != 0) {
+                rc = -1;
+                goto xml_osc_error;
+        }
+
+        if ((rc = xml_command("setup", "OSTDEV", net->server, NULL)) != 0) {
+                rc = -1;
+                goto xml_osc_error;
+        }
+
+xml_osc_error:
+        free(net);
+        return rc;
+}
+
+static int xml_mdc(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        network_t *net = NULL;
+        int rc = 0;
+
+        printf("--- Setting up MDC ---\n");
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "network")) {
+                        net = xml_network(doc, cur);
+                        if (net == NULL)
+                                return -1;
+                }
+                cur = cur->next;
+        } 
+
+        if (net == NULL) {
+                printf("error: <network> tag required\n");
+                return -1;
+        }
+
+        if ((rc = xml_command("newdev", NULL)) != 0) {
+                rc = -1;
+                goto xml_mdc_error;
+        }
+
+        if ((rc = xml_command("attach", "mdc", "MDCDEV", NULL)) != 0) {
+                rc = -1;
+                goto xml_mdc_error;
+        }
+
+        if ((rc = xml_command("setup", "MDSDEV", net->server, NULL)) != 0) {
+                rc = -1;
+                goto xml_mdc_error;
+        }
+
+xml_mdc_error:
+        free(net);
+        return rc;
+}
+
+static int xml_lov(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        printf("--- Setting up LOV ---\n");
+        return 0;
+}
+
+static int xml_router(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        printf("--- Setting up ROUTER ---\n");
+        return 0;
+}
+
+static int xml_ldlm(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        int rc;
+
+        printf("--- Setting up LDLM ---\n");
+        if ((rc = xml_command("newdev", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("attach", "ldlm", "LDLMDEV", NULL)) != 0)
+                return rc;
+
+        if ((rc = xml_command("setup", NULL)) != 0)
+                return rc;
+
+        return 0;
+}
+
+static int xml_service(xmlDocPtr doc, xmlNodePtr root, 
+                       int serv_num, char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root;
+        char *id, *uuid;
+
+        while (cur != NULL) {
+                id = xmlGetProp(cur, "id");
+                uuid = xmlGetProp(cur, "uuid");
+
+                if (xmlStrcmp(id, serv_id) ||
+                    xmlStrcmp(uuid, serv_uuid)) {
+                        cur = cur->next;
+                        continue;
+                }
+
+                if (!xmlStrcmp(cur->name, "mds"))
+                        return xml_mds(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "obd"))
+                        return xml_obd(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "ost"))
+                        return xml_ost(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "osc"))
+                        return xml_osc(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "mdc"))
+                        return xml_mdc(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "lov"))
+                        return xml_lov(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "router"))
+                        return xml_router(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "ldlm"))
+                        return xml_ldlm(doc, cur, serv_id, serv_uuid);
+                else
+                        return -1;        
+
+                cur = cur->next;
+        }
+
+        printf("error: No XML config branch for id=%s uuid=%s\n",
+                serv_id, serv_uuid); 
+        return -1; 
+}
+
+static int xml_profile(xmlDocPtr doc, xmlNodePtr root, 
+                       int prof_num, char *prof_id, char *prof_uuid) {
+        xmlNodePtr parent, cur = root;
+        char *id, *uuid, *srv_id, *srv_uuid;
+        int rc = 0, num;
+
+        while (cur != NULL) {
+                id = xmlGetProp(cur, "id");
+                uuid = xmlGetProp(cur, "uuid");
+
+                if (xmlStrcmp(cur->name, "profile") || 
+                    xmlStrcmp(id, prof_id)          ||
+                    xmlStrcmp(uuid, prof_uuid)) {
+                        cur = cur->next;
+                        continue;
+                }
+
+                /* FIXME: Doesn't understand mountpoints yet
+                 *        xml_mountpoint(doc, root, ...);
+                 */    
+
+                /* Setup each service in turn
+                 * FIXME: Should be sorted by "num" attr, we shouldn't
+                 *        assume they're in order in the XML document.
+                 */
+                parent = cur;
+                cur = cur->xmlChildrenNode;
+                while (cur != NULL) {
+                        if (!xmlStrcmp(cur->name, "service_id")) {
+                                num = atoi(xmlGetProp(cur, "num"));
+                                rc = xml_service(doc, root, num,
+                                        srv_id = xmlGetProp(cur, "id"),
+                                        srv_uuid = xmlGetProp(cur, "uuid"));
+                                if (rc != 0) {
+                                        printf("error: service config\n");
+                                        return rc;
+                                }
+                        }
+
+                        cur = cur->next;
+                }
+
+                cur = parent->next;
+        }
+
+        return rc; 
+}
+
+static int xml_node(xmlDocPtr doc, xmlNodePtr root) {
+        xmlNodePtr parent, cur = root;
+        char *id, *uuid;
+        int rc = 0, num;
+        
+        /* Walk the node tags looking for ours */
+        while (cur != NULL) {
+                if (xmlStrcmp(cur->name, "node")) {
+                        cur = cur->next;
+                        continue;
+                }
+
+                id = xmlGetProp(cur, "id");
+                if (id == NULL)
+                        return -1;
+
+                uuid = xmlGetProp(cur, "uuid");
+                if (uuid == NULL)
+                        return -1;
+
+                /* FIXME: Verify our ID and UUID against /etc/lustre/id
+                 *        so we're sure we are who we think we are.
+                 */
+
+                /* Setup each profile in turn
+                 * FIXME: Should be sorted by "num" attr, we shouldn't
+                 *        assume they're in order in the XML document.
+                 */
+                parent = cur;
+                cur = cur->xmlChildrenNode;
+                while (cur != NULL) {
+                        if (!xmlStrcmp(cur->name, "profile_id")) {
+                                num = atoi(xmlGetProp(cur, "num"));
+                                rc = xml_profile(doc, root, num,
+                                                 xmlGetProp(cur, "id"),
+                                                 xmlGetProp(cur, "uuid"));
+                                if (rc != 0)
+                                        return rc;
+                        }
+
+                        cur = cur->next;
+                }
+
+                cur = parent->next;
+        }
+
+        return rc;
+}
+
+static int do_xml(char *func, char *file)
+{
+        xmlDocPtr doc;
+        xmlNodePtr cur;
+        int rc;
+
+        doc = xmlParseFile(file);
+        if (doc == NULL) {
+                fprintf(stderr, "error: Unable to parse XML\n");
+                return -1; 
+        }
+
+        cur = xmlDocGetRootElement(doc);
+        if (cur == NULL) {
+                fprintf(stderr, "error: Empty XML\n");
+                xmlFreeDoc(doc);
+                return -1;
+        }
+        
+        if (xmlStrcmp(cur->name, (const xmlChar *)"lustre")) {
+                fprintf(stderr, "error: Root node != <lustre>\n");
+                xmlFreeDoc(doc);
+                return -1;
+        }
+
+        /* FIXME: Validate the XML against the DTD here */
+    
+        /* FIXME: Merge all the text nodes under each branch and 
+         *        prune empty nodes.  Just to make the parsing more
+         *        tolerant, the exact location of nested tags isn't
+         *        critical for this.
+         */
+        
+        rc = xml_node(doc, cur->xmlChildrenNode);
+        xmlFreeDoc(doc);
+
+        return rc;
+}
+
 static int do_device(char *func, int dev)
 {
         struct obd_ioctl_data data;
@@ -306,6 +760,17 @@ static int jt_disconnect(int argc, char **argv)
         return do_disconnect(argv[0], 0);
 }
 
+static int jt__xml(int argc, char **argv)
+{
+        if (argc < 2) {
+                fprintf(stderr, "usage: %s <xml file> <command [args ...]>\n",
+                        cmdname(argv[0]));
+                return -1;
+        }
+
+        return do_xml("xml", argv[1]);
+}
+
 static int jt__device(int argc, char **argv)
 {
         char *arg2[3];
@@ -942,6 +1407,7 @@ static int jt_quit(int argc, char **argv)
 
 command_t cmdlist[] = {
         /* Metacommands */
+        {"--xml", jt__xml, 0, "--xml <xml file> <command [args ...]>"},
         {"--device", jt__device, 0, "--device <devno> <command [args ...]>"},
         {"--threads", jt__threads, 0,
                 "--threads <threads> <devno> <command [args ...]>"},