1 // SPDX-License-Identifier: GPL-2.0
4 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
5 * Use is subject to license terms.
7 * Copyright (c) 2012, Intel Corporation.
11 * This file is part of Lustre, http://www.lustre.org/
13 * Test client & Server
15 * Author: Liang Zhen <liangzhen@clusterfs.com>
20 #define LST_PING_TEST_MAGIC 0xbabeface
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");
26 struct lst_ping_data {
27 spinlock_t pnd_lock; /* serialize */
28 int pnd_counter; /* sequence counter */
31 static struct lst_ping_data lst_ping_data;
34 ping_client_init(struct sfw_test_instance *tsi)
36 struct sfw_session *sn = tsi->tsi_batch->bat_session;
38 LASSERT(tsi->tsi_is_client);
39 LASSERT(sn != NULL && (sn->sn_features & ~LST_FEATS_MASK) == 0);
41 spin_lock_init(&lst_ping_data.pnd_lock);
42 lst_ping_data.pnd_counter = 0;
48 ping_client_fini(struct sfw_test_instance *tsi)
50 struct sfw_session *sn = tsi->tsi_batch->bat_session;
54 LASSERT (tsi->tsi_is_client);
56 errors = atomic_read(&sn->sn_ping_errors);
58 CWARN ("%d pings have failed.\n", errors);
60 CDEBUG (D_NET, "Ping test finished OK.\n");
64 ping_client_prep_rpc(struct sfw_test_unit *tsu, struct lnet_process_id dest,
65 struct srpc_client_rpc **rpc)
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;
74 LASSERT((sn->sn_features & ~LST_FEATS_MASK) == 0);
76 rc = sfw_create_test_rpc(tsu, dest, sn->sn_features, 0, 0, rpc);
80 req = &(*rpc)->crpc_reqstmsg.msg_body.ping_reqst;
82 req->pnr_magic = LST_PING_TEST_MAGIC;
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);
88 ktime_get_real_ts64(&ts);
89 req->pnr_time_sec = ts.tv_sec;
90 req->pnr_time_nsec = ts.tv_nsec;
96 ping_client_done_rpc(struct sfw_test_unit *tsu, struct srpc_client_rpc *rpc)
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;
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);
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);
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);
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),
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)));
146 ping_server_handle(struct srpc_server_rpc *rpc)
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;
154 LASSERT (sv->sv_id == SRPC_SERVICE_PING);
156 if (reqstmsg->msg_magic != SRPC_MSG_MAGIC) {
157 LASSERT (reqstmsg->msg_magic == __swab32(SRPC_MSG_MAGIC));
159 __swab32s(&req->pnr_seq);
160 __swab32s(&req->pnr_magic);
161 __swab64s(&req->pnr_time_sec);
162 __swab64s(&req->pnr_time_nsec);
164 LASSERT (reqstmsg->msg_type == srpc_service2request(sv->sv_id));
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));
172 rep->pnr_seq = req->pnr_seq;
173 rep->pnr_magic = LST_PING_TEST_MAGIC;
175 if ((reqstmsg->msg_ses_feats & ~LST_FEATS_MASK) != 0) {
176 replymsg->msg_ses_feats = LST_FEATS_MASK;
177 rep->pnr_status = EPROTO;
181 replymsg->msg_ses_feats = reqstmsg->msg_ses_feats;
183 CDEBUG(D_NET, "Get ping %d from %s\n",
184 req->pnr_seq, libcfs_id2str(rpc->srpc_peer));
188 struct sfw_test_client_ops ping_test_client;
190 void ping_init_test_client(void)
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;
198 struct srpc_service ping_test_service;
200 void ping_init_test_service(void)
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;