Whamcloud - gitweb
Barebones XML support added to obdctl via the --xml option. Validation of
[fs/lustre-release.git] / lustre / utils / obdctl.c
index 31fbfc3..66bb5c0 100644 (file)
@@ -22,6 +22,7 @@
  *
  */
 
+
 #include <stdlib.h>
 #include <sys/ioctl.h>
 #include <fcntl.h>
@@ -37,6 +38,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];
 char *buf = rawbuf;
 int max = 8192;
 int thread;
+int rc = 0;
 
 #define IOCINIT(data)                                                   \
 do {                                                                    \
@@ -146,9 +156,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)
@@ -193,7 +209,14 @@ static int get_verbose(const char *arg)
         else if (arg[0] == 's' || arg[0] == 'q')
                 verbose = 0;
         else
-                verbose = strtoul(arg, NULL, 0);
+                verbose = (int) strtoul(arg, NULL, 0);
+
+        if (verbose < 0)
+                printf("Print status every %d seconds\n", -verbose);
+        else if (verbose == 1)
+                printf("Print status every operation\n");
+        else if (verbose > 1)
+                printf("Print status every %d operations\n", verbose);
 
         return verbose;
 }
@@ -201,7 +224,6 @@ static int get_verbose(const char *arg)
 static int do_disconnect(char *func, int verbose)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         if (connid == -1)
                 return 0;
@@ -224,6 +246,286 @@ static int do_disconnect(char *func, int verbose)
 
 extern command_t cmdlist[];
 
