lnet/selftest/ping_test.c

   1 // SPDX-License-Identifier: GPL-2.0
   2
   3 /*
   4  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
   5  * Use is subject to license terms.
   6  *
   7  * Copyright (c) 2012, Intel Corporation.
   8  */
   9
  10 /*
  11  * This file is part of Lustre, http://www.lustre.org/
  12  *
  13  * Test client & Server
  14  *
  15  * Author: Liang Zhen <liangzhen@clusterfs.com>
  16  */
  17
  18 #include "selftest.h"
  19
  20 #define LST_PING_TEST_MAGIC     0xbabeface
  21
  22 static int ping_srv_workitems = SFW_TEST_WI_MAX;
  23 module_param(ping_srv_workitems, int, 0644);
  24 MODULE_PARM_DESC(ping_srv_workitems, "# PING server workitems");
  25
  26 struct lst_ping_data {
  27         spinlock_t      pnd_lock;       /* serialize */
  28         int             pnd_counter;    /* sequence counter */
  29 };
  30
  31 static struct lst_ping_data lst_ping_data;
  32
  33 static int
  34 ping_client_init(struct sfw_test_instance *tsi)
  35 {
  36         struct sfw_session *sn = tsi->tsi_batch->bat_session;
  37
  38         LASSERT(tsi->tsi_is_client);
  39         LASSERT(sn != NULL && (sn->sn_features & ~LST_FEATS_MASK) == 0);
  40
  41         spin_lock_init(&lst_ping_data.pnd_lock);
  42         lst_ping_data.pnd_counter = 0;
  43
  44         return 0;
  45 }
  46
  47 static void
  48 ping_client_fini(struct sfw_test_instance *tsi)
  49 {
  50         struct sfw_session *sn = tsi->tsi_batch->bat_session;
  51         int            errors;
  52
  53         LASSERT (sn != NULL);
  54         LASSERT (tsi->tsi_is_client);
  55
  56         errors = atomic_read(&sn->sn_ping_errors);
  57         if (errors)
  58                 CWARN ("%d pings have failed.\n", errors);
  59         else
  60                 CDEBUG (D_NET, "Ping test finished OK.\n");
  61 }
  62
  63 static int
  64 ping_client_prep_rpc(struct sfw_test_unit *tsu, struct lnet_process_id dest,
  65                      struct srpc_client_rpc **rpc)
  66 {
  67         struct srpc_ping_reqst *req;
  68         struct sfw_test_instance *tsi = tsu->tsu_instance;
  69         struct sfw_session *sn = tsi->tsi_batch->bat_session;
  70         struct timespec64 ts;
  71         int rc;
  72
  73         LASSERT(sn != NULL);
  74         LASSERT((sn->sn_features & ~LST_FEATS_MASK) == 0);
  75
  76         rc = sfw_create_test_rpc(tsu, dest, sn->sn_features, 0, 0, rpc);
  77         if (rc != 0)
  78                 return rc;
  79
  80         req = &(*rpc)->crpc_reqstmsg.msg_body.ping_reqst;
  81
  82         req->pnr_magic = LST_PING_TEST_MAGIC;
  83
  84         spin_lock(&lst_ping_data.pnd_lock);
  85         req->pnr_seq = lst_ping_data.pnd_counter++;
  86         spin_unlock(&lst_ping_data.pnd_lock);
  87
  88         ktime_get_real_ts64(&ts);
  89         req->pnr_time_sec  = ts.tv_sec;
  90         req->pnr_time_nsec = ts.tv_nsec;
  91
  92         return rc;
  93 }
  94
  95 static void
  96 ping_client_done_rpc(struct sfw_test_unit *tsu, struct srpc_client_rpc *rpc)
  97 {
  98         struct sfw_test_instance *tsi = tsu->tsu_instance;
  99         struct sfw_session *sn = tsi->tsi_batch->bat_session;
 100         struct srpc_ping_reqst *reqst = &rpc->crpc_reqstmsg.msg_body.ping_reqst;
 101         struct srpc_ping_reply *reply = &rpc->crpc_replymsg.msg_body.ping_reply;
 102         struct timespec64 ts;
 103
 104         LASSERT(sn != NULL);
 105
 106         if (rpc->crpc_status != 0) {
 107                 if (!tsi->tsi_stopping) /* rpc could have been aborted */
 108                         atomic_inc(&sn->sn_ping_errors);
 109                 CERROR ("Unable to ping %s (%d): %d\n",
 110                         libcfs_id2str(rpc->crpc_dest),
 111                         reqst->pnr_seq, rpc->crpc_status);
 112                 return;
 113         }
 114
 115         if (rpc->crpc_replymsg.msg_magic != SRPC_MSG_MAGIC) {
 116                 __swab32s(&reply->pnr_seq);
 117                 __swab32s(&reply->pnr_magic);
 118                 __swab32s(&reply->pnr_status);
 119         }
 120
 121         if (reply->pnr_magic != LST_PING_TEST_MAGIC) {
 122                 rpc->crpc_status = -EBADMSG;
 123                 atomic_inc(&sn->sn_ping_errors);
 124                 CERROR ("Bad magic %u from %s, %u expected.\n",
 125                         reply->pnr_magic, libcfs_id2str(rpc->crpc_dest),
 126                         LST_PING_TEST_MAGIC);
 127                 return;
 128         }
 129
 130         if (reply->pnr_seq != reqst->pnr_seq) {
 131                 rpc->crpc_status = -EBADMSG;
 132                 atomic_inc(&sn->sn_ping_errors);
 133                 CERROR ("Bad seq %u from %s, %u expected.\n",
 134                         reply->pnr_seq, libcfs_id2str(rpc->crpc_dest),
 135                         reqst->pnr_seq);
 136                 return;
 137         }
 138
 139         ktime_get_real_ts64(&ts);
 140         CDEBUG(D_NET, "%d reply in %llu nsec\n", reply->pnr_seq,
 141                (u64)((ts.tv_sec - reqst->pnr_time_sec) * NSEC_PER_SEC +
 142                     (ts.tv_nsec - reqst->pnr_time_nsec)));
 143 }
 144
 145 static int
 146 ping_server_handle(struct srpc_server_rpc *rpc)
 147 {
 148         struct srpc_service *sv  = rpc->srpc_scd->scd_svc;
 149         struct srpc_msg *reqstmsg = &rpc->srpc_reqstbuf->buf_msg;
 150         struct srpc_msg *replymsg = &rpc->srpc_replymsg;
 151         struct srpc_ping_reqst *req = &reqstmsg->msg_body.ping_reqst;
 152         struct srpc_ping_reply *rep = &rpc->srpc_replymsg.msg_body.ping_reply;
 153
 154         LASSERT (sv->sv_id == SRPC_SERVICE_PING);
 155
 156         if (reqstmsg->msg_magic != SRPC_MSG_MAGIC) {
 157                 LASSERT (reqstmsg->msg_magic == __swab32(SRPC_MSG_MAGIC));
 158
 159                 __swab32s(&req->pnr_seq);
 160                 __swab32s(&req->pnr_magic);
 161                 __swab64s(&req->pnr_time_sec);
 162                 __swab64s(&req->pnr_time_nsec);
 163         }
 164         LASSERT (reqstmsg->msg_type == srpc_service2request(sv->sv_id));
 165
 166         if (req->pnr_magic != LST_PING_TEST_MAGIC) {
 167                 CERROR ("Unexpect magic %08x from %s\n",
 168                         req->pnr_magic, libcfs_id2str(rpc->srpc_peer));
 169                 return -EINVAL;
 170         }
 171
 172         rep->pnr_seq   = req->pnr_seq;
 173         rep->pnr_magic = LST_PING_TEST_MAGIC;
 174
 175         if ((reqstmsg->msg_ses_feats & ~LST_FEATS_MASK) != 0) {
 176                 replymsg->msg_ses_feats = LST_FEATS_MASK;
 177                 rep->pnr_status = EPROTO;
 178                 return 0;
 179         }
 180
 181         replymsg->msg_ses_feats = reqstmsg->msg_ses_feats;
 182
 183         CDEBUG(D_NET, "Get ping %d from %s\n",
 184                req->pnr_seq, libcfs_id2str(rpc->srpc_peer));
 185         return 0;
 186 }
 187
 188 struct sfw_test_client_ops ping_test_client;
 189
 190 void ping_init_test_client(void)
 191 {
 192         ping_test_client.tso_init     = ping_client_init;
 193         ping_test_client.tso_fini     = ping_client_fini;
 194         ping_test_client.tso_prep_rpc = ping_client_prep_rpc;
 195         ping_test_client.tso_done_rpc = ping_client_done_rpc;
 196 }
 197
 198 struct srpc_service ping_test_service;
 199
 200 void ping_init_test_service(void)
 201 {
 202         ping_test_service.sv_id       = SRPC_SERVICE_PING;
 203         ping_test_service.sv_name     = "ping_test";
 204         ping_test_service.sv_handler  = ping_server_handle;
 205         ping_test_service.sv_wi_total = ping_srv_workitems;
 206 }