/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 *
 *  Copyright (C) 2002 Cluster File Systems, Inc.
 *
 *   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.
 *
 */

#define EXPORT_SYMTAB
#define DEBUG_SUBSYSTEM S_RPC

#include <linux/module.h>
#include <linux/lustre_ha.h>

extern int ptlrpc_init_portals(void);
extern void ptlrpc_exit_portals(void);

int connmgr_setup(struct obd_device *obddev, obd_count len, void *buf)
{
        struct recovd_obd *recovd = &obddev->u.recovd;
        int err;
        ENTRY;

        memset(recovd, 0, sizeof(*recovd));

        OBD_ALLOC(recovd->recovd_client, sizeof(*recovd->recovd_client));
        if (!recovd)
                RETURN(-ENOMEM);

        err = recovd_setup(recovd);
        if (err)
                GOTO(err_free, err);

        recovd->recovd_service =
                ptlrpc_init_svc(128 * 1024,CONNMGR_REQUEST_PORTAL,
                                CONNMGR_REPLY_PORTAL, "self", connmgr_handle);
        if (!recovd->recovd_service) {
                CERROR("failed to start service\n");
                GOTO(err_recovd, err = -EINVAL);
        }

        ptlrpc_init_client(NULL, NULL, CONNMGR_REQUEST_PORTAL, 
                           CONNMGR_REPLY_PORTAL, recovd->recovd_client);
        recovd->recovd_client->cli_name = "connmgr"; 

        err = ptlrpc_start_thread(obddev, recovd->recovd_service, "lustre_connmgr");
        if (err) {
                CERROR("cannot start thread\n");
                GOTO(err_svc, err);
        }

        MOD_INC_USE_COUNT;
        ptlrpc_connmgr = recovd;
        RETURN(0);

 err_svc: 
        rpc_unregister_service(recovd->recovd_service);
 err_recovd: 
        recovd_cleanup(recovd); 
 err_free:
        if (recovd->recovd_client)
                OBD_FREE(recovd->recovd_client, sizeof(*recovd->recovd_client));
        RETURN(err);
}

int connmgr_cleanup(struct obd_device *dev)
{
        struct recovd_obd *recovd = &dev->u.recovd;
        int err;

        err = recovd_cleanup(recovd); 
        if (err) 
                LBUG();

        ptlrpc_stop_all_threads(recovd->recovd_service);
        rpc_unregister_service(recovd->recovd_service);
        if (!list_empty(&recovd->recovd_service->srv_reqs)) {
                // XXX reply with errors and clean up
                CERROR("Request list not empty!\n");
        }

        OBD_FREE(recovd->recovd_service, sizeof(*recovd->recovd_service));
        ptlrpc_cleanup_client(recovd->recovd_client);
        OBD_FREE(recovd->recovd_client, sizeof(*recovd->recovd_client));
        MOD_DEC_USE_COUNT;
        RETURN(0);
}


int connmgr_iocontrol(long cmd, struct obd_conn *conn, int len, void *karg,
                      void *uarg)
{
        struct recovd_obd *recovd = &conn->oc_dev->u.recovd;

        ENTRY;
        if (cmd == OBD_RECOVD_NEWCONN) { 
                spin_lock(&recovd->recovd_lock);
                recovd->recovd_flags |= RECOVD_UPCALL_ANSWER;
                recovd->recovd_wakeup_flag = 1;
                wake_up(&recovd->recovd_waitq);
                spin_unlock(&recovd->recovd_lock);
                EXIT;
        }
        return 0;
}


/* use obd ops to offer management infrastructure */
static struct obd_ops recovd_obd_ops = {
        o_setup:       connmgr_setup,
        o_cleanup:     connmgr_cleanup,
        o_iocontrol:     connmgr_iocontrol,
};

static int __init ptlrpc_init(void)
{
        int rc; 
        rc = ptlrpc_init_portals();
        if (rc) 
                RETURN(rc);
        ptlrpc_init_connection();
        obd_register_type(&recovd_obd_ops, LUSTRE_HA_NAME);
        return 0;
}

static void __exit ptlrpc_exit(void)
{
        obd_unregister_type(LUSTRE_HA_NAME);
        ptlrpc_exit_portals();
        ptlrpc_cleanup_connection();
}

/* connmgr.c */
EXPORT_SYMBOL(ptlrpc_connmgr);
EXPORT_SYMBOL(connmgr_connect);
EXPORT_SYMBOL(connmgr_handle);
EXPORT_SYMBOL(recovd_cli_fail);
EXPORT_SYMBOL(recovd_cli_manage);
EXPORT_SYMBOL(recovd_cli_fixed);
EXPORT_SYMBOL(recovd_setup);
EXPORT_SYMBOL(recovd_cleanup);

/* connection.c */
EXPORT_SYMBOL(ptlrpc_readdress_connection);
EXPORT_SYMBOL(ptlrpc_get_connection);
EXPORT_SYMBOL(ptlrpc_put_connection);
EXPORT_SYMBOL(ptlrpc_connection_addref);
EXPORT_SYMBOL(ptlrpc_init_connection);
EXPORT_SYMBOL(ptlrpc_cleanup_connection);

/* niobuf.c */
EXPORT_SYMBOL(ptlrpc_send_bulk);
EXPORT_SYMBOL(ptlrpc_register_bulk);
EXPORT_SYMBOL(ptlrpc_abort_bulk);
EXPORT_SYMBOL(ptlrpc_reply);
EXPORT_SYMBOL(ptlrpc_error);
EXPORT_SYMBOL(ptlrpc_resend_req);
EXPORT_SYMBOL(ptl_send_rpc);
EXPORT_SYMBOL(ptlrpc_link_svc_me);

/* client.c */
EXPORT_SYMBOL(ptlrpc_init_client);
EXPORT_SYMBOL(ptlrpc_cleanup_client);
EXPORT_SYMBOL(ptlrpc_req_to_uuid);
EXPORT_SYMBOL(ptlrpc_uuid_to_connection);
EXPORT_SYMBOL(ptlrpc_queue_wait);
EXPORT_SYMBOL(ptlrpc_continue_req);
EXPORT_SYMBOL(ptlrpc_replay_req);
EXPORT_SYMBOL(ptlrpc_restart_req);
EXPORT_SYMBOL(ptlrpc_prep_req);
EXPORT_SYMBOL(ptlrpc_free_req);
EXPORT_SYMBOL(ptlrpc_req_finished);
EXPORT_SYMBOL(ptlrpc_prep_bulk);
EXPORT_SYMBOL(ptlrpc_free_bulk);
EXPORT_SYMBOL(ptlrpc_prep_bulk_page);
EXPORT_SYMBOL(ptlrpc_free_bulk_page);
EXPORT_SYMBOL(ptlrpc_check_status);

/* service.c */
EXPORT_SYMBOL(ptlrpc_init_svc);
EXPORT_SYMBOL(ptlrpc_stop_all_threads);
EXPORT_SYMBOL(ptlrpc_start_thread);
EXPORT_SYMBOL(rpc_unregister_service);

/* pack_generic.c */
EXPORT_SYMBOL(lustre_pack_msg);
EXPORT_SYMBOL(lustre_msg_size);
EXPORT_SYMBOL(lustre_unpack_msg);
EXPORT_SYMBOL(lustre_msg_buf);

MODULE_AUTHOR("Peter J. Braam <braam@clusterfs.com>");
MODULE_DESCRIPTION("Lustre Request Processor v1.0");
MODULE_LICENSE("GPL"); 

module_init(ptlrpc_init);
module_exit(ptlrpc_exit);