ping_test.c

lustre 1.6.5 source code

/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 *
 * Author: Liang Zhen <[e-mail]>
 *
 * This file is part of Lustre, http://www.lustre.org
 *
 * Test client & Server
 */

#include "selftest.h"

#define LST_PING_TEST_MAGIC     0xbabeface

typedef struct {
        spinlock_t      pnd_lock;       /* serialize */
        int             pnd_counter;    /* sequence counter */
} lst_ping_data_t;

static lst_ping_data_t  lst_ping_data;

static int
ping_client_init(sfw_test_instance_t *tsi)
{
        LASSERT (tsi->tsi_is_client);

        spin_lock_init(&lst_ping_data.pnd_lock);
        lst_ping_data.pnd_counter = 0;

        return 0;
}

static void
ping_client_fini (sfw_test_instance_t *tsi)
{
        sfw_session_t *sn = tsi->tsi_batch->bat_session;
        int            errors;

        LASSERT (sn != NULL);
        LASSERT (tsi->tsi_is_client);

        errors = atomic_read(&sn->sn_ping_errors);
        if (errors)
                CWARN ("%d pings have failed.\n", errors);
        else
                CDEBUG (D_NET, "Ping test finished OK.\n");
}

static int
ping_client_prep_rpc(sfw_test_unit_t *tsu,
                     lnet_process_id_t dest, srpc_client_rpc_t **rpc)
{
        srpc_ping_reqst_t *req;
        struct timeval     tv;
        int                rc;

        rc = sfw_create_test_rpc(tsu, dest, 0, 0, rpc);
        if (rc != 0)
                return rc;

        req = &(*rpc)->crpc_reqstmsg.msg_body.ping_reqst;

        req->pnr_magic = LST_PING_TEST_MAGIC;

        spin_lock(&lst_ping_data.pnd_lock);
        req->pnr_seq = lst_ping_data.pnd_counter ++;
        spin_unlock(&lst_ping_data.pnd_lock);

        cfs_fs_timeval(&tv);
        req->pnr_time_sec  = tv.tv_sec;
        req->pnr_time_usec = tv.tv_usec;

        return rc;
}

static void
ping_client_done_rpc (sfw_test_unit_t *tsu, srpc_client_rpc_t *rpc)
{
        sfw_test_instance_t *tsi = tsu->tsu_instance;
        sfw_session_t       *sn = tsi->tsi_batch->bat_session;
        srpc_ping_reqst_t   *reqst = &rpc->crpc_reqstmsg.msg_body.ping_reqst;
        srpc_ping_reply_t   *reply = &rpc->crpc_replymsg.msg_body.ping_reply;
        struct timeval       tv;

        LASSERT (sn != NULL);

        if (rpc->crpc_status != 0) {
                if (!tsi->tsi_stopping) /* rpc could have been aborted */
                        atomic_inc(&sn->sn_ping_errors);
                CERROR ("Unable to ping %s (%d): %d\n",
                        libcfs_id2str(rpc->crpc_dest),
                        reqst->pnr_seq, rpc->crpc_status);
                return;
        } 

        if (rpc->crpc_replymsg.msg_magic != SRPC_MSG_MAGIC) {
                __swab32s(&reply->pnr_seq);
                __swab32s(&reply->pnr_magic);
                __swab32s(&reply->pnr_status);
        }
        
        if (reply->pnr_magic != LST_PING_TEST_MAGIC) {
                rpc->crpc_status = -EBADMSG;
                atomic_inc(&sn->sn_ping_errors);
                CERROR ("Bad magic %u from %s, %u expected.\n",
                        reply->pnr_magic, libcfs_id2str(rpc->crpc_dest),
                        LST_PING_TEST_MAGIC);
                return;
        } 
        
        if (reply->pnr_seq != reqst->pnr_seq) {
                rpc->crpc_status = -EBADMSG;
                atomic_inc(&sn->sn_ping_errors);
                CERROR ("Bad seq %u from %s, %u expected.\n",
                        reply->pnr_seq, libcfs_id2str(rpc->crpc_dest),
                        reqst->pnr_seq);
                return;
        }

        cfs_fs_timeval(&tv);
        CDEBUG (D_NET, "%d reply in %u usec\n", reply->pnr_seq,
                (unsigned)((tv.tv_sec - (unsigned)reqst->pnr_time_sec) * 1000000
                           + (tv.tv_usec - reqst->pnr_time_usec)));
        return;
}

static int
ping_server_handle (srpc_server_rpc_t *rpc)
{
        srpc_service_t    *sv  = rpc->srpc_service;
        srpc_msg_t        *reqstmsg = &rpc->srpc_reqstbuf->buf_msg;
        srpc_ping_reqst_t *req = &reqstmsg->msg_body.ping_reqst;
        srpc_ping_reply_t *rep = &rpc->srpc_replymsg.msg_body.ping_reply;

        LASSERT (sv->sv_id == SRPC_SERVICE_PING);

        if (reqstmsg->msg_magic != SRPC_MSG_MAGIC) {
                LASSERT (reqstmsg->msg_magic == __swab32(SRPC_MSG_MAGIC));

                __swab32s(&reqstmsg->msg_type);
                __swab32s(&req->pnr_seq);
                __swab32s(&req->pnr_magic);
                __swab64s(&req->pnr_time_sec);
                __swab64s(&req->pnr_time_usec);
        }
        LASSERT (reqstmsg->msg_type == srpc_service2request(sv->sv_id));

        if (req->pnr_magic != LST_PING_TEST_MAGIC) {
                CERROR ("Unexpect magic %08x from %s\n",
                        req->pnr_magic, libcfs_id2str(rpc->srpc_peer));
                return -EINVAL;
        }

        rep->pnr_seq   = req->pnr_seq;
        rep->pnr_magic = LST_PING_TEST_MAGIC;

        CDEBUG (D_NET, "Get ping %d from %s\n",
                req->pnr_seq, libcfs_id2str(rpc->srpc_peer));
        return 0;
}

sfw_test_client_ops_t ping_test_client = 
{
        .tso_init       = ping_client_init,
        .tso_fini       = ping_client_fini,
        .tso_prep_rpc   = ping_client_prep_rpc,
        .tso_done_rpc   = ping_client_done_rpc,
};

srpc_service_t ping_test_service = 
{
        .sv_name        = "ping test",
        .sv_handler     = ping_server_handle,
        .sv_id          = SRPC_SERVICE_PING,
};