Whamcloud - gitweb
LU-1346 libcfs: replace libcfs wrappers with kernel API
[fs/lustre-release.git] / lnet / selftest / ping_test.c
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.sun.com/software/products/lustre/docs/GPLv2.pdf
19  *
20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
21  * CA 95054 USA or visit www.sun.com if you need additional information or
22  * have any questions.
23  *
24  * GPL HEADER END
25  */
26 /*
27  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
28  * Use is subject to license terms.
29  */
30 /*
31  * This file is part of Lustre, http://www.lustre.org/
32  * Lustre is a trademark of Sun Microsystems, Inc.
33  *
34  * lnet/selftest/conctl.c
35  *
36  * Test client & Server
37  *
38  * Author: Liang Zhen <liangzhen@clusterfs.com>
39  */
40
41 #include "selftest.h"
42
43 #define LST_PING_TEST_MAGIC     0xbabeface
44
45 int ping_srv_workitems = SFW_TEST_WI_MAX;
46 CFS_MODULE_PARM(ping_srv_workitems, "i", int, 0644, "# PING server workitems");
47
48 typedef struct {
49         spinlock_t      pnd_lock;       /* serialize */
50         int             pnd_counter;    /* sequence counter */
51 } lst_ping_data_t;
52
53 static lst_ping_data_t  lst_ping_data;
54
55 static int
56 ping_client_init(sfw_test_instance_t *tsi)
57 {
58         sfw_session_t *sn = tsi->tsi_batch->bat_session;
59
60         LASSERT(tsi->tsi_is_client);
61         LASSERT(sn != NULL && (sn->sn_features & ~LST_FEATS_MASK) == 0);
62
63         spin_lock_init(&lst_ping_data.pnd_lock);
64         lst_ping_data.pnd_counter = 0;
65
66         return 0;
67 }
68
69 static void
70 ping_client_fini (sfw_test_instance_t *tsi)
71 {
72         sfw_session_t *sn = tsi->tsi_batch->bat_session;
73         int            errors;
74
75         LASSERT (sn != NULL);
76         LASSERT (tsi->tsi_is_client);
77
78         errors = cfs_atomic_read(&sn->sn_ping_errors);
79         if (errors)
80                 CWARN ("%d pings have failed.\n", errors);
81         else
82                 CDEBUG (D_NET, "Ping test finished OK.\n");
83 }
84
85 static int
86 ping_client_prep_rpc(sfw_test_unit_t *tsu,
87                      lnet_process_id_t dest, srpc_client_rpc_t **rpc)
88 {
89         srpc_ping_reqst_t   *req;
90         sfw_test_instance_t *tsi = tsu->tsu_instance;
91         sfw_session_t       *sn  = tsi->tsi_batch->bat_session;
92         struct timeval       tv;
93         int                  rc;
94
95         LASSERT(sn != NULL);
96         LASSERT((sn->sn_features & ~LST_FEATS_MASK) == 0);
97
98         rc = sfw_create_test_rpc(tsu, dest, sn->sn_features, 0, 0, rpc);
99         if (rc != 0)
100                 return rc;
101
102         req = &(*rpc)->crpc_reqstmsg.msg_body.ping_reqst;
103
104         req->pnr_magic = LST_PING_TEST_MAGIC;
105
106         spin_lock(&lst_ping_data.pnd_lock);
107         req->pnr_seq = lst_ping_data.pnd_counter++;
108         spin_unlock(&lst_ping_data.pnd_lock);
109
110         cfs_fs_timeval(&tv);
111         req->pnr_time_sec  = tv.tv_sec;
112         req->pnr_time_usec = tv.tv_usec;
113
114         return rc;
115 }
116
117 static void
118 ping_client_done_rpc (sfw_test_unit_t *tsu, srpc_client_rpc_t *rpc)
119 {
120         sfw_test_instance_t *tsi = tsu->tsu_instance;
121         sfw_session_t       *sn = tsi->tsi_batch->bat_session;
122         srpc_ping_reqst_t   *reqst = &rpc->crpc_reqstmsg.msg_body.ping_reqst;
123         srpc_ping_reply_t   *reply = &rpc->crpc_replymsg.msg_body.ping_reply;
124         struct timeval       tv;
125
126         LASSERT (sn != NULL);
127
128         if (rpc->crpc_status != 0) {
129                 if (!tsi->tsi_stopping) /* rpc could have been aborted */
130                         cfs_atomic_inc(&sn->sn_ping_errors);
131                 CERROR ("Unable to ping %s (%d): %d\n",
132                         libcfs_id2str(rpc->crpc_dest),
133                         reqst->pnr_seq, rpc->crpc_status);
134                 return;
135         }
136
137         if (rpc->crpc_replymsg.msg_magic != SRPC_MSG_MAGIC) {
138                 __swab32s(&reply->pnr_seq);
139                 __swab32s(&reply->pnr_magic);
140                 __swab32s(&reply->pnr_status);
141         }
142
143         if (reply->pnr_magic != LST_PING_TEST_MAGIC) {
144                 rpc->crpc_status = -EBADMSG;
145                 cfs_atomic_inc(&sn->sn_ping_errors);
146                 CERROR ("Bad magic %u from %s, %u expected.\n",
147                         reply->pnr_magic, libcfs_id2str(rpc->crpc_dest),
148                         LST_PING_TEST_MAGIC);
149                 return;
150         }
151
152         if (reply->pnr_seq != reqst->pnr_seq) {
153                 rpc->crpc_status = -EBADMSG;
154                 cfs_atomic_inc(&sn->sn_ping_errors);
155                 CERROR ("Bad seq %u from %s, %u expected.\n",
156                         reply->pnr_seq, libcfs_id2str(rpc->crpc_dest),
157                         reqst->pnr_seq);
158                 return;
159         }
160
161         cfs_fs_timeval(&tv);
162         CDEBUG (D_NET, "%d reply in %u usec\n", reply->pnr_seq,
163                 (unsigned)((tv.tv_sec - (unsigned)reqst->pnr_time_sec) * 1000000
164                            + (tv.tv_usec - reqst->pnr_time_usec)));
165         return;
166 }
167
168 static int
169 ping_server_handle(struct srpc_server_rpc *rpc)
170 {
171         struct srpc_service     *sv  = rpc->srpc_scd->scd_svc;
172         srpc_msg_t        *reqstmsg = &rpc->srpc_reqstbuf->buf_msg;
173         srpc_msg_t        *replymsg = &rpc->srpc_replymsg;
174         srpc_ping_reqst_t *req = &reqstmsg->msg_body.ping_reqst;
175         srpc_ping_reply_t *rep = &rpc->srpc_replymsg.msg_body.ping_reply;
176
177         LASSERT (sv->sv_id == SRPC_SERVICE_PING);
178
179         if (reqstmsg->msg_magic != SRPC_MSG_MAGIC) {
180                 LASSERT (reqstmsg->msg_magic == __swab32(SRPC_MSG_MAGIC));
181
182                 __swab32s(&req->pnr_seq);
183                 __swab32s(&req->pnr_magic);
184                 __swab64s(&req->pnr_time_sec);
185                 __swab64s(&req->pnr_time_usec);
186         }
187         LASSERT (reqstmsg->msg_type == srpc_service2request(sv->sv_id));
188
189         if (req->pnr_magic != LST_PING_TEST_MAGIC) {
190                 CERROR ("Unexpect magic %08x from %s\n",
191                         req->pnr_magic, libcfs_id2str(rpc->srpc_peer));
192                 return -EINVAL;
193         }
194
195         rep->pnr_seq   = req->pnr_seq;
196         rep->pnr_magic = LST_PING_TEST_MAGIC;
197
198         if ((reqstmsg->msg_ses_feats & ~LST_FEATS_MASK) != 0) {
199                 replymsg->msg_ses_feats = LST_FEATS_MASK;
200                 rep->pnr_status = EPROTO;
201                 return 0;
202         }
203
204         replymsg->msg_ses_feats = reqstmsg->msg_ses_feats;
205
206         CDEBUG(D_NET, "Get ping %d from %s\n",
207                req->pnr_seq, libcfs_id2str(rpc->srpc_peer));
208         return 0;
209 }
210
211 sfw_test_client_ops_t ping_test_client;
212 void ping_init_test_client(void)
213 {
214         ping_test_client.tso_init     = ping_client_init;
215         ping_test_client.tso_fini     = ping_client_fini;
216         ping_test_client.tso_prep_rpc = ping_client_prep_rpc;
217         ping_test_client.tso_done_rpc = ping_client_done_rpc;
218 }
219
220 srpc_service_t ping_test_service;
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 }