/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 * 
 * Copyright (C) 2001  Cluster File Systems, Inc.
 *
 * This code is issued under the GNU General Public License.
 * See the file COPYING in this distribution
 *
 * These are the only exported functions, they provide some generic
 * infrastructure for managing object devices
 *
 * Object Devices Class Driver
 *              Copyright (C) 2002 Cluster File Systems, Inc.
 */

#define EXPORT_SYMTAB
#include <linux/config.h> /* for CONFIG_PROC_FS */
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/kernel.h>
#include <linux/major.h>
#include <linux/kmod.h>   /* for request_module() */
#include <linux/sched.h>
#include <linux/lp.h>
#include <linux/slab.h>
#include <linux/ioport.h>
#include <linux/fcntl.h>
#include <linux/delay.h>
#include <linux/skbuff.h>
#include <linux/proc_fs.h>
#include <linux/fs.h>
#include <linux/poll.h>
#include <linux/init.h>
#include <linux/list.h>
#include <asm/io.h>
#include <asm/system.h>
#include <asm/poll.h>
#include <asm/uaccess.h>
#include <linux/miscdevice.h>

#define DEBUG_SUBSYSTEM S_CLASS

#include <linux/obd_support.h>
#include <linux/obd_class.h>

struct obd_device obd_dev[MAX_OBD_DEVICES];
extern struct obd_type *class_nm_to_type(char *nm);
struct list_head obd_types;

unsigned long obd_memory = 0;
unsigned long obd_fail_loc = 0;


/*  opening /dev/obd */
static int obd_class_open(struct inode * inode, struct file * file)
{
        ENTRY;

        file->private_data = NULL;
        MOD_INC_USE_COUNT;
        RETURN(0);
}

/*  closing /dev/obd */
static int obd_class_release(struct inode * inode, struct file * file)
{
        ENTRY;

        if (file->private_data)
                file->private_data = NULL;

        MOD_DEC_USE_COUNT;
        RETURN(0);
}


inline void obd_data2conn(struct lustre_handle *conn, struct obd_ioctl_data *data)
{
        conn->addr = data->ioc_addr;
        conn->cookie = data->ioc_cookie;
}


inline void obd_conn2data(struct obd_ioctl_data *data, struct lustre_handle *conn)
{
        data->ioc_addr = conn->addr;
        data->ioc_cookie = conn->cookie;
}

