1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2 * vim:expandtab:shiftwidth=8:tabstop=8:
6 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License version 2 only,
10 * as published by the Free Software Foundation.
12 * This program is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * General Public License version 2 for more details (a copy is included
16 * in the LICENSE file that accompanied this code).
18 * You should have received a copy of the GNU General Public License
19 * version 2 along with this program; If not, see [sun.com URL with a
22 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
23 * CA 95054 USA or visit www.sun.com if you need additional information or
29 * Copyright 2008 Sun Microsystems, Inc. All rights reserved
30 * Use is subject to license terms.
33 * This file is part of Lustre, http://www.lustre.org/
34 * Lustre is a trademark of Sun Microsystems, Inc.
38 * Author: Isaac Huang <isaac@clusterfs.com>
41 #define DEBUG_SUBSYSTEM S_LNET
46 #define SRPC_PEER_HASH_SIZE 101 /* # peer lists */
57 struct smoketest_rpc {
58 spinlock_t rpc_glock; /* global lock */
59 srpc_service_t *rpc_services[SRPC_SERVICE_MAX_ID + 1];
60 struct list_head *rpc_peers; /* hash table of known peers */
61 lnet_handle_eq_t rpc_lnet_eq; /* _the_ LNet event queue */
62 srpc_state_t rpc_state;
63 srpc_counters_t rpc_counters;
64 __u64 rpc_matchbits; /* matchbits counter */
67 static int srpc_peer_credits = 16;
68 CFS_MODULE_PARM(srpc_peer_credits, "i", int, 0444,
69 "# in-flight RPCs per peer (16 by default)");
72 int srpc_handle_rpc (swi_workitem_t *wi);
74 void srpc_get_counters (srpc_counters_t *cnt)
76 spin_lock(&srpc_data.rpc_glock);
77 *cnt = srpc_data.rpc_counters;
78 spin_unlock(&srpc_data.rpc_glock);
81 void srpc_set_counters (const srpc_counters_t *cnt)
83 spin_lock(&srpc_data.rpc_glock);
84 srpc_data.rpc_counters = *cnt;
85 spin_unlock(&srpc_data.rpc_glock);
89 srpc_add_bulk_page (srpc_bulk_t *bk, cfs_page_t *pg, int i)
91 LASSERT (i >= 0 && i < bk->bk_niov);
94 bk->bk_iovs[i].kiov_offset = 0;
95 bk->bk_iovs[i].kiov_page = pg;
96 bk->bk_iovs[i].kiov_len = CFS_PAGE_SIZE;
98 LASSERT (bk->bk_pages != NULL);
100 bk->bk_pages[i] = pg;
101 bk->bk_iovs[i].iov_len = CFS_PAGE_SIZE;
102 bk->bk_iovs[i].iov_base = cfs_page_address(pg);
108 srpc_free_bulk (srpc_bulk_t *bk)
113 LASSERT (bk != NULL);
115 LASSERT (bk->bk_pages != NULL);
118 for (i = 0; i < bk->bk_niov; i++) {
120 pg = bk->bk_iovs[i].kiov_page;
122 pg = bk->bk_pages[i];
124 if (pg == NULL) break;
130 LIBCFS_FREE(bk->bk_pages, sizeof(cfs_page_t *) * bk->bk_niov);
132 LIBCFS_FREE(bk, offsetof(srpc_bulk_t, bk_iovs[bk->bk_niov]));
137 srpc_alloc_bulk (int npages, int sink)
143 LASSERT (npages > 0 && npages <= LNET_MAX_IOV);
145 LIBCFS_ALLOC(bk, offsetof(srpc_bulk_t, bk_iovs[npages]));
147 CERROR ("Can't allocate descriptor for %d pages\n", npages);
151 memset(bk, 0, offsetof(srpc_bulk_t, bk_iovs[npages]));
153 bk->bk_niov = npages;
154 bk->bk_len = npages * CFS_PAGE_SIZE;
156 LIBCFS_ALLOC(pages, sizeof(cfs_page_t *) * npages);
158 LIBCFS_FREE(bk, offsetof(srpc_bulk_t, bk_iovs[npages]));
159 CERROR ("Can't allocate page array for %d pages\n", npages);
163 memset(pages, 0, sizeof(cfs_page_t *) * npages);
164 bk->bk_pages = pages;
169 for (i = 0; i < npages; i++) {
170 cfs_page_t *pg = cfs_alloc_page(CFS_ALLOC_STD);
173 CERROR ("Can't allocate page %d of %d\n", i, npages);
178 srpc_add_bulk_page(bk, pg, i);
185 static inline struct list_head *
186 srpc_nid2peerlist (lnet_nid_t nid)
188 unsigned int hash = ((unsigned int)nid) % SRPC_PEER_HASH_SIZE;
190 return &srpc_data.rpc_peers[hash];
193 static inline srpc_peer_t *
194 srpc_create_peer (lnet_nid_t nid)
198 LASSERT (nid != LNET_NID_ANY);
200 LIBCFS_ALLOC(peer, sizeof(srpc_peer_t));
202 CERROR ("Failed to allocate peer structure for %s\n",
203 libcfs_nid2str(nid));
207 memset(peer, 0, sizeof(srpc_peer_t));
209 peer->stp_credits = srpc_peer_credits;
211 spin_lock_init(&peer->stp_lock);
212 CFS_INIT_LIST_HEAD(&peer->stp_rpcq);
213 CFS_INIT_LIST_HEAD(&peer->stp_ctl_rpcq);
218 srpc_find_peer_locked (lnet_nid_t nid)
220 struct list_head *peer_list = srpc_nid2peerlist(nid);
223 LASSERT (nid != LNET_NID_ANY);
225 list_for_each_entry (peer, peer_list, stp_list) {
226 if (peer->stp_nid == nid)
234 srpc_nid2peer (lnet_nid_t nid)
237 srpc_peer_t *new_peer;
239 spin_lock(&srpc_data.rpc_glock);
240 peer = srpc_find_peer_locked(nid);
241 spin_unlock(&srpc_data.rpc_glock);
246 new_peer = srpc_create_peer(nid);
248 spin_lock(&srpc_data.rpc_glock);
250 peer = srpc_find_peer_locked(nid);
252 spin_unlock(&srpc_data.rpc_glock);
253 if (new_peer != NULL)
254 LIBCFS_FREE(new_peer, sizeof(srpc_peer_t));
259 if (new_peer == NULL) {
260 spin_unlock(&srpc_data.rpc_glock);
264 list_add_tail(&new_peer->stp_list, srpc_nid2peerlist(nid));
265 spin_unlock(&srpc_data.rpc_glock);
274 spin_lock(&srpc_data.rpc_glock);
275 id = srpc_data.rpc_matchbits++;
276 spin_unlock(&srpc_data.rpc_glock);
281 srpc_init_server_rpc (srpc_server_rpc_t *rpc,
282 srpc_service_t *sv, srpc_buffer_t *buffer)
284 memset(rpc, 0, sizeof(*rpc));
285 swi_init_workitem(&rpc->srpc_wi, rpc, srpc_handle_rpc);
287 rpc->srpc_ev.ev_fired = 1; /* no event expected now */
289 rpc->srpc_service = sv;
290 rpc->srpc_reqstbuf = buffer;
291 rpc->srpc_peer = buffer->buf_peer;
292 rpc->srpc_self = buffer->buf_self;
293 rpc->srpc_replymdh = LNET_INVALID_HANDLE;
297 srpc_add_service (srpc_service_t *sv)
301 srpc_server_rpc_t *rpc;
303 LASSERT (sv->sv_concur > 0);
304 LASSERT (0 <= id && id <= SRPC_SERVICE_MAX_ID);
306 spin_lock(&srpc_data.rpc_glock);
308 LASSERT (srpc_data.rpc_state == SRPC_STATE_RUNNING);
310 if (srpc_data.rpc_services[id] != NULL) {
311 spin_unlock(&srpc_data.rpc_glock);
315 srpc_data.rpc_services[id] = sv;
316 spin_unlock(&srpc_data.rpc_glock);
319 sv->sv_nposted_msg = 0;
320 sv->sv_shuttingdown = 0;
321 spin_lock_init(&sv->sv_lock);
322 CFS_INIT_LIST_HEAD(&sv->sv_free_rpcq);
323 CFS_INIT_LIST_HEAD(&sv->sv_active_rpcq);
324 CFS_INIT_LIST_HEAD(&sv->sv_posted_msgq);
325 CFS_INIT_LIST_HEAD(&sv->sv_blocked_msgq);
327 sv->sv_ev.ev_data = sv;
328 sv->sv_ev.ev_type = SRPC_REQUEST_RCVD;
330 for (i = 0; i < sv->sv_concur; i++) {
331 LIBCFS_ALLOC(rpc, sizeof(*rpc));
332 if (rpc == NULL) goto enomem;
334 list_add(&rpc->srpc_list, &sv->sv_free_rpcq);
337 CDEBUG (D_NET, "Adding service: id %d, name %s, concurrency %d\n",
338 id, sv->sv_name, sv->sv_concur);
342 while (!list_empty(&sv->sv_free_rpcq)) {
343 rpc = list_entry(sv->sv_free_rpcq.next,
344 srpc_server_rpc_t, srpc_list);
345 list_del(&rpc->srpc_list);
346 LIBCFS_FREE(rpc, sizeof(*rpc));
349 spin_lock(&srpc_data.rpc_glock);
350 srpc_data.rpc_services[id] = NULL;
351 spin_unlock(&srpc_data.rpc_glock);
356 srpc_remove_service (srpc_service_t *sv)
360 spin_lock(&srpc_data.rpc_glock);
362 if (srpc_data.rpc_services[id] != sv) {
363 spin_unlock(&srpc_data.rpc_glock);
367 srpc_data.rpc_services[id] = NULL;
368 spin_unlock(&srpc_data.rpc_glock);
373 srpc_post_passive_rdma(int portal, __u64 matchbits, void *buf,
374 int len, int options, lnet_process_id_t peer,
375 lnet_handle_md_t *mdh, srpc_event_t *ev)
379 lnet_handle_me_t meh;
381 rc = LNetMEAttach(portal, peer, matchbits, 0,
382 LNET_UNLINK, LNET_INS_AFTER, &meh);
384 CERROR ("LNetMEAttach failed: %d\n", rc);
385 LASSERT (rc == -ENOMEM);
393 md.options = options;
394 md.eq_handle = srpc_data.rpc_lnet_eq;
396 rc = LNetMDAttach(meh, md, LNET_UNLINK, mdh);
398 CERROR ("LNetMDAttach failed: %d\n", rc);
399 LASSERT (rc == -ENOMEM);
401 rc = LNetMEUnlink(meh);
407 "Posted passive RDMA: peer %s, portal %d, matchbits "LPX64"\n",
408 libcfs_id2str(peer), portal, matchbits);
413 srpc_post_active_rdma(int portal, __u64 matchbits, void *buf, int len,
414 int options, lnet_process_id_t peer, lnet_nid_t self,
415 lnet_handle_md_t *mdh, srpc_event_t *ev)
423 md.eq_handle = srpc_data.rpc_lnet_eq;
424 md.threshold = ((options & LNET_MD_OP_GET) != 0) ? 2 : 1;
425 md.options = options & ~(LNET_MD_OP_PUT | LNET_MD_OP_GET);
427 rc = LNetMDBind(md, LNET_UNLINK, mdh);
429 CERROR ("LNetMDBind failed: %d\n", rc);
430 LASSERT (rc == -ENOMEM);
434 /* this is kind of an abuse of the LNET_MD_OP_{PUT,GET} options.
435 * they're only meaningful for MDs attached to an ME (i.e. passive
437 if ((options & LNET_MD_OP_PUT) != 0) {
438 rc = LNetPut(self, *mdh, LNET_NOACK_REQ, peer,
439 portal, matchbits, 0, 0);
441 LASSERT ((options & LNET_MD_OP_GET) != 0);
443 rc = LNetGet(self, *mdh, peer, portal, matchbits, 0);
447 CERROR ("LNet%s(%s, %d, "LPD64") failed: %d\n",
448 ((options & LNET_MD_OP_PUT) != 0) ? "Put" : "Get",
449 libcfs_id2str(peer), portal, matchbits, rc);
451 /* The forthcoming unlink event will complete this operation
452 * with failure, so fall through and return success here.
454 rc = LNetMDUnlink(*mdh);
458 "Posted active RDMA: peer %s, portal %u, matchbits "LPX64"\n",
459 libcfs_id2str(peer), portal, matchbits);
465 srpc_post_active_rqtbuf(lnet_process_id_t peer, int service, void *buf,
466 int len, lnet_handle_md_t *mdh, srpc_event_t *ev)
471 if (service > SRPC_FRAMEWORK_SERVICE_MAX_ID)
472 portal = SRPC_REQUEST_PORTAL;
474 portal = SRPC_FRAMEWORK_REQUEST_PORTAL;
476 rc = srpc_post_active_rdma(portal, service, buf, len,
477 LNET_MD_OP_PUT, peer,
478 LNET_NID_ANY, mdh, ev);
483 srpc_post_passive_rqtbuf(int service, void *buf, int len,
484 lnet_handle_md_t *mdh, srpc_event_t *ev)
488 lnet_process_id_t any = {.nid = LNET_NID_ANY,
489 .pid = LNET_PID_ANY};
491 if (service > SRPC_FRAMEWORK_SERVICE_MAX_ID)
492 portal = SRPC_REQUEST_PORTAL;
494 portal = SRPC_FRAMEWORK_REQUEST_PORTAL;
496 rc = srpc_post_passive_rdma(portal, service, buf, len,
497 LNET_MD_OP_PUT, any, mdh, ev);
502 srpc_service_post_buffer (srpc_service_t *sv, srpc_buffer_t *buf)
504 srpc_msg_t *msg = &buf->buf_msg;
507 LASSERT (!sv->sv_shuttingdown);
509 buf->buf_mdh = LNET_INVALID_HANDLE;
510 list_add(&buf->buf_list, &sv->sv_posted_msgq);
511 sv->sv_nposted_msg++;
512 spin_unlock(&sv->sv_lock);
514 rc = srpc_post_passive_rqtbuf(sv->sv_id, msg, sizeof(*msg),
515 &buf->buf_mdh, &sv->sv_ev);
517 /* At this point, a RPC (new or delayed) may have arrived in
518 * msg and its event handler has been called. So we must add
519 * buf to sv_posted_msgq _before_ dropping sv_lock */
521 spin_lock(&sv->sv_lock);
524 if (sv->sv_shuttingdown) {
525 spin_unlock(&sv->sv_lock);
527 /* srpc_shutdown_service might have tried to unlink me
528 * when my buf_mdh was still invalid */
529 LNetMDUnlink(buf->buf_mdh);
531 spin_lock(&sv->sv_lock);
536 sv->sv_nposted_msg--;
537 if (sv->sv_shuttingdown) return rc;
539 list_del(&buf->buf_list);
541 spin_unlock(&sv->sv_lock);
542 LIBCFS_FREE(buf, sizeof(*buf));
543 spin_lock(&sv->sv_lock);
548 srpc_service_add_buffers (srpc_service_t *sv, int nbuffer)
554 LASSERTF (nbuffer > 0,
555 "nbuffer must be positive: %d\n", nbuffer);
557 for (posted = 0; posted < nbuffer; posted++) {
558 LIBCFS_ALLOC(buf, sizeof(*buf));
559 if (buf == NULL) break;
561 spin_lock(&sv->sv_lock);
562 rc = srpc_service_post_buffer(sv, buf);
563 spin_unlock(&sv->sv_lock);
572 srpc_service_remove_buffers (srpc_service_t *sv, int nbuffer)
574 LASSERTF (nbuffer > 0,
575 "nbuffer must be positive: %d\n", nbuffer);
577 spin_lock(&sv->sv_lock);
579 LASSERT (sv->sv_nprune >= 0);
580 LASSERT (!sv->sv_shuttingdown);
582 sv->sv_nprune += nbuffer;
584 spin_unlock(&sv->sv_lock);
588 /* returns 1 if sv has finished, otherwise 0 */
590 srpc_finish_service (srpc_service_t *sv)
592 srpc_server_rpc_t *rpc;
595 spin_lock(&sv->sv_lock);
597 LASSERT (sv->sv_shuttingdown); /* srpc_shutdown_service called */
599 if (sv->sv_nposted_msg != 0 || !list_empty(&sv->sv_active_rpcq)) {
601 "waiting for %d posted buffers to unlink and "
602 "in-flight RPCs to die.\n",
605 if (!list_empty(&sv->sv_active_rpcq)) {
606 rpc = list_entry(sv->sv_active_rpcq.next,
607 srpc_server_rpc_t, srpc_list);
609 "Active RPC on shutdown: sv %s, peer %s, "
610 "wi %s scheduled %d running %d, "
611 "ev fired %d type %d status %d lnet %d\n",
612 sv->sv_name, libcfs_id2str(rpc->srpc_peer),
613 swi_state2str(rpc->srpc_wi.wi_state),
614 rpc->srpc_wi.wi_scheduled,
615 rpc->srpc_wi.wi_running,
616 rpc->srpc_ev.ev_fired,
617 rpc->srpc_ev.ev_type,
618 rpc->srpc_ev.ev_status,
619 rpc->srpc_ev.ev_lnet);
622 spin_unlock(&sv->sv_lock);
626 spin_unlock(&sv->sv_lock); /* no lock needed from now on */
631 if (!list_empty(&sv->sv_posted_msgq))
632 q = &sv->sv_posted_msgq;
633 else if (!list_empty(&sv->sv_blocked_msgq))
634 q = &sv->sv_blocked_msgq;
638 buf = list_entry(q->next, srpc_buffer_t, buf_list);
639 list_del(&buf->buf_list);
641 LIBCFS_FREE(buf, sizeof(*buf));
644 while (!list_empty(&sv->sv_free_rpcq)) {
645 rpc = list_entry(sv->sv_free_rpcq.next,
646 srpc_server_rpc_t, srpc_list);
647 list_del(&rpc->srpc_list);
648 LIBCFS_FREE(rpc, sizeof(*rpc));
654 /* called with sv->sv_lock held */
656 srpc_service_recycle_buffer (srpc_service_t *sv, srpc_buffer_t *buf)
658 if (sv->sv_shuttingdown) goto free;
660 if (sv->sv_nprune == 0) {
661 if (srpc_service_post_buffer(sv, buf) != 0)
662 CWARN ("Failed to post %s buffer\n", sv->sv_name);
668 spin_unlock(&sv->sv_lock);
669 LIBCFS_FREE(buf, sizeof(*buf));
670 spin_lock(&sv->sv_lock);
674 srpc_shutdown_service (srpc_service_t *sv)
676 srpc_server_rpc_t *rpc;
679 spin_lock(&sv->sv_lock);
681 CDEBUG (D_NET, "Shutting down service: id %d, name %s\n",
682 sv->sv_id, sv->sv_name);
684 sv->sv_shuttingdown = 1; /* i.e. no new active RPC */
686 /* schedule in-flight RPCs to notice the shutdown */
687 list_for_each_entry (rpc, &sv->sv_active_rpcq, srpc_list) {
688 swi_schedule_workitem(&rpc->srpc_wi);
691 spin_unlock(&sv->sv_lock);
693 /* OK to traverse sv_posted_msgq without lock, since no one
694 * touches sv_posted_msgq now */
695 list_for_each_entry (buf, &sv->sv_posted_msgq, buf_list)
696 LNetMDUnlink(buf->buf_mdh);
702 srpc_send_request (srpc_client_rpc_t *rpc)
704 srpc_event_t *ev = &rpc->crpc_reqstev;
709 ev->ev_type = SRPC_REQUEST_SENT;
711 rc = srpc_post_active_rqtbuf(rpc->crpc_dest, rpc->crpc_service,
712 &rpc->crpc_reqstmsg, sizeof(srpc_msg_t),
713 &rpc->crpc_reqstmdh, ev);
715 LASSERT (rc == -ENOMEM);
716 ev->ev_fired = 1; /* no more event expected */
722 srpc_prepare_reply (srpc_client_rpc_t *rpc)
724 srpc_event_t *ev = &rpc->crpc_replyev;
725 __u64 *id = &rpc->crpc_reqstmsg.msg_body.reqst.rpyid;
730 ev->ev_type = SRPC_REPLY_RCVD;
732 *id = srpc_next_id();
734 rc = srpc_post_passive_rdma(SRPC_RDMA_PORTAL, *id,
735 &rpc->crpc_replymsg, sizeof(srpc_msg_t),
736 LNET_MD_OP_PUT, rpc->crpc_dest,
737 &rpc->crpc_replymdh, ev);
739 LASSERT (rc == -ENOMEM);
740 ev->ev_fired = 1; /* no more event expected */
746 srpc_prepare_bulk (srpc_client_rpc_t *rpc)
748 srpc_bulk_t *bk = &rpc->crpc_bulk;
749 srpc_event_t *ev = &rpc->crpc_bulkev;
750 __u64 *id = &rpc->crpc_reqstmsg.msg_body.reqst.bulkid;
754 LASSERT (bk->bk_niov <= LNET_MAX_IOV);
756 if (bk->bk_niov == 0) return 0; /* nothing to do */
758 opt = bk->bk_sink ? LNET_MD_OP_PUT : LNET_MD_OP_GET;
762 opt |= LNET_MD_IOVEC;
767 ev->ev_type = SRPC_BULK_REQ_RCVD;
769 *id = srpc_next_id();
771 rc = srpc_post_passive_rdma(SRPC_RDMA_PORTAL, *id,
772 &bk->bk_iovs[0], bk->bk_niov, opt,
773 rpc->crpc_dest, &bk->bk_mdh, ev);
775 LASSERT (rc == -ENOMEM);
776 ev->ev_fired = 1; /* no more event expected */
782 srpc_do_bulk (srpc_server_rpc_t *rpc)
784 srpc_event_t *ev = &rpc->srpc_ev;
785 srpc_bulk_t *bk = rpc->srpc_bulk;
786 __u64 id = rpc->srpc_reqstbuf->buf_msg.msg_body.reqst.bulkid;
790 LASSERT (bk != NULL);
792 opt = bk->bk_sink ? LNET_MD_OP_GET : LNET_MD_OP_PUT;
796 opt |= LNET_MD_IOVEC;
801 ev->ev_type = bk->bk_sink ? SRPC_BULK_GET_RPLD : SRPC_BULK_PUT_SENT;
803 rc = srpc_post_active_rdma(SRPC_RDMA_PORTAL, id,
804 &bk->bk_iovs[0], bk->bk_niov, opt,
805 rpc->srpc_peer, rpc->srpc_self,
808 ev->ev_fired = 1; /* no more event expected */
812 /* called with srpc_service_t::sv_lock held */
814 srpc_schedule_server_rpc (srpc_server_rpc_t *rpc)
816 srpc_service_t *sv = rpc->srpc_service;
818 if (sv->sv_id > SRPC_FRAMEWORK_SERVICE_MAX_ID)
819 swi_schedule_workitem(&rpc->srpc_wi);
820 else /* framework RPCs are handled one by one */
821 swi_schedule_serial_workitem(&rpc->srpc_wi);
826 /* only called from srpc_handle_rpc */
828 srpc_server_rpc_done (srpc_server_rpc_t *rpc, int status)
830 srpc_service_t *sv = rpc->srpc_service;
831 srpc_buffer_t *buffer;
833 LASSERT (status != 0 || rpc->srpc_wi.wi_state == SWI_STATE_DONE);
835 rpc->srpc_status = status;
837 CDEBUG (status == 0 ? D_NET : D_NETERROR,
838 "Server RPC done: service %s, peer %s, status %s:%d\n",
839 sv->sv_name, libcfs_id2str(rpc->srpc_peer),
840 swi_state2str(rpc->srpc_wi.wi_state), status);
843 spin_lock(&srpc_data.rpc_glock);
844 srpc_data.rpc_counters.rpcs_dropped++;
845 spin_unlock(&srpc_data.rpc_glock);
848 if (rpc->srpc_done != NULL)
849 (*rpc->srpc_done) (rpc);
850 LASSERT (rpc->srpc_bulk == NULL);
852 spin_lock(&sv->sv_lock);
854 if (rpc->srpc_reqstbuf != NULL) {
855 /* NB might drop sv_lock in srpc_service_recycle_buffer, but
856 * sv won't go away for sv_active_rpcq must not be empty */
857 srpc_service_recycle_buffer(sv, rpc->srpc_reqstbuf);
858 rpc->srpc_reqstbuf = NULL;
861 list_del(&rpc->srpc_list); /* from sv->sv_active_rpcq */
864 * No one can schedule me now since:
865 * - I'm not on sv_active_rpcq.
866 * - all LNet events have been fired.
867 * Cancel pending schedules and prevent future schedule attempts:
869 LASSERT (rpc->srpc_ev.ev_fired);
870 swi_kill_workitem(&rpc->srpc_wi);
872 if (!sv->sv_shuttingdown && !list_empty(&sv->sv_blocked_msgq)) {
873 buffer = list_entry(sv->sv_blocked_msgq.next,
874 srpc_buffer_t, buf_list);
875 list_del(&buffer->buf_list);
877 srpc_init_server_rpc(rpc, sv, buffer);
878 list_add_tail(&rpc->srpc_list, &sv->sv_active_rpcq);
879 srpc_schedule_server_rpc(rpc);
881 list_add(&rpc->srpc_list, &sv->sv_free_rpcq);
884 spin_unlock(&sv->sv_lock);
888 /* handles an incoming RPC */
890 srpc_handle_rpc (swi_workitem_t *wi)
892 srpc_server_rpc_t *rpc = wi->wi_data;
893 srpc_service_t *sv = rpc->srpc_service;
894 srpc_event_t *ev = &rpc->srpc_ev;
897 LASSERT (wi == &rpc->srpc_wi);
899 spin_lock(&sv->sv_lock);
901 if (sv->sv_shuttingdown) {
902 spin_unlock(&sv->sv_lock);
904 if (rpc->srpc_bulk != NULL)
905 LNetMDUnlink(rpc->srpc_bulk->bk_mdh);
906 LNetMDUnlink(rpc->srpc_replymdh);
908 if (ev->ev_fired) { /* no more event, OK to finish */
909 srpc_server_rpc_done(rpc, -ESHUTDOWN);
915 spin_unlock(&sv->sv_lock);
917 switch (wi->wi_state) {
920 case SWI_STATE_NEWBORN: {
922 srpc_generic_reply_t *reply;
924 msg = &rpc->srpc_reqstbuf->buf_msg;
925 reply = &rpc->srpc_replymsg.msg_body.reply;
927 if (msg->msg_version != SRPC_MSG_VERSION &&
928 msg->msg_version != __swab32(SRPC_MSG_VERSION)) {
929 CWARN ("Version mismatch: %u, %u expected, from %s\n",
930 msg->msg_version, SRPC_MSG_VERSION,
931 libcfs_id2str(rpc->srpc_peer));
932 reply->status = EPROTO;
935 rc = (*sv->sv_handler) (rpc);
936 LASSERT (reply->status == 0 || !rpc->srpc_bulk);
940 srpc_server_rpc_done(rpc, rc);
944 wi->wi_state = SWI_STATE_BULK_STARTED;
946 if (rpc->srpc_bulk != NULL) {
947 rc = srpc_do_bulk(rpc);
949 return 0; /* wait for bulk */
951 LASSERT (ev->ev_fired);
955 case SWI_STATE_BULK_STARTED:
956 /* we cannot LASSERT ev_fired right here because it
957 * may be set only upon an event with unlinked==1 */
959 if (rpc->srpc_bulk != NULL) {
962 if (sv->sv_bulk_ready != NULL)
963 rc = (*sv->sv_bulk_ready) (rpc, rc);
967 srpc_server_rpc_done(rpc, rc);
971 rpc->srpc_status = rc;
972 wi->wi_state = SWI_STATE_BULK_ERRORED;
973 LNetMDUnlink(rpc->srpc_bulk->bk_mdh);
974 return 0; /* wait for UNLINK event */
978 LASSERT (rpc->srpc_bulk == NULL || ev->ev_fired);
980 wi->wi_state = SWI_STATE_REPLY_SUBMITTED;
981 rc = srpc_send_reply(rpc);
983 return 0; /* wait for reply */
984 srpc_server_rpc_done(rpc, rc);
987 case SWI_STATE_REPLY_SUBMITTED:
988 LASSERT (ev->ev_fired);
990 wi->wi_state = SWI_STATE_DONE;
991 srpc_server_rpc_done(rpc, ev->ev_status);
994 case SWI_STATE_BULK_ERRORED:
995 LASSERT (rpc->srpc_bulk != NULL && ev->ev_fired);
996 LASSERT (rpc->srpc_status != 0);
998 srpc_server_rpc_done(rpc, rpc->srpc_status);
1006 srpc_client_rpc_expired (void *data)
1008 srpc_client_rpc_t *rpc = data;
1010 CWARN ("Client RPC expired: service %d, peer %s, timeout %d.\n",
1011 rpc->crpc_service, libcfs_id2str(rpc->crpc_dest),
1014 spin_lock(&rpc->crpc_lock);
1016 rpc->crpc_timeout = 0;
1017 srpc_abort_rpc(rpc, -ETIMEDOUT);
1019 spin_unlock(&rpc->crpc_lock);
1021 spin_lock(&srpc_data.rpc_glock);
1022 srpc_data.rpc_counters.rpcs_expired++;
1023 spin_unlock(&srpc_data.rpc_glock);
1028 srpc_add_client_rpc_timer (srpc_client_rpc_t *rpc)
1030 stt_timer_t *timer = &rpc->crpc_timer;
1032 if (rpc->crpc_timeout == 0) return;
1034 CFS_INIT_LIST_HEAD(&timer->stt_list);
1035 timer->stt_data = rpc;
1036 timer->stt_func = srpc_client_rpc_expired;
1037 timer->stt_expires = cfs_time_add(rpc->crpc_timeout,
1038 cfs_time_current_sec());
1039 stt_add_timer(timer);
1044 * Called with rpc->crpc_lock held.
1046 * Upon exit the RPC expiry timer is not queued and the handler is not
1047 * running on any CPU. */
1049 srpc_del_client_rpc_timer (srpc_client_rpc_t *rpc)
1051 /* timer not planted or already exploded */
1052 if (rpc->crpc_timeout == 0) return;
1054 /* timer sucessfully defused */
1055 if (stt_del_timer(&rpc->crpc_timer)) return;
1058 /* timer detonated, wait for it to explode */
1059 while (rpc->crpc_timeout != 0) {
1060 spin_unlock(&rpc->crpc_lock);
1064 spin_lock(&rpc->crpc_lock);
1067 LBUG(); /* impossible in single-threaded runtime */
1073 srpc_check_sends (srpc_peer_t *peer, int credits)
1075 struct list_head *q;
1076 srpc_client_rpc_t *rpc;
1078 LASSERT (credits >= 0);
1079 LASSERT (srpc_data.rpc_state == SRPC_STATE_RUNNING);
1081 spin_lock(&peer->stp_lock);
1082 peer->stp_credits += credits;
1084 while (peer->stp_credits) {
1085 if (!list_empty(&peer->stp_ctl_rpcq))
1086 q = &peer->stp_ctl_rpcq;
1087 else if (!list_empty(&peer->stp_rpcq))
1088 q = &peer->stp_rpcq;
1092 peer->stp_credits--;
1094 rpc = list_entry(q->next, srpc_client_rpc_t, crpc_privl);
1095 list_del_init(&rpc->crpc_privl);
1096 srpc_client_rpc_decref(rpc); /* --ref for peer->*rpcq */
1098 swi_schedule_workitem(&rpc->crpc_wi);
1101 spin_unlock(&peer->stp_lock);
1106 srpc_client_rpc_done (srpc_client_rpc_t *rpc, int status)
1108 swi_workitem_t *wi = &rpc->crpc_wi;
1109 srpc_peer_t *peer = rpc->crpc_peer;
1111 LASSERT (status != 0 || wi->wi_state == SWI_STATE_DONE);
1113 spin_lock(&rpc->crpc_lock);
1115 rpc->crpc_closed = 1;
1116 if (rpc->crpc_status == 0)
1117 rpc->crpc_status = status;
1119 srpc_del_client_rpc_timer(rpc);
1121 CDEBUG ((status == 0) ? D_NET : D_NETERROR,
1122 "Client RPC done: service %d, peer %s, status %s:%d:%d\n",
1123 rpc->crpc_service, libcfs_id2str(rpc->crpc_dest),
1124 swi_state2str(wi->wi_state), rpc->crpc_aborted, status);
1127 * No one can schedule me now since:
1128 * - RPC timer has been defused.
1129 * - all LNet events have been fired.
1130 * - crpc_closed has been set, preventing srpc_abort_rpc from
1132 * Cancel pending schedules and prevent future schedule attempts:
1134 LASSERT (!srpc_event_pending(rpc));
1135 swi_kill_workitem(wi);
1137 spin_unlock(&rpc->crpc_lock);
1139 (*rpc->crpc_done) (rpc);
1142 srpc_check_sends(peer, 1);
1146 /* sends an outgoing RPC */
1148 srpc_send_rpc (swi_workitem_t *wi)
1151 srpc_client_rpc_t *rpc = wi->wi_data;
1152 srpc_msg_t *reply = &rpc->crpc_replymsg;
1153 int do_bulk = rpc->crpc_bulk.bk_niov > 0;
1155 LASSERT (rpc != NULL);
1156 LASSERT (wi == &rpc->crpc_wi);
1158 spin_lock(&rpc->crpc_lock);
1160 if (rpc->crpc_aborted) {
1161 spin_unlock(&rpc->crpc_lock);
1165 spin_unlock(&rpc->crpc_lock);
1167 switch (wi->wi_state) {
1170 case SWI_STATE_NEWBORN:
1171 LASSERT (!srpc_event_pending(rpc));
1173 rc = srpc_prepare_reply(rpc);
1175 srpc_client_rpc_done(rpc, rc);
1179 rc = srpc_prepare_bulk(rpc);
1182 wi->wi_state = SWI_STATE_REQUEST_SUBMITTED;
1183 rc = srpc_send_request(rpc);
1186 case SWI_STATE_REQUEST_SUBMITTED:
1187 /* CAVEAT EMPTOR: rqtev, rpyev, and bulkev may come in any
1188 * order; however, they're processed in a strict order:
1189 * rqt, rpy, and bulk. */
1190 if (!rpc->crpc_reqstev.ev_fired) break;
1192 rc = rpc->crpc_reqstev.ev_status;
1195 wi->wi_state = SWI_STATE_REQUEST_SENT;
1196 /* perhaps more events, fall thru */
1197 case SWI_STATE_REQUEST_SENT: {
1198 srpc_msg_type_t type = srpc_service2reply(rpc->crpc_service);
1200 if (!rpc->crpc_replyev.ev_fired) break;
1202 rc = rpc->crpc_replyev.ev_status;
1205 if ((reply->msg_type != type &&
1206 reply->msg_type != __swab32(type)) ||
1207 (reply->msg_magic != SRPC_MSG_MAGIC &&
1208 reply->msg_magic != __swab32(SRPC_MSG_MAGIC))) {
1209 CWARN ("Bad message from %s: type %u (%d expected),"
1210 " magic %u (%d expected).\n",
1211 libcfs_id2str(rpc->crpc_dest),
1212 reply->msg_type, type,
1213 reply->msg_magic, SRPC_MSG_MAGIC);
1218 if (do_bulk && reply->msg_body.reply.status != 0) {
1219 CWARN ("Remote error %d at %s, unlink bulk buffer in "
1220 "case peer didn't initiate bulk transfer\n",
1221 reply->msg_body.reply.status,
1222 libcfs_id2str(rpc->crpc_dest));
1223 LNetMDUnlink(rpc->crpc_bulk.bk_mdh);
1226 wi->wi_state = SWI_STATE_REPLY_RECEIVED;
1228 case SWI_STATE_REPLY_RECEIVED:
1229 if (do_bulk && !rpc->crpc_bulkev.ev_fired) break;
1231 rc = do_bulk ? rpc->crpc_bulkev.ev_status : 0;
1233 /* Bulk buffer was unlinked due to remote error. Clear error
1234 * since reply buffer still contains valid data.
1235 * NB rpc->crpc_done shouldn't look into bulk data in case of
1237 if (do_bulk && rpc->crpc_bulkev.ev_lnet == LNET_EVENT_UNLINK &&
1238 rpc->crpc_status == 0 && reply->msg_body.reply.status != 0)
1241 wi->wi_state = SWI_STATE_DONE;
1242 srpc_client_rpc_done(rpc, rc);
1247 spin_lock(&rpc->crpc_lock);
1248 srpc_abort_rpc(rpc, rc);
1249 spin_unlock(&rpc->crpc_lock);
1253 if (rpc->crpc_aborted) {
1254 LNetMDUnlink(rpc->crpc_reqstmdh);
1255 LNetMDUnlink(rpc->crpc_replymdh);
1256 LNetMDUnlink(rpc->crpc_bulk.bk_mdh);
1258 if (!srpc_event_pending(rpc)) {
1259 srpc_client_rpc_done(rpc, -EINTR);
1267 srpc_create_client_rpc (lnet_process_id_t peer, int service,
1268 int nbulkiov, int bulklen,
1269 void (*rpc_done)(srpc_client_rpc_t *),
1270 void (*rpc_fini)(srpc_client_rpc_t *), void *priv)
1272 srpc_client_rpc_t *rpc;
1274 LIBCFS_ALLOC(rpc, offsetof(srpc_client_rpc_t,
1275 crpc_bulk.bk_iovs[nbulkiov]));
1279 srpc_init_client_rpc(rpc, peer, service, nbulkiov,
1280 bulklen, rpc_done, rpc_fini, priv);
1284 /* called with rpc->crpc_lock held */
1286 srpc_queue_rpc (srpc_peer_t *peer, srpc_client_rpc_t *rpc)
1288 int service = rpc->crpc_service;
1290 LASSERT (peer->stp_nid == rpc->crpc_dest.nid);
1291 LASSERT (srpc_data.rpc_state == SRPC_STATE_RUNNING);
1293 rpc->crpc_peer = peer;
1295 spin_lock(&peer->stp_lock);
1297 /* Framework RPCs that alter session state shall take precedence
1298 * over test RPCs and framework query RPCs */
1299 if (service <= SRPC_FRAMEWORK_SERVICE_MAX_ID &&
1300 service != SRPC_SERVICE_DEBUG &&
1301 service != SRPC_SERVICE_QUERY_STAT)
1302 list_add_tail(&rpc->crpc_privl, &peer->stp_ctl_rpcq);
1304 list_add_tail(&rpc->crpc_privl, &peer->stp_rpcq);
1306 srpc_client_rpc_addref(rpc); /* ++ref for peer->*rpcq */
1307 spin_unlock(&peer->stp_lock);
1311 /* called with rpc->crpc_lock held */
1313 srpc_abort_rpc (srpc_client_rpc_t *rpc, int why)
1315 srpc_peer_t *peer = rpc->crpc_peer;
1319 if (rpc->crpc_aborted || /* already aborted */
1320 rpc->crpc_closed) /* callback imminent */
1324 "Aborting RPC: service %d, peer %s, state %s, why %d\n",
1325 rpc->crpc_service, libcfs_id2str(rpc->crpc_dest),
1326 swi_state2str(rpc->crpc_wi.wi_state), why);
1328 rpc->crpc_aborted = 1;
1329 rpc->crpc_status = why;
1332 spin_lock(&peer->stp_lock);
1334 if (!list_empty(&rpc->crpc_privl)) { /* still queued */
1335 list_del_init(&rpc->crpc_privl);
1336 srpc_client_rpc_decref(rpc); /* --ref for peer->*rpcq */
1337 rpc->crpc_peer = NULL; /* no credit taken */
1340 spin_unlock(&peer->stp_lock);
1343 swi_schedule_workitem(&rpc->crpc_wi);
1347 /* called with rpc->crpc_lock held */
1349 srpc_post_rpc (srpc_client_rpc_t *rpc)
1353 LASSERT (!rpc->crpc_aborted);
1354 LASSERT (rpc->crpc_peer == NULL);
1355 LASSERT (srpc_data.rpc_state == SRPC_STATE_RUNNING);
1356 LASSERT ((rpc->crpc_bulk.bk_len & ~CFS_PAGE_MASK) == 0);
1358 CDEBUG (D_NET, "Posting RPC: peer %s, service %d, timeout %d\n",
1359 libcfs_id2str(rpc->crpc_dest), rpc->crpc_service,
1362 srpc_add_client_rpc_timer(rpc);
1364 peer = srpc_nid2peer(rpc->crpc_dest.nid);
1366 srpc_abort_rpc(rpc, -ENOMEM);
1370 srpc_queue_rpc(peer, rpc);
1372 spin_unlock(&rpc->crpc_lock);
1373 srpc_check_sends(peer, 0);
1374 spin_lock(&rpc->crpc_lock);
1380 srpc_send_reply (srpc_server_rpc_t *rpc)
1382 srpc_event_t *ev = &rpc->srpc_ev;
1383 srpc_msg_t *msg = &rpc->srpc_replymsg;
1384 srpc_buffer_t *buffer = rpc->srpc_reqstbuf;
1385 srpc_service_t *sv = rpc->srpc_service;
1389 LASSERT (buffer != NULL);
1390 rpyid = buffer->buf_msg.msg_body.reqst.rpyid;
1392 spin_lock(&sv->sv_lock);
1394 if (!sv->sv_shuttingdown &&
1395 sv->sv_id > SRPC_FRAMEWORK_SERVICE_MAX_ID) {
1396 /* Repost buffer before replying since test client
1397 * might send me another RPC once it gets the reply */
1398 if (srpc_service_post_buffer(sv, buffer) != 0)
1399 CWARN ("Failed to repost %s buffer\n", sv->sv_name);
1400 rpc->srpc_reqstbuf = NULL;
1403 spin_unlock(&sv->sv_lock);
1407 ev->ev_type = SRPC_REPLY_SENT;
1409 msg->msg_magic = SRPC_MSG_MAGIC;
1410 msg->msg_version = SRPC_MSG_VERSION;
1411 msg->msg_type = srpc_service2reply(sv->sv_id);
1413 rc = srpc_post_active_rdma(SRPC_RDMA_PORTAL, rpyid, msg,
1414 sizeof(*msg), LNET_MD_OP_PUT,
1415 rpc->srpc_peer, rpc->srpc_self,
1416 &rpc->srpc_replymdh, ev);
1418 ev->ev_fired = 1; /* no more event expected */
1422 /* when in kernel always called with LNET_LOCK() held, and in thread context */
1424 srpc_lnet_ev_handler (lnet_event_t *ev)
1426 srpc_event_t *rpcev = ev->md.user_ptr;
1427 srpc_client_rpc_t *crpc;
1428 srpc_server_rpc_t *srpc;
1429 srpc_buffer_t *buffer;
1432 srpc_msg_type_t type;
1435 LASSERT (!in_interrupt());
1437 if (ev->status != 0) {
1438 spin_lock(&srpc_data.rpc_glock);
1439 srpc_data.rpc_counters.errors++;
1440 spin_unlock(&srpc_data.rpc_glock);
1443 rpcev->ev_lnet = ev->type;
1445 switch (rpcev->ev_type) {
1448 case SRPC_REQUEST_SENT:
1449 if (ev->status == 0 && ev->type != LNET_EVENT_UNLINK) {
1450 spin_lock(&srpc_data.rpc_glock);
1451 srpc_data.rpc_counters.rpcs_sent++;
1452 spin_unlock(&srpc_data.rpc_glock);
1454 case SRPC_REPLY_RCVD:
1455 case SRPC_BULK_REQ_RCVD:
1456 crpc = rpcev->ev_data;
1458 LASSERT (rpcev == &crpc->crpc_reqstev ||
1459 rpcev == &crpc->crpc_replyev ||
1460 rpcev == &crpc->crpc_bulkev);
1462 spin_lock(&crpc->crpc_lock);
1464 LASSERT (rpcev->ev_fired == 0);
1465 rpcev->ev_fired = 1;
1466 rpcev->ev_status = (ev->type == LNET_EVENT_UNLINK) ?
1467 -EINTR : ev->status;
1468 swi_schedule_workitem(&crpc->crpc_wi);
1470 spin_unlock(&crpc->crpc_lock);
1473 case SRPC_REQUEST_RCVD:
1474 sv = rpcev->ev_data;
1476 LASSERT (rpcev == &sv->sv_ev);
1478 spin_lock(&sv->sv_lock);
1480 LASSERT (ev->unlinked);
1481 LASSERT (ev->type == LNET_EVENT_PUT ||
1482 ev->type == LNET_EVENT_UNLINK);
1483 LASSERT (ev->type != LNET_EVENT_UNLINK ||
1484 sv->sv_shuttingdown);
1486 buffer = container_of(ev->md.start, srpc_buffer_t, buf_msg);
1487 buffer->buf_peer = ev->initiator;
1488 buffer->buf_self = ev->target.nid;
1490 sv->sv_nposted_msg--;
1491 LASSERT (sv->sv_nposted_msg >= 0);
1493 if (sv->sv_shuttingdown) {
1494 /* Leave buffer on sv->sv_posted_msgq since
1495 * srpc_finish_service needs to traverse it. */
1496 spin_unlock(&sv->sv_lock);
1500 list_del(&buffer->buf_list); /* from sv->sv_posted_msgq */
1501 msg = &buffer->buf_msg;
1502 type = srpc_service2request(sv->sv_id);
1504 if (ev->status != 0 || ev->mlength != sizeof(*msg) ||
1505 (msg->msg_type != type &&
1506 msg->msg_type != __swab32(type)) ||
1507 (msg->msg_magic != SRPC_MSG_MAGIC &&
1508 msg->msg_magic != __swab32(SRPC_MSG_MAGIC))) {
1509 CERROR ("Dropping RPC (%s) from %s: "
1510 "status %d mlength %d type %u magic %u.\n",
1511 sv->sv_name, libcfs_id2str(ev->initiator),
1512 ev->status, ev->mlength,
1513 msg->msg_type, msg->msg_magic);
1515 /* NB might drop sv_lock in srpc_service_recycle_buffer,
1516 * sv_nposted_msg++ as an implicit reference to prevent
1517 * sv from disappearing under me */
1518 sv->sv_nposted_msg++;
1519 srpc_service_recycle_buffer(sv, buffer);
1520 sv->sv_nposted_msg--;
1521 spin_unlock(&sv->sv_lock);
1523 if (ev->status == 0) { /* status!=0 counted already */
1524 spin_lock(&srpc_data.rpc_glock);
1525 srpc_data.rpc_counters.errors++;
1526 spin_unlock(&srpc_data.rpc_glock);
1531 if (!list_empty(&sv->sv_free_rpcq)) {
1532 srpc = list_entry(sv->sv_free_rpcq.next,
1533 srpc_server_rpc_t, srpc_list);
1534 list_del(&srpc->srpc_list);
1536 srpc_init_server_rpc(srpc, sv, buffer);
1537 list_add_tail(&srpc->srpc_list, &sv->sv_active_rpcq);
1538 srpc_schedule_server_rpc(srpc);
1540 list_add_tail(&buffer->buf_list, &sv->sv_blocked_msgq);
1543 spin_unlock(&sv->sv_lock);
1545 spin_lock(&srpc_data.rpc_glock);
1546 srpc_data.rpc_counters.rpcs_rcvd++;
1547 spin_unlock(&srpc_data.rpc_glock);
1550 case SRPC_BULK_GET_RPLD:
1551 LASSERT (ev->type == LNET_EVENT_SEND ||
1552 ev->type == LNET_EVENT_REPLY ||
1553 ev->type == LNET_EVENT_UNLINK);
1555 if (ev->type == LNET_EVENT_SEND && !ev->unlinked) {
1556 if (ev->status == 0)
1557 break; /* wait for the final LNET_EVENT_REPLY */
1559 fired_flag = 0; /* LNET_EVENT_REPLY may arrive
1560 (optimized GET case) */
1562 case SRPC_BULK_PUT_SENT:
1563 if (ev->status == 0 && ev->type != LNET_EVENT_UNLINK) {
1564 spin_lock(&srpc_data.rpc_glock);
1566 if (rpcev->ev_type == SRPC_BULK_GET_RPLD)
1567 srpc_data.rpc_counters.bulk_get += ev->mlength;
1569 srpc_data.rpc_counters.bulk_put += ev->mlength;
1571 spin_unlock(&srpc_data.rpc_glock);
1573 case SRPC_REPLY_SENT:
1574 srpc = rpcev->ev_data;
1575 sv = srpc->srpc_service;
1577 LASSERT (rpcev == &srpc->srpc_ev);
1579 spin_lock(&sv->sv_lock);
1581 rpcev->ev_fired = 1;
1583 rpcev->ev_status = (ev->type == LNET_EVENT_UNLINK) ?
1584 -EINTR : ev->status;
1586 srpc_schedule_server_rpc(srpc);
1587 spin_unlock(&sv->sv_lock);
1597 srpc_check_event (int timeout)
1603 rc = LNetEQPoll(&srpc_data.rpc_lnet_eq, 1,
1604 timeout * 1000, &ev, &i);
1605 if (rc == 0) return 0;
1607 LASSERT (rc == -EOVERFLOW || rc == 1);
1609 /* We can't affort to miss any events... */
1610 if (rc == -EOVERFLOW) {
1611 CERROR ("Dropped an event!!!\n");
1615 srpc_lnet_ev_handler(&ev);
1630 s = getenv("SRPC_PEER_CREDITS");
1631 srpc_peer_credits = (s != NULL) ? atoi(s) : srpc_peer_credits;
1634 if (srpc_peer_credits <= 0) {
1635 CERROR("Peer credits must be positive: %d\n", srpc_peer_credits);
1639 memset(&srpc_data, 0, sizeof(struct smoketest_rpc));
1640 spin_lock_init(&srpc_data.rpc_glock);
1642 /* 1 second pause to avoid timestamp reuse */
1643 cfs_pause(cfs_time_seconds(1));
1644 srpc_data.rpc_matchbits = ((__u64) cfs_time_current_sec()) << 48;
1646 srpc_data.rpc_state = SRPC_STATE_NONE;
1648 LIBCFS_ALLOC(srpc_data.rpc_peers,
1649 sizeof(struct list_head) * SRPC_PEER_HASH_SIZE);
1650 if (srpc_data.rpc_peers == NULL) {
1651 CERROR ("Failed to alloc peer hash.\n");
1655 for (i = 0; i < SRPC_PEER_HASH_SIZE; i++)
1656 CFS_INIT_LIST_HEAD(&srpc_data.rpc_peers[i]);
1659 rc = LNetNIInit(LUSTRE_SRV_LNET_PID);
1661 if (the_lnet.ln_server_mode_flag)
1662 rc = LNetNIInit(LUSTRE_SRV_LNET_PID);
1664 rc = LNetNIInit(getpid() | LNET_PID_USERFLAG);
1667 CERROR ("LNetNIInit() has failed: %d\n", rc);
1668 LIBCFS_FREE(srpc_data.rpc_peers,
1669 sizeof(struct list_head) * SRPC_PEER_HASH_SIZE);
1673 srpc_data.rpc_state = SRPC_STATE_NI_INIT;
1675 srpc_data.rpc_lnet_eq = LNET_EQ_NONE;
1677 rc = LNetEQAlloc(16, srpc_lnet_ev_handler, &srpc_data.rpc_lnet_eq);
1679 rc = LNetEQAlloc(10240, LNET_EQ_HANDLER_NONE, &srpc_data.rpc_lnet_eq);
1682 CERROR("LNetEQAlloc() has failed: %d\n", rc);
1686 rc = LNetSetLazyPortal(SRPC_FRAMEWORK_REQUEST_PORTAL);
1689 srpc_data.rpc_state = SRPC_STATE_EQ_INIT;
1695 srpc_data.rpc_state = SRPC_STATE_WI_INIT;
1703 srpc_data.rpc_state = SRPC_STATE_RUNNING;
1709 srpc_shutdown (void)
1715 state = srpc_data.rpc_state;
1716 srpc_data.rpc_state = SRPC_STATE_STOPPING;
1721 case SRPC_STATE_RUNNING:
1722 spin_lock(&srpc_data.rpc_glock);
1724 for (i = 0; i <= SRPC_SERVICE_MAX_ID; i++) {
1725 srpc_service_t *sv = srpc_data.rpc_services[i];
1727 LASSERTF (sv == NULL,
1728 "service not empty: id %d, name %s\n",
1732 spin_unlock(&srpc_data.rpc_glock);
1736 case SRPC_STATE_WI_INIT:
1739 case SRPC_STATE_EQ_INIT:
1740 rc = LNetClearLazyPortal(SRPC_FRAMEWORK_REQUEST_PORTAL);
1742 rc = LNetEQFree(srpc_data.rpc_lnet_eq);
1743 LASSERT (rc == 0); /* the EQ should have no user by now */
1745 case SRPC_STATE_NI_INIT:
1750 /* srpc_peer_t's are kept in hash until shutdown */
1751 for (i = 0; i < SRPC_PEER_HASH_SIZE; i++) {
1754 while (!list_empty(&srpc_data.rpc_peers[i])) {
1755 peer = list_entry(srpc_data.rpc_peers[i].next,
1756 srpc_peer_t, stp_list);
1757 list_del(&peer->stp_list);
1759 LASSERT (list_empty(&peer->stp_rpcq));
1760 LASSERT (list_empty(&peer->stp_ctl_rpcq));
1761 LASSERT (peer->stp_credits == srpc_peer_credits);
1763 LIBCFS_FREE(peer, sizeof(srpc_peer_t));
1767 LIBCFS_FREE(srpc_data.rpc_peers,
1768 sizeof(struct list_head) * SRPC_PEER_HASH_SIZE);