/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 *
 *  Copyright (C) 2002 Cluster File Systems, Inc.
 *   Author: Peter J. Braam <braam@clusterfs.com>
 *   Author: Phil Schwan <phil@clusterfs.com>
 *   Author: Andreas Dilger <adilger@clusterfs.com>
 *   Author: Robert Read <rread@clusterfs.com>
 *
 *   This file is part of Lustre, http://www.lustre.org.
 *
 *   Lustre is free software; you can redistribute it and/or
 *   modify it under the terms of version 2 of the GNU General Public
 *   License as published by the Free Software Foundation.
 *
 *   Lustre is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with Lustre; if not, write to the Free Software
 *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */


#include <stdlib.h>
#include <sys/ioctl.h>
#include <stdio.h>
#include <stdarg.h>

#ifndef __KERNEL__
#include <liblustre.h>
#endif
#include <linux/lustre_lib.h>
#include <linux/lustre_cfg.h>
#include <linux/lustre_idl.h>
#include <linux/lustre_dlm.h>
#include <linux/obd.h>          /* for struct lov_stripe_md */
#include <linux/lustre_build_version.h>

#include <unistd.h>
#include <sys/un.h>
#include <time.h>
#include <sys/time.h>
#include <errno.h>
#include <string.h>


#include "obdctl.h"
#include <portals/ptlctl.h>
#include "parser.h"
#include <stdio.h>

static char * lcfg_devname;

void lcfg_set_devname(char *name)
{
	if (lcfg_devname)
		free(lcfg_devname);
	lcfg_devname = strdup(name);
}


int jt_lcfg_device(int argc, char **argv)
{
	char *name;

        if (argc == 1) {
		printf("current device is %s\n", lcfg_devname? : "not set");
		return 0;
	} else if (argc != 2) {
                return CMD_HELP;
	}

	name = argv[1];

	/* quietly strip the unnecessary '$' */
	if (*name == '$')
		name++;

	lcfg_set_devname(name);

        return 0;
}

/* NOOP */
int jt_lcfg_newdev(int argc, char **argv)
{
        return 0;
}

int jt_lcfg_attach(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        int rc;

        LCFG_INIT(lcfg, LCFG_ATTACH, lcfg_devname);

        if (argc != 2 && argc != 3 && argc != 4)
                return CMD_HELP;

        lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
        lcfg.lcfg_inlbuf1 = argv[1];
        if (argc >= 3) {
                lcfg.lcfg_dev_namelen = strlen(argv[2]) + 1;
                lcfg.lcfg_dev_name = argv[2];
        } else {
                fprintf(stderr, "error: %s: LCFG_ATTACH requires a name\n",
                        jt_cmdname(argv[0]));
		return -EINVAL;
	}

        if (argc == 4) {
                lcfg.lcfg_inllen2 = strlen(argv[3]) + 1;
                lcfg.lcfg_inlbuf2 = argv[3];
        }

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: LCFG_ATTACH %s\n",
                        jt_cmdname(argv[0]), strerror(rc = errno));
        } else if (argc == 3) {
                char name[1024];

		lcfg_set_devname(argv[2]);
                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);
                }
        } else {
		lcfg_set_devname(argv[2]);
	}

        return rc;
}

int jt_lcfg_setup(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        int rc;

        if (lcfg_devname == NULL) {
                fprintf(stderr, "%s: please use 'cfg_device name' to set the "
                        "device name for config commands.\n",
                        jt_cmdname(argv[0]));
		return -EINVAL;
        }

        LCFG_INIT(lcfg, LCFG_SETUP, lcfg_devname);

        if (argc > 7)
                return CMD_HELP;

        if (argc > 1) {
                lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
                lcfg.lcfg_inlbuf1 = argv[1];
        }
        if (argc > 2) {
                lcfg.lcfg_inllen2 = strlen(argv[2]) + 1;
                lcfg.lcfg_inlbuf2 = argv[2];
        }
        if (argc > 3) {
                lcfg.lcfg_inllen3 = strlen(argv[3]) + 1;
                lcfg.lcfg_inlbuf3 = argv[3];
        }
        if (argc > 4) {
                lcfg.lcfg_inllen4 = strlen(argv[4]) + 1;
                lcfg.lcfg_inlbuf4 = argv[4];
        }
        if (argc > 5) {
                lcfg.lcfg_inllen5 = strlen(argv[5]) + 1;
                lcfg.lcfg_inlbuf5 = argv[5];
        }
        if (argc > 6) {
                lcfg.lcfg_inllen6 = strlen(argv[6]) + 1;
                lcfg.lcfg_inlbuf6 = argv[6];
        }
        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0)
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));

        return rc;
}