/* to control /dev/obd */
static int obd_class_ioctl (struct inode * inode, struct file * filp,
                            unsigned int cmd, unsigned long arg)
{
        /* NOTE this must be larger than any of the ioctl data structs */
        char buf[1024];
        int len = 1024;
        struct obd_ioctl_data *data;
        struct obd_device *obd = filp->private_data;
        struct lustre_handle conn;
        int rw = OBD_BRW_READ;
        int err = 0;
        ENTRY;

        memset(buf, 0, sizeof(buf));

        if (!obd && cmd != OBD_IOC_DEVICE && cmd != TCGETS &&
            cmd != OBD_IOC_LIST &&
            cmd != OBD_IOC_NAME2DEV && cmd != OBD_IOC_NEWDEV) {
                CERROR("OBD ioctl: No device\n");
                RETURN(-EINVAL);
        }
        if (obd_ioctl_getdata(buf, &len, (void *)arg)) {
                CERROR("OBD ioctl: data error\n");
                RETURN(-EINVAL);
        }
        data = (struct obd_ioctl_data *)buf;

        switch (cmd) {
        case TCGETS:
                RETURN(-EINVAL);
        case OBD_IOC_DEVICE: {
                CDEBUG(D_IOCTL, "\n");
                if (data->ioc_dev >= MAX_OBD_DEVICES || data->ioc_dev < 0) {
                        CERROR("OBD ioctl: DEVICE insufficient devices\n");
                        RETURN(-EINVAL);
                }
                CDEBUG(D_IOCTL, "device %d\n", data->ioc_dev);

                filp->private_data = &obd_dev[data->ioc_dev];
                RETURN(0);
        }

        case OBD_IOC_LIST: {
                int i;
                char *buf = data->ioc_bulk;
                int remains = data->ioc_inllen1;

                if (!data->ioc_inlbuf1) {
                        CERROR("No buffer passed!\n");
                        RETURN(-EINVAL);
                }


                for (i = 0 ; i < MAX_OBD_DEVICES ; i++) {
                        int l;
                        struct obd_device *obd = &obd_dev[i];
                        if (!obd->obd_type) 
                                continue;
                        l = snprintf(buf, remains, "%2d %s %s %s\n",
                                     i, obd->obd_type->typ_name, 
                                     obd->obd_name, obd->obd_uuid);
                        buf +=l;
                        remains -=l;
                        if (remains <= 0) { 
                                CERROR("not enough space for device listing\n");
                                break;
                        }
                }

                err = copy_to_user((int *)arg, data, len);
                RETURN(err);
        }


        case OBD_IOC_NAME2DEV: {
                /* Resolve a device name.  This does not change the
                 * currently selected device.
                 */
                int dev;

                if (!data->ioc_inlbuf1) {
                        CERROR("No name passed!\n");
                        RETURN(-EINVAL);
                }
                CDEBUG(D_IOCTL, "device name %s\n", data->ioc_inlbuf1);
                dev = class_name2dev(data->ioc_inlbuf1);
                data->ioc_dev = dev;
                if (dev == -1) {
                        CDEBUG(D_IOCTL, "No device for name %s!\n",
                               data->ioc_inlbuf1);
                        RETURN(-EINVAL);
                }

                CDEBUG(D_IOCTL, "device name %s, dev %d\n", data->ioc_inlbuf1,
                       dev);
                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_UUID2DEV: {
                /* Resolve a device uuid.  This does not change the
                 * currently selected device.
                 */
                int dev;

                if (!data->ioc_inlbuf1) {
                        CERROR("No UUID passed!\n");
                        RETURN(-EINVAL);
                }
                CDEBUG(D_IOCTL, "device name %s\n", data->ioc_inlbuf1);
                dev = class_uuid2dev(data->ioc_inlbuf1);
                data->ioc_dev = dev;
                if (dev == -1) {
                        CDEBUG(D_IOCTL, "No device for name %s!\n",
                               data->ioc_inlbuf1);
                        RETURN(-EINVAL);
                }

                CDEBUG(D_IOCTL, "device name %s, dev %d\n", data->ioc_inlbuf1,
                       dev);
                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_NEWDEV: {
                int dev = -1;
                int i;

                filp->private_data = NULL;
                for (i = 0 ; i < MAX_OBD_DEVICES ; i++) {
                        struct obd_device *obd = &obd_dev[i];
                        if (!obd->obd_type) {
                                filp->private_data = obd;
                                dev = i;
                                break;
                        }
                }


                data->ioc_dev = dev;
                if (dev == -1)
                        RETURN(-EINVAL);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_ATTACH: {
                struct obd_type *type;

                /* have we attached a type to this device */
                if (obd->obd_flags & OBD_ATTACHED) {
                        CERROR("OBD: Device %d already typed as %s.\n",
                               obd->obd_minor, MKSTR(obd->obd_type->typ_name));
                        RETURN(-EBUSY);
                }

                CDEBUG(D_IOCTL, "attach type %s name: %s uuid: %s\n", 
                       MKSTR(data->ioc_inlbuf1),
                       MKSTR(data->ioc_inlbuf2), MKSTR(data->ioc_inlbuf3));

                /* find the type */
                type = class_nm_to_type(data->ioc_inlbuf1);
                if (!type) {
                        CERROR("OBD: unknown type dev %d\n", obd->obd_minor);
                        RETURN(-EINVAL);
                }

                obd->obd_type = type;
                obd->obd_multi_count = 0;
                INIT_LIST_HEAD(&obd->obd_exports);

                /* do the attach */
                if (OBT(obd) && OBP(obd, attach))
                        err = OBP(obd,attach)(obd, sizeof(*data), data);
                if (err) {
                        obd->obd_type = NULL;
                } else {
                        obd->obd_flags |= OBD_ATTACHED;
                        type->typ_refcnt++;
                        CDEBUG(D_IOCTL, "OBD: dev %d attached type %s\n",
                               obd->obd_minor, data->ioc_inlbuf1);
                        obd->obd_proc_entry =
                                proc_lustre_register_obd_device(obd);
                        if (data->ioc_inlbuf2) {
                                int len = strlen(data->ioc_inlbuf2) + 1;
                                OBD_ALLOC(obd->obd_name, len + 1);
                                if (!obd->obd_name) {
                                        CERROR("no memory\n");
                                        LBUG();
                                }
                                memcpy(obd->obd_name, data->ioc_inlbuf2, len + 1);
                        }
                        if (data->ioc_inlbuf3) {
                                int len = strlen(data->ioc_inlbuf3);
                                if (len > 37) { 
                                        CERROR("uuid should be shorter than 37 bytes\n");
                                        if (obd->obd_name)
                                                OBD_FREE(obd->obd_name, 
                                                         strlen(obd->obd_name) + 1);
                                        RETURN(-EINVAL);
                                }
                                memcpy(obd->obd_uuid, data->ioc_inlbuf3, 
                                       sizeof(obd->obd_uuid));
                        }

                        MOD_INC_USE_COUNT;
                }

                RETURN(err);
        }

        case OBD_IOC_DETACH: {
                ENTRY;
                if (obd->obd_flags & OBD_SET_UP) {
                        CERROR("OBD device %d still set up\n", obd->obd_minor);
                        RETURN(-EBUSY);
                }
                if (! (obd->obd_flags & OBD_ATTACHED) ) {
                        CERROR("OBD device %d not attached\n", obd->obd_minor);
                        RETURN(-ENODEV);
                }
                if ( !list_empty(&obd->obd_exports) ) {
                        CERROR("OBD device %d has exports\n",
                               obd->obd_minor);
                        RETURN(-EBUSY);
                }

                if (obd->obd_name) {
                        OBD_FREE(obd->obd_name, strlen(obd->obd_name)+ 1);
                        obd->obd_name = NULL;
                }

                if (obd->obd_proc_entry)
                        proc_lustre_release_obd_device(obd);

                obd->obd_flags &= ~OBD_ATTACHED;
                obd->obd_type->typ_refcnt--;
                obd->obd_type = NULL;
                MOD_DEC_USE_COUNT;
                RETURN(0);
        }

        case OBD_IOC_SETUP: {
                /* have we attached a type to this device? */
                if (!(obd->obd_flags & OBD_ATTACHED)) {
                        CERROR("Device %d not attached\n", obd->obd_minor);
                        RETURN(-ENODEV);
                }

                /* has this been done already? */
                if ( obd->obd_flags & OBD_SET_UP ) {
                        CERROR("Device %d already setup (type %s)\n",
                               obd->obd_minor, obd->obd_type->typ_name);
                        RETURN(-EBUSY);
                }

                if ( OBT(obd) && OBP(obd, setup) )
                        err = obd_setup(obd, sizeof(*data), data);

                if (!err) {
                        obd->obd_type->typ_refcnt++;
                        obd->obd_flags |= OBD_SET_UP;
                }

                RETURN(err);
        }
        case OBD_IOC_CLEANUP: {
                /* have we attached a type to this device? */
                if (!(obd->obd_flags & OBD_ATTACHED)) {
                        CERROR("Device %d not attached\n", obd->obd_minor);
                        RETURN(-ENODEV);
                }

                if ( OBT(obd) && OBP(obd, cleanup) )
                        err = obd_cleanup(obd);

                if (!err) {
                        obd->obd_flags &= ~OBD_SET_UP;
                        obd->obd_type->typ_refcnt--;
                }
                RETURN(err);
        }

        case OBD_IOC_CONNECT: {
                obd_data2conn(&conn, data); 

                err = obd_connect(&conn, obd);

                CDEBUG(D_IOCTL, "assigned export %Lx\n", conn.addr);
                obd_conn2data(data, &conn);
                if (err)
                        RETURN(err);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_DISCONNECT: {
                obd_data2conn(&conn, data);
                err = obd_disconnect(&conn);
                RETURN(err);
        }

        case OBD_IOC_DEC_USE_COUNT: {
                MOD_DEC_USE_COUNT;
                RETURN(0);
        }

        case OBD_IOC_CREATE: {
                obd_data2conn(&conn, data);

                err = obd_create(&conn, &data->ioc_obdo1);
                if (err)
                        RETURN(err);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_GETATTR: {

                obd_data2conn(&conn, data);
                err = obd_getattr(&conn, &data->ioc_obdo1);
                if (err)
                        RETURN(err);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_SETATTR: {
                obd_data2conn(&conn, data);
                err = obd_setattr(&conn, &data->ioc_obdo1);
                if (err)
                        RETURN(err);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_DESTROY: {
                obd_data2conn(&conn, data);

                err = obd_destroy(&conn, &data->ioc_obdo1);
                if (err)
                        RETURN(err);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }

        case OBD_IOC_BRW_WRITE:
                rw = OBD_BRW_WRITE;
        case OBD_IOC_BRW_READ: {
                /* FIXME: use a better ioctl data struct than obd_ioctl_data.
                 *        We don't really support multiple-obdo I/Os here,
                 *        for example offset and count are not per-obdo.
                 */
                struct obdo     *obdos[2] = { NULL, NULL };
                obd_count       oa_bufs[2] = { 0, 0 };
                struct page     **bufs = NULL;
                obd_size        *counts = NULL;
                obd_off         *offsets = NULL;
                obd_flag        *flags = NULL;
                int             num = 1;
                int             pages;
                int             i, j;

                obd_data2conn(&conn, data);

                pages = oa_bufs[0] = data->ioc_plen1 / PAGE_SIZE;
                if (data->ioc_obdo2.o_id) {
                        num = 2;
                        oa_bufs[1] = data->ioc_plen2 / PAGE_SIZE;
                        pages += oa_bufs[1];
                }

                CDEBUG(D_INODE, "BRW %s with %dx%d pages\n",
                       rw == OBD_BRW_READ ? "read" : "write",
                       num, oa_bufs[0]);
                OBD_ALLOC(bufs, pages * sizeof(*bufs));
                OBD_ALLOC(counts, pages * sizeof(*counts));
                OBD_ALLOC(offsets, pages * sizeof(*offsets));
                OBD_ALLOC(flags, pages * sizeof(*flags));
                if (!bufs || !counts || !offsets || !flags) {
                        CERROR("no memory for %d BRW per-page data\n", pages);
                        err = -ENOMEM;
                        GOTO(brw_free, err);
                }

                obdos[0] = &data->ioc_obdo1;
                if (num > 1)
                        obdos[1] = &data->ioc_obdo2;

                for (i = 0, pages = 0; i < num; i++) {
                        unsigned long off;
                        void *from;

                        from = (&data->ioc_pbuf1)[i];
                        off = data->ioc_offset;

                        for (j = 0; j < oa_bufs[i];
                             j++, pages++, off += PAGE_SIZE, from += PAGE_SIZE){
                                unsigned long to;

                                to = __get_free_pages(GFP_KERNEL, 0);
                                if (!to) {
                                /*      ||
                                    copy_from_user((void *)to,from,PAGE_SIZE))
                                        free_pages(to, 0);
                                 */
                                        CERROR("no memory for brw pages\n");
                                        err = -ENOMEM;
                                        GOTO(brw_cleanup, err);
                                }
                                bufs[pages] = virt_to_page(to);
                                counts[pages] = PAGE_SIZE;
                                offsets[pages] = off;
                                flags[pages] = 0;
                        }
                }

                err = obd_brw(rw, &conn, num, obdos, oa_bufs, bufs,
                              counts, offsets, flags, NULL);

                EXIT;
        brw_cleanup:
                i = pages;
                while (i-- > 0)
                        __free_pages(bufs[i], 0);
        brw_free:
                OBD_FREE(bufs, pages * sizeof(*bufs));
                OBD_FREE(counts, pages * sizeof(*counts));
                OBD_FREE(offsets, pages * sizeof(*offsets));
                OBD_FREE(flags, pages * sizeof(*flags));
                return err;
        }
        default: {
                obd_data2conn(&conn, data);

                err = obd_iocontrol(cmd, &conn, sizeof(*data), data, NULL);
                if (err)
                        RETURN(err);

                err = copy_to_user((int *)arg, data, sizeof(*data));
                RETURN(err);
        }
        }
} /* obd_class_ioctl */



/* declare character device */
static struct file_operations obd_psdev_fops = {
        ioctl: obd_class_ioctl,       /* ioctl */
        open: obd_class_open,        /* open */
        release: obd_class_release,     /* release */
};

/* modules setup */
#define OBD_MINOR 241
static struct miscdevice obd_psdev = {
        OBD_MINOR,
        "obd_psdev",
        &obd_psdev_fops
};


EXPORT_SYMBOL(obd_dev);
EXPORT_SYMBOL(obdo_cachep);
EXPORT_SYMBOL(obd_memory);
EXPORT_SYMBOL(obd_fail_loc);

EXPORT_SYMBOL(class_register_type);
EXPORT_SYMBOL(class_unregister_type);
EXPORT_SYMBOL(class_name2dev);
EXPORT_SYMBOL(class_uuid2dev);
EXPORT_SYMBOL(class_connect);
EXPORT_SYMBOL(class_conn2export);
EXPORT_SYMBOL(class_conn2obd);
EXPORT_SYMBOL(class_disconnect);
EXPORT_SYMBOL(class_multi_setup);
EXPORT_SYMBOL(class_multi_cleanup);

static int __init init_obdclass(void)
{
        int err;
        int i;

        printk(KERN_INFO "OBD class driver  v0.9, info@clusterfs.com\n");

        INIT_LIST_HEAD(&obd_types);

        if ((err = misc_register(&obd_psdev))) {
                CERROR("cannot register %d err %d\n", OBD_MINOR, err);
                return err;
        }

        for (i = 0; i < MAX_OBD_DEVICES; i++) {
                memset(&(obd_dev[i]), 0, sizeof(obd_dev[i]));
                obd_dev[i].obd_minor = i;
                INIT_LIST_HEAD(&obd_dev[i].obd_exports);
        }

        err = obd_init_caches();
        if (err)
                return err;
        obd_sysctl_init();
        return 0;
}

static void __exit cleanup_obdclass(void)
{
        int i;
        ENTRY;

        misc_deregister(&obd_psdev);
        for (i = 0; i < MAX_OBD_DEVICES; i++) {
                struct obd_device *obd = &obd_dev[i];
                if (obd->obd_type && (obd->obd_flags & OBD_SET_UP) &&
                    OBT(obd) && OBP(obd, detach)) {
                        /* XXX should this call generic detach otherwise? */
                        OBP(obd, detach)(obd);
                }
        }

        obd_cleanup_caches();
        obd_sysctl_clean();
        CERROR("obd memory leaked: %ld bytes\n", obd_memory);
        EXIT;
}

MODULE_AUTHOR("Cluster File Systems, Inc. <info@clusterfs.com>");
MODULE_DESCRIPTION("Lustre Class Driver v1.0");
MODULE_LICENSE("GPL");

module_init(init_obdclass);
module_exit(cleanup_obdclass);