lustre/ptlrpc/events.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   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.
  11  *
  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).
  17  *
  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
  21  *
  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
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright (c) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  */
  36
  37 #define DEBUG_SUBSYSTEM S_RPC
  38
  39 #ifndef __KERNEL__
  40 # include <liblustre.h>
  41 #else
  42 # include <libcfs/libcfs.h>
  43 # ifdef __mips64__
  44 #  include <linux/kernel.h>
  45 # endif
  46 #endif
  47
  48 #include <obd_class.h>
  49 #include <lustre_net.h>
  50 #include <lustre_sec.h>
  51 #include "ptlrpc_internal.h"
  52
  53 lnet_handle_eq_t   ptlrpc_eq_h;
  54
  55 /*
  56  *  Client's outgoing request callback
  57  */
  58 void request_out_callback(lnet_event_t *ev)
  59 {
  60         struct ptlrpc_cb_id   *cbid = ev->md.user_ptr;
  61         struct ptlrpc_request *req = cbid->cbid_arg;
  62         ENTRY;
  63
  64         LASSERT (ev->type == LNET_EVENT_SEND ||
  65                  ev->type == LNET_EVENT_UNLINK);
  66         LASSERT (ev->unlinked);
  67
  68         DEBUG_REQ(D_NET, req, "type %d, status %d", ev->type, ev->status);
  69
  70         sptlrpc_request_out_callback(req);
  71         req->rq_real_sent = cfs_time_current_sec();
  72
  73         if (ev->type == LNET_EVENT_UNLINK || ev->status != 0) {
  74
  75                 /* Failed send: make it seem like the reply timed out, just
  76                  * like failing sends in client.c does currently...  */
  77
  78                 cfs_spin_lock(&req->rq_lock);
  79                 req->rq_net_err = 1;
  80                 cfs_spin_unlock(&req->rq_lock);
  81
  82                 ptlrpc_client_wake_req(req);
  83         }
  84
  85         ptlrpc_req_finished(req);
  86
  87         EXIT;
  88 }
  89
  90 /*
  91  * Client's incoming reply callback
  92  */
  93 void reply_in_callback(lnet_event_t *ev)
  94 {
  95         struct ptlrpc_cb_id   *cbid = ev->md.user_ptr;
  96         struct ptlrpc_request *req = cbid->cbid_arg;
  97         ENTRY;
  98
  99         DEBUG_REQ(D_NET, req, "type %d, status %d", ev->type, ev->status);
 100
 101         LASSERT (ev->type == LNET_EVENT_PUT || ev->type == LNET_EVENT_UNLINK);
 102         LASSERT (ev->md.start == req->rq_repbuf);
 103         LASSERT (ev->offset + ev->mlength <= req->rq_repbuf_len);
 104         /* We've set LNET_MD_MANAGE_REMOTE for all outgoing requests
 105            for adaptive timeouts' early reply. */
 106         LASSERT((ev->md.options & LNET_MD_MANAGE_REMOTE) != 0);
 107
 108         cfs_spin_lock(&req->rq_lock);
 109
 110         req->rq_receiving_reply = 0;
 111         req->rq_early = 0;
 112         if (ev->unlinked)
 113                 req->rq_must_unlink = 0;
 114
 115         if (ev->status)
 116                 goto out_wake;
 117
 118         if (ev->type == LNET_EVENT_UNLINK) {
 119                 LASSERT(ev->unlinked);
 120                 DEBUG_REQ(D_NET, req, "unlink");
 121                 goto out_wake;
 122         }
 123
 124         if (ev->mlength < ev->rlength ) {
 125                 CDEBUG(D_RPCTRACE, "truncate req %p rpc %d - %d+%d\n", req,
 126                        req->rq_replen, ev->rlength, ev->offset);
 127                 req->rq_reply_truncate = 1;
 128                 req->rq_replied = 1;
 129                 req->rq_status = -EOVERFLOW;
 130                 req->rq_nob_received = ev->rlength + ev->offset;
 131                 goto out_wake;
 132         }
 133
 134         if ((ev->offset == 0) &&
 135             ((lustre_msghdr_get_flags(req->rq_reqmsg) & MSGHDR_AT_SUPPORT))) {
 136                 /* Early reply */
 137                 DEBUG_REQ(D_ADAPTTO, req,
 138                           "Early reply received: mlen=%u offset=%d replen=%d "
 139                           "replied=%d unlinked=%d", ev->mlength, ev->offset,
 140                           req->rq_replen, req->rq_replied, ev->unlinked);
 141
 142                 req->rq_early_count++; /* number received, client side */
 143
 144                 if (req->rq_replied)   /* already got the real reply */
 145                         goto out_wake;
 146
 147                 req->rq_early = 1;
 148                 req->rq_reply_off = ev->offset;
 149                 req->rq_nob_received = ev->mlength;
 150                 /* And we're still receiving */
 151                 req->rq_receiving_reply = 1;
 152         } else {
 153                 /* Real reply */
 154                 req->rq_rep_swab_mask = 0;
 155                 req->rq_replied = 1;
 156                 req->rq_reply_off = ev->offset;
 157                 req->rq_nob_received = ev->mlength;
 158                 /* LNetMDUnlink can't be called under the LNET_LOCK,
 159                    so we must unlink in ptlrpc_unregister_reply */
 160                 DEBUG_REQ(D_INFO, req,
 161                           "reply in flags=%x mlen=%u offset=%d replen=%d",
 162                           lustre_msg_get_flags(req->rq_reqmsg),
 163                           ev->mlength, ev->offset, req->rq_replen);
 164         }
 165
 166         req->rq_import->imp_last_reply_time = cfs_time_current_sec();
 167
 168 out_wake:
 169         /* NB don't unlock till after wakeup; req can disappear under us
 170          * since we don't have our own ref */
 171         ptlrpc_client_wake_req(req);
 172         cfs_spin_unlock(&req->rq_lock);
 173         EXIT;
 174 }
 175
 176 /*
 177  * Client's bulk has been written/read
 178  */
 179 void client_bulk_callback (lnet_event_t *ev)
 180 {
 181         struct ptlrpc_cb_id     *cbid = ev->md.user_ptr;
 182         struct ptlrpc_bulk_desc *desc = cbid->cbid_arg;
 183         ENTRY;
 184
 185         LASSERT ((desc->bd_type == BULK_PUT_SINK &&
 186                   ev->type == LNET_EVENT_PUT) ||
 187                  (desc->bd_type == BULK_GET_SOURCE &&
 188                   ev->type == LNET_EVENT_GET) ||
 189                  ev->type == LNET_EVENT_UNLINK);
 190         LASSERT (ev->unlinked);
 191
 192         CDEBUG((ev->status == 0) ? D_NET : D_ERROR,
 193                "event type %d, status %d, desc %p\n",
 194                ev->type, ev->status, desc);
 195
 196         cfs_spin_lock(&desc->bd_lock);
 197
 198         LASSERT(desc->bd_network_rw);
 199         desc->bd_network_rw = 0;
 200
 201         if (ev->type != LNET_EVENT_UNLINK && ev->status == 0) {
 202                 desc->bd_success = 1;
 203                 desc->bd_nob_transferred = ev->mlength;
 204                 desc->bd_sender = ev->sender;
 205         }
 206
 207         /* release the encrypted pages for write */
 208         if (desc->bd_req->rq_bulk_write)
 209                 sptlrpc_enc_pool_put_pages(desc);
 210
 211         /* NB don't unlock till after wakeup; desc can disappear under us
 212          * otherwise */
 213         ptlrpc_client_wake_req(desc->bd_req);
 214
 215         cfs_spin_unlock(&desc->bd_lock);
 216         EXIT;
 217 }
 218
 219 /*
 220  * Server's incoming request callback
 221  */
 222 void request_in_callback(lnet_event_t *ev)
 223 {
 224         struct ptlrpc_cb_id               *cbid = ev->md.user_ptr;
 225         struct ptlrpc_request_buffer_desc *rqbd = cbid->cbid_arg;
 226         struct ptlrpc_service             *service = rqbd->rqbd_service;
 227         struct ptlrpc_request             *req;
 228         ENTRY;
 229
 230         LASSERT (ev->type == LNET_EVENT_PUT ||
 231                  ev->type == LNET_EVENT_UNLINK);
 232         LASSERT ((char *)ev->md.start >= rqbd->rqbd_buffer);
 233         LASSERT ((char *)ev->md.start + ev->offset + ev->mlength <=
 234                  rqbd->rqbd_buffer + service->srv_buf_size);
 235
 236         CDEBUG((ev->status == 0) ? D_NET : D_ERROR,
 237                "event type %d, status %d, service %s\n",
 238                ev->type, ev->status, service->srv_name);
 239
 240         if (ev->unlinked) {
 241                 /* If this is the last request message to fit in the
 242                  * request buffer we can use the request object embedded in
 243                  * rqbd.  Note that if we failed to allocate a request,
 244                  * we'd have to re-post the rqbd, which we can't do in this
 245                  * context. */
 246                 req = &rqbd->rqbd_req;
 247                 memset(req, 0, sizeof (*req));
 248         } else {
 249                 LASSERT (ev->type == LNET_EVENT_PUT);
 250                 if (ev->status != 0) {
 251                         /* We moaned above already... */
 252                         return;
 253                 }
 254                 OBD_ALLOC_GFP(req, sizeof(*req), CFS_ALLOC_ATOMIC_TRY);
 255                 if (req == NULL) {
 256                         CERROR("Can't allocate incoming request descriptor: "
 257                                "Dropping %s RPC from %s\n",
 258                                service->srv_name,
 259                                libcfs_id2str(ev->initiator));
 260                         return;
 261                 }
 262         }
 263
 264         /* NB we ABSOLUTELY RELY on req being zeroed, so pointers are NULL,
 265          * flags are reset and scalars are zero.  We only set the message
 266          * size to non-zero if this was a successful receive. */
 267         req->rq_xid = ev->match_bits;
 268         req->rq_reqbuf = ev->md.start + ev->offset;
 269         if (ev->type == LNET_EVENT_PUT && ev->status == 0)
 270                 req->rq_reqdata_len = ev->mlength;
 271         cfs_gettimeofday(&req->rq_arrival_time);
 272         req->rq_peer = ev->initiator;
 273         req->rq_self = ev->target.nid;
 274         req->rq_rqbd = rqbd;
 275         req->rq_phase = RQ_PHASE_NEW;
 276 #ifdef CRAY_XT3
 277         req->rq_uid = ev->uid;
 278 #endif
 279         cfs_spin_lock_init(&req->rq_lock);
 280         CFS_INIT_LIST_HEAD(&req->rq_timed_list);
 281         cfs_atomic_set(&req->rq_refcount, 1);
 282         if (ev->type == LNET_EVENT_PUT)
 283                 CDEBUG(D_INFO, "incoming req@%p x"LPU64" msgsize %u\n",
 284                        req, req->rq_xid, ev->mlength);
 285
 286         CDEBUG(D_RPCTRACE, "peer: %s\n", libcfs_id2str(req->rq_peer));
 287
 288         cfs_spin_lock(&service->srv_lock);
 289
 290         req->rq_history_seq = service->srv_request_seq++;
 291         cfs_list_add_tail(&req->rq_history_list, &service->srv_request_history);
 292
 293         if (ev->unlinked) {
 294                 service->srv_nrqbd_receiving--;
 295                 CDEBUG(D_INFO, "Buffer complete: %d buffers still posted\n",
 296                        service->srv_nrqbd_receiving);
 297
 298                 /* Normally, don't complain about 0 buffers posted; LNET won't
 299                  * drop incoming reqs since we set the portal lazy */
 300                 if (test_req_buffer_pressure &&
 301                     ev->type != LNET_EVENT_UNLINK &&
 302                     service->srv_nrqbd_receiving == 0)
 303                         CWARN("All %s request buffers busy\n",
 304                               service->srv_name);
 305
 306                 /* req takes over the network's ref on rqbd */
 307         } else {
 308                 /* req takes a ref on rqbd */
 309                 rqbd->rqbd_refcount++;
 310         }
 311
 312         cfs_list_add_tail(&req->rq_list, &service->srv_req_in_queue);
 313         service->srv_n_queued_reqs++;
 314
 315         /* NB everything can disappear under us once the request
 316          * has been queued and we unlock, so do the wake now... */
 317         cfs_waitq_signal(&service->srv_waitq);
 318
 319         cfs_spin_unlock(&service->srv_lock);
 320         EXIT;
 321 }
 322
 323 /*
 324  *  Server's outgoing reply callback
 325  */
 326 void reply_out_callback(lnet_event_t *ev)
 327 {
 328         struct ptlrpc_cb_id       *cbid = ev->md.user_ptr;
 329         struct ptlrpc_reply_state *rs = cbid->cbid_arg;
 330         struct ptlrpc_service     *svc = rs->rs_service;
 331         ENTRY;
 332
 333         LASSERT (ev->type == LNET_EVENT_SEND ||
 334                  ev->type == LNET_EVENT_ACK ||
 335                  ev->type == LNET_EVENT_UNLINK);
 336
 337         if (!rs->rs_difficult) {
 338                 /* 'Easy' replies have no further processing so I drop the
 339                  * net's ref on 'rs' */
 340                 LASSERT (ev->unlinked);
 341                 ptlrpc_rs_decref(rs);
 342                 EXIT;
 343                 return;
 344         }
 345
 346         LASSERT (rs->rs_on_net);
 347
 348         if (ev->unlinked) {
 349                 /* Last network callback. The net's ref on 'rs' stays put
 350                  * until ptlrpc_handle_rs() is done with it */
 351                 cfs_spin_lock(&svc->srv_rs_lock);
 352                 cfs_spin_lock(&rs->rs_lock);
 353                 rs->rs_on_net = 0;
 354                 if (!rs->rs_no_ack ||
 355                     rs->rs_transno <= rs->rs_export->exp_obd->obd_last_committed)
 356                         ptlrpc_schedule_difficult_reply (rs);
 357                 cfs_spin_unlock(&rs->rs_lock);
 358                 cfs_spin_unlock(&svc->srv_rs_lock);
 359         }
 360
 361         EXIT;
 362 }
 363
 364 /*
 365  * Server's bulk completion callback
 366  */
 367 void server_bulk_callback (lnet_event_t *ev)
 368 {
 369         struct ptlrpc_cb_id     *cbid = ev->md.user_ptr;
 370         struct ptlrpc_bulk_desc *desc = cbid->cbid_arg;
 371         ENTRY;
 372
 373         LASSERT (ev->type == LNET_EVENT_SEND ||
 374                  ev->type == LNET_EVENT_UNLINK ||
 375                  (desc->bd_type == BULK_PUT_SOURCE &&
 376                   ev->type == LNET_EVENT_ACK) ||
 377                  (desc->bd_type == BULK_GET_SINK &&
 378                   ev->type == LNET_EVENT_REPLY));
 379
 380         CDEBUG((ev->status == 0) ? D_NET : D_ERROR,
 381                "event type %d, status %d, desc %p\n",
 382                ev->type, ev->status, desc);
 383
 384         cfs_spin_lock(&desc->bd_lock);
 385
 386         if ((ev->type == LNET_EVENT_ACK ||
 387              ev->type == LNET_EVENT_REPLY) &&
 388             ev->status == 0) {
 389                 /* We heard back from the peer, so even if we get this
 390                  * before the SENT event (oh yes we can), we know we
 391                  * read/wrote the peer buffer and how much... */
 392                 desc->bd_success = 1;
 393                 desc->bd_nob_transferred = ev->mlength;
 394                 desc->bd_sender = ev->sender;
 395         }
 396
 397         if (ev->unlinked) {
 398                 /* This is the last callback no matter what... */
 399                 desc->bd_network_rw = 0;
 400                 cfs_waitq_signal(&desc->bd_waitq);
 401         }
 402
 403         cfs_spin_unlock(&desc->bd_lock);
 404         EXIT;
 405 }
 406
 407 static void ptlrpc_master_callback(lnet_event_t *ev)
 408 {
 409         struct ptlrpc_cb_id *cbid = ev->md.user_ptr;
 410         void (*callback)(lnet_event_t *ev) = cbid->cbid_fn;
 411
 412         /* Honestly, it's best to find out early. */
 413         LASSERT (cbid->cbid_arg != LP_POISON);
 414         LASSERT (callback == request_out_callback ||
 415                  callback == reply_in_callback ||
 416                  callback == client_bulk_callback ||
 417                  callback == request_in_callback ||
 418                  callback == reply_out_callback ||
 419                  callback == server_bulk_callback);
 420
 421         callback (ev);
 422 }
 423
 424 int ptlrpc_uuid_to_peer (struct obd_uuid *uuid,
 425                          lnet_process_id_t *peer, lnet_nid_t *self)
 426 {
 427         int               best_dist = 0;
 428         __u32             best_order = 0;
 429         int               count = 0;
 430         int               rc = -ENOENT;
 431         int               portals_compatibility;
 432         int               dist;
 433         __u32             order;
 434         lnet_nid_t        dst_nid;
 435         lnet_nid_t        src_nid;
 436
 437         portals_compatibility = LNetCtl(IOC_LIBCFS_PORTALS_COMPATIBILITY, NULL);
 438
 439         peer->pid = LUSTRE_SRV_LNET_PID;
 440
 441         /* Choose the matching UUID that's closest */
 442         while (lustre_uuid_to_peer(uuid->uuid, &dst_nid, count++) == 0) {
 443                 dist = LNetDist(dst_nid, &src_nid, &order);
 444                 if (dist < 0)
 445                         continue;
 446
 447                 if (dist == 0) {                /* local! use loopback LND */
 448                         peer->nid = *self = LNET_MKNID(LNET_MKNET(LOLND, 0), 0);
 449                         rc = 0;
 450                         break;
 451                 }
 452
 453                 if (rc < 0 ||
 454                     dist < best_dist ||
 455                     (dist == best_dist && order < best_order)) {
 456                         best_dist = dist;
 457                         best_order = order;
 458
 459                         if (portals_compatibility > 1) {
 460                                 /* Strong portals compatibility: Zero the nid's
 461                                  * NET, so if I'm reading new config logs, or
 462                                  * getting configured by (new) lconf I can
 463                                  * still talk to old servers. */
 464                                 dst_nid = LNET_MKNID(0, LNET_NIDADDR(dst_nid));
 465                                 src_nid = LNET_MKNID(0, LNET_NIDADDR(src_nid));
 466                         }
 467                         peer->nid = dst_nid;
 468                         *self = src_nid;
 469                         rc = 0;
 470                 }
 471         }
 472
 473         CDEBUG(D_NET,"%s->%s\n", uuid->uuid, libcfs_id2str(*peer));
 474         return rc;
 475 }
 476
 477 void ptlrpc_ni_fini(void)
 478 {
 479         cfs_waitq_t         waitq;
 480         struct l_wait_info  lwi;
 481         int                 rc;
 482         int                 retries;
 483
 484         /* Wait for the event queue to become idle since there may still be
 485          * messages in flight with pending events (i.e. the fire-and-forget
 486          * messages == client requests and "non-difficult" server
 487          * replies */
 488
 489         for (retries = 0;; retries++) {
 490                 rc = LNetEQFree(ptlrpc_eq_h);
 491                 switch (rc) {
 492                 default:
 493                         LBUG();
 494
 495                 case 0:
 496                         LNetNIFini();
 497                         return;
 498
 499                 case -EBUSY:
 500                         if (retries != 0)
 501                                 CWARN("Event queue still busy\n");
 502
 503                         /* Wait for a bit */
 504                         cfs_waitq_init(&waitq);
 505                         lwi = LWI_TIMEOUT(cfs_time_seconds(2), NULL, NULL);
 506                         l_wait_event(waitq, 0, &lwi);
 507                         break;
 508                 }
 509         }
 510         /* notreached */
 511 }
 512
 513 lnet_pid_t ptl_get_pid(void)
 514 {
 515         lnet_pid_t        pid;
 516
 517 #ifndef  __KERNEL__
 518         pid = getpid();
 519 #else
 520         pid = LUSTRE_SRV_LNET_PID;
 521 #endif
 522         return pid;
 523 }
 524
 525 int ptlrpc_ni_init(void)
 526 {
 527         int              rc;
 528         lnet_pid_t       pid;
 529
 530         pid = ptl_get_pid();
 531         CDEBUG(D_NET, "My pid is: %x\n", pid);
 532
 533         /* We're not passing any limits yet... */
 534         rc = LNetNIInit(pid);
 535         if (rc < 0) {
 536                 CDEBUG (D_NET, "Can't init network interface: %d\n", rc);
 537                 return (-ENOENT);
 538         }
 539
 540         /* CAVEAT EMPTOR: how we process portals events is _radically_
 541          * different depending on... */
 542 #ifdef __KERNEL__
 543         /* kernel portals calls our master callback when events are added to
 544          * the event queue.  In fact lustre never pulls events off this queue,
 545          * so it's only sized for some debug history. */
 546         rc = LNetEQAlloc(1024, ptlrpc_master_callback, &ptlrpc_eq_h);
 547 #else
 548         /* liblustre calls the master callback when it removes events from the
 549          * event queue.  The event queue has to be big enough not to drop
 550          * anything */
 551         rc = LNetEQAlloc(10240, LNET_EQ_HANDLER_NONE, &ptlrpc_eq_h);
 552 #endif
 553         if (rc == 0)
 554                 return 0;
 555
 556         CERROR ("Failed to allocate event queue: %d\n", rc);
 557         LNetNIFini();
 558
 559         return (-ENOMEM);
 560 }
 561
 562 #ifndef __KERNEL__
 563 CFS_LIST_HEAD(liblustre_wait_callbacks);
 564 CFS_LIST_HEAD(liblustre_idle_callbacks);
 565 void *liblustre_services_callback;
 566
 567 void *
 568 liblustre_register_waitidle_callback (cfs_list_t *callback_list,
 569                                       const char *name,
 570                                       int (*fn)(void *arg), void *arg)
 571 {
 572         struct liblustre_wait_callback *llwc;
 573
 574         OBD_ALLOC(llwc, sizeof(*llwc));
 575         LASSERT (llwc != NULL);
 576
 577         llwc->llwc_name = name;
 578         llwc->llwc_fn = fn;
 579         llwc->llwc_arg = arg;
 580         cfs_list_add_tail(&llwc->llwc_list, callback_list);
 581
 582         return (llwc);
 583 }
 584
 585 void
 586 liblustre_deregister_waitidle_callback (void *opaque)
 587 {
 588         struct liblustre_wait_callback *llwc = opaque;
 589
 590         cfs_list_del(&llwc->llwc_list);
 591         OBD_FREE(llwc, sizeof(*llwc));
 592 }
 593
 594 void *
 595 liblustre_register_wait_callback (const char *name,
 596                                   int (*fn)(void *arg), void *arg)
 597 {
 598         return liblustre_register_waitidle_callback(&liblustre_wait_callbacks,
 599                                                     name, fn, arg);
 600 }
 601
 602 void
 603 liblustre_deregister_wait_callback (void *opaque)
 604 {
 605         liblustre_deregister_waitidle_callback(opaque);
 606 }
 607
 608 void *
 609 liblustre_register_idle_callback (const char *name,
 610                                   int (*fn)(void *arg), void *arg)
 611 {
 612         return liblustre_register_waitidle_callback(&liblustre_idle_callbacks,
 613                                                     name, fn, arg);
 614 }
 615
 616 void
 617 liblustre_deregister_idle_callback (void *opaque)
 618 {
 619         liblustre_deregister_waitidle_callback(opaque);
 620 }
 621
 622 int
 623 liblustre_check_events (int timeout)
 624 {
 625         lnet_event_t ev;
 626         int         rc;
 627         int         i;
 628         ENTRY;
 629
 630         rc = LNetEQPoll(&ptlrpc_eq_h, 1, timeout * 1000, &ev, &i);
 631         if (rc == 0)
 632                 RETURN(0);
 633
 634         LASSERT (rc == -EOVERFLOW || rc == 1);
 635
 636         /* liblustre: no asynch callback so we can't afford to miss any
 637          * events... */
 638         if (rc == -EOVERFLOW) {
 639                 CERROR ("Dropped an event!!!\n");
 640                 abort();
 641         }
 642
 643         ptlrpc_master_callback (&ev);
 644         RETURN(1);
 645 }
 646
 647 int liblustre_waiting = 0;
 648
 649 int
 650 liblustre_wait_event (int timeout)
 651 {
 652         cfs_list_t                     *tmp;
 653         struct liblustre_wait_callback *llwc;
 654         int                             found_something = 0;
 655
 656         /* single threaded recursion check... */
 657         liblustre_waiting = 1;
 658
 659         for (;;) {
 660                 /* Deal with all pending events */
 661                 while (liblustre_check_events(0))
 662                         found_something = 1;
 663
 664                 /* Give all registered callbacks a bite at the cherry */
 665                 cfs_list_for_each(tmp, &liblustre_wait_callbacks) {
 666                         llwc = cfs_list_entry(tmp,
 667                                               struct liblustre_wait_callback,
 668                                               llwc_list);
 669
 670                         if (llwc->llwc_fn(llwc->llwc_arg))
 671                                 found_something = 1;
 672                 }
 673
 674                 if (found_something || timeout == 0)
 675                         break;
 676
 677                 /* Nothing so far, but I'm allowed to block... */
 678                 found_something = liblustre_check_events(timeout);
 679                 if (!found_something)           /* still nothing */
 680                         break;                  /* I timed out */
 681         }
 682
 683         liblustre_waiting = 0;
 684
 685         return found_something;
 686 }
 687
 688 void
 689 liblustre_wait_idle(void)
 690 {
 691         static int recursed = 0;
 692
 693         cfs_list_t                     *tmp;
 694         struct liblustre_wait_callback *llwc;
 695         int                             idle = 0;
 696
 697         LASSERT(!recursed);
 698         recursed = 1;
 699
 700         do {
 701                 liblustre_wait_event(0);
 702
 703                 idle = 1;
 704
 705                 cfs_list_for_each(tmp, &liblustre_idle_callbacks) {
 706                         llwc = cfs_list_entry(tmp,
 707                                               struct liblustre_wait_callback,
 708                                               llwc_list);
 709
 710                         if (!llwc->llwc_fn(llwc->llwc_arg)) {
 711                                 idle = 0;
 712                                 break;
 713                         }
 714                 }
 715
 716         } while (!idle);
 717
 718         recursed = 0;
 719 }
 720
 721 #endif /* __KERNEL__ */
 722
 723 int ptlrpc_init_portals(void)
 724 {
 725         int   rc = ptlrpc_ni_init();
 726
 727         if (rc != 0) {
 728                 CERROR("network initialisation failed\n");
 729                 return -EIO;
 730         }
 731 #ifndef __KERNEL__
 732         liblustre_services_callback =
 733                 liblustre_register_wait_callback("liblustre_check_services",
 734                                                  &liblustre_check_services,
 735                                                  NULL);
 736         cfs_init_completion_module(liblustre_wait_event);
 737 #endif
 738         rc = ptlrpcd_addref();
 739         if (rc == 0)
 740                 return 0;
 741
 742         CERROR("rpcd initialisation failed\n");
 743 #ifndef __KERNEL__
 744         liblustre_deregister_wait_callback(liblustre_services_callback);
 745 #endif
 746         ptlrpc_ni_fini();
 747         return rc;
 748 }
 749
 750 void ptlrpc_exit_portals(void)
 751 {
 752 #ifndef __KERNEL__
 753         liblustre_deregister_wait_callback(liblustre_services_callback);
 754 #endif
 755         ptlrpcd_decref();
 756         ptlrpc_ni_fini();
 757 }