int jt_obd_detach(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        int rc;

        if (lcfg_devname == NULL) {
                fprintf(stderr, "%s: please use 'cfg_device name' to set the "
                        "device name for config commands.\n",
                        jt_cmdname(argv[0]));
		return -EINVAL;
        }

        LCFG_INIT(lcfg, LCFG_DETACH, lcfg_devname);

        if (argc != 1)
                return CMD_HELP;

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0)
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));

        return rc;
}

int jt_obd_cleanup(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        char force = 'F';
        char failover = 'A';
        char flags[3];
        int flag_cnt = 0, n;
        int rc;

        if (lcfg_devname == NULL) {
                fprintf(stderr, "%s: please use 'cfg_device name' to set the "
                        "device name for config commands.\n",
                        jt_cmdname(argv[0]));
		return -EINVAL;
        }

        LCFG_INIT(lcfg, LCFG_CLEANUP, lcfg_devname);

        if (argc < 1 || argc > 3)
                return CMD_HELP;

        for (n = 1; n < argc; n++)
                if (strcmp(argv[n], "force") == 0) {
                        flags[flag_cnt++] = force;
                } else if (strcmp(argv[n], "failover") == 0) {
                        flags[flag_cnt++] = failover;
                } else {
                        fprintf(stderr, "unknown option: %s", argv[n]);
                        return CMD_HELP;
                }

	lcfg.lcfg_inllen1 = flag_cnt;
        if (flag_cnt)
                lcfg.lcfg_inlbuf1 = flags;

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0)
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));

        return rc;
}

static
int do_add_uuid(char * func, char *uuid, ptl_nid_t nid, int nal)
{
        char tmp[64];
        int rc;
        struct lustre_cfg lcfg;

        LCFG_INIT(lcfg, LCFG_ADD_UUID, lcfg_devname);
        lcfg.lcfg_nid = nid;
        lcfg.lcfg_inllen1 = strlen(uuid) + 1;
        lcfg.lcfg_inlbuf1 = uuid;
        lcfg.lcfg_nal = nal;

        rc = lcfg_ioctl(func, OBD_DEV_ID, &lcfg);
        if (rc) {
                fprintf(stderr, "IOC_PORTAL_ADD_UUID failed: %s\n",
                        strerror(errno));
                return -1;
        }

        printf ("Added uuid %s: %s\n", uuid, ptl_nid2str (tmp, nid));
        return 0;
}

int jt_lcfg_add_uuid(int argc, char **argv)
{
        ptl_nid_t nid = 0;
        int nal;

        if (argc != 4) {
                return CMD_HELP;
        }

        if (ptl_parse_nid (&nid, argv[2]) != 0) {
                fprintf (stderr, "Can't parse NID %s\n", argv[2]);
                        return (-1);
        }

        nal = ptl_name2nal(argv[3]);

        if (nal <= 0) {
                fprintf (stderr, "Can't parse NAL %s\n", argv[3]);
                return -1;
        }

        return do_add_uuid(argv[0], argv[1], nid, nal);
}

int obd_add_uuid(char *uuid, ptl_nid_t nid, int nal)
{
        return do_add_uuid("obd_add_uuid", uuid, nid, nal);
}