+/* Given a XML document and the root of a mds branch do the setup */
+static int xml_mds(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        char *fstype = NULL, *device = NULL;
+        char *newdev[1] = { "newdev" };
+        char *attach[3] = { "attach", "mds", "MDSDEV" };
+        char *setup[3] = { "setup", NULL, NULL };
+        int rc;
+
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "fstype"))
+                        fstype = cur->content;
+
+                if (!xmlStrcmp(cur->name, "device"))
+                        device = cur->content;
+        } 
+
+        if ((fstype == NULL) || (device == NULL))
+                return -1;
+
+        /* Invoke commands as if passed interactively */
+        rc = jt_newdev(1, newdev);
+        if (rc != 0)
+                return rc;
+        
+        rc = jt_attach(3, attach);
+        if (rc != 0)
+                return rc;
+
+        setup[1] = device;
+        setup[2] = fstype;
+        rc = jt_setup(3, setup);
+        if (rc != 0)
+                return rc;
+
+        return 0;
+}
+        
+/* Given a XML document and the root of a obd branch do the setup */
+static int xml_obd(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        /* FIXME: Fill in the guts... */
+        return 0;
+}
+
+/* Given a XML document and the root of a ost branch do the setup */
+static int xml_ost(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        /* FIXME: Fill in the guts... */
+        return 0;
+}
+
+/* Given a XML document and the root of a osc branch do the setup */
+static int xml_osc(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        xmlNodePtr cur = root->xmlChildrenNode;
+        char *type = NULL, *address = NULL;
+        char *newdev[1] = { "newdev" };
+        char *attach[3] = { "attach", "osc", "OSCDEV" };
+        char *setup[2] = { "setup", "-1" };
+        int rc;
+
+        while (cur != NULL) {
+                if (!xmlStrcmp(cur->name, "network")) {
+                        type = xmlGetProp(cur, "type");
+                        if (type == NULL)
+                                return -1;
+
+                        address = xmlGetProp(cur, "address");
+                        if (address == NULL)
+                                return -1;
+
+                        /* FIXME: Do the portals setup here as well? */
+                }
+                        
+                cur = cur->next;
+        } 
+
+        /* Invoke commands as if passed interactively */
+        rc = jt_newdev(1, newdev);
+        if (rc != 0)
+                return rc;
+        
+        rc = jt_attach(2, attach);
+        if (rc != 0)
+                return rc;
+        
+        rc = jt_setup(1, setup);
+        if (rc != 0)
+                return rc;
+
+        return 0;
+}
+
+/* Given a XML document and the root of a lov branch do the setup */
+static int xml_lov(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        /* FIXME: Fill in the guts... */
+        return 0;
+}
+
+/* Given a XML document and the root of a router branch do the setup */
+static int xml_router(xmlDocPtr doc, xmlNodePtr root, 
+                   char *serv_id, char *serv_uuid) {
+        /* FIXME: Fill in the guts... */
+        return 0;
+}
+
+/* Given a XML document and service id/uuid setup the service */
+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"))
+                        rc = xml_mds(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "obd"))
+                        rc = xml_obd(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "ost"))
+                        rc = xml_ost(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "osc"))
+                        rc = xml_osc(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "lov"))
+                        rc = xml_lov(doc, cur, serv_id, serv_uuid);
+                else if (!xmlStrcmp(cur->name, "router"))
+                        rc = xml_router(doc, cur, serv_id, serv_uuid);
+                else
+                        return -1;        
+
+                cur = cur->next;
+        }
+
+        return rc; 
+}
+
+/* Given a XML document and profile id/uuid setup the profile */
+static int xml_profile(xmlDocPtr doc, xmlNodePtr root, 
+                       int prof_num, char *prof_id, char *prof_uuid) {
+        xmlNodePtr parent, cur = root;
+        char *id, *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,
+                                                 xmlGetProp(cur, "id"),
+                                                 xmlGetProp(cur, "uuid"));
+                                if (rc != 0)
+                                        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. 
+         */
+        
+        rc = xml_node(doc, cur->xmlChildrenNode);
+        xmlFreeDoc(doc);
+
+        return rc;
+}
+
 static int do_device(char *func, int dev)
 {
         struct obd_ioctl_data data;
@@ -245,8 +547,6 @@ static int do_device(char *func, int dev)
 
 static int jt_device(int argc, char **argv)
 {
-        int rc;
-
         do_disconnect(argv[0], 1);
 
         if (argc != 2) {
@@ -266,7 +566,6 @@ static int jt_device(int argc, char **argv)
 static int jt_connect(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
 
@@ -297,10 +596,21 @@ 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];
-        int rc, ret;
+        int ret;
 
         if (argc < 3) {
                 fprintf(stderr, "usage: %s devno <command [args ...]>\n",
@@ -331,7 +641,6 @@ static int jt__threads(int argc, char **argv)
         int threads, next_thread;
         int verbose;
         int i, j;
-        int rc;
 
         if (argc < 5) {
                 fprintf(stderr,
@@ -402,7 +711,6 @@ static int jt__threads(int argc, char **argv)
 static int jt_detach(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
 
@@ -427,7 +735,6 @@ static int jt_detach(int argc, char **argv)
 static int jt_cleanup(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
 
@@ -447,7 +754,6 @@ static int jt_cleanup(int argc, char **argv)
 static int jt_newdev(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         if (getfd(argv[0]))
                 return -1;
@@ -473,7 +779,6 @@ static int jt_newdev(int argc, char **argv)
 static int jt_attach(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
 
@@ -499,16 +804,16 @@ static int jt_attach(int argc, char **argv)
         if (rc < 0)
                 fprintf(stderr, "error: %s: %x %s\n", cmdname(argv[0]),
                         OBD_IOC_ATTACH, strerror(rc = errno));
-        else if (argc == 3) { 
+        else if (argc == 3) {
                 char name[1024];
-                if (strlen(argv[2]) > 128) { 
+                if (strlen(argv[2]) > 128) {
                         printf("Name too long to set environment\n");
                         return -EINVAL;
                 }
                 snprintf(name, 512, "LUSTRE_DEV_%s", argv[2]);
-                rc = setenv(name, argv[1], 1); 
-                if (rc) { 
-                        printf("error setting env variable %s\n", name); 
+                rc = setenv(name, argv[1], 1);
+                if (rc) {
+                        printf("error setting env variable %s\n", name);
                 }
         }
 
@@ -520,7 +825,6 @@ static int jt_attach(int argc, char **argv)
 static int do_name2dev(char *func, char *name)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         if (getfd(func))
                 return -1;
@@ -548,8 +852,6 @@ static int do_name2dev(char *func, char *name)
 
 static int jt_name2dev(int argc, char **argv)
 {
-        int rc;
-
         if (argc != 2) {
                 fprintf(stderr, "usage: %s name\n", cmdname(argv[0]));
                 return -1;
@@ -568,7 +870,6 @@ static int jt_name2dev(int argc, char **argv)
 static int jt_setup(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
 
@@ -617,7 +918,6 @@ static int jt_create(int argc, char **argv)
         int count = 1, next_count;
         int verbose;
         int i;
-        int rc;
 
         IOCINIT(data);
         if (argc < 2 || argc > 4) {
@@ -656,7 +956,6 @@ static int jt_create(int argc, char **argv)
 static int jt_setattr(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
         if (argc != 2) {
@@ -666,7 +965,7 @@ static int jt_setattr(int argc, char **argv)
 
         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
         data.ioc_obdo1.o_mode = S_IFREG | strtoul(argv[2], NULL, 0);
-        data.ioc_obdo1.o_valid = OBD_MD_FLMODE; 
+        data.ioc_obdo1.o_valid = OBD_MD_FLMODE;
 
         rc = ioctl(fd, OBD_IOC_SETATTR , &data);
         if (rc < 0)
@@ -679,7 +978,6 @@ static int jt_setattr(int argc, char **argv)
 static int jt_destroy(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
         if (argc != 2) {
@@ -701,7 +999,6 @@ static int jt_destroy(int argc, char **argv)
 static int jt_getattr(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         if (argc != 2) {
                 fprintf(stderr, "usage: %s id\n", cmdname(argv[0]));
@@ -710,7 +1007,7 @@ static int jt_getattr(int argc, char **argv)
 
         IOCINIT(data);
         data.ioc_obdo1.o_id = strtoul(argv[1], NULL, 0);
-        /* to help obd filter */ 
+        /* to help obd filter */
         data.ioc_obdo1.o_mode = 0100644;
         data.ioc_obdo1.o_valid = 0xffffffff;
         printf("%s: object id %Ld\n", cmdname(argv[0]), data.ioc_obdo1.o_id);
@@ -732,7 +1029,6 @@ static int jt_test_getattr(int argc, char **argv)
         struct timeval start, next_time;
         int i, count, next_count;
         int verbose;
-        int rc;
 
         if (argc != 2 && argc != 3) {
                 fprintf(stderr, "usage: %s count [verbose]\n",cmdname(argv[0]));
@@ -742,7 +1038,10 @@ static int jt_test_getattr(int argc, char **argv)
         IOCINIT(data);
         count = strtoul(argv[1], NULL, 0);
 
-        verbose = get_verbose(argv[2]);
+        if (argc == 3)
+                verbose = get_verbose(argv[2]);
+        else
+                verbose = 1;
 
         data.ioc_obdo1.o_valid = 0xffffffff;
         data.ioc_obdo1.o_id = 2;
@@ -758,9 +1057,11 @@ static int jt_test_getattr(int argc, char **argv)
                         fprintf(stderr, "error: %s: #%d - %s\n",
                                 cmdname(argv[0]), i, strerror(rc = errno));
                         break;
-                } else if (be_verbose(verbose, &next_time, i,&next_count,count))
+                } else {
+                        if (be_verbose(verbose, &next_time, i,&next_count,count))
                         printf("%s: got attr #%d\n", cmdname(argv[0]), i);
-        }
+               }
+       }
 
         if (!rc) {
                 struct timeval end;
@@ -787,7 +1088,6 @@ static int jt_test_brw(int argc, char **argv)
         int verbose = 1, write = 0, rw;
         int i, o, p;
         int len;
-        int rc;
 
         if (argc < 2 || argc > 6) {
                 fprintf(stderr,
@@ -834,7 +1134,7 @@ static int jt_test_brw(int argc, char **argv)
         data.ioc_plen1 = len;
         data.ioc_pbuf1 = bulk;
         if (obdos > 1) {
-                data.ioc_obdo2.o_id = 2;
+                data.ioc_obdo2.o_id = 3;
                 data.ioc_plen2 = len;
                 data.ioc_pbuf2 = bulk + len;
         }
@@ -851,6 +1151,10 @@ static int jt_test_brw(int argc, char **argv)
          * We will put in the start time (and loop count inside the loop)
          * at the beginning of each page so that we will be able to validate
          * (at some later time) whether the data actually made it or not.
+         *
+         * XXX we do not currently use any of this memory in OBD_IOC_BRW_*
+         *     just to avoid the overhead of the copy_{to,from}_user.  It
+         *     can be fixed if we ever need to send real data around.
          */
         for (o = 0, b = bulk; o < obdos; o++)
                 for (p = 0; p < pages; p++, b += PAGE_SIZE)
@@ -898,7 +1202,6 @@ static int jt_test_brw(int argc, char **argv)
 static int jt_test_ldlm(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
         if (argc != 1) {
@@ -916,7 +1219,6 @@ static int jt_test_ldlm(int argc, char **argv)
 static int jt_newconn(int argc, char **argv)
 {
         struct obd_ioctl_data data;
-        int rc;
 
         IOCINIT(data);
         if (argc != 1) {
@@ -924,7 +1226,7 @@ static int jt_newconn(int argc, char **argv)
                 return -1;
         }
 
-        rc = ioctl(fd, OBD_RECOVD_NEWCONN , &data);
+        rc = ioctl(fd, OBD_IOC_RECOVD_NEWCONN , &data);
         if (rc < 0)
                 fprintf(stderr, "error: %s: %s\n", cmdname(argv[0]),
                         strerror(rc = errno));
@@ -932,8 +1234,16 @@ static int jt_newconn(int argc, char **argv)
         return rc;
 }
 
+static int jt_quit(int argc, char **argv)
+{
+        Parser_quit(argc, argv);
+
+        return rc;
+}
+
 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 ...]>"},
@@ -964,15 +1274,15 @@ command_t cmdlist[] = {
 
         /* User interface commands */
         {"help", Parser_help, 0, "help"},
-        {"exit", Parser_quit, 0, "quit"},
-        {"quit", Parser_quit, 0, "quit"},
+        {"exit", jt_quit, 0, "quit"},
+        {"quit", jt_quit, 0, "quit"},
         { 0, 0, 0, NULL }
 };
 
 
 static void signal_server(int sig)
 {
-        if (sig == SIGINT) { 
+        if (sig == SIGINT) {
                 do_disconnect("sigint", 1);
                 exit(1);
         } else {
@@ -983,7 +1293,6 @@ static void signal_server(int sig)
 int main(int argc, char **argv)
 {
         struct sigaction sigact;
-        int rc = 0;
 
         sigact.sa_handler = signal_server;
         sigfillset(&sigact.sa_mask);
@@ -995,7 +1304,7 @@ int main(int argc, char **argv)
                 rc = Parser_execarg(argc - 1, argv + 1, cmdlist);
         } else {
                 Parser_init("obdctl > ", cmdlist);
-                Parser_commands();
+                rc = Parser_commands();
         }
 
         do_disconnect(argv[0], 1);