1 /*
2 * GPL HEADER START
3 *
4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 only,
8 * as published by the Free Software Foundation.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License version 2 for more details (a copy is included
14 * in the LICENSE file that accompanied this code).
15 *
16 * You should have received a copy of the GNU General Public License
17 * version 2 along with this program; If not, see
18 * http://www.gnu.org/licenses/gpl-2.0.html
19 *
20 * GPL HEADER END
21 */
22 /*
23 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
24 * Use is subject to license terms.
25 *
26 * Copyright (c) 2012, Intel Corporation.
27 */
28 /*
29 * This file is part of Lustre, http://www.lustre.org/
30 * Lustre is a trademark of Sun Microsystems, Inc.
31 *
32 * lnet/selftest/conctl.c
33 *
34 * Test client & Server
35 *
36 * Author: Liang Zhen <liangzhen@clusterfs.com>
37 */
38
39 #include "selftest.h"
40
41 #define LST_PING_TEST_MAGIC 0xbabeface
42
43 static int ping_srv_workitems = SFW_TEST_WI_MAX;
44 module_param(ping_srv_workitems, int, 0644);
45 MODULE_PARM_DESC(ping_srv_workitems, "# PING server workitems");
46
47 struct lst_ping_data {
48 spinlock_t pnd_lock; /* serialize */
49 int pnd_counter; /* sequence counter */
50 };
51
52 static struct lst_ping_data lst_ping_data;
53
54 static int
ping_client_init(struct sfw_test_instance * tsi)55 ping_client_init(struct sfw_test_instance *tsi)
56 {
57 struct sfw_session *sn = tsi->tsi_batch->bat_session;
58
59 LASSERT(tsi->tsi_is_client);
60 LASSERT(sn && !(sn->sn_features & ~LST_FEATS_MASK));
61
62 spin_lock_init(&lst_ping_data.pnd_lock);
63 lst_ping_data.pnd_counter = 0;
64
65 return 0;
66 }
67
68 static void
ping_client_fini(struct sfw_test_instance * tsi)69 ping_client_fini(struct sfw_test_instance *tsi)
70 {
71 struct sfw_session *sn = tsi->tsi_batch->bat_session;
72 int errors;
73
74 LASSERT(sn);
75 LASSERT(tsi->tsi_is_client);
76
77 errors = atomic_read(&sn->sn_ping_errors);
78 if (errors)
79 CWARN("%d pings have failed.\n", errors);
80 else
81 CDEBUG(D_NET, "Ping test finished OK.\n");
82 }
83
84 static int
ping_client_prep_rpc(struct sfw_test_unit * tsu,lnet_process_id_t dest,struct srpc_client_rpc ** rpc)85 ping_client_prep_rpc(struct sfw_test_unit *tsu, lnet_process_id_t dest,
86 struct srpc_client_rpc **rpc)
87 {
88 struct srpc_ping_reqst *req;
89 struct sfw_test_instance *tsi = tsu->tsu_instance;
90 struct sfw_session *sn = tsi->tsi_batch->bat_session;
91 struct timespec64 ts;
92 int rc;
93
94 LASSERT(sn);
95 LASSERT(!(sn->sn_features & ~LST_FEATS_MASK));
96
97 rc = sfw_create_test_rpc(tsu, dest, sn->sn_features, 0, 0, rpc);
98 if (rc)
99 return rc;
100
101 req = &(*rpc)->crpc_reqstmsg.msg_body.ping_reqst;
102
103 req->pnr_magic = LST_PING_TEST_MAGIC;
104
105 spin_lock(&lst_ping_data.pnd_lock);
106 req->pnr_seq = lst_ping_data.pnd_counter++;
107 spin_unlock(&lst_ping_data.pnd_lock);
108
109 ktime_get_real_ts64(&ts);
110 req->pnr_time_sec = ts.tv_sec;
111 req->pnr_time_usec = ts.tv_nsec / NSEC_PER_USEC;
112
113 return rc;
114 }
115
116 static void
ping_client_done_rpc(struct sfw_test_unit * tsu,struct srpc_client_rpc * rpc)117 ping_client_done_rpc(struct sfw_test_unit *tsu, struct srpc_client_rpc *rpc)
118 {
119 struct sfw_test_instance *tsi = tsu->tsu_instance;
120 struct sfw_session *sn = tsi->tsi_batch->bat_session;
121 struct srpc_ping_reqst *reqst = &rpc->crpc_reqstmsg.msg_body.ping_reqst;
122 struct srpc_ping_reply *reply = &rpc->crpc_replymsg.msg_body.ping_reply;
123 struct timespec64 ts;
124
125 LASSERT(sn);
126
127 if (rpc->crpc_status) {
128 if (!tsi->tsi_stopping) /* rpc could have been aborted */
129 atomic_inc(&sn->sn_ping_errors);
130 CERROR("Unable to ping %s (%d): %d\n",
131 libcfs_id2str(rpc->crpc_dest),
132 reqst->pnr_seq, rpc->crpc_status);
133 return;
134 }
135
136 if (rpc->crpc_replymsg.msg_magic != SRPC_MSG_MAGIC) {
137 __swab32s(&reply->pnr_seq);
138 __swab32s(&reply->pnr_magic);
139 __swab32s(&reply->pnr_status);
140 }
141
142 if (reply->pnr_magic != LST_PING_TEST_MAGIC) {
143 rpc->crpc_status = -EBADMSG;
144 atomic_inc(&sn->sn_ping_errors);
145 CERROR("Bad magic %u from %s, %u expected.\n",
146 reply->pnr_magic, libcfs_id2str(rpc->crpc_dest),
147 LST_PING_TEST_MAGIC);
148 return;
149 }
150
151 if (reply->pnr_seq != reqst->pnr_seq) {
152 rpc->crpc_status = -EBADMSG;
153 atomic_inc(&sn->sn_ping_errors);
154 CERROR("Bad seq %u from %s, %u expected.\n",
155 reply->pnr_seq, libcfs_id2str(rpc->crpc_dest),
156 reqst->pnr_seq);
157 return;
158 }
159
160 ktime_get_real_ts64(&ts);
161 CDEBUG(D_NET, "%d reply in %u usec\n", reply->pnr_seq,
162 (unsigned)((ts.tv_sec - reqst->pnr_time_sec) * 1000000 +
163 (ts.tv_nsec / NSEC_PER_USEC - reqst->pnr_time_usec)));
164 }
165
166 static int
ping_server_handle(struct srpc_server_rpc * rpc)167 ping_server_handle(struct srpc_server_rpc *rpc)
168 {
169 struct srpc_service *sv = rpc->srpc_scd->scd_svc;
170 struct srpc_msg *reqstmsg = &rpc->srpc_reqstbuf->buf_msg;
171 struct srpc_msg *replymsg = &rpc->srpc_replymsg;
172 struct srpc_ping_reqst *req = &reqstmsg->msg_body.ping_reqst;
173 struct srpc_ping_reply *rep = &rpc->srpc_replymsg.msg_body.ping_reply;
174
175 LASSERT(sv->sv_id == SRPC_SERVICE_PING);
176
177 if (reqstmsg->msg_magic != SRPC_MSG_MAGIC) {
178 LASSERT(reqstmsg->msg_magic == __swab32(SRPC_MSG_MAGIC));
179
180 __swab32s(&req->pnr_seq);
181 __swab32s(&req->pnr_magic);
182 __swab64s(&req->pnr_time_sec);
183 __swab64s(&req->pnr_time_usec);
184 }
185 LASSERT(reqstmsg->msg_type == srpc_service2request(sv->sv_id));
186
187 if (req->pnr_magic != LST_PING_TEST_MAGIC) {
188 CERROR("Unexpected magic %08x from %s\n",
189 req->pnr_magic, libcfs_id2str(rpc->srpc_peer));
190 return -EINVAL;
191 }
192
193 rep->pnr_seq = req->pnr_seq;
194 rep->pnr_magic = LST_PING_TEST_MAGIC;
195
196 if (reqstmsg->msg_ses_feats & ~LST_FEATS_MASK) {
197 replymsg->msg_ses_feats = LST_FEATS_MASK;
198 rep->pnr_status = EPROTO;
199 return 0;
200 }
201
202 replymsg->msg_ses_feats = reqstmsg->msg_ses_feats;
203
204 CDEBUG(D_NET, "Get ping %d from %s\n",
205 req->pnr_seq, libcfs_id2str(rpc->srpc_peer));
206 return 0;
207 }
208
209 struct sfw_test_client_ops ping_test_client;
210
ping_init_test_client(void)211 void ping_init_test_client(void)
212 {
213 ping_test_client.tso_init = ping_client_init;
214 ping_test_client.tso_fini = ping_client_fini;
215 ping_test_client.tso_prep_rpc = ping_client_prep_rpc;
216 ping_test_client.tso_done_rpc = ping_client_done_rpc;
217 }
218
219 struct srpc_service ping_test_service;
220
ping_init_test_service(void)221 void ping_init_test_service(void)
222 {
223 ping_test_service.sv_id = SRPC_SERVICE_PING;
224 ping_test_service.sv_name = "ping_test";
225 ping_test_service.sv_handler = ping_server_handle;
226 ping_test_service.sv_wi_total = ping_srv_workitems;
227 }
228