int jt_lcfg_del_uuid(int argc, char **argv)
{
        int rc;
        struct lustre_cfg lcfg;

        if (argc != 2) {
                fprintf(stderr, "usage: %s <uuid>\n", argv[0]);
                return 0;
        }

        LCFG_INIT(lcfg, LCFG_DEL_UUID, lcfg_devname);

        if (strcmp (argv[1], "_all_"))
        {
                lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
                lcfg.lcfg_inlbuf1 = argv[1];
        }

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc) {
                fprintf(stderr, "IOC_PORTAL_DEL_UUID failed: %s\n",
                        strerror(errno));
                return -1;
        }
        return 0;
}

int jt_lcfg_lov_setup(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        struct lov_desc desc;
        int rc;
        char *end;
                                                                                                                                                                                                     
        /* argv: lov_setup <LOV uuid> <stripe count> <stripe size>
         *                 <stripe offset> <pattern> [ <max tgt index> ]
         */
        if (argc <= 6)
                return CMD_HELP;
                                                                                                                                                                                                     
        if (strlen(argv[1]) > sizeof(desc.ld_uuid) - 1) {
                fprintf(stderr,
                        "error: %s: LOV uuid '%s' longer than "LPSZ" chars\n",
                        jt_cmdname(argv[0]), argv[1], sizeof(desc.ld_uuid) - 1);
                return -EINVAL;
        }
                                                                                                                                                                                                     
        memset(&desc, 0, sizeof(desc));
        obd_str2uuid(&desc.ld_uuid, argv[1]);
        desc.ld_default_stripe_count = strtoul(argv[2], &end, 0);
        if (*end) {
                fprintf(stderr, "error: %s: bad default stripe count '%s'\n",
                        jt_cmdname(argv[0]), argv[2]);
                return CMD_HELP;
        }
                                                                                                                                                                                                     
        desc.ld_default_stripe_size = strtoull(argv[3], &end, 0);
        if (*end) {
                fprintf(stderr, "error: %s: bad default stripe size '%s'\n",
                        jt_cmdname(argv[0]), argv[3]);
                return CMD_HELP;
        }
        if (desc.ld_default_stripe_size < 4096) {
                fprintf(stderr,
                        "error: %s: default stripe size "LPU64" too small\n",
                        jt_cmdname(argv[0]), desc.ld_default_stripe_size);
                return -EINVAL;
        } else if ((long)desc.ld_default_stripe_size <
                   desc.ld_default_stripe_size) {
                fprintf(stderr,
                        "error: %s: default stripe size "LPU64" too large\n",
                        jt_cmdname(argv[0]), desc.ld_default_stripe_size);
                return -EINVAL;
        }
        desc.ld_default_stripe_offset = strtoull(argv[4], &end, 0);
        if (*end) {
                fprintf(stderr, "error: %s: bad default stripe offset '%s'\n",
                        jt_cmdname(argv[0]), argv[4]);
                return CMD_HELP;
        }
        desc.ld_pattern = strtoul(argv[5], &end, 0);
        if (*end) {
                fprintf(stderr, "error: %s: bad stripe pattern '%s'\n",
                        jt_cmdname(argv[0]), argv[5]);
                return CMD_HELP;
        }
                                                                                                                                                                                                     
        if (argc > 7) {
                desc.ld_tgt_count = argc - 6;
                if (desc.ld_default_stripe_count > desc.ld_tgt_count) {
                        fprintf(stderr,
                                "error: %s: default stripe count %u > "
                                "OST count %u\n", jt_cmdname(argv[0]),
                                desc.ld_default_stripe_count,
                                desc.ld_tgt_count);
                        return -EINVAL;
                }
        }
                                                                                                                                                                                                     
        LCFG_INIT(lcfg, LCFG_SETUP, lcfg_devname);
                                                                                                                                                                                                     
        lcfg.lcfg_inllen1 = sizeof(desc);
        lcfg.lcfg_inlbuf1 = (char *)&desc;
                                                                                                                                                                                                     
        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc)
                fprintf(stderr, "error: %s: ioctl error: %s\n",
                        jt_cmdname(argv[0]), strerror(rc = errno));
        return rc;
}

