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
20 * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
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 (c) 2007, 2010, Oracle and/or its affiliates. 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
54 struct smoketest_rpc {
55 cfs_spinlock_t rpc_glock; /* global lock */
56 srpc_service_t *rpc_services[SRPC_SERVICE_MAX_ID + 1];
57 lnet_handle_eq_t rpc_lnet_eq; /* _the_ LNet event queue */
58 srpc_state_t rpc_state;
59 srpc_counters_t rpc_counters;
60 __u64 rpc_matchbits; /* matchbits counter */
64 int srpc_handle_rpc (swi_workitem_t *wi);
66 void srpc_get_counters (srpc_counters_t *cnt)
68 cfs_spin_lock(&srpc_data.rpc_glock);
69 *cnt = srpc_data.rpc_counters;
70 cfs_spin_unlock(&srpc_data.rpc_glock);
73 void srpc_set_counters (const srpc_counters_t *cnt)
75 cfs_spin_lock(&srpc_data.rpc_glock);
76 srpc_data.rpc_counters = *cnt;
77 cfs_spin_unlock(&srpc_data.rpc_glock);
81 srpc_add_bulk_page (srpc_bulk_t *bk, cfs_page_t *pg, int i)
83 LASSERT (i >= 0 && i < bk->bk_niov);
86 bk->bk_iovs[i].kiov_offset = 0;
87 bk->bk_iovs[i].kiov_page = pg;
88 bk->bk_iovs[i].kiov_len = CFS_PAGE_SIZE;
90 LASSERT (bk->bk_pages != NULL);
93 bk->bk_iovs[i].iov_len = CFS_PAGE_SIZE;
94 bk->bk_iovs[i].iov_base = cfs_page_address(pg);
100 srpc_free_bulk (srpc_bulk_t *bk)
105 LASSERT (bk != NULL);
107 LASSERT (bk->bk_pages != NULL);
110 for (i = 0; i < bk->bk_niov; i++) {
112 pg = bk->bk_iovs[i].kiov_page;
114 pg = bk->bk_pages[i];
116 if (pg == NULL) break;
122 LIBCFS_FREE(bk->bk_pages, sizeof(cfs_page_t *) * bk->bk_niov);
124 LIBCFS_FREE(bk, offsetof(srpc_bulk_t, bk_iovs[bk->bk_niov]));
129 srpc_alloc_bulk (int npages, int sink)
135 LASSERT (npages > 0 && npages <= LNET_MAX_IOV);
137 LIBCFS_ALLOC(bk, offsetof(srpc_bulk_t, bk_iovs[npages]));
139 CERROR ("Can't allocate descriptor for %d pages\n", npages);
143 memset(bk, 0, offsetof(srpc_bulk_t, bk_iovs[npages]));
145 bk->bk_niov = npages;
146 bk->bk_len = npages * CFS_PAGE_SIZE;
148 LIBCFS_ALLOC(pages, sizeof(cfs_page_t *) * npages);
150 LIBCFS_FREE(bk, offsetof(srpc_bulk_t, bk_iovs[npages]));
151 CERROR ("Can't allocate page array for %d pages\n", npages);
155 memset(pages, 0, sizeof(cfs_page_t *) * npages);
156 bk->bk_pages = pages;
161 for (i = 0; i < npages; i++) {
162 cfs_page_t *pg = cfs_alloc_page(CFS_ALLOC_STD);
165 CERROR ("Can't allocate page %d of %d\n", i, npages);
170 srpc_add_bulk_page(bk, pg, i);
181 cfs_spin_lock(&srpc_data.rpc_glock);
182 id = srpc_data.rpc_matchbits++;
183 cfs_spin_unlock(&srpc_data.rpc_glock);
188 srpc_init_server_rpc (srpc_server_rpc_t *rpc,
189 srpc_service_t *sv, srpc_buffer_t *buffer)
191 memset(rpc, 0, sizeof(*rpc));
192 swi_init_workitem(&rpc->srpc_wi, rpc, srpc_handle_rpc,
193 sv->sv_id <= SRPC_FRAMEWORK_SERVICE_MAX_ID ?
194 CFS_WI_SCHED_SERIAL : CFS_WI_SCHED_ANY);
196 rpc->srpc_ev.ev_fired = 1; /* no event expected now */
198 rpc->srpc_service = sv;
199 rpc->srpc_reqstbuf = buffer;
200 rpc->srpc_peer = buffer->buf_peer;
201 rpc->srpc_self = buffer->buf_self;
202 LNetInvalidateHandle(&rpc->srpc_replymdh);
206 srpc_add_service (srpc_service_t *sv)
210 srpc_server_rpc_t *rpc;
212 LASSERT (sv->sv_concur > 0);
213 LASSERT (0 <= id && id <= SRPC_SERVICE_MAX_ID);
215 cfs_spin_lock(&srpc_data.rpc_glock);
217 LASSERT (srpc_data.rpc_state == SRPC_STATE_RUNNING);
219 if (srpc_data.rpc_services[id] != NULL) {
220 cfs_spin_unlock(&srpc_data.rpc_glock);
224 srpc_data.rpc_services[id] = sv;
225 cfs_spin_unlock(&srpc_data.rpc_glock);
228 sv->sv_nposted_msg = 0;
229 sv->sv_shuttingdown = 0;
230 cfs_spin_lock_init(&sv->sv_lock);
231 CFS_INIT_LIST_HEAD(&sv->sv_free_rpcq);
232 CFS_INIT_LIST_HEAD(&sv->sv_active_rpcq);
233 CFS_INIT_LIST_HEAD(&sv->sv_posted_msgq);
234 CFS_INIT_LIST_HEAD(&sv->sv_blocked_msgq);
236 sv->sv_ev.ev_data = sv;
237 sv->sv_ev.ev_type = SRPC_REQUEST_RCVD;
239 for (i = 0; i < sv->sv_concur; i++) {
240 LIBCFS_ALLOC(rpc, sizeof(*rpc));
241 if (rpc == NULL) goto enomem;
243 cfs_list_add(&rpc->srpc_list, &sv->sv_free_rpcq);
246 CDEBUG (D_NET, "Adding service: id %d, name %s, concurrency %d\n",
247 id, sv->sv_name, sv->sv_concur);
251 while (!cfs_list_empty(&sv->sv_free_rpcq)) {
252 rpc = cfs_list_entry(sv->sv_free_rpcq.next,
253 srpc_server_rpc_t, srpc_list);
254 cfs_list_del(&rpc->srpc_list);
255 LIBCFS_FREE(rpc, sizeof(*rpc));
258 cfs_spin_lock(&srpc_data.rpc_glock);
259 srpc_data.rpc_services[id] = NULL;
260 cfs_spin_unlock(&srpc_data.rpc_glock);
265 srpc_remove_service (srpc_service_t *sv)
269 cfs_spin_lock(&srpc_data.rpc_glock);
271 if (srpc_data.rpc_services[id] != sv) {
272 cfs_spin_unlock(&srpc_data.rpc_glock);
276 srpc_data.rpc_services[id] = NULL;
277 cfs_spin_unlock(&srpc_data.rpc_glock);
282 srpc_post_passive_rdma(int portal, __u64 matchbits, void *buf,
283 int len, int options, lnet_process_id_t peer,
284 lnet_handle_md_t *mdh, srpc_event_t *ev)
288 lnet_handle_me_t meh;
290 rc = LNetMEAttach(portal, peer, matchbits, 0,
291 LNET_UNLINK, LNET_INS_AFTER, &meh);
293 CERROR ("LNetMEAttach failed: %d\n", rc);
294 LASSERT (rc == -ENOMEM);
302 md.options = options;
303 md.eq_handle = srpc_data.rpc_lnet_eq;
305 rc = LNetMDAttach(meh, md, LNET_UNLINK, mdh);
307 CERROR ("LNetMDAttach failed: %d\n", rc);
308 LASSERT (rc == -ENOMEM);
310 rc = LNetMEUnlink(meh);
316 "Posted passive RDMA: peer %s, portal %d, matchbits "LPX64"\n",
317 libcfs_id2str(peer), portal, matchbits);
322 srpc_post_active_rdma(int portal, __u64 matchbits, void *buf, int len,
323 int options, lnet_process_id_t peer, lnet_nid_t self,
324 lnet_handle_md_t *mdh, srpc_event_t *ev)
332 md.eq_handle = srpc_data.rpc_lnet_eq;
333 md.threshold = ((options & LNET_MD_OP_GET) != 0) ? 2 : 1;
334 md.options = options & ~(LNET_MD_OP_PUT | LNET_MD_OP_GET);
336 rc = LNetMDBind(md, LNET_UNLINK, mdh);
338 CERROR ("LNetMDBind failed: %d\n", rc);
339 LASSERT (rc == -ENOMEM);
343 /* this is kind of an abuse of the LNET_MD_OP_{PUT,GET} options.
344 * they're only meaningful for MDs attached to an ME (i.e. passive
346 if ((options & LNET_MD_OP_PUT) != 0) {
347 rc = LNetPut(self, *mdh, LNET_NOACK_REQ, peer,
348 portal, matchbits, 0, 0);
350 LASSERT ((options & LNET_MD_OP_GET) != 0);
352 rc = LNetGet(self, *mdh, peer, portal, matchbits, 0);
356 CERROR ("LNet%s(%s, %d, "LPD64") failed: %d\n",
357 ((options & LNET_MD_OP_PUT) != 0) ? "Put" : "Get",
358 libcfs_id2str(peer), portal, matchbits, rc);
360 /* The forthcoming unlink event will complete this operation
361 * with failure, so fall through and return success here.
363 rc = LNetMDUnlink(*mdh);
367 "Posted active RDMA: peer %s, portal %u, matchbits "LPX64"\n",
368 libcfs_id2str(peer), portal, matchbits);
374 srpc_post_active_rqtbuf(lnet_process_id_t peer, int service, void *buf,
375 int len, lnet_handle_md_t *mdh, srpc_event_t *ev)
380 if (service > SRPC_FRAMEWORK_SERVICE_MAX_ID)
381 portal = SRPC_REQUEST_PORTAL;
383 portal = SRPC_FRAMEWORK_REQUEST_PORTAL;
385 rc = srpc_post_active_rdma(portal, service, buf, len,
386 LNET_MD_OP_PUT, peer,
387 LNET_NID_ANY, mdh, ev);
392 srpc_post_passive_rqtbuf(int service, void *buf, int len,
393 lnet_handle_md_t *mdh, srpc_event_t *ev)
397 lnet_process_id_t any = {0};
399 any.nid = LNET_NID_ANY;
400 any.pid = LNET_PID_ANY;
402 if (service > SRPC_FRAMEWORK_SERVICE_MAX_ID)
403 portal = SRPC_REQUEST_PORTAL;
405 portal = SRPC_FRAMEWORK_REQUEST_PORTAL;
407 rc = srpc_post_passive_rdma(portal, service, buf, len,
408 LNET_MD_OP_PUT, any, mdh, ev);
413 srpc_service_post_buffer (srpc_service_t *sv, srpc_buffer_t *buf)
415 srpc_msg_t *msg = &buf->buf_msg;
418 LASSERT (!sv->sv_shuttingdown);
420 LNetInvalidateHandle(&buf->buf_mdh);
421 cfs_list_add(&buf->buf_list, &sv->sv_posted_msgq);
422 sv->sv_nposted_msg++;
423 cfs_spin_unlock(&sv->sv_lock);
425 rc = srpc_post_passive_rqtbuf(sv->sv_id, msg, sizeof(*msg),
426 &buf->buf_mdh, &sv->sv_ev);
428 /* At this point, a RPC (new or delayed) may have arrived in
429 * msg and its event handler has been called. So we must add
430 * buf to sv_posted_msgq _before_ dropping sv_lock */
432 cfs_spin_lock(&sv->sv_lock);
435 if (sv->sv_shuttingdown) {
436 cfs_spin_unlock(&sv->sv_lock);
438 /* srpc_shutdown_service might have tried to unlink me
439 * when my buf_mdh was still invalid */
440 LNetMDUnlink(buf->buf_mdh);
442 cfs_spin_lock(&sv->sv_lock);
447 sv->sv_nposted_msg--;
448 if (sv->sv_shuttingdown) return rc;
450 cfs_list_del(&buf->buf_list);
452 cfs_spin_unlock(&sv->sv_lock);
453 LIBCFS_FREE(buf, sizeof(*buf));
454 cfs_spin_lock(&sv->sv_lock);
459 srpc_service_add_buffers (srpc_service_t *sv, int nbuffer)
465 LASSERTF (nbuffer > 0,
466 "nbuffer must be positive: %d\n", nbuffer);
468 for (posted = 0; posted < nbuffer; posted++) {
469 LIBCFS_ALLOC(buf, sizeof(*buf));
470 if (buf == NULL) break;
472 cfs_spin_lock(&sv->sv_lock);
473 rc = srpc_service_post_buffer(sv, buf);
474 cfs_spin_unlock(&sv->sv_lock);
483 srpc_service_remove_buffers (srpc_service_t *sv, int nbuffer)
485 LASSERTF (nbuffer > 0,
486 "nbuffer must be positive: %d\n", nbuffer);
488 cfs_spin_lock(&sv->sv_lock);
490 LASSERT (sv->sv_nprune >= 0);
491 LASSERT (!sv->sv_shuttingdown);
493 sv->sv_nprune += nbuffer;
495 cfs_spin_unlock(&sv->sv_lock);
499 /* returns 1 if sv has finished, otherwise 0 */
501 srpc_finish_service (srpc_service_t *sv)
503 srpc_server_rpc_t *rpc;
506 cfs_spin_lock(&sv->sv_lock);
508 LASSERT (sv->sv_shuttingdown); /* srpc_shutdown_service called */
510 if (sv->sv_nposted_msg != 0 || !cfs_list_empty(&sv->sv_active_rpcq)) {
512 "waiting for %d posted buffers to unlink and "
513 "in-flight RPCs to die.\n",
516 if (!cfs_list_empty(&sv->sv_active_rpcq)) {
517 rpc = cfs_list_entry(sv->sv_active_rpcq.next,
518 srpc_server_rpc_t, srpc_list);
520 "Active RPC %p on shutdown: sv %s, peer %s, "
521 "wi %s scheduled %d running %d, "
522 "ev fired %d type %d status %d lnet %d\n",
523 rpc, sv->sv_name, libcfs_id2str(rpc->srpc_peer),
524 swi_state2str(rpc->srpc_wi.swi_state),
525 rpc->srpc_wi.swi_workitem.wi_scheduled,
526 rpc->srpc_wi.swi_workitem.wi_running,
527 rpc->srpc_ev.ev_fired,
528 rpc->srpc_ev.ev_type,
529 rpc->srpc_ev.ev_status,
530 rpc->srpc_ev.ev_lnet);
533 cfs_spin_unlock(&sv->sv_lock);
537 cfs_spin_unlock(&sv->sv_lock); /* no lock needed from now on */
542 if (!cfs_list_empty(&sv->sv_posted_msgq))
543 q = &sv->sv_posted_msgq;
544 else if (!cfs_list_empty(&sv->sv_blocked_msgq))
545 q = &sv->sv_blocked_msgq;
549 buf = cfs_list_entry(q->next, srpc_buffer_t, buf_list);
550 cfs_list_del(&buf->buf_list);
552 LIBCFS_FREE(buf, sizeof(*buf));
555 while (!cfs_list_empty(&sv->sv_free_rpcq)) {
556 rpc = cfs_list_entry(sv->sv_free_rpcq.next,
557 srpc_server_rpc_t, srpc_list);
558 cfs_list_del(&rpc->srpc_list);
559 LIBCFS_FREE(rpc, sizeof(*rpc));
565 /* called with sv->sv_lock held */
567 srpc_service_recycle_buffer (srpc_service_t *sv, srpc_buffer_t *buf)
569 if (sv->sv_shuttingdown) goto free;
571 if (sv->sv_nprune == 0) {
572 if (srpc_service_post_buffer(sv, buf) != 0)
573 CWARN ("Failed to post %s buffer\n", sv->sv_name);
579 cfs_spin_unlock(&sv->sv_lock);
580 LIBCFS_FREE(buf, sizeof(*buf));
581 cfs_spin_lock(&sv->sv_lock);
584 /* called with srpc_service_t::sv_lock held */
586 srpc_schedule_server_rpc (srpc_server_rpc_t *rpc)
588 swi_schedule_workitem(&rpc->srpc_wi);
592 srpc_abort_service (srpc_service_t *sv)
594 srpc_server_rpc_t *rpc;
596 cfs_spin_lock(&sv->sv_lock);
598 CDEBUG(D_NET, "Aborting service: id %d, name %s\n",
599 sv->sv_id, sv->sv_name);
601 /* schedule in-flight RPCs to notice the abort, NB:
602 * racing with incoming RPCs; complete fix should make test
603 * RPCs carry session ID in its headers */
604 cfs_list_for_each_entry (rpc, &sv->sv_active_rpcq, srpc_list) {
605 rpc->srpc_aborted = 1;
606 srpc_schedule_server_rpc(rpc);
609 cfs_spin_unlock(&sv->sv_lock);
614 srpc_shutdown_service (srpc_service_t *sv)
616 srpc_server_rpc_t *rpc;
619 cfs_spin_lock(&sv->sv_lock);
621 CDEBUG(D_NET, "Shutting down service: id %d, name %s\n",
622 sv->sv_id, sv->sv_name);
624 sv->sv_shuttingdown = 1; /* i.e. no new active RPC */
626 /* schedule in-flight RPCs to notice the shutdown */
627 cfs_list_for_each_entry_typed (rpc, &sv->sv_active_rpcq,
628 srpc_server_rpc_t, srpc_list) {
629 srpc_schedule_server_rpc(rpc);
632 cfs_spin_unlock(&sv->sv_lock);
634 /* OK to traverse sv_posted_msgq without lock, since no one
635 * touches sv_posted_msgq now */
636 cfs_list_for_each_entry_typed (buf, &sv->sv_posted_msgq,
637 srpc_buffer_t, buf_list)
638 LNetMDUnlink(buf->buf_mdh);
644 srpc_send_request (srpc_client_rpc_t *rpc)
646 srpc_event_t *ev = &rpc->crpc_reqstev;
651 ev->ev_type = SRPC_REQUEST_SENT;
653 rc = srpc_post_active_rqtbuf(rpc->crpc_dest, rpc->crpc_service,
654 &rpc->crpc_reqstmsg, sizeof(srpc_msg_t),
655 &rpc->crpc_reqstmdh, ev);
657 LASSERT (rc == -ENOMEM);
658 ev->ev_fired = 1; /* no more event expected */
664 srpc_prepare_reply (srpc_client_rpc_t *rpc)
666 srpc_event_t *ev = &rpc->crpc_replyev;
667 __u64 *id = &rpc->crpc_reqstmsg.msg_body.reqst.rpyid;
672 ev->ev_type = SRPC_REPLY_RCVD;
674 *id = srpc_next_id();
676 rc = srpc_post_passive_rdma(SRPC_RDMA_PORTAL, *id,
677 &rpc->crpc_replymsg, sizeof(srpc_msg_t),
678 LNET_MD_OP_PUT, rpc->crpc_dest,
679 &rpc->crpc_replymdh, ev);
681 LASSERT (rc == -ENOMEM);
682 ev->ev_fired = 1; /* no more event expected */
688 srpc_prepare_bulk (srpc_client_rpc_t *rpc)
690 srpc_bulk_t *bk = &rpc->crpc_bulk;
691 srpc_event_t *ev = &rpc->crpc_bulkev;
692 __u64 *id = &rpc->crpc_reqstmsg.msg_body.reqst.bulkid;
696 LASSERT (bk->bk_niov <= LNET_MAX_IOV);
698 if (bk->bk_niov == 0) return 0; /* nothing to do */
700 opt = bk->bk_sink ? LNET_MD_OP_PUT : LNET_MD_OP_GET;
704 opt |= LNET_MD_IOVEC;
709 ev->ev_type = SRPC_BULK_REQ_RCVD;
711 *id = srpc_next_id();
713 rc = srpc_post_passive_rdma(SRPC_RDMA_PORTAL, *id,
714 &bk->bk_iovs[0], bk->bk_niov, opt,
715 rpc->crpc_dest, &bk->bk_mdh, ev);
717 LASSERT (rc == -ENOMEM);
718 ev->ev_fired = 1; /* no more event expected */
724 srpc_do_bulk (srpc_server_rpc_t *rpc)
726 srpc_event_t *ev = &rpc->srpc_ev;
727 srpc_bulk_t *bk = rpc->srpc_bulk;
728 __u64 id = rpc->srpc_reqstbuf->buf_msg.msg_body.reqst.bulkid;
732 LASSERT (bk != NULL);
734 opt = bk->bk_sink ? LNET_MD_OP_GET : LNET_MD_OP_PUT;
738 opt |= LNET_MD_IOVEC;
743 ev->ev_type = bk->bk_sink ? SRPC_BULK_GET_RPLD : SRPC_BULK_PUT_SENT;
745 rc = srpc_post_active_rdma(SRPC_RDMA_PORTAL, id,
746 &bk->bk_iovs[0], bk->bk_niov, opt,
747 rpc->srpc_peer, rpc->srpc_self,
750 ev->ev_fired = 1; /* no more event expected */
754 /* only called from srpc_handle_rpc */
756 srpc_server_rpc_done (srpc_server_rpc_t *rpc, int status)
758 srpc_service_t *sv = rpc->srpc_service;
759 srpc_buffer_t *buffer;
761 LASSERT (status != 0 || rpc->srpc_wi.swi_state == SWI_STATE_DONE);
763 rpc->srpc_status = status;
765 CDEBUG (status == 0 ? D_NET : D_NETERROR,
766 "Server RPC %p done: service %s, peer %s, status %s:%d\n",
767 rpc, sv->sv_name, libcfs_id2str(rpc->srpc_peer),
768 swi_state2str(rpc->srpc_wi.swi_state), status);
771 cfs_spin_lock(&srpc_data.rpc_glock);
772 srpc_data.rpc_counters.rpcs_dropped++;
773 cfs_spin_unlock(&srpc_data.rpc_glock);
776 if (rpc->srpc_done != NULL)
777 (*rpc->srpc_done) (rpc);
778 LASSERT (rpc->srpc_bulk == NULL);
780 cfs_spin_lock(&sv->sv_lock);
782 if (rpc->srpc_reqstbuf != NULL) {
783 /* NB might drop sv_lock in srpc_service_recycle_buffer, but
784 * sv won't go away for sv_active_rpcq must not be empty */
785 srpc_service_recycle_buffer(sv, rpc->srpc_reqstbuf);
786 rpc->srpc_reqstbuf = NULL;
789 cfs_list_del(&rpc->srpc_list); /* from sv->sv_active_rpcq */
792 * No one can schedule me now since:
793 * - I'm not on sv_active_rpcq.
794 * - all LNet events have been fired.
795 * Cancel pending schedules and prevent future schedule attempts:
797 LASSERT (rpc->srpc_ev.ev_fired);
798 swi_kill_workitem(&rpc->srpc_wi);
800 if (!sv->sv_shuttingdown && !cfs_list_empty(&sv->sv_blocked_msgq)) {
801 buffer = cfs_list_entry(sv->sv_blocked_msgq.next,
802 srpc_buffer_t, buf_list);
803 cfs_list_del(&buffer->buf_list);
805 srpc_init_server_rpc(rpc, sv, buffer);
806 cfs_list_add_tail(&rpc->srpc_list, &sv->sv_active_rpcq);
807 srpc_schedule_server_rpc(rpc);
809 cfs_list_add(&rpc->srpc_list, &sv->sv_free_rpcq);
812 cfs_spin_unlock(&sv->sv_lock);
816 /* handles an incoming RPC */
818 srpc_handle_rpc (swi_workitem_t *wi)
820 srpc_server_rpc_t *rpc = wi->swi_workitem.wi_data;
821 srpc_service_t *sv = rpc->srpc_service;
822 srpc_event_t *ev = &rpc->srpc_ev;
825 LASSERT (wi == &rpc->srpc_wi);
827 cfs_spin_lock(&sv->sv_lock);
829 if (sv->sv_shuttingdown || rpc->srpc_aborted) {
830 cfs_spin_unlock(&sv->sv_lock);
832 if (rpc->srpc_bulk != NULL)
833 LNetMDUnlink(rpc->srpc_bulk->bk_mdh);
834 LNetMDUnlink(rpc->srpc_replymdh);
836 if (ev->ev_fired) { /* no more event, OK to finish */
837 srpc_server_rpc_done(rpc, -ESHUTDOWN);
843 cfs_spin_unlock(&sv->sv_lock);
845 switch (wi->swi_state) {
848 case SWI_STATE_NEWBORN: {
850 srpc_generic_reply_t *reply;
852 msg = &rpc->srpc_reqstbuf->buf_msg;
853 reply = &rpc->srpc_replymsg.msg_body.reply;
855 if (msg->msg_magic == 0) {
856 /* moaned already in srpc_lnet_ev_handler */
858 } else if (msg->msg_version != SRPC_MSG_VERSION &&
859 msg->msg_version != __swab32(SRPC_MSG_VERSION)) {
860 CWARN ("Version mismatch: %u, %u expected, from %s\n",
861 msg->msg_version, SRPC_MSG_VERSION,
862 libcfs_id2str(rpc->srpc_peer));
863 reply->status = EPROTO;
866 rc = (*sv->sv_handler) (rpc);
867 LASSERT (reply->status == 0 || !rpc->srpc_bulk);
871 srpc_server_rpc_done(rpc, rc);
875 wi->swi_state = SWI_STATE_BULK_STARTED;
877 if (rpc->srpc_bulk != NULL) {
878 rc = srpc_do_bulk(rpc);
880 return 0; /* wait for bulk */
882 LASSERT (ev->ev_fired);
886 case SWI_STATE_BULK_STARTED:
887 LASSERT (rpc->srpc_bulk == NULL || ev->ev_fired);
889 if (rpc->srpc_bulk != NULL) {
892 if (sv->sv_bulk_ready != NULL)
893 rc = (*sv->sv_bulk_ready) (rpc, rc);
896 srpc_server_rpc_done(rpc, rc);
901 wi->swi_state = SWI_STATE_REPLY_SUBMITTED;
902 rc = srpc_send_reply(rpc);
904 return 0; /* wait for reply */
905 srpc_server_rpc_done(rpc, rc);
908 case SWI_STATE_REPLY_SUBMITTED:
910 CERROR("RPC %p: bulk %p, service %d\n",
911 rpc, rpc->srpc_bulk, rpc->srpc_service->sv_id);
912 CERROR("Event: status %d, type %d, lnet %d\n",
913 ev->ev_status, ev->ev_type, ev->ev_lnet);
914 LASSERT (ev->ev_fired);
917 wi->swi_state = SWI_STATE_DONE;
918 srpc_server_rpc_done(rpc, ev->ev_status);
926 srpc_client_rpc_expired (void *data)
928 srpc_client_rpc_t *rpc = data;
930 CWARN ("Client RPC expired: service %d, peer %s, timeout %d.\n",
931 rpc->crpc_service, libcfs_id2str(rpc->crpc_dest),
934 cfs_spin_lock(&rpc->crpc_lock);
936 rpc->crpc_timeout = 0;
937 srpc_abort_rpc(rpc, -ETIMEDOUT);
939 cfs_spin_unlock(&rpc->crpc_lock);
941 cfs_spin_lock(&srpc_data.rpc_glock);
942 srpc_data.rpc_counters.rpcs_expired++;
943 cfs_spin_unlock(&srpc_data.rpc_glock);
948 srpc_add_client_rpc_timer (srpc_client_rpc_t *rpc)
950 stt_timer_t *timer = &rpc->crpc_timer;
952 if (rpc->crpc_timeout == 0) return;
954 CFS_INIT_LIST_HEAD(&timer->stt_list);
955 timer->stt_data = rpc;
956 timer->stt_func = srpc_client_rpc_expired;
957 timer->stt_expires = cfs_time_add(rpc->crpc_timeout,
958 cfs_time_current_sec());
959 stt_add_timer(timer);
964 * Called with rpc->crpc_lock held.
966 * Upon exit the RPC expiry timer is not queued and the handler is not
967 * running on any CPU. */
969 srpc_del_client_rpc_timer (srpc_client_rpc_t *rpc)
971 /* timer not planted or already exploded */
972 if (rpc->crpc_timeout == 0) return;
974 /* timer sucessfully defused */
975 if (stt_del_timer(&rpc->crpc_timer)) return;
978 /* timer detonated, wait for it to explode */
979 while (rpc->crpc_timeout != 0) {
980 cfs_spin_unlock(&rpc->crpc_lock);
984 cfs_spin_lock(&rpc->crpc_lock);
987 LBUG(); /* impossible in single-threaded runtime */
993 srpc_client_rpc_done (srpc_client_rpc_t *rpc, int status)
995 swi_workitem_t *wi = &rpc->crpc_wi;
997 LASSERT (status != 0 || wi->swi_state == SWI_STATE_DONE);
999 cfs_spin_lock(&rpc->crpc_lock);
1001 rpc->crpc_closed = 1;
1002 if (rpc->crpc_status == 0)
1003 rpc->crpc_status = status;
1005 srpc_del_client_rpc_timer(rpc);
1007 CDEBUG ((status == 0) ? D_NET : D_NETERROR,
1008 "Client RPC done: service %d, peer %s, status %s:%d:%d\n",
1009 rpc->crpc_service, libcfs_id2str(rpc->crpc_dest),
1010 swi_state2str(wi->swi_state), rpc->crpc_aborted, status);
1013 * No one can schedule me now since:
1014 * - RPC timer has been defused.
1015 * - all LNet events have been fired.
1016 * - crpc_closed has been set, preventing srpc_abort_rpc from
1018 * Cancel pending schedules and prevent future schedule attempts:
1020 LASSERT (!srpc_event_pending(rpc));
1021 swi_kill_workitem(wi);
1023 cfs_spin_unlock(&rpc->crpc_lock);
1025 (*rpc->crpc_done) (rpc);
1029 /* sends an outgoing RPC */
1031 srpc_send_rpc (swi_workitem_t *wi)
1034 srpc_client_rpc_t *rpc = wi->swi_workitem.wi_data;
1035 srpc_msg_t *reply = &rpc->crpc_replymsg;
1036 int do_bulk = rpc->crpc_bulk.bk_niov > 0;
1038 LASSERT (rpc != NULL);
1039 LASSERT (wi == &rpc->crpc_wi);
1041 cfs_spin_lock(&rpc->crpc_lock);
1043 if (rpc->crpc_aborted) {
1044 cfs_spin_unlock(&rpc->crpc_lock);
1048 cfs_spin_unlock(&rpc->crpc_lock);
1050 switch (wi->swi_state) {
1053 case SWI_STATE_NEWBORN:
1054 LASSERT (!srpc_event_pending(rpc));
1056 rc = srpc_prepare_reply(rpc);
1058 srpc_client_rpc_done(rpc, rc);
1062 rc = srpc_prepare_bulk(rpc);
1065 wi->swi_state = SWI_STATE_REQUEST_SUBMITTED;
1066 rc = srpc_send_request(rpc);
1069 case SWI_STATE_REQUEST_SUBMITTED:
1070 /* CAVEAT EMPTOR: rqtev, rpyev, and bulkev may come in any
1071 * order; however, they're processed in a strict order:
1072 * rqt, rpy, and bulk. */
1073 if (!rpc->crpc_reqstev.ev_fired) break;
1075 rc = rpc->crpc_reqstev.ev_status;
1078 wi->swi_state = SWI_STATE_REQUEST_SENT;
1079 /* perhaps more events, fall thru */
1080 case SWI_STATE_REQUEST_SENT: {
1081 srpc_msg_type_t type = srpc_service2reply(rpc->crpc_service);
1083 if (!rpc->crpc_replyev.ev_fired) break;
1085 rc = rpc->crpc_replyev.ev_status;
1088 if ((reply->msg_type != type &&
1089 reply->msg_type != __swab32(type)) ||
1090 (reply->msg_magic != SRPC_MSG_MAGIC &&
1091 reply->msg_magic != __swab32(SRPC_MSG_MAGIC))) {
1092 CWARN ("Bad message from %s: type %u (%d expected),"
1093 " magic %u (%d expected).\n",
1094 libcfs_id2str(rpc->crpc_dest),
1095 reply->msg_type, type,
1096 reply->msg_magic, SRPC_MSG_MAGIC);
1101 if (do_bulk && reply->msg_body.reply.status != 0) {
1102 CWARN ("Remote error %d at %s, unlink bulk buffer in "
1103 "case peer didn't initiate bulk transfer\n",
1104 reply->msg_body.reply.status,
1105 libcfs_id2str(rpc->crpc_dest));
1106 LNetMDUnlink(rpc->crpc_bulk.bk_mdh);
1109 wi->swi_state = SWI_STATE_REPLY_RECEIVED;
1111 case SWI_STATE_REPLY_RECEIVED:
1112 if (do_bulk && !rpc->crpc_bulkev.ev_fired) break;
1114 rc = do_bulk ? rpc->crpc_bulkev.ev_status : 0;
1116 /* Bulk buffer was unlinked due to remote error. Clear error
1117 * since reply buffer still contains valid data.
1118 * NB rpc->crpc_done shouldn't look into bulk data in case of
1120 if (do_bulk && rpc->crpc_bulkev.ev_lnet == LNET_EVENT_UNLINK &&
1121 rpc->crpc_status == 0 && reply->msg_body.reply.status != 0)
1124 wi->swi_state = SWI_STATE_DONE;
1125 srpc_client_rpc_done(rpc, rc);
1130 cfs_spin_lock(&rpc->crpc_lock);
1131 srpc_abort_rpc(rpc, rc);
1132 cfs_spin_unlock(&rpc->crpc_lock);
1136 if (rpc->crpc_aborted) {
1137 LNetMDUnlink(rpc->crpc_reqstmdh);
1138 LNetMDUnlink(rpc->crpc_replymdh);
1139 LNetMDUnlink(rpc->crpc_bulk.bk_mdh);
1141 if (!srpc_event_pending(rpc)) {
1142 srpc_client_rpc_done(rpc, -EINTR);
1150 srpc_create_client_rpc (lnet_process_id_t peer, int service,
1151 int nbulkiov, int bulklen,
1152 void (*rpc_done)(srpc_client_rpc_t *),
1153 void (*rpc_fini)(srpc_client_rpc_t *), void *priv)
1155 srpc_client_rpc_t *rpc;
1157 LIBCFS_ALLOC(rpc, offsetof(srpc_client_rpc_t,
1158 crpc_bulk.bk_iovs[nbulkiov]));
1162 srpc_init_client_rpc(rpc, peer, service, nbulkiov,
1163 bulklen, rpc_done, rpc_fini, priv);
1167 /* called with rpc->crpc_lock held */
1169 srpc_abort_rpc (srpc_client_rpc_t *rpc, int why)
1173 if (rpc->crpc_aborted || /* already aborted */
1174 rpc->crpc_closed) /* callback imminent */
1178 "Aborting RPC: service %d, peer %s, state %s, why %d\n",
1179 rpc->crpc_service, libcfs_id2str(rpc->crpc_dest),
1180 swi_state2str(rpc->crpc_wi.swi_state), why);
1182 rpc->crpc_aborted = 1;
1183 rpc->crpc_status = why;
1184 swi_schedule_workitem(&rpc->crpc_wi);
1188 /* called with rpc->crpc_lock held */
1190 srpc_post_rpc (srpc_client_rpc_t *rpc)
1192 LASSERT (!rpc->crpc_aborted);
1193 LASSERT (srpc_data.rpc_state == SRPC_STATE_RUNNING);
1194 LASSERT ((rpc->crpc_bulk.bk_len & ~CFS_PAGE_MASK) == 0);
1196 CDEBUG (D_NET, "Posting RPC: peer %s, service %d, timeout %d\n",
1197 libcfs_id2str(rpc->crpc_dest), rpc->crpc_service,
1200 srpc_add_client_rpc_timer(rpc);
1201 swi_schedule_workitem(&rpc->crpc_wi);
1207 srpc_send_reply (srpc_server_rpc_t *rpc)
1209 srpc_event_t *ev = &rpc->srpc_ev;
1210 srpc_msg_t *msg = &rpc->srpc_replymsg;
1211 srpc_buffer_t *buffer = rpc->srpc_reqstbuf;
1212 srpc_service_t *sv = rpc->srpc_service;
1216 LASSERT (buffer != NULL);
1217 rpyid = buffer->buf_msg.msg_body.reqst.rpyid;
1219 cfs_spin_lock(&sv->sv_lock);
1221 if (!sv->sv_shuttingdown &&
1222 sv->sv_id > SRPC_FRAMEWORK_SERVICE_MAX_ID) {
1223 /* Repost buffer before replying since test client
1224 * might send me another RPC once it gets the reply */
1225 if (srpc_service_post_buffer(sv, buffer) != 0)
1226 CWARN ("Failed to repost %s buffer\n", sv->sv_name);
1227 rpc->srpc_reqstbuf = NULL;
1230 cfs_spin_unlock(&sv->sv_lock);
1234 ev->ev_type = SRPC_REPLY_SENT;
1236 msg->msg_magic = SRPC_MSG_MAGIC;
1237 msg->msg_version = SRPC_MSG_VERSION;
1238 msg->msg_type = srpc_service2reply(sv->sv_id);
1240 rc = srpc_post_active_rdma(SRPC_RDMA_PORTAL, rpyid, msg,
1241 sizeof(*msg), LNET_MD_OP_PUT,
1242 rpc->srpc_peer, rpc->srpc_self,
1243 &rpc->srpc_replymdh, ev);
1245 ev->ev_fired = 1; /* no more event expected */
1249 /* when in kernel always called with LNET_LOCK() held, and in thread context */
1251 srpc_lnet_ev_handler (lnet_event_t *ev)
1253 srpc_event_t *rpcev = ev->md.user_ptr;
1254 srpc_client_rpc_t *crpc;
1255 srpc_server_rpc_t *srpc;
1256 srpc_buffer_t *buffer;
1259 srpc_msg_type_t type;
1261 LASSERT (!cfs_in_interrupt());
1263 if (ev->status != 0) {
1264 cfs_spin_lock(&srpc_data.rpc_glock);
1265 srpc_data.rpc_counters.errors++;
1266 cfs_spin_unlock(&srpc_data.rpc_glock);
1269 rpcev->ev_lnet = ev->type;
1271 switch (rpcev->ev_type) {
1273 CERROR("Unknown event: status %d, type %d, lnet %d\n",
1274 rpcev->ev_status, rpcev->ev_type, rpcev->ev_lnet);
1276 case SRPC_REQUEST_SENT:
1277 if (ev->status == 0 && ev->type != LNET_EVENT_UNLINK) {
1278 cfs_spin_lock(&srpc_data.rpc_glock);
1279 srpc_data.rpc_counters.rpcs_sent++;
1280 cfs_spin_unlock(&srpc_data.rpc_glock);
1282 case SRPC_REPLY_RCVD:
1283 case SRPC_BULK_REQ_RCVD:
1284 crpc = rpcev->ev_data;
1286 if (rpcev != &crpc->crpc_reqstev &&
1287 rpcev != &crpc->crpc_replyev &&
1288 rpcev != &crpc->crpc_bulkev) {
1289 CERROR("rpcev %p, crpc %p, reqstev %p, replyev %p, bulkev %p\n",
1290 rpcev, crpc, &crpc->crpc_reqstev,
1291 &crpc->crpc_replyev, &crpc->crpc_bulkev);
1292 CERROR("Bad event: status %d, type %d, lnet %d\n",
1293 rpcev->ev_status, rpcev->ev_type, rpcev->ev_lnet);
1297 cfs_spin_lock(&crpc->crpc_lock);
1299 LASSERT (rpcev->ev_fired == 0);
1300 rpcev->ev_fired = 1;
1301 rpcev->ev_status = (ev->type == LNET_EVENT_UNLINK) ?
1302 -EINTR : ev->status;
1303 swi_schedule_workitem(&crpc->crpc_wi);
1305 cfs_spin_unlock(&crpc->crpc_lock);
1308 case SRPC_REQUEST_RCVD:
1309 sv = rpcev->ev_data;
1311 LASSERT (rpcev == &sv->sv_ev);
1313 cfs_spin_lock(&sv->sv_lock);
1315 LASSERT (ev->unlinked);
1316 LASSERT (ev->type == LNET_EVENT_PUT ||
1317 ev->type == LNET_EVENT_UNLINK);
1318 LASSERT (ev->type != LNET_EVENT_UNLINK ||
1319 sv->sv_shuttingdown);
1321 buffer = container_of(ev->md.start, srpc_buffer_t, buf_msg);
1322 buffer->buf_peer = ev->initiator;
1323 buffer->buf_self = ev->target.nid;
1325 sv->sv_nposted_msg--;
1326 LASSERT (sv->sv_nposted_msg >= 0);
1328 if (sv->sv_shuttingdown) {
1329 /* Leave buffer on sv->sv_posted_msgq since
1330 * srpc_finish_service needs to traverse it. */
1331 cfs_spin_unlock(&sv->sv_lock);
1335 cfs_list_del(&buffer->buf_list); /* from sv->sv_posted_msgq */
1336 msg = &buffer->buf_msg;
1337 type = srpc_service2request(sv->sv_id);
1339 if (ev->status != 0 || ev->mlength != sizeof(*msg) ||
1340 (msg->msg_type != type &&
1341 msg->msg_type != __swab32(type)) ||
1342 (msg->msg_magic != SRPC_MSG_MAGIC &&
1343 msg->msg_magic != __swab32(SRPC_MSG_MAGIC))) {
1344 CERROR ("Dropping RPC (%s) from %s: "
1345 "status %d mlength %d type %u magic %u.\n",
1346 sv->sv_name, libcfs_id2str(ev->initiator),
1347 ev->status, ev->mlength,
1348 msg->msg_type, msg->msg_magic);
1350 /* NB can't call srpc_service_recycle_buffer here since
1351 * it may call LNetM[DE]Attach. The invalid magic tells
1352 * srpc_handle_rpc to drop this RPC */
1356 if (!cfs_list_empty(&sv->sv_free_rpcq)) {
1357 srpc = cfs_list_entry(sv->sv_free_rpcq.next,
1358 srpc_server_rpc_t, srpc_list);
1359 cfs_list_del(&srpc->srpc_list);
1361 srpc_init_server_rpc(srpc, sv, buffer);
1362 cfs_list_add_tail(&srpc->srpc_list,
1363 &sv->sv_active_rpcq);
1364 srpc_schedule_server_rpc(srpc);
1366 cfs_list_add_tail(&buffer->buf_list,
1367 &sv->sv_blocked_msgq);
1370 cfs_spin_unlock(&sv->sv_lock);
1372 cfs_spin_lock(&srpc_data.rpc_glock);
1373 srpc_data.rpc_counters.rpcs_rcvd++;
1374 cfs_spin_unlock(&srpc_data.rpc_glock);
1377 case SRPC_BULK_GET_RPLD:
1378 LASSERT (ev->type == LNET_EVENT_SEND ||
1379 ev->type == LNET_EVENT_REPLY ||
1380 ev->type == LNET_EVENT_UNLINK);
1383 break; /* wait for final event */
1385 case SRPC_BULK_PUT_SENT:
1386 if (ev->status == 0 && ev->type != LNET_EVENT_UNLINK) {
1387 cfs_spin_lock(&srpc_data.rpc_glock);
1389 if (rpcev->ev_type == SRPC_BULK_GET_RPLD)
1390 srpc_data.rpc_counters.bulk_get += ev->mlength;
1392 srpc_data.rpc_counters.bulk_put += ev->mlength;
1394 cfs_spin_unlock(&srpc_data.rpc_glock);
1396 case SRPC_REPLY_SENT:
1397 srpc = rpcev->ev_data;
1398 sv = srpc->srpc_service;
1400 LASSERT (rpcev == &srpc->srpc_ev);
1402 cfs_spin_lock(&sv->sv_lock);
1404 rpcev->ev_fired = 1;
1405 rpcev->ev_status = (ev->type == LNET_EVENT_UNLINK) ?
1406 -EINTR : ev->status;
1407 srpc_schedule_server_rpc(srpc);
1409 cfs_spin_unlock(&sv->sv_lock);
1419 srpc_check_event (int timeout)
1425 rc = LNetEQPoll(&srpc_data.rpc_lnet_eq, 1,
1426 timeout * 1000, &ev, &i);
1427 if (rc == 0) return 0;
1429 LASSERT (rc == -EOVERFLOW || rc == 1);
1431 /* We can't affort to miss any events... */
1432 if (rc == -EOVERFLOW) {
1433 CERROR ("Dropped an event!!!\n");
1437 srpc_lnet_ev_handler(&ev);
1448 memset(&srpc_data, 0, sizeof(struct smoketest_rpc));
1449 cfs_spin_lock_init(&srpc_data.rpc_glock);
1451 /* 1 second pause to avoid timestamp reuse */
1452 cfs_pause(cfs_time_seconds(1));
1453 srpc_data.rpc_matchbits = ((__u64) cfs_time_current_sec()) << 48;
1455 srpc_data.rpc_state = SRPC_STATE_NONE;
1458 rc = LNetNIInit(LUSTRE_SRV_LNET_PID);
1460 if (the_lnet.ln_server_mode_flag)
1461 rc = LNetNIInit(LUSTRE_SRV_LNET_PID);
1463 rc = LNetNIInit(getpid() | LNET_PID_USERFLAG);
1466 CERROR ("LNetNIInit() has failed: %d\n", rc);
1470 srpc_data.rpc_state = SRPC_STATE_NI_INIT;
1472 LNetInvalidateHandle(&srpc_data.rpc_lnet_eq);
1474 rc = LNetEQAlloc(16, srpc_lnet_ev_handler, &srpc_data.rpc_lnet_eq);
1476 rc = LNetEQAlloc(10240, LNET_EQ_HANDLER_NONE, &srpc_data.rpc_lnet_eq);
1479 CERROR("LNetEQAlloc() has failed: %d\n", rc);
1483 rc = LNetSetLazyPortal(SRPC_FRAMEWORK_REQUEST_PORTAL);
1486 srpc_data.rpc_state = SRPC_STATE_EQ_INIT;
1494 srpc_data.rpc_state = SRPC_STATE_RUNNING;
1500 srpc_shutdown (void)
1506 state = srpc_data.rpc_state;
1507 srpc_data.rpc_state = SRPC_STATE_STOPPING;
1512 case SRPC_STATE_RUNNING:
1513 cfs_spin_lock(&srpc_data.rpc_glock);
1515 for (i = 0; i <= SRPC_SERVICE_MAX_ID; i++) {
1516 srpc_service_t *sv = srpc_data.rpc_services[i];
1518 LASSERTF (sv == NULL,
1519 "service not empty: id %d, name %s\n",
1523 cfs_spin_unlock(&srpc_data.rpc_glock);
1527 case SRPC_STATE_EQ_INIT:
1528 rc = LNetClearLazyPortal(SRPC_FRAMEWORK_REQUEST_PORTAL);
1530 rc = LNetEQFree(srpc_data.rpc_lnet_eq);
1531 LASSERT (rc == 0); /* the EQ should have no user by now */
1533 case SRPC_STATE_NI_INIT: