Whamcloud - gitweb
b=17631
[fs/lustre-release.git] / lustre / ptlrpc / client.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  2008 Sun Microsystems, Inc. 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 #ifndef __KERNEL__
39 #include <errno.h>
40 #include <signal.h>
41 #include <liblustre.h>
42 #endif
43
44 #include <obd_support.h>
45 #include <obd_class.h>
46 #include <lustre_lib.h>
47 #include <lustre_ha.h>
48 #include <lustre_import.h>
49 #include <lustre_req_layout.h>
50
51 #include "ptlrpc_internal.h"
52
53 void ptlrpc_init_client(int req_portal, int rep_portal, char *name,
54                         struct ptlrpc_client *cl)
55 {
56         cl->cli_request_portal = req_portal;
57         cl->cli_reply_portal   = rep_portal;
58         cl->cli_name           = name;
59 }
60
61 struct ptlrpc_connection *ptlrpc_uuid_to_connection(struct obd_uuid *uuid)
62 {
63         struct ptlrpc_connection *c;
64         lnet_nid_t                self;
65         lnet_process_id_t         peer;
66         int                       err;
67
68         err = ptlrpc_uuid_to_peer(uuid, &peer, &self);
69         if (err != 0) {
70                 CERROR("cannot find peer %s!\n", uuid->uuid);
71                 return NULL;
72         }
73
74         c = ptlrpc_connection_get(peer, self, uuid);
75         if (c) {
76                 memcpy(c->c_remote_uuid.uuid,
77                        uuid->uuid, sizeof(c->c_remote_uuid.uuid));
78         }
79
80         CDEBUG(D_INFO, "%s -> %p\n", uuid->uuid, c);
81
82         return c;
83 }
84
85 static inline struct ptlrpc_bulk_desc *new_bulk(int npages, int type, int portal)
86 {
87         struct ptlrpc_bulk_desc *desc;
88
89         OBD_ALLOC(desc, offsetof (struct ptlrpc_bulk_desc, bd_iov[npages]));
90         if (!desc)
91                 return NULL;
92
93         spin_lock_init(&desc->bd_lock);
94         cfs_waitq_init(&desc->bd_waitq);
95         desc->bd_max_iov = npages;
96         desc->bd_iov_count = 0;
97         LNetInvalidateHandle(&desc->bd_md_h);
98         desc->bd_portal = portal;
99         desc->bd_type = type;
100
101         return desc;
102 }
103
104 struct ptlrpc_bulk_desc *ptlrpc_prep_bulk_imp(struct ptlrpc_request *req,
105                                               int npages, int type, int portal)
106 {
107         struct obd_import *imp = req->rq_import;
108         struct ptlrpc_bulk_desc *desc;
109
110         ENTRY;
111         LASSERT(type == BULK_PUT_SINK || type == BULK_GET_SOURCE);
112         desc = new_bulk(npages, type, portal);
113         if (desc == NULL)
114                 RETURN(NULL);
115
116         desc->bd_import_generation = req->rq_import_generation;
117         desc->bd_import = class_import_get(imp);
118         desc->bd_req = req;
119
120         desc->bd_cbid.cbid_fn  = client_bulk_callback;
121         desc->bd_cbid.cbid_arg = desc;
122
123         /* This makes req own desc, and free it when she frees herself */
124         req->rq_bulk = desc;
125
126         return desc;
127 }
128
129 struct ptlrpc_bulk_desc *ptlrpc_prep_bulk_exp(struct ptlrpc_request *req,
130                                               int npages, int type, int portal)
131 {
132         struct obd_export *exp = req->rq_export;
133         struct ptlrpc_bulk_desc *desc;
134
135         ENTRY;
136         LASSERT(type == BULK_PUT_SOURCE || type == BULK_GET_SINK);
137
138         desc = new_bulk(npages, type, portal);
139         if (desc == NULL)
140                 RETURN(NULL);
141
142         desc->bd_export = class_export_get(exp);
143         desc->bd_req = req;
144
145         desc->bd_cbid.cbid_fn  = server_bulk_callback;
146         desc->bd_cbid.cbid_arg = desc;
147
148         /* NB we don't assign rq_bulk here; server-side requests are
149          * re-used, and the handler frees the bulk desc explicitly. */
150
151         return desc;
152 }
153
154 void ptlrpc_prep_bulk_page(struct ptlrpc_bulk_desc *desc,
155                            cfs_page_t *page, int pageoffset, int len)
156 {
157         LASSERT(desc->bd_iov_count < desc->bd_max_iov);
158         LASSERT(page != NULL);
159         LASSERT(pageoffset >= 0);
160         LASSERT(len > 0);
161         LASSERT(pageoffset + len <= CFS_PAGE_SIZE);
162
163         desc->bd_nob += len;
164
165         ptlrpc_add_bulk_page(desc, page, pageoffset, len);
166 }
167
168 void ptlrpc_free_bulk(struct ptlrpc_bulk_desc *desc)
169 {
170         ENTRY;
171
172         LASSERT(desc != NULL);
173         LASSERT(desc->bd_iov_count != LI_POISON); /* not freed already */
174         LASSERT(!desc->bd_network_rw);         /* network hands off or */
175         LASSERT((desc->bd_export != NULL) ^ (desc->bd_import != NULL));
176
177         sptlrpc_enc_pool_put_pages(desc);
178
179         if (desc->bd_export)
180                 class_export_put(desc->bd_export);
181         else
182                 class_import_put(desc->bd_import);
183
184         OBD_FREE(desc, offsetof(struct ptlrpc_bulk_desc,
185                                 bd_iov[desc->bd_max_iov]));
186         EXIT;
187 }
188
189 /* Set server timelimit for this req */
190 void ptlrpc_at_set_req_timeout(struct ptlrpc_request *req)
191 {
192         __u32 serv_est;
193         int idx;
194         struct imp_at *at;
195
196         LASSERT(req->rq_import);
197
198         if (AT_OFF) {
199                 /* non-AT settings */
200                 req->rq_timeout = req->rq_import->imp_server_timeout ?
201                         obd_timeout / 2 : obd_timeout;
202                 lustre_msg_set_timeout(req->rq_reqmsg, req->rq_timeout);
203                 return;
204         }
205
206         at = &req->rq_import->imp_at;
207         idx = import_at_get_index(req->rq_import,
208                                   req->rq_request_portal);
209         serv_est = at_get(&at->iat_service_estimate[idx]);
210         req->rq_timeout = at_est2timeout(serv_est);
211         /* We could get even fancier here, using history to predict increased
212            loading... */
213
214         /* Let the server know what this RPC timeout is by putting it in the
215            reqmsg*/
216         lustre_msg_set_timeout(req->rq_reqmsg, req->rq_timeout);
217 }
218
219 /* Adjust max service estimate based on server value */
220 static void ptlrpc_at_adj_service(struct ptlrpc_request *req,
221                                   unsigned int serv_est)
222 {
223         int idx;
224         unsigned int oldse;
225         struct imp_at *at;
226
227         /* do estimate only if is not in recovery */
228         if (!(req->rq_send_state & (LUSTRE_IMP_FULL | LUSTRE_IMP_CONNECTING)))
229                 return;
230
231         LASSERT(req->rq_import);
232         at = &req->rq_import->imp_at;
233
234         idx = import_at_get_index(req->rq_import, req->rq_request_portal);
235         /* max service estimates are tracked on the server side,
236            so just keep minimal history here */
237         oldse = at_add(&at->iat_service_estimate[idx], serv_est);
238         if (oldse != 0)
239                 CDEBUG(D_ADAPTTO, "The RPC service estimate for %s ptl %d "
240                        "has changed from %d to %d\n",
241                        req->rq_import->imp_obd->obd_name,req->rq_request_portal,
242                        oldse, at_get(&at->iat_service_estimate[idx]));
243 }
244
245 /* Expected network latency per remote node (secs) */
246 int ptlrpc_at_get_net_latency(struct ptlrpc_request *req)
247 {
248         return AT_OFF ? 0 : at_get(&req->rq_import->imp_at.iat_net_latency);
249 }
250
251 /* Adjust expected network latency */
252 static void ptlrpc_at_adj_net_latency(struct ptlrpc_request *req,
253                                       unsigned int service_time)
254 {
255         unsigned int nl, oldnl;
256         struct imp_at *at;
257         time_t now = cfs_time_current_sec();
258
259         LASSERT(req->rq_import);
260         at = &req->rq_import->imp_at;
261
262         /* Network latency is total time less server processing time */
263         nl = max_t(int, now - req->rq_sent - service_time, 0) +1/*st rounding*/;
264         if (service_time > now - req->rq_sent + 3 /* bz16408 */)
265                 CWARN("Reported service time %u > total measured time "
266                       CFS_DURATION_T"\n", service_time,
267                       cfs_time_sub(now, req->rq_sent));
268
269         oldnl = at_add(&at->iat_net_latency, nl);
270         if (oldnl != 0)
271                 CDEBUG(D_ADAPTTO, "The network latency for %s (nid %s) "
272                        "has changed from %d to %d\n",
273                        req->rq_import->imp_obd->obd_name,
274                        obd_uuid2str(
275                                &req->rq_import->imp_connection->c_remote_uuid),
276                        oldnl, at_get(&at->iat_net_latency));
277 }
278
279 static int unpack_reply(struct ptlrpc_request *req)
280 {
281         int rc;
282
283         /* Clear reply swab mask; we may have already swabbed an early reply */
284         req->rq_rep_swab_mask = 0;
285
286         rc = lustre_unpack_msg(req->rq_repmsg, req->rq_replen);
287         if (rc) {
288                 DEBUG_REQ(D_ERROR, req, "unpack_rep failed: %d", rc);
289                 return(-EPROTO);
290         }
291
292         rc = lustre_unpack_rep_ptlrpc_body(req, MSG_PTLRPC_BODY_OFF);
293         if (rc) {
294                 DEBUG_REQ(D_ERROR, req, "unpack ptlrpc body failed: %d", rc);
295                 return(-EPROTO);
296         }
297         return 0;
298 }
299
300 /*
301  * Handle an early reply message, called with the rq_lock held.
302  * If anything goes wrong just ignore it - same as if it never happened
303  */
304 static int ptlrpc_at_recv_early_reply(struct ptlrpc_request *req)
305 {
306         struct ptlrpc_request *early_req;
307         time_t                 olddl;
308         int                    rc;
309         ENTRY;
310
311         req->rq_early = 0;
312         spin_unlock(&req->rq_lock);
313
314         rc = sptlrpc_cli_unwrap_early_reply(req, &early_req);
315         if (rc) {
316                 spin_lock(&req->rq_lock);
317                 RETURN(rc);
318         }
319
320         rc = unpack_reply(early_req);
321         if (rc == 0) {
322                 /* Expecting to increase the service time estimate here */
323                 ptlrpc_at_adj_service(req,
324                         lustre_msg_get_timeout(early_req->rq_repmsg));
325                 ptlrpc_at_adj_net_latency(req,
326                         lustre_msg_get_service_time(early_req->rq_repmsg));
327         }
328
329         sptlrpc_cli_finish_early_reply(early_req);
330
331         spin_lock(&req->rq_lock);
332
333         if (rc == 0) {
334                 /* Adjust the local timeout for this req */
335                 ptlrpc_at_set_req_timeout(req);
336
337                 olddl = req->rq_deadline;
338                 /* server assumes it now has rq_timeout from when it sent the
339                    early reply, so client should give it at least that long. */
340                 req->rq_deadline = cfs_time_current_sec() + req->rq_timeout +
341                             ptlrpc_at_get_net_latency(req);
342
343                 DEBUG_REQ(D_ADAPTTO, req,
344                           "Early reply #%d, new deadline in "CFS_DURATION_T"s "
345                           "("CFS_DURATION_T"s)", req->rq_early_count,
346                           cfs_time_sub(req->rq_deadline,
347                                        cfs_time_current_sec()),
348                           cfs_time_sub(req->rq_deadline, olddl));
349         }
350
351         RETURN(rc);
352 }
353
354 void ptlrpc_free_rq_pool(struct ptlrpc_request_pool *pool)
355 {
356         struct list_head *l, *tmp;
357         struct ptlrpc_request *req;
358
359         LASSERT(pool != NULL);
360
361         spin_lock(&pool->prp_lock);
362         list_for_each_safe(l, tmp, &pool->prp_req_list) {
363                 req = list_entry(l, struct ptlrpc_request, rq_list);
364                 list_del(&req->rq_list);
365                 LASSERT(req->rq_reqbuf);
366                 LASSERT(req->rq_reqbuf_len == pool->prp_rq_size);
367                 OBD_FREE(req->rq_reqbuf, pool->prp_rq_size);
368                 OBD_FREE(req, sizeof(*req));
369         }
370         spin_unlock(&pool->prp_lock);
371         OBD_FREE(pool, sizeof(*pool));
372 }
373
374 void ptlrpc_add_rqs_to_pool(struct ptlrpc_request_pool *pool, int num_rq)
375 {
376         int i;
377         int size = 1;
378
379         while (size < pool->prp_rq_size + SPTLRPC_MAX_PAYLOAD)
380                 size <<= 1;
381
382         LASSERTF(list_empty(&pool->prp_req_list) || size == pool->prp_rq_size,
383                  "Trying to change pool size with nonempty pool "
384                  "from %d to %d bytes\n", pool->prp_rq_size, size);
385
386         spin_lock(&pool->prp_lock);
387         pool->prp_rq_size = size;
388         for (i = 0; i < num_rq; i++) {
389                 struct ptlrpc_request *req;
390                 struct lustre_msg *msg;
391
392                 spin_unlock(&pool->prp_lock);
393                 OBD_ALLOC(req, sizeof(struct ptlrpc_request));
394                 if (!req)
395                         return;
396                 OBD_ALLOC_GFP(msg, size, CFS_ALLOC_STD);
397                 if (!msg) {
398                         OBD_FREE(req, sizeof(struct ptlrpc_request));
399                         return;
400                 }
401                 req->rq_reqbuf = msg;
402                 req->rq_reqbuf_len = size;
403                 req->rq_pool = pool;
404                 spin_lock(&pool->prp_lock);
405                 list_add_tail(&req->rq_list, &pool->prp_req_list);
406         }
407         spin_unlock(&pool->prp_lock);
408         return;
409 }
410
411 struct ptlrpc_request_pool *
412 ptlrpc_init_rq_pool(int num_rq, int msgsize,
413                     void (*populate_pool)(struct ptlrpc_request_pool *, int))
414 {
415         struct ptlrpc_request_pool *pool;
416
417         OBD_ALLOC(pool, sizeof (struct ptlrpc_request_pool));
418         if (!pool)
419                 return NULL;
420
421         /* Request next power of two for the allocation, because internally
422            kernel would do exactly this */
423
424         spin_lock_init(&pool->prp_lock);
425         CFS_INIT_LIST_HEAD(&pool->prp_req_list);
426         pool->prp_rq_size = msgsize;
427         pool->prp_populate = populate_pool;
428
429         populate_pool(pool, num_rq);
430
431         if (list_empty(&pool->prp_req_list)) {
432                 /* have not allocated a single request for the pool */
433                 OBD_FREE(pool, sizeof (struct ptlrpc_request_pool));
434                 pool = NULL;
435         }
436         return pool;
437 }
438
439 static struct ptlrpc_request *
440 ptlrpc_prep_req_from_pool(struct ptlrpc_request_pool *pool)
441 {
442         struct ptlrpc_request *request;
443         struct lustre_msg *reqbuf;
444
445         if (!pool)
446                 return NULL;
447
448         spin_lock(&pool->prp_lock);
449
450         /* See if we have anything in a pool, and bail out if nothing,
451          * in writeout path, where this matters, this is safe to do, because
452          * nothing is lost in this case, and when some in-flight requests
453          * complete, this code will be called again. */
454         if (unlikely(list_empty(&pool->prp_req_list))) {
455                 spin_unlock(&pool->prp_lock);
456                 return NULL;
457         }
458
459         request = list_entry(pool->prp_req_list.next, struct ptlrpc_request,
460                              rq_list);
461         list_del_init(&request->rq_list);
462         spin_unlock(&pool->prp_lock);
463
464         LASSERT(request->rq_reqbuf);
465         LASSERT(request->rq_pool);
466
467         reqbuf = request->rq_reqbuf;
468         memset(request, 0, sizeof(*request));
469         request->rq_reqbuf = reqbuf;
470         request->rq_reqbuf_len = pool->prp_rq_size;
471         request->rq_pool = pool;
472
473         return request;
474 }
475
476 static void __ptlrpc_free_req_to_pool(struct ptlrpc_request *request)
477 {
478         struct ptlrpc_request_pool *pool = request->rq_pool;
479
480         spin_lock(&pool->prp_lock);
481         LASSERT(list_empty(&request->rq_list));
482         LASSERT(!request->rq_receiving_reply);
483         list_add_tail(&request->rq_list, &pool->prp_req_list);
484         spin_unlock(&pool->prp_lock);
485 }
486
487 static int __ptlrpc_request_bufs_pack(struct ptlrpc_request *request,
488                                       __u32 version, int opcode,
489                                       int count, __u32 *lengths, char **bufs,
490                                       struct ptlrpc_cli_ctx *ctx)
491 {
492         struct obd_import  *imp = request->rq_import;
493         int                 rc;
494         ENTRY;
495
496         if (unlikely(ctx))
497                 request->rq_cli_ctx = sptlrpc_cli_ctx_get(ctx);
498         else {
499                 rc = sptlrpc_req_get_ctx(request);
500                 if (rc)
501                         GOTO(out_free, rc);
502         }
503
504         sptlrpc_req_set_flavor(request, opcode);
505
506         rc = lustre_pack_request(request, imp->imp_msg_magic, count,
507                                  lengths, bufs);
508         if (rc) {
509                 LASSERT(!request->rq_pool);
510                 GOTO(out_ctx, rc);
511         }
512
513         lustre_msg_add_version(request->rq_reqmsg, version);
514         request->rq_send_state = LUSTRE_IMP_FULL;
515         request->rq_type = PTL_RPC_MSG_REQUEST;
516         request->rq_export = NULL;
517
518         request->rq_req_cbid.cbid_fn  = request_out_callback;
519         request->rq_req_cbid.cbid_arg = request;
520
521         request->rq_reply_cbid.cbid_fn  = reply_in_callback;
522         request->rq_reply_cbid.cbid_arg = request;
523
524         request->rq_reply_deadline = 0;
525         request->rq_phase = RQ_PHASE_NEW;
526         request->rq_next_phase = RQ_PHASE_UNDEFINED;
527
528         request->rq_request_portal = imp->imp_client->cli_request_portal;
529         request->rq_reply_portal = imp->imp_client->cli_reply_portal;
530
531         ptlrpc_at_set_req_timeout(request);
532
533         spin_lock_init(&request->rq_lock);
534         CFS_INIT_LIST_HEAD(&request->rq_list);
535         CFS_INIT_LIST_HEAD(&request->rq_timed_list);
536         CFS_INIT_LIST_HEAD(&request->rq_replay_list);
537         CFS_INIT_LIST_HEAD(&request->rq_mod_list);
538         CFS_INIT_LIST_HEAD(&request->rq_ctx_chain);
539         CFS_INIT_LIST_HEAD(&request->rq_set_chain);
540         CFS_INIT_LIST_HEAD(&request->rq_history_list);
541         CFS_INIT_LIST_HEAD(&request->rq_exp_list);
542         cfs_waitq_init(&request->rq_reply_waitq);
543         request->rq_xid = ptlrpc_next_xid();
544         atomic_set(&request->rq_refcount, 1);
545
546         lustre_msg_set_opc(request->rq_reqmsg, opcode);
547
548         RETURN(0);
549 out_ctx:
550         sptlrpc_cli_ctx_put(request->rq_cli_ctx, 1);
551 out_free:
552         class_import_put(imp);
553         return rc;
554 }
555
556 int ptlrpc_request_bufs_pack(struct ptlrpc_request *request,
557                              __u32 version, int opcode, char **bufs,
558                              struct ptlrpc_cli_ctx *ctx)
559 {
560         int count;
561
562         count = req_capsule_filled_sizes(&request->rq_pill, RCL_CLIENT);
563         return __ptlrpc_request_bufs_pack(request, version, opcode, count,
564                                           request->rq_pill.rc_area[RCL_CLIENT],
565                                           bufs, ctx);
566 }
567 EXPORT_SYMBOL(ptlrpc_request_bufs_pack);
568
569 int ptlrpc_request_pack(struct ptlrpc_request *request,
570                         __u32 version, int opcode)
571 {
572         return ptlrpc_request_bufs_pack(request, version, opcode, NULL, NULL);
573 }
574
575 static inline
576 struct ptlrpc_request *__ptlrpc_request_alloc(struct obd_import *imp,
577                                               struct ptlrpc_request_pool *pool)
578 {
579         struct ptlrpc_request *request = NULL;
580
581         if (pool)
582                 request = ptlrpc_prep_req_from_pool(pool);
583
584         if (!request)
585                 OBD_ALLOC_PTR(request);
586
587         if (request) {
588                 LASSERTF((unsigned long)imp > 0x1000, "%p", imp);
589                 LASSERT(imp != LP_POISON);
590                 LASSERTF((unsigned long)imp->imp_client > 0x1000, "%p",
591                         imp->imp_client);
592                 LASSERT(imp->imp_client != LP_POISON);
593
594                 request->rq_import = class_import_get(imp);
595         } else {
596                 CERROR("request allocation out of memory\n");
597         }
598
599         return request;
600 }
601
602 static struct ptlrpc_request *
603 ptlrpc_request_alloc_internal(struct obd_import *imp,
604                               struct ptlrpc_request_pool * pool,
605                               const struct req_format *format)
606 {
607         struct ptlrpc_request *request;
608
609         request = __ptlrpc_request_alloc(imp, pool);
610         if (request == NULL)
611                 return NULL;
612
613         req_capsule_init(&request->rq_pill, request, RCL_CLIENT);
614         req_capsule_set(&request->rq_pill, format);
615         return request;
616 }
617
618 struct ptlrpc_request *ptlrpc_request_alloc(struct obd_import *imp,
619                                             const struct req_format *format)
620 {
621         return ptlrpc_request_alloc_internal(imp, NULL, format);
622 }
623
624 struct ptlrpc_request *ptlrpc_request_alloc_pool(struct obd_import *imp,
625                                             struct ptlrpc_request_pool * pool,
626                                             const struct req_format *format)
627 {
628         return ptlrpc_request_alloc_internal(imp, pool, format);
629 }
630
631 void ptlrpc_request_free(struct ptlrpc_request *request)
632 {
633         if (request->rq_pool)
634                 __ptlrpc_free_req_to_pool(request);
635         else
636                 OBD_FREE_PTR(request);
637 }
638
639 struct ptlrpc_request *ptlrpc_request_alloc_pack(struct obd_import *imp,
640                                                 const struct req_format *format,
641                                                 __u32 version, int opcode)
642 {
643         struct ptlrpc_request *req = ptlrpc_request_alloc(imp, format);
644         int                    rc;
645
646         if (req) {
647                 rc = ptlrpc_request_pack(req, version, opcode);
648                 if (rc) {
649                         ptlrpc_request_free(req);
650                         req = NULL;
651                 }
652         }
653         return req;
654 }
655
656 struct ptlrpc_request *
657 ptlrpc_prep_req_pool(struct obd_import *imp,
658                      __u32 version, int opcode,
659                      int count, __u32 *lengths, char **bufs,
660                      struct ptlrpc_request_pool *pool)
661 {
662         struct ptlrpc_request *request;
663         int                    rc;
664
665         request = __ptlrpc_request_alloc(imp, pool);
666         if (!request)
667                 return NULL;
668
669         rc = __ptlrpc_request_bufs_pack(request, version, opcode, count,
670                                         lengths, bufs, NULL);
671         if (rc) {
672                 ptlrpc_request_free(request);
673                 request = NULL;
674         }
675         return request;
676 }
677
678 struct ptlrpc_request *
679 ptlrpc_prep_req(struct obd_import *imp, __u32 version, int opcode, int count,
680                 __u32 *lengths, char **bufs)
681 {
682         return ptlrpc_prep_req_pool(imp, version, opcode, count, lengths, bufs,
683                                     NULL);
684 }
685
686 struct ptlrpc_request_set *ptlrpc_prep_set(void)
687 {
688         struct ptlrpc_request_set *set;
689
690         ENTRY;
691         OBD_ALLOC(set, sizeof *set);
692         if (!set)
693                 RETURN(NULL);
694         CFS_INIT_LIST_HEAD(&set->set_requests);
695         cfs_waitq_init(&set->set_waitq);
696         set->set_remaining = 0;
697         spin_lock_init(&set->set_new_req_lock);
698         CFS_INIT_LIST_HEAD(&set->set_new_requests);
699         CFS_INIT_LIST_HEAD(&set->set_cblist);
700
701         RETURN(set);
702 }
703
704 /* Finish with this set; opposite of prep_set. */
705 void ptlrpc_set_destroy(struct ptlrpc_request_set *set)
706 {
707         struct list_head *tmp;
708         struct list_head *next;
709         int               expected_phase;
710         int               n = 0;
711         ENTRY;
712
713         /* Requests on the set should either all be completed, or all be new */
714         expected_phase = (set->set_remaining == 0) ?
715                          RQ_PHASE_COMPLETE : RQ_PHASE_NEW;
716         list_for_each (tmp, &set->set_requests) {
717                 struct ptlrpc_request *req =
718                         list_entry(tmp, struct ptlrpc_request, rq_set_chain);
719
720                 LASSERT(req->rq_phase == expected_phase);
721                 n++;
722         }
723
724         LASSERT(set->set_remaining == 0 || set->set_remaining == n);
725
726         list_for_each_safe(tmp, next, &set->set_requests) {
727                 struct ptlrpc_request *req =
728                         list_entry(tmp, struct ptlrpc_request, rq_set_chain);
729                 list_del_init(&req->rq_set_chain);
730
731                 LASSERT(req->rq_phase == expected_phase);
732
733                 if (req->rq_phase == RQ_PHASE_NEW) {
734
735                         if (req->rq_interpret_reply != NULL) {
736                                 ptlrpc_interpterer_t interpreter =
737                                         req->rq_interpret_reply;
738
739                                 /* higher level (i.e. LOV) failed;
740                                  * let the sub reqs clean up */
741                                 req->rq_status = -EBADR;
742                                 interpreter(NULL, req, &req->rq_async_args,
743                                             req->rq_status);
744                         }
745                         set->set_remaining--;
746                 }
747
748                 req->rq_set = NULL;
749                 ptlrpc_req_finished (req);
750         }
751
752         LASSERT(set->set_remaining == 0);
753
754         OBD_FREE(set, sizeof(*set));
755         EXIT;
756 }
757
758 int ptlrpc_set_add_cb(struct ptlrpc_request_set *set,
759                       set_interpreter_func fn, void *data)
760 {
761         struct ptlrpc_set_cbdata *cbdata;
762
763         OBD_ALLOC_PTR(cbdata);
764         if (cbdata == NULL)
765                 RETURN(-ENOMEM);
766
767         cbdata->psc_interpret = fn;
768         cbdata->psc_data = data;
769         list_add_tail(&cbdata->psc_item, &set->set_cblist);
770
771         RETURN(0);
772 }
773
774 void ptlrpc_set_add_req(struct ptlrpc_request_set *set,
775                         struct ptlrpc_request *req)
776 {
777         /* The set takes over the caller's request reference */
778         list_add_tail(&req->rq_set_chain, &set->set_requests);
779         req->rq_set = set;
780         set->set_remaining++;
781 }
782
783 /**
784  * Lock so many callers can add things, the context that owns the set
785  * is supposed to notice these and move them into the set proper.
786  */
787 int ptlrpc_set_add_new_req(struct ptlrpcd_ctl *pc,
788                            struct ptlrpc_request *req)
789 {
790         struct ptlrpc_request_set *set = pc->pc_set;
791
792         /*
793          * Let caller know that we stopped and will not handle this request.
794          * It needs to take care itself of request.
795          */
796         if (test_bit(LIOD_STOP, &pc->pc_flags))
797                 return -EALREADY;
798
799         spin_lock(&set->set_new_req_lock);
800         /*
801          * The set takes over the caller's request reference.
802          */
803         list_add_tail(&req->rq_set_chain, &set->set_new_requests);
804         req->rq_set = set;
805         spin_unlock(&set->set_new_req_lock);
806
807         /*
808          * Let thead know that we added something and better it to wake up
809          * and process.
810          */
811         cfs_waitq_signal(&set->set_waitq);
812         return 0;
813 }
814
815 /*
816  * Based on the current state of the import, determine if the request
817  * can be sent, is an error, or should be delayed.
818  *
819  * Returns true if this request should be delayed. If false, and
820  * *status is set, then the request can not be sent and *status is the
821  * error code.  If false and status is 0, then request can be sent.
822  *
823  * The imp->imp_lock must be held.
824  */
825 static int ptlrpc_import_delay_req(struct obd_import *imp,
826                                    struct ptlrpc_request *req, int *status)
827 {
828         int delay = 0;
829         ENTRY;
830
831         LASSERT (status != NULL);
832         *status = 0;
833
834         if (req->rq_ctx_init || req->rq_ctx_fini) {
835                 /* always allow ctx init/fini rpc go through */
836         } else if (imp->imp_state == LUSTRE_IMP_NEW) {
837                 DEBUG_REQ(D_ERROR, req, "Uninitialized import.");
838                 *status = -EIO;
839                 LBUG();
840         } else if (imp->imp_state == LUSTRE_IMP_CLOSED) {
841                 DEBUG_REQ(D_ERROR, req, "IMP_CLOSED ");
842                 *status = -EIO;
843         } else if (req->rq_send_state == LUSTRE_IMP_CONNECTING &&
844                    imp->imp_state == LUSTRE_IMP_CONNECTING) {
845                 /* allow CONNECT even if import is invalid */ ;
846                 if (atomic_read(&imp->imp_inval_count) != 0) {
847                         DEBUG_REQ(D_ERROR, req, "invalidate in flight");
848                         *status = -EIO;
849                 }
850         } else if ((imp->imp_invalid && (!imp->imp_recon_bk)) ||
851                                          imp->imp_obd->obd_no_recov) {
852                 /* If the import has been invalidated (such as by an OST
853                  * failure), and if the import(MGC) tried all of its connection
854                  * list (Bug 13464), the request must fail with -ESHUTDOWN.
855                  * This indicates the requests should be discarded; an -EIO
856                  * may result in a resend of the request. */
857                 if (!imp->imp_deactive)
858                           DEBUG_REQ(D_ERROR, req, "IMP_INVALID");
859                 *status = -ESHUTDOWN; /* bz 12940 */
860         } else if (req->rq_import_generation != imp->imp_generation) {
861                 DEBUG_REQ(D_ERROR, req, "req wrong generation:");
862                 *status = -EIO;
863         } else if (req->rq_send_state != imp->imp_state) {
864                 /* invalidate in progress - any requests should be drop */
865                 if (atomic_read(&imp->imp_inval_count) != 0) {
866                         DEBUG_REQ(D_ERROR, req, "invalidate in flight");
867                         *status = -EIO;
868                 } else if (imp->imp_dlm_fake || req->rq_no_delay) {
869                         *status = -EWOULDBLOCK;
870                 } else {
871                         delay = 1;
872                 }
873         }
874
875         RETURN(delay);
876 }
877
878 static int ptlrpc_check_reply(struct ptlrpc_request *req)
879 {
880         int rc = 0;
881         ENTRY;
882
883         /* serialise with network callback */
884         spin_lock(&req->rq_lock);
885
886         if (ptlrpc_client_replied(req))
887                 GOTO(out, rc = 1);
888
889         if (req->rq_net_err && !req->rq_timedout) {
890                 spin_unlock(&req->rq_lock);
891                 rc = ptlrpc_expire_one_request(req, 0);
892                 spin_lock(&req->rq_lock);
893                 GOTO(out, rc);
894         }
895
896         if (req->rq_err)
897                 GOTO(out, rc = 1);
898
899         if (req->rq_resend)
900                 GOTO(out, rc = 1);
901
902         if (req->rq_restart)
903                 GOTO(out, rc = 1);
904
905         if (ptlrpc_client_early(req)) {
906                 ptlrpc_at_recv_early_reply(req);
907                 GOTO(out, rc = 0); /* keep waiting */
908         }
909
910         EXIT;
911  out:
912         spin_unlock(&req->rq_lock);
913         DEBUG_REQ(D_NET, req, "rc = %d for", rc);
914         return rc;
915 }
916
917 static int ptlrpc_check_status(struct ptlrpc_request *req)
918 {
919         int err;
920         ENTRY;
921
922         err = lustre_msg_get_status(req->rq_repmsg);
923         if (lustre_msg_get_type(req->rq_repmsg) == PTL_RPC_MSG_ERR) {
924                 struct obd_import *imp = req->rq_import;
925                 __u32 opc = lustre_msg_get_opc(req->rq_reqmsg);
926                 LCONSOLE_ERROR_MSG(0x011,"an error occurred while communicating"
927                                 " with %s. The %s operation failed with %d\n",
928                                 libcfs_nid2str(imp->imp_connection->c_peer.nid),
929                                 ll_opcode2str(opc), err);
930                 RETURN(err < 0 ? err : -EINVAL);
931         }
932
933         if (err < 0) {
934                 DEBUG_REQ(D_INFO, req, "status is %d", err);
935         } else if (err > 0) {
936                 /* XXX: translate this error from net to host */
937                 DEBUG_REQ(D_INFO, req, "status is %d", err);
938         }
939
940         RETURN(err);
941 }
942
943 /**
944  * Callback function called when client receives RPC reply for \a req.
945  */
946 static int after_reply(struct ptlrpc_request *req)
947 {
948         struct obd_import *imp = req->rq_import;
949         struct obd_device *obd = req->rq_import->imp_obd;
950         int rc;
951         struct timeval work_start;
952         long timediff;
953         ENTRY;
954
955         LASSERT(!req->rq_receiving_reply);
956         LASSERT(obd);
957         LASSERT(req->rq_nob_received <= req->rq_repbuf_len);
958
959         /*
960          * NB Until this point, the whole of the incoming message,
961          * including buflens, status etc is in the sender's byte order.
962          */
963
964         rc = sptlrpc_cli_unwrap_reply(req);
965         if (rc) {
966                 DEBUG_REQ(D_ERROR, req, "unwrap reply failed (%d):", rc);
967                 RETURN(rc);
968         }
969
970         /*
971          * Security layer unwrap might ask resend this request.
972          */
973         if (req->rq_resend)
974                 RETURN(0);
975
976         rc = unpack_reply(req);
977         if (rc)
978                 RETURN(rc);
979
980         do_gettimeofday(&work_start);
981         timediff = cfs_timeval_sub(&work_start, &req->rq_arrival_time, NULL);
982         if (obd->obd_svc_stats != NULL)
983                 lprocfs_counter_add(obd->obd_svc_stats, PTLRPC_REQWAIT_CNTR,
984                                     timediff);
985
986         if (lustre_msg_get_type(req->rq_repmsg) != PTL_RPC_MSG_REPLY &&
987             lustre_msg_get_type(req->rq_repmsg) != PTL_RPC_MSG_ERR) {
988                 DEBUG_REQ(D_ERROR, req, "invalid packet received (type=%u)",
989                           lustre_msg_get_type(req->rq_repmsg));
990                 RETURN(-EPROTO);
991         }
992
993         OBD_FAIL_TIMEOUT(OBD_FAIL_PTLRPC_PAUSE_REP, obd_fail_val);
994         ptlrpc_at_adj_service(req, lustre_msg_get_timeout(req->rq_repmsg));
995         ptlrpc_at_adj_net_latency(req,
996                                   lustre_msg_get_service_time(req->rq_repmsg));
997
998         rc = ptlrpc_check_status(req);
999         imp->imp_connect_error = rc;
1000
1001         if (rc) {
1002                 /*
1003                  * Either we've been evicted, or the server has failed for
1004                  * some reason. Try to reconnect, and if that fails, punt to
1005                  * the upcall.
1006                  */
1007                 if (ll_rpc_recoverable_error(rc)) {
1008                         if (req->rq_send_state != LUSTRE_IMP_FULL ||
1009                             imp->imp_obd->obd_no_recov || imp->imp_dlm_fake) {
1010                                 RETURN(rc);
1011                         }
1012                         ptlrpc_request_handle_notconn(req);
1013                         RETURN(rc);
1014                 }
1015         } else {
1016                 /*
1017                  * Let's look if server sent slv. Do it only for RPC with
1018                  * rc == 0.
1019                  */
1020                 ldlm_cli_update_pool(req);
1021         }
1022
1023         /*
1024          * Store transno in reqmsg for replay.
1025          */
1026         req->rq_transno = lustre_msg_get_transno(req->rq_repmsg);
1027         lustre_msg_set_transno(req->rq_reqmsg, req->rq_transno);
1028
1029         if (req->rq_import->imp_replayable) {
1030                 spin_lock(&imp->imp_lock);
1031                 /*
1032                  * No point in adding already-committed requests to the replay
1033                  * list, we will just remove them immediately. b=9829
1034                  */
1035                 if (req->rq_transno != 0 &&
1036                     (req->rq_transno >
1037                      lustre_msg_get_last_committed(req->rq_repmsg) ||
1038                      req->rq_replay))
1039                         ptlrpc_retain_replayable_request(req, imp);
1040                 else if (req->rq_commit_cb != NULL) {
1041                         spin_unlock(&imp->imp_lock);
1042                         req->rq_commit_cb(req);
1043                         spin_lock(&imp->imp_lock);
1044                 }
1045
1046                 /*
1047                  * Replay-enabled imports return commit-status information.
1048                  */
1049                 if (lustre_msg_get_last_committed(req->rq_repmsg)) {
1050                         imp->imp_peer_committed_transno =
1051                                 lustre_msg_get_last_committed(req->rq_repmsg);
1052                 }
1053                 ptlrpc_free_committed(imp);
1054                 spin_unlock(&imp->imp_lock);
1055         }
1056
1057         RETURN(rc);
1058 }
1059
1060 static int ptlrpc_send_new_req(struct ptlrpc_request *req)
1061 {
1062         struct obd_import     *imp;
1063         int rc;
1064         ENTRY;
1065
1066         LASSERT(req->rq_phase == RQ_PHASE_NEW);
1067         if (req->rq_sent && (req->rq_sent > cfs_time_current_sec()))
1068                 RETURN (0);
1069
1070         ptlrpc_rqphase_move(req, RQ_PHASE_RPC);
1071
1072         imp = req->rq_import;
1073         spin_lock(&imp->imp_lock);
1074
1075         req->rq_import_generation = imp->imp_generation;
1076
1077         if (ptlrpc_import_delay_req(imp, req, &rc)) {
1078                 spin_lock(&req->rq_lock);
1079                 req->rq_waiting = 1;
1080                 spin_unlock(&req->rq_lock);
1081
1082                 DEBUG_REQ(D_HA, req, "req from PID %d waiting for recovery: "
1083                           "(%s != %s)", lustre_msg_get_status(req->rq_reqmsg),
1084                           ptlrpc_import_state_name(req->rq_send_state),
1085                           ptlrpc_import_state_name(imp->imp_state));
1086                 LASSERT(list_empty(&req->rq_list));
1087                 list_add_tail(&req->rq_list, &imp->imp_delayed_list);
1088                 atomic_inc(&req->rq_import->imp_inflight);
1089                 spin_unlock(&imp->imp_lock);
1090                 RETURN(0);
1091         }
1092
1093         if (rc != 0) {
1094                 spin_unlock(&imp->imp_lock);
1095                 req->rq_status = rc;
1096                 ptlrpc_rqphase_move(req, RQ_PHASE_INTERPRET);
1097                 RETURN(rc);
1098         }
1099
1100         LASSERT(list_empty(&req->rq_list));
1101         list_add_tail(&req->rq_list, &imp->imp_sending_list);
1102         atomic_inc(&req->rq_import->imp_inflight);
1103         spin_unlock(&imp->imp_lock);
1104
1105         lustre_msg_set_status(req->rq_reqmsg, cfs_curproc_pid());
1106
1107         rc = sptlrpc_req_refresh_ctx(req, -1);
1108         if (rc) {
1109                 if (req->rq_err) {
1110                         req->rq_status = rc;
1111                         RETURN(1);
1112                 } else {
1113                         req->rq_wait_ctx = 1;
1114                         RETURN(0);
1115                 }
1116         }
1117
1118         CDEBUG(D_RPCTRACE, "Sending RPC pname:cluuid:pid:xid:nid:opc"
1119                " %s:%s:%d:"LPU64":%s:%d\n", cfs_curproc_comm(),
1120                imp->imp_obd->obd_uuid.uuid,
1121                lustre_msg_get_status(req->rq_reqmsg), req->rq_xid,
1122                libcfs_nid2str(imp->imp_connection->c_peer.nid),
1123                lustre_msg_get_opc(req->rq_reqmsg));
1124
1125         rc = ptl_send_rpc(req, 0);
1126         if (rc) {
1127                 DEBUG_REQ(D_HA, req, "send failed (%d); expect timeout", rc);
1128                 req->rq_net_err = 1;
1129                 RETURN(rc);
1130         }
1131         RETURN(0);
1132 }
1133
1134 /* this sends any unsent RPCs in @set and returns TRUE if all are sent */
1135 int ptlrpc_check_set(const struct lu_env *env, struct ptlrpc_request_set *set)
1136 {
1137         struct list_head *tmp;
1138         int force_timer_recalc = 0;
1139         ENTRY;
1140
1141         if (set->set_remaining == 0)
1142                 RETURN(1);
1143
1144         list_for_each(tmp, &set->set_requests) {
1145                 struct ptlrpc_request *req =
1146                         list_entry(tmp, struct ptlrpc_request, rq_set_chain);
1147                 struct obd_import *imp = req->rq_import;
1148                 int rc = 0;
1149
1150                 if (req->rq_phase == RQ_PHASE_NEW &&
1151                     ptlrpc_send_new_req(req)) {
1152                         force_timer_recalc = 1;
1153                 }
1154
1155                 /* delayed send - skip */
1156                 if (req->rq_phase == RQ_PHASE_NEW && req->rq_sent)
1157                         continue;
1158
1159                 if (!(req->rq_phase == RQ_PHASE_RPC ||
1160                       req->rq_phase == RQ_PHASE_BULK ||
1161                       req->rq_phase == RQ_PHASE_INTERPRET ||
1162                       req->rq_phase == RQ_PHASE_UNREGISTERING ||
1163                       req->rq_phase == RQ_PHASE_COMPLETE)) {
1164                         DEBUG_REQ(D_ERROR, req, "bad phase %x", req->rq_phase);
1165                         LBUG();
1166                 }
1167
1168                 if (req->rq_phase == RQ_PHASE_UNREGISTERING) {
1169                         LASSERT(req->rq_next_phase != req->rq_phase);
1170                         LASSERT(req->rq_next_phase != RQ_PHASE_UNDEFINED);
1171
1172                         /* 
1173                          * Skip processing until reply is unlinked. We
1174                          * can't return to pool before that and we can't
1175                          * call interpret before that. We need to make
1176                          * sure that all rdma transfers finished and will
1177                          * not corrupt any data. 
1178                          */
1179                         if (ptlrpc_client_recv_or_unlink(req) ||
1180                             ptlrpc_client_bulk_active(req))
1181                                 continue;
1182                      
1183                         /* 
1184                          * Turn fail_loc off to prevent it from looping
1185                          * forever. 
1186                          */
1187                         if (OBD_FAIL_CHECK(OBD_FAIL_PTLRPC_LONG_REPL_UNLINK)) {
1188                                 OBD_FAIL_CHECK_ORSET(OBD_FAIL_PTLRPC_LONG_REPL_UNLINK,
1189                                                      OBD_FAIL_ONCE);
1190                         }
1191                         if (OBD_FAIL_CHECK(OBD_FAIL_PTLRPC_LONG_BULK_UNLINK)) {
1192                                 OBD_FAIL_CHECK_ORSET(OBD_FAIL_PTLRPC_LONG_BULK_UNLINK,
1193                                                      OBD_FAIL_ONCE);
1194                         }
1195
1196                         /* 
1197                          * Move to next phase if reply was successfully 
1198                          * unlinked. 
1199                          */
1200                         ptlrpc_rqphase_move(req, req->rq_next_phase);
1201                 }
1202
1203                 if (req->rq_phase == RQ_PHASE_COMPLETE)
1204                         continue;
1205
1206                 if (req->rq_phase == RQ_PHASE_INTERPRET)
1207                         GOTO(interpret, req->rq_status);
1208
1209                 /* 
1210                  * Note that this also will start async reply unlink. 
1211                  */
1212                 if (req->rq_net_err && !req->rq_timedout) {
1213                         ptlrpc_expire_one_request(req, 1);
1214
1215                         /* 
1216                          * Check if we still need to wait for unlink. 
1217                          */
1218                         if (ptlrpc_client_recv_or_unlink(req) ||
1219                             ptlrpc_client_bulk_active(req))
1220                                 continue;
1221                 }
1222
1223                 if (req->rq_err) {
1224                         req->rq_replied = 0;
1225                         if (req->rq_status == 0)
1226                                 req->rq_status = -EIO;
1227                         ptlrpc_rqphase_move(req, RQ_PHASE_INTERPRET);
1228                         GOTO(interpret, req->rq_status);
1229                 }
1230
1231                 /* ptlrpc_queue_wait->l_wait_event guarantees that rq_intr
1232                  * will only be set after rq_timedout, but the synchronous IO
1233                  * waiting path sets rq_intr irrespective of whether ptlrpcd
1234                  * has seen a timeout.  our policy is to only interpret
1235                  * interrupted rpcs after they have timed out */
1236                 if (req->rq_intr && (req->rq_timedout || req->rq_waiting ||
1237                                      req->rq_wait_ctx)) {
1238                         req->rq_status = -EINTR;
1239                         ptlrpc_rqphase_move(req, RQ_PHASE_INTERPRET);
1240                         GOTO(interpret, req->rq_status);
1241                 }
1242
1243                 if (req->rq_phase == RQ_PHASE_RPC) {
1244                         if (req->rq_timedout || req->rq_resend ||
1245                             req->rq_waiting || req->rq_wait_ctx) {
1246                                 int status;
1247
1248                                 if (!ptlrpc_unregister_reply(req, 1))
1249                                         continue;
1250
1251                                 spin_lock(&imp->imp_lock);
1252
1253                                 if (ptlrpc_import_delay_req(imp, req, &status)){
1254                                         spin_unlock(&imp->imp_lock);
1255                                         continue;
1256                                 }
1257
1258                                 if (status != 0)  {
1259                                         req->rq_status = status;
1260                                         ptlrpc_rqphase_move(req, 
1261                                                 RQ_PHASE_INTERPRET);
1262                                         spin_unlock(&imp->imp_lock);
1263                                         GOTO(interpret, req->rq_status);
1264                                 }
1265                                 if (req->rq_no_resend && !req->rq_wait_ctx) {
1266                                         req->rq_status = -ENOTCONN;
1267                                         ptlrpc_rqphase_move(req, 
1268                                                 RQ_PHASE_INTERPRET);
1269                                         spin_unlock(&imp->imp_lock);
1270                                         GOTO(interpret, req->rq_status);
1271                                 }
1272
1273                                 list_del_init(&req->rq_list);
1274                                 list_add_tail(&req->rq_list,
1275                                               &imp->imp_sending_list);
1276
1277                                 spin_unlock(&imp->imp_lock);
1278
1279                                 req->rq_waiting = 0;
1280
1281                                 if (req->rq_timedout||req->rq_resend) {
1282                                         /* This is re-sending anyways, 
1283                                          * let's mark req as resend. */
1284                                         req->rq_resend = 1;
1285                                         lustre_msg_add_flags(req->rq_reqmsg,
1286                                                              MSG_RESENT);
1287                                         if (req->rq_bulk) {
1288                                                 __u64 old_xid;
1289
1290                                                 if (!ptlrpc_unregister_bulk(req, 1))
1291                                                         continue;
1292
1293                                                 /* ensure previous bulk fails */
1294                                                 old_xid = req->rq_xid;
1295                                                 req->rq_xid = ptlrpc_next_xid();
1296                                                 CDEBUG(D_HA, "resend bulk "
1297                                                        "old x"LPU64
1298                                                        " new x"LPU64"\n",
1299                                                        old_xid, req->rq_xid);
1300                                         }
1301                                 }
1302                                 /*
1303                                  * rq_wait_ctx is only touched by ptlrpcd,
1304                                  * so no lock is needed here.
1305                                  */
1306                                 status = sptlrpc_req_refresh_ctx(req, -1);
1307                                 if (status) {
1308                                         if (req->rq_err) {
1309                                                 req->rq_status = status;
1310                                                 force_timer_recalc = 1;
1311                                         } else {
1312                                                 req->rq_wait_ctx = 1;
1313                                         }
1314
1315                                         continue;
1316                                 } else {
1317                                         req->rq_wait_ctx = 0;
1318                                 }
1319
1320                                 rc = ptl_send_rpc(req, 0);
1321                                 if (rc) {
1322                                         DEBUG_REQ(D_HA, req, "send failed (%d)",
1323                                                   rc);
1324                                         force_timer_recalc = 1;
1325                                         req->rq_net_err = 1;
1326                                 }
1327                                 /* need to reset the timeout */
1328                                 force_timer_recalc = 1;
1329                         }
1330
1331                         spin_lock(&req->rq_lock);
1332
1333                         if (ptlrpc_client_early(req)) {
1334                                 ptlrpc_at_recv_early_reply(req);
1335                                 spin_unlock(&req->rq_lock);
1336                                 continue;
1337                         }
1338
1339                         /* Still waiting for a reply? */
1340                         if (ptlrpc_client_recv(req)) {
1341                                 spin_unlock(&req->rq_lock);
1342                                 continue;
1343                         }
1344
1345                         /* Did we actually receive a reply? */
1346                         if (!ptlrpc_client_replied(req)) {
1347                                 spin_unlock(&req->rq_lock);
1348                                 continue;
1349                         }
1350
1351                         spin_unlock(&req->rq_lock);
1352
1353                         req->rq_status = after_reply(req);
1354                         if (req->rq_resend) {
1355                                 /* Add this req to the delayed list so
1356                                    it can be errored if the import is
1357                                    evicted after recovery. */
1358                                 spin_lock(&imp->imp_lock);
1359                                 list_del_init(&req->rq_list);
1360                                 list_add_tail(&req->rq_list,
1361                                               &imp->imp_delayed_list);
1362                                 spin_unlock(&imp->imp_lock);
1363                                 continue;
1364                         }
1365
1366                         /* If there is no bulk associated with this request,
1367                          * then we're done and should let the interpreter
1368                          * process the reply. Similarly if the RPC returned
1369                          * an error, and therefore the bulk will never arrive.
1370                          */
1371                         if (req->rq_bulk == NULL || req->rq_status != 0) {
1372                                 ptlrpc_rqphase_move(req, RQ_PHASE_INTERPRET);
1373                                 GOTO(interpret, req->rq_status);
1374                         }
1375
1376                         ptlrpc_rqphase_move(req, RQ_PHASE_BULK);
1377                 }
1378
1379                 LASSERT(req->rq_phase == RQ_PHASE_BULK);
1380                 if (ptlrpc_client_bulk_active(req))
1381                         continue;
1382
1383                 if (!req->rq_bulk->bd_success) {
1384                         /* The RPC reply arrived OK, but the bulk screwed
1385                          * up!  Dead wierd since the server told us the RPC
1386                          * was good after getting the REPLY for her GET or
1387                          * the ACK for her PUT. */
1388                         DEBUG_REQ(D_ERROR, req, "bulk transfer failed");
1389                         LBUG();
1390                 }
1391
1392                 ptlrpc_rqphase_move(req, RQ_PHASE_INTERPRET);
1393
1394         interpret:
1395                 LASSERT(req->rq_phase == RQ_PHASE_INTERPRET);
1396
1397                 /* This moves to "unregistering" phase we need to wait for
1398                  * reply unlink. */
1399                 if (!ptlrpc_unregister_reply(req, 1))
1400                         continue;
1401
1402                 if (!ptlrpc_unregister_bulk(req, 1))
1403                         continue;
1404
1405                 /* When calling interpret receiving already should be
1406                  * finished. */
1407                 LASSERT(!req->rq_receiving_reply);
1408
1409                 if (req->rq_interpret_reply != NULL) {
1410                         ptlrpc_interpterer_t interpreter =
1411                                 req->rq_interpret_reply;
1412                         req->rq_status = interpreter(env, req,
1413                                                      &req->rq_async_args,
1414                                                      req->rq_status);
1415                 }
1416                 ptlrpc_rqphase_move(req, RQ_PHASE_COMPLETE);
1417
1418                 CDEBUG(D_RPCTRACE, "Completed RPC pname:cluuid:pid:xid:nid:"
1419                        "opc %s:%s:%d:"LPU64":%s:%d\n", cfs_curproc_comm(),
1420                        imp->imp_obd->obd_uuid.uuid,
1421                        lustre_msg_get_status(req->rq_reqmsg), req->rq_xid,
1422                        libcfs_nid2str(imp->imp_connection->c_peer.nid),
1423                        lustre_msg_get_opc(req->rq_reqmsg));
1424
1425                 spin_lock(&imp->imp_lock);
1426                 /* Request already may be not on sending or delaying list. This
1427                  * may happen in the case of marking it errorneous for the case
1428                  * ptlrpc_import_delay_req(req, status) find it impossible to 
1429                  * allow sending this rpc and returns *status != 0. */
1430                 if (!list_empty(&req->rq_list)) {
1431                         list_del_init(&req->rq_list);
1432                         atomic_dec(&imp->imp_inflight);
1433                 }
1434                 spin_unlock(&imp->imp_lock);
1435
1436                 set->set_remaining--;
1437                 cfs_waitq_signal(&imp->imp_recovery_waitq);
1438         }
1439
1440         /* If we hit an error, we want to recover promptly. */
1441         RETURN(set->set_remaining == 0 || force_timer_recalc);
1442 }
1443
1444 /* Return 1 if we should give up, else 0 */
1445 int ptlrpc_expire_one_request(struct ptlrpc_request *req, int async_unlink)
1446 {
1447         struct obd_import *imp = req->rq_import;
1448         int rc = 0;
1449         ENTRY;
1450
1451         DEBUG_REQ(D_ERROR|D_NETERROR, req,
1452                   "%s (sent at "CFS_TIME_T", "CFS_DURATION_T"s ago)",
1453                   req->rq_net_err ? "network error" : "timeout",
1454                   req->rq_sent, cfs_time_sub(cfs_time_current_sec(),
1455                   req->rq_sent));
1456
1457         if (imp) {
1458                 LCONSOLE_WARN("Request x"LPU64" sent from %s to NID %s "
1459                               CFS_DURATION_T"s ago has timed out "
1460                               "(limit "CFS_DURATION_T"s).\n", req->rq_xid,
1461                               req->rq_import->imp_obd->obd_name,
1462                               libcfs_nid2str(imp->imp_connection->c_peer.nid),
1463                               cfs_time_sub(cfs_time_current_sec(), req->rq_sent),
1464                               cfs_time_sub(req->rq_deadline, req->rq_sent));
1465         }
1466
1467         if (imp != NULL && obd_debug_peer_on_timeout)
1468                 LNetCtl(IOC_LIBCFS_DEBUG_PEER, &imp->imp_connection->c_peer);
1469
1470         spin_lock(&req->rq_lock);
1471         req->rq_timedout = 1;
1472         spin_unlock(&req->rq_lock);
1473
1474         ptlrpc_unregister_reply(req, async_unlink);
1475         ptlrpc_unregister_bulk(req, async_unlink);
1476
1477         if (obd_dump_on_timeout)
1478                 libcfs_debug_dumplog();
1479
1480         if (imp == NULL) {
1481                 DEBUG_REQ(D_HA, req, "NULL import: already cleaned up?");
1482                 RETURN(1);
1483         }
1484
1485         /* The DLM server doesn't want recovery run on its imports. */
1486         if (imp->imp_dlm_fake)
1487                 RETURN(1);
1488
1489         /* If this request is for recovery or other primordial tasks,
1490          * then error it out here. */
1491         if (req->rq_ctx_init || req->rq_ctx_fini ||
1492             req->rq_send_state != LUSTRE_IMP_FULL ||
1493             imp->imp_obd->obd_no_recov) {
1494                 DEBUG_REQ(D_RPCTRACE, req, "err -110, sent_state=%s (now=%s)",
1495                           ptlrpc_import_state_name(req->rq_send_state),
1496                           ptlrpc_import_state_name(imp->imp_state));
1497                 spin_lock(&req->rq_lock);
1498                 req->rq_status = -ETIMEDOUT;
1499                 req->rq_err = 1;
1500                 spin_unlock(&req->rq_lock);
1501                 RETURN(1);
1502         }
1503
1504         /* if a request can't be resent we can't wait for an answer after
1505            the timeout */
1506         if (req->rq_no_resend) {
1507                 DEBUG_REQ(D_RPCTRACE, req, "TIMEOUT-NORESEND:");
1508                 rc = 1;
1509         }
1510
1511         ptlrpc_fail_import(imp, lustre_msg_get_conn_cnt(req->rq_reqmsg));
1512
1513         RETURN(rc);
1514 }
1515
1516 int ptlrpc_expired_set(void *data)
1517 {
1518         struct ptlrpc_request_set *set = data;
1519         struct list_head          *tmp;
1520         time_t                     now = cfs_time_current_sec();
1521         ENTRY;
1522
1523         LASSERT(set != NULL);
1524
1525         /* 
1526          * A timeout expired. See which reqs it applies to... 
1527          */
1528         list_for_each (tmp, &set->set_requests) {
1529                 struct ptlrpc_request *req =
1530                         list_entry(tmp, struct ptlrpc_request, rq_set_chain);
1531
1532                 /* Request in-flight? */
1533                 if (!((req->rq_phase == RQ_PHASE_RPC &&
1534                        !req->rq_waiting && !req->rq_resend) ||
1535                       (req->rq_phase == RQ_PHASE_BULK)))
1536                         continue;
1537   
1538                 if (req->rq_timedout ||     /* already dealt with */
1539                     req->rq_deadline > now) /* not expired */
1540                         continue;
1541
1542                 /* Deal with this guy. Do it asynchronously to not block
1543                  * ptlrpcd thread. */
1544                 ptlrpc_expire_one_request(req, 1);
1545         }
1546
1547         /* 
1548          * When waiting for a whole set, we always to break out of the
1549          * sleep so we can recalculate the timeout, or enable interrupts
1550          * if everyone's timed out.
1551          */
1552         RETURN(1);
1553 }
1554
1555 void ptlrpc_mark_interrupted(struct ptlrpc_request *req)
1556 {
1557         spin_lock(&req->rq_lock);
1558         req->rq_intr = 1;
1559         spin_unlock(&req->rq_lock);
1560 }
1561
1562 void ptlrpc_interrupted_set(void *data)
1563 {
1564         struct ptlrpc_request_set *set = data;
1565         struct list_head *tmp;
1566
1567         LASSERT(set != NULL);
1568         CERROR("INTERRUPTED SET %p\n", set);
1569
1570         list_for_each(tmp, &set->set_requests) {
1571                 struct ptlrpc_request *req =
1572                         list_entry(tmp, struct ptlrpc_request, rq_set_chain);
1573
1574                 if (req->rq_phase != RQ_PHASE_RPC && 
1575                     req->rq_phase != RQ_PHASE_UNREGISTERING)
1576                         continue;
1577
1578                 ptlrpc_mark_interrupted(req);
1579         }
1580 }
1581
1582 /** 
1583  * Get the smallest timeout in the set; this does NOT set a timeout. 
1584  */
1585 int ptlrpc_set_next_timeout(struct ptlrpc_request_set *set)
1586 {
1587         struct list_head      *tmp;
1588         time_t                 now = cfs_time_current_sec();
1589         int                    timeout = 0;
1590         struct ptlrpc_request *req;
1591         int                    deadline;
1592         ENTRY;
1593
1594         SIGNAL_MASK_ASSERT(); /* XXX BUG 1511 */
1595
1596         list_for_each(tmp, &set->set_requests) {
1597                 req = list_entry(tmp, struct ptlrpc_request, rq_set_chain);
1598
1599                 /* 
1600                  * Request in-flight? 
1601                  */
1602                 if (!(((req->rq_phase == RQ_PHASE_RPC) && !req->rq_waiting) ||
1603                       (req->rq_phase == RQ_PHASE_BULK) ||
1604                       (req->rq_phase == RQ_PHASE_NEW)))
1605                         continue;
1606
1607                 /* 
1608                  * Already timed out. 
1609                  */
1610                 if (req->rq_timedout)
1611                         continue;
1612
1613                 /* 
1614                  * Waiting for ctx. 
1615                  */
1616                 if (req->rq_wait_ctx)
1617                         continue;
1618
1619                 if (req->rq_phase == RQ_PHASE_NEW)
1620                         deadline = req->rq_sent;
1621                 else
1622                         deadline = req->rq_sent + req->rq_timeout;
1623
1624                 if (deadline <= now)    /* actually expired already */
1625                         timeout = 1;    /* ASAP */
1626                 else if (timeout == 0 || timeout > deadline - now)
1627                         timeout = deadline - now;
1628         }
1629         RETURN(timeout);
1630 }
1631
1632 int ptlrpc_set_wait(struct ptlrpc_request_set *set)
1633 {
1634         struct list_head      *tmp;
1635         struct ptlrpc_request *req;
1636         struct l_wait_info     lwi;
1637         int                    rc, timeout;
1638         ENTRY;
1639
1640         if (list_empty(&set->set_requests))
1641                 RETURN(0);
1642
1643         list_for_each(tmp, &set->set_requests) {
1644                 req = list_entry(tmp, struct ptlrpc_request, rq_set_chain);
1645                 if (req->rq_phase == RQ_PHASE_NEW)
1646                         (void)ptlrpc_send_new_req(req);
1647         }
1648
1649         do {
1650                 timeout = ptlrpc_set_next_timeout(set);
1651
1652                 /* wait until all complete, interrupted, or an in-flight
1653                  * req times out */
1654                 CDEBUG(D_RPCTRACE, "set %p going to sleep for %d seconds\n",
1655                        set, timeout);
1656                 lwi = LWI_TIMEOUT_INTR(cfs_time_seconds(timeout ? timeout : 1),
1657                                        ptlrpc_expired_set,
1658                                        ptlrpc_interrupted_set, set);
1659                 rc = l_wait_event(set->set_waitq,
1660                                   ptlrpc_check_set(NULL, set), &lwi);
1661
1662                 LASSERT(rc == 0 || rc == -EINTR || rc == -ETIMEDOUT);
1663
1664                 /* -EINTR => all requests have been flagged rq_intr so next
1665                  * check completes.
1666                  * -ETIMEOUTD => someone timed out.  When all reqs have
1667                  * timed out, signals are enabled allowing completion with
1668                  * EINTR.
1669                  * I don't really care if we go once more round the loop in
1670                  * the error cases -eeb. */
1671         } while (rc != 0 || set->set_remaining != 0);
1672
1673         LASSERT(set->set_remaining == 0);
1674
1675         rc = 0;
1676         list_for_each(tmp, &set->set_requests) {
1677                 req = list_entry(tmp, struct ptlrpc_request, rq_set_chain);
1678
1679                 LASSERT(req->rq_phase == RQ_PHASE_COMPLETE);
1680                 if (req->rq_status != 0)
1681                         rc = req->rq_status;
1682         }
1683
1684         if (set->set_interpret != NULL) {
1685                 int (*interpreter)(struct ptlrpc_request_set *set,void *,int) =
1686                         set->set_interpret;
1687                 rc = interpreter (set, set->set_arg, rc);
1688         } else {
1689                 struct ptlrpc_set_cbdata *cbdata, *n;
1690                 int err;
1691
1692                 list_for_each_entry_safe(cbdata, n,
1693                                          &set->set_cblist, psc_item) {
1694                         list_del_init(&cbdata->psc_item);
1695                         err = cbdata->psc_interpret(set, cbdata->psc_data, rc);
1696                         if (err && !rc)
1697                                 rc = err;
1698                         OBD_FREE_PTR(cbdata);
1699                 }
1700         }
1701
1702         RETURN(rc);
1703 }
1704
1705 static void __ptlrpc_free_req(struct ptlrpc_request *request, int locked)
1706 {
1707         ENTRY;
1708         if (request == NULL) {
1709                 EXIT;
1710                 return;
1711         }
1712
1713         LASSERTF(!request->rq_receiving_reply, "req %p\n", request);
1714         LASSERTF(request->rq_rqbd == NULL, "req %p\n",request);/* client-side */
1715         LASSERTF(list_empty(&request->rq_list), "req %p\n", request);
1716         LASSERTF(list_empty(&request->rq_set_chain), "req %p\n", request);
1717         LASSERTF(list_empty(&request->rq_exp_list), "req %p\n", request);
1718         LASSERTF(!request->rq_replay, "req %p\n", request);
1719         LASSERT(request->rq_cli_ctx);
1720
1721         req_capsule_fini(&request->rq_pill);
1722
1723         /* We must take it off the imp_replay_list first.  Otherwise, we'll set
1724          * request->rq_reqmsg to NULL while osc_close is dereferencing it. */
1725         if (request->rq_import != NULL) {
1726                 if (!locked)
1727                         spin_lock(&request->rq_import->imp_lock);
1728                 list_del_init(&request->rq_mod_list);
1729                 list_del_init(&request->rq_replay_list);
1730                 if (!locked)
1731                         spin_unlock(&request->rq_import->imp_lock);
1732         }
1733         LASSERTF(list_empty(&request->rq_replay_list), "req %p\n", request);
1734
1735         if (atomic_read(&request->rq_refcount) != 0) {
1736                 DEBUG_REQ(D_ERROR, request,
1737                           "freeing request with nonzero refcount");
1738                 LBUG();
1739         }
1740
1741         if (request->rq_repbuf != NULL)
1742                 sptlrpc_cli_free_repbuf(request);
1743         if (request->rq_export != NULL) {
1744                 class_export_put(request->rq_export);
1745                 request->rq_export = NULL;
1746         }
1747         if (request->rq_import != NULL) {
1748                 class_import_put(request->rq_import);
1749                 request->rq_import = NULL;
1750         }
1751         if (request->rq_bulk != NULL)
1752                 ptlrpc_free_bulk(request->rq_bulk);
1753
1754         if (request->rq_reqbuf != NULL || request->rq_clrbuf != NULL)
1755                 sptlrpc_cli_free_reqbuf(request);
1756
1757         sptlrpc_req_put_ctx(request, !locked);
1758
1759         if (request->rq_pool)
1760                 __ptlrpc_free_req_to_pool(request);
1761         else
1762                 OBD_FREE(request, sizeof(*request));
1763         EXIT;
1764 }
1765
1766 static int __ptlrpc_req_finished(struct ptlrpc_request *request, int locked);
1767 void ptlrpc_req_finished_with_imp_lock(struct ptlrpc_request *request)
1768 {
1769         LASSERT_SPIN_LOCKED(&request->rq_import->imp_lock);
1770         (void)__ptlrpc_req_finished(request, 1);
1771 }
1772
1773 static int __ptlrpc_req_finished(struct ptlrpc_request *request, int locked)
1774 {
1775         ENTRY;
1776         if (request == NULL)
1777                 RETURN(1);
1778
1779         if (request == LP_POISON ||
1780             request->rq_reqmsg == LP_POISON) {
1781                 CERROR("dereferencing freed request (bug 575)\n");
1782                 LBUG();
1783                 RETURN(1);
1784         }
1785
1786         DEBUG_REQ(D_INFO, request, "refcount now %u",
1787                   atomic_read(&request->rq_refcount) - 1);
1788
1789         if (atomic_dec_and_test(&request->rq_refcount)) {
1790                 __ptlrpc_free_req(request, locked);
1791                 RETURN(1);
1792         }
1793
1794         RETURN(0);
1795 }
1796
1797 void ptlrpc_req_finished(struct ptlrpc_request *request)
1798 {
1799         __ptlrpc_req_finished(request, 0);
1800 }
1801
1802 __u64 ptlrpc_req_xid(struct ptlrpc_request *request)
1803 {
1804         return request->rq_xid;
1805 }
1806 EXPORT_SYMBOL(ptlrpc_req_xid);
1807
1808 /* Disengage the client's reply buffer from the network
1809  * NB does _NOT_ unregister any client-side bulk.
1810  * IDEMPOTENT, but _not_ safe against concurrent callers.
1811  * The request owner (i.e. the thread doing the I/O) must call...
1812  */
1813 int ptlrpc_unregister_reply(struct ptlrpc_request *request, int async)
1814 {
1815         int                rc;
1816         cfs_waitq_t       *wq;
1817         struct l_wait_info lwi;
1818
1819         /* 
1820          * Might sleep. 
1821          */
1822         LASSERT(!in_interrupt());
1823
1824         /* 
1825          * Let's setup deadline for reply unlink. 
1826          */
1827         if (OBD_FAIL_CHECK(OBD_FAIL_PTLRPC_LONG_REPL_UNLINK) && 
1828             async && request->rq_reply_deadline == 0)
1829                 request->rq_reply_deadline = cfs_time_current_sec()+LONG_UNLINK;
1830
1831         /* 
1832          * Nothing left to do. 
1833          */
1834         if (!ptlrpc_client_recv_or_unlink(request))
1835                 RETURN(1);
1836
1837         LNetMDUnlink(request->rq_reply_md_h);
1838
1839         /* 
1840          * Let's check it once again. 
1841          */
1842         if (!ptlrpc_client_recv_or_unlink(request))
1843                 RETURN(1);
1844
1845         /* 
1846          * Move to "Unregistering" phase as reply was not unlinked yet. 
1847          */
1848         ptlrpc_rqphase_move(request, RQ_PHASE_UNREGISTERING);
1849
1850         /* 
1851          * Do not wait for unlink to finish. 
1852          */
1853         if (async)
1854                 RETURN(0);
1855
1856         /* 
1857          * We have to l_wait_event() whatever the result, to give liblustre
1858          * a chance to run reply_in_callback(), and to make sure we've
1859          * unlinked before returning a req to the pool. 
1860          */
1861         if (request->rq_set != NULL)
1862                 wq = &request->rq_set->set_waitq;
1863         else
1864                 wq = &request->rq_reply_waitq;
1865
1866         for (;;) {
1867                 /* Network access will complete in finite time but the HUGE
1868                  * timeout lets us CWARN for visibility of sluggish NALs */
1869                 lwi = LWI_TIMEOUT_INTERVAL(cfs_time_seconds(LONG_UNLINK),
1870                                            cfs_time_seconds(1), NULL, NULL);
1871                 rc = l_wait_event(*wq, !ptlrpc_client_recv_or_unlink(request),
1872                                   &lwi);
1873                 if (rc == 0) {
1874                         ptlrpc_rqphase_move(request, request->rq_next_phase);
1875                         RETURN(1);
1876                 }
1877   
1878                 LASSERT(rc == -ETIMEDOUT);
1879                 DEBUG_REQ(D_WARNING, request, "Unexpectedly long timeout "
1880                           "rvcng=%d unlnk=%d", request->rq_receiving_reply,
1881                           request->rq_must_unlink);
1882         }
1883         RETURN(0);
1884 }
1885
1886 /* caller must hold imp->imp_lock */
1887 void ptlrpc_free_committed(struct obd_import *imp)
1888 {
1889         struct list_head *tmp, *saved;
1890         struct ptlrpc_request *req;
1891         struct ptlrpc_request *last_req = NULL; /* temporary fire escape */
1892         ENTRY;
1893
1894         LASSERT(imp != NULL);
1895
1896         LASSERT_SPIN_LOCKED(&imp->imp_lock);
1897
1898
1899         if (imp->imp_peer_committed_transno == imp->imp_last_transno_checked &&
1900             imp->imp_generation == imp->imp_last_generation_checked) {
1901                 CDEBUG(D_RPCTRACE, "%s: skip recheck: last_committed "LPU64"\n",
1902                        imp->imp_obd->obd_name, imp->imp_peer_committed_transno);
1903                 EXIT;
1904                 return;
1905         }
1906         CDEBUG(D_RPCTRACE, "%s: committing for last_committed "LPU64" gen %d\n",
1907                imp->imp_obd->obd_name, imp->imp_peer_committed_transno,
1908                imp->imp_generation);
1909         imp->imp_last_transno_checked = imp->imp_peer_committed_transno;
1910         imp->imp_last_generation_checked = imp->imp_generation;
1911
1912         list_for_each_safe(tmp, saved, &imp->imp_replay_list) {
1913                 req = list_entry(tmp, struct ptlrpc_request, rq_replay_list);
1914
1915                 /* XXX ok to remove when 1357 resolved - rread 05/29/03  */
1916                 LASSERT(req != last_req);
1917                 last_req = req;
1918
1919                 if (req->rq_import_generation < imp->imp_generation) {
1920                         DEBUG_REQ(D_RPCTRACE, req, "free request with old gen");
1921                         GOTO(free_req, 0);
1922                 }
1923
1924                 if (req->rq_replay) {
1925                         DEBUG_REQ(D_RPCTRACE, req, "keeping (FL_REPLAY)");
1926                         continue;
1927                 }
1928
1929                 /* not yet committed */
1930                 if (req->rq_transno > imp->imp_peer_committed_transno) {
1931                         DEBUG_REQ(D_RPCTRACE, req, "stopping search");
1932                         break;
1933                 }
1934
1935                 DEBUG_REQ(D_RPCTRACE, req, "commit (last_committed "LPU64")",
1936                           imp->imp_peer_committed_transno);
1937 free_req:
1938                 spin_lock(&req->rq_lock);
1939                 req->rq_replay = 0;
1940                 spin_unlock(&req->rq_lock);
1941                 if (req->rq_commit_cb != NULL)
1942                         req->rq_commit_cb(req);
1943                 list_del_init(&req->rq_replay_list);
1944                 __ptlrpc_req_finished(req, 1);
1945         }
1946
1947         EXIT;
1948         return;
1949 }
1950
1951 void ptlrpc_cleanup_client(struct obd_import *imp)
1952 {
1953         ENTRY;
1954         EXIT;
1955         return;
1956 }
1957
1958 void ptlrpc_resend_req(struct ptlrpc_request *req)
1959 {
1960         DEBUG_REQ(D_HA, req, "going to resend");
1961         lustre_msg_set_handle(req->rq_reqmsg, &(struct lustre_handle){ 0 });
1962         req->rq_status = -EAGAIN;
1963
1964         spin_lock(&req->rq_lock);
1965         req->rq_resend = 1;
1966         req->rq_net_err = 0;
1967         req->rq_timedout = 0;
1968         if (req->rq_bulk) {
1969                 __u64 old_xid = req->rq_xid;
1970
1971                 /* ensure previous bulk fails */
1972                 req->rq_xid = ptlrpc_next_xid();
1973                 CDEBUG(D_HA, "resend bulk old x"LPU64" new x"LPU64"\n",
1974                        old_xid, req->rq_xid);
1975         }
1976         ptlrpc_client_wake_req(req);
1977         spin_unlock(&req->rq_lock);
1978 }
1979
1980 /* XXX: this function and rq_status are currently unused */
1981 void ptlrpc_restart_req(struct ptlrpc_request *req)
1982 {
1983         DEBUG_REQ(D_HA, req, "restarting (possibly-)completed request");
1984         req->rq_status = -ERESTARTSYS;
1985
1986         spin_lock(&req->rq_lock);
1987         req->rq_restart = 1;
1988         req->rq_timedout = 0;
1989         ptlrpc_client_wake_req(req);
1990         spin_unlock(&req->rq_lock);
1991 }
1992
1993 static int expired_request(void *data)
1994 {
1995         struct ptlrpc_request *req = data;
1996         ENTRY;
1997
1998         /* 
1999          * Some failure can suspend regular timeouts.
2000          */
2001         if (ptlrpc_check_suspend())
2002                 RETURN(1);
2003
2004         /* 
2005          * Deadline may have changed with an early reply. 
2006          */
2007         if (req->rq_deadline > cfs_time_current_sec())
2008                 RETURN(1);
2009
2010         RETURN(ptlrpc_expire_one_request(req, 0));
2011 }
2012
2013 static void interrupted_request(void *data)
2014 {
2015         struct ptlrpc_request *req = data;
2016         DEBUG_REQ(D_HA, req, "request interrupted");
2017         spin_lock(&req->rq_lock);
2018         req->rq_intr = 1;
2019         spin_unlock(&req->rq_lock);
2020 }
2021
2022 struct ptlrpc_request *ptlrpc_request_addref(struct ptlrpc_request *req)
2023 {
2024         ENTRY;
2025         atomic_inc(&req->rq_refcount);
2026         RETURN(req);
2027 }
2028
2029 void ptlrpc_retain_replayable_request(struct ptlrpc_request *req,
2030                                       struct obd_import *imp)
2031 {
2032         struct list_head *tmp;
2033
2034         LASSERT_SPIN_LOCKED(&imp->imp_lock);
2035
2036         /* clear this for new requests that were resent as well
2037            as resent replayed requests. */
2038         lustre_msg_clear_flags(req->rq_reqmsg, MSG_RESENT);
2039
2040         /* don't re-add requests that have been replayed */
2041         if (!list_empty(&req->rq_replay_list))
2042                 return;
2043
2044         lustre_msg_add_flags(req->rq_reqmsg, MSG_REPLAY);
2045
2046         LASSERT(imp->imp_replayable);
2047         /* Balanced in ptlrpc_free_committed, usually. */
2048         ptlrpc_request_addref(req);
2049         list_for_each_prev(tmp, &imp->imp_replay_list) {
2050                 struct ptlrpc_request *iter =
2051                         list_entry(tmp, struct ptlrpc_request, rq_replay_list);
2052
2053                 /* We may have duplicate transnos if we create and then
2054                  * open a file, or for closes retained if to match creating
2055                  * opens, so use req->rq_xid as a secondary key.
2056                  * (See bugs 684, 685, and 428.)
2057                  * XXX no longer needed, but all opens need transnos!
2058                  */
2059                 if (iter->rq_transno > req->rq_transno)
2060                         continue;
2061
2062                 if (iter->rq_transno == req->rq_transno) {
2063                         LASSERT(iter->rq_xid != req->rq_xid);
2064                         if (iter->rq_xid > req->rq_xid)
2065                                 continue;
2066                 }
2067
2068                 list_add(&req->rq_replay_list, &iter->rq_replay_list);
2069                 return;
2070         }
2071
2072         list_add_tail(&req->rq_replay_list, &imp->imp_replay_list);
2073 }
2074
2075 int ptlrpc_queue_wait(struct ptlrpc_request *req)
2076 {
2077         int rc = 0;
2078         int brc;
2079         struct l_wait_info lwi;
2080         struct obd_import *imp = req->rq_import;
2081         cfs_duration_t timeout = CFS_TICK;
2082         long timeoutl;
2083         ENTRY;
2084
2085         LASSERT(req->rq_set == NULL);
2086         LASSERT(!req->rq_receiving_reply);
2087
2088         /* for distributed debugging */
2089         lustre_msg_set_status(req->rq_reqmsg, cfs_curproc_pid());
2090         LASSERT(imp->imp_obd != NULL);
2091         CDEBUG(D_RPCTRACE, "Sending RPC pname:cluuid:pid:xid:nid:opc "
2092                "%s:%s:%d:"LPU64":%s:%d\n", cfs_curproc_comm(),
2093                imp->imp_obd->obd_uuid.uuid,
2094                lustre_msg_get_status(req->rq_reqmsg), req->rq_xid,
2095                libcfs_nid2str(imp->imp_connection->c_peer.nid),
2096                lustre_msg_get_opc(req->rq_reqmsg));
2097
2098         /* Mark phase here for a little debug help */
2099         ptlrpc_rqphase_move(req, RQ_PHASE_RPC);
2100
2101         spin_lock(&imp->imp_lock);
2102         req->rq_import_generation = imp->imp_generation;
2103 restart:
2104         if (ptlrpc_import_delay_req(imp, req, &rc)) {
2105                 list_del_init(&req->rq_list);
2106                 list_add_tail(&req->rq_list, &imp->imp_delayed_list);
2107                 atomic_inc(&imp->imp_inflight);
2108                 spin_unlock(&imp->imp_lock);
2109
2110                 DEBUG_REQ(D_HA, req, "\"%s\" waiting for recovery: (%s != %s)",
2111                           cfs_curproc_comm(),
2112                           ptlrpc_import_state_name(req->rq_send_state),
2113                           ptlrpc_import_state_name(imp->imp_state));
2114                 lwi = LWI_INTR(interrupted_request, req);
2115                 rc = l_wait_event(req->rq_reply_waitq,
2116                                   (req->rq_send_state == imp->imp_state ||
2117                                    req->rq_err || req->rq_intr),
2118                                   &lwi);
2119                 DEBUG_REQ(D_HA, req, "\"%s\" awake: (%s == %s or %d/%d == 1)",
2120                           cfs_curproc_comm(),
2121                           ptlrpc_import_state_name(imp->imp_state),
2122                           ptlrpc_import_state_name(req->rq_send_state),
2123                           req->rq_err, req->rq_intr);
2124
2125                 spin_lock(&imp->imp_lock);
2126                 list_del_init(&req->rq_list);
2127                 atomic_dec(&imp->imp_inflight);
2128
2129                 if (req->rq_err) {
2130                         /* rq_status was set locally */
2131                         rc = -EIO;
2132                 }
2133                 else if (req->rq_intr) {
2134                         rc = -EINTR;
2135                 }
2136                 else if (req->rq_no_resend) {
2137                         spin_unlock(&imp->imp_lock);
2138                         GOTO(out, rc = -ETIMEDOUT);
2139                 }
2140                 else {
2141                         GOTO(restart, rc);
2142                 }
2143         }
2144
2145         if (rc != 0) {
2146                 spin_unlock(&imp->imp_lock);
2147                 req->rq_status = rc; // XXX this ok?
2148                 GOTO(out, rc);
2149         }
2150
2151         if (req->rq_resend) {
2152                 lustre_msg_add_flags(req->rq_reqmsg, MSG_RESENT);
2153
2154                 if (req->rq_bulk != NULL) {
2155                         ptlrpc_unregister_bulk(req, 0);
2156
2157                         /* bulk requests are supposed to be
2158                          * idempotent, so we are free to bump the xid
2159                          * here, which we need to do before
2160                          * registering the bulk again (bug 6371).
2161                          * print the old xid first for sanity.
2162                          */
2163                         DEBUG_REQ(D_HA, req, "bumping xid for bulk: ");
2164                         req->rq_xid = ptlrpc_next_xid();
2165                 }
2166
2167                 DEBUG_REQ(D_HA, req, "resending: ");
2168         }
2169
2170         /* XXX this is the same as ptlrpc_set_wait */
2171         LASSERT(list_empty(&req->rq_list));
2172         list_add_tail(&req->rq_list, &imp->imp_sending_list);
2173         atomic_inc(&imp->imp_inflight);
2174         spin_unlock(&imp->imp_lock);
2175
2176         rc = sptlrpc_req_refresh_ctx(req, 0);
2177         if (rc) {
2178                 if (req->rq_err) {
2179                         /* we got fatal ctx refresh error, directly jump out
2180                          * thus we can pass back the actual error code.
2181                          */
2182                         spin_lock(&imp->imp_lock);
2183                         list_del_init(&req->rq_list);
2184                         spin_unlock(&imp->imp_lock);
2185
2186                         CERROR("Failed to refresh ctx of req %p: %d\n", req, rc);
2187                         GOTO(out, rc);
2188                 }
2189                 /* simulating we got error during send rpc */
2190                 goto after_send;
2191         }
2192
2193         rc = ptl_send_rpc(req, 0);
2194         if (rc)
2195                 DEBUG_REQ(D_HA, req, "send failed (%d); recovering", rc);
2196 repeat:
2197         timeoutl = req->rq_deadline - cfs_time_current_sec();
2198         timeout = (timeoutl <= 0 || rc) ? CFS_TICK :
2199                 cfs_time_seconds(timeoutl);
2200         DEBUG_REQ(D_NET, req,
2201                   "-- sleeping for "CFS_DURATION_T" ticks", timeout);
2202         lwi = LWI_TIMEOUT_INTR(timeout, expired_request, interrupted_request,
2203                                req);
2204         brc = l_wait_event(req->rq_reply_waitq, ptlrpc_check_reply(req), &lwi);
2205         if (brc == -ETIMEDOUT && ((req->rq_deadline > cfs_time_current_sec()) ||
2206                                  ptlrpc_check_and_wait_suspend(req)))
2207                 goto repeat;
2208
2209 after_send:
2210         CDEBUG(D_RPCTRACE, "Completed RPC pname:cluuid:pid:xid:nid:opc "
2211                "%s:%s:%d:"LPU64":%s:%d\n", cfs_curproc_comm(),
2212                imp->imp_obd->obd_uuid.uuid,
2213                lustre_msg_get_status(req->rq_reqmsg), req->rq_xid,
2214                libcfs_nid2str(imp->imp_connection->c_peer.nid),
2215                lustre_msg_get_opc(req->rq_reqmsg));
2216
2217         /* If the reply was received normally, this just grabs the spinlock
2218          * (ensuring the reply callback has returned), sees that
2219          * req->rq_receiving_reply is clear and returns. */
2220         ptlrpc_unregister_reply(req, 0);
2221
2222         spin_lock(&imp->imp_lock);
2223         list_del_init(&req->rq_list);
2224         atomic_dec(&imp->imp_inflight);
2225         spin_unlock(&imp->imp_lock);
2226
2227         if (req->rq_err) {
2228                 DEBUG_REQ(D_RPCTRACE, req, "err rc=%d status=%d",
2229                           rc, req->rq_status);
2230                 GOTO(out, rc = rc ? rc : -EIO);
2231         }
2232
2233         if (req->rq_intr) {
2234                 /* Should only be interrupted if we timed out. */
2235                 if (!req->rq_timedout)
2236                         DEBUG_REQ(D_ERROR, req,
2237                                   "rq_intr set but rq_timedout not");
2238                 GOTO(out, rc = -EINTR);
2239         }
2240
2241         /* Resend if we need to */
2242         if (req->rq_resend) {
2243                 /* ...unless we were specifically told otherwise. */
2244                 if (req->rq_no_resend)
2245                         GOTO(out, rc = -ETIMEDOUT);
2246                 spin_lock(&imp->imp_lock);
2247                 goto restart;
2248         }
2249
2250         if (req->rq_timedout) {                 /* non-recoverable timeout */
2251                 GOTO(out, rc = -ETIMEDOUT);
2252         }
2253
2254         if (!ptlrpc_client_replied(req)) {
2255                 /* How can this be? -eeb */
2256                 DEBUG_REQ(D_ERROR, req, "!rq_replied: ");
2257                 LBUG();
2258                 GOTO(out, rc = req->rq_status);
2259         }
2260
2261         rc = after_reply(req);
2262         /* NB may return +ve success rc */
2263         if (req->rq_resend) {
2264                 spin_lock(&imp->imp_lock);
2265                 goto restart;
2266         }
2267
2268  out:
2269         if (req->rq_bulk != NULL) {
2270                 if (rc >= 0) {
2271                         /* success so far.  Note that anything going wrong
2272                          * with bulk now, is EXTREMELY strange, since the
2273                          * server must have believed that the bulk
2274                          * tranferred OK before she replied with success to
2275                          * me. */
2276                         lwi = LWI_TIMEOUT(timeout, NULL, NULL);
2277                         brc = l_wait_event(req->rq_reply_waitq,
2278                                            !ptlrpc_client_bulk_active(req),
2279                                            &lwi);
2280                         LASSERT(brc == 0 || brc == -ETIMEDOUT);
2281                         if (brc != 0) {
2282                                 LASSERT(brc == -ETIMEDOUT);
2283                                 DEBUG_REQ(D_ERROR, req, "bulk timed out");
2284                                 rc = brc;
2285                         } else if (!req->rq_bulk->bd_success) {
2286                                 DEBUG_REQ(D_ERROR, req, "bulk transfer failed");
2287                                 rc = -EIO;
2288                         }
2289                 }
2290                 if (rc < 0)
2291                         ptlrpc_unregister_bulk(req, 0);
2292         }
2293
2294         LASSERT(!req->rq_receiving_reply);
2295         ptlrpc_rqphase_move(req, RQ_PHASE_INTERPRET);
2296         cfs_waitq_signal(&imp->imp_recovery_waitq);
2297         RETURN(rc);
2298 }
2299
2300 struct ptlrpc_replay_async_args {
2301         int praa_old_state;
2302         int praa_old_status;
2303 };
2304
2305 static int ptlrpc_replay_interpret(const struct lu_env *env,
2306                                    struct ptlrpc_request *req,
2307                                    void * data, int rc)
2308 {
2309         struct ptlrpc_replay_async_args *aa = data;
2310         struct obd_import *imp = req->rq_import;
2311
2312         ENTRY;
2313         atomic_dec(&imp->imp_replay_inflight);
2314
2315         if (!ptlrpc_client_replied(req)) {
2316                 CERROR("request replay timed out, restarting recovery\n");
2317                 GOTO(out, rc = -ETIMEDOUT);
2318         }
2319
2320         if (lustre_msg_get_type(req->rq_repmsg) == PTL_RPC_MSG_ERR &&
2321             (lustre_msg_get_status(req->rq_repmsg) == -ENOTCONN ||
2322              lustre_msg_get_status(req->rq_repmsg) == -ENODEV))
2323                 GOTO(out, rc = lustre_msg_get_status(req->rq_repmsg));
2324
2325         /* The transno had better not change over replay. */
2326         LASSERT(lustre_msg_get_transno(req->rq_reqmsg) ==
2327                 lustre_msg_get_transno(req->rq_repmsg));
2328
2329         DEBUG_REQ(D_HA, req, "got rep");
2330
2331         /* let the callback do fixups, possibly including in the request */
2332         if (req->rq_replay_cb)
2333                 req->rq_replay_cb(req);
2334
2335         if (ptlrpc_client_replied(req) &&
2336             lustre_msg_get_status(req->rq_repmsg) != aa->praa_old_status) {
2337                 DEBUG_REQ(D_ERROR, req, "status %d, old was %d",
2338                           lustre_msg_get_status(req->rq_repmsg),
2339                           aa->praa_old_status);
2340         } else {
2341                 /* Put it back for re-replay. */
2342                 lustre_msg_set_status(req->rq_repmsg, aa->praa_old_status);
2343         }
2344
2345         /*
2346          * Errors while replay can set transno to 0, but
2347          * imp_last_replay_transno shouldn't be set to 0 anyway
2348          */
2349         if (req->rq_transno > 0) {
2350                 spin_lock(&imp->imp_lock);
2351                 LASSERT(req->rq_transno <= imp->imp_last_replay_transno);
2352                 imp->imp_last_replay_transno = req->rq_transno;
2353                 spin_unlock(&imp->imp_lock);
2354         } else
2355                 CERROR("Transno is 0 during replay!\n");
2356         /* continue with recovery */
2357         rc = ptlrpc_import_recovery_state_machine(imp);
2358  out:
2359         req->rq_send_state = aa->praa_old_state;
2360
2361         if (rc != 0)
2362                 /* this replay failed, so restart recovery */
2363                 ptlrpc_connect_import(imp, NULL);
2364
2365         RETURN(rc);
2366 }
2367
2368 int ptlrpc_replay_req(struct ptlrpc_request *req)
2369 {
2370         struct ptlrpc_replay_async_args *aa;
2371         ENTRY;
2372
2373         LASSERT(req->rq_import->imp_state == LUSTRE_IMP_REPLAY);
2374         /* Not handling automatic bulk replay yet (or ever?) */
2375         LASSERT(req->rq_bulk == NULL);
2376
2377         LASSERT (sizeof (*aa) <= sizeof (req->rq_async_args));
2378         aa = ptlrpc_req_async_args(req);
2379         memset(aa, 0, sizeof *aa);
2380
2381         /* Prepare request to be resent with ptlrpcd */
2382         aa->praa_old_state = req->rq_send_state;
2383         req->rq_send_state = LUSTRE_IMP_REPLAY;
2384         req->rq_phase = RQ_PHASE_NEW;
2385         req->rq_next_phase = RQ_PHASE_UNDEFINED;
2386         if (req->rq_repmsg)
2387                 aa->praa_old_status = lustre_msg_get_status(req->rq_repmsg);
2388         req->rq_status = 0;
2389         req->rq_interpret_reply = ptlrpc_replay_interpret;
2390         /* Readjust the timeout for current conditions */
2391         ptlrpc_at_set_req_timeout(req);
2392
2393         DEBUG_REQ(D_HA, req, "REPLAY");
2394
2395         atomic_inc(&req->rq_import->imp_replay_inflight);
2396         ptlrpc_request_addref(req); /* ptlrpcd needs a ref */
2397
2398         ptlrpcd_add_req(req, PSCOPE_OTHER);
2399         RETURN(0);
2400 }
2401
2402 void ptlrpc_abort_inflight(struct obd_import *imp)
2403 {
2404         struct list_head *tmp, *n;
2405         ENTRY;
2406
2407         /* Make sure that no new requests get processed for this import.
2408          * ptlrpc_{queue,set}_wait must (and does) hold imp_lock while testing
2409          * this flag and then putting requests on sending_list or delayed_list.
2410          */
2411         spin_lock(&imp->imp_lock);
2412
2413         /* XXX locking?  Maybe we should remove each request with the list
2414          * locked?  Also, how do we know if the requests on the list are
2415          * being freed at this time?
2416          */
2417         list_for_each_safe(tmp, n, &imp->imp_sending_list) {
2418                 struct ptlrpc_request *req =
2419                         list_entry(tmp, struct ptlrpc_request, rq_list);
2420
2421                 DEBUG_REQ(D_RPCTRACE, req, "inflight");
2422
2423                 spin_lock (&req->rq_lock);
2424                 if (req->rq_import_generation < imp->imp_generation) {
2425                         req->rq_err = 1;
2426                         req->rq_status = -EINTR;
2427                         ptlrpc_client_wake_req(req);
2428                 }
2429                 spin_unlock (&req->rq_lock);
2430         }
2431
2432         list_for_each_safe(tmp, n, &imp->imp_delayed_list) {
2433                 struct ptlrpc_request *req =
2434                         list_entry(tmp, struct ptlrpc_request, rq_list);
2435
2436                 DEBUG_REQ(D_RPCTRACE, req, "aborting waiting req");
2437
2438                 spin_lock (&req->rq_lock);
2439                 if (req->rq_import_generation < imp->imp_generation) {
2440                         req->rq_err = 1;
2441                         req->rq_status = -EINTR;
2442                         ptlrpc_client_wake_req(req);
2443                 }
2444                 spin_unlock (&req->rq_lock);
2445         }
2446
2447         /* Last chance to free reqs left on the replay list, but we
2448          * will still leak reqs that haven't committed.  */
2449         if (imp->imp_replayable)
2450                 ptlrpc_free_committed(imp);
2451
2452         spin_unlock(&imp->imp_lock);
2453
2454         EXIT;
2455 }
2456
2457 void ptlrpc_abort_set(struct ptlrpc_request_set *set)
2458 {
2459         struct list_head *tmp, *pos;
2460
2461         LASSERT(set != NULL);
2462
2463         list_for_each_safe(pos, tmp, &set->set_requests) {
2464                 struct ptlrpc_request *req =
2465                         list_entry(pos, struct ptlrpc_request, rq_set_chain);
2466
2467                 spin_lock(&req->rq_lock);
2468                 if (req->rq_phase != RQ_PHASE_RPC) {
2469                         spin_unlock(&req->rq_lock);
2470                         continue;
2471                 }
2472
2473                 req->rq_err = 1;
2474                 req->rq_status = -EINTR;
2475                 ptlrpc_client_wake_req(req);
2476                 spin_unlock(&req->rq_lock);
2477         }
2478 }
2479
2480 static __u64 ptlrpc_last_xid;
2481 static spinlock_t ptlrpc_last_xid_lock;
2482
2483 /* Initialize the XID for the node.  This is common among all requests on
2484  * this node, and only requires the property that it is monotonically
2485  * increasing.  It does not need to be sequential.  Since this is also used
2486  * as the RDMA match bits, it is important that a single client NOT have
2487  * the same match bits for two different in-flight requests, hence we do
2488  * NOT want to have an XID per target or similar.
2489  *
2490  * To avoid an unlikely collision between match bits after a client reboot
2491  * (which would cause old to be delivered into the wrong buffer) we initialize
2492  * the XID based on the current time, assuming a maximum RPC rate of 1M RPC/s.
2493  * If the time is clearly incorrect, we instead use a 62-bit random number.
2494  * In the worst case the random number will overflow 1M RPCs per second in
2495  * 9133 years, or permutations thereof.
2496  */
2497 #define YEAR_2004 (1ULL << 30)
2498 void ptlrpc_init_xid(void)
2499 {
2500         time_t now = cfs_time_current_sec();
2501
2502         spin_lock_init(&ptlrpc_last_xid_lock);
2503         if (now < YEAR_2004) {
2504                 ll_get_random_bytes(&ptlrpc_last_xid, sizeof(ptlrpc_last_xid));
2505                 ptlrpc_last_xid >>= 2;
2506                 ptlrpc_last_xid |= (1ULL << 61);
2507         } else {
2508                 ptlrpc_last_xid = (now << 20);
2509         }
2510 }
2511
2512 __u64 ptlrpc_next_xid(void)
2513 {
2514         __u64 tmp;
2515         spin_lock(&ptlrpc_last_xid_lock);
2516         tmp = ++ptlrpc_last_xid;
2517         spin_unlock(&ptlrpc_last_xid_lock);
2518         return tmp;
2519 }
2520
2521 __u64 ptlrpc_sample_next_xid(void)
2522 {
2523 #if BITS_PER_LONG == 32
2524         /* need to avoid possible word tearing on 32-bit systems */
2525         __u64 tmp;
2526         spin_lock(&ptlrpc_last_xid_lock);
2527         tmp = ptlrpc_last_xid + 1;
2528         spin_unlock(&ptlrpc_last_xid_lock);
2529         return tmp;
2530 #else
2531         /* No need to lock, since returned value is racy anyways */
2532         return ptlrpc_last_xid + 1;
2533 #endif
2534 }
2535 EXPORT_SYMBOL(ptlrpc_sample_next_xid);