int jt_lcfg_lmv_setup(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        struct lmv_desc desc;
        struct obd_uuid *uuidarray, *ptr;
        int rc, i;

        LCFG_INIT(lcfg, LCFG_SETUP, lcfg_devname);

        if (argc <= 2)
                return CMD_HELP;

        if (strlen(argv[1]) > sizeof(desc.ld_uuid) - 1) {
                fprintf(stderr,
                        "error: %s: LMV uuid '%s' longer than "LPSZ" chars\n",
                        jt_cmdname(argv[0]), argv[1], sizeof(desc.ld_uuid) - 1);
                return -EINVAL;
        }

        memset(&desc, 0, sizeof(desc));
        obd_str2uuid(&desc.ld_uuid, argv[1]);
        desc.ld_tgt_count = argc - 2;
        printf("LMV: %d uuids:\n", desc.ld_tgt_count);

        /* NOTE: it is possible to overwrite the default striping parameters,
         *       but EXTREME care must be taken when saving the OST UUID list.
         *       It must be EXACTLY the same, or have only additions at the
         *       end of the list, or only overwrite individual OST entries
         *       that are restored from backups of the previous OST.
         */
        uuidarray = calloc(desc.ld_tgt_count, sizeof(*uuidarray));
        if (!uuidarray) {
                fprintf(stderr, "error: %s: no memory for %d UUIDs\n",
                        jt_cmdname(argv[0]), desc.ld_tgt_count);
                rc = -ENOMEM;
                goto out;
        }
        for (i = 2, ptr = uuidarray; i < argc; i++, ptr++) {
                if (strlen(argv[i]) >= sizeof(*ptr)) {
                        fprintf(stderr, "error: %s: arg %d (%s) too long\n",
                                jt_cmdname(argv[0]), i, argv[i]);
                        rc = -EINVAL;
                        goto out;
                }
                printf("  %s\n", argv[i]);
                strcpy((char *)ptr, argv[i]);
        }

        lcfg.lcfg_inllen1 = sizeof(desc);
        lcfg.lcfg_inlbuf1 = (char *)&desc;
        lcfg.lcfg_inllen2 = desc.ld_tgt_count * sizeof(*uuidarray);
        lcfg.lcfg_inlbuf2 = (char *)uuidarray;

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc)
                fprintf(stderr, "error: %s: ioctl error: %s\n",
                        jt_cmdname(argv[0]), strerror(rc = errno));
out:
        free(uuidarray);
        return rc;
}

int jt_lcfg_lov_modify_tgts(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        char *end;
        int index;
        int gen;
        int rc;

        /* argv: lov_modify_tgts <op> <LOV name> <OBD uuid> <index> <gen> */
        if (argc != 6)
                return CMD_HELP;

        if (!strncmp(argv[1], "add", 4)) {
                LCFG_INIT(lcfg, LCFG_LOV_ADD_OBD, argv[2]);
        } else if (!strncmp(argv[1], "del", 4)) {
                LCFG_INIT(lcfg, LCFG_LOV_DEL_OBD, argv[2]);
        } else {
                fprintf(stderr, "error: %s: bad operation '%s'\n",
                        jt_cmdname(argv[0]), argv[1]);
                return CMD_HELP;
        }

        lcfg.lcfg_inlbuf1 = argv[3];
        lcfg.lcfg_inllen1 = strlen(lcfg.lcfg_inlbuf1) + 1;
        if (lcfg.lcfg_inllen1 > sizeof(struct obd_uuid)) {
                fprintf(stderr,
                        "error: %s: OBD uuid '%s' longer than "LPSZ" chars\n",
                        jt_cmdname(argv[0]), argv[3],
                        sizeof(struct obd_uuid) - 1);
                return -EINVAL;
        }

        index = strtoul(argv[4], &end, 0);
        if (*end) {
                fprintf(stderr, "error: %s: bad OBD index '%s'\n",
                        jt_cmdname(argv[0]), argv[4]);
                return CMD_HELP;
        }
        lcfg.lcfg_inlbuf2 = argv[4];
        lcfg.lcfg_inllen2 = strlen(lcfg.lcfg_inlbuf2);

        gen = strtoul(argv[5], &end, 0);
        if (*end) {
                fprintf(stderr, "error: %s: bad OBD generation '%s'\n",
                        jt_cmdname(argv[0]), argv[5]);
                return CMD_HELP;
        }
        lcfg.lcfg_inlbuf3 = argv[5];
        lcfg.lcfg_inllen3 = strlen(lcfg.lcfg_inlbuf3);

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc)
                fprintf(stderr, "error: %s: ioctl error: %s\n",
                        jt_cmdname(argv[0]), strerror(rc = errno));

        return rc;
}

int jt_lcfg_mount_option(int argc, char **argv)
{
        int rc;
        struct lustre_cfg lcfg;

        LCFG_INIT(lcfg, LCFG_MOUNTOPT, lcfg_devname);

        if (argc < 3 || argc > 4)
                return CMD_HELP;

        /* profile name */
        lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
        lcfg.lcfg_inlbuf1 = argv[1];
        /* osc name */
        lcfg.lcfg_inllen2 = strlen(argv[2]) + 1;
        lcfg.lcfg_inlbuf2 = argv[2];
        if (argc == 4) {
                /* mdc name */
                lcfg.lcfg_inllen3 = strlen(argv[3]) + 1;
                lcfg.lcfg_inlbuf3 = argv[3];
        }
        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));
        }

        return rc;
}

int jt_lcfg_del_mount_option(int argc, char **argv)
{
        int rc;
        struct lustre_cfg lcfg;

        LCFG_INIT(lcfg, LCFG_DEL_MOUNTOPT, lcfg_devname);

        if (argc != 2)
                return CMD_HELP;

        /* profile name */
        lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
        lcfg.lcfg_inlbuf1 = argv[1];

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));
        }

        return rc;
}

int jt_lcfg_set_timeout(int argc, char **argv)
{
        int rc;
        struct lustre_cfg lcfg;

        LCFG_INIT(lcfg, LCFG_SET_TIMEOUT, lcfg_devname);

        if (argc != 2)
                return CMD_HELP;

        lcfg.lcfg_num = atoi(argv[1]);

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));
        }

        return rc;
}


int jt_lcfg_set_lustre_upcall(int argc, char **argv)
{
        int rc;
        struct lustre_cfg lcfg;

        LCFG_INIT(lcfg, LCFG_SET_UPCALL, lcfg_devname);

        if (argc != 2)
                return CMD_HELP;

        /* profile name */
        lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
        lcfg.lcfg_inlbuf1 = argv[1];

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));
        }

        return rc;
}
int jt_lcfg_add_conn(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        int priority;
        int rc;

        if (argc == 2)
                priority = 0;
        else if (argc == 3)
                priority = 1;
        else
                return CMD_HELP;

        if (lcfg_devname == NULL) {
                fprintf(stderr, "%s: please use 'cfg_device name' to set the "
                        "device name for config commands.\n",
                        jt_cmdname(argv[0]));
               return -EINVAL;
        }

        LCFG_INIT(lcfg, LCFG_ADD_CONN, lcfg_devname);

        /* connection uuid */
        lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
        lcfg.lcfg_inlbuf1 = argv[1];
        lcfg.lcfg_inllen2 = sizeof(int);
        lcfg.lcfg_inlbuf2 = (char*) &priority;

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));
        }

        return rc;
}

int jt_lcfg_del_conn(int argc, char **argv)
{
        struct lustre_cfg lcfg;
        int rc;

        if (argc != 2)
                return CMD_HELP;

        if (lcfg_devname == NULL) {
                fprintf(stderr, "%s: please use 'cfg_device name' to set the "
                        "device name for config commands.\n",
                        jt_cmdname(argv[0]));
               return -EINVAL;
        }

        LCFG_INIT(lcfg, LCFG_DEL_CONN, lcfg_devname);

        /* connection uuid */
        lcfg.lcfg_inllen1 = strlen(argv[1]) + 1;
        lcfg.lcfg_inlbuf1 = argv[1];

        rc = lcfg_ioctl(argv[0], OBD_DEV_ID, &lcfg);
        if (rc < 0) {
                fprintf(stderr, "error: %s: %s\n", jt_cmdname(argv[0]),
                        strerror(rc = errno));
        }

        return rc;
}