Loading...
1// SPDX-License-Identifier: GPL-2.0
2
3#include <linux/ceph/ceph_debug.h>
4
5#include <linux/module.h>
6#include <linux/err.h>
7#include <linux/highmem.h>
8#include <linux/mm.h>
9#include <linux/pagemap.h>
10#include <linux/slab.h>
11#include <linux/uaccess.h>
12#ifdef CONFIG_BLOCK
13#include <linux/bio.h>
14#endif
15
16#include <linux/ceph/ceph_features.h>
17#include <linux/ceph/libceph.h>
18#include <linux/ceph/osd_client.h>
19#include <linux/ceph/messenger.h>
20#include <linux/ceph/decode.h>
21#include <linux/ceph/auth.h>
22#include <linux/ceph/pagelist.h>
23#include <linux/ceph/striper.h>
24
25#define OSD_OPREPLY_FRONT_LEN 512
26
27static struct kmem_cache *ceph_osd_request_cache;
28
29static const struct ceph_connection_operations osd_con_ops;
30
31/*
32 * Implement client access to distributed object storage cluster.
33 *
34 * All data objects are stored within a cluster/cloud of OSDs, or
35 * "object storage devices." (Note that Ceph OSDs have _nothing_ to
36 * do with the T10 OSD extensions to SCSI.) Ceph OSDs are simply
37 * remote daemons serving up and coordinating consistent and safe
38 * access to storage.
39 *
40 * Cluster membership and the mapping of data objects onto storage devices
41 * are described by the osd map.
42 *
43 * We keep track of pending OSD requests (read, write), resubmit
44 * requests to different OSDs when the cluster topology/data layout
45 * change, or retry the affected requests when the communications
46 * channel with an OSD is reset.
47 */
48
49static void link_request(struct ceph_osd *osd, struct ceph_osd_request *req);
50static void unlink_request(struct ceph_osd *osd, struct ceph_osd_request *req);
51static void link_linger(struct ceph_osd *osd,
52 struct ceph_osd_linger_request *lreq);
53static void unlink_linger(struct ceph_osd *osd,
54 struct ceph_osd_linger_request *lreq);
55static void clear_backoffs(struct ceph_osd *osd);
56
57#if 1
58static inline bool rwsem_is_wrlocked(struct rw_semaphore *sem)
59{
60 bool wrlocked = true;
61
62 if (unlikely(down_read_trylock(sem))) {
63 wrlocked = false;
64 up_read(sem);
65 }
66
67 return wrlocked;
68}
69static inline void verify_osdc_locked(struct ceph_osd_client *osdc)
70{
71 WARN_ON(!rwsem_is_locked(&osdc->lock));
72}
73static inline void verify_osdc_wrlocked(struct ceph_osd_client *osdc)
74{
75 WARN_ON(!rwsem_is_wrlocked(&osdc->lock));
76}
77static inline void verify_osd_locked(struct ceph_osd *osd)
78{
79 struct ceph_osd_client *osdc = osd->o_osdc;
80
81 WARN_ON(!(mutex_is_locked(&osd->lock) &&
82 rwsem_is_locked(&osdc->lock)) &&
83 !rwsem_is_wrlocked(&osdc->lock));
84}
85static inline void verify_lreq_locked(struct ceph_osd_linger_request *lreq)
86{
87 WARN_ON(!mutex_is_locked(&lreq->lock));
88}
89#else
90static inline void verify_osdc_locked(struct ceph_osd_client *osdc) { }
91static inline void verify_osdc_wrlocked(struct ceph_osd_client *osdc) { }
92static inline void verify_osd_locked(struct ceph_osd *osd) { }
93static inline void verify_lreq_locked(struct ceph_osd_linger_request *lreq) { }
94#endif
95
96/*
97 * calculate the mapping of a file extent onto an object, and fill out the
98 * request accordingly. shorten extent as necessary if it crosses an
99 * object boundary.
100 *
101 * fill osd op in request message.
102 */
103static int calc_layout(struct ceph_file_layout *layout, u64 off, u64 *plen,
104 u64 *objnum, u64 *objoff, u64 *objlen)
105{
106 u64 orig_len = *plen;
107 u32 xlen;
108
109 /* object extent? */
110 ceph_calc_file_object_mapping(layout, off, orig_len, objnum,
111 objoff, &xlen);
112 *objlen = xlen;
113 if (*objlen < orig_len) {
114 *plen = *objlen;
115 dout(" skipping last %llu, final file extent %llu~%llu\n",
116 orig_len - *plen, off, *plen);
117 }
118
119 dout("calc_layout objnum=%llx %llu~%llu\n", *objnum, *objoff, *objlen);
120 return 0;
121}
122
123static void ceph_osd_data_init(struct ceph_osd_data *osd_data)
124{
125 memset(osd_data, 0, sizeof (*osd_data));
126 osd_data->type = CEPH_OSD_DATA_TYPE_NONE;
127}
128
129/*
130 * Consumes @pages if @own_pages is true.
131 */
132static void ceph_osd_data_pages_init(struct ceph_osd_data *osd_data,
133 struct page **pages, u64 length, u32 alignment,
134 bool pages_from_pool, bool own_pages)
135{
136 osd_data->type = CEPH_OSD_DATA_TYPE_PAGES;
137 osd_data->pages = pages;
138 osd_data->length = length;
139 osd_data->alignment = alignment;
140 osd_data->pages_from_pool = pages_from_pool;
141 osd_data->own_pages = own_pages;
142}
143
144/*
145 * Consumes a ref on @pagelist.
146 */
147static void ceph_osd_data_pagelist_init(struct ceph_osd_data *osd_data,
148 struct ceph_pagelist *pagelist)
149{
150 osd_data->type = CEPH_OSD_DATA_TYPE_PAGELIST;
151 osd_data->pagelist = pagelist;
152}
153
154#ifdef CONFIG_BLOCK
155static void ceph_osd_data_bio_init(struct ceph_osd_data *osd_data,
156 struct ceph_bio_iter *bio_pos,
157 u32 bio_length)
158{
159 osd_data->type = CEPH_OSD_DATA_TYPE_BIO;
160 osd_data->bio_pos = *bio_pos;
161 osd_data->bio_length = bio_length;
162}
163#endif /* CONFIG_BLOCK */
164
165static void ceph_osd_data_bvecs_init(struct ceph_osd_data *osd_data,
166 struct ceph_bvec_iter *bvec_pos,
167 u32 num_bvecs)
168{
169 osd_data->type = CEPH_OSD_DATA_TYPE_BVECS;
170 osd_data->bvec_pos = *bvec_pos;
171 osd_data->num_bvecs = num_bvecs;
172}
173
174static struct ceph_osd_data *
175osd_req_op_raw_data_in(struct ceph_osd_request *osd_req, unsigned int which)
176{
177 BUG_ON(which >= osd_req->r_num_ops);
178
179 return &osd_req->r_ops[which].raw_data_in;
180}
181
182struct ceph_osd_data *
183osd_req_op_extent_osd_data(struct ceph_osd_request *osd_req,
184 unsigned int which)
185{
186 return osd_req_op_data(osd_req, which, extent, osd_data);
187}
188EXPORT_SYMBOL(osd_req_op_extent_osd_data);
189
190void osd_req_op_raw_data_in_pages(struct ceph_osd_request *osd_req,
191 unsigned int which, struct page **pages,
192 u64 length, u32 alignment,
193 bool pages_from_pool, bool own_pages)
194{
195 struct ceph_osd_data *osd_data;
196
197 osd_data = osd_req_op_raw_data_in(osd_req, which);
198 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
199 pages_from_pool, own_pages);
200}
201EXPORT_SYMBOL(osd_req_op_raw_data_in_pages);
202
203void osd_req_op_extent_osd_data_pages(struct ceph_osd_request *osd_req,
204 unsigned int which, struct page **pages,
205 u64 length, u32 alignment,
206 bool pages_from_pool, bool own_pages)
207{
208 struct ceph_osd_data *osd_data;
209
210 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
211 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
212 pages_from_pool, own_pages);
213}
214EXPORT_SYMBOL(osd_req_op_extent_osd_data_pages);
215
216void osd_req_op_extent_osd_data_pagelist(struct ceph_osd_request *osd_req,
217 unsigned int which, struct ceph_pagelist *pagelist)
218{
219 struct ceph_osd_data *osd_data;
220
221 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
222 ceph_osd_data_pagelist_init(osd_data, pagelist);
223}
224EXPORT_SYMBOL(osd_req_op_extent_osd_data_pagelist);
225
226#ifdef CONFIG_BLOCK
227void osd_req_op_extent_osd_data_bio(struct ceph_osd_request *osd_req,
228 unsigned int which,
229 struct ceph_bio_iter *bio_pos,
230 u32 bio_length)
231{
232 struct ceph_osd_data *osd_data;
233
234 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
235 ceph_osd_data_bio_init(osd_data, bio_pos, bio_length);
236}
237EXPORT_SYMBOL(osd_req_op_extent_osd_data_bio);
238#endif /* CONFIG_BLOCK */
239
240void osd_req_op_extent_osd_data_bvecs(struct ceph_osd_request *osd_req,
241 unsigned int which,
242 struct bio_vec *bvecs, u32 num_bvecs,
243 u32 bytes)
244{
245 struct ceph_osd_data *osd_data;
246 struct ceph_bvec_iter it = {
247 .bvecs = bvecs,
248 .iter = { .bi_size = bytes },
249 };
250
251 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
252 ceph_osd_data_bvecs_init(osd_data, &it, num_bvecs);
253}
254EXPORT_SYMBOL(osd_req_op_extent_osd_data_bvecs);
255
256void osd_req_op_extent_osd_data_bvec_pos(struct ceph_osd_request *osd_req,
257 unsigned int which,
258 struct ceph_bvec_iter *bvec_pos)
259{
260 struct ceph_osd_data *osd_data;
261
262 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
263 ceph_osd_data_bvecs_init(osd_data, bvec_pos, 0);
264}
265EXPORT_SYMBOL(osd_req_op_extent_osd_data_bvec_pos);
266
267static void osd_req_op_cls_request_info_pagelist(
268 struct ceph_osd_request *osd_req,
269 unsigned int which, struct ceph_pagelist *pagelist)
270{
271 struct ceph_osd_data *osd_data;
272
273 osd_data = osd_req_op_data(osd_req, which, cls, request_info);
274 ceph_osd_data_pagelist_init(osd_data, pagelist);
275}
276
277void osd_req_op_cls_request_data_pagelist(
278 struct ceph_osd_request *osd_req,
279 unsigned int which, struct ceph_pagelist *pagelist)
280{
281 struct ceph_osd_data *osd_data;
282
283 osd_data = osd_req_op_data(osd_req, which, cls, request_data);
284 ceph_osd_data_pagelist_init(osd_data, pagelist);
285 osd_req->r_ops[which].cls.indata_len += pagelist->length;
286 osd_req->r_ops[which].indata_len += pagelist->length;
287}
288EXPORT_SYMBOL(osd_req_op_cls_request_data_pagelist);
289
290void osd_req_op_cls_request_data_pages(struct ceph_osd_request *osd_req,
291 unsigned int which, struct page **pages, u64 length,
292 u32 alignment, bool pages_from_pool, bool own_pages)
293{
294 struct ceph_osd_data *osd_data;
295
296 osd_data = osd_req_op_data(osd_req, which, cls, request_data);
297 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
298 pages_from_pool, own_pages);
299 osd_req->r_ops[which].cls.indata_len += length;
300 osd_req->r_ops[which].indata_len += length;
301}
302EXPORT_SYMBOL(osd_req_op_cls_request_data_pages);
303
304void osd_req_op_cls_request_data_bvecs(struct ceph_osd_request *osd_req,
305 unsigned int which,
306 struct bio_vec *bvecs, u32 num_bvecs,
307 u32 bytes)
308{
309 struct ceph_osd_data *osd_data;
310 struct ceph_bvec_iter it = {
311 .bvecs = bvecs,
312 .iter = { .bi_size = bytes },
313 };
314
315 osd_data = osd_req_op_data(osd_req, which, cls, request_data);
316 ceph_osd_data_bvecs_init(osd_data, &it, num_bvecs);
317 osd_req->r_ops[which].cls.indata_len += bytes;
318 osd_req->r_ops[which].indata_len += bytes;
319}
320EXPORT_SYMBOL(osd_req_op_cls_request_data_bvecs);
321
322void osd_req_op_cls_response_data_pages(struct ceph_osd_request *osd_req,
323 unsigned int which, struct page **pages, u64 length,
324 u32 alignment, bool pages_from_pool, bool own_pages)
325{
326 struct ceph_osd_data *osd_data;
327
328 osd_data = osd_req_op_data(osd_req, which, cls, response_data);
329 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
330 pages_from_pool, own_pages);
331}
332EXPORT_SYMBOL(osd_req_op_cls_response_data_pages);
333
334static u64 ceph_osd_data_length(struct ceph_osd_data *osd_data)
335{
336 switch (osd_data->type) {
337 case CEPH_OSD_DATA_TYPE_NONE:
338 return 0;
339 case CEPH_OSD_DATA_TYPE_PAGES:
340 return osd_data->length;
341 case CEPH_OSD_DATA_TYPE_PAGELIST:
342 return (u64)osd_data->pagelist->length;
343#ifdef CONFIG_BLOCK
344 case CEPH_OSD_DATA_TYPE_BIO:
345 return (u64)osd_data->bio_length;
346#endif /* CONFIG_BLOCK */
347 case CEPH_OSD_DATA_TYPE_BVECS:
348 return osd_data->bvec_pos.iter.bi_size;
349 default:
350 WARN(true, "unrecognized data type %d\n", (int)osd_data->type);
351 return 0;
352 }
353}
354
355static void ceph_osd_data_release(struct ceph_osd_data *osd_data)
356{
357 if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGES && osd_data->own_pages) {
358 int num_pages;
359
360 num_pages = calc_pages_for((u64)osd_data->alignment,
361 (u64)osd_data->length);
362 ceph_release_page_vector(osd_data->pages, num_pages);
363 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGELIST) {
364 ceph_pagelist_release(osd_data->pagelist);
365 }
366 ceph_osd_data_init(osd_data);
367}
368
369static void osd_req_op_data_release(struct ceph_osd_request *osd_req,
370 unsigned int which)
371{
372 struct ceph_osd_req_op *op;
373
374 BUG_ON(which >= osd_req->r_num_ops);
375 op = &osd_req->r_ops[which];
376
377 switch (op->op) {
378 case CEPH_OSD_OP_READ:
379 case CEPH_OSD_OP_WRITE:
380 case CEPH_OSD_OP_WRITEFULL:
381 ceph_osd_data_release(&op->extent.osd_data);
382 break;
383 case CEPH_OSD_OP_CALL:
384 ceph_osd_data_release(&op->cls.request_info);
385 ceph_osd_data_release(&op->cls.request_data);
386 ceph_osd_data_release(&op->cls.response_data);
387 break;
388 case CEPH_OSD_OP_SETXATTR:
389 case CEPH_OSD_OP_CMPXATTR:
390 ceph_osd_data_release(&op->xattr.osd_data);
391 break;
392 case CEPH_OSD_OP_STAT:
393 ceph_osd_data_release(&op->raw_data_in);
394 break;
395 case CEPH_OSD_OP_NOTIFY_ACK:
396 ceph_osd_data_release(&op->notify_ack.request_data);
397 break;
398 case CEPH_OSD_OP_NOTIFY:
399 ceph_osd_data_release(&op->notify.request_data);
400 ceph_osd_data_release(&op->notify.response_data);
401 break;
402 case CEPH_OSD_OP_LIST_WATCHERS:
403 ceph_osd_data_release(&op->list_watchers.response_data);
404 break;
405 case CEPH_OSD_OP_COPY_FROM2:
406 ceph_osd_data_release(&op->copy_from.osd_data);
407 break;
408 default:
409 break;
410 }
411}
412
413/*
414 * Assumes @t is zero-initialized.
415 */
416static void target_init(struct ceph_osd_request_target *t)
417{
418 ceph_oid_init(&t->base_oid);
419 ceph_oloc_init(&t->base_oloc);
420 ceph_oid_init(&t->target_oid);
421 ceph_oloc_init(&t->target_oloc);
422
423 ceph_osds_init(&t->acting);
424 ceph_osds_init(&t->up);
425 t->size = -1;
426 t->min_size = -1;
427
428 t->osd = CEPH_HOMELESS_OSD;
429}
430
431static void target_copy(struct ceph_osd_request_target *dest,
432 const struct ceph_osd_request_target *src)
433{
434 ceph_oid_copy(&dest->base_oid, &src->base_oid);
435 ceph_oloc_copy(&dest->base_oloc, &src->base_oloc);
436 ceph_oid_copy(&dest->target_oid, &src->target_oid);
437 ceph_oloc_copy(&dest->target_oloc, &src->target_oloc);
438
439 dest->pgid = src->pgid; /* struct */
440 dest->spgid = src->spgid; /* struct */
441 dest->pg_num = src->pg_num;
442 dest->pg_num_mask = src->pg_num_mask;
443 ceph_osds_copy(&dest->acting, &src->acting);
444 ceph_osds_copy(&dest->up, &src->up);
445 dest->size = src->size;
446 dest->min_size = src->min_size;
447 dest->sort_bitwise = src->sort_bitwise;
448 dest->recovery_deletes = src->recovery_deletes;
449
450 dest->flags = src->flags;
451 dest->used_replica = src->used_replica;
452 dest->paused = src->paused;
453
454 dest->epoch = src->epoch;
455 dest->last_force_resend = src->last_force_resend;
456
457 dest->osd = src->osd;
458}
459
460static void target_destroy(struct ceph_osd_request_target *t)
461{
462 ceph_oid_destroy(&t->base_oid);
463 ceph_oloc_destroy(&t->base_oloc);
464 ceph_oid_destroy(&t->target_oid);
465 ceph_oloc_destroy(&t->target_oloc);
466}
467
468/*
469 * requests
470 */
471static void request_release_checks(struct ceph_osd_request *req)
472{
473 WARN_ON(!RB_EMPTY_NODE(&req->r_node));
474 WARN_ON(!RB_EMPTY_NODE(&req->r_mc_node));
475 WARN_ON(!list_empty(&req->r_private_item));
476 WARN_ON(req->r_osd);
477}
478
479static void ceph_osdc_release_request(struct kref *kref)
480{
481 struct ceph_osd_request *req = container_of(kref,
482 struct ceph_osd_request, r_kref);
483 unsigned int which;
484
485 dout("%s %p (r_request %p r_reply %p)\n", __func__, req,
486 req->r_request, req->r_reply);
487 request_release_checks(req);
488
489 if (req->r_request)
490 ceph_msg_put(req->r_request);
491 if (req->r_reply)
492 ceph_msg_put(req->r_reply);
493
494 for (which = 0; which < req->r_num_ops; which++)
495 osd_req_op_data_release(req, which);
496
497 target_destroy(&req->r_t);
498 ceph_put_snap_context(req->r_snapc);
499
500 if (req->r_mempool)
501 mempool_free(req, req->r_osdc->req_mempool);
502 else if (req->r_num_ops <= CEPH_OSD_SLAB_OPS)
503 kmem_cache_free(ceph_osd_request_cache, req);
504 else
505 kfree(req);
506}
507
508void ceph_osdc_get_request(struct ceph_osd_request *req)
509{
510 dout("%s %p (was %d)\n", __func__, req,
511 kref_read(&req->r_kref));
512 kref_get(&req->r_kref);
513}
514EXPORT_SYMBOL(ceph_osdc_get_request);
515
516void ceph_osdc_put_request(struct ceph_osd_request *req)
517{
518 if (req) {
519 dout("%s %p (was %d)\n", __func__, req,
520 kref_read(&req->r_kref));
521 kref_put(&req->r_kref, ceph_osdc_release_request);
522 }
523}
524EXPORT_SYMBOL(ceph_osdc_put_request);
525
526static void request_init(struct ceph_osd_request *req)
527{
528 /* req only, each op is zeroed in osd_req_op_init() */
529 memset(req, 0, sizeof(*req));
530
531 kref_init(&req->r_kref);
532 init_completion(&req->r_completion);
533 RB_CLEAR_NODE(&req->r_node);
534 RB_CLEAR_NODE(&req->r_mc_node);
535 INIT_LIST_HEAD(&req->r_private_item);
536
537 target_init(&req->r_t);
538}
539
540/*
541 * This is ugly, but it allows us to reuse linger registration and ping
542 * requests, keeping the structure of the code around send_linger{_ping}()
543 * reasonable. Setting up a min_nr=2 mempool for each linger request
544 * and dealing with copying ops (this blasts req only, watch op remains
545 * intact) isn't any better.
546 */
547static void request_reinit(struct ceph_osd_request *req)
548{
549 struct ceph_osd_client *osdc = req->r_osdc;
550 bool mempool = req->r_mempool;
551 unsigned int num_ops = req->r_num_ops;
552 u64 snapid = req->r_snapid;
553 struct ceph_snap_context *snapc = req->r_snapc;
554 bool linger = req->r_linger;
555 struct ceph_msg *request_msg = req->r_request;
556 struct ceph_msg *reply_msg = req->r_reply;
557
558 dout("%s req %p\n", __func__, req);
559 WARN_ON(kref_read(&req->r_kref) != 1);
560 request_release_checks(req);
561
562 WARN_ON(kref_read(&request_msg->kref) != 1);
563 WARN_ON(kref_read(&reply_msg->kref) != 1);
564 target_destroy(&req->r_t);
565
566 request_init(req);
567 req->r_osdc = osdc;
568 req->r_mempool = mempool;
569 req->r_num_ops = num_ops;
570 req->r_snapid = snapid;
571 req->r_snapc = snapc;
572 req->r_linger = linger;
573 req->r_request = request_msg;
574 req->r_reply = reply_msg;
575}
576
577struct ceph_osd_request *ceph_osdc_alloc_request(struct ceph_osd_client *osdc,
578 struct ceph_snap_context *snapc,
579 unsigned int num_ops,
580 bool use_mempool,
581 gfp_t gfp_flags)
582{
583 struct ceph_osd_request *req;
584
585 if (use_mempool) {
586 BUG_ON(num_ops > CEPH_OSD_SLAB_OPS);
587 req = mempool_alloc(osdc->req_mempool, gfp_flags);
588 } else if (num_ops <= CEPH_OSD_SLAB_OPS) {
589 req = kmem_cache_alloc(ceph_osd_request_cache, gfp_flags);
590 } else {
591 BUG_ON(num_ops > CEPH_OSD_MAX_OPS);
592 req = kmalloc(struct_size(req, r_ops, num_ops), gfp_flags);
593 }
594 if (unlikely(!req))
595 return NULL;
596
597 request_init(req);
598 req->r_osdc = osdc;
599 req->r_mempool = use_mempool;
600 req->r_num_ops = num_ops;
601 req->r_snapid = CEPH_NOSNAP;
602 req->r_snapc = ceph_get_snap_context(snapc);
603
604 dout("%s req %p\n", __func__, req);
605 return req;
606}
607EXPORT_SYMBOL(ceph_osdc_alloc_request);
608
609static int ceph_oloc_encoding_size(const struct ceph_object_locator *oloc)
610{
611 return 8 + 4 + 4 + 4 + (oloc->pool_ns ? oloc->pool_ns->len : 0);
612}
613
614static int __ceph_osdc_alloc_messages(struct ceph_osd_request *req, gfp_t gfp,
615 int num_request_data_items,
616 int num_reply_data_items)
617{
618 struct ceph_osd_client *osdc = req->r_osdc;
619 struct ceph_msg *msg;
620 int msg_size;
621
622 WARN_ON(req->r_request || req->r_reply);
623 WARN_ON(ceph_oid_empty(&req->r_base_oid));
624 WARN_ON(ceph_oloc_empty(&req->r_base_oloc));
625
626 /* create request message */
627 msg_size = CEPH_ENCODING_START_BLK_LEN +
628 CEPH_PGID_ENCODING_LEN + 1; /* spgid */
629 msg_size += 4 + 4 + 4; /* hash, osdmap_epoch, flags */
630 msg_size += CEPH_ENCODING_START_BLK_LEN +
631 sizeof(struct ceph_osd_reqid); /* reqid */
632 msg_size += sizeof(struct ceph_blkin_trace_info); /* trace */
633 msg_size += 4 + sizeof(struct ceph_timespec); /* client_inc, mtime */
634 msg_size += CEPH_ENCODING_START_BLK_LEN +
635 ceph_oloc_encoding_size(&req->r_base_oloc); /* oloc */
636 msg_size += 4 + req->r_base_oid.name_len; /* oid */
637 msg_size += 2 + req->r_num_ops * sizeof(struct ceph_osd_op);
638 msg_size += 8; /* snapid */
639 msg_size += 8; /* snap_seq */
640 msg_size += 4 + 8 * (req->r_snapc ? req->r_snapc->num_snaps : 0);
641 msg_size += 4 + 8; /* retry_attempt, features */
642
643 if (req->r_mempool)
644 msg = ceph_msgpool_get(&osdc->msgpool_op, msg_size,
645 num_request_data_items);
646 else
647 msg = ceph_msg_new2(CEPH_MSG_OSD_OP, msg_size,
648 num_request_data_items, gfp, true);
649 if (!msg)
650 return -ENOMEM;
651
652 memset(msg->front.iov_base, 0, msg->front.iov_len);
653 req->r_request = msg;
654
655 /* create reply message */
656 msg_size = OSD_OPREPLY_FRONT_LEN;
657 msg_size += req->r_base_oid.name_len;
658 msg_size += req->r_num_ops * sizeof(struct ceph_osd_op);
659
660 if (req->r_mempool)
661 msg = ceph_msgpool_get(&osdc->msgpool_op_reply, msg_size,
662 num_reply_data_items);
663 else
664 msg = ceph_msg_new2(CEPH_MSG_OSD_OPREPLY, msg_size,
665 num_reply_data_items, gfp, true);
666 if (!msg)
667 return -ENOMEM;
668
669 req->r_reply = msg;
670
671 return 0;
672}
673
674static bool osd_req_opcode_valid(u16 opcode)
675{
676 switch (opcode) {
677#define GENERATE_CASE(op, opcode, str) case CEPH_OSD_OP_##op: return true;
678__CEPH_FORALL_OSD_OPS(GENERATE_CASE)
679#undef GENERATE_CASE
680 default:
681 return false;
682 }
683}
684
685static void get_num_data_items(struct ceph_osd_request *req,
686 int *num_request_data_items,
687 int *num_reply_data_items)
688{
689 struct ceph_osd_req_op *op;
690
691 *num_request_data_items = 0;
692 *num_reply_data_items = 0;
693
694 for (op = req->r_ops; op != &req->r_ops[req->r_num_ops]; op++) {
695 switch (op->op) {
696 /* request */
697 case CEPH_OSD_OP_WRITE:
698 case CEPH_OSD_OP_WRITEFULL:
699 case CEPH_OSD_OP_SETXATTR:
700 case CEPH_OSD_OP_CMPXATTR:
701 case CEPH_OSD_OP_NOTIFY_ACK:
702 case CEPH_OSD_OP_COPY_FROM2:
703 *num_request_data_items += 1;
704 break;
705
706 /* reply */
707 case CEPH_OSD_OP_STAT:
708 case CEPH_OSD_OP_READ:
709 case CEPH_OSD_OP_LIST_WATCHERS:
710 *num_reply_data_items += 1;
711 break;
712
713 /* both */
714 case CEPH_OSD_OP_NOTIFY:
715 *num_request_data_items += 1;
716 *num_reply_data_items += 1;
717 break;
718 case CEPH_OSD_OP_CALL:
719 *num_request_data_items += 2;
720 *num_reply_data_items += 1;
721 break;
722
723 default:
724 WARN_ON(!osd_req_opcode_valid(op->op));
725 break;
726 }
727 }
728}
729
730/*
731 * oid, oloc and OSD op opcode(s) must be filled in before this function
732 * is called.
733 */
734int ceph_osdc_alloc_messages(struct ceph_osd_request *req, gfp_t gfp)
735{
736 int num_request_data_items, num_reply_data_items;
737
738 get_num_data_items(req, &num_request_data_items, &num_reply_data_items);
739 return __ceph_osdc_alloc_messages(req, gfp, num_request_data_items,
740 num_reply_data_items);
741}
742EXPORT_SYMBOL(ceph_osdc_alloc_messages);
743
744/*
745 * This is an osd op init function for opcodes that have no data or
746 * other information associated with them. It also serves as a
747 * common init routine for all the other init functions, below.
748 */
749struct ceph_osd_req_op *
750osd_req_op_init(struct ceph_osd_request *osd_req, unsigned int which,
751 u16 opcode, u32 flags)
752{
753 struct ceph_osd_req_op *op;
754
755 BUG_ON(which >= osd_req->r_num_ops);
756 BUG_ON(!osd_req_opcode_valid(opcode));
757
758 op = &osd_req->r_ops[which];
759 memset(op, 0, sizeof (*op));
760 op->op = opcode;
761 op->flags = flags;
762
763 return op;
764}
765EXPORT_SYMBOL(osd_req_op_init);
766
767void osd_req_op_extent_init(struct ceph_osd_request *osd_req,
768 unsigned int which, u16 opcode,
769 u64 offset, u64 length,
770 u64 truncate_size, u32 truncate_seq)
771{
772 struct ceph_osd_req_op *op = osd_req_op_init(osd_req, which,
773 opcode, 0);
774 size_t payload_len = 0;
775
776 BUG_ON(opcode != CEPH_OSD_OP_READ && opcode != CEPH_OSD_OP_WRITE &&
777 opcode != CEPH_OSD_OP_WRITEFULL && opcode != CEPH_OSD_OP_ZERO &&
778 opcode != CEPH_OSD_OP_TRUNCATE);
779
780 op->extent.offset = offset;
781 op->extent.length = length;
782 op->extent.truncate_size = truncate_size;
783 op->extent.truncate_seq = truncate_seq;
784 if (opcode == CEPH_OSD_OP_WRITE || opcode == CEPH_OSD_OP_WRITEFULL)
785 payload_len += length;
786
787 op->indata_len = payload_len;
788}
789EXPORT_SYMBOL(osd_req_op_extent_init);
790
791void osd_req_op_extent_update(struct ceph_osd_request *osd_req,
792 unsigned int which, u64 length)
793{
794 struct ceph_osd_req_op *op;
795 u64 previous;
796
797 BUG_ON(which >= osd_req->r_num_ops);
798 op = &osd_req->r_ops[which];
799 previous = op->extent.length;
800
801 if (length == previous)
802 return; /* Nothing to do */
803 BUG_ON(length > previous);
804
805 op->extent.length = length;
806 if (op->op == CEPH_OSD_OP_WRITE || op->op == CEPH_OSD_OP_WRITEFULL)
807 op->indata_len -= previous - length;
808}
809EXPORT_SYMBOL(osd_req_op_extent_update);
810
811void osd_req_op_extent_dup_last(struct ceph_osd_request *osd_req,
812 unsigned int which, u64 offset_inc)
813{
814 struct ceph_osd_req_op *op, *prev_op;
815
816 BUG_ON(which + 1 >= osd_req->r_num_ops);
817
818 prev_op = &osd_req->r_ops[which];
819 op = osd_req_op_init(osd_req, which + 1, prev_op->op, prev_op->flags);
820 /* dup previous one */
821 op->indata_len = prev_op->indata_len;
822 op->outdata_len = prev_op->outdata_len;
823 op->extent = prev_op->extent;
824 /* adjust offset */
825 op->extent.offset += offset_inc;
826 op->extent.length -= offset_inc;
827
828 if (op->op == CEPH_OSD_OP_WRITE || op->op == CEPH_OSD_OP_WRITEFULL)
829 op->indata_len -= offset_inc;
830}
831EXPORT_SYMBOL(osd_req_op_extent_dup_last);
832
833int osd_req_op_cls_init(struct ceph_osd_request *osd_req, unsigned int which,
834 const char *class, const char *method)
835{
836 struct ceph_osd_req_op *op;
837 struct ceph_pagelist *pagelist;
838 size_t payload_len = 0;
839 size_t size;
840 int ret;
841
842 op = osd_req_op_init(osd_req, which, CEPH_OSD_OP_CALL, 0);
843
844 pagelist = ceph_pagelist_alloc(GFP_NOFS);
845 if (!pagelist)
846 return -ENOMEM;
847
848 op->cls.class_name = class;
849 size = strlen(class);
850 BUG_ON(size > (size_t) U8_MAX);
851 op->cls.class_len = size;
852 ret = ceph_pagelist_append(pagelist, class, size);
853 if (ret)
854 goto err_pagelist_free;
855 payload_len += size;
856
857 op->cls.method_name = method;
858 size = strlen(method);
859 BUG_ON(size > (size_t) U8_MAX);
860 op->cls.method_len = size;
861 ret = ceph_pagelist_append(pagelist, method, size);
862 if (ret)
863 goto err_pagelist_free;
864 payload_len += size;
865
866 osd_req_op_cls_request_info_pagelist(osd_req, which, pagelist);
867 op->indata_len = payload_len;
868 return 0;
869
870err_pagelist_free:
871 ceph_pagelist_release(pagelist);
872 return ret;
873}
874EXPORT_SYMBOL(osd_req_op_cls_init);
875
876int osd_req_op_xattr_init(struct ceph_osd_request *osd_req, unsigned int which,
877 u16 opcode, const char *name, const void *value,
878 size_t size, u8 cmp_op, u8 cmp_mode)
879{
880 struct ceph_osd_req_op *op = osd_req_op_init(osd_req, which,
881 opcode, 0);
882 struct ceph_pagelist *pagelist;
883 size_t payload_len;
884 int ret;
885
886 BUG_ON(opcode != CEPH_OSD_OP_SETXATTR && opcode != CEPH_OSD_OP_CMPXATTR);
887
888 pagelist = ceph_pagelist_alloc(GFP_NOFS);
889 if (!pagelist)
890 return -ENOMEM;
891
892 payload_len = strlen(name);
893 op->xattr.name_len = payload_len;
894 ret = ceph_pagelist_append(pagelist, name, payload_len);
895 if (ret)
896 goto err_pagelist_free;
897
898 op->xattr.value_len = size;
899 ret = ceph_pagelist_append(pagelist, value, size);
900 if (ret)
901 goto err_pagelist_free;
902 payload_len += size;
903
904 op->xattr.cmp_op = cmp_op;
905 op->xattr.cmp_mode = cmp_mode;
906
907 ceph_osd_data_pagelist_init(&op->xattr.osd_data, pagelist);
908 op->indata_len = payload_len;
909 return 0;
910
911err_pagelist_free:
912 ceph_pagelist_release(pagelist);
913 return ret;
914}
915EXPORT_SYMBOL(osd_req_op_xattr_init);
916
917/*
918 * @watch_opcode: CEPH_OSD_WATCH_OP_*
919 */
920static void osd_req_op_watch_init(struct ceph_osd_request *req, int which,
921 u64 cookie, u8 watch_opcode)
922{
923 struct ceph_osd_req_op *op;
924
925 op = osd_req_op_init(req, which, CEPH_OSD_OP_WATCH, 0);
926 op->watch.cookie = cookie;
927 op->watch.op = watch_opcode;
928 op->watch.gen = 0;
929}
930
931/*
932 * @flags: CEPH_OSD_OP_ALLOC_HINT_FLAG_*
933 */
934void osd_req_op_alloc_hint_init(struct ceph_osd_request *osd_req,
935 unsigned int which,
936 u64 expected_object_size,
937 u64 expected_write_size,
938 u32 flags)
939{
940 struct ceph_osd_req_op *op;
941
942 op = osd_req_op_init(osd_req, which, CEPH_OSD_OP_SETALLOCHINT, 0);
943 op->alloc_hint.expected_object_size = expected_object_size;
944 op->alloc_hint.expected_write_size = expected_write_size;
945 op->alloc_hint.flags = flags;
946
947 /*
948 * CEPH_OSD_OP_SETALLOCHINT op is advisory and therefore deemed
949 * not worth a feature bit. Set FAILOK per-op flag to make
950 * sure older osds don't trip over an unsupported opcode.
951 */
952 op->flags |= CEPH_OSD_OP_FLAG_FAILOK;
953}
954EXPORT_SYMBOL(osd_req_op_alloc_hint_init);
955
956static void ceph_osdc_msg_data_add(struct ceph_msg *msg,
957 struct ceph_osd_data *osd_data)
958{
959 u64 length = ceph_osd_data_length(osd_data);
960
961 if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGES) {
962 BUG_ON(length > (u64) SIZE_MAX);
963 if (length)
964 ceph_msg_data_add_pages(msg, osd_data->pages,
965 length, osd_data->alignment, false);
966 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGELIST) {
967 BUG_ON(!length);
968 ceph_msg_data_add_pagelist(msg, osd_data->pagelist);
969#ifdef CONFIG_BLOCK
970 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_BIO) {
971 ceph_msg_data_add_bio(msg, &osd_data->bio_pos, length);
972#endif
973 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_BVECS) {
974 ceph_msg_data_add_bvecs(msg, &osd_data->bvec_pos);
975 } else {
976 BUG_ON(osd_data->type != CEPH_OSD_DATA_TYPE_NONE);
977 }
978}
979
980static u32 osd_req_encode_op(struct ceph_osd_op *dst,
981 const struct ceph_osd_req_op *src)
982{
983 switch (src->op) {
984 case CEPH_OSD_OP_STAT:
985 break;
986 case CEPH_OSD_OP_READ:
987 case CEPH_OSD_OP_WRITE:
988 case CEPH_OSD_OP_WRITEFULL:
989 case CEPH_OSD_OP_ZERO:
990 case CEPH_OSD_OP_TRUNCATE:
991 dst->extent.offset = cpu_to_le64(src->extent.offset);
992 dst->extent.length = cpu_to_le64(src->extent.length);
993 dst->extent.truncate_size =
994 cpu_to_le64(src->extent.truncate_size);
995 dst->extent.truncate_seq =
996 cpu_to_le32(src->extent.truncate_seq);
997 break;
998 case CEPH_OSD_OP_CALL:
999 dst->cls.class_len = src->cls.class_len;
1000 dst->cls.method_len = src->cls.method_len;
1001 dst->cls.indata_len = cpu_to_le32(src->cls.indata_len);
1002 break;
1003 case CEPH_OSD_OP_WATCH:
1004 dst->watch.cookie = cpu_to_le64(src->watch.cookie);
1005 dst->watch.ver = cpu_to_le64(0);
1006 dst->watch.op = src->watch.op;
1007 dst->watch.gen = cpu_to_le32(src->watch.gen);
1008 break;
1009 case CEPH_OSD_OP_NOTIFY_ACK:
1010 break;
1011 case CEPH_OSD_OP_NOTIFY:
1012 dst->notify.cookie = cpu_to_le64(src->notify.cookie);
1013 break;
1014 case CEPH_OSD_OP_LIST_WATCHERS:
1015 break;
1016 case CEPH_OSD_OP_SETALLOCHINT:
1017 dst->alloc_hint.expected_object_size =
1018 cpu_to_le64(src->alloc_hint.expected_object_size);
1019 dst->alloc_hint.expected_write_size =
1020 cpu_to_le64(src->alloc_hint.expected_write_size);
1021 dst->alloc_hint.flags = cpu_to_le32(src->alloc_hint.flags);
1022 break;
1023 case CEPH_OSD_OP_SETXATTR:
1024 case CEPH_OSD_OP_CMPXATTR:
1025 dst->xattr.name_len = cpu_to_le32(src->xattr.name_len);
1026 dst->xattr.value_len = cpu_to_le32(src->xattr.value_len);
1027 dst->xattr.cmp_op = src->xattr.cmp_op;
1028 dst->xattr.cmp_mode = src->xattr.cmp_mode;
1029 break;
1030 case CEPH_OSD_OP_CREATE:
1031 case CEPH_OSD_OP_DELETE:
1032 break;
1033 case CEPH_OSD_OP_COPY_FROM2:
1034 dst->copy_from.snapid = cpu_to_le64(src->copy_from.snapid);
1035 dst->copy_from.src_version =
1036 cpu_to_le64(src->copy_from.src_version);
1037 dst->copy_from.flags = src->copy_from.flags;
1038 dst->copy_from.src_fadvise_flags =
1039 cpu_to_le32(src->copy_from.src_fadvise_flags);
1040 break;
1041 default:
1042 pr_err("unsupported osd opcode %s\n",
1043 ceph_osd_op_name(src->op));
1044 WARN_ON(1);
1045
1046 return 0;
1047 }
1048
1049 dst->op = cpu_to_le16(src->op);
1050 dst->flags = cpu_to_le32(src->flags);
1051 dst->payload_len = cpu_to_le32(src->indata_len);
1052
1053 return src->indata_len;
1054}
1055
1056/*
1057 * build new request AND message, calculate layout, and adjust file
1058 * extent as needed.
1059 *
1060 * if the file was recently truncated, we include information about its
1061 * old and new size so that the object can be updated appropriately. (we
1062 * avoid synchronously deleting truncated objects because it's slow.)
1063 */
1064struct ceph_osd_request *ceph_osdc_new_request(struct ceph_osd_client *osdc,
1065 struct ceph_file_layout *layout,
1066 struct ceph_vino vino,
1067 u64 off, u64 *plen,
1068 unsigned int which, int num_ops,
1069 int opcode, int flags,
1070 struct ceph_snap_context *snapc,
1071 u32 truncate_seq,
1072 u64 truncate_size,
1073 bool use_mempool)
1074{
1075 struct ceph_osd_request *req;
1076 u64 objnum = 0;
1077 u64 objoff = 0;
1078 u64 objlen = 0;
1079 int r;
1080
1081 BUG_ON(opcode != CEPH_OSD_OP_READ && opcode != CEPH_OSD_OP_WRITE &&
1082 opcode != CEPH_OSD_OP_ZERO && opcode != CEPH_OSD_OP_TRUNCATE &&
1083 opcode != CEPH_OSD_OP_CREATE && opcode != CEPH_OSD_OP_DELETE);
1084
1085 req = ceph_osdc_alloc_request(osdc, snapc, num_ops, use_mempool,
1086 GFP_NOFS);
1087 if (!req) {
1088 r = -ENOMEM;
1089 goto fail;
1090 }
1091
1092 /* calculate max write size */
1093 r = calc_layout(layout, off, plen, &objnum, &objoff, &objlen);
1094 if (r)
1095 goto fail;
1096
1097 if (opcode == CEPH_OSD_OP_CREATE || opcode == CEPH_OSD_OP_DELETE) {
1098 osd_req_op_init(req, which, opcode, 0);
1099 } else {
1100 u32 object_size = layout->object_size;
1101 u32 object_base = off - objoff;
1102 if (!(truncate_seq == 1 && truncate_size == -1ULL)) {
1103 if (truncate_size <= object_base) {
1104 truncate_size = 0;
1105 } else {
1106 truncate_size -= object_base;
1107 if (truncate_size > object_size)
1108 truncate_size = object_size;
1109 }
1110 }
1111 osd_req_op_extent_init(req, which, opcode, objoff, objlen,
1112 truncate_size, truncate_seq);
1113 }
1114
1115 req->r_base_oloc.pool = layout->pool_id;
1116 req->r_base_oloc.pool_ns = ceph_try_get_string(layout->pool_ns);
1117 ceph_oid_printf(&req->r_base_oid, "%llx.%08llx", vino.ino, objnum);
1118 req->r_flags = flags | osdc->client->options->read_from_replica;
1119
1120 req->r_snapid = vino.snap;
1121 if (flags & CEPH_OSD_FLAG_WRITE)
1122 req->r_data_offset = off;
1123
1124 if (num_ops > 1)
1125 /*
1126 * This is a special case for ceph_writepages_start(), but it
1127 * also covers ceph_uninline_data(). If more multi-op request
1128 * use cases emerge, we will need a separate helper.
1129 */
1130 r = __ceph_osdc_alloc_messages(req, GFP_NOFS, num_ops, 0);
1131 else
1132 r = ceph_osdc_alloc_messages(req, GFP_NOFS);
1133 if (r)
1134 goto fail;
1135
1136 return req;
1137
1138fail:
1139 ceph_osdc_put_request(req);
1140 return ERR_PTR(r);
1141}
1142EXPORT_SYMBOL(ceph_osdc_new_request);
1143
1144/*
1145 * We keep osd requests in an rbtree, sorted by ->r_tid.
1146 */
1147DEFINE_RB_FUNCS(request, struct ceph_osd_request, r_tid, r_node)
1148DEFINE_RB_FUNCS(request_mc, struct ceph_osd_request, r_tid, r_mc_node)
1149
1150/*
1151 * Call @fn on each OSD request as long as @fn returns 0.
1152 */
1153static void for_each_request(struct ceph_osd_client *osdc,
1154 int (*fn)(struct ceph_osd_request *req, void *arg),
1155 void *arg)
1156{
1157 struct rb_node *n, *p;
1158
1159 for (n = rb_first(&osdc->osds); n; n = rb_next(n)) {
1160 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
1161
1162 for (p = rb_first(&osd->o_requests); p; ) {
1163 struct ceph_osd_request *req =
1164 rb_entry(p, struct ceph_osd_request, r_node);
1165
1166 p = rb_next(p);
1167 if (fn(req, arg))
1168 return;
1169 }
1170 }
1171
1172 for (p = rb_first(&osdc->homeless_osd.o_requests); p; ) {
1173 struct ceph_osd_request *req =
1174 rb_entry(p, struct ceph_osd_request, r_node);
1175
1176 p = rb_next(p);
1177 if (fn(req, arg))
1178 return;
1179 }
1180}
1181
1182static bool osd_homeless(struct ceph_osd *osd)
1183{
1184 return osd->o_osd == CEPH_HOMELESS_OSD;
1185}
1186
1187static bool osd_registered(struct ceph_osd *osd)
1188{
1189 verify_osdc_locked(osd->o_osdc);
1190
1191 return !RB_EMPTY_NODE(&osd->o_node);
1192}
1193
1194/*
1195 * Assumes @osd is zero-initialized.
1196 */
1197static void osd_init(struct ceph_osd *osd)
1198{
1199 refcount_set(&osd->o_ref, 1);
1200 RB_CLEAR_NODE(&osd->o_node);
1201 osd->o_requests = RB_ROOT;
1202 osd->o_linger_requests = RB_ROOT;
1203 osd->o_backoff_mappings = RB_ROOT;
1204 osd->o_backoffs_by_id = RB_ROOT;
1205 INIT_LIST_HEAD(&osd->o_osd_lru);
1206 INIT_LIST_HEAD(&osd->o_keepalive_item);
1207 osd->o_incarnation = 1;
1208 mutex_init(&osd->lock);
1209}
1210
1211static void osd_cleanup(struct ceph_osd *osd)
1212{
1213 WARN_ON(!RB_EMPTY_NODE(&osd->o_node));
1214 WARN_ON(!RB_EMPTY_ROOT(&osd->o_requests));
1215 WARN_ON(!RB_EMPTY_ROOT(&osd->o_linger_requests));
1216 WARN_ON(!RB_EMPTY_ROOT(&osd->o_backoff_mappings));
1217 WARN_ON(!RB_EMPTY_ROOT(&osd->o_backoffs_by_id));
1218 WARN_ON(!list_empty(&osd->o_osd_lru));
1219 WARN_ON(!list_empty(&osd->o_keepalive_item));
1220
1221 if (osd->o_auth.authorizer) {
1222 WARN_ON(osd_homeless(osd));
1223 ceph_auth_destroy_authorizer(osd->o_auth.authorizer);
1224 }
1225}
1226
1227/*
1228 * Track open sessions with osds.
1229 */
1230static struct ceph_osd *create_osd(struct ceph_osd_client *osdc, int onum)
1231{
1232 struct ceph_osd *osd;
1233
1234 WARN_ON(onum == CEPH_HOMELESS_OSD);
1235
1236 osd = kzalloc(sizeof(*osd), GFP_NOIO | __GFP_NOFAIL);
1237 osd_init(osd);
1238 osd->o_osdc = osdc;
1239 osd->o_osd = onum;
1240
1241 ceph_con_init(&osd->o_con, osd, &osd_con_ops, &osdc->client->msgr);
1242
1243 return osd;
1244}
1245
1246static struct ceph_osd *get_osd(struct ceph_osd *osd)
1247{
1248 if (refcount_inc_not_zero(&osd->o_ref)) {
1249 dout("get_osd %p %d -> %d\n", osd, refcount_read(&osd->o_ref)-1,
1250 refcount_read(&osd->o_ref));
1251 return osd;
1252 } else {
1253 dout("get_osd %p FAIL\n", osd);
1254 return NULL;
1255 }
1256}
1257
1258static void put_osd(struct ceph_osd *osd)
1259{
1260 dout("put_osd %p %d -> %d\n", osd, refcount_read(&osd->o_ref),
1261 refcount_read(&osd->o_ref) - 1);
1262 if (refcount_dec_and_test(&osd->o_ref)) {
1263 osd_cleanup(osd);
1264 kfree(osd);
1265 }
1266}
1267
1268DEFINE_RB_FUNCS(osd, struct ceph_osd, o_osd, o_node)
1269
1270static void __move_osd_to_lru(struct ceph_osd *osd)
1271{
1272 struct ceph_osd_client *osdc = osd->o_osdc;
1273
1274 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1275 BUG_ON(!list_empty(&osd->o_osd_lru));
1276
1277 spin_lock(&osdc->osd_lru_lock);
1278 list_add_tail(&osd->o_osd_lru, &osdc->osd_lru);
1279 spin_unlock(&osdc->osd_lru_lock);
1280
1281 osd->lru_ttl = jiffies + osdc->client->options->osd_idle_ttl;
1282}
1283
1284static void maybe_move_osd_to_lru(struct ceph_osd *osd)
1285{
1286 if (RB_EMPTY_ROOT(&osd->o_requests) &&
1287 RB_EMPTY_ROOT(&osd->o_linger_requests))
1288 __move_osd_to_lru(osd);
1289}
1290
1291static void __remove_osd_from_lru(struct ceph_osd *osd)
1292{
1293 struct ceph_osd_client *osdc = osd->o_osdc;
1294
1295 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1296
1297 spin_lock(&osdc->osd_lru_lock);
1298 if (!list_empty(&osd->o_osd_lru))
1299 list_del_init(&osd->o_osd_lru);
1300 spin_unlock(&osdc->osd_lru_lock);
1301}
1302
1303/*
1304 * Close the connection and assign any leftover requests to the
1305 * homeless session.
1306 */
1307static void close_osd(struct ceph_osd *osd)
1308{
1309 struct ceph_osd_client *osdc = osd->o_osdc;
1310 struct rb_node *n;
1311
1312 verify_osdc_wrlocked(osdc);
1313 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1314
1315 ceph_con_close(&osd->o_con);
1316
1317 for (n = rb_first(&osd->o_requests); n; ) {
1318 struct ceph_osd_request *req =
1319 rb_entry(n, struct ceph_osd_request, r_node);
1320
1321 n = rb_next(n); /* unlink_request() */
1322
1323 dout(" reassigning req %p tid %llu\n", req, req->r_tid);
1324 unlink_request(osd, req);
1325 link_request(&osdc->homeless_osd, req);
1326 }
1327 for (n = rb_first(&osd->o_linger_requests); n; ) {
1328 struct ceph_osd_linger_request *lreq =
1329 rb_entry(n, struct ceph_osd_linger_request, node);
1330
1331 n = rb_next(n); /* unlink_linger() */
1332
1333 dout(" reassigning lreq %p linger_id %llu\n", lreq,
1334 lreq->linger_id);
1335 unlink_linger(osd, lreq);
1336 link_linger(&osdc->homeless_osd, lreq);
1337 }
1338 clear_backoffs(osd);
1339
1340 __remove_osd_from_lru(osd);
1341 erase_osd(&osdc->osds, osd);
1342 put_osd(osd);
1343}
1344
1345/*
1346 * reset osd connect
1347 */
1348static int reopen_osd(struct ceph_osd *osd)
1349{
1350 struct ceph_entity_addr *peer_addr;
1351
1352 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1353
1354 if (RB_EMPTY_ROOT(&osd->o_requests) &&
1355 RB_EMPTY_ROOT(&osd->o_linger_requests)) {
1356 close_osd(osd);
1357 return -ENODEV;
1358 }
1359
1360 peer_addr = &osd->o_osdc->osdmap->osd_addr[osd->o_osd];
1361 if (!memcmp(peer_addr, &osd->o_con.peer_addr, sizeof (*peer_addr)) &&
1362 !ceph_con_opened(&osd->o_con)) {
1363 struct rb_node *n;
1364
1365 dout("osd addr hasn't changed and connection never opened, "
1366 "letting msgr retry\n");
1367 /* touch each r_stamp for handle_timeout()'s benfit */
1368 for (n = rb_first(&osd->o_requests); n; n = rb_next(n)) {
1369 struct ceph_osd_request *req =
1370 rb_entry(n, struct ceph_osd_request, r_node);
1371 req->r_stamp = jiffies;
1372 }
1373
1374 return -EAGAIN;
1375 }
1376
1377 ceph_con_close(&osd->o_con);
1378 ceph_con_open(&osd->o_con, CEPH_ENTITY_TYPE_OSD, osd->o_osd, peer_addr);
1379 osd->o_incarnation++;
1380
1381 return 0;
1382}
1383
1384static struct ceph_osd *lookup_create_osd(struct ceph_osd_client *osdc, int o,
1385 bool wrlocked)
1386{
1387 struct ceph_osd *osd;
1388
1389 if (wrlocked)
1390 verify_osdc_wrlocked(osdc);
1391 else
1392 verify_osdc_locked(osdc);
1393
1394 if (o != CEPH_HOMELESS_OSD)
1395 osd = lookup_osd(&osdc->osds, o);
1396 else
1397 osd = &osdc->homeless_osd;
1398 if (!osd) {
1399 if (!wrlocked)
1400 return ERR_PTR(-EAGAIN);
1401
1402 osd = create_osd(osdc, o);
1403 insert_osd(&osdc->osds, osd);
1404 ceph_con_open(&osd->o_con, CEPH_ENTITY_TYPE_OSD, osd->o_osd,
1405 &osdc->osdmap->osd_addr[osd->o_osd]);
1406 }
1407
1408 dout("%s osdc %p osd%d -> osd %p\n", __func__, osdc, o, osd);
1409 return osd;
1410}
1411
1412/*
1413 * Create request <-> OSD session relation.
1414 *
1415 * @req has to be assigned a tid, @osd may be homeless.
1416 */
1417static void link_request(struct ceph_osd *osd, struct ceph_osd_request *req)
1418{
1419 verify_osd_locked(osd);
1420 WARN_ON(!req->r_tid || req->r_osd);
1421 dout("%s osd %p osd%d req %p tid %llu\n", __func__, osd, osd->o_osd,
1422 req, req->r_tid);
1423
1424 if (!osd_homeless(osd))
1425 __remove_osd_from_lru(osd);
1426 else
1427 atomic_inc(&osd->o_osdc->num_homeless);
1428
1429 get_osd(osd);
1430 insert_request(&osd->o_requests, req);
1431 req->r_osd = osd;
1432}
1433
1434static void unlink_request(struct ceph_osd *osd, struct ceph_osd_request *req)
1435{
1436 verify_osd_locked(osd);
1437 WARN_ON(req->r_osd != osd);
1438 dout("%s osd %p osd%d req %p tid %llu\n", __func__, osd, osd->o_osd,
1439 req, req->r_tid);
1440
1441 req->r_osd = NULL;
1442 erase_request(&osd->o_requests, req);
1443 put_osd(osd);
1444
1445 if (!osd_homeless(osd))
1446 maybe_move_osd_to_lru(osd);
1447 else
1448 atomic_dec(&osd->o_osdc->num_homeless);
1449}
1450
1451static bool __pool_full(struct ceph_pg_pool_info *pi)
1452{
1453 return pi->flags & CEPH_POOL_FLAG_FULL;
1454}
1455
1456static bool have_pool_full(struct ceph_osd_client *osdc)
1457{
1458 struct rb_node *n;
1459
1460 for (n = rb_first(&osdc->osdmap->pg_pools); n; n = rb_next(n)) {
1461 struct ceph_pg_pool_info *pi =
1462 rb_entry(n, struct ceph_pg_pool_info, node);
1463
1464 if (__pool_full(pi))
1465 return true;
1466 }
1467
1468 return false;
1469}
1470
1471static bool pool_full(struct ceph_osd_client *osdc, s64 pool_id)
1472{
1473 struct ceph_pg_pool_info *pi;
1474
1475 pi = ceph_pg_pool_by_id(osdc->osdmap, pool_id);
1476 if (!pi)
1477 return false;
1478
1479 return __pool_full(pi);
1480}
1481
1482/*
1483 * Returns whether a request should be blocked from being sent
1484 * based on the current osdmap and osd_client settings.
1485 */
1486static bool target_should_be_paused(struct ceph_osd_client *osdc,
1487 const struct ceph_osd_request_target *t,
1488 struct ceph_pg_pool_info *pi)
1489{
1490 bool pauserd = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD);
1491 bool pausewr = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR) ||
1492 ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
1493 __pool_full(pi);
1494
1495 WARN_ON(pi->id != t->target_oloc.pool);
1496 return ((t->flags & CEPH_OSD_FLAG_READ) && pauserd) ||
1497 ((t->flags & CEPH_OSD_FLAG_WRITE) && pausewr) ||
1498 (osdc->osdmap->epoch < osdc->epoch_barrier);
1499}
1500
1501static int pick_random_replica(const struct ceph_osds *acting)
1502{
1503 int i = prandom_u32() % acting->size;
1504
1505 dout("%s picked osd%d, primary osd%d\n", __func__,
1506 acting->osds[i], acting->primary);
1507 return i;
1508}
1509
1510/*
1511 * Picks the closest replica based on client's location given by
1512 * crush_location option. Prefers the primary if the locality is
1513 * the same.
1514 */
1515static int pick_closest_replica(struct ceph_osd_client *osdc,
1516 const struct ceph_osds *acting)
1517{
1518 struct ceph_options *opt = osdc->client->options;
1519 int best_i, best_locality;
1520 int i = 0, locality;
1521
1522 do {
1523 locality = ceph_get_crush_locality(osdc->osdmap,
1524 acting->osds[i],
1525 &opt->crush_locs);
1526 if (i == 0 ||
1527 (locality >= 0 && best_locality < 0) ||
1528 (locality >= 0 && best_locality >= 0 &&
1529 locality < best_locality)) {
1530 best_i = i;
1531 best_locality = locality;
1532 }
1533 } while (++i < acting->size);
1534
1535 dout("%s picked osd%d with locality %d, primary osd%d\n", __func__,
1536 acting->osds[best_i], best_locality, acting->primary);
1537 return best_i;
1538}
1539
1540enum calc_target_result {
1541 CALC_TARGET_NO_ACTION = 0,
1542 CALC_TARGET_NEED_RESEND,
1543 CALC_TARGET_POOL_DNE,
1544};
1545
1546static enum calc_target_result calc_target(struct ceph_osd_client *osdc,
1547 struct ceph_osd_request_target *t,
1548 bool any_change)
1549{
1550 struct ceph_pg_pool_info *pi;
1551 struct ceph_pg pgid, last_pgid;
1552 struct ceph_osds up, acting;
1553 bool is_read = t->flags & CEPH_OSD_FLAG_READ;
1554 bool is_write = t->flags & CEPH_OSD_FLAG_WRITE;
1555 bool force_resend = false;
1556 bool unpaused = false;
1557 bool legacy_change = false;
1558 bool split = false;
1559 bool sort_bitwise = ceph_osdmap_flag(osdc, CEPH_OSDMAP_SORTBITWISE);
1560 bool recovery_deletes = ceph_osdmap_flag(osdc,
1561 CEPH_OSDMAP_RECOVERY_DELETES);
1562 enum calc_target_result ct_res;
1563
1564 t->epoch = osdc->osdmap->epoch;
1565 pi = ceph_pg_pool_by_id(osdc->osdmap, t->base_oloc.pool);
1566 if (!pi) {
1567 t->osd = CEPH_HOMELESS_OSD;
1568 ct_res = CALC_TARGET_POOL_DNE;
1569 goto out;
1570 }
1571
1572 if (osdc->osdmap->epoch == pi->last_force_request_resend) {
1573 if (t->last_force_resend < pi->last_force_request_resend) {
1574 t->last_force_resend = pi->last_force_request_resend;
1575 force_resend = true;
1576 } else if (t->last_force_resend == 0) {
1577 force_resend = true;
1578 }
1579 }
1580
1581 /* apply tiering */
1582 ceph_oid_copy(&t->target_oid, &t->base_oid);
1583 ceph_oloc_copy(&t->target_oloc, &t->base_oloc);
1584 if ((t->flags & CEPH_OSD_FLAG_IGNORE_OVERLAY) == 0) {
1585 if (is_read && pi->read_tier >= 0)
1586 t->target_oloc.pool = pi->read_tier;
1587 if (is_write && pi->write_tier >= 0)
1588 t->target_oloc.pool = pi->write_tier;
1589
1590 pi = ceph_pg_pool_by_id(osdc->osdmap, t->target_oloc.pool);
1591 if (!pi) {
1592 t->osd = CEPH_HOMELESS_OSD;
1593 ct_res = CALC_TARGET_POOL_DNE;
1594 goto out;
1595 }
1596 }
1597
1598 __ceph_object_locator_to_pg(pi, &t->target_oid, &t->target_oloc, &pgid);
1599 last_pgid.pool = pgid.pool;
1600 last_pgid.seed = ceph_stable_mod(pgid.seed, t->pg_num, t->pg_num_mask);
1601
1602 ceph_pg_to_up_acting_osds(osdc->osdmap, pi, &pgid, &up, &acting);
1603 if (any_change &&
1604 ceph_is_new_interval(&t->acting,
1605 &acting,
1606 &t->up,
1607 &up,
1608 t->size,
1609 pi->size,
1610 t->min_size,
1611 pi->min_size,
1612 t->pg_num,
1613 pi->pg_num,
1614 t->sort_bitwise,
1615 sort_bitwise,
1616 t->recovery_deletes,
1617 recovery_deletes,
1618 &last_pgid))
1619 force_resend = true;
1620
1621 if (t->paused && !target_should_be_paused(osdc, t, pi)) {
1622 t->paused = false;
1623 unpaused = true;
1624 }
1625 legacy_change = ceph_pg_compare(&t->pgid, &pgid) ||
1626 ceph_osds_changed(&t->acting, &acting,
1627 t->used_replica || any_change);
1628 if (t->pg_num)
1629 split = ceph_pg_is_split(&last_pgid, t->pg_num, pi->pg_num);
1630
1631 if (legacy_change || force_resend || split) {
1632 t->pgid = pgid; /* struct */
1633 ceph_pg_to_primary_shard(osdc->osdmap, pi, &pgid, &t->spgid);
1634 ceph_osds_copy(&t->acting, &acting);
1635 ceph_osds_copy(&t->up, &up);
1636 t->size = pi->size;
1637 t->min_size = pi->min_size;
1638 t->pg_num = pi->pg_num;
1639 t->pg_num_mask = pi->pg_num_mask;
1640 t->sort_bitwise = sort_bitwise;
1641 t->recovery_deletes = recovery_deletes;
1642
1643 if ((t->flags & (CEPH_OSD_FLAG_BALANCE_READS |
1644 CEPH_OSD_FLAG_LOCALIZE_READS)) &&
1645 !is_write && pi->type == CEPH_POOL_TYPE_REP &&
1646 acting.size > 1) {
1647 int pos;
1648
1649 WARN_ON(!is_read || acting.osds[0] != acting.primary);
1650 if (t->flags & CEPH_OSD_FLAG_BALANCE_READS) {
1651 pos = pick_random_replica(&acting);
1652 } else {
1653 pos = pick_closest_replica(osdc, &acting);
1654 }
1655 t->osd = acting.osds[pos];
1656 t->used_replica = pos > 0;
1657 } else {
1658 t->osd = acting.primary;
1659 t->used_replica = false;
1660 }
1661 }
1662
1663 if (unpaused || legacy_change || force_resend || split)
1664 ct_res = CALC_TARGET_NEED_RESEND;
1665 else
1666 ct_res = CALC_TARGET_NO_ACTION;
1667
1668out:
1669 dout("%s t %p -> %d%d%d%d ct_res %d osd%d\n", __func__, t, unpaused,
1670 legacy_change, force_resend, split, ct_res, t->osd);
1671 return ct_res;
1672}
1673
1674static struct ceph_spg_mapping *alloc_spg_mapping(void)
1675{
1676 struct ceph_spg_mapping *spg;
1677
1678 spg = kmalloc(sizeof(*spg), GFP_NOIO);
1679 if (!spg)
1680 return NULL;
1681
1682 RB_CLEAR_NODE(&spg->node);
1683 spg->backoffs = RB_ROOT;
1684 return spg;
1685}
1686
1687static void free_spg_mapping(struct ceph_spg_mapping *spg)
1688{
1689 WARN_ON(!RB_EMPTY_NODE(&spg->node));
1690 WARN_ON(!RB_EMPTY_ROOT(&spg->backoffs));
1691
1692 kfree(spg);
1693}
1694
1695/*
1696 * rbtree of ceph_spg_mapping for handling map<spg_t, ...>, similar to
1697 * ceph_pg_mapping. Used to track OSD backoffs -- a backoff [range] is
1698 * defined only within a specific spgid; it does not pass anything to
1699 * children on split, or to another primary.
1700 */
1701DEFINE_RB_FUNCS2(spg_mapping, struct ceph_spg_mapping, spgid, ceph_spg_compare,
1702 RB_BYPTR, const struct ceph_spg *, node)
1703
1704static u64 hoid_get_bitwise_key(const struct ceph_hobject_id *hoid)
1705{
1706 return hoid->is_max ? 0x100000000ull : hoid->hash_reverse_bits;
1707}
1708
1709static void hoid_get_effective_key(const struct ceph_hobject_id *hoid,
1710 void **pkey, size_t *pkey_len)
1711{
1712 if (hoid->key_len) {
1713 *pkey = hoid->key;
1714 *pkey_len = hoid->key_len;
1715 } else {
1716 *pkey = hoid->oid;
1717 *pkey_len = hoid->oid_len;
1718 }
1719}
1720
1721static int compare_names(const void *name1, size_t name1_len,
1722 const void *name2, size_t name2_len)
1723{
1724 int ret;
1725
1726 ret = memcmp(name1, name2, min(name1_len, name2_len));
1727 if (!ret) {
1728 if (name1_len < name2_len)
1729 ret = -1;
1730 else if (name1_len > name2_len)
1731 ret = 1;
1732 }
1733 return ret;
1734}
1735
1736static int hoid_compare(const struct ceph_hobject_id *lhs,
1737 const struct ceph_hobject_id *rhs)
1738{
1739 void *effective_key1, *effective_key2;
1740 size_t effective_key1_len, effective_key2_len;
1741 int ret;
1742
1743 if (lhs->is_max < rhs->is_max)
1744 return -1;
1745 if (lhs->is_max > rhs->is_max)
1746 return 1;
1747
1748 if (lhs->pool < rhs->pool)
1749 return -1;
1750 if (lhs->pool > rhs->pool)
1751 return 1;
1752
1753 if (hoid_get_bitwise_key(lhs) < hoid_get_bitwise_key(rhs))
1754 return -1;
1755 if (hoid_get_bitwise_key(lhs) > hoid_get_bitwise_key(rhs))
1756 return 1;
1757
1758 ret = compare_names(lhs->nspace, lhs->nspace_len,
1759 rhs->nspace, rhs->nspace_len);
1760 if (ret)
1761 return ret;
1762
1763 hoid_get_effective_key(lhs, &effective_key1, &effective_key1_len);
1764 hoid_get_effective_key(rhs, &effective_key2, &effective_key2_len);
1765 ret = compare_names(effective_key1, effective_key1_len,
1766 effective_key2, effective_key2_len);
1767 if (ret)
1768 return ret;
1769
1770 ret = compare_names(lhs->oid, lhs->oid_len, rhs->oid, rhs->oid_len);
1771 if (ret)
1772 return ret;
1773
1774 if (lhs->snapid < rhs->snapid)
1775 return -1;
1776 if (lhs->snapid > rhs->snapid)
1777 return 1;
1778
1779 return 0;
1780}
1781
1782/*
1783 * For decoding ->begin and ->end of MOSDBackoff only -- no MIN/MAX
1784 * compat stuff here.
1785 *
1786 * Assumes @hoid is zero-initialized.
1787 */
1788static int decode_hoid(void **p, void *end, struct ceph_hobject_id *hoid)
1789{
1790 u8 struct_v;
1791 u32 struct_len;
1792 int ret;
1793
1794 ret = ceph_start_decoding(p, end, 4, "hobject_t", &struct_v,
1795 &struct_len);
1796 if (ret)
1797 return ret;
1798
1799 if (struct_v < 4) {
1800 pr_err("got struct_v %d < 4 of hobject_t\n", struct_v);
1801 goto e_inval;
1802 }
1803
1804 hoid->key = ceph_extract_encoded_string(p, end, &hoid->key_len,
1805 GFP_NOIO);
1806 if (IS_ERR(hoid->key)) {
1807 ret = PTR_ERR(hoid->key);
1808 hoid->key = NULL;
1809 return ret;
1810 }
1811
1812 hoid->oid = ceph_extract_encoded_string(p, end, &hoid->oid_len,
1813 GFP_NOIO);
1814 if (IS_ERR(hoid->oid)) {
1815 ret = PTR_ERR(hoid->oid);
1816 hoid->oid = NULL;
1817 return ret;
1818 }
1819
1820 ceph_decode_64_safe(p, end, hoid->snapid, e_inval);
1821 ceph_decode_32_safe(p, end, hoid->hash, e_inval);
1822 ceph_decode_8_safe(p, end, hoid->is_max, e_inval);
1823
1824 hoid->nspace = ceph_extract_encoded_string(p, end, &hoid->nspace_len,
1825 GFP_NOIO);
1826 if (IS_ERR(hoid->nspace)) {
1827 ret = PTR_ERR(hoid->nspace);
1828 hoid->nspace = NULL;
1829 return ret;
1830 }
1831
1832 ceph_decode_64_safe(p, end, hoid->pool, e_inval);
1833
1834 ceph_hoid_build_hash_cache(hoid);
1835 return 0;
1836
1837e_inval:
1838 return -EINVAL;
1839}
1840
1841static int hoid_encoding_size(const struct ceph_hobject_id *hoid)
1842{
1843 return 8 + 4 + 1 + 8 + /* snapid, hash, is_max, pool */
1844 4 + hoid->key_len + 4 + hoid->oid_len + 4 + hoid->nspace_len;
1845}
1846
1847static void encode_hoid(void **p, void *end, const struct ceph_hobject_id *hoid)
1848{
1849 ceph_start_encoding(p, 4, 3, hoid_encoding_size(hoid));
1850 ceph_encode_string(p, end, hoid->key, hoid->key_len);
1851 ceph_encode_string(p, end, hoid->oid, hoid->oid_len);
1852 ceph_encode_64(p, hoid->snapid);
1853 ceph_encode_32(p, hoid->hash);
1854 ceph_encode_8(p, hoid->is_max);
1855 ceph_encode_string(p, end, hoid->nspace, hoid->nspace_len);
1856 ceph_encode_64(p, hoid->pool);
1857}
1858
1859static void free_hoid(struct ceph_hobject_id *hoid)
1860{
1861 if (hoid) {
1862 kfree(hoid->key);
1863 kfree(hoid->oid);
1864 kfree(hoid->nspace);
1865 kfree(hoid);
1866 }
1867}
1868
1869static struct ceph_osd_backoff *alloc_backoff(void)
1870{
1871 struct ceph_osd_backoff *backoff;
1872
1873 backoff = kzalloc(sizeof(*backoff), GFP_NOIO);
1874 if (!backoff)
1875 return NULL;
1876
1877 RB_CLEAR_NODE(&backoff->spg_node);
1878 RB_CLEAR_NODE(&backoff->id_node);
1879 return backoff;
1880}
1881
1882static void free_backoff(struct ceph_osd_backoff *backoff)
1883{
1884 WARN_ON(!RB_EMPTY_NODE(&backoff->spg_node));
1885 WARN_ON(!RB_EMPTY_NODE(&backoff->id_node));
1886
1887 free_hoid(backoff->begin);
1888 free_hoid(backoff->end);
1889 kfree(backoff);
1890}
1891
1892/*
1893 * Within a specific spgid, backoffs are managed by ->begin hoid.
1894 */
1895DEFINE_RB_INSDEL_FUNCS2(backoff, struct ceph_osd_backoff, begin, hoid_compare,
1896 RB_BYVAL, spg_node);
1897
1898static struct ceph_osd_backoff *lookup_containing_backoff(struct rb_root *root,
1899 const struct ceph_hobject_id *hoid)
1900{
1901 struct rb_node *n = root->rb_node;
1902
1903 while (n) {
1904 struct ceph_osd_backoff *cur =
1905 rb_entry(n, struct ceph_osd_backoff, spg_node);
1906 int cmp;
1907
1908 cmp = hoid_compare(hoid, cur->begin);
1909 if (cmp < 0) {
1910 n = n->rb_left;
1911 } else if (cmp > 0) {
1912 if (hoid_compare(hoid, cur->end) < 0)
1913 return cur;
1914
1915 n = n->rb_right;
1916 } else {
1917 return cur;
1918 }
1919 }
1920
1921 return NULL;
1922}
1923
1924/*
1925 * Each backoff has a unique id within its OSD session.
1926 */
1927DEFINE_RB_FUNCS(backoff_by_id, struct ceph_osd_backoff, id, id_node)
1928
1929static void clear_backoffs(struct ceph_osd *osd)
1930{
1931 while (!RB_EMPTY_ROOT(&osd->o_backoff_mappings)) {
1932 struct ceph_spg_mapping *spg =
1933 rb_entry(rb_first(&osd->o_backoff_mappings),
1934 struct ceph_spg_mapping, node);
1935
1936 while (!RB_EMPTY_ROOT(&spg->backoffs)) {
1937 struct ceph_osd_backoff *backoff =
1938 rb_entry(rb_first(&spg->backoffs),
1939 struct ceph_osd_backoff, spg_node);
1940
1941 erase_backoff(&spg->backoffs, backoff);
1942 erase_backoff_by_id(&osd->o_backoffs_by_id, backoff);
1943 free_backoff(backoff);
1944 }
1945 erase_spg_mapping(&osd->o_backoff_mappings, spg);
1946 free_spg_mapping(spg);
1947 }
1948}
1949
1950/*
1951 * Set up a temporary, non-owning view into @t.
1952 */
1953static void hoid_fill_from_target(struct ceph_hobject_id *hoid,
1954 const struct ceph_osd_request_target *t)
1955{
1956 hoid->key = NULL;
1957 hoid->key_len = 0;
1958 hoid->oid = t->target_oid.name;
1959 hoid->oid_len = t->target_oid.name_len;
1960 hoid->snapid = CEPH_NOSNAP;
1961 hoid->hash = t->pgid.seed;
1962 hoid->is_max = false;
1963 if (t->target_oloc.pool_ns) {
1964 hoid->nspace = t->target_oloc.pool_ns->str;
1965 hoid->nspace_len = t->target_oloc.pool_ns->len;
1966 } else {
1967 hoid->nspace = NULL;
1968 hoid->nspace_len = 0;
1969 }
1970 hoid->pool = t->target_oloc.pool;
1971 ceph_hoid_build_hash_cache(hoid);
1972}
1973
1974static bool should_plug_request(struct ceph_osd_request *req)
1975{
1976 struct ceph_osd *osd = req->r_osd;
1977 struct ceph_spg_mapping *spg;
1978 struct ceph_osd_backoff *backoff;
1979 struct ceph_hobject_id hoid;
1980
1981 spg = lookup_spg_mapping(&osd->o_backoff_mappings, &req->r_t.spgid);
1982 if (!spg)
1983 return false;
1984
1985 hoid_fill_from_target(&hoid, &req->r_t);
1986 backoff = lookup_containing_backoff(&spg->backoffs, &hoid);
1987 if (!backoff)
1988 return false;
1989
1990 dout("%s req %p tid %llu backoff osd%d spgid %llu.%xs%d id %llu\n",
1991 __func__, req, req->r_tid, osd->o_osd, backoff->spgid.pgid.pool,
1992 backoff->spgid.pgid.seed, backoff->spgid.shard, backoff->id);
1993 return true;
1994}
1995
1996/*
1997 * Keep get_num_data_items() in sync with this function.
1998 */
1999static void setup_request_data(struct ceph_osd_request *req)
2000{
2001 struct ceph_msg *request_msg = req->r_request;
2002 struct ceph_msg *reply_msg = req->r_reply;
2003 struct ceph_osd_req_op *op;
2004
2005 if (req->r_request->num_data_items || req->r_reply->num_data_items)
2006 return;
2007
2008 WARN_ON(request_msg->data_length || reply_msg->data_length);
2009 for (op = req->r_ops; op != &req->r_ops[req->r_num_ops]; op++) {
2010 switch (op->op) {
2011 /* request */
2012 case CEPH_OSD_OP_WRITE:
2013 case CEPH_OSD_OP_WRITEFULL:
2014 WARN_ON(op->indata_len != op->extent.length);
2015 ceph_osdc_msg_data_add(request_msg,
2016 &op->extent.osd_data);
2017 break;
2018 case CEPH_OSD_OP_SETXATTR:
2019 case CEPH_OSD_OP_CMPXATTR:
2020 WARN_ON(op->indata_len != op->xattr.name_len +
2021 op->xattr.value_len);
2022 ceph_osdc_msg_data_add(request_msg,
2023 &op->xattr.osd_data);
2024 break;
2025 case CEPH_OSD_OP_NOTIFY_ACK:
2026 ceph_osdc_msg_data_add(request_msg,
2027 &op->notify_ack.request_data);
2028 break;
2029 case CEPH_OSD_OP_COPY_FROM2:
2030 ceph_osdc_msg_data_add(request_msg,
2031 &op->copy_from.osd_data);
2032 break;
2033
2034 /* reply */
2035 case CEPH_OSD_OP_STAT:
2036 ceph_osdc_msg_data_add(reply_msg,
2037 &op->raw_data_in);
2038 break;
2039 case CEPH_OSD_OP_READ:
2040 ceph_osdc_msg_data_add(reply_msg,
2041 &op->extent.osd_data);
2042 break;
2043 case CEPH_OSD_OP_LIST_WATCHERS:
2044 ceph_osdc_msg_data_add(reply_msg,
2045 &op->list_watchers.response_data);
2046 break;
2047
2048 /* both */
2049 case CEPH_OSD_OP_CALL:
2050 WARN_ON(op->indata_len != op->cls.class_len +
2051 op->cls.method_len +
2052 op->cls.indata_len);
2053 ceph_osdc_msg_data_add(request_msg,
2054 &op->cls.request_info);
2055 /* optional, can be NONE */
2056 ceph_osdc_msg_data_add(request_msg,
2057 &op->cls.request_data);
2058 /* optional, can be NONE */
2059 ceph_osdc_msg_data_add(reply_msg,
2060 &op->cls.response_data);
2061 break;
2062 case CEPH_OSD_OP_NOTIFY:
2063 ceph_osdc_msg_data_add(request_msg,
2064 &op->notify.request_data);
2065 ceph_osdc_msg_data_add(reply_msg,
2066 &op->notify.response_data);
2067 break;
2068 }
2069 }
2070}
2071
2072static void encode_pgid(void **p, const struct ceph_pg *pgid)
2073{
2074 ceph_encode_8(p, 1);
2075 ceph_encode_64(p, pgid->pool);
2076 ceph_encode_32(p, pgid->seed);
2077 ceph_encode_32(p, -1); /* preferred */
2078}
2079
2080static void encode_spgid(void **p, const struct ceph_spg *spgid)
2081{
2082 ceph_start_encoding(p, 1, 1, CEPH_PGID_ENCODING_LEN + 1);
2083 encode_pgid(p, &spgid->pgid);
2084 ceph_encode_8(p, spgid->shard);
2085}
2086
2087static void encode_oloc(void **p, void *end,
2088 const struct ceph_object_locator *oloc)
2089{
2090 ceph_start_encoding(p, 5, 4, ceph_oloc_encoding_size(oloc));
2091 ceph_encode_64(p, oloc->pool);
2092 ceph_encode_32(p, -1); /* preferred */
2093 ceph_encode_32(p, 0); /* key len */
2094 if (oloc->pool_ns)
2095 ceph_encode_string(p, end, oloc->pool_ns->str,
2096 oloc->pool_ns->len);
2097 else
2098 ceph_encode_32(p, 0);
2099}
2100
2101static void encode_request_partial(struct ceph_osd_request *req,
2102 struct ceph_msg *msg)
2103{
2104 void *p = msg->front.iov_base;
2105 void *const end = p + msg->front_alloc_len;
2106 u32 data_len = 0;
2107 int i;
2108
2109 if (req->r_flags & CEPH_OSD_FLAG_WRITE) {
2110 /* snapshots aren't writeable */
2111 WARN_ON(req->r_snapid != CEPH_NOSNAP);
2112 } else {
2113 WARN_ON(req->r_mtime.tv_sec || req->r_mtime.tv_nsec ||
2114 req->r_data_offset || req->r_snapc);
2115 }
2116
2117 setup_request_data(req);
2118
2119 encode_spgid(&p, &req->r_t.spgid); /* actual spg */
2120 ceph_encode_32(&p, req->r_t.pgid.seed); /* raw hash */
2121 ceph_encode_32(&p, req->r_osdc->osdmap->epoch);
2122 ceph_encode_32(&p, req->r_flags);
2123
2124 /* reqid */
2125 ceph_start_encoding(&p, 2, 2, sizeof(struct ceph_osd_reqid));
2126 memset(p, 0, sizeof(struct ceph_osd_reqid));
2127 p += sizeof(struct ceph_osd_reqid);
2128
2129 /* trace */
2130 memset(p, 0, sizeof(struct ceph_blkin_trace_info));
2131 p += sizeof(struct ceph_blkin_trace_info);
2132
2133 ceph_encode_32(&p, 0); /* client_inc, always 0 */
2134 ceph_encode_timespec64(p, &req->r_mtime);
2135 p += sizeof(struct ceph_timespec);
2136
2137 encode_oloc(&p, end, &req->r_t.target_oloc);
2138 ceph_encode_string(&p, end, req->r_t.target_oid.name,
2139 req->r_t.target_oid.name_len);
2140
2141 /* ops, can imply data */
2142 ceph_encode_16(&p, req->r_num_ops);
2143 for (i = 0; i < req->r_num_ops; i++) {
2144 data_len += osd_req_encode_op(p, &req->r_ops[i]);
2145 p += sizeof(struct ceph_osd_op);
2146 }
2147
2148 ceph_encode_64(&p, req->r_snapid); /* snapid */
2149 if (req->r_snapc) {
2150 ceph_encode_64(&p, req->r_snapc->seq);
2151 ceph_encode_32(&p, req->r_snapc->num_snaps);
2152 for (i = 0; i < req->r_snapc->num_snaps; i++)
2153 ceph_encode_64(&p, req->r_snapc->snaps[i]);
2154 } else {
2155 ceph_encode_64(&p, 0); /* snap_seq */
2156 ceph_encode_32(&p, 0); /* snaps len */
2157 }
2158
2159 ceph_encode_32(&p, req->r_attempts); /* retry_attempt */
2160 BUG_ON(p > end - 8); /* space for features */
2161
2162 msg->hdr.version = cpu_to_le16(8); /* MOSDOp v8 */
2163 /* front_len is finalized in encode_request_finish() */
2164 msg->front.iov_len = p - msg->front.iov_base;
2165 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
2166 msg->hdr.data_len = cpu_to_le32(data_len);
2167 /*
2168 * The header "data_off" is a hint to the receiver allowing it
2169 * to align received data into its buffers such that there's no
2170 * need to re-copy it before writing it to disk (direct I/O).
2171 */
2172 msg->hdr.data_off = cpu_to_le16(req->r_data_offset);
2173
2174 dout("%s req %p msg %p oid %s oid_len %d\n", __func__, req, msg,
2175 req->r_t.target_oid.name, req->r_t.target_oid.name_len);
2176}
2177
2178static void encode_request_finish(struct ceph_msg *msg)
2179{
2180 void *p = msg->front.iov_base;
2181 void *const partial_end = p + msg->front.iov_len;
2182 void *const end = p + msg->front_alloc_len;
2183
2184 if (CEPH_HAVE_FEATURE(msg->con->peer_features, RESEND_ON_SPLIT)) {
2185 /* luminous OSD -- encode features and be done */
2186 p = partial_end;
2187 ceph_encode_64(&p, msg->con->peer_features);
2188 } else {
2189 struct {
2190 char spgid[CEPH_ENCODING_START_BLK_LEN +
2191 CEPH_PGID_ENCODING_LEN + 1];
2192 __le32 hash;
2193 __le32 epoch;
2194 __le32 flags;
2195 char reqid[CEPH_ENCODING_START_BLK_LEN +
2196 sizeof(struct ceph_osd_reqid)];
2197 char trace[sizeof(struct ceph_blkin_trace_info)];
2198 __le32 client_inc;
2199 struct ceph_timespec mtime;
2200 } __packed head;
2201 struct ceph_pg pgid;
2202 void *oloc, *oid, *tail;
2203 int oloc_len, oid_len, tail_len;
2204 int len;
2205
2206 /*
2207 * Pre-luminous OSD -- reencode v8 into v4 using @head
2208 * as a temporary buffer. Encode the raw PG; the rest
2209 * is just a matter of moving oloc, oid and tail blobs
2210 * around.
2211 */
2212 memcpy(&head, p, sizeof(head));
2213 p += sizeof(head);
2214
2215 oloc = p;
2216 p += CEPH_ENCODING_START_BLK_LEN;
2217 pgid.pool = ceph_decode_64(&p);
2218 p += 4 + 4; /* preferred, key len */
2219 len = ceph_decode_32(&p);
2220 p += len; /* nspace */
2221 oloc_len = p - oloc;
2222
2223 oid = p;
2224 len = ceph_decode_32(&p);
2225 p += len;
2226 oid_len = p - oid;
2227
2228 tail = p;
2229 tail_len = partial_end - p;
2230
2231 p = msg->front.iov_base;
2232 ceph_encode_copy(&p, &head.client_inc, sizeof(head.client_inc));
2233 ceph_encode_copy(&p, &head.epoch, sizeof(head.epoch));
2234 ceph_encode_copy(&p, &head.flags, sizeof(head.flags));
2235 ceph_encode_copy(&p, &head.mtime, sizeof(head.mtime));
2236
2237 /* reassert_version */
2238 memset(p, 0, sizeof(struct ceph_eversion));
2239 p += sizeof(struct ceph_eversion);
2240
2241 BUG_ON(p >= oloc);
2242 memmove(p, oloc, oloc_len);
2243 p += oloc_len;
2244
2245 pgid.seed = le32_to_cpu(head.hash);
2246 encode_pgid(&p, &pgid); /* raw pg */
2247
2248 BUG_ON(p >= oid);
2249 memmove(p, oid, oid_len);
2250 p += oid_len;
2251
2252 /* tail -- ops, snapid, snapc, retry_attempt */
2253 BUG_ON(p >= tail);
2254 memmove(p, tail, tail_len);
2255 p += tail_len;
2256
2257 msg->hdr.version = cpu_to_le16(4); /* MOSDOp v4 */
2258 }
2259
2260 BUG_ON(p > end);
2261 msg->front.iov_len = p - msg->front.iov_base;
2262 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
2263
2264 dout("%s msg %p tid %llu %u+%u+%u v%d\n", __func__, msg,
2265 le64_to_cpu(msg->hdr.tid), le32_to_cpu(msg->hdr.front_len),
2266 le32_to_cpu(msg->hdr.middle_len), le32_to_cpu(msg->hdr.data_len),
2267 le16_to_cpu(msg->hdr.version));
2268}
2269
2270/*
2271 * @req has to be assigned a tid and registered.
2272 */
2273static void send_request(struct ceph_osd_request *req)
2274{
2275 struct ceph_osd *osd = req->r_osd;
2276
2277 verify_osd_locked(osd);
2278 WARN_ON(osd->o_osd != req->r_t.osd);
2279
2280 /* backoff? */
2281 if (should_plug_request(req))
2282 return;
2283
2284 /*
2285 * We may have a previously queued request message hanging
2286 * around. Cancel it to avoid corrupting the msgr.
2287 */
2288 if (req->r_sent)
2289 ceph_msg_revoke(req->r_request);
2290
2291 req->r_flags |= CEPH_OSD_FLAG_KNOWN_REDIR;
2292 if (req->r_attempts)
2293 req->r_flags |= CEPH_OSD_FLAG_RETRY;
2294 else
2295 WARN_ON(req->r_flags & CEPH_OSD_FLAG_RETRY);
2296
2297 encode_request_partial(req, req->r_request);
2298
2299 dout("%s req %p tid %llu to pgid %llu.%x spgid %llu.%xs%d osd%d e%u flags 0x%x attempt %d\n",
2300 __func__, req, req->r_tid, req->r_t.pgid.pool, req->r_t.pgid.seed,
2301 req->r_t.spgid.pgid.pool, req->r_t.spgid.pgid.seed,
2302 req->r_t.spgid.shard, osd->o_osd, req->r_t.epoch, req->r_flags,
2303 req->r_attempts);
2304
2305 req->r_t.paused = false;
2306 req->r_stamp = jiffies;
2307 req->r_attempts++;
2308
2309 req->r_sent = osd->o_incarnation;
2310 req->r_request->hdr.tid = cpu_to_le64(req->r_tid);
2311 ceph_con_send(&osd->o_con, ceph_msg_get(req->r_request));
2312}
2313
2314static void maybe_request_map(struct ceph_osd_client *osdc)
2315{
2316 bool continuous = false;
2317
2318 verify_osdc_locked(osdc);
2319 WARN_ON(!osdc->osdmap->epoch);
2320
2321 if (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
2322 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD) ||
2323 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR)) {
2324 dout("%s osdc %p continuous\n", __func__, osdc);
2325 continuous = true;
2326 } else {
2327 dout("%s osdc %p onetime\n", __func__, osdc);
2328 }
2329
2330 if (ceph_monc_want_map(&osdc->client->monc, CEPH_SUB_OSDMAP,
2331 osdc->osdmap->epoch + 1, continuous))
2332 ceph_monc_renew_subs(&osdc->client->monc);
2333}
2334
2335static void complete_request(struct ceph_osd_request *req, int err);
2336static void send_map_check(struct ceph_osd_request *req);
2337
2338static void __submit_request(struct ceph_osd_request *req, bool wrlocked)
2339{
2340 struct ceph_osd_client *osdc = req->r_osdc;
2341 struct ceph_osd *osd;
2342 enum calc_target_result ct_res;
2343 int err = 0;
2344 bool need_send = false;
2345 bool promoted = false;
2346
2347 WARN_ON(req->r_tid);
2348 dout("%s req %p wrlocked %d\n", __func__, req, wrlocked);
2349
2350again:
2351 ct_res = calc_target(osdc, &req->r_t, false);
2352 if (ct_res == CALC_TARGET_POOL_DNE && !wrlocked)
2353 goto promote;
2354
2355 osd = lookup_create_osd(osdc, req->r_t.osd, wrlocked);
2356 if (IS_ERR(osd)) {
2357 WARN_ON(PTR_ERR(osd) != -EAGAIN || wrlocked);
2358 goto promote;
2359 }
2360
2361 if (osdc->abort_err) {
2362 dout("req %p abort_err %d\n", req, osdc->abort_err);
2363 err = osdc->abort_err;
2364 } else if (osdc->osdmap->epoch < osdc->epoch_barrier) {
2365 dout("req %p epoch %u barrier %u\n", req, osdc->osdmap->epoch,
2366 osdc->epoch_barrier);
2367 req->r_t.paused = true;
2368 maybe_request_map(osdc);
2369 } else if ((req->r_flags & CEPH_OSD_FLAG_WRITE) &&
2370 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR)) {
2371 dout("req %p pausewr\n", req);
2372 req->r_t.paused = true;
2373 maybe_request_map(osdc);
2374 } else if ((req->r_flags & CEPH_OSD_FLAG_READ) &&
2375 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD)) {
2376 dout("req %p pauserd\n", req);
2377 req->r_t.paused = true;
2378 maybe_request_map(osdc);
2379 } else if ((req->r_flags & CEPH_OSD_FLAG_WRITE) &&
2380 !(req->r_flags & (CEPH_OSD_FLAG_FULL_TRY |
2381 CEPH_OSD_FLAG_FULL_FORCE)) &&
2382 (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
2383 pool_full(osdc, req->r_t.base_oloc.pool))) {
2384 dout("req %p full/pool_full\n", req);
2385 if (ceph_test_opt(osdc->client, ABORT_ON_FULL)) {
2386 err = -ENOSPC;
2387 } else {
2388 pr_warn_ratelimited("FULL or reached pool quota\n");
2389 req->r_t.paused = true;
2390 maybe_request_map(osdc);
2391 }
2392 } else if (!osd_homeless(osd)) {
2393 need_send = true;
2394 } else {
2395 maybe_request_map(osdc);
2396 }
2397
2398 mutex_lock(&osd->lock);
2399 /*
2400 * Assign the tid atomically with send_request() to protect
2401 * multiple writes to the same object from racing with each
2402 * other, resulting in out of order ops on the OSDs.
2403 */
2404 req->r_tid = atomic64_inc_return(&osdc->last_tid);
2405 link_request(osd, req);
2406 if (need_send)
2407 send_request(req);
2408 else if (err)
2409 complete_request(req, err);
2410 mutex_unlock(&osd->lock);
2411
2412 if (!err && ct_res == CALC_TARGET_POOL_DNE)
2413 send_map_check(req);
2414
2415 if (promoted)
2416 downgrade_write(&osdc->lock);
2417 return;
2418
2419promote:
2420 up_read(&osdc->lock);
2421 down_write(&osdc->lock);
2422 wrlocked = true;
2423 promoted = true;
2424 goto again;
2425}
2426
2427static void account_request(struct ceph_osd_request *req)
2428{
2429 WARN_ON(req->r_flags & (CEPH_OSD_FLAG_ACK | CEPH_OSD_FLAG_ONDISK));
2430 WARN_ON(!(req->r_flags & (CEPH_OSD_FLAG_READ | CEPH_OSD_FLAG_WRITE)));
2431
2432 req->r_flags |= CEPH_OSD_FLAG_ONDISK;
2433 atomic_inc(&req->r_osdc->num_requests);
2434
2435 req->r_start_stamp = jiffies;
2436 req->r_start_latency = ktime_get();
2437}
2438
2439static void submit_request(struct ceph_osd_request *req, bool wrlocked)
2440{
2441 ceph_osdc_get_request(req);
2442 account_request(req);
2443 __submit_request(req, wrlocked);
2444}
2445
2446static void finish_request(struct ceph_osd_request *req)
2447{
2448 struct ceph_osd_client *osdc = req->r_osdc;
2449
2450 WARN_ON(lookup_request_mc(&osdc->map_checks, req->r_tid));
2451 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
2452
2453 req->r_end_latency = ktime_get();
2454
2455 if (req->r_osd)
2456 unlink_request(req->r_osd, req);
2457 atomic_dec(&osdc->num_requests);
2458
2459 /*
2460 * If an OSD has failed or returned and a request has been sent
2461 * twice, it's possible to get a reply and end up here while the
2462 * request message is queued for delivery. We will ignore the
2463 * reply, so not a big deal, but better to try and catch it.
2464 */
2465 ceph_msg_revoke(req->r_request);
2466 ceph_msg_revoke_incoming(req->r_reply);
2467}
2468
2469static void __complete_request(struct ceph_osd_request *req)
2470{
2471 dout("%s req %p tid %llu cb %ps result %d\n", __func__, req,
2472 req->r_tid, req->r_callback, req->r_result);
2473
2474 if (req->r_callback)
2475 req->r_callback(req);
2476 complete_all(&req->r_completion);
2477 ceph_osdc_put_request(req);
2478}
2479
2480static void complete_request_workfn(struct work_struct *work)
2481{
2482 struct ceph_osd_request *req =
2483 container_of(work, struct ceph_osd_request, r_complete_work);
2484
2485 __complete_request(req);
2486}
2487
2488/*
2489 * This is open-coded in handle_reply().
2490 */
2491static void complete_request(struct ceph_osd_request *req, int err)
2492{
2493 dout("%s req %p tid %llu err %d\n", __func__, req, req->r_tid, err);
2494
2495 req->r_result = err;
2496 finish_request(req);
2497
2498 INIT_WORK(&req->r_complete_work, complete_request_workfn);
2499 queue_work(req->r_osdc->completion_wq, &req->r_complete_work);
2500}
2501
2502static void cancel_map_check(struct ceph_osd_request *req)
2503{
2504 struct ceph_osd_client *osdc = req->r_osdc;
2505 struct ceph_osd_request *lookup_req;
2506
2507 verify_osdc_wrlocked(osdc);
2508
2509 lookup_req = lookup_request_mc(&osdc->map_checks, req->r_tid);
2510 if (!lookup_req)
2511 return;
2512
2513 WARN_ON(lookup_req != req);
2514 erase_request_mc(&osdc->map_checks, req);
2515 ceph_osdc_put_request(req);
2516}
2517
2518static void cancel_request(struct ceph_osd_request *req)
2519{
2520 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
2521
2522 cancel_map_check(req);
2523 finish_request(req);
2524 complete_all(&req->r_completion);
2525 ceph_osdc_put_request(req);
2526}
2527
2528static void abort_request(struct ceph_osd_request *req, int err)
2529{
2530 dout("%s req %p tid %llu err %d\n", __func__, req, req->r_tid, err);
2531
2532 cancel_map_check(req);
2533 complete_request(req, err);
2534}
2535
2536static int abort_fn(struct ceph_osd_request *req, void *arg)
2537{
2538 int err = *(int *)arg;
2539
2540 abort_request(req, err);
2541 return 0; /* continue iteration */
2542}
2543
2544/*
2545 * Abort all in-flight requests with @err and arrange for all future
2546 * requests to be failed immediately.
2547 */
2548void ceph_osdc_abort_requests(struct ceph_osd_client *osdc, int err)
2549{
2550 dout("%s osdc %p err %d\n", __func__, osdc, err);
2551 down_write(&osdc->lock);
2552 for_each_request(osdc, abort_fn, &err);
2553 osdc->abort_err = err;
2554 up_write(&osdc->lock);
2555}
2556EXPORT_SYMBOL(ceph_osdc_abort_requests);
2557
2558void ceph_osdc_clear_abort_err(struct ceph_osd_client *osdc)
2559{
2560 down_write(&osdc->lock);
2561 osdc->abort_err = 0;
2562 up_write(&osdc->lock);
2563}
2564EXPORT_SYMBOL(ceph_osdc_clear_abort_err);
2565
2566static void update_epoch_barrier(struct ceph_osd_client *osdc, u32 eb)
2567{
2568 if (likely(eb > osdc->epoch_barrier)) {
2569 dout("updating epoch_barrier from %u to %u\n",
2570 osdc->epoch_barrier, eb);
2571 osdc->epoch_barrier = eb;
2572 /* Request map if we're not to the barrier yet */
2573 if (eb > osdc->osdmap->epoch)
2574 maybe_request_map(osdc);
2575 }
2576}
2577
2578void ceph_osdc_update_epoch_barrier(struct ceph_osd_client *osdc, u32 eb)
2579{
2580 down_read(&osdc->lock);
2581 if (unlikely(eb > osdc->epoch_barrier)) {
2582 up_read(&osdc->lock);
2583 down_write(&osdc->lock);
2584 update_epoch_barrier(osdc, eb);
2585 up_write(&osdc->lock);
2586 } else {
2587 up_read(&osdc->lock);
2588 }
2589}
2590EXPORT_SYMBOL(ceph_osdc_update_epoch_barrier);
2591
2592/*
2593 * We can end up releasing caps as a result of abort_request().
2594 * In that case, we probably want to ensure that the cap release message
2595 * has an updated epoch barrier in it, so set the epoch barrier prior to
2596 * aborting the first request.
2597 */
2598static int abort_on_full_fn(struct ceph_osd_request *req, void *arg)
2599{
2600 struct ceph_osd_client *osdc = req->r_osdc;
2601 bool *victims = arg;
2602
2603 if ((req->r_flags & CEPH_OSD_FLAG_WRITE) &&
2604 (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
2605 pool_full(osdc, req->r_t.base_oloc.pool))) {
2606 if (!*victims) {
2607 update_epoch_barrier(osdc, osdc->osdmap->epoch);
2608 *victims = true;
2609 }
2610 abort_request(req, -ENOSPC);
2611 }
2612
2613 return 0; /* continue iteration */
2614}
2615
2616/*
2617 * Drop all pending requests that are stalled waiting on a full condition to
2618 * clear, and complete them with ENOSPC as the return code. Set the
2619 * osdc->epoch_barrier to the latest map epoch that we've seen if any were
2620 * cancelled.
2621 */
2622static void ceph_osdc_abort_on_full(struct ceph_osd_client *osdc)
2623{
2624 bool victims = false;
2625
2626 if (ceph_test_opt(osdc->client, ABORT_ON_FULL) &&
2627 (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) || have_pool_full(osdc)))
2628 for_each_request(osdc, abort_on_full_fn, &victims);
2629}
2630
2631static void check_pool_dne(struct ceph_osd_request *req)
2632{
2633 struct ceph_osd_client *osdc = req->r_osdc;
2634 struct ceph_osdmap *map = osdc->osdmap;
2635
2636 verify_osdc_wrlocked(osdc);
2637 WARN_ON(!map->epoch);
2638
2639 if (req->r_attempts) {
2640 /*
2641 * We sent a request earlier, which means that
2642 * previously the pool existed, and now it does not
2643 * (i.e., it was deleted).
2644 */
2645 req->r_map_dne_bound = map->epoch;
2646 dout("%s req %p tid %llu pool disappeared\n", __func__, req,
2647 req->r_tid);
2648 } else {
2649 dout("%s req %p tid %llu map_dne_bound %u have %u\n", __func__,
2650 req, req->r_tid, req->r_map_dne_bound, map->epoch);
2651 }
2652
2653 if (req->r_map_dne_bound) {
2654 if (map->epoch >= req->r_map_dne_bound) {
2655 /* we had a new enough map */
2656 pr_info_ratelimited("tid %llu pool does not exist\n",
2657 req->r_tid);
2658 complete_request(req, -ENOENT);
2659 }
2660 } else {
2661 send_map_check(req);
2662 }
2663}
2664
2665static void map_check_cb(struct ceph_mon_generic_request *greq)
2666{
2667 struct ceph_osd_client *osdc = &greq->monc->client->osdc;
2668 struct ceph_osd_request *req;
2669 u64 tid = greq->private_data;
2670
2671 WARN_ON(greq->result || !greq->u.newest);
2672
2673 down_write(&osdc->lock);
2674 req = lookup_request_mc(&osdc->map_checks, tid);
2675 if (!req) {
2676 dout("%s tid %llu dne\n", __func__, tid);
2677 goto out_unlock;
2678 }
2679
2680 dout("%s req %p tid %llu map_dne_bound %u newest %llu\n", __func__,
2681 req, req->r_tid, req->r_map_dne_bound, greq->u.newest);
2682 if (!req->r_map_dne_bound)
2683 req->r_map_dne_bound = greq->u.newest;
2684 erase_request_mc(&osdc->map_checks, req);
2685 check_pool_dne(req);
2686
2687 ceph_osdc_put_request(req);
2688out_unlock:
2689 up_write(&osdc->lock);
2690}
2691
2692static void send_map_check(struct ceph_osd_request *req)
2693{
2694 struct ceph_osd_client *osdc = req->r_osdc;
2695 struct ceph_osd_request *lookup_req;
2696 int ret;
2697
2698 verify_osdc_wrlocked(osdc);
2699
2700 lookup_req = lookup_request_mc(&osdc->map_checks, req->r_tid);
2701 if (lookup_req) {
2702 WARN_ON(lookup_req != req);
2703 return;
2704 }
2705
2706 ceph_osdc_get_request(req);
2707 insert_request_mc(&osdc->map_checks, req);
2708 ret = ceph_monc_get_version_async(&osdc->client->monc, "osdmap",
2709 map_check_cb, req->r_tid);
2710 WARN_ON(ret);
2711}
2712
2713/*
2714 * lingering requests, watch/notify v2 infrastructure
2715 */
2716static void linger_release(struct kref *kref)
2717{
2718 struct ceph_osd_linger_request *lreq =
2719 container_of(kref, struct ceph_osd_linger_request, kref);
2720
2721 dout("%s lreq %p reg_req %p ping_req %p\n", __func__, lreq,
2722 lreq->reg_req, lreq->ping_req);
2723 WARN_ON(!RB_EMPTY_NODE(&lreq->node));
2724 WARN_ON(!RB_EMPTY_NODE(&lreq->osdc_node));
2725 WARN_ON(!RB_EMPTY_NODE(&lreq->mc_node));
2726 WARN_ON(!list_empty(&lreq->scan_item));
2727 WARN_ON(!list_empty(&lreq->pending_lworks));
2728 WARN_ON(lreq->osd);
2729
2730 if (lreq->reg_req)
2731 ceph_osdc_put_request(lreq->reg_req);
2732 if (lreq->ping_req)
2733 ceph_osdc_put_request(lreq->ping_req);
2734 target_destroy(&lreq->t);
2735 kfree(lreq);
2736}
2737
2738static void linger_put(struct ceph_osd_linger_request *lreq)
2739{
2740 if (lreq)
2741 kref_put(&lreq->kref, linger_release);
2742}
2743
2744static struct ceph_osd_linger_request *
2745linger_get(struct ceph_osd_linger_request *lreq)
2746{
2747 kref_get(&lreq->kref);
2748 return lreq;
2749}
2750
2751static struct ceph_osd_linger_request *
2752linger_alloc(struct ceph_osd_client *osdc)
2753{
2754 struct ceph_osd_linger_request *lreq;
2755
2756 lreq = kzalloc(sizeof(*lreq), GFP_NOIO);
2757 if (!lreq)
2758 return NULL;
2759
2760 kref_init(&lreq->kref);
2761 mutex_init(&lreq->lock);
2762 RB_CLEAR_NODE(&lreq->node);
2763 RB_CLEAR_NODE(&lreq->osdc_node);
2764 RB_CLEAR_NODE(&lreq->mc_node);
2765 INIT_LIST_HEAD(&lreq->scan_item);
2766 INIT_LIST_HEAD(&lreq->pending_lworks);
2767 init_completion(&lreq->reg_commit_wait);
2768 init_completion(&lreq->notify_finish_wait);
2769
2770 lreq->osdc = osdc;
2771 target_init(&lreq->t);
2772
2773 dout("%s lreq %p\n", __func__, lreq);
2774 return lreq;
2775}
2776
2777DEFINE_RB_INSDEL_FUNCS(linger, struct ceph_osd_linger_request, linger_id, node)
2778DEFINE_RB_FUNCS(linger_osdc, struct ceph_osd_linger_request, linger_id, osdc_node)
2779DEFINE_RB_FUNCS(linger_mc, struct ceph_osd_linger_request, linger_id, mc_node)
2780
2781/*
2782 * Create linger request <-> OSD session relation.
2783 *
2784 * @lreq has to be registered, @osd may be homeless.
2785 */
2786static void link_linger(struct ceph_osd *osd,
2787 struct ceph_osd_linger_request *lreq)
2788{
2789 verify_osd_locked(osd);
2790 WARN_ON(!lreq->linger_id || lreq->osd);
2791 dout("%s osd %p osd%d lreq %p linger_id %llu\n", __func__, osd,
2792 osd->o_osd, lreq, lreq->linger_id);
2793
2794 if (!osd_homeless(osd))
2795 __remove_osd_from_lru(osd);
2796 else
2797 atomic_inc(&osd->o_osdc->num_homeless);
2798
2799 get_osd(osd);
2800 insert_linger(&osd->o_linger_requests, lreq);
2801 lreq->osd = osd;
2802}
2803
2804static void unlink_linger(struct ceph_osd *osd,
2805 struct ceph_osd_linger_request *lreq)
2806{
2807 verify_osd_locked(osd);
2808 WARN_ON(lreq->osd != osd);
2809 dout("%s osd %p osd%d lreq %p linger_id %llu\n", __func__, osd,
2810 osd->o_osd, lreq, lreq->linger_id);
2811
2812 lreq->osd = NULL;
2813 erase_linger(&osd->o_linger_requests, lreq);
2814 put_osd(osd);
2815
2816 if (!osd_homeless(osd))
2817 maybe_move_osd_to_lru(osd);
2818 else
2819 atomic_dec(&osd->o_osdc->num_homeless);
2820}
2821
2822static bool __linger_registered(struct ceph_osd_linger_request *lreq)
2823{
2824 verify_osdc_locked(lreq->osdc);
2825
2826 return !RB_EMPTY_NODE(&lreq->osdc_node);
2827}
2828
2829static bool linger_registered(struct ceph_osd_linger_request *lreq)
2830{
2831 struct ceph_osd_client *osdc = lreq->osdc;
2832 bool registered;
2833
2834 down_read(&osdc->lock);
2835 registered = __linger_registered(lreq);
2836 up_read(&osdc->lock);
2837
2838 return registered;
2839}
2840
2841static void linger_register(struct ceph_osd_linger_request *lreq)
2842{
2843 struct ceph_osd_client *osdc = lreq->osdc;
2844
2845 verify_osdc_wrlocked(osdc);
2846 WARN_ON(lreq->linger_id);
2847
2848 linger_get(lreq);
2849 lreq->linger_id = ++osdc->last_linger_id;
2850 insert_linger_osdc(&osdc->linger_requests, lreq);
2851}
2852
2853static void linger_unregister(struct ceph_osd_linger_request *lreq)
2854{
2855 struct ceph_osd_client *osdc = lreq->osdc;
2856
2857 verify_osdc_wrlocked(osdc);
2858
2859 erase_linger_osdc(&osdc->linger_requests, lreq);
2860 linger_put(lreq);
2861}
2862
2863static void cancel_linger_request(struct ceph_osd_request *req)
2864{
2865 struct ceph_osd_linger_request *lreq = req->r_priv;
2866
2867 WARN_ON(!req->r_linger);
2868 cancel_request(req);
2869 linger_put(lreq);
2870}
2871
2872struct linger_work {
2873 struct work_struct work;
2874 struct ceph_osd_linger_request *lreq;
2875 struct list_head pending_item;
2876 unsigned long queued_stamp;
2877
2878 union {
2879 struct {
2880 u64 notify_id;
2881 u64 notifier_id;
2882 void *payload; /* points into @msg front */
2883 size_t payload_len;
2884
2885 struct ceph_msg *msg; /* for ceph_msg_put() */
2886 } notify;
2887 struct {
2888 int err;
2889 } error;
2890 };
2891};
2892
2893static struct linger_work *lwork_alloc(struct ceph_osd_linger_request *lreq,
2894 work_func_t workfn)
2895{
2896 struct linger_work *lwork;
2897
2898 lwork = kzalloc(sizeof(*lwork), GFP_NOIO);
2899 if (!lwork)
2900 return NULL;
2901
2902 INIT_WORK(&lwork->work, workfn);
2903 INIT_LIST_HEAD(&lwork->pending_item);
2904 lwork->lreq = linger_get(lreq);
2905
2906 return lwork;
2907}
2908
2909static void lwork_free(struct linger_work *lwork)
2910{
2911 struct ceph_osd_linger_request *lreq = lwork->lreq;
2912
2913 mutex_lock(&lreq->lock);
2914 list_del(&lwork->pending_item);
2915 mutex_unlock(&lreq->lock);
2916
2917 linger_put(lreq);
2918 kfree(lwork);
2919}
2920
2921static void lwork_queue(struct linger_work *lwork)
2922{
2923 struct ceph_osd_linger_request *lreq = lwork->lreq;
2924 struct ceph_osd_client *osdc = lreq->osdc;
2925
2926 verify_lreq_locked(lreq);
2927 WARN_ON(!list_empty(&lwork->pending_item));
2928
2929 lwork->queued_stamp = jiffies;
2930 list_add_tail(&lwork->pending_item, &lreq->pending_lworks);
2931 queue_work(osdc->notify_wq, &lwork->work);
2932}
2933
2934static void do_watch_notify(struct work_struct *w)
2935{
2936 struct linger_work *lwork = container_of(w, struct linger_work, work);
2937 struct ceph_osd_linger_request *lreq = lwork->lreq;
2938
2939 if (!linger_registered(lreq)) {
2940 dout("%s lreq %p not registered\n", __func__, lreq);
2941 goto out;
2942 }
2943
2944 WARN_ON(!lreq->is_watch);
2945 dout("%s lreq %p notify_id %llu notifier_id %llu payload_len %zu\n",
2946 __func__, lreq, lwork->notify.notify_id, lwork->notify.notifier_id,
2947 lwork->notify.payload_len);
2948 lreq->wcb(lreq->data, lwork->notify.notify_id, lreq->linger_id,
2949 lwork->notify.notifier_id, lwork->notify.payload,
2950 lwork->notify.payload_len);
2951
2952out:
2953 ceph_msg_put(lwork->notify.msg);
2954 lwork_free(lwork);
2955}
2956
2957static void do_watch_error(struct work_struct *w)
2958{
2959 struct linger_work *lwork = container_of(w, struct linger_work, work);
2960 struct ceph_osd_linger_request *lreq = lwork->lreq;
2961
2962 if (!linger_registered(lreq)) {
2963 dout("%s lreq %p not registered\n", __func__, lreq);
2964 goto out;
2965 }
2966
2967 dout("%s lreq %p err %d\n", __func__, lreq, lwork->error.err);
2968 lreq->errcb(lreq->data, lreq->linger_id, lwork->error.err);
2969
2970out:
2971 lwork_free(lwork);
2972}
2973
2974static void queue_watch_error(struct ceph_osd_linger_request *lreq)
2975{
2976 struct linger_work *lwork;
2977
2978 lwork = lwork_alloc(lreq, do_watch_error);
2979 if (!lwork) {
2980 pr_err("failed to allocate error-lwork\n");
2981 return;
2982 }
2983
2984 lwork->error.err = lreq->last_error;
2985 lwork_queue(lwork);
2986}
2987
2988static void linger_reg_commit_complete(struct ceph_osd_linger_request *lreq,
2989 int result)
2990{
2991 if (!completion_done(&lreq->reg_commit_wait)) {
2992 lreq->reg_commit_error = (result <= 0 ? result : 0);
2993 complete_all(&lreq->reg_commit_wait);
2994 }
2995}
2996
2997static void linger_commit_cb(struct ceph_osd_request *req)
2998{
2999 struct ceph_osd_linger_request *lreq = req->r_priv;
3000
3001 mutex_lock(&lreq->lock);
3002 dout("%s lreq %p linger_id %llu result %d\n", __func__, lreq,
3003 lreq->linger_id, req->r_result);
3004 linger_reg_commit_complete(lreq, req->r_result);
3005 lreq->committed = true;
3006
3007 if (!lreq->is_watch) {
3008 struct ceph_osd_data *osd_data =
3009 osd_req_op_data(req, 0, notify, response_data);
3010 void *p = page_address(osd_data->pages[0]);
3011
3012 WARN_ON(req->r_ops[0].op != CEPH_OSD_OP_NOTIFY ||
3013 osd_data->type != CEPH_OSD_DATA_TYPE_PAGES);
3014
3015 /* make note of the notify_id */
3016 if (req->r_ops[0].outdata_len >= sizeof(u64)) {
3017 lreq->notify_id = ceph_decode_64(&p);
3018 dout("lreq %p notify_id %llu\n", lreq,
3019 lreq->notify_id);
3020 } else {
3021 dout("lreq %p no notify_id\n", lreq);
3022 }
3023 }
3024
3025 mutex_unlock(&lreq->lock);
3026 linger_put(lreq);
3027}
3028
3029static int normalize_watch_error(int err)
3030{
3031 /*
3032 * Translate ENOENT -> ENOTCONN so that a delete->disconnection
3033 * notification and a failure to reconnect because we raced with
3034 * the delete appear the same to the user.
3035 */
3036 if (err == -ENOENT)
3037 err = -ENOTCONN;
3038
3039 return err;
3040}
3041
3042static void linger_reconnect_cb(struct ceph_osd_request *req)
3043{
3044 struct ceph_osd_linger_request *lreq = req->r_priv;
3045
3046 mutex_lock(&lreq->lock);
3047 dout("%s lreq %p linger_id %llu result %d last_error %d\n", __func__,
3048 lreq, lreq->linger_id, req->r_result, lreq->last_error);
3049 if (req->r_result < 0) {
3050 if (!lreq->last_error) {
3051 lreq->last_error = normalize_watch_error(req->r_result);
3052 queue_watch_error(lreq);
3053 }
3054 }
3055
3056 mutex_unlock(&lreq->lock);
3057 linger_put(lreq);
3058}
3059
3060static void send_linger(struct ceph_osd_linger_request *lreq)
3061{
3062 struct ceph_osd_request *req = lreq->reg_req;
3063 struct ceph_osd_req_op *op = &req->r_ops[0];
3064
3065 verify_osdc_wrlocked(req->r_osdc);
3066 dout("%s lreq %p linger_id %llu\n", __func__, lreq, lreq->linger_id);
3067
3068 if (req->r_osd)
3069 cancel_linger_request(req);
3070
3071 request_reinit(req);
3072 target_copy(&req->r_t, &lreq->t);
3073 req->r_mtime = lreq->mtime;
3074
3075 mutex_lock(&lreq->lock);
3076 if (lreq->is_watch && lreq->committed) {
3077 WARN_ON(op->op != CEPH_OSD_OP_WATCH ||
3078 op->watch.cookie != lreq->linger_id);
3079 op->watch.op = CEPH_OSD_WATCH_OP_RECONNECT;
3080 op->watch.gen = ++lreq->register_gen;
3081 dout("lreq %p reconnect register_gen %u\n", lreq,
3082 op->watch.gen);
3083 req->r_callback = linger_reconnect_cb;
3084 } else {
3085 if (!lreq->is_watch)
3086 lreq->notify_id = 0;
3087 else
3088 WARN_ON(op->watch.op != CEPH_OSD_WATCH_OP_WATCH);
3089 dout("lreq %p register\n", lreq);
3090 req->r_callback = linger_commit_cb;
3091 }
3092 mutex_unlock(&lreq->lock);
3093
3094 req->r_priv = linger_get(lreq);
3095 req->r_linger = true;
3096
3097 submit_request(req, true);
3098}
3099
3100static void linger_ping_cb(struct ceph_osd_request *req)
3101{
3102 struct ceph_osd_linger_request *lreq = req->r_priv;
3103
3104 mutex_lock(&lreq->lock);
3105 dout("%s lreq %p linger_id %llu result %d ping_sent %lu last_error %d\n",
3106 __func__, lreq, lreq->linger_id, req->r_result, lreq->ping_sent,
3107 lreq->last_error);
3108 if (lreq->register_gen == req->r_ops[0].watch.gen) {
3109 if (!req->r_result) {
3110 lreq->watch_valid_thru = lreq->ping_sent;
3111 } else if (!lreq->last_error) {
3112 lreq->last_error = normalize_watch_error(req->r_result);
3113 queue_watch_error(lreq);
3114 }
3115 } else {
3116 dout("lreq %p register_gen %u ignoring old pong %u\n", lreq,
3117 lreq->register_gen, req->r_ops[0].watch.gen);
3118 }
3119
3120 mutex_unlock(&lreq->lock);
3121 linger_put(lreq);
3122}
3123
3124static void send_linger_ping(struct ceph_osd_linger_request *lreq)
3125{
3126 struct ceph_osd_client *osdc = lreq->osdc;
3127 struct ceph_osd_request *req = lreq->ping_req;
3128 struct ceph_osd_req_op *op = &req->r_ops[0];
3129
3130 if (ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD)) {
3131 dout("%s PAUSERD\n", __func__);
3132 return;
3133 }
3134
3135 lreq->ping_sent = jiffies;
3136 dout("%s lreq %p linger_id %llu ping_sent %lu register_gen %u\n",
3137 __func__, lreq, lreq->linger_id, lreq->ping_sent,
3138 lreq->register_gen);
3139
3140 if (req->r_osd)
3141 cancel_linger_request(req);
3142
3143 request_reinit(req);
3144 target_copy(&req->r_t, &lreq->t);
3145
3146 WARN_ON(op->op != CEPH_OSD_OP_WATCH ||
3147 op->watch.cookie != lreq->linger_id ||
3148 op->watch.op != CEPH_OSD_WATCH_OP_PING);
3149 op->watch.gen = lreq->register_gen;
3150 req->r_callback = linger_ping_cb;
3151 req->r_priv = linger_get(lreq);
3152 req->r_linger = true;
3153
3154 ceph_osdc_get_request(req);
3155 account_request(req);
3156 req->r_tid = atomic64_inc_return(&osdc->last_tid);
3157 link_request(lreq->osd, req);
3158 send_request(req);
3159}
3160
3161static void linger_submit(struct ceph_osd_linger_request *lreq)
3162{
3163 struct ceph_osd_client *osdc = lreq->osdc;
3164 struct ceph_osd *osd;
3165
3166 down_write(&osdc->lock);
3167 linger_register(lreq);
3168 if (lreq->is_watch) {
3169 lreq->reg_req->r_ops[0].watch.cookie = lreq->linger_id;
3170 lreq->ping_req->r_ops[0].watch.cookie = lreq->linger_id;
3171 } else {
3172 lreq->reg_req->r_ops[0].notify.cookie = lreq->linger_id;
3173 }
3174
3175 calc_target(osdc, &lreq->t, false);
3176 osd = lookup_create_osd(osdc, lreq->t.osd, true);
3177 link_linger(osd, lreq);
3178
3179 send_linger(lreq);
3180 up_write(&osdc->lock);
3181}
3182
3183static void cancel_linger_map_check(struct ceph_osd_linger_request *lreq)
3184{
3185 struct ceph_osd_client *osdc = lreq->osdc;
3186 struct ceph_osd_linger_request *lookup_lreq;
3187
3188 verify_osdc_wrlocked(osdc);
3189
3190 lookup_lreq = lookup_linger_mc(&osdc->linger_map_checks,
3191 lreq->linger_id);
3192 if (!lookup_lreq)
3193 return;
3194
3195 WARN_ON(lookup_lreq != lreq);
3196 erase_linger_mc(&osdc->linger_map_checks, lreq);
3197 linger_put(lreq);
3198}
3199
3200/*
3201 * @lreq has to be both registered and linked.
3202 */
3203static void __linger_cancel(struct ceph_osd_linger_request *lreq)
3204{
3205 if (lreq->is_watch && lreq->ping_req->r_osd)
3206 cancel_linger_request(lreq->ping_req);
3207 if (lreq->reg_req->r_osd)
3208 cancel_linger_request(lreq->reg_req);
3209 cancel_linger_map_check(lreq);
3210 unlink_linger(lreq->osd, lreq);
3211 linger_unregister(lreq);
3212}
3213
3214static void linger_cancel(struct ceph_osd_linger_request *lreq)
3215{
3216 struct ceph_osd_client *osdc = lreq->osdc;
3217
3218 down_write(&osdc->lock);
3219 if (__linger_registered(lreq))
3220 __linger_cancel(lreq);
3221 up_write(&osdc->lock);
3222}
3223
3224static void send_linger_map_check(struct ceph_osd_linger_request *lreq);
3225
3226static void check_linger_pool_dne(struct ceph_osd_linger_request *lreq)
3227{
3228 struct ceph_osd_client *osdc = lreq->osdc;
3229 struct ceph_osdmap *map = osdc->osdmap;
3230
3231 verify_osdc_wrlocked(osdc);
3232 WARN_ON(!map->epoch);
3233
3234 if (lreq->register_gen) {
3235 lreq->map_dne_bound = map->epoch;
3236 dout("%s lreq %p linger_id %llu pool disappeared\n", __func__,
3237 lreq, lreq->linger_id);
3238 } else {
3239 dout("%s lreq %p linger_id %llu map_dne_bound %u have %u\n",
3240 __func__, lreq, lreq->linger_id, lreq->map_dne_bound,
3241 map->epoch);
3242 }
3243
3244 if (lreq->map_dne_bound) {
3245 if (map->epoch >= lreq->map_dne_bound) {
3246 /* we had a new enough map */
3247 pr_info("linger_id %llu pool does not exist\n",
3248 lreq->linger_id);
3249 linger_reg_commit_complete(lreq, -ENOENT);
3250 __linger_cancel(lreq);
3251 }
3252 } else {
3253 send_linger_map_check(lreq);
3254 }
3255}
3256
3257static void linger_map_check_cb(struct ceph_mon_generic_request *greq)
3258{
3259 struct ceph_osd_client *osdc = &greq->monc->client->osdc;
3260 struct ceph_osd_linger_request *lreq;
3261 u64 linger_id = greq->private_data;
3262
3263 WARN_ON(greq->result || !greq->u.newest);
3264
3265 down_write(&osdc->lock);
3266 lreq = lookup_linger_mc(&osdc->linger_map_checks, linger_id);
3267 if (!lreq) {
3268 dout("%s linger_id %llu dne\n", __func__, linger_id);
3269 goto out_unlock;
3270 }
3271
3272 dout("%s lreq %p linger_id %llu map_dne_bound %u newest %llu\n",
3273 __func__, lreq, lreq->linger_id, lreq->map_dne_bound,
3274 greq->u.newest);
3275 if (!lreq->map_dne_bound)
3276 lreq->map_dne_bound = greq->u.newest;
3277 erase_linger_mc(&osdc->linger_map_checks, lreq);
3278 check_linger_pool_dne(lreq);
3279
3280 linger_put(lreq);
3281out_unlock:
3282 up_write(&osdc->lock);
3283}
3284
3285static void send_linger_map_check(struct ceph_osd_linger_request *lreq)
3286{
3287 struct ceph_osd_client *osdc = lreq->osdc;
3288 struct ceph_osd_linger_request *lookup_lreq;
3289 int ret;
3290
3291 verify_osdc_wrlocked(osdc);
3292
3293 lookup_lreq = lookup_linger_mc(&osdc->linger_map_checks,
3294 lreq->linger_id);
3295 if (lookup_lreq) {
3296 WARN_ON(lookup_lreq != lreq);
3297 return;
3298 }
3299
3300 linger_get(lreq);
3301 insert_linger_mc(&osdc->linger_map_checks, lreq);
3302 ret = ceph_monc_get_version_async(&osdc->client->monc, "osdmap",
3303 linger_map_check_cb, lreq->linger_id);
3304 WARN_ON(ret);
3305}
3306
3307static int linger_reg_commit_wait(struct ceph_osd_linger_request *lreq)
3308{
3309 int ret;
3310
3311 dout("%s lreq %p linger_id %llu\n", __func__, lreq, lreq->linger_id);
3312 ret = wait_for_completion_interruptible(&lreq->reg_commit_wait);
3313 return ret ?: lreq->reg_commit_error;
3314}
3315
3316static int linger_notify_finish_wait(struct ceph_osd_linger_request *lreq)
3317{
3318 int ret;
3319
3320 dout("%s lreq %p linger_id %llu\n", __func__, lreq, lreq->linger_id);
3321 ret = wait_for_completion_interruptible(&lreq->notify_finish_wait);
3322 return ret ?: lreq->notify_finish_error;
3323}
3324
3325/*
3326 * Timeout callback, called every N seconds. When 1 or more OSD
3327 * requests has been active for more than N seconds, we send a keepalive
3328 * (tag + timestamp) to its OSD to ensure any communications channel
3329 * reset is detected.
3330 */
3331static void handle_timeout(struct work_struct *work)
3332{
3333 struct ceph_osd_client *osdc =
3334 container_of(work, struct ceph_osd_client, timeout_work.work);
3335 struct ceph_options *opts = osdc->client->options;
3336 unsigned long cutoff = jiffies - opts->osd_keepalive_timeout;
3337 unsigned long expiry_cutoff = jiffies - opts->osd_request_timeout;
3338 LIST_HEAD(slow_osds);
3339 struct rb_node *n, *p;
3340
3341 dout("%s osdc %p\n", __func__, osdc);
3342 down_write(&osdc->lock);
3343
3344 /*
3345 * ping osds that are a bit slow. this ensures that if there
3346 * is a break in the TCP connection we will notice, and reopen
3347 * a connection with that osd (from the fault callback).
3348 */
3349 for (n = rb_first(&osdc->osds); n; n = rb_next(n)) {
3350 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
3351 bool found = false;
3352
3353 for (p = rb_first(&osd->o_requests); p; ) {
3354 struct ceph_osd_request *req =
3355 rb_entry(p, struct ceph_osd_request, r_node);
3356
3357 p = rb_next(p); /* abort_request() */
3358
3359 if (time_before(req->r_stamp, cutoff)) {
3360 dout(" req %p tid %llu on osd%d is laggy\n",
3361 req, req->r_tid, osd->o_osd);
3362 found = true;
3363 }
3364 if (opts->osd_request_timeout &&
3365 time_before(req->r_start_stamp, expiry_cutoff)) {
3366 pr_err_ratelimited("tid %llu on osd%d timeout\n",
3367 req->r_tid, osd->o_osd);
3368 abort_request(req, -ETIMEDOUT);
3369 }
3370 }
3371 for (p = rb_first(&osd->o_linger_requests); p; p = rb_next(p)) {
3372 struct ceph_osd_linger_request *lreq =
3373 rb_entry(p, struct ceph_osd_linger_request, node);
3374
3375 dout(" lreq %p linger_id %llu is served by osd%d\n",
3376 lreq, lreq->linger_id, osd->o_osd);
3377 found = true;
3378
3379 mutex_lock(&lreq->lock);
3380 if (lreq->is_watch && lreq->committed && !lreq->last_error)
3381 send_linger_ping(lreq);
3382 mutex_unlock(&lreq->lock);
3383 }
3384
3385 if (found)
3386 list_move_tail(&osd->o_keepalive_item, &slow_osds);
3387 }
3388
3389 if (opts->osd_request_timeout) {
3390 for (p = rb_first(&osdc->homeless_osd.o_requests); p; ) {
3391 struct ceph_osd_request *req =
3392 rb_entry(p, struct ceph_osd_request, r_node);
3393
3394 p = rb_next(p); /* abort_request() */
3395
3396 if (time_before(req->r_start_stamp, expiry_cutoff)) {
3397 pr_err_ratelimited("tid %llu on osd%d timeout\n",
3398 req->r_tid, osdc->homeless_osd.o_osd);
3399 abort_request(req, -ETIMEDOUT);
3400 }
3401 }
3402 }
3403
3404 if (atomic_read(&osdc->num_homeless) || !list_empty(&slow_osds))
3405 maybe_request_map(osdc);
3406
3407 while (!list_empty(&slow_osds)) {
3408 struct ceph_osd *osd = list_first_entry(&slow_osds,
3409 struct ceph_osd,
3410 o_keepalive_item);
3411 list_del_init(&osd->o_keepalive_item);
3412 ceph_con_keepalive(&osd->o_con);
3413 }
3414
3415 up_write(&osdc->lock);
3416 schedule_delayed_work(&osdc->timeout_work,
3417 osdc->client->options->osd_keepalive_timeout);
3418}
3419
3420static void handle_osds_timeout(struct work_struct *work)
3421{
3422 struct ceph_osd_client *osdc =
3423 container_of(work, struct ceph_osd_client,
3424 osds_timeout_work.work);
3425 unsigned long delay = osdc->client->options->osd_idle_ttl / 4;
3426 struct ceph_osd *osd, *nosd;
3427
3428 dout("%s osdc %p\n", __func__, osdc);
3429 down_write(&osdc->lock);
3430 list_for_each_entry_safe(osd, nosd, &osdc->osd_lru, o_osd_lru) {
3431 if (time_before(jiffies, osd->lru_ttl))
3432 break;
3433
3434 WARN_ON(!RB_EMPTY_ROOT(&osd->o_requests));
3435 WARN_ON(!RB_EMPTY_ROOT(&osd->o_linger_requests));
3436 close_osd(osd);
3437 }
3438
3439 up_write(&osdc->lock);
3440 schedule_delayed_work(&osdc->osds_timeout_work,
3441 round_jiffies_relative(delay));
3442}
3443
3444static int ceph_oloc_decode(void **p, void *end,
3445 struct ceph_object_locator *oloc)
3446{
3447 u8 struct_v, struct_cv;
3448 u32 len;
3449 void *struct_end;
3450 int ret = 0;
3451
3452 ceph_decode_need(p, end, 1 + 1 + 4, e_inval);
3453 struct_v = ceph_decode_8(p);
3454 struct_cv = ceph_decode_8(p);
3455 if (struct_v < 3) {
3456 pr_warn("got v %d < 3 cv %d of ceph_object_locator\n",
3457 struct_v, struct_cv);
3458 goto e_inval;
3459 }
3460 if (struct_cv > 6) {
3461 pr_warn("got v %d cv %d > 6 of ceph_object_locator\n",
3462 struct_v, struct_cv);
3463 goto e_inval;
3464 }
3465 len = ceph_decode_32(p);
3466 ceph_decode_need(p, end, len, e_inval);
3467 struct_end = *p + len;
3468
3469 oloc->pool = ceph_decode_64(p);
3470 *p += 4; /* skip preferred */
3471
3472 len = ceph_decode_32(p);
3473 if (len > 0) {
3474 pr_warn("ceph_object_locator::key is set\n");
3475 goto e_inval;
3476 }
3477
3478 if (struct_v >= 5) {
3479 bool changed = false;
3480
3481 len = ceph_decode_32(p);
3482 if (len > 0) {
3483 ceph_decode_need(p, end, len, e_inval);
3484 if (!oloc->pool_ns ||
3485 ceph_compare_string(oloc->pool_ns, *p, len))
3486 changed = true;
3487 *p += len;
3488 } else {
3489 if (oloc->pool_ns)
3490 changed = true;
3491 }
3492 if (changed) {
3493 /* redirect changes namespace */
3494 pr_warn("ceph_object_locator::nspace is changed\n");
3495 goto e_inval;
3496 }
3497 }
3498
3499 if (struct_v >= 6) {
3500 s64 hash = ceph_decode_64(p);
3501 if (hash != -1) {
3502 pr_warn("ceph_object_locator::hash is set\n");
3503 goto e_inval;
3504 }
3505 }
3506
3507 /* skip the rest */
3508 *p = struct_end;
3509out:
3510 return ret;
3511
3512e_inval:
3513 ret = -EINVAL;
3514 goto out;
3515}
3516
3517static int ceph_redirect_decode(void **p, void *end,
3518 struct ceph_request_redirect *redir)
3519{
3520 u8 struct_v, struct_cv;
3521 u32 len;
3522 void *struct_end;
3523 int ret;
3524
3525 ceph_decode_need(p, end, 1 + 1 + 4, e_inval);
3526 struct_v = ceph_decode_8(p);
3527 struct_cv = ceph_decode_8(p);
3528 if (struct_cv > 1) {
3529 pr_warn("got v %d cv %d > 1 of ceph_request_redirect\n",
3530 struct_v, struct_cv);
3531 goto e_inval;
3532 }
3533 len = ceph_decode_32(p);
3534 ceph_decode_need(p, end, len, e_inval);
3535 struct_end = *p + len;
3536
3537 ret = ceph_oloc_decode(p, end, &redir->oloc);
3538 if (ret)
3539 goto out;
3540
3541 len = ceph_decode_32(p);
3542 if (len > 0) {
3543 pr_warn("ceph_request_redirect::object_name is set\n");
3544 goto e_inval;
3545 }
3546
3547 /* skip the rest */
3548 *p = struct_end;
3549out:
3550 return ret;
3551
3552e_inval:
3553 ret = -EINVAL;
3554 goto out;
3555}
3556
3557struct MOSDOpReply {
3558 struct ceph_pg pgid;
3559 u64 flags;
3560 int result;
3561 u32 epoch;
3562 int num_ops;
3563 u32 outdata_len[CEPH_OSD_MAX_OPS];
3564 s32 rval[CEPH_OSD_MAX_OPS];
3565 int retry_attempt;
3566 struct ceph_eversion replay_version;
3567 u64 user_version;
3568 struct ceph_request_redirect redirect;
3569};
3570
3571static int decode_MOSDOpReply(const struct ceph_msg *msg, struct MOSDOpReply *m)
3572{
3573 void *p = msg->front.iov_base;
3574 void *const end = p + msg->front.iov_len;
3575 u16 version = le16_to_cpu(msg->hdr.version);
3576 struct ceph_eversion bad_replay_version;
3577 u8 decode_redir;
3578 u32 len;
3579 int ret;
3580 int i;
3581
3582 ceph_decode_32_safe(&p, end, len, e_inval);
3583 ceph_decode_need(&p, end, len, e_inval);
3584 p += len; /* skip oid */
3585
3586 ret = ceph_decode_pgid(&p, end, &m->pgid);
3587 if (ret)
3588 return ret;
3589
3590 ceph_decode_64_safe(&p, end, m->flags, e_inval);
3591 ceph_decode_32_safe(&p, end, m->result, e_inval);
3592 ceph_decode_need(&p, end, sizeof(bad_replay_version), e_inval);
3593 memcpy(&bad_replay_version, p, sizeof(bad_replay_version));
3594 p += sizeof(bad_replay_version);
3595 ceph_decode_32_safe(&p, end, m->epoch, e_inval);
3596
3597 ceph_decode_32_safe(&p, end, m->num_ops, e_inval);
3598 if (m->num_ops > ARRAY_SIZE(m->outdata_len))
3599 goto e_inval;
3600
3601 ceph_decode_need(&p, end, m->num_ops * sizeof(struct ceph_osd_op),
3602 e_inval);
3603 for (i = 0; i < m->num_ops; i++) {
3604 struct ceph_osd_op *op = p;
3605
3606 m->outdata_len[i] = le32_to_cpu(op->payload_len);
3607 p += sizeof(*op);
3608 }
3609
3610 ceph_decode_32_safe(&p, end, m->retry_attempt, e_inval);
3611 for (i = 0; i < m->num_ops; i++)
3612 ceph_decode_32_safe(&p, end, m->rval[i], e_inval);
3613
3614 if (version >= 5) {
3615 ceph_decode_need(&p, end, sizeof(m->replay_version), e_inval);
3616 memcpy(&m->replay_version, p, sizeof(m->replay_version));
3617 p += sizeof(m->replay_version);
3618 ceph_decode_64_safe(&p, end, m->user_version, e_inval);
3619 } else {
3620 m->replay_version = bad_replay_version; /* struct */
3621 m->user_version = le64_to_cpu(m->replay_version.version);
3622 }
3623
3624 if (version >= 6) {
3625 if (version >= 7)
3626 ceph_decode_8_safe(&p, end, decode_redir, e_inval);
3627 else
3628 decode_redir = 1;
3629 } else {
3630 decode_redir = 0;
3631 }
3632
3633 if (decode_redir) {
3634 ret = ceph_redirect_decode(&p, end, &m->redirect);
3635 if (ret)
3636 return ret;
3637 } else {
3638 ceph_oloc_init(&m->redirect.oloc);
3639 }
3640
3641 return 0;
3642
3643e_inval:
3644 return -EINVAL;
3645}
3646
3647/*
3648 * Handle MOSDOpReply. Set ->r_result and call the callback if it is
3649 * specified.
3650 */
3651static void handle_reply(struct ceph_osd *osd, struct ceph_msg *msg)
3652{
3653 struct ceph_osd_client *osdc = osd->o_osdc;
3654 struct ceph_osd_request *req;
3655 struct MOSDOpReply m;
3656 u64 tid = le64_to_cpu(msg->hdr.tid);
3657 u32 data_len = 0;
3658 int ret;
3659 int i;
3660
3661 dout("%s msg %p tid %llu\n", __func__, msg, tid);
3662
3663 down_read(&osdc->lock);
3664 if (!osd_registered(osd)) {
3665 dout("%s osd%d unknown\n", __func__, osd->o_osd);
3666 goto out_unlock_osdc;
3667 }
3668 WARN_ON(osd->o_osd != le64_to_cpu(msg->hdr.src.num));
3669
3670 mutex_lock(&osd->lock);
3671 req = lookup_request(&osd->o_requests, tid);
3672 if (!req) {
3673 dout("%s osd%d tid %llu unknown\n", __func__, osd->o_osd, tid);
3674 goto out_unlock_session;
3675 }
3676
3677 m.redirect.oloc.pool_ns = req->r_t.target_oloc.pool_ns;
3678 ret = decode_MOSDOpReply(msg, &m);
3679 m.redirect.oloc.pool_ns = NULL;
3680 if (ret) {
3681 pr_err("failed to decode MOSDOpReply for tid %llu: %d\n",
3682 req->r_tid, ret);
3683 ceph_msg_dump(msg);
3684 goto fail_request;
3685 }
3686 dout("%s req %p tid %llu flags 0x%llx pgid %llu.%x epoch %u attempt %d v %u'%llu uv %llu\n",
3687 __func__, req, req->r_tid, m.flags, m.pgid.pool, m.pgid.seed,
3688 m.epoch, m.retry_attempt, le32_to_cpu(m.replay_version.epoch),
3689 le64_to_cpu(m.replay_version.version), m.user_version);
3690
3691 if (m.retry_attempt >= 0) {
3692 if (m.retry_attempt != req->r_attempts - 1) {
3693 dout("req %p tid %llu retry_attempt %d != %d, ignoring\n",
3694 req, req->r_tid, m.retry_attempt,
3695 req->r_attempts - 1);
3696 goto out_unlock_session;
3697 }
3698 } else {
3699 WARN_ON(1); /* MOSDOpReply v4 is assumed */
3700 }
3701
3702 if (!ceph_oloc_empty(&m.redirect.oloc)) {
3703 dout("req %p tid %llu redirect pool %lld\n", req, req->r_tid,
3704 m.redirect.oloc.pool);
3705 unlink_request(osd, req);
3706 mutex_unlock(&osd->lock);
3707
3708 /*
3709 * Not ceph_oloc_copy() - changing pool_ns is not
3710 * supported.
3711 */
3712 req->r_t.target_oloc.pool = m.redirect.oloc.pool;
3713 req->r_flags |= CEPH_OSD_FLAG_REDIRECTED |
3714 CEPH_OSD_FLAG_IGNORE_OVERLAY |
3715 CEPH_OSD_FLAG_IGNORE_CACHE;
3716 req->r_tid = 0;
3717 __submit_request(req, false);
3718 goto out_unlock_osdc;
3719 }
3720
3721 if (m.result == -EAGAIN) {
3722 dout("req %p tid %llu EAGAIN\n", req, req->r_tid);
3723 unlink_request(osd, req);
3724 mutex_unlock(&osd->lock);
3725
3726 /*
3727 * The object is missing on the replica or not (yet)
3728 * readable. Clear pgid to force a resend to the primary
3729 * via legacy_change.
3730 */
3731 req->r_t.pgid.pool = 0;
3732 req->r_t.pgid.seed = 0;
3733 WARN_ON(!req->r_t.used_replica);
3734 req->r_flags &= ~(CEPH_OSD_FLAG_BALANCE_READS |
3735 CEPH_OSD_FLAG_LOCALIZE_READS);
3736 req->r_tid = 0;
3737 __submit_request(req, false);
3738 goto out_unlock_osdc;
3739 }
3740
3741 if (m.num_ops != req->r_num_ops) {
3742 pr_err("num_ops %d != %d for tid %llu\n", m.num_ops,
3743 req->r_num_ops, req->r_tid);
3744 goto fail_request;
3745 }
3746 for (i = 0; i < req->r_num_ops; i++) {
3747 dout(" req %p tid %llu op %d rval %d len %u\n", req,
3748 req->r_tid, i, m.rval[i], m.outdata_len[i]);
3749 req->r_ops[i].rval = m.rval[i];
3750 req->r_ops[i].outdata_len = m.outdata_len[i];
3751 data_len += m.outdata_len[i];
3752 }
3753 if (data_len != le32_to_cpu(msg->hdr.data_len)) {
3754 pr_err("sum of lens %u != %u for tid %llu\n", data_len,
3755 le32_to_cpu(msg->hdr.data_len), req->r_tid);
3756 goto fail_request;
3757 }
3758 dout("%s req %p tid %llu result %d data_len %u\n", __func__,
3759 req, req->r_tid, m.result, data_len);
3760
3761 /*
3762 * Since we only ever request ONDISK, we should only ever get
3763 * one (type of) reply back.
3764 */
3765 WARN_ON(!(m.flags & CEPH_OSD_FLAG_ONDISK));
3766 req->r_result = m.result ?: data_len;
3767 finish_request(req);
3768 mutex_unlock(&osd->lock);
3769 up_read(&osdc->lock);
3770
3771 __complete_request(req);
3772 return;
3773
3774fail_request:
3775 complete_request(req, -EIO);
3776out_unlock_session:
3777 mutex_unlock(&osd->lock);
3778out_unlock_osdc:
3779 up_read(&osdc->lock);
3780}
3781
3782static void set_pool_was_full(struct ceph_osd_client *osdc)
3783{
3784 struct rb_node *n;
3785
3786 for (n = rb_first(&osdc->osdmap->pg_pools); n; n = rb_next(n)) {
3787 struct ceph_pg_pool_info *pi =
3788 rb_entry(n, struct ceph_pg_pool_info, node);
3789
3790 pi->was_full = __pool_full(pi);
3791 }
3792}
3793
3794static bool pool_cleared_full(struct ceph_osd_client *osdc, s64 pool_id)
3795{
3796 struct ceph_pg_pool_info *pi;
3797
3798 pi = ceph_pg_pool_by_id(osdc->osdmap, pool_id);
3799 if (!pi)
3800 return false;
3801
3802 return pi->was_full && !__pool_full(pi);
3803}
3804
3805static enum calc_target_result
3806recalc_linger_target(struct ceph_osd_linger_request *lreq)
3807{
3808 struct ceph_osd_client *osdc = lreq->osdc;
3809 enum calc_target_result ct_res;
3810
3811 ct_res = calc_target(osdc, &lreq->t, true);
3812 if (ct_res == CALC_TARGET_NEED_RESEND) {
3813 struct ceph_osd *osd;
3814
3815 osd = lookup_create_osd(osdc, lreq->t.osd, true);
3816 if (osd != lreq->osd) {
3817 unlink_linger(lreq->osd, lreq);
3818 link_linger(osd, lreq);
3819 }
3820 }
3821
3822 return ct_res;
3823}
3824
3825/*
3826 * Requeue requests whose mapping to an OSD has changed.
3827 */
3828static void scan_requests(struct ceph_osd *osd,
3829 bool force_resend,
3830 bool cleared_full,
3831 bool check_pool_cleared_full,
3832 struct rb_root *need_resend,
3833 struct list_head *need_resend_linger)
3834{
3835 struct ceph_osd_client *osdc = osd->o_osdc;
3836 struct rb_node *n;
3837 bool force_resend_writes;
3838
3839 for (n = rb_first(&osd->o_linger_requests); n; ) {
3840 struct ceph_osd_linger_request *lreq =
3841 rb_entry(n, struct ceph_osd_linger_request, node);
3842 enum calc_target_result ct_res;
3843
3844 n = rb_next(n); /* recalc_linger_target() */
3845
3846 dout("%s lreq %p linger_id %llu\n", __func__, lreq,
3847 lreq->linger_id);
3848 ct_res = recalc_linger_target(lreq);
3849 switch (ct_res) {
3850 case CALC_TARGET_NO_ACTION:
3851 force_resend_writes = cleared_full ||
3852 (check_pool_cleared_full &&
3853 pool_cleared_full(osdc, lreq->t.base_oloc.pool));
3854 if (!force_resend && !force_resend_writes)
3855 break;
3856
3857 fallthrough;
3858 case CALC_TARGET_NEED_RESEND:
3859 cancel_linger_map_check(lreq);
3860 /*
3861 * scan_requests() for the previous epoch(s)
3862 * may have already added it to the list, since
3863 * it's not unlinked here.
3864 */
3865 if (list_empty(&lreq->scan_item))
3866 list_add_tail(&lreq->scan_item, need_resend_linger);
3867 break;
3868 case CALC_TARGET_POOL_DNE:
3869 list_del_init(&lreq->scan_item);
3870 check_linger_pool_dne(lreq);
3871 break;
3872 }
3873 }
3874
3875 for (n = rb_first(&osd->o_requests); n; ) {
3876 struct ceph_osd_request *req =
3877 rb_entry(n, struct ceph_osd_request, r_node);
3878 enum calc_target_result ct_res;
3879
3880 n = rb_next(n); /* unlink_request(), check_pool_dne() */
3881
3882 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
3883 ct_res = calc_target(osdc, &req->r_t, false);
3884 switch (ct_res) {
3885 case CALC_TARGET_NO_ACTION:
3886 force_resend_writes = cleared_full ||
3887 (check_pool_cleared_full &&
3888 pool_cleared_full(osdc, req->r_t.base_oloc.pool));
3889 if (!force_resend &&
3890 (!(req->r_flags & CEPH_OSD_FLAG_WRITE) ||
3891 !force_resend_writes))
3892 break;
3893
3894 fallthrough;
3895 case CALC_TARGET_NEED_RESEND:
3896 cancel_map_check(req);
3897 unlink_request(osd, req);
3898 insert_request(need_resend, req);
3899 break;
3900 case CALC_TARGET_POOL_DNE:
3901 check_pool_dne(req);
3902 break;
3903 }
3904 }
3905}
3906
3907static int handle_one_map(struct ceph_osd_client *osdc,
3908 void *p, void *end, bool incremental,
3909 struct rb_root *need_resend,
3910 struct list_head *need_resend_linger)
3911{
3912 struct ceph_osdmap *newmap;
3913 struct rb_node *n;
3914 bool skipped_map = false;
3915 bool was_full;
3916
3917 was_full = ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL);
3918 set_pool_was_full(osdc);
3919
3920 if (incremental)
3921 newmap = osdmap_apply_incremental(&p, end, osdc->osdmap);
3922 else
3923 newmap = ceph_osdmap_decode(&p, end);
3924 if (IS_ERR(newmap))
3925 return PTR_ERR(newmap);
3926
3927 if (newmap != osdc->osdmap) {
3928 /*
3929 * Preserve ->was_full before destroying the old map.
3930 * For pools that weren't in the old map, ->was_full
3931 * should be false.
3932 */
3933 for (n = rb_first(&newmap->pg_pools); n; n = rb_next(n)) {
3934 struct ceph_pg_pool_info *pi =
3935 rb_entry(n, struct ceph_pg_pool_info, node);
3936 struct ceph_pg_pool_info *old_pi;
3937
3938 old_pi = ceph_pg_pool_by_id(osdc->osdmap, pi->id);
3939 if (old_pi)
3940 pi->was_full = old_pi->was_full;
3941 else
3942 WARN_ON(pi->was_full);
3943 }
3944
3945 if (osdc->osdmap->epoch &&
3946 osdc->osdmap->epoch + 1 < newmap->epoch) {
3947 WARN_ON(incremental);
3948 skipped_map = true;
3949 }
3950
3951 ceph_osdmap_destroy(osdc->osdmap);
3952 osdc->osdmap = newmap;
3953 }
3954
3955 was_full &= !ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL);
3956 scan_requests(&osdc->homeless_osd, skipped_map, was_full, true,
3957 need_resend, need_resend_linger);
3958
3959 for (n = rb_first(&osdc->osds); n; ) {
3960 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
3961
3962 n = rb_next(n); /* close_osd() */
3963
3964 scan_requests(osd, skipped_map, was_full, true, need_resend,
3965 need_resend_linger);
3966 if (!ceph_osd_is_up(osdc->osdmap, osd->o_osd) ||
3967 memcmp(&osd->o_con.peer_addr,
3968 ceph_osd_addr(osdc->osdmap, osd->o_osd),
3969 sizeof(struct ceph_entity_addr)))
3970 close_osd(osd);
3971 }
3972
3973 return 0;
3974}
3975
3976static void kick_requests(struct ceph_osd_client *osdc,
3977 struct rb_root *need_resend,
3978 struct list_head *need_resend_linger)
3979{
3980 struct ceph_osd_linger_request *lreq, *nlreq;
3981 enum calc_target_result ct_res;
3982 struct rb_node *n;
3983
3984 /* make sure need_resend targets reflect latest map */
3985 for (n = rb_first(need_resend); n; ) {
3986 struct ceph_osd_request *req =
3987 rb_entry(n, struct ceph_osd_request, r_node);
3988
3989 n = rb_next(n);
3990
3991 if (req->r_t.epoch < osdc->osdmap->epoch) {
3992 ct_res = calc_target(osdc, &req->r_t, false);
3993 if (ct_res == CALC_TARGET_POOL_DNE) {
3994 erase_request(need_resend, req);
3995 check_pool_dne(req);
3996 }
3997 }
3998 }
3999
4000 for (n = rb_first(need_resend); n; ) {
4001 struct ceph_osd_request *req =
4002 rb_entry(n, struct ceph_osd_request, r_node);
4003 struct ceph_osd *osd;
4004
4005 n = rb_next(n);
4006 erase_request(need_resend, req); /* before link_request() */
4007
4008 osd = lookup_create_osd(osdc, req->r_t.osd, true);
4009 link_request(osd, req);
4010 if (!req->r_linger) {
4011 if (!osd_homeless(osd) && !req->r_t.paused)
4012 send_request(req);
4013 } else {
4014 cancel_linger_request(req);
4015 }
4016 }
4017
4018 list_for_each_entry_safe(lreq, nlreq, need_resend_linger, scan_item) {
4019 if (!osd_homeless(lreq->osd))
4020 send_linger(lreq);
4021
4022 list_del_init(&lreq->scan_item);
4023 }
4024}
4025
4026/*
4027 * Process updated osd map.
4028 *
4029 * The message contains any number of incremental and full maps, normally
4030 * indicating some sort of topology change in the cluster. Kick requests
4031 * off to different OSDs as needed.
4032 */
4033void ceph_osdc_handle_map(struct ceph_osd_client *osdc, struct ceph_msg *msg)
4034{
4035 void *p = msg->front.iov_base;
4036 void *const end = p + msg->front.iov_len;
4037 u32 nr_maps, maplen;
4038 u32 epoch;
4039 struct ceph_fsid fsid;
4040 struct rb_root need_resend = RB_ROOT;
4041 LIST_HEAD(need_resend_linger);
4042 bool handled_incremental = false;
4043 bool was_pauserd, was_pausewr;
4044 bool pauserd, pausewr;
4045 int err;
4046
4047 dout("%s have %u\n", __func__, osdc->osdmap->epoch);
4048 down_write(&osdc->lock);
4049
4050 /* verify fsid */
4051 ceph_decode_need(&p, end, sizeof(fsid), bad);
4052 ceph_decode_copy(&p, &fsid, sizeof(fsid));
4053 if (ceph_check_fsid(osdc->client, &fsid) < 0)
4054 goto bad;
4055
4056 was_pauserd = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD);
4057 was_pausewr = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR) ||
4058 ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
4059 have_pool_full(osdc);
4060
4061 /* incremental maps */
4062 ceph_decode_32_safe(&p, end, nr_maps, bad);
4063 dout(" %d inc maps\n", nr_maps);
4064 while (nr_maps > 0) {
4065 ceph_decode_need(&p, end, 2*sizeof(u32), bad);
4066 epoch = ceph_decode_32(&p);
4067 maplen = ceph_decode_32(&p);
4068 ceph_decode_need(&p, end, maplen, bad);
4069 if (osdc->osdmap->epoch &&
4070 osdc->osdmap->epoch + 1 == epoch) {
4071 dout("applying incremental map %u len %d\n",
4072 epoch, maplen);
4073 err = handle_one_map(osdc, p, p + maplen, true,
4074 &need_resend, &need_resend_linger);
4075 if (err)
4076 goto bad;
4077 handled_incremental = true;
4078 } else {
4079 dout("ignoring incremental map %u len %d\n",
4080 epoch, maplen);
4081 }
4082 p += maplen;
4083 nr_maps--;
4084 }
4085 if (handled_incremental)
4086 goto done;
4087
4088 /* full maps */
4089 ceph_decode_32_safe(&p, end, nr_maps, bad);
4090 dout(" %d full maps\n", nr_maps);
4091 while (nr_maps) {
4092 ceph_decode_need(&p, end, 2*sizeof(u32), bad);
4093 epoch = ceph_decode_32(&p);
4094 maplen = ceph_decode_32(&p);
4095 ceph_decode_need(&p, end, maplen, bad);
4096 if (nr_maps > 1) {
4097 dout("skipping non-latest full map %u len %d\n",
4098 epoch, maplen);
4099 } else if (osdc->osdmap->epoch >= epoch) {
4100 dout("skipping full map %u len %d, "
4101 "older than our %u\n", epoch, maplen,
4102 osdc->osdmap->epoch);
4103 } else {
4104 dout("taking full map %u len %d\n", epoch, maplen);
4105 err = handle_one_map(osdc, p, p + maplen, false,
4106 &need_resend, &need_resend_linger);
4107 if (err)
4108 goto bad;
4109 }
4110 p += maplen;
4111 nr_maps--;
4112 }
4113
4114done:
4115 /*
4116 * subscribe to subsequent osdmap updates if full to ensure
4117 * we find out when we are no longer full and stop returning
4118 * ENOSPC.
4119 */
4120 pauserd = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD);
4121 pausewr = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR) ||
4122 ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
4123 have_pool_full(osdc);
4124 if (was_pauserd || was_pausewr || pauserd || pausewr ||
4125 osdc->osdmap->epoch < osdc->epoch_barrier)
4126 maybe_request_map(osdc);
4127
4128 kick_requests(osdc, &need_resend, &need_resend_linger);
4129
4130 ceph_osdc_abort_on_full(osdc);
4131 ceph_monc_got_map(&osdc->client->monc, CEPH_SUB_OSDMAP,
4132 osdc->osdmap->epoch);
4133 up_write(&osdc->lock);
4134 wake_up_all(&osdc->client->auth_wq);
4135 return;
4136
4137bad:
4138 pr_err("osdc handle_map corrupt msg\n");
4139 ceph_msg_dump(msg);
4140 up_write(&osdc->lock);
4141}
4142
4143/*
4144 * Resubmit requests pending on the given osd.
4145 */
4146static void kick_osd_requests(struct ceph_osd *osd)
4147{
4148 struct rb_node *n;
4149
4150 clear_backoffs(osd);
4151
4152 for (n = rb_first(&osd->o_requests); n; ) {
4153 struct ceph_osd_request *req =
4154 rb_entry(n, struct ceph_osd_request, r_node);
4155
4156 n = rb_next(n); /* cancel_linger_request() */
4157
4158 if (!req->r_linger) {
4159 if (!req->r_t.paused)
4160 send_request(req);
4161 } else {
4162 cancel_linger_request(req);
4163 }
4164 }
4165 for (n = rb_first(&osd->o_linger_requests); n; n = rb_next(n)) {
4166 struct ceph_osd_linger_request *lreq =
4167 rb_entry(n, struct ceph_osd_linger_request, node);
4168
4169 send_linger(lreq);
4170 }
4171}
4172
4173/*
4174 * If the osd connection drops, we need to resubmit all requests.
4175 */
4176static void osd_fault(struct ceph_connection *con)
4177{
4178 struct ceph_osd *osd = con->private;
4179 struct ceph_osd_client *osdc = osd->o_osdc;
4180
4181 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
4182
4183 down_write(&osdc->lock);
4184 if (!osd_registered(osd)) {
4185 dout("%s osd%d unknown\n", __func__, osd->o_osd);
4186 goto out_unlock;
4187 }
4188
4189 if (!reopen_osd(osd))
4190 kick_osd_requests(osd);
4191 maybe_request_map(osdc);
4192
4193out_unlock:
4194 up_write(&osdc->lock);
4195}
4196
4197struct MOSDBackoff {
4198 struct ceph_spg spgid;
4199 u32 map_epoch;
4200 u8 op;
4201 u64 id;
4202 struct ceph_hobject_id *begin;
4203 struct ceph_hobject_id *end;
4204};
4205
4206static int decode_MOSDBackoff(const struct ceph_msg *msg, struct MOSDBackoff *m)
4207{
4208 void *p = msg->front.iov_base;
4209 void *const end = p + msg->front.iov_len;
4210 u8 struct_v;
4211 u32 struct_len;
4212 int ret;
4213
4214 ret = ceph_start_decoding(&p, end, 1, "spg_t", &struct_v, &struct_len);
4215 if (ret)
4216 return ret;
4217
4218 ret = ceph_decode_pgid(&p, end, &m->spgid.pgid);
4219 if (ret)
4220 return ret;
4221
4222 ceph_decode_8_safe(&p, end, m->spgid.shard, e_inval);
4223 ceph_decode_32_safe(&p, end, m->map_epoch, e_inval);
4224 ceph_decode_8_safe(&p, end, m->op, e_inval);
4225 ceph_decode_64_safe(&p, end, m->id, e_inval);
4226
4227 m->begin = kzalloc(sizeof(*m->begin), GFP_NOIO);
4228 if (!m->begin)
4229 return -ENOMEM;
4230
4231 ret = decode_hoid(&p, end, m->begin);
4232 if (ret) {
4233 free_hoid(m->begin);
4234 return ret;
4235 }
4236
4237 m->end = kzalloc(sizeof(*m->end), GFP_NOIO);
4238 if (!m->end) {
4239 free_hoid(m->begin);
4240 return -ENOMEM;
4241 }
4242
4243 ret = decode_hoid(&p, end, m->end);
4244 if (ret) {
4245 free_hoid(m->begin);
4246 free_hoid(m->end);
4247 return ret;
4248 }
4249
4250 return 0;
4251
4252e_inval:
4253 return -EINVAL;
4254}
4255
4256static struct ceph_msg *create_backoff_message(
4257 const struct ceph_osd_backoff *backoff,
4258 u32 map_epoch)
4259{
4260 struct ceph_msg *msg;
4261 void *p, *end;
4262 int msg_size;
4263
4264 msg_size = CEPH_ENCODING_START_BLK_LEN +
4265 CEPH_PGID_ENCODING_LEN + 1; /* spgid */
4266 msg_size += 4 + 1 + 8; /* map_epoch, op, id */
4267 msg_size += CEPH_ENCODING_START_BLK_LEN +
4268 hoid_encoding_size(backoff->begin);
4269 msg_size += CEPH_ENCODING_START_BLK_LEN +
4270 hoid_encoding_size(backoff->end);
4271
4272 msg = ceph_msg_new(CEPH_MSG_OSD_BACKOFF, msg_size, GFP_NOIO, true);
4273 if (!msg)
4274 return NULL;
4275
4276 p = msg->front.iov_base;
4277 end = p + msg->front_alloc_len;
4278
4279 encode_spgid(&p, &backoff->spgid);
4280 ceph_encode_32(&p, map_epoch);
4281 ceph_encode_8(&p, CEPH_OSD_BACKOFF_OP_ACK_BLOCK);
4282 ceph_encode_64(&p, backoff->id);
4283 encode_hoid(&p, end, backoff->begin);
4284 encode_hoid(&p, end, backoff->end);
4285 BUG_ON(p != end);
4286
4287 msg->front.iov_len = p - msg->front.iov_base;
4288 msg->hdr.version = cpu_to_le16(1); /* MOSDBackoff v1 */
4289 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
4290
4291 return msg;
4292}
4293
4294static void handle_backoff_block(struct ceph_osd *osd, struct MOSDBackoff *m)
4295{
4296 struct ceph_spg_mapping *spg;
4297 struct ceph_osd_backoff *backoff;
4298 struct ceph_msg *msg;
4299
4300 dout("%s osd%d spgid %llu.%xs%d id %llu\n", __func__, osd->o_osd,
4301 m->spgid.pgid.pool, m->spgid.pgid.seed, m->spgid.shard, m->id);
4302
4303 spg = lookup_spg_mapping(&osd->o_backoff_mappings, &m->spgid);
4304 if (!spg) {
4305 spg = alloc_spg_mapping();
4306 if (!spg) {
4307 pr_err("%s failed to allocate spg\n", __func__);
4308 return;
4309 }
4310 spg->spgid = m->spgid; /* struct */
4311 insert_spg_mapping(&osd->o_backoff_mappings, spg);
4312 }
4313
4314 backoff = alloc_backoff();
4315 if (!backoff) {
4316 pr_err("%s failed to allocate backoff\n", __func__);
4317 return;
4318 }
4319 backoff->spgid = m->spgid; /* struct */
4320 backoff->id = m->id;
4321 backoff->begin = m->begin;
4322 m->begin = NULL; /* backoff now owns this */
4323 backoff->end = m->end;
4324 m->end = NULL; /* ditto */
4325
4326 insert_backoff(&spg->backoffs, backoff);
4327 insert_backoff_by_id(&osd->o_backoffs_by_id, backoff);
4328
4329 /*
4330 * Ack with original backoff's epoch so that the OSD can
4331 * discard this if there was a PG split.
4332 */
4333 msg = create_backoff_message(backoff, m->map_epoch);
4334 if (!msg) {
4335 pr_err("%s failed to allocate msg\n", __func__);
4336 return;
4337 }
4338 ceph_con_send(&osd->o_con, msg);
4339}
4340
4341static bool target_contained_by(const struct ceph_osd_request_target *t,
4342 const struct ceph_hobject_id *begin,
4343 const struct ceph_hobject_id *end)
4344{
4345 struct ceph_hobject_id hoid;
4346 int cmp;
4347
4348 hoid_fill_from_target(&hoid, t);
4349 cmp = hoid_compare(&hoid, begin);
4350 return !cmp || (cmp > 0 && hoid_compare(&hoid, end) < 0);
4351}
4352
4353static void handle_backoff_unblock(struct ceph_osd *osd,
4354 const struct MOSDBackoff *m)
4355{
4356 struct ceph_spg_mapping *spg;
4357 struct ceph_osd_backoff *backoff;
4358 struct rb_node *n;
4359
4360 dout("%s osd%d spgid %llu.%xs%d id %llu\n", __func__, osd->o_osd,
4361 m->spgid.pgid.pool, m->spgid.pgid.seed, m->spgid.shard, m->id);
4362
4363 backoff = lookup_backoff_by_id(&osd->o_backoffs_by_id, m->id);
4364 if (!backoff) {
4365 pr_err("%s osd%d spgid %llu.%xs%d id %llu backoff dne\n",
4366 __func__, osd->o_osd, m->spgid.pgid.pool,
4367 m->spgid.pgid.seed, m->spgid.shard, m->id);
4368 return;
4369 }
4370
4371 if (hoid_compare(backoff->begin, m->begin) &&
4372 hoid_compare(backoff->end, m->end)) {
4373 pr_err("%s osd%d spgid %llu.%xs%d id %llu bad range?\n",
4374 __func__, osd->o_osd, m->spgid.pgid.pool,
4375 m->spgid.pgid.seed, m->spgid.shard, m->id);
4376 /* unblock it anyway... */
4377 }
4378
4379 spg = lookup_spg_mapping(&osd->o_backoff_mappings, &backoff->spgid);
4380 BUG_ON(!spg);
4381
4382 erase_backoff(&spg->backoffs, backoff);
4383 erase_backoff_by_id(&osd->o_backoffs_by_id, backoff);
4384 free_backoff(backoff);
4385
4386 if (RB_EMPTY_ROOT(&spg->backoffs)) {
4387 erase_spg_mapping(&osd->o_backoff_mappings, spg);
4388 free_spg_mapping(spg);
4389 }
4390
4391 for (n = rb_first(&osd->o_requests); n; n = rb_next(n)) {
4392 struct ceph_osd_request *req =
4393 rb_entry(n, struct ceph_osd_request, r_node);
4394
4395 if (!ceph_spg_compare(&req->r_t.spgid, &m->spgid)) {
4396 /*
4397 * Match against @m, not @backoff -- the PG may
4398 * have split on the OSD.
4399 */
4400 if (target_contained_by(&req->r_t, m->begin, m->end)) {
4401 /*
4402 * If no other installed backoff applies,
4403 * resend.
4404 */
4405 send_request(req);
4406 }
4407 }
4408 }
4409}
4410
4411static void handle_backoff(struct ceph_osd *osd, struct ceph_msg *msg)
4412{
4413 struct ceph_osd_client *osdc = osd->o_osdc;
4414 struct MOSDBackoff m;
4415 int ret;
4416
4417 down_read(&osdc->lock);
4418 if (!osd_registered(osd)) {
4419 dout("%s osd%d unknown\n", __func__, osd->o_osd);
4420 up_read(&osdc->lock);
4421 return;
4422 }
4423 WARN_ON(osd->o_osd != le64_to_cpu(msg->hdr.src.num));
4424
4425 mutex_lock(&osd->lock);
4426 ret = decode_MOSDBackoff(msg, &m);
4427 if (ret) {
4428 pr_err("failed to decode MOSDBackoff: %d\n", ret);
4429 ceph_msg_dump(msg);
4430 goto out_unlock;
4431 }
4432
4433 switch (m.op) {
4434 case CEPH_OSD_BACKOFF_OP_BLOCK:
4435 handle_backoff_block(osd, &m);
4436 break;
4437 case CEPH_OSD_BACKOFF_OP_UNBLOCK:
4438 handle_backoff_unblock(osd, &m);
4439 break;
4440 default:
4441 pr_err("%s osd%d unknown op %d\n", __func__, osd->o_osd, m.op);
4442 }
4443
4444 free_hoid(m.begin);
4445 free_hoid(m.end);
4446
4447out_unlock:
4448 mutex_unlock(&osd->lock);
4449 up_read(&osdc->lock);
4450}
4451
4452/*
4453 * Process osd watch notifications
4454 */
4455static void handle_watch_notify(struct ceph_osd_client *osdc,
4456 struct ceph_msg *msg)
4457{
4458 void *p = msg->front.iov_base;
4459 void *const end = p + msg->front.iov_len;
4460 struct ceph_osd_linger_request *lreq;
4461 struct linger_work *lwork;
4462 u8 proto_ver, opcode;
4463 u64 cookie, notify_id;
4464 u64 notifier_id = 0;
4465 s32 return_code = 0;
4466 void *payload = NULL;
4467 u32 payload_len = 0;
4468
4469 ceph_decode_8_safe(&p, end, proto_ver, bad);
4470 ceph_decode_8_safe(&p, end, opcode, bad);
4471 ceph_decode_64_safe(&p, end, cookie, bad);
4472 p += 8; /* skip ver */
4473 ceph_decode_64_safe(&p, end, notify_id, bad);
4474
4475 if (proto_ver >= 1) {
4476 ceph_decode_32_safe(&p, end, payload_len, bad);
4477 ceph_decode_need(&p, end, payload_len, bad);
4478 payload = p;
4479 p += payload_len;
4480 }
4481
4482 if (le16_to_cpu(msg->hdr.version) >= 2)
4483 ceph_decode_32_safe(&p, end, return_code, bad);
4484
4485 if (le16_to_cpu(msg->hdr.version) >= 3)
4486 ceph_decode_64_safe(&p, end, notifier_id, bad);
4487
4488 down_read(&osdc->lock);
4489 lreq = lookup_linger_osdc(&osdc->linger_requests, cookie);
4490 if (!lreq) {
4491 dout("%s opcode %d cookie %llu dne\n", __func__, opcode,
4492 cookie);
4493 goto out_unlock_osdc;
4494 }
4495
4496 mutex_lock(&lreq->lock);
4497 dout("%s opcode %d cookie %llu lreq %p is_watch %d\n", __func__,
4498 opcode, cookie, lreq, lreq->is_watch);
4499 if (opcode == CEPH_WATCH_EVENT_DISCONNECT) {
4500 if (!lreq->last_error) {
4501 lreq->last_error = -ENOTCONN;
4502 queue_watch_error(lreq);
4503 }
4504 } else if (!lreq->is_watch) {
4505 /* CEPH_WATCH_EVENT_NOTIFY_COMPLETE */
4506 if (lreq->notify_id && lreq->notify_id != notify_id) {
4507 dout("lreq %p notify_id %llu != %llu, ignoring\n", lreq,
4508 lreq->notify_id, notify_id);
4509 } else if (!completion_done(&lreq->notify_finish_wait)) {
4510 struct ceph_msg_data *data =
4511 msg->num_data_items ? &msg->data[0] : NULL;
4512
4513 if (data) {
4514 if (lreq->preply_pages) {
4515 WARN_ON(data->type !=
4516 CEPH_MSG_DATA_PAGES);
4517 *lreq->preply_pages = data->pages;
4518 *lreq->preply_len = data->length;
4519 data->own_pages = false;
4520 }
4521 }
4522 lreq->notify_finish_error = return_code;
4523 complete_all(&lreq->notify_finish_wait);
4524 }
4525 } else {
4526 /* CEPH_WATCH_EVENT_NOTIFY */
4527 lwork = lwork_alloc(lreq, do_watch_notify);
4528 if (!lwork) {
4529 pr_err("failed to allocate notify-lwork\n");
4530 goto out_unlock_lreq;
4531 }
4532
4533 lwork->notify.notify_id = notify_id;
4534 lwork->notify.notifier_id = notifier_id;
4535 lwork->notify.payload = payload;
4536 lwork->notify.payload_len = payload_len;
4537 lwork->notify.msg = ceph_msg_get(msg);
4538 lwork_queue(lwork);
4539 }
4540
4541out_unlock_lreq:
4542 mutex_unlock(&lreq->lock);
4543out_unlock_osdc:
4544 up_read(&osdc->lock);
4545 return;
4546
4547bad:
4548 pr_err("osdc handle_watch_notify corrupt msg\n");
4549}
4550
4551/*
4552 * Register request, send initial attempt.
4553 */
4554int ceph_osdc_start_request(struct ceph_osd_client *osdc,
4555 struct ceph_osd_request *req,
4556 bool nofail)
4557{
4558 down_read(&osdc->lock);
4559 submit_request(req, false);
4560 up_read(&osdc->lock);
4561
4562 return 0;
4563}
4564EXPORT_SYMBOL(ceph_osdc_start_request);
4565
4566/*
4567 * Unregister a registered request. The request is not completed:
4568 * ->r_result isn't set and __complete_request() isn't called.
4569 */
4570void ceph_osdc_cancel_request(struct ceph_osd_request *req)
4571{
4572 struct ceph_osd_client *osdc = req->r_osdc;
4573
4574 down_write(&osdc->lock);
4575 if (req->r_osd)
4576 cancel_request(req);
4577 up_write(&osdc->lock);
4578}
4579EXPORT_SYMBOL(ceph_osdc_cancel_request);
4580
4581/*
4582 * @timeout: in jiffies, 0 means "wait forever"
4583 */
4584static int wait_request_timeout(struct ceph_osd_request *req,
4585 unsigned long timeout)
4586{
4587 long left;
4588
4589 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
4590 left = wait_for_completion_killable_timeout(&req->r_completion,
4591 ceph_timeout_jiffies(timeout));
4592 if (left <= 0) {
4593 left = left ?: -ETIMEDOUT;
4594 ceph_osdc_cancel_request(req);
4595 } else {
4596 left = req->r_result; /* completed */
4597 }
4598
4599 return left;
4600}
4601
4602/*
4603 * wait for a request to complete
4604 */
4605int ceph_osdc_wait_request(struct ceph_osd_client *osdc,
4606 struct ceph_osd_request *req)
4607{
4608 return wait_request_timeout(req, 0);
4609}
4610EXPORT_SYMBOL(ceph_osdc_wait_request);
4611
4612/*
4613 * sync - wait for all in-flight requests to flush. avoid starvation.
4614 */
4615void ceph_osdc_sync(struct ceph_osd_client *osdc)
4616{
4617 struct rb_node *n, *p;
4618 u64 last_tid = atomic64_read(&osdc->last_tid);
4619
4620again:
4621 down_read(&osdc->lock);
4622 for (n = rb_first(&osdc->osds); n; n = rb_next(n)) {
4623 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
4624
4625 mutex_lock(&osd->lock);
4626 for (p = rb_first(&osd->o_requests); p; p = rb_next(p)) {
4627 struct ceph_osd_request *req =
4628 rb_entry(p, struct ceph_osd_request, r_node);
4629
4630 if (req->r_tid > last_tid)
4631 break;
4632
4633 if (!(req->r_flags & CEPH_OSD_FLAG_WRITE))
4634 continue;
4635
4636 ceph_osdc_get_request(req);
4637 mutex_unlock(&osd->lock);
4638 up_read(&osdc->lock);
4639 dout("%s waiting on req %p tid %llu last_tid %llu\n",
4640 __func__, req, req->r_tid, last_tid);
4641 wait_for_completion(&req->r_completion);
4642 ceph_osdc_put_request(req);
4643 goto again;
4644 }
4645
4646 mutex_unlock(&osd->lock);
4647 }
4648
4649 up_read(&osdc->lock);
4650 dout("%s done last_tid %llu\n", __func__, last_tid);
4651}
4652EXPORT_SYMBOL(ceph_osdc_sync);
4653
4654static struct ceph_osd_request *
4655alloc_linger_request(struct ceph_osd_linger_request *lreq)
4656{
4657 struct ceph_osd_request *req;
4658
4659 req = ceph_osdc_alloc_request(lreq->osdc, NULL, 1, false, GFP_NOIO);
4660 if (!req)
4661 return NULL;
4662
4663 ceph_oid_copy(&req->r_base_oid, &lreq->t.base_oid);
4664 ceph_oloc_copy(&req->r_base_oloc, &lreq->t.base_oloc);
4665 return req;
4666}
4667
4668static struct ceph_osd_request *
4669alloc_watch_request(struct ceph_osd_linger_request *lreq, u8 watch_opcode)
4670{
4671 struct ceph_osd_request *req;
4672
4673 req = alloc_linger_request(lreq);
4674 if (!req)
4675 return NULL;
4676
4677 /*
4678 * Pass 0 for cookie because we don't know it yet, it will be
4679 * filled in by linger_submit().
4680 */
4681 osd_req_op_watch_init(req, 0, 0, watch_opcode);
4682
4683 if (ceph_osdc_alloc_messages(req, GFP_NOIO)) {
4684 ceph_osdc_put_request(req);
4685 return NULL;
4686 }
4687
4688 return req;
4689}
4690
4691/*
4692 * Returns a handle, caller owns a ref.
4693 */
4694struct ceph_osd_linger_request *
4695ceph_osdc_watch(struct ceph_osd_client *osdc,
4696 struct ceph_object_id *oid,
4697 struct ceph_object_locator *oloc,
4698 rados_watchcb2_t wcb,
4699 rados_watcherrcb_t errcb,
4700 void *data)
4701{
4702 struct ceph_osd_linger_request *lreq;
4703 int ret;
4704
4705 lreq = linger_alloc(osdc);
4706 if (!lreq)
4707 return ERR_PTR(-ENOMEM);
4708
4709 lreq->is_watch = true;
4710 lreq->wcb = wcb;
4711 lreq->errcb = errcb;
4712 lreq->data = data;
4713 lreq->watch_valid_thru = jiffies;
4714
4715 ceph_oid_copy(&lreq->t.base_oid, oid);
4716 ceph_oloc_copy(&lreq->t.base_oloc, oloc);
4717 lreq->t.flags = CEPH_OSD_FLAG_WRITE;
4718 ktime_get_real_ts64(&lreq->mtime);
4719
4720 lreq->reg_req = alloc_watch_request(lreq, CEPH_OSD_WATCH_OP_WATCH);
4721 if (!lreq->reg_req) {
4722 ret = -ENOMEM;
4723 goto err_put_lreq;
4724 }
4725
4726 lreq->ping_req = alloc_watch_request(lreq, CEPH_OSD_WATCH_OP_PING);
4727 if (!lreq->ping_req) {
4728 ret = -ENOMEM;
4729 goto err_put_lreq;
4730 }
4731
4732 linger_submit(lreq);
4733 ret = linger_reg_commit_wait(lreq);
4734 if (ret) {
4735 linger_cancel(lreq);
4736 goto err_put_lreq;
4737 }
4738
4739 return lreq;
4740
4741err_put_lreq:
4742 linger_put(lreq);
4743 return ERR_PTR(ret);
4744}
4745EXPORT_SYMBOL(ceph_osdc_watch);
4746
4747/*
4748 * Releases a ref.
4749 *
4750 * Times out after mount_timeout to preserve rbd unmap behaviour
4751 * introduced in 2894e1d76974 ("rbd: timeout watch teardown on unmap
4752 * with mount_timeout").
4753 */
4754int ceph_osdc_unwatch(struct ceph_osd_client *osdc,
4755 struct ceph_osd_linger_request *lreq)
4756{
4757 struct ceph_options *opts = osdc->client->options;
4758 struct ceph_osd_request *req;
4759 int ret;
4760
4761 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
4762 if (!req)
4763 return -ENOMEM;
4764
4765 ceph_oid_copy(&req->r_base_oid, &lreq->t.base_oid);
4766 ceph_oloc_copy(&req->r_base_oloc, &lreq->t.base_oloc);
4767 req->r_flags = CEPH_OSD_FLAG_WRITE;
4768 ktime_get_real_ts64(&req->r_mtime);
4769 osd_req_op_watch_init(req, 0, lreq->linger_id,
4770 CEPH_OSD_WATCH_OP_UNWATCH);
4771
4772 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
4773 if (ret)
4774 goto out_put_req;
4775
4776 ceph_osdc_start_request(osdc, req, false);
4777 linger_cancel(lreq);
4778 linger_put(lreq);
4779 ret = wait_request_timeout(req, opts->mount_timeout);
4780
4781out_put_req:
4782 ceph_osdc_put_request(req);
4783 return ret;
4784}
4785EXPORT_SYMBOL(ceph_osdc_unwatch);
4786
4787static int osd_req_op_notify_ack_init(struct ceph_osd_request *req, int which,
4788 u64 notify_id, u64 cookie, void *payload,
4789 u32 payload_len)
4790{
4791 struct ceph_osd_req_op *op;
4792 struct ceph_pagelist *pl;
4793 int ret;
4794
4795 op = osd_req_op_init(req, which, CEPH_OSD_OP_NOTIFY_ACK, 0);
4796
4797 pl = ceph_pagelist_alloc(GFP_NOIO);
4798 if (!pl)
4799 return -ENOMEM;
4800
4801 ret = ceph_pagelist_encode_64(pl, notify_id);
4802 ret |= ceph_pagelist_encode_64(pl, cookie);
4803 if (payload) {
4804 ret |= ceph_pagelist_encode_32(pl, payload_len);
4805 ret |= ceph_pagelist_append(pl, payload, payload_len);
4806 } else {
4807 ret |= ceph_pagelist_encode_32(pl, 0);
4808 }
4809 if (ret) {
4810 ceph_pagelist_release(pl);
4811 return -ENOMEM;
4812 }
4813
4814 ceph_osd_data_pagelist_init(&op->notify_ack.request_data, pl);
4815 op->indata_len = pl->length;
4816 return 0;
4817}
4818
4819int ceph_osdc_notify_ack(struct ceph_osd_client *osdc,
4820 struct ceph_object_id *oid,
4821 struct ceph_object_locator *oloc,
4822 u64 notify_id,
4823 u64 cookie,
4824 void *payload,
4825 u32 payload_len)
4826{
4827 struct ceph_osd_request *req;
4828 int ret;
4829
4830 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
4831 if (!req)
4832 return -ENOMEM;
4833
4834 ceph_oid_copy(&req->r_base_oid, oid);
4835 ceph_oloc_copy(&req->r_base_oloc, oloc);
4836 req->r_flags = CEPH_OSD_FLAG_READ;
4837
4838 ret = osd_req_op_notify_ack_init(req, 0, notify_id, cookie, payload,
4839 payload_len);
4840 if (ret)
4841 goto out_put_req;
4842
4843 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
4844 if (ret)
4845 goto out_put_req;
4846
4847 ceph_osdc_start_request(osdc, req, false);
4848 ret = ceph_osdc_wait_request(osdc, req);
4849
4850out_put_req:
4851 ceph_osdc_put_request(req);
4852 return ret;
4853}
4854EXPORT_SYMBOL(ceph_osdc_notify_ack);
4855
4856static int osd_req_op_notify_init(struct ceph_osd_request *req, int which,
4857 u64 cookie, u32 prot_ver, u32 timeout,
4858 void *payload, u32 payload_len)
4859{
4860 struct ceph_osd_req_op *op;
4861 struct ceph_pagelist *pl;
4862 int ret;
4863
4864 op = osd_req_op_init(req, which, CEPH_OSD_OP_NOTIFY, 0);
4865 op->notify.cookie = cookie;
4866
4867 pl = ceph_pagelist_alloc(GFP_NOIO);
4868 if (!pl)
4869 return -ENOMEM;
4870
4871 ret = ceph_pagelist_encode_32(pl, 1); /* prot_ver */
4872 ret |= ceph_pagelist_encode_32(pl, timeout);
4873 ret |= ceph_pagelist_encode_32(pl, payload_len);
4874 ret |= ceph_pagelist_append(pl, payload, payload_len);
4875 if (ret) {
4876 ceph_pagelist_release(pl);
4877 return -ENOMEM;
4878 }
4879
4880 ceph_osd_data_pagelist_init(&op->notify.request_data, pl);
4881 op->indata_len = pl->length;
4882 return 0;
4883}
4884
4885/*
4886 * @timeout: in seconds
4887 *
4888 * @preply_{pages,len} are initialized both on success and error.
4889 * The caller is responsible for:
4890 *
4891 * ceph_release_page_vector(reply_pages, calc_pages_for(0, reply_len))
4892 */
4893int ceph_osdc_notify(struct ceph_osd_client *osdc,
4894 struct ceph_object_id *oid,
4895 struct ceph_object_locator *oloc,
4896 void *payload,
4897 u32 payload_len,
4898 u32 timeout,
4899 struct page ***preply_pages,
4900 size_t *preply_len)
4901{
4902 struct ceph_osd_linger_request *lreq;
4903 struct page **pages;
4904 int ret;
4905
4906 WARN_ON(!timeout);
4907 if (preply_pages) {
4908 *preply_pages = NULL;
4909 *preply_len = 0;
4910 }
4911
4912 lreq = linger_alloc(osdc);
4913 if (!lreq)
4914 return -ENOMEM;
4915
4916 lreq->preply_pages = preply_pages;
4917 lreq->preply_len = preply_len;
4918
4919 ceph_oid_copy(&lreq->t.base_oid, oid);
4920 ceph_oloc_copy(&lreq->t.base_oloc, oloc);
4921 lreq->t.flags = CEPH_OSD_FLAG_READ;
4922
4923 lreq->reg_req = alloc_linger_request(lreq);
4924 if (!lreq->reg_req) {
4925 ret = -ENOMEM;
4926 goto out_put_lreq;
4927 }
4928
4929 /*
4930 * Pass 0 for cookie because we don't know it yet, it will be
4931 * filled in by linger_submit().
4932 */
4933 ret = osd_req_op_notify_init(lreq->reg_req, 0, 0, 1, timeout,
4934 payload, payload_len);
4935 if (ret)
4936 goto out_put_lreq;
4937
4938 /* for notify_id */
4939 pages = ceph_alloc_page_vector(1, GFP_NOIO);
4940 if (IS_ERR(pages)) {
4941 ret = PTR_ERR(pages);
4942 goto out_put_lreq;
4943 }
4944 ceph_osd_data_pages_init(osd_req_op_data(lreq->reg_req, 0, notify,
4945 response_data),
4946 pages, PAGE_SIZE, 0, false, true);
4947
4948 ret = ceph_osdc_alloc_messages(lreq->reg_req, GFP_NOIO);
4949 if (ret)
4950 goto out_put_lreq;
4951
4952 linger_submit(lreq);
4953 ret = linger_reg_commit_wait(lreq);
4954 if (!ret)
4955 ret = linger_notify_finish_wait(lreq);
4956 else
4957 dout("lreq %p failed to initiate notify %d\n", lreq, ret);
4958
4959 linger_cancel(lreq);
4960out_put_lreq:
4961 linger_put(lreq);
4962 return ret;
4963}
4964EXPORT_SYMBOL(ceph_osdc_notify);
4965
4966/*
4967 * Return the number of milliseconds since the watch was last
4968 * confirmed, or an error. If there is an error, the watch is no
4969 * longer valid, and should be destroyed with ceph_osdc_unwatch().
4970 */
4971int ceph_osdc_watch_check(struct ceph_osd_client *osdc,
4972 struct ceph_osd_linger_request *lreq)
4973{
4974 unsigned long stamp, age;
4975 int ret;
4976
4977 down_read(&osdc->lock);
4978 mutex_lock(&lreq->lock);
4979 stamp = lreq->watch_valid_thru;
4980 if (!list_empty(&lreq->pending_lworks)) {
4981 struct linger_work *lwork =
4982 list_first_entry(&lreq->pending_lworks,
4983 struct linger_work,
4984 pending_item);
4985
4986 if (time_before(lwork->queued_stamp, stamp))
4987 stamp = lwork->queued_stamp;
4988 }
4989 age = jiffies - stamp;
4990 dout("%s lreq %p linger_id %llu age %lu last_error %d\n", __func__,
4991 lreq, lreq->linger_id, age, lreq->last_error);
4992 /* we are truncating to msecs, so return a safe upper bound */
4993 ret = lreq->last_error ?: 1 + jiffies_to_msecs(age);
4994
4995 mutex_unlock(&lreq->lock);
4996 up_read(&osdc->lock);
4997 return ret;
4998}
4999
5000static int decode_watcher(void **p, void *end, struct ceph_watch_item *item)
5001{
5002 u8 struct_v;
5003 u32 struct_len;
5004 int ret;
5005
5006 ret = ceph_start_decoding(p, end, 2, "watch_item_t",
5007 &struct_v, &struct_len);
5008 if (ret)
5009 goto bad;
5010
5011 ret = -EINVAL;
5012 ceph_decode_copy_safe(p, end, &item->name, sizeof(item->name), bad);
5013 ceph_decode_64_safe(p, end, item->cookie, bad);
5014 ceph_decode_skip_32(p, end, bad); /* skip timeout seconds */
5015
5016 if (struct_v >= 2) {
5017 ret = ceph_decode_entity_addr(p, end, &item->addr);
5018 if (ret)
5019 goto bad;
5020 } else {
5021 ret = 0;
5022 }
5023
5024 dout("%s %s%llu cookie %llu addr %s\n", __func__,
5025 ENTITY_NAME(item->name), item->cookie,
5026 ceph_pr_addr(&item->addr));
5027bad:
5028 return ret;
5029}
5030
5031static int decode_watchers(void **p, void *end,
5032 struct ceph_watch_item **watchers,
5033 u32 *num_watchers)
5034{
5035 u8 struct_v;
5036 u32 struct_len;
5037 int i;
5038 int ret;
5039
5040 ret = ceph_start_decoding(p, end, 1, "obj_list_watch_response_t",
5041 &struct_v, &struct_len);
5042 if (ret)
5043 return ret;
5044
5045 *num_watchers = ceph_decode_32(p);
5046 *watchers = kcalloc(*num_watchers, sizeof(**watchers), GFP_NOIO);
5047 if (!*watchers)
5048 return -ENOMEM;
5049
5050 for (i = 0; i < *num_watchers; i++) {
5051 ret = decode_watcher(p, end, *watchers + i);
5052 if (ret) {
5053 kfree(*watchers);
5054 return ret;
5055 }
5056 }
5057
5058 return 0;
5059}
5060
5061/*
5062 * On success, the caller is responsible for:
5063 *
5064 * kfree(watchers);
5065 */
5066int ceph_osdc_list_watchers(struct ceph_osd_client *osdc,
5067 struct ceph_object_id *oid,
5068 struct ceph_object_locator *oloc,
5069 struct ceph_watch_item **watchers,
5070 u32 *num_watchers)
5071{
5072 struct ceph_osd_request *req;
5073 struct page **pages;
5074 int ret;
5075
5076 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
5077 if (!req)
5078 return -ENOMEM;
5079
5080 ceph_oid_copy(&req->r_base_oid, oid);
5081 ceph_oloc_copy(&req->r_base_oloc, oloc);
5082 req->r_flags = CEPH_OSD_FLAG_READ;
5083
5084 pages = ceph_alloc_page_vector(1, GFP_NOIO);
5085 if (IS_ERR(pages)) {
5086 ret = PTR_ERR(pages);
5087 goto out_put_req;
5088 }
5089
5090 osd_req_op_init(req, 0, CEPH_OSD_OP_LIST_WATCHERS, 0);
5091 ceph_osd_data_pages_init(osd_req_op_data(req, 0, list_watchers,
5092 response_data),
5093 pages, PAGE_SIZE, 0, false, true);
5094
5095 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
5096 if (ret)
5097 goto out_put_req;
5098
5099 ceph_osdc_start_request(osdc, req, false);
5100 ret = ceph_osdc_wait_request(osdc, req);
5101 if (ret >= 0) {
5102 void *p = page_address(pages[0]);
5103 void *const end = p + req->r_ops[0].outdata_len;
5104
5105 ret = decode_watchers(&p, end, watchers, num_watchers);
5106 }
5107
5108out_put_req:
5109 ceph_osdc_put_request(req);
5110 return ret;
5111}
5112EXPORT_SYMBOL(ceph_osdc_list_watchers);
5113
5114/*
5115 * Call all pending notify callbacks - for use after a watch is
5116 * unregistered, to make sure no more callbacks for it will be invoked
5117 */
5118void ceph_osdc_flush_notifies(struct ceph_osd_client *osdc)
5119{
5120 dout("%s osdc %p\n", __func__, osdc);
5121 flush_workqueue(osdc->notify_wq);
5122}
5123EXPORT_SYMBOL(ceph_osdc_flush_notifies);
5124
5125void ceph_osdc_maybe_request_map(struct ceph_osd_client *osdc)
5126{
5127 down_read(&osdc->lock);
5128 maybe_request_map(osdc);
5129 up_read(&osdc->lock);
5130}
5131EXPORT_SYMBOL(ceph_osdc_maybe_request_map);
5132
5133/*
5134 * Execute an OSD class method on an object.
5135 *
5136 * @flags: CEPH_OSD_FLAG_*
5137 * @resp_len: in/out param for reply length
5138 */
5139int ceph_osdc_call(struct ceph_osd_client *osdc,
5140 struct ceph_object_id *oid,
5141 struct ceph_object_locator *oloc,
5142 const char *class, const char *method,
5143 unsigned int flags,
5144 struct page *req_page, size_t req_len,
5145 struct page **resp_pages, size_t *resp_len)
5146{
5147 struct ceph_osd_request *req;
5148 int ret;
5149
5150 if (req_len > PAGE_SIZE)
5151 return -E2BIG;
5152
5153 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
5154 if (!req)
5155 return -ENOMEM;
5156
5157 ceph_oid_copy(&req->r_base_oid, oid);
5158 ceph_oloc_copy(&req->r_base_oloc, oloc);
5159 req->r_flags = flags;
5160
5161 ret = osd_req_op_cls_init(req, 0, class, method);
5162 if (ret)
5163 goto out_put_req;
5164
5165 if (req_page)
5166 osd_req_op_cls_request_data_pages(req, 0, &req_page, req_len,
5167 0, false, false);
5168 if (resp_pages)
5169 osd_req_op_cls_response_data_pages(req, 0, resp_pages,
5170 *resp_len, 0, false, false);
5171
5172 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
5173 if (ret)
5174 goto out_put_req;
5175
5176 ceph_osdc_start_request(osdc, req, false);
5177 ret = ceph_osdc_wait_request(osdc, req);
5178 if (ret >= 0) {
5179 ret = req->r_ops[0].rval;
5180 if (resp_pages)
5181 *resp_len = req->r_ops[0].outdata_len;
5182 }
5183
5184out_put_req:
5185 ceph_osdc_put_request(req);
5186 return ret;
5187}
5188EXPORT_SYMBOL(ceph_osdc_call);
5189
5190/*
5191 * reset all osd connections
5192 */
5193void ceph_osdc_reopen_osds(struct ceph_osd_client *osdc)
5194{
5195 struct rb_node *n;
5196
5197 down_write(&osdc->lock);
5198 for (n = rb_first(&osdc->osds); n; ) {
5199 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
5200
5201 n = rb_next(n);
5202 if (!reopen_osd(osd))
5203 kick_osd_requests(osd);
5204 }
5205 up_write(&osdc->lock);
5206}
5207
5208/*
5209 * init, shutdown
5210 */
5211int ceph_osdc_init(struct ceph_osd_client *osdc, struct ceph_client *client)
5212{
5213 int err;
5214
5215 dout("init\n");
5216 osdc->client = client;
5217 init_rwsem(&osdc->lock);
5218 osdc->osds = RB_ROOT;
5219 INIT_LIST_HEAD(&osdc->osd_lru);
5220 spin_lock_init(&osdc->osd_lru_lock);
5221 osd_init(&osdc->homeless_osd);
5222 osdc->homeless_osd.o_osdc = osdc;
5223 osdc->homeless_osd.o_osd = CEPH_HOMELESS_OSD;
5224 osdc->last_linger_id = CEPH_LINGER_ID_START;
5225 osdc->linger_requests = RB_ROOT;
5226 osdc->map_checks = RB_ROOT;
5227 osdc->linger_map_checks = RB_ROOT;
5228 INIT_DELAYED_WORK(&osdc->timeout_work, handle_timeout);
5229 INIT_DELAYED_WORK(&osdc->osds_timeout_work, handle_osds_timeout);
5230
5231 err = -ENOMEM;
5232 osdc->osdmap = ceph_osdmap_alloc();
5233 if (!osdc->osdmap)
5234 goto out;
5235
5236 osdc->req_mempool = mempool_create_slab_pool(10,
5237 ceph_osd_request_cache);
5238 if (!osdc->req_mempool)
5239 goto out_map;
5240
5241 err = ceph_msgpool_init(&osdc->msgpool_op, CEPH_MSG_OSD_OP,
5242 PAGE_SIZE, CEPH_OSD_SLAB_OPS, 10, "osd_op");
5243 if (err < 0)
5244 goto out_mempool;
5245 err = ceph_msgpool_init(&osdc->msgpool_op_reply, CEPH_MSG_OSD_OPREPLY,
5246 PAGE_SIZE, CEPH_OSD_SLAB_OPS, 10,
5247 "osd_op_reply");
5248 if (err < 0)
5249 goto out_msgpool;
5250
5251 err = -ENOMEM;
5252 osdc->notify_wq = create_singlethread_workqueue("ceph-watch-notify");
5253 if (!osdc->notify_wq)
5254 goto out_msgpool_reply;
5255
5256 osdc->completion_wq = create_singlethread_workqueue("ceph-completion");
5257 if (!osdc->completion_wq)
5258 goto out_notify_wq;
5259
5260 schedule_delayed_work(&osdc->timeout_work,
5261 osdc->client->options->osd_keepalive_timeout);
5262 schedule_delayed_work(&osdc->osds_timeout_work,
5263 round_jiffies_relative(osdc->client->options->osd_idle_ttl));
5264
5265 return 0;
5266
5267out_notify_wq:
5268 destroy_workqueue(osdc->notify_wq);
5269out_msgpool_reply:
5270 ceph_msgpool_destroy(&osdc->msgpool_op_reply);
5271out_msgpool:
5272 ceph_msgpool_destroy(&osdc->msgpool_op);
5273out_mempool:
5274 mempool_destroy(osdc->req_mempool);
5275out_map:
5276 ceph_osdmap_destroy(osdc->osdmap);
5277out:
5278 return err;
5279}
5280
5281void ceph_osdc_stop(struct ceph_osd_client *osdc)
5282{
5283 destroy_workqueue(osdc->completion_wq);
5284 destroy_workqueue(osdc->notify_wq);
5285 cancel_delayed_work_sync(&osdc->timeout_work);
5286 cancel_delayed_work_sync(&osdc->osds_timeout_work);
5287
5288 down_write(&osdc->lock);
5289 while (!RB_EMPTY_ROOT(&osdc->osds)) {
5290 struct ceph_osd *osd = rb_entry(rb_first(&osdc->osds),
5291 struct ceph_osd, o_node);
5292 close_osd(osd);
5293 }
5294 up_write(&osdc->lock);
5295 WARN_ON(refcount_read(&osdc->homeless_osd.o_ref) != 1);
5296 osd_cleanup(&osdc->homeless_osd);
5297
5298 WARN_ON(!list_empty(&osdc->osd_lru));
5299 WARN_ON(!RB_EMPTY_ROOT(&osdc->linger_requests));
5300 WARN_ON(!RB_EMPTY_ROOT(&osdc->map_checks));
5301 WARN_ON(!RB_EMPTY_ROOT(&osdc->linger_map_checks));
5302 WARN_ON(atomic_read(&osdc->num_requests));
5303 WARN_ON(atomic_read(&osdc->num_homeless));
5304
5305 ceph_osdmap_destroy(osdc->osdmap);
5306 mempool_destroy(osdc->req_mempool);
5307 ceph_msgpool_destroy(&osdc->msgpool_op);
5308 ceph_msgpool_destroy(&osdc->msgpool_op_reply);
5309}
5310
5311static int osd_req_op_copy_from_init(struct ceph_osd_request *req,
5312 u64 src_snapid, u64 src_version,
5313 struct ceph_object_id *src_oid,
5314 struct ceph_object_locator *src_oloc,
5315 u32 src_fadvise_flags,
5316 u32 dst_fadvise_flags,
5317 u32 truncate_seq, u64 truncate_size,
5318 u8 copy_from_flags)
5319{
5320 struct ceph_osd_req_op *op;
5321 struct page **pages;
5322 void *p, *end;
5323
5324 pages = ceph_alloc_page_vector(1, GFP_KERNEL);
5325 if (IS_ERR(pages))
5326 return PTR_ERR(pages);
5327
5328 op = osd_req_op_init(req, 0, CEPH_OSD_OP_COPY_FROM2,
5329 dst_fadvise_flags);
5330 op->copy_from.snapid = src_snapid;
5331 op->copy_from.src_version = src_version;
5332 op->copy_from.flags = copy_from_flags;
5333 op->copy_from.src_fadvise_flags = src_fadvise_flags;
5334
5335 p = page_address(pages[0]);
5336 end = p + PAGE_SIZE;
5337 ceph_encode_string(&p, end, src_oid->name, src_oid->name_len);
5338 encode_oloc(&p, end, src_oloc);
5339 ceph_encode_32(&p, truncate_seq);
5340 ceph_encode_64(&p, truncate_size);
5341 op->indata_len = PAGE_SIZE - (end - p);
5342
5343 ceph_osd_data_pages_init(&op->copy_from.osd_data, pages,
5344 op->indata_len, 0, false, true);
5345 return 0;
5346}
5347
5348int ceph_osdc_copy_from(struct ceph_osd_client *osdc,
5349 u64 src_snapid, u64 src_version,
5350 struct ceph_object_id *src_oid,
5351 struct ceph_object_locator *src_oloc,
5352 u32 src_fadvise_flags,
5353 struct ceph_object_id *dst_oid,
5354 struct ceph_object_locator *dst_oloc,
5355 u32 dst_fadvise_flags,
5356 u32 truncate_seq, u64 truncate_size,
5357 u8 copy_from_flags)
5358{
5359 struct ceph_osd_request *req;
5360 int ret;
5361
5362 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_KERNEL);
5363 if (!req)
5364 return -ENOMEM;
5365
5366 req->r_flags = CEPH_OSD_FLAG_WRITE;
5367
5368 ceph_oloc_copy(&req->r_t.base_oloc, dst_oloc);
5369 ceph_oid_copy(&req->r_t.base_oid, dst_oid);
5370
5371 ret = osd_req_op_copy_from_init(req, src_snapid, src_version, src_oid,
5372 src_oloc, src_fadvise_flags,
5373 dst_fadvise_flags, truncate_seq,
5374 truncate_size, copy_from_flags);
5375 if (ret)
5376 goto out;
5377
5378 ret = ceph_osdc_alloc_messages(req, GFP_KERNEL);
5379 if (ret)
5380 goto out;
5381
5382 ceph_osdc_start_request(osdc, req, false);
5383 ret = ceph_osdc_wait_request(osdc, req);
5384
5385out:
5386 ceph_osdc_put_request(req);
5387 return ret;
5388}
5389EXPORT_SYMBOL(ceph_osdc_copy_from);
5390
5391int __init ceph_osdc_setup(void)
5392{
5393 size_t size = sizeof(struct ceph_osd_request) +
5394 CEPH_OSD_SLAB_OPS * sizeof(struct ceph_osd_req_op);
5395
5396 BUG_ON(ceph_osd_request_cache);
5397 ceph_osd_request_cache = kmem_cache_create("ceph_osd_request", size,
5398 0, 0, NULL);
5399
5400 return ceph_osd_request_cache ? 0 : -ENOMEM;
5401}
5402
5403void ceph_osdc_cleanup(void)
5404{
5405 BUG_ON(!ceph_osd_request_cache);
5406 kmem_cache_destroy(ceph_osd_request_cache);
5407 ceph_osd_request_cache = NULL;
5408}
5409
5410/*
5411 * handle incoming message
5412 */
5413static void dispatch(struct ceph_connection *con, struct ceph_msg *msg)
5414{
5415 struct ceph_osd *osd = con->private;
5416 struct ceph_osd_client *osdc = osd->o_osdc;
5417 int type = le16_to_cpu(msg->hdr.type);
5418
5419 switch (type) {
5420 case CEPH_MSG_OSD_MAP:
5421 ceph_osdc_handle_map(osdc, msg);
5422 break;
5423 case CEPH_MSG_OSD_OPREPLY:
5424 handle_reply(osd, msg);
5425 break;
5426 case CEPH_MSG_OSD_BACKOFF:
5427 handle_backoff(osd, msg);
5428 break;
5429 case CEPH_MSG_WATCH_NOTIFY:
5430 handle_watch_notify(osdc, msg);
5431 break;
5432
5433 default:
5434 pr_err("received unknown message type %d %s\n", type,
5435 ceph_msg_type_name(type));
5436 }
5437
5438 ceph_msg_put(msg);
5439}
5440
5441/*
5442 * Lookup and return message for incoming reply. Don't try to do
5443 * anything about a larger than preallocated data portion of the
5444 * message at the moment - for now, just skip the message.
5445 */
5446static struct ceph_msg *get_reply(struct ceph_connection *con,
5447 struct ceph_msg_header *hdr,
5448 int *skip)
5449{
5450 struct ceph_osd *osd = con->private;
5451 struct ceph_osd_client *osdc = osd->o_osdc;
5452 struct ceph_msg *m = NULL;
5453 struct ceph_osd_request *req;
5454 int front_len = le32_to_cpu(hdr->front_len);
5455 int data_len = le32_to_cpu(hdr->data_len);
5456 u64 tid = le64_to_cpu(hdr->tid);
5457
5458 down_read(&osdc->lock);
5459 if (!osd_registered(osd)) {
5460 dout("%s osd%d unknown, skipping\n", __func__, osd->o_osd);
5461 *skip = 1;
5462 goto out_unlock_osdc;
5463 }
5464 WARN_ON(osd->o_osd != le64_to_cpu(hdr->src.num));
5465
5466 mutex_lock(&osd->lock);
5467 req = lookup_request(&osd->o_requests, tid);
5468 if (!req) {
5469 dout("%s osd%d tid %llu unknown, skipping\n", __func__,
5470 osd->o_osd, tid);
5471 *skip = 1;
5472 goto out_unlock_session;
5473 }
5474
5475 ceph_msg_revoke_incoming(req->r_reply);
5476
5477 if (front_len > req->r_reply->front_alloc_len) {
5478 pr_warn("%s osd%d tid %llu front %d > preallocated %d\n",
5479 __func__, osd->o_osd, req->r_tid, front_len,
5480 req->r_reply->front_alloc_len);
5481 m = ceph_msg_new(CEPH_MSG_OSD_OPREPLY, front_len, GFP_NOFS,
5482 false);
5483 if (!m)
5484 goto out_unlock_session;
5485 ceph_msg_put(req->r_reply);
5486 req->r_reply = m;
5487 }
5488
5489 if (data_len > req->r_reply->data_length) {
5490 pr_warn("%s osd%d tid %llu data %d > preallocated %zu, skipping\n",
5491 __func__, osd->o_osd, req->r_tid, data_len,
5492 req->r_reply->data_length);
5493 m = NULL;
5494 *skip = 1;
5495 goto out_unlock_session;
5496 }
5497
5498 m = ceph_msg_get(req->r_reply);
5499 dout("get_reply tid %lld %p\n", tid, m);
5500
5501out_unlock_session:
5502 mutex_unlock(&osd->lock);
5503out_unlock_osdc:
5504 up_read(&osdc->lock);
5505 return m;
5506}
5507
5508static struct ceph_msg *alloc_msg_with_page_vector(struct ceph_msg_header *hdr)
5509{
5510 struct ceph_msg *m;
5511 int type = le16_to_cpu(hdr->type);
5512 u32 front_len = le32_to_cpu(hdr->front_len);
5513 u32 data_len = le32_to_cpu(hdr->data_len);
5514
5515 m = ceph_msg_new2(type, front_len, 1, GFP_NOIO, false);
5516 if (!m)
5517 return NULL;
5518
5519 if (data_len) {
5520 struct page **pages;
5521
5522 pages = ceph_alloc_page_vector(calc_pages_for(0, data_len),
5523 GFP_NOIO);
5524 if (IS_ERR(pages)) {
5525 ceph_msg_put(m);
5526 return NULL;
5527 }
5528
5529 ceph_msg_data_add_pages(m, pages, data_len, 0, true);
5530 }
5531
5532 return m;
5533}
5534
5535static struct ceph_msg *alloc_msg(struct ceph_connection *con,
5536 struct ceph_msg_header *hdr,
5537 int *skip)
5538{
5539 struct ceph_osd *osd = con->private;
5540 int type = le16_to_cpu(hdr->type);
5541
5542 *skip = 0;
5543 switch (type) {
5544 case CEPH_MSG_OSD_MAP:
5545 case CEPH_MSG_OSD_BACKOFF:
5546 case CEPH_MSG_WATCH_NOTIFY:
5547 return alloc_msg_with_page_vector(hdr);
5548 case CEPH_MSG_OSD_OPREPLY:
5549 return get_reply(con, hdr, skip);
5550 default:
5551 pr_warn("%s osd%d unknown msg type %d, skipping\n", __func__,
5552 osd->o_osd, type);
5553 *skip = 1;
5554 return NULL;
5555 }
5556}
5557
5558/*
5559 * Wrappers to refcount containing ceph_osd struct
5560 */
5561static struct ceph_connection *get_osd_con(struct ceph_connection *con)
5562{
5563 struct ceph_osd *osd = con->private;
5564 if (get_osd(osd))
5565 return con;
5566 return NULL;
5567}
5568
5569static void put_osd_con(struct ceph_connection *con)
5570{
5571 struct ceph_osd *osd = con->private;
5572 put_osd(osd);
5573}
5574
5575/*
5576 * authentication
5577 */
5578/*
5579 * Note: returned pointer is the address of a structure that's
5580 * managed separately. Caller must *not* attempt to free it.
5581 */
5582static struct ceph_auth_handshake *get_authorizer(struct ceph_connection *con,
5583 int *proto, int force_new)
5584{
5585 struct ceph_osd *o = con->private;
5586 struct ceph_osd_client *osdc = o->o_osdc;
5587 struct ceph_auth_client *ac = osdc->client->monc.auth;
5588 struct ceph_auth_handshake *auth = &o->o_auth;
5589
5590 if (force_new && auth->authorizer) {
5591 ceph_auth_destroy_authorizer(auth->authorizer);
5592 auth->authorizer = NULL;
5593 }
5594 if (!auth->authorizer) {
5595 int ret = ceph_auth_create_authorizer(ac, CEPH_ENTITY_TYPE_OSD,
5596 auth);
5597 if (ret)
5598 return ERR_PTR(ret);
5599 } else {
5600 int ret = ceph_auth_update_authorizer(ac, CEPH_ENTITY_TYPE_OSD,
5601 auth);
5602 if (ret)
5603 return ERR_PTR(ret);
5604 }
5605 *proto = ac->protocol;
5606
5607 return auth;
5608}
5609
5610static int add_authorizer_challenge(struct ceph_connection *con,
5611 void *challenge_buf, int challenge_buf_len)
5612{
5613 struct ceph_osd *o = con->private;
5614 struct ceph_osd_client *osdc = o->o_osdc;
5615 struct ceph_auth_client *ac = osdc->client->monc.auth;
5616
5617 return ceph_auth_add_authorizer_challenge(ac, o->o_auth.authorizer,
5618 challenge_buf, challenge_buf_len);
5619}
5620
5621static int verify_authorizer_reply(struct ceph_connection *con)
5622{
5623 struct ceph_osd *o = con->private;
5624 struct ceph_osd_client *osdc = o->o_osdc;
5625 struct ceph_auth_client *ac = osdc->client->monc.auth;
5626
5627 return ceph_auth_verify_authorizer_reply(ac, o->o_auth.authorizer);
5628}
5629
5630static int invalidate_authorizer(struct ceph_connection *con)
5631{
5632 struct ceph_osd *o = con->private;
5633 struct ceph_osd_client *osdc = o->o_osdc;
5634 struct ceph_auth_client *ac = osdc->client->monc.auth;
5635
5636 ceph_auth_invalidate_authorizer(ac, CEPH_ENTITY_TYPE_OSD);
5637 return ceph_monc_validate_auth(&osdc->client->monc);
5638}
5639
5640static void osd_reencode_message(struct ceph_msg *msg)
5641{
5642 int type = le16_to_cpu(msg->hdr.type);
5643
5644 if (type == CEPH_MSG_OSD_OP)
5645 encode_request_finish(msg);
5646}
5647
5648static int osd_sign_message(struct ceph_msg *msg)
5649{
5650 struct ceph_osd *o = msg->con->private;
5651 struct ceph_auth_handshake *auth = &o->o_auth;
5652
5653 return ceph_auth_sign_message(auth, msg);
5654}
5655
5656static int osd_check_message_signature(struct ceph_msg *msg)
5657{
5658 struct ceph_osd *o = msg->con->private;
5659 struct ceph_auth_handshake *auth = &o->o_auth;
5660
5661 return ceph_auth_check_message_signature(auth, msg);
5662}
5663
5664static const struct ceph_connection_operations osd_con_ops = {
5665 .get = get_osd_con,
5666 .put = put_osd_con,
5667 .dispatch = dispatch,
5668 .get_authorizer = get_authorizer,
5669 .add_authorizer_challenge = add_authorizer_challenge,
5670 .verify_authorizer_reply = verify_authorizer_reply,
5671 .invalidate_authorizer = invalidate_authorizer,
5672 .alloc_msg = alloc_msg,
5673 .reencode_message = osd_reencode_message,
5674 .sign_message = osd_sign_message,
5675 .check_message_signature = osd_check_message_signature,
5676 .fault = osd_fault,
5677};
1// SPDX-License-Identifier: GPL-2.0
2
3#include <linux/ceph/ceph_debug.h>
4
5#include <linux/module.h>
6#include <linux/err.h>
7#include <linux/highmem.h>
8#include <linux/mm.h>
9#include <linux/pagemap.h>
10#include <linux/slab.h>
11#include <linux/uaccess.h>
12#ifdef CONFIG_BLOCK
13#include <linux/bio.h>
14#endif
15
16#include <linux/ceph/ceph_features.h>
17#include <linux/ceph/libceph.h>
18#include <linux/ceph/osd_client.h>
19#include <linux/ceph/messenger.h>
20#include <linux/ceph/decode.h>
21#include <linux/ceph/auth.h>
22#include <linux/ceph/pagelist.h>
23#include <linux/ceph/striper.h>
24
25#define OSD_OPREPLY_FRONT_LEN 512
26
27static struct kmem_cache *ceph_osd_request_cache;
28
29static const struct ceph_connection_operations osd_con_ops;
30
31/*
32 * Implement client access to distributed object storage cluster.
33 *
34 * All data objects are stored within a cluster/cloud of OSDs, or
35 * "object storage devices." (Note that Ceph OSDs have _nothing_ to
36 * do with the T10 OSD extensions to SCSI.) Ceph OSDs are simply
37 * remote daemons serving up and coordinating consistent and safe
38 * access to storage.
39 *
40 * Cluster membership and the mapping of data objects onto storage devices
41 * are described by the osd map.
42 *
43 * We keep track of pending OSD requests (read, write), resubmit
44 * requests to different OSDs when the cluster topology/data layout
45 * change, or retry the affected requests when the communications
46 * channel with an OSD is reset.
47 */
48
49static void link_request(struct ceph_osd *osd, struct ceph_osd_request *req);
50static void unlink_request(struct ceph_osd *osd, struct ceph_osd_request *req);
51static void link_linger(struct ceph_osd *osd,
52 struct ceph_osd_linger_request *lreq);
53static void unlink_linger(struct ceph_osd *osd,
54 struct ceph_osd_linger_request *lreq);
55static void clear_backoffs(struct ceph_osd *osd);
56
57#if 1
58static inline bool rwsem_is_wrlocked(struct rw_semaphore *sem)
59{
60 bool wrlocked = true;
61
62 if (unlikely(down_read_trylock(sem))) {
63 wrlocked = false;
64 up_read(sem);
65 }
66
67 return wrlocked;
68}
69static inline void verify_osdc_locked(struct ceph_osd_client *osdc)
70{
71 WARN_ON(!rwsem_is_locked(&osdc->lock));
72}
73static inline void verify_osdc_wrlocked(struct ceph_osd_client *osdc)
74{
75 WARN_ON(!rwsem_is_wrlocked(&osdc->lock));
76}
77static inline void verify_osd_locked(struct ceph_osd *osd)
78{
79 struct ceph_osd_client *osdc = osd->o_osdc;
80
81 WARN_ON(!(mutex_is_locked(&osd->lock) &&
82 rwsem_is_locked(&osdc->lock)) &&
83 !rwsem_is_wrlocked(&osdc->lock));
84}
85static inline void verify_lreq_locked(struct ceph_osd_linger_request *lreq)
86{
87 WARN_ON(!mutex_is_locked(&lreq->lock));
88}
89#else
90static inline void verify_osdc_locked(struct ceph_osd_client *osdc) { }
91static inline void verify_osdc_wrlocked(struct ceph_osd_client *osdc) { }
92static inline void verify_osd_locked(struct ceph_osd *osd) { }
93static inline void verify_lreq_locked(struct ceph_osd_linger_request *lreq) { }
94#endif
95
96/*
97 * calculate the mapping of a file extent onto an object, and fill out the
98 * request accordingly. shorten extent as necessary if it crosses an
99 * object boundary.
100 *
101 * fill osd op in request message.
102 */
103static int calc_layout(struct ceph_file_layout *layout, u64 off, u64 *plen,
104 u64 *objnum, u64 *objoff, u64 *objlen)
105{
106 u64 orig_len = *plen;
107 u32 xlen;
108
109 /* object extent? */
110 ceph_calc_file_object_mapping(layout, off, orig_len, objnum,
111 objoff, &xlen);
112 *objlen = xlen;
113 if (*objlen < orig_len) {
114 *plen = *objlen;
115 dout(" skipping last %llu, final file extent %llu~%llu\n",
116 orig_len - *plen, off, *plen);
117 }
118
119 dout("calc_layout objnum=%llx %llu~%llu\n", *objnum, *objoff, *objlen);
120 return 0;
121}
122
123static void ceph_osd_data_init(struct ceph_osd_data *osd_data)
124{
125 memset(osd_data, 0, sizeof (*osd_data));
126 osd_data->type = CEPH_OSD_DATA_TYPE_NONE;
127}
128
129/*
130 * Consumes @pages if @own_pages is true.
131 */
132static void ceph_osd_data_pages_init(struct ceph_osd_data *osd_data,
133 struct page **pages, u64 length, u32 alignment,
134 bool pages_from_pool, bool own_pages)
135{
136 osd_data->type = CEPH_OSD_DATA_TYPE_PAGES;
137 osd_data->pages = pages;
138 osd_data->length = length;
139 osd_data->alignment = alignment;
140 osd_data->pages_from_pool = pages_from_pool;
141 osd_data->own_pages = own_pages;
142}
143
144/*
145 * Consumes a ref on @pagelist.
146 */
147static void ceph_osd_data_pagelist_init(struct ceph_osd_data *osd_data,
148 struct ceph_pagelist *pagelist)
149{
150 osd_data->type = CEPH_OSD_DATA_TYPE_PAGELIST;
151 osd_data->pagelist = pagelist;
152}
153
154#ifdef CONFIG_BLOCK
155static void ceph_osd_data_bio_init(struct ceph_osd_data *osd_data,
156 struct ceph_bio_iter *bio_pos,
157 u32 bio_length)
158{
159 osd_data->type = CEPH_OSD_DATA_TYPE_BIO;
160 osd_data->bio_pos = *bio_pos;
161 osd_data->bio_length = bio_length;
162}
163#endif /* CONFIG_BLOCK */
164
165static void ceph_osd_data_bvecs_init(struct ceph_osd_data *osd_data,
166 struct ceph_bvec_iter *bvec_pos,
167 u32 num_bvecs)
168{
169 osd_data->type = CEPH_OSD_DATA_TYPE_BVECS;
170 osd_data->bvec_pos = *bvec_pos;
171 osd_data->num_bvecs = num_bvecs;
172}
173
174static void ceph_osd_iter_init(struct ceph_osd_data *osd_data,
175 struct iov_iter *iter)
176{
177 osd_data->type = CEPH_OSD_DATA_TYPE_ITER;
178 osd_data->iter = *iter;
179}
180
181static struct ceph_osd_data *
182osd_req_op_raw_data_in(struct ceph_osd_request *osd_req, unsigned int which)
183{
184 BUG_ON(which >= osd_req->r_num_ops);
185
186 return &osd_req->r_ops[which].raw_data_in;
187}
188
189struct ceph_osd_data *
190osd_req_op_extent_osd_data(struct ceph_osd_request *osd_req,
191 unsigned int which)
192{
193 return osd_req_op_data(osd_req, which, extent, osd_data);
194}
195EXPORT_SYMBOL(osd_req_op_extent_osd_data);
196
197void osd_req_op_raw_data_in_pages(struct ceph_osd_request *osd_req,
198 unsigned int which, struct page **pages,
199 u64 length, u32 alignment,
200 bool pages_from_pool, bool own_pages)
201{
202 struct ceph_osd_data *osd_data;
203
204 osd_data = osd_req_op_raw_data_in(osd_req, which);
205 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
206 pages_from_pool, own_pages);
207}
208EXPORT_SYMBOL(osd_req_op_raw_data_in_pages);
209
210void osd_req_op_extent_osd_data_pages(struct ceph_osd_request *osd_req,
211 unsigned int which, struct page **pages,
212 u64 length, u32 alignment,
213 bool pages_from_pool, bool own_pages)
214{
215 struct ceph_osd_data *osd_data;
216
217 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
218 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
219 pages_from_pool, own_pages);
220}
221EXPORT_SYMBOL(osd_req_op_extent_osd_data_pages);
222
223void osd_req_op_extent_osd_data_pagelist(struct ceph_osd_request *osd_req,
224 unsigned int which, struct ceph_pagelist *pagelist)
225{
226 struct ceph_osd_data *osd_data;
227
228 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
229 ceph_osd_data_pagelist_init(osd_data, pagelist);
230}
231EXPORT_SYMBOL(osd_req_op_extent_osd_data_pagelist);
232
233#ifdef CONFIG_BLOCK
234void osd_req_op_extent_osd_data_bio(struct ceph_osd_request *osd_req,
235 unsigned int which,
236 struct ceph_bio_iter *bio_pos,
237 u32 bio_length)
238{
239 struct ceph_osd_data *osd_data;
240
241 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
242 ceph_osd_data_bio_init(osd_data, bio_pos, bio_length);
243}
244EXPORT_SYMBOL(osd_req_op_extent_osd_data_bio);
245#endif /* CONFIG_BLOCK */
246
247void osd_req_op_extent_osd_data_bvecs(struct ceph_osd_request *osd_req,
248 unsigned int which,
249 struct bio_vec *bvecs, u32 num_bvecs,
250 u32 bytes)
251{
252 struct ceph_osd_data *osd_data;
253 struct ceph_bvec_iter it = {
254 .bvecs = bvecs,
255 .iter = { .bi_size = bytes },
256 };
257
258 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
259 ceph_osd_data_bvecs_init(osd_data, &it, num_bvecs);
260}
261EXPORT_SYMBOL(osd_req_op_extent_osd_data_bvecs);
262
263void osd_req_op_extent_osd_data_bvec_pos(struct ceph_osd_request *osd_req,
264 unsigned int which,
265 struct ceph_bvec_iter *bvec_pos)
266{
267 struct ceph_osd_data *osd_data;
268
269 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
270 ceph_osd_data_bvecs_init(osd_data, bvec_pos, 0);
271}
272EXPORT_SYMBOL(osd_req_op_extent_osd_data_bvec_pos);
273
274/**
275 * osd_req_op_extent_osd_iter - Set up an operation with an iterator buffer
276 * @osd_req: The request to set up
277 * @which: Index of the operation in which to set the iter
278 * @iter: The buffer iterator
279 */
280void osd_req_op_extent_osd_iter(struct ceph_osd_request *osd_req,
281 unsigned int which, struct iov_iter *iter)
282{
283 struct ceph_osd_data *osd_data;
284
285 osd_data = osd_req_op_data(osd_req, which, extent, osd_data);
286 ceph_osd_iter_init(osd_data, iter);
287}
288EXPORT_SYMBOL(osd_req_op_extent_osd_iter);
289
290static void osd_req_op_cls_request_info_pagelist(
291 struct ceph_osd_request *osd_req,
292 unsigned int which, struct ceph_pagelist *pagelist)
293{
294 struct ceph_osd_data *osd_data;
295
296 osd_data = osd_req_op_data(osd_req, which, cls, request_info);
297 ceph_osd_data_pagelist_init(osd_data, pagelist);
298}
299
300void osd_req_op_cls_request_data_pagelist(
301 struct ceph_osd_request *osd_req,
302 unsigned int which, struct ceph_pagelist *pagelist)
303{
304 struct ceph_osd_data *osd_data;
305
306 osd_data = osd_req_op_data(osd_req, which, cls, request_data);
307 ceph_osd_data_pagelist_init(osd_data, pagelist);
308 osd_req->r_ops[which].cls.indata_len += pagelist->length;
309 osd_req->r_ops[which].indata_len += pagelist->length;
310}
311EXPORT_SYMBOL(osd_req_op_cls_request_data_pagelist);
312
313void osd_req_op_cls_request_data_pages(struct ceph_osd_request *osd_req,
314 unsigned int which, struct page **pages, u64 length,
315 u32 alignment, bool pages_from_pool, bool own_pages)
316{
317 struct ceph_osd_data *osd_data;
318
319 osd_data = osd_req_op_data(osd_req, which, cls, request_data);
320 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
321 pages_from_pool, own_pages);
322 osd_req->r_ops[which].cls.indata_len += length;
323 osd_req->r_ops[which].indata_len += length;
324}
325EXPORT_SYMBOL(osd_req_op_cls_request_data_pages);
326
327void osd_req_op_cls_request_data_bvecs(struct ceph_osd_request *osd_req,
328 unsigned int which,
329 struct bio_vec *bvecs, u32 num_bvecs,
330 u32 bytes)
331{
332 struct ceph_osd_data *osd_data;
333 struct ceph_bvec_iter it = {
334 .bvecs = bvecs,
335 .iter = { .bi_size = bytes },
336 };
337
338 osd_data = osd_req_op_data(osd_req, which, cls, request_data);
339 ceph_osd_data_bvecs_init(osd_data, &it, num_bvecs);
340 osd_req->r_ops[which].cls.indata_len += bytes;
341 osd_req->r_ops[which].indata_len += bytes;
342}
343EXPORT_SYMBOL(osd_req_op_cls_request_data_bvecs);
344
345void osd_req_op_cls_response_data_pages(struct ceph_osd_request *osd_req,
346 unsigned int which, struct page **pages, u64 length,
347 u32 alignment, bool pages_from_pool, bool own_pages)
348{
349 struct ceph_osd_data *osd_data;
350
351 osd_data = osd_req_op_data(osd_req, which, cls, response_data);
352 ceph_osd_data_pages_init(osd_data, pages, length, alignment,
353 pages_from_pool, own_pages);
354}
355EXPORT_SYMBOL(osd_req_op_cls_response_data_pages);
356
357static u64 ceph_osd_data_length(struct ceph_osd_data *osd_data)
358{
359 switch (osd_data->type) {
360 case CEPH_OSD_DATA_TYPE_NONE:
361 return 0;
362 case CEPH_OSD_DATA_TYPE_PAGES:
363 return osd_data->length;
364 case CEPH_OSD_DATA_TYPE_PAGELIST:
365 return (u64)osd_data->pagelist->length;
366#ifdef CONFIG_BLOCK
367 case CEPH_OSD_DATA_TYPE_BIO:
368 return (u64)osd_data->bio_length;
369#endif /* CONFIG_BLOCK */
370 case CEPH_OSD_DATA_TYPE_BVECS:
371 return osd_data->bvec_pos.iter.bi_size;
372 case CEPH_OSD_DATA_TYPE_ITER:
373 return iov_iter_count(&osd_data->iter);
374 default:
375 WARN(true, "unrecognized data type %d\n", (int)osd_data->type);
376 return 0;
377 }
378}
379
380static void ceph_osd_data_release(struct ceph_osd_data *osd_data)
381{
382 if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGES && osd_data->own_pages) {
383 int num_pages;
384
385 num_pages = calc_pages_for((u64)osd_data->alignment,
386 (u64)osd_data->length);
387 ceph_release_page_vector(osd_data->pages, num_pages);
388 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGELIST) {
389 ceph_pagelist_release(osd_data->pagelist);
390 }
391 ceph_osd_data_init(osd_data);
392}
393
394static void osd_req_op_data_release(struct ceph_osd_request *osd_req,
395 unsigned int which)
396{
397 struct ceph_osd_req_op *op;
398
399 BUG_ON(which >= osd_req->r_num_ops);
400 op = &osd_req->r_ops[which];
401
402 switch (op->op) {
403 case CEPH_OSD_OP_READ:
404 case CEPH_OSD_OP_SPARSE_READ:
405 case CEPH_OSD_OP_WRITE:
406 case CEPH_OSD_OP_WRITEFULL:
407 kfree(op->extent.sparse_ext);
408 ceph_osd_data_release(&op->extent.osd_data);
409 break;
410 case CEPH_OSD_OP_CALL:
411 ceph_osd_data_release(&op->cls.request_info);
412 ceph_osd_data_release(&op->cls.request_data);
413 ceph_osd_data_release(&op->cls.response_data);
414 break;
415 case CEPH_OSD_OP_SETXATTR:
416 case CEPH_OSD_OP_CMPXATTR:
417 ceph_osd_data_release(&op->xattr.osd_data);
418 break;
419 case CEPH_OSD_OP_STAT:
420 ceph_osd_data_release(&op->raw_data_in);
421 break;
422 case CEPH_OSD_OP_NOTIFY_ACK:
423 ceph_osd_data_release(&op->notify_ack.request_data);
424 break;
425 case CEPH_OSD_OP_NOTIFY:
426 ceph_osd_data_release(&op->notify.request_data);
427 ceph_osd_data_release(&op->notify.response_data);
428 break;
429 case CEPH_OSD_OP_LIST_WATCHERS:
430 ceph_osd_data_release(&op->list_watchers.response_data);
431 break;
432 case CEPH_OSD_OP_COPY_FROM2:
433 ceph_osd_data_release(&op->copy_from.osd_data);
434 break;
435 default:
436 break;
437 }
438}
439
440/*
441 * Assumes @t is zero-initialized.
442 */
443static void target_init(struct ceph_osd_request_target *t)
444{
445 ceph_oid_init(&t->base_oid);
446 ceph_oloc_init(&t->base_oloc);
447 ceph_oid_init(&t->target_oid);
448 ceph_oloc_init(&t->target_oloc);
449
450 ceph_osds_init(&t->acting);
451 ceph_osds_init(&t->up);
452 t->size = -1;
453 t->min_size = -1;
454
455 t->osd = CEPH_HOMELESS_OSD;
456}
457
458static void target_copy(struct ceph_osd_request_target *dest,
459 const struct ceph_osd_request_target *src)
460{
461 ceph_oid_copy(&dest->base_oid, &src->base_oid);
462 ceph_oloc_copy(&dest->base_oloc, &src->base_oloc);
463 ceph_oid_copy(&dest->target_oid, &src->target_oid);
464 ceph_oloc_copy(&dest->target_oloc, &src->target_oloc);
465
466 dest->pgid = src->pgid; /* struct */
467 dest->spgid = src->spgid; /* struct */
468 dest->pg_num = src->pg_num;
469 dest->pg_num_mask = src->pg_num_mask;
470 ceph_osds_copy(&dest->acting, &src->acting);
471 ceph_osds_copy(&dest->up, &src->up);
472 dest->size = src->size;
473 dest->min_size = src->min_size;
474 dest->sort_bitwise = src->sort_bitwise;
475 dest->recovery_deletes = src->recovery_deletes;
476
477 dest->flags = src->flags;
478 dest->used_replica = src->used_replica;
479 dest->paused = src->paused;
480
481 dest->epoch = src->epoch;
482 dest->last_force_resend = src->last_force_resend;
483
484 dest->osd = src->osd;
485}
486
487static void target_destroy(struct ceph_osd_request_target *t)
488{
489 ceph_oid_destroy(&t->base_oid);
490 ceph_oloc_destroy(&t->base_oloc);
491 ceph_oid_destroy(&t->target_oid);
492 ceph_oloc_destroy(&t->target_oloc);
493}
494
495/*
496 * requests
497 */
498static void request_release_checks(struct ceph_osd_request *req)
499{
500 WARN_ON(!RB_EMPTY_NODE(&req->r_node));
501 WARN_ON(!RB_EMPTY_NODE(&req->r_mc_node));
502 WARN_ON(!list_empty(&req->r_private_item));
503 WARN_ON(req->r_osd);
504}
505
506static void ceph_osdc_release_request(struct kref *kref)
507{
508 struct ceph_osd_request *req = container_of(kref,
509 struct ceph_osd_request, r_kref);
510 unsigned int which;
511
512 dout("%s %p (r_request %p r_reply %p)\n", __func__, req,
513 req->r_request, req->r_reply);
514 request_release_checks(req);
515
516 if (req->r_request)
517 ceph_msg_put(req->r_request);
518 if (req->r_reply)
519 ceph_msg_put(req->r_reply);
520
521 for (which = 0; which < req->r_num_ops; which++)
522 osd_req_op_data_release(req, which);
523
524 target_destroy(&req->r_t);
525 ceph_put_snap_context(req->r_snapc);
526
527 if (req->r_mempool)
528 mempool_free(req, req->r_osdc->req_mempool);
529 else if (req->r_num_ops <= CEPH_OSD_SLAB_OPS)
530 kmem_cache_free(ceph_osd_request_cache, req);
531 else
532 kfree(req);
533}
534
535void ceph_osdc_get_request(struct ceph_osd_request *req)
536{
537 dout("%s %p (was %d)\n", __func__, req,
538 kref_read(&req->r_kref));
539 kref_get(&req->r_kref);
540}
541EXPORT_SYMBOL(ceph_osdc_get_request);
542
543void ceph_osdc_put_request(struct ceph_osd_request *req)
544{
545 if (req) {
546 dout("%s %p (was %d)\n", __func__, req,
547 kref_read(&req->r_kref));
548 kref_put(&req->r_kref, ceph_osdc_release_request);
549 }
550}
551EXPORT_SYMBOL(ceph_osdc_put_request);
552
553static void request_init(struct ceph_osd_request *req)
554{
555 /* req only, each op is zeroed in osd_req_op_init() */
556 memset(req, 0, sizeof(*req));
557
558 kref_init(&req->r_kref);
559 init_completion(&req->r_completion);
560 RB_CLEAR_NODE(&req->r_node);
561 RB_CLEAR_NODE(&req->r_mc_node);
562 INIT_LIST_HEAD(&req->r_private_item);
563
564 target_init(&req->r_t);
565}
566
567struct ceph_osd_request *ceph_osdc_alloc_request(struct ceph_osd_client *osdc,
568 struct ceph_snap_context *snapc,
569 unsigned int num_ops,
570 bool use_mempool,
571 gfp_t gfp_flags)
572{
573 struct ceph_osd_request *req;
574
575 if (use_mempool) {
576 BUG_ON(num_ops > CEPH_OSD_SLAB_OPS);
577 req = mempool_alloc(osdc->req_mempool, gfp_flags);
578 } else if (num_ops <= CEPH_OSD_SLAB_OPS) {
579 req = kmem_cache_alloc(ceph_osd_request_cache, gfp_flags);
580 } else {
581 BUG_ON(num_ops > CEPH_OSD_MAX_OPS);
582 req = kmalloc(struct_size(req, r_ops, num_ops), gfp_flags);
583 }
584 if (unlikely(!req))
585 return NULL;
586
587 request_init(req);
588 req->r_osdc = osdc;
589 req->r_mempool = use_mempool;
590 req->r_num_ops = num_ops;
591 req->r_snapid = CEPH_NOSNAP;
592 req->r_snapc = ceph_get_snap_context(snapc);
593
594 dout("%s req %p\n", __func__, req);
595 return req;
596}
597EXPORT_SYMBOL(ceph_osdc_alloc_request);
598
599static int ceph_oloc_encoding_size(const struct ceph_object_locator *oloc)
600{
601 return 8 + 4 + 4 + 4 + (oloc->pool_ns ? oloc->pool_ns->len : 0);
602}
603
604static int __ceph_osdc_alloc_messages(struct ceph_osd_request *req, gfp_t gfp,
605 int num_request_data_items,
606 int num_reply_data_items)
607{
608 struct ceph_osd_client *osdc = req->r_osdc;
609 struct ceph_msg *msg;
610 int msg_size;
611
612 WARN_ON(req->r_request || req->r_reply);
613 WARN_ON(ceph_oid_empty(&req->r_base_oid));
614 WARN_ON(ceph_oloc_empty(&req->r_base_oloc));
615
616 /* create request message */
617 msg_size = CEPH_ENCODING_START_BLK_LEN +
618 CEPH_PGID_ENCODING_LEN + 1; /* spgid */
619 msg_size += 4 + 4 + 4; /* hash, osdmap_epoch, flags */
620 msg_size += CEPH_ENCODING_START_BLK_LEN +
621 sizeof(struct ceph_osd_reqid); /* reqid */
622 msg_size += sizeof(struct ceph_blkin_trace_info); /* trace */
623 msg_size += 4 + sizeof(struct ceph_timespec); /* client_inc, mtime */
624 msg_size += CEPH_ENCODING_START_BLK_LEN +
625 ceph_oloc_encoding_size(&req->r_base_oloc); /* oloc */
626 msg_size += 4 + req->r_base_oid.name_len; /* oid */
627 msg_size += 2 + req->r_num_ops * sizeof(struct ceph_osd_op);
628 msg_size += 8; /* snapid */
629 msg_size += 8; /* snap_seq */
630 msg_size += 4 + 8 * (req->r_snapc ? req->r_snapc->num_snaps : 0);
631 msg_size += 4 + 8; /* retry_attempt, features */
632
633 if (req->r_mempool)
634 msg = ceph_msgpool_get(&osdc->msgpool_op, msg_size,
635 num_request_data_items);
636 else
637 msg = ceph_msg_new2(CEPH_MSG_OSD_OP, msg_size,
638 num_request_data_items, gfp, true);
639 if (!msg)
640 return -ENOMEM;
641
642 memset(msg->front.iov_base, 0, msg->front.iov_len);
643 req->r_request = msg;
644
645 /* create reply message */
646 msg_size = OSD_OPREPLY_FRONT_LEN;
647 msg_size += req->r_base_oid.name_len;
648 msg_size += req->r_num_ops * sizeof(struct ceph_osd_op);
649
650 if (req->r_mempool)
651 msg = ceph_msgpool_get(&osdc->msgpool_op_reply, msg_size,
652 num_reply_data_items);
653 else
654 msg = ceph_msg_new2(CEPH_MSG_OSD_OPREPLY, msg_size,
655 num_reply_data_items, gfp, true);
656 if (!msg)
657 return -ENOMEM;
658
659 req->r_reply = msg;
660
661 return 0;
662}
663
664static bool osd_req_opcode_valid(u16 opcode)
665{
666 switch (opcode) {
667#define GENERATE_CASE(op, opcode, str) case CEPH_OSD_OP_##op: return true;
668__CEPH_FORALL_OSD_OPS(GENERATE_CASE)
669#undef GENERATE_CASE
670 default:
671 return false;
672 }
673}
674
675static void get_num_data_items(struct ceph_osd_request *req,
676 int *num_request_data_items,
677 int *num_reply_data_items)
678{
679 struct ceph_osd_req_op *op;
680
681 *num_request_data_items = 0;
682 *num_reply_data_items = 0;
683
684 for (op = req->r_ops; op != &req->r_ops[req->r_num_ops]; op++) {
685 switch (op->op) {
686 /* request */
687 case CEPH_OSD_OP_WRITE:
688 case CEPH_OSD_OP_WRITEFULL:
689 case CEPH_OSD_OP_SETXATTR:
690 case CEPH_OSD_OP_CMPXATTR:
691 case CEPH_OSD_OP_NOTIFY_ACK:
692 case CEPH_OSD_OP_COPY_FROM2:
693 *num_request_data_items += 1;
694 break;
695
696 /* reply */
697 case CEPH_OSD_OP_STAT:
698 case CEPH_OSD_OP_READ:
699 case CEPH_OSD_OP_SPARSE_READ:
700 case CEPH_OSD_OP_LIST_WATCHERS:
701 *num_reply_data_items += 1;
702 break;
703
704 /* both */
705 case CEPH_OSD_OP_NOTIFY:
706 *num_request_data_items += 1;
707 *num_reply_data_items += 1;
708 break;
709 case CEPH_OSD_OP_CALL:
710 *num_request_data_items += 2;
711 *num_reply_data_items += 1;
712 break;
713
714 default:
715 WARN_ON(!osd_req_opcode_valid(op->op));
716 break;
717 }
718 }
719}
720
721/*
722 * oid, oloc and OSD op opcode(s) must be filled in before this function
723 * is called.
724 */
725int ceph_osdc_alloc_messages(struct ceph_osd_request *req, gfp_t gfp)
726{
727 int num_request_data_items, num_reply_data_items;
728
729 get_num_data_items(req, &num_request_data_items, &num_reply_data_items);
730 return __ceph_osdc_alloc_messages(req, gfp, num_request_data_items,
731 num_reply_data_items);
732}
733EXPORT_SYMBOL(ceph_osdc_alloc_messages);
734
735/*
736 * This is an osd op init function for opcodes that have no data or
737 * other information associated with them. It also serves as a
738 * common init routine for all the other init functions, below.
739 */
740struct ceph_osd_req_op *
741osd_req_op_init(struct ceph_osd_request *osd_req, unsigned int which,
742 u16 opcode, u32 flags)
743{
744 struct ceph_osd_req_op *op;
745
746 BUG_ON(which >= osd_req->r_num_ops);
747 BUG_ON(!osd_req_opcode_valid(opcode));
748
749 op = &osd_req->r_ops[which];
750 memset(op, 0, sizeof (*op));
751 op->op = opcode;
752 op->flags = flags;
753
754 return op;
755}
756EXPORT_SYMBOL(osd_req_op_init);
757
758void osd_req_op_extent_init(struct ceph_osd_request *osd_req,
759 unsigned int which, u16 opcode,
760 u64 offset, u64 length,
761 u64 truncate_size, u32 truncate_seq)
762{
763 struct ceph_osd_req_op *op = osd_req_op_init(osd_req, which,
764 opcode, 0);
765 size_t payload_len = 0;
766
767 BUG_ON(opcode != CEPH_OSD_OP_READ && opcode != CEPH_OSD_OP_WRITE &&
768 opcode != CEPH_OSD_OP_WRITEFULL && opcode != CEPH_OSD_OP_ZERO &&
769 opcode != CEPH_OSD_OP_TRUNCATE && opcode != CEPH_OSD_OP_SPARSE_READ);
770
771 op->extent.offset = offset;
772 op->extent.length = length;
773 op->extent.truncate_size = truncate_size;
774 op->extent.truncate_seq = truncate_seq;
775 if (opcode == CEPH_OSD_OP_WRITE || opcode == CEPH_OSD_OP_WRITEFULL)
776 payload_len += length;
777
778 op->indata_len = payload_len;
779}
780EXPORT_SYMBOL(osd_req_op_extent_init);
781
782void osd_req_op_extent_update(struct ceph_osd_request *osd_req,
783 unsigned int which, u64 length)
784{
785 struct ceph_osd_req_op *op;
786 u64 previous;
787
788 BUG_ON(which >= osd_req->r_num_ops);
789 op = &osd_req->r_ops[which];
790 previous = op->extent.length;
791
792 if (length == previous)
793 return; /* Nothing to do */
794 BUG_ON(length > previous);
795
796 op->extent.length = length;
797 if (op->op == CEPH_OSD_OP_WRITE || op->op == CEPH_OSD_OP_WRITEFULL)
798 op->indata_len -= previous - length;
799}
800EXPORT_SYMBOL(osd_req_op_extent_update);
801
802void osd_req_op_extent_dup_last(struct ceph_osd_request *osd_req,
803 unsigned int which, u64 offset_inc)
804{
805 struct ceph_osd_req_op *op, *prev_op;
806
807 BUG_ON(which + 1 >= osd_req->r_num_ops);
808
809 prev_op = &osd_req->r_ops[which];
810 op = osd_req_op_init(osd_req, which + 1, prev_op->op, prev_op->flags);
811 /* dup previous one */
812 op->indata_len = prev_op->indata_len;
813 op->outdata_len = prev_op->outdata_len;
814 op->extent = prev_op->extent;
815 /* adjust offset */
816 op->extent.offset += offset_inc;
817 op->extent.length -= offset_inc;
818
819 if (op->op == CEPH_OSD_OP_WRITE || op->op == CEPH_OSD_OP_WRITEFULL)
820 op->indata_len -= offset_inc;
821}
822EXPORT_SYMBOL(osd_req_op_extent_dup_last);
823
824int osd_req_op_cls_init(struct ceph_osd_request *osd_req, unsigned int which,
825 const char *class, const char *method)
826{
827 struct ceph_osd_req_op *op;
828 struct ceph_pagelist *pagelist;
829 size_t payload_len = 0;
830 size_t size;
831 int ret;
832
833 op = osd_req_op_init(osd_req, which, CEPH_OSD_OP_CALL, 0);
834
835 pagelist = ceph_pagelist_alloc(GFP_NOFS);
836 if (!pagelist)
837 return -ENOMEM;
838
839 op->cls.class_name = class;
840 size = strlen(class);
841 BUG_ON(size > (size_t) U8_MAX);
842 op->cls.class_len = size;
843 ret = ceph_pagelist_append(pagelist, class, size);
844 if (ret)
845 goto err_pagelist_free;
846 payload_len += size;
847
848 op->cls.method_name = method;
849 size = strlen(method);
850 BUG_ON(size > (size_t) U8_MAX);
851 op->cls.method_len = size;
852 ret = ceph_pagelist_append(pagelist, method, size);
853 if (ret)
854 goto err_pagelist_free;
855 payload_len += size;
856
857 osd_req_op_cls_request_info_pagelist(osd_req, which, pagelist);
858 op->indata_len = payload_len;
859 return 0;
860
861err_pagelist_free:
862 ceph_pagelist_release(pagelist);
863 return ret;
864}
865EXPORT_SYMBOL(osd_req_op_cls_init);
866
867int osd_req_op_xattr_init(struct ceph_osd_request *osd_req, unsigned int which,
868 u16 opcode, const char *name, const void *value,
869 size_t size, u8 cmp_op, u8 cmp_mode)
870{
871 struct ceph_osd_req_op *op = osd_req_op_init(osd_req, which,
872 opcode, 0);
873 struct ceph_pagelist *pagelist;
874 size_t payload_len;
875 int ret;
876
877 BUG_ON(opcode != CEPH_OSD_OP_SETXATTR && opcode != CEPH_OSD_OP_CMPXATTR);
878
879 pagelist = ceph_pagelist_alloc(GFP_NOFS);
880 if (!pagelist)
881 return -ENOMEM;
882
883 payload_len = strlen(name);
884 op->xattr.name_len = payload_len;
885 ret = ceph_pagelist_append(pagelist, name, payload_len);
886 if (ret)
887 goto err_pagelist_free;
888
889 op->xattr.value_len = size;
890 ret = ceph_pagelist_append(pagelist, value, size);
891 if (ret)
892 goto err_pagelist_free;
893 payload_len += size;
894
895 op->xattr.cmp_op = cmp_op;
896 op->xattr.cmp_mode = cmp_mode;
897
898 ceph_osd_data_pagelist_init(&op->xattr.osd_data, pagelist);
899 op->indata_len = payload_len;
900 return 0;
901
902err_pagelist_free:
903 ceph_pagelist_release(pagelist);
904 return ret;
905}
906EXPORT_SYMBOL(osd_req_op_xattr_init);
907
908/*
909 * @watch_opcode: CEPH_OSD_WATCH_OP_*
910 */
911static void osd_req_op_watch_init(struct ceph_osd_request *req, int which,
912 u8 watch_opcode, u64 cookie, u32 gen)
913{
914 struct ceph_osd_req_op *op;
915
916 op = osd_req_op_init(req, which, CEPH_OSD_OP_WATCH, 0);
917 op->watch.cookie = cookie;
918 op->watch.op = watch_opcode;
919 op->watch.gen = gen;
920}
921
922/*
923 * prot_ver, timeout and notify payload (may be empty) should already be
924 * encoded in @request_pl
925 */
926static void osd_req_op_notify_init(struct ceph_osd_request *req, int which,
927 u64 cookie, struct ceph_pagelist *request_pl)
928{
929 struct ceph_osd_req_op *op;
930
931 op = osd_req_op_init(req, which, CEPH_OSD_OP_NOTIFY, 0);
932 op->notify.cookie = cookie;
933
934 ceph_osd_data_pagelist_init(&op->notify.request_data, request_pl);
935 op->indata_len = request_pl->length;
936}
937
938/*
939 * @flags: CEPH_OSD_OP_ALLOC_HINT_FLAG_*
940 */
941void osd_req_op_alloc_hint_init(struct ceph_osd_request *osd_req,
942 unsigned int which,
943 u64 expected_object_size,
944 u64 expected_write_size,
945 u32 flags)
946{
947 struct ceph_osd_req_op *op;
948
949 op = osd_req_op_init(osd_req, which, CEPH_OSD_OP_SETALLOCHINT, 0);
950 op->alloc_hint.expected_object_size = expected_object_size;
951 op->alloc_hint.expected_write_size = expected_write_size;
952 op->alloc_hint.flags = flags;
953
954 /*
955 * CEPH_OSD_OP_SETALLOCHINT op is advisory and therefore deemed
956 * not worth a feature bit. Set FAILOK per-op flag to make
957 * sure older osds don't trip over an unsupported opcode.
958 */
959 op->flags |= CEPH_OSD_OP_FLAG_FAILOK;
960}
961EXPORT_SYMBOL(osd_req_op_alloc_hint_init);
962
963static void ceph_osdc_msg_data_add(struct ceph_msg *msg,
964 struct ceph_osd_data *osd_data)
965{
966 u64 length = ceph_osd_data_length(osd_data);
967
968 if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGES) {
969 BUG_ON(length > (u64) SIZE_MAX);
970 if (length)
971 ceph_msg_data_add_pages(msg, osd_data->pages,
972 length, osd_data->alignment, false);
973 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_PAGELIST) {
974 BUG_ON(!length);
975 ceph_msg_data_add_pagelist(msg, osd_data->pagelist);
976#ifdef CONFIG_BLOCK
977 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_BIO) {
978 ceph_msg_data_add_bio(msg, &osd_data->bio_pos, length);
979#endif
980 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_BVECS) {
981 ceph_msg_data_add_bvecs(msg, &osd_data->bvec_pos);
982 } else if (osd_data->type == CEPH_OSD_DATA_TYPE_ITER) {
983 ceph_msg_data_add_iter(msg, &osd_data->iter);
984 } else {
985 BUG_ON(osd_data->type != CEPH_OSD_DATA_TYPE_NONE);
986 }
987}
988
989static u32 osd_req_encode_op(struct ceph_osd_op *dst,
990 const struct ceph_osd_req_op *src)
991{
992 switch (src->op) {
993 case CEPH_OSD_OP_STAT:
994 break;
995 case CEPH_OSD_OP_READ:
996 case CEPH_OSD_OP_SPARSE_READ:
997 case CEPH_OSD_OP_WRITE:
998 case CEPH_OSD_OP_WRITEFULL:
999 case CEPH_OSD_OP_ZERO:
1000 case CEPH_OSD_OP_TRUNCATE:
1001 dst->extent.offset = cpu_to_le64(src->extent.offset);
1002 dst->extent.length = cpu_to_le64(src->extent.length);
1003 dst->extent.truncate_size =
1004 cpu_to_le64(src->extent.truncate_size);
1005 dst->extent.truncate_seq =
1006 cpu_to_le32(src->extent.truncate_seq);
1007 break;
1008 case CEPH_OSD_OP_CALL:
1009 dst->cls.class_len = src->cls.class_len;
1010 dst->cls.method_len = src->cls.method_len;
1011 dst->cls.indata_len = cpu_to_le32(src->cls.indata_len);
1012 break;
1013 case CEPH_OSD_OP_WATCH:
1014 dst->watch.cookie = cpu_to_le64(src->watch.cookie);
1015 dst->watch.ver = cpu_to_le64(0);
1016 dst->watch.op = src->watch.op;
1017 dst->watch.gen = cpu_to_le32(src->watch.gen);
1018 break;
1019 case CEPH_OSD_OP_NOTIFY_ACK:
1020 break;
1021 case CEPH_OSD_OP_NOTIFY:
1022 dst->notify.cookie = cpu_to_le64(src->notify.cookie);
1023 break;
1024 case CEPH_OSD_OP_LIST_WATCHERS:
1025 break;
1026 case CEPH_OSD_OP_SETALLOCHINT:
1027 dst->alloc_hint.expected_object_size =
1028 cpu_to_le64(src->alloc_hint.expected_object_size);
1029 dst->alloc_hint.expected_write_size =
1030 cpu_to_le64(src->alloc_hint.expected_write_size);
1031 dst->alloc_hint.flags = cpu_to_le32(src->alloc_hint.flags);
1032 break;
1033 case CEPH_OSD_OP_SETXATTR:
1034 case CEPH_OSD_OP_CMPXATTR:
1035 dst->xattr.name_len = cpu_to_le32(src->xattr.name_len);
1036 dst->xattr.value_len = cpu_to_le32(src->xattr.value_len);
1037 dst->xattr.cmp_op = src->xattr.cmp_op;
1038 dst->xattr.cmp_mode = src->xattr.cmp_mode;
1039 break;
1040 case CEPH_OSD_OP_CREATE:
1041 case CEPH_OSD_OP_DELETE:
1042 break;
1043 case CEPH_OSD_OP_COPY_FROM2:
1044 dst->copy_from.snapid = cpu_to_le64(src->copy_from.snapid);
1045 dst->copy_from.src_version =
1046 cpu_to_le64(src->copy_from.src_version);
1047 dst->copy_from.flags = src->copy_from.flags;
1048 dst->copy_from.src_fadvise_flags =
1049 cpu_to_le32(src->copy_from.src_fadvise_flags);
1050 break;
1051 case CEPH_OSD_OP_ASSERT_VER:
1052 dst->assert_ver.unused = cpu_to_le64(0);
1053 dst->assert_ver.ver = cpu_to_le64(src->assert_ver.ver);
1054 break;
1055 default:
1056 pr_err("unsupported osd opcode %s\n",
1057 ceph_osd_op_name(src->op));
1058 WARN_ON(1);
1059
1060 return 0;
1061 }
1062
1063 dst->op = cpu_to_le16(src->op);
1064 dst->flags = cpu_to_le32(src->flags);
1065 dst->payload_len = cpu_to_le32(src->indata_len);
1066
1067 return src->indata_len;
1068}
1069
1070/*
1071 * build new request AND message, calculate layout, and adjust file
1072 * extent as needed.
1073 *
1074 * if the file was recently truncated, we include information about its
1075 * old and new size so that the object can be updated appropriately. (we
1076 * avoid synchronously deleting truncated objects because it's slow.)
1077 */
1078struct ceph_osd_request *ceph_osdc_new_request(struct ceph_osd_client *osdc,
1079 struct ceph_file_layout *layout,
1080 struct ceph_vino vino,
1081 u64 off, u64 *plen,
1082 unsigned int which, int num_ops,
1083 int opcode, int flags,
1084 struct ceph_snap_context *snapc,
1085 u32 truncate_seq,
1086 u64 truncate_size,
1087 bool use_mempool)
1088{
1089 struct ceph_osd_request *req;
1090 u64 objnum = 0;
1091 u64 objoff = 0;
1092 u64 objlen = 0;
1093 int r;
1094
1095 BUG_ON(opcode != CEPH_OSD_OP_READ && opcode != CEPH_OSD_OP_WRITE &&
1096 opcode != CEPH_OSD_OP_ZERO && opcode != CEPH_OSD_OP_TRUNCATE &&
1097 opcode != CEPH_OSD_OP_CREATE && opcode != CEPH_OSD_OP_DELETE &&
1098 opcode != CEPH_OSD_OP_SPARSE_READ);
1099
1100 req = ceph_osdc_alloc_request(osdc, snapc, num_ops, use_mempool,
1101 GFP_NOFS);
1102 if (!req) {
1103 r = -ENOMEM;
1104 goto fail;
1105 }
1106
1107 /* calculate max write size */
1108 r = calc_layout(layout, off, plen, &objnum, &objoff, &objlen);
1109 if (r)
1110 goto fail;
1111
1112 if (opcode == CEPH_OSD_OP_CREATE || opcode == CEPH_OSD_OP_DELETE) {
1113 osd_req_op_init(req, which, opcode, 0);
1114 } else {
1115 u32 object_size = layout->object_size;
1116 u32 object_base = off - objoff;
1117 if (!(truncate_seq == 1 && truncate_size == -1ULL)) {
1118 if (truncate_size <= object_base) {
1119 truncate_size = 0;
1120 } else {
1121 truncate_size -= object_base;
1122 if (truncate_size > object_size)
1123 truncate_size = object_size;
1124 }
1125 }
1126 osd_req_op_extent_init(req, which, opcode, objoff, objlen,
1127 truncate_size, truncate_seq);
1128 }
1129
1130 req->r_base_oloc.pool = layout->pool_id;
1131 req->r_base_oloc.pool_ns = ceph_try_get_string(layout->pool_ns);
1132 ceph_oid_printf(&req->r_base_oid, "%llx.%08llx", vino.ino, objnum);
1133 req->r_flags = flags | osdc->client->options->read_from_replica;
1134
1135 req->r_snapid = vino.snap;
1136 if (flags & CEPH_OSD_FLAG_WRITE)
1137 req->r_data_offset = off;
1138
1139 if (num_ops > 1) {
1140 int num_req_ops, num_rep_ops;
1141
1142 /*
1143 * If this is a multi-op write request, assume that we'll need
1144 * request ops. If it's a multi-op read then assume we'll need
1145 * reply ops. Anything else and call it -EINVAL.
1146 */
1147 if (flags & CEPH_OSD_FLAG_WRITE) {
1148 num_req_ops = num_ops;
1149 num_rep_ops = 0;
1150 } else if (flags & CEPH_OSD_FLAG_READ) {
1151 num_req_ops = 0;
1152 num_rep_ops = num_ops;
1153 } else {
1154 r = -EINVAL;
1155 goto fail;
1156 }
1157
1158 r = __ceph_osdc_alloc_messages(req, GFP_NOFS, num_req_ops,
1159 num_rep_ops);
1160 } else {
1161 r = ceph_osdc_alloc_messages(req, GFP_NOFS);
1162 }
1163 if (r)
1164 goto fail;
1165
1166 return req;
1167
1168fail:
1169 ceph_osdc_put_request(req);
1170 return ERR_PTR(r);
1171}
1172EXPORT_SYMBOL(ceph_osdc_new_request);
1173
1174int __ceph_alloc_sparse_ext_map(struct ceph_osd_req_op *op, int cnt)
1175{
1176 WARN_ON(op->op != CEPH_OSD_OP_SPARSE_READ);
1177
1178 op->extent.sparse_ext_cnt = cnt;
1179 op->extent.sparse_ext = kmalloc_array(cnt,
1180 sizeof(*op->extent.sparse_ext),
1181 GFP_NOFS);
1182 if (!op->extent.sparse_ext)
1183 return -ENOMEM;
1184 return 0;
1185}
1186EXPORT_SYMBOL(__ceph_alloc_sparse_ext_map);
1187
1188/*
1189 * We keep osd requests in an rbtree, sorted by ->r_tid.
1190 */
1191DEFINE_RB_FUNCS(request, struct ceph_osd_request, r_tid, r_node)
1192DEFINE_RB_FUNCS(request_mc, struct ceph_osd_request, r_tid, r_mc_node)
1193
1194/*
1195 * Call @fn on each OSD request as long as @fn returns 0.
1196 */
1197static void for_each_request(struct ceph_osd_client *osdc,
1198 int (*fn)(struct ceph_osd_request *req, void *arg),
1199 void *arg)
1200{
1201 struct rb_node *n, *p;
1202
1203 for (n = rb_first(&osdc->osds); n; n = rb_next(n)) {
1204 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
1205
1206 for (p = rb_first(&osd->o_requests); p; ) {
1207 struct ceph_osd_request *req =
1208 rb_entry(p, struct ceph_osd_request, r_node);
1209
1210 p = rb_next(p);
1211 if (fn(req, arg))
1212 return;
1213 }
1214 }
1215
1216 for (p = rb_first(&osdc->homeless_osd.o_requests); p; ) {
1217 struct ceph_osd_request *req =
1218 rb_entry(p, struct ceph_osd_request, r_node);
1219
1220 p = rb_next(p);
1221 if (fn(req, arg))
1222 return;
1223 }
1224}
1225
1226static bool osd_homeless(struct ceph_osd *osd)
1227{
1228 return osd->o_osd == CEPH_HOMELESS_OSD;
1229}
1230
1231static bool osd_registered(struct ceph_osd *osd)
1232{
1233 verify_osdc_locked(osd->o_osdc);
1234
1235 return !RB_EMPTY_NODE(&osd->o_node);
1236}
1237
1238/*
1239 * Assumes @osd is zero-initialized.
1240 */
1241static void osd_init(struct ceph_osd *osd)
1242{
1243 refcount_set(&osd->o_ref, 1);
1244 RB_CLEAR_NODE(&osd->o_node);
1245 spin_lock_init(&osd->o_requests_lock);
1246 osd->o_requests = RB_ROOT;
1247 osd->o_linger_requests = RB_ROOT;
1248 osd->o_backoff_mappings = RB_ROOT;
1249 osd->o_backoffs_by_id = RB_ROOT;
1250 INIT_LIST_HEAD(&osd->o_osd_lru);
1251 INIT_LIST_HEAD(&osd->o_keepalive_item);
1252 osd->o_incarnation = 1;
1253 mutex_init(&osd->lock);
1254}
1255
1256static void ceph_init_sparse_read(struct ceph_sparse_read *sr)
1257{
1258 kfree(sr->sr_extent);
1259 memset(sr, '\0', sizeof(*sr));
1260 sr->sr_state = CEPH_SPARSE_READ_HDR;
1261}
1262
1263static void osd_cleanup(struct ceph_osd *osd)
1264{
1265 WARN_ON(!RB_EMPTY_NODE(&osd->o_node));
1266 WARN_ON(!RB_EMPTY_ROOT(&osd->o_requests));
1267 WARN_ON(!RB_EMPTY_ROOT(&osd->o_linger_requests));
1268 WARN_ON(!RB_EMPTY_ROOT(&osd->o_backoff_mappings));
1269 WARN_ON(!RB_EMPTY_ROOT(&osd->o_backoffs_by_id));
1270 WARN_ON(!list_empty(&osd->o_osd_lru));
1271 WARN_ON(!list_empty(&osd->o_keepalive_item));
1272
1273 ceph_init_sparse_read(&osd->o_sparse_read);
1274
1275 if (osd->o_auth.authorizer) {
1276 WARN_ON(osd_homeless(osd));
1277 ceph_auth_destroy_authorizer(osd->o_auth.authorizer);
1278 }
1279}
1280
1281/*
1282 * Track open sessions with osds.
1283 */
1284static struct ceph_osd *create_osd(struct ceph_osd_client *osdc, int onum)
1285{
1286 struct ceph_osd *osd;
1287
1288 WARN_ON(onum == CEPH_HOMELESS_OSD);
1289
1290 osd = kzalloc(sizeof(*osd), GFP_NOIO | __GFP_NOFAIL);
1291 osd_init(osd);
1292 osd->o_osdc = osdc;
1293 osd->o_osd = onum;
1294 osd->o_sparse_op_idx = -1;
1295
1296 ceph_init_sparse_read(&osd->o_sparse_read);
1297
1298 ceph_con_init(&osd->o_con, osd, &osd_con_ops, &osdc->client->msgr);
1299
1300 return osd;
1301}
1302
1303static struct ceph_osd *get_osd(struct ceph_osd *osd)
1304{
1305 if (refcount_inc_not_zero(&osd->o_ref)) {
1306 dout("get_osd %p %d -> %d\n", osd, refcount_read(&osd->o_ref)-1,
1307 refcount_read(&osd->o_ref));
1308 return osd;
1309 } else {
1310 dout("get_osd %p FAIL\n", osd);
1311 return NULL;
1312 }
1313}
1314
1315static void put_osd(struct ceph_osd *osd)
1316{
1317 dout("put_osd %p %d -> %d\n", osd, refcount_read(&osd->o_ref),
1318 refcount_read(&osd->o_ref) - 1);
1319 if (refcount_dec_and_test(&osd->o_ref)) {
1320 osd_cleanup(osd);
1321 kfree(osd);
1322 }
1323}
1324
1325DEFINE_RB_FUNCS(osd, struct ceph_osd, o_osd, o_node)
1326
1327static void __move_osd_to_lru(struct ceph_osd *osd)
1328{
1329 struct ceph_osd_client *osdc = osd->o_osdc;
1330
1331 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1332 BUG_ON(!list_empty(&osd->o_osd_lru));
1333
1334 spin_lock(&osdc->osd_lru_lock);
1335 list_add_tail(&osd->o_osd_lru, &osdc->osd_lru);
1336 spin_unlock(&osdc->osd_lru_lock);
1337
1338 osd->lru_ttl = jiffies + osdc->client->options->osd_idle_ttl;
1339}
1340
1341static void maybe_move_osd_to_lru(struct ceph_osd *osd)
1342{
1343 if (RB_EMPTY_ROOT(&osd->o_requests) &&
1344 RB_EMPTY_ROOT(&osd->o_linger_requests))
1345 __move_osd_to_lru(osd);
1346}
1347
1348static void __remove_osd_from_lru(struct ceph_osd *osd)
1349{
1350 struct ceph_osd_client *osdc = osd->o_osdc;
1351
1352 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1353
1354 spin_lock(&osdc->osd_lru_lock);
1355 if (!list_empty(&osd->o_osd_lru))
1356 list_del_init(&osd->o_osd_lru);
1357 spin_unlock(&osdc->osd_lru_lock);
1358}
1359
1360/*
1361 * Close the connection and assign any leftover requests to the
1362 * homeless session.
1363 */
1364static void close_osd(struct ceph_osd *osd)
1365{
1366 struct ceph_osd_client *osdc = osd->o_osdc;
1367 struct rb_node *n;
1368
1369 verify_osdc_wrlocked(osdc);
1370 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1371
1372 ceph_con_close(&osd->o_con);
1373
1374 for (n = rb_first(&osd->o_requests); n; ) {
1375 struct ceph_osd_request *req =
1376 rb_entry(n, struct ceph_osd_request, r_node);
1377
1378 n = rb_next(n); /* unlink_request() */
1379
1380 dout(" reassigning req %p tid %llu\n", req, req->r_tid);
1381 unlink_request(osd, req);
1382 link_request(&osdc->homeless_osd, req);
1383 }
1384 for (n = rb_first(&osd->o_linger_requests); n; ) {
1385 struct ceph_osd_linger_request *lreq =
1386 rb_entry(n, struct ceph_osd_linger_request, node);
1387
1388 n = rb_next(n); /* unlink_linger() */
1389
1390 dout(" reassigning lreq %p linger_id %llu\n", lreq,
1391 lreq->linger_id);
1392 unlink_linger(osd, lreq);
1393 link_linger(&osdc->homeless_osd, lreq);
1394 }
1395 clear_backoffs(osd);
1396
1397 __remove_osd_from_lru(osd);
1398 erase_osd(&osdc->osds, osd);
1399 put_osd(osd);
1400}
1401
1402/*
1403 * reset osd connect
1404 */
1405static int reopen_osd(struct ceph_osd *osd)
1406{
1407 struct ceph_entity_addr *peer_addr;
1408
1409 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
1410
1411 if (RB_EMPTY_ROOT(&osd->o_requests) &&
1412 RB_EMPTY_ROOT(&osd->o_linger_requests)) {
1413 close_osd(osd);
1414 return -ENODEV;
1415 }
1416
1417 peer_addr = &osd->o_osdc->osdmap->osd_addr[osd->o_osd];
1418 if (!memcmp(peer_addr, &osd->o_con.peer_addr, sizeof (*peer_addr)) &&
1419 !ceph_con_opened(&osd->o_con)) {
1420 struct rb_node *n;
1421
1422 dout("osd addr hasn't changed and connection never opened, "
1423 "letting msgr retry\n");
1424 /* touch each r_stamp for handle_timeout()'s benfit */
1425 for (n = rb_first(&osd->o_requests); n; n = rb_next(n)) {
1426 struct ceph_osd_request *req =
1427 rb_entry(n, struct ceph_osd_request, r_node);
1428 req->r_stamp = jiffies;
1429 }
1430
1431 return -EAGAIN;
1432 }
1433
1434 ceph_con_close(&osd->o_con);
1435 ceph_con_open(&osd->o_con, CEPH_ENTITY_TYPE_OSD, osd->o_osd, peer_addr);
1436 osd->o_incarnation++;
1437
1438 return 0;
1439}
1440
1441static struct ceph_osd *lookup_create_osd(struct ceph_osd_client *osdc, int o,
1442 bool wrlocked)
1443{
1444 struct ceph_osd *osd;
1445
1446 if (wrlocked)
1447 verify_osdc_wrlocked(osdc);
1448 else
1449 verify_osdc_locked(osdc);
1450
1451 if (o != CEPH_HOMELESS_OSD)
1452 osd = lookup_osd(&osdc->osds, o);
1453 else
1454 osd = &osdc->homeless_osd;
1455 if (!osd) {
1456 if (!wrlocked)
1457 return ERR_PTR(-EAGAIN);
1458
1459 osd = create_osd(osdc, o);
1460 insert_osd(&osdc->osds, osd);
1461 ceph_con_open(&osd->o_con, CEPH_ENTITY_TYPE_OSD, osd->o_osd,
1462 &osdc->osdmap->osd_addr[osd->o_osd]);
1463 }
1464
1465 dout("%s osdc %p osd%d -> osd %p\n", __func__, osdc, o, osd);
1466 return osd;
1467}
1468
1469/*
1470 * Create request <-> OSD session relation.
1471 *
1472 * @req has to be assigned a tid, @osd may be homeless.
1473 */
1474static void link_request(struct ceph_osd *osd, struct ceph_osd_request *req)
1475{
1476 verify_osd_locked(osd);
1477 WARN_ON(!req->r_tid || req->r_osd);
1478 dout("%s osd %p osd%d req %p tid %llu\n", __func__, osd, osd->o_osd,
1479 req, req->r_tid);
1480
1481 if (!osd_homeless(osd))
1482 __remove_osd_from_lru(osd);
1483 else
1484 atomic_inc(&osd->o_osdc->num_homeless);
1485
1486 get_osd(osd);
1487 spin_lock(&osd->o_requests_lock);
1488 insert_request(&osd->o_requests, req);
1489 spin_unlock(&osd->o_requests_lock);
1490 req->r_osd = osd;
1491}
1492
1493static void unlink_request(struct ceph_osd *osd, struct ceph_osd_request *req)
1494{
1495 verify_osd_locked(osd);
1496 WARN_ON(req->r_osd != osd);
1497 dout("%s osd %p osd%d req %p tid %llu\n", __func__, osd, osd->o_osd,
1498 req, req->r_tid);
1499
1500 req->r_osd = NULL;
1501 spin_lock(&osd->o_requests_lock);
1502 erase_request(&osd->o_requests, req);
1503 spin_unlock(&osd->o_requests_lock);
1504 put_osd(osd);
1505
1506 if (!osd_homeless(osd))
1507 maybe_move_osd_to_lru(osd);
1508 else
1509 atomic_dec(&osd->o_osdc->num_homeless);
1510}
1511
1512static bool __pool_full(struct ceph_pg_pool_info *pi)
1513{
1514 return pi->flags & CEPH_POOL_FLAG_FULL;
1515}
1516
1517static bool have_pool_full(struct ceph_osd_client *osdc)
1518{
1519 struct rb_node *n;
1520
1521 for (n = rb_first(&osdc->osdmap->pg_pools); n; n = rb_next(n)) {
1522 struct ceph_pg_pool_info *pi =
1523 rb_entry(n, struct ceph_pg_pool_info, node);
1524
1525 if (__pool_full(pi))
1526 return true;
1527 }
1528
1529 return false;
1530}
1531
1532static bool pool_full(struct ceph_osd_client *osdc, s64 pool_id)
1533{
1534 struct ceph_pg_pool_info *pi;
1535
1536 pi = ceph_pg_pool_by_id(osdc->osdmap, pool_id);
1537 if (!pi)
1538 return false;
1539
1540 return __pool_full(pi);
1541}
1542
1543/*
1544 * Returns whether a request should be blocked from being sent
1545 * based on the current osdmap and osd_client settings.
1546 */
1547static bool target_should_be_paused(struct ceph_osd_client *osdc,
1548 const struct ceph_osd_request_target *t,
1549 struct ceph_pg_pool_info *pi)
1550{
1551 bool pauserd = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD);
1552 bool pausewr = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR) ||
1553 ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
1554 __pool_full(pi);
1555
1556 WARN_ON(pi->id != t->target_oloc.pool);
1557 return ((t->flags & CEPH_OSD_FLAG_READ) && pauserd) ||
1558 ((t->flags & CEPH_OSD_FLAG_WRITE) && pausewr) ||
1559 (osdc->osdmap->epoch < osdc->epoch_barrier);
1560}
1561
1562static int pick_random_replica(const struct ceph_osds *acting)
1563{
1564 int i = get_random_u32_below(acting->size);
1565
1566 dout("%s picked osd%d, primary osd%d\n", __func__,
1567 acting->osds[i], acting->primary);
1568 return i;
1569}
1570
1571/*
1572 * Picks the closest replica based on client's location given by
1573 * crush_location option. Prefers the primary if the locality is
1574 * the same.
1575 */
1576static int pick_closest_replica(struct ceph_osd_client *osdc,
1577 const struct ceph_osds *acting)
1578{
1579 struct ceph_options *opt = osdc->client->options;
1580 int best_i, best_locality;
1581 int i = 0, locality;
1582
1583 do {
1584 locality = ceph_get_crush_locality(osdc->osdmap,
1585 acting->osds[i],
1586 &opt->crush_locs);
1587 if (i == 0 ||
1588 (locality >= 0 && best_locality < 0) ||
1589 (locality >= 0 && best_locality >= 0 &&
1590 locality < best_locality)) {
1591 best_i = i;
1592 best_locality = locality;
1593 }
1594 } while (++i < acting->size);
1595
1596 dout("%s picked osd%d with locality %d, primary osd%d\n", __func__,
1597 acting->osds[best_i], best_locality, acting->primary);
1598 return best_i;
1599}
1600
1601enum calc_target_result {
1602 CALC_TARGET_NO_ACTION = 0,
1603 CALC_TARGET_NEED_RESEND,
1604 CALC_TARGET_POOL_DNE,
1605};
1606
1607static enum calc_target_result calc_target(struct ceph_osd_client *osdc,
1608 struct ceph_osd_request_target *t,
1609 bool any_change)
1610{
1611 struct ceph_pg_pool_info *pi;
1612 struct ceph_pg pgid, last_pgid;
1613 struct ceph_osds up, acting;
1614 bool is_read = t->flags & CEPH_OSD_FLAG_READ;
1615 bool is_write = t->flags & CEPH_OSD_FLAG_WRITE;
1616 bool force_resend = false;
1617 bool unpaused = false;
1618 bool legacy_change = false;
1619 bool split = false;
1620 bool sort_bitwise = ceph_osdmap_flag(osdc, CEPH_OSDMAP_SORTBITWISE);
1621 bool recovery_deletes = ceph_osdmap_flag(osdc,
1622 CEPH_OSDMAP_RECOVERY_DELETES);
1623 enum calc_target_result ct_res;
1624
1625 t->epoch = osdc->osdmap->epoch;
1626 pi = ceph_pg_pool_by_id(osdc->osdmap, t->base_oloc.pool);
1627 if (!pi) {
1628 t->osd = CEPH_HOMELESS_OSD;
1629 ct_res = CALC_TARGET_POOL_DNE;
1630 goto out;
1631 }
1632
1633 if (osdc->osdmap->epoch == pi->last_force_request_resend) {
1634 if (t->last_force_resend < pi->last_force_request_resend) {
1635 t->last_force_resend = pi->last_force_request_resend;
1636 force_resend = true;
1637 } else if (t->last_force_resend == 0) {
1638 force_resend = true;
1639 }
1640 }
1641
1642 /* apply tiering */
1643 ceph_oid_copy(&t->target_oid, &t->base_oid);
1644 ceph_oloc_copy(&t->target_oloc, &t->base_oloc);
1645 if ((t->flags & CEPH_OSD_FLAG_IGNORE_OVERLAY) == 0) {
1646 if (is_read && pi->read_tier >= 0)
1647 t->target_oloc.pool = pi->read_tier;
1648 if (is_write && pi->write_tier >= 0)
1649 t->target_oloc.pool = pi->write_tier;
1650
1651 pi = ceph_pg_pool_by_id(osdc->osdmap, t->target_oloc.pool);
1652 if (!pi) {
1653 t->osd = CEPH_HOMELESS_OSD;
1654 ct_res = CALC_TARGET_POOL_DNE;
1655 goto out;
1656 }
1657 }
1658
1659 __ceph_object_locator_to_pg(pi, &t->target_oid, &t->target_oloc, &pgid);
1660 last_pgid.pool = pgid.pool;
1661 last_pgid.seed = ceph_stable_mod(pgid.seed, t->pg_num, t->pg_num_mask);
1662
1663 ceph_pg_to_up_acting_osds(osdc->osdmap, pi, &pgid, &up, &acting);
1664 if (any_change &&
1665 ceph_is_new_interval(&t->acting,
1666 &acting,
1667 &t->up,
1668 &up,
1669 t->size,
1670 pi->size,
1671 t->min_size,
1672 pi->min_size,
1673 t->pg_num,
1674 pi->pg_num,
1675 t->sort_bitwise,
1676 sort_bitwise,
1677 t->recovery_deletes,
1678 recovery_deletes,
1679 &last_pgid))
1680 force_resend = true;
1681
1682 if (t->paused && !target_should_be_paused(osdc, t, pi)) {
1683 t->paused = false;
1684 unpaused = true;
1685 }
1686 legacy_change = ceph_pg_compare(&t->pgid, &pgid) ||
1687 ceph_osds_changed(&t->acting, &acting,
1688 t->used_replica || any_change);
1689 if (t->pg_num)
1690 split = ceph_pg_is_split(&last_pgid, t->pg_num, pi->pg_num);
1691
1692 if (legacy_change || force_resend || split) {
1693 t->pgid = pgid; /* struct */
1694 ceph_pg_to_primary_shard(osdc->osdmap, pi, &pgid, &t->spgid);
1695 ceph_osds_copy(&t->acting, &acting);
1696 ceph_osds_copy(&t->up, &up);
1697 t->size = pi->size;
1698 t->min_size = pi->min_size;
1699 t->pg_num = pi->pg_num;
1700 t->pg_num_mask = pi->pg_num_mask;
1701 t->sort_bitwise = sort_bitwise;
1702 t->recovery_deletes = recovery_deletes;
1703
1704 if ((t->flags & (CEPH_OSD_FLAG_BALANCE_READS |
1705 CEPH_OSD_FLAG_LOCALIZE_READS)) &&
1706 !is_write && pi->type == CEPH_POOL_TYPE_REP &&
1707 acting.size > 1) {
1708 int pos;
1709
1710 WARN_ON(!is_read || acting.osds[0] != acting.primary);
1711 if (t->flags & CEPH_OSD_FLAG_BALANCE_READS) {
1712 pos = pick_random_replica(&acting);
1713 } else {
1714 pos = pick_closest_replica(osdc, &acting);
1715 }
1716 t->osd = acting.osds[pos];
1717 t->used_replica = pos > 0;
1718 } else {
1719 t->osd = acting.primary;
1720 t->used_replica = false;
1721 }
1722 }
1723
1724 if (unpaused || legacy_change || force_resend || split)
1725 ct_res = CALC_TARGET_NEED_RESEND;
1726 else
1727 ct_res = CALC_TARGET_NO_ACTION;
1728
1729out:
1730 dout("%s t %p -> %d%d%d%d ct_res %d osd%d\n", __func__, t, unpaused,
1731 legacy_change, force_resend, split, ct_res, t->osd);
1732 return ct_res;
1733}
1734
1735static struct ceph_spg_mapping *alloc_spg_mapping(void)
1736{
1737 struct ceph_spg_mapping *spg;
1738
1739 spg = kmalloc(sizeof(*spg), GFP_NOIO);
1740 if (!spg)
1741 return NULL;
1742
1743 RB_CLEAR_NODE(&spg->node);
1744 spg->backoffs = RB_ROOT;
1745 return spg;
1746}
1747
1748static void free_spg_mapping(struct ceph_spg_mapping *spg)
1749{
1750 WARN_ON(!RB_EMPTY_NODE(&spg->node));
1751 WARN_ON(!RB_EMPTY_ROOT(&spg->backoffs));
1752
1753 kfree(spg);
1754}
1755
1756/*
1757 * rbtree of ceph_spg_mapping for handling map<spg_t, ...>, similar to
1758 * ceph_pg_mapping. Used to track OSD backoffs -- a backoff [range] is
1759 * defined only within a specific spgid; it does not pass anything to
1760 * children on split, or to another primary.
1761 */
1762DEFINE_RB_FUNCS2(spg_mapping, struct ceph_spg_mapping, spgid, ceph_spg_compare,
1763 RB_BYPTR, const struct ceph_spg *, node)
1764
1765static u64 hoid_get_bitwise_key(const struct ceph_hobject_id *hoid)
1766{
1767 return hoid->is_max ? 0x100000000ull : hoid->hash_reverse_bits;
1768}
1769
1770static void hoid_get_effective_key(const struct ceph_hobject_id *hoid,
1771 void **pkey, size_t *pkey_len)
1772{
1773 if (hoid->key_len) {
1774 *pkey = hoid->key;
1775 *pkey_len = hoid->key_len;
1776 } else {
1777 *pkey = hoid->oid;
1778 *pkey_len = hoid->oid_len;
1779 }
1780}
1781
1782static int compare_names(const void *name1, size_t name1_len,
1783 const void *name2, size_t name2_len)
1784{
1785 int ret;
1786
1787 ret = memcmp(name1, name2, min(name1_len, name2_len));
1788 if (!ret) {
1789 if (name1_len < name2_len)
1790 ret = -1;
1791 else if (name1_len > name2_len)
1792 ret = 1;
1793 }
1794 return ret;
1795}
1796
1797static int hoid_compare(const struct ceph_hobject_id *lhs,
1798 const struct ceph_hobject_id *rhs)
1799{
1800 void *effective_key1, *effective_key2;
1801 size_t effective_key1_len, effective_key2_len;
1802 int ret;
1803
1804 if (lhs->is_max < rhs->is_max)
1805 return -1;
1806 if (lhs->is_max > rhs->is_max)
1807 return 1;
1808
1809 if (lhs->pool < rhs->pool)
1810 return -1;
1811 if (lhs->pool > rhs->pool)
1812 return 1;
1813
1814 if (hoid_get_bitwise_key(lhs) < hoid_get_bitwise_key(rhs))
1815 return -1;
1816 if (hoid_get_bitwise_key(lhs) > hoid_get_bitwise_key(rhs))
1817 return 1;
1818
1819 ret = compare_names(lhs->nspace, lhs->nspace_len,
1820 rhs->nspace, rhs->nspace_len);
1821 if (ret)
1822 return ret;
1823
1824 hoid_get_effective_key(lhs, &effective_key1, &effective_key1_len);
1825 hoid_get_effective_key(rhs, &effective_key2, &effective_key2_len);
1826 ret = compare_names(effective_key1, effective_key1_len,
1827 effective_key2, effective_key2_len);
1828 if (ret)
1829 return ret;
1830
1831 ret = compare_names(lhs->oid, lhs->oid_len, rhs->oid, rhs->oid_len);
1832 if (ret)
1833 return ret;
1834
1835 if (lhs->snapid < rhs->snapid)
1836 return -1;
1837 if (lhs->snapid > rhs->snapid)
1838 return 1;
1839
1840 return 0;
1841}
1842
1843/*
1844 * For decoding ->begin and ->end of MOSDBackoff only -- no MIN/MAX
1845 * compat stuff here.
1846 *
1847 * Assumes @hoid is zero-initialized.
1848 */
1849static int decode_hoid(void **p, void *end, struct ceph_hobject_id *hoid)
1850{
1851 u8 struct_v;
1852 u32 struct_len;
1853 int ret;
1854
1855 ret = ceph_start_decoding(p, end, 4, "hobject_t", &struct_v,
1856 &struct_len);
1857 if (ret)
1858 return ret;
1859
1860 if (struct_v < 4) {
1861 pr_err("got struct_v %d < 4 of hobject_t\n", struct_v);
1862 goto e_inval;
1863 }
1864
1865 hoid->key = ceph_extract_encoded_string(p, end, &hoid->key_len,
1866 GFP_NOIO);
1867 if (IS_ERR(hoid->key)) {
1868 ret = PTR_ERR(hoid->key);
1869 hoid->key = NULL;
1870 return ret;
1871 }
1872
1873 hoid->oid = ceph_extract_encoded_string(p, end, &hoid->oid_len,
1874 GFP_NOIO);
1875 if (IS_ERR(hoid->oid)) {
1876 ret = PTR_ERR(hoid->oid);
1877 hoid->oid = NULL;
1878 return ret;
1879 }
1880
1881 ceph_decode_64_safe(p, end, hoid->snapid, e_inval);
1882 ceph_decode_32_safe(p, end, hoid->hash, e_inval);
1883 ceph_decode_8_safe(p, end, hoid->is_max, e_inval);
1884
1885 hoid->nspace = ceph_extract_encoded_string(p, end, &hoid->nspace_len,
1886 GFP_NOIO);
1887 if (IS_ERR(hoid->nspace)) {
1888 ret = PTR_ERR(hoid->nspace);
1889 hoid->nspace = NULL;
1890 return ret;
1891 }
1892
1893 ceph_decode_64_safe(p, end, hoid->pool, e_inval);
1894
1895 ceph_hoid_build_hash_cache(hoid);
1896 return 0;
1897
1898e_inval:
1899 return -EINVAL;
1900}
1901
1902static int hoid_encoding_size(const struct ceph_hobject_id *hoid)
1903{
1904 return 8 + 4 + 1 + 8 + /* snapid, hash, is_max, pool */
1905 4 + hoid->key_len + 4 + hoid->oid_len + 4 + hoid->nspace_len;
1906}
1907
1908static void encode_hoid(void **p, void *end, const struct ceph_hobject_id *hoid)
1909{
1910 ceph_start_encoding(p, 4, 3, hoid_encoding_size(hoid));
1911 ceph_encode_string(p, end, hoid->key, hoid->key_len);
1912 ceph_encode_string(p, end, hoid->oid, hoid->oid_len);
1913 ceph_encode_64(p, hoid->snapid);
1914 ceph_encode_32(p, hoid->hash);
1915 ceph_encode_8(p, hoid->is_max);
1916 ceph_encode_string(p, end, hoid->nspace, hoid->nspace_len);
1917 ceph_encode_64(p, hoid->pool);
1918}
1919
1920static void free_hoid(struct ceph_hobject_id *hoid)
1921{
1922 if (hoid) {
1923 kfree(hoid->key);
1924 kfree(hoid->oid);
1925 kfree(hoid->nspace);
1926 kfree(hoid);
1927 }
1928}
1929
1930static struct ceph_osd_backoff *alloc_backoff(void)
1931{
1932 struct ceph_osd_backoff *backoff;
1933
1934 backoff = kzalloc(sizeof(*backoff), GFP_NOIO);
1935 if (!backoff)
1936 return NULL;
1937
1938 RB_CLEAR_NODE(&backoff->spg_node);
1939 RB_CLEAR_NODE(&backoff->id_node);
1940 return backoff;
1941}
1942
1943static void free_backoff(struct ceph_osd_backoff *backoff)
1944{
1945 WARN_ON(!RB_EMPTY_NODE(&backoff->spg_node));
1946 WARN_ON(!RB_EMPTY_NODE(&backoff->id_node));
1947
1948 free_hoid(backoff->begin);
1949 free_hoid(backoff->end);
1950 kfree(backoff);
1951}
1952
1953/*
1954 * Within a specific spgid, backoffs are managed by ->begin hoid.
1955 */
1956DEFINE_RB_INSDEL_FUNCS2(backoff, struct ceph_osd_backoff, begin, hoid_compare,
1957 RB_BYVAL, spg_node);
1958
1959static struct ceph_osd_backoff *lookup_containing_backoff(struct rb_root *root,
1960 const struct ceph_hobject_id *hoid)
1961{
1962 struct rb_node *n = root->rb_node;
1963
1964 while (n) {
1965 struct ceph_osd_backoff *cur =
1966 rb_entry(n, struct ceph_osd_backoff, spg_node);
1967 int cmp;
1968
1969 cmp = hoid_compare(hoid, cur->begin);
1970 if (cmp < 0) {
1971 n = n->rb_left;
1972 } else if (cmp > 0) {
1973 if (hoid_compare(hoid, cur->end) < 0)
1974 return cur;
1975
1976 n = n->rb_right;
1977 } else {
1978 return cur;
1979 }
1980 }
1981
1982 return NULL;
1983}
1984
1985/*
1986 * Each backoff has a unique id within its OSD session.
1987 */
1988DEFINE_RB_FUNCS(backoff_by_id, struct ceph_osd_backoff, id, id_node)
1989
1990static void clear_backoffs(struct ceph_osd *osd)
1991{
1992 while (!RB_EMPTY_ROOT(&osd->o_backoff_mappings)) {
1993 struct ceph_spg_mapping *spg =
1994 rb_entry(rb_first(&osd->o_backoff_mappings),
1995 struct ceph_spg_mapping, node);
1996
1997 while (!RB_EMPTY_ROOT(&spg->backoffs)) {
1998 struct ceph_osd_backoff *backoff =
1999 rb_entry(rb_first(&spg->backoffs),
2000 struct ceph_osd_backoff, spg_node);
2001
2002 erase_backoff(&spg->backoffs, backoff);
2003 erase_backoff_by_id(&osd->o_backoffs_by_id, backoff);
2004 free_backoff(backoff);
2005 }
2006 erase_spg_mapping(&osd->o_backoff_mappings, spg);
2007 free_spg_mapping(spg);
2008 }
2009}
2010
2011/*
2012 * Set up a temporary, non-owning view into @t.
2013 */
2014static void hoid_fill_from_target(struct ceph_hobject_id *hoid,
2015 const struct ceph_osd_request_target *t)
2016{
2017 hoid->key = NULL;
2018 hoid->key_len = 0;
2019 hoid->oid = t->target_oid.name;
2020 hoid->oid_len = t->target_oid.name_len;
2021 hoid->snapid = CEPH_NOSNAP;
2022 hoid->hash = t->pgid.seed;
2023 hoid->is_max = false;
2024 if (t->target_oloc.pool_ns) {
2025 hoid->nspace = t->target_oloc.pool_ns->str;
2026 hoid->nspace_len = t->target_oloc.pool_ns->len;
2027 } else {
2028 hoid->nspace = NULL;
2029 hoid->nspace_len = 0;
2030 }
2031 hoid->pool = t->target_oloc.pool;
2032 ceph_hoid_build_hash_cache(hoid);
2033}
2034
2035static bool should_plug_request(struct ceph_osd_request *req)
2036{
2037 struct ceph_osd *osd = req->r_osd;
2038 struct ceph_spg_mapping *spg;
2039 struct ceph_osd_backoff *backoff;
2040 struct ceph_hobject_id hoid;
2041
2042 spg = lookup_spg_mapping(&osd->o_backoff_mappings, &req->r_t.spgid);
2043 if (!spg)
2044 return false;
2045
2046 hoid_fill_from_target(&hoid, &req->r_t);
2047 backoff = lookup_containing_backoff(&spg->backoffs, &hoid);
2048 if (!backoff)
2049 return false;
2050
2051 dout("%s req %p tid %llu backoff osd%d spgid %llu.%xs%d id %llu\n",
2052 __func__, req, req->r_tid, osd->o_osd, backoff->spgid.pgid.pool,
2053 backoff->spgid.pgid.seed, backoff->spgid.shard, backoff->id);
2054 return true;
2055}
2056
2057/*
2058 * Keep get_num_data_items() in sync with this function.
2059 */
2060static void setup_request_data(struct ceph_osd_request *req)
2061{
2062 struct ceph_msg *request_msg = req->r_request;
2063 struct ceph_msg *reply_msg = req->r_reply;
2064 struct ceph_osd_req_op *op;
2065
2066 if (req->r_request->num_data_items || req->r_reply->num_data_items)
2067 return;
2068
2069 WARN_ON(request_msg->data_length || reply_msg->data_length);
2070 for (op = req->r_ops; op != &req->r_ops[req->r_num_ops]; op++) {
2071 switch (op->op) {
2072 /* request */
2073 case CEPH_OSD_OP_WRITE:
2074 case CEPH_OSD_OP_WRITEFULL:
2075 WARN_ON(op->indata_len != op->extent.length);
2076 ceph_osdc_msg_data_add(request_msg,
2077 &op->extent.osd_data);
2078 break;
2079 case CEPH_OSD_OP_SETXATTR:
2080 case CEPH_OSD_OP_CMPXATTR:
2081 WARN_ON(op->indata_len != op->xattr.name_len +
2082 op->xattr.value_len);
2083 ceph_osdc_msg_data_add(request_msg,
2084 &op->xattr.osd_data);
2085 break;
2086 case CEPH_OSD_OP_NOTIFY_ACK:
2087 ceph_osdc_msg_data_add(request_msg,
2088 &op->notify_ack.request_data);
2089 break;
2090 case CEPH_OSD_OP_COPY_FROM2:
2091 ceph_osdc_msg_data_add(request_msg,
2092 &op->copy_from.osd_data);
2093 break;
2094
2095 /* reply */
2096 case CEPH_OSD_OP_STAT:
2097 ceph_osdc_msg_data_add(reply_msg,
2098 &op->raw_data_in);
2099 break;
2100 case CEPH_OSD_OP_READ:
2101 case CEPH_OSD_OP_SPARSE_READ:
2102 ceph_osdc_msg_data_add(reply_msg,
2103 &op->extent.osd_data);
2104 break;
2105 case CEPH_OSD_OP_LIST_WATCHERS:
2106 ceph_osdc_msg_data_add(reply_msg,
2107 &op->list_watchers.response_data);
2108 break;
2109
2110 /* both */
2111 case CEPH_OSD_OP_CALL:
2112 WARN_ON(op->indata_len != op->cls.class_len +
2113 op->cls.method_len +
2114 op->cls.indata_len);
2115 ceph_osdc_msg_data_add(request_msg,
2116 &op->cls.request_info);
2117 /* optional, can be NONE */
2118 ceph_osdc_msg_data_add(request_msg,
2119 &op->cls.request_data);
2120 /* optional, can be NONE */
2121 ceph_osdc_msg_data_add(reply_msg,
2122 &op->cls.response_data);
2123 break;
2124 case CEPH_OSD_OP_NOTIFY:
2125 ceph_osdc_msg_data_add(request_msg,
2126 &op->notify.request_data);
2127 ceph_osdc_msg_data_add(reply_msg,
2128 &op->notify.response_data);
2129 break;
2130 }
2131 }
2132}
2133
2134static void encode_pgid(void **p, const struct ceph_pg *pgid)
2135{
2136 ceph_encode_8(p, 1);
2137 ceph_encode_64(p, pgid->pool);
2138 ceph_encode_32(p, pgid->seed);
2139 ceph_encode_32(p, -1); /* preferred */
2140}
2141
2142static void encode_spgid(void **p, const struct ceph_spg *spgid)
2143{
2144 ceph_start_encoding(p, 1, 1, CEPH_PGID_ENCODING_LEN + 1);
2145 encode_pgid(p, &spgid->pgid);
2146 ceph_encode_8(p, spgid->shard);
2147}
2148
2149static void encode_oloc(void **p, void *end,
2150 const struct ceph_object_locator *oloc)
2151{
2152 ceph_start_encoding(p, 5, 4, ceph_oloc_encoding_size(oloc));
2153 ceph_encode_64(p, oloc->pool);
2154 ceph_encode_32(p, -1); /* preferred */
2155 ceph_encode_32(p, 0); /* key len */
2156 if (oloc->pool_ns)
2157 ceph_encode_string(p, end, oloc->pool_ns->str,
2158 oloc->pool_ns->len);
2159 else
2160 ceph_encode_32(p, 0);
2161}
2162
2163static void encode_request_partial(struct ceph_osd_request *req,
2164 struct ceph_msg *msg)
2165{
2166 void *p = msg->front.iov_base;
2167 void *const end = p + msg->front_alloc_len;
2168 u32 data_len = 0;
2169 int i;
2170
2171 if (req->r_flags & CEPH_OSD_FLAG_WRITE) {
2172 /* snapshots aren't writeable */
2173 WARN_ON(req->r_snapid != CEPH_NOSNAP);
2174 } else {
2175 WARN_ON(req->r_mtime.tv_sec || req->r_mtime.tv_nsec ||
2176 req->r_data_offset || req->r_snapc);
2177 }
2178
2179 setup_request_data(req);
2180
2181 encode_spgid(&p, &req->r_t.spgid); /* actual spg */
2182 ceph_encode_32(&p, req->r_t.pgid.seed); /* raw hash */
2183 ceph_encode_32(&p, req->r_osdc->osdmap->epoch);
2184 ceph_encode_32(&p, req->r_flags);
2185
2186 /* reqid */
2187 ceph_start_encoding(&p, 2, 2, sizeof(struct ceph_osd_reqid));
2188 memset(p, 0, sizeof(struct ceph_osd_reqid));
2189 p += sizeof(struct ceph_osd_reqid);
2190
2191 /* trace */
2192 memset(p, 0, sizeof(struct ceph_blkin_trace_info));
2193 p += sizeof(struct ceph_blkin_trace_info);
2194
2195 ceph_encode_32(&p, 0); /* client_inc, always 0 */
2196 ceph_encode_timespec64(p, &req->r_mtime);
2197 p += sizeof(struct ceph_timespec);
2198
2199 encode_oloc(&p, end, &req->r_t.target_oloc);
2200 ceph_encode_string(&p, end, req->r_t.target_oid.name,
2201 req->r_t.target_oid.name_len);
2202
2203 /* ops, can imply data */
2204 ceph_encode_16(&p, req->r_num_ops);
2205 for (i = 0; i < req->r_num_ops; i++) {
2206 data_len += osd_req_encode_op(p, &req->r_ops[i]);
2207 p += sizeof(struct ceph_osd_op);
2208 }
2209
2210 ceph_encode_64(&p, req->r_snapid); /* snapid */
2211 if (req->r_snapc) {
2212 ceph_encode_64(&p, req->r_snapc->seq);
2213 ceph_encode_32(&p, req->r_snapc->num_snaps);
2214 for (i = 0; i < req->r_snapc->num_snaps; i++)
2215 ceph_encode_64(&p, req->r_snapc->snaps[i]);
2216 } else {
2217 ceph_encode_64(&p, 0); /* snap_seq */
2218 ceph_encode_32(&p, 0); /* snaps len */
2219 }
2220
2221 ceph_encode_32(&p, req->r_attempts); /* retry_attempt */
2222 BUG_ON(p > end - 8); /* space for features */
2223
2224 msg->hdr.version = cpu_to_le16(8); /* MOSDOp v8 */
2225 /* front_len is finalized in encode_request_finish() */
2226 msg->front.iov_len = p - msg->front.iov_base;
2227 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
2228 msg->hdr.data_len = cpu_to_le32(data_len);
2229 /*
2230 * The header "data_off" is a hint to the receiver allowing it
2231 * to align received data into its buffers such that there's no
2232 * need to re-copy it before writing it to disk (direct I/O).
2233 */
2234 msg->hdr.data_off = cpu_to_le16(req->r_data_offset);
2235
2236 dout("%s req %p msg %p oid %s oid_len %d\n", __func__, req, msg,
2237 req->r_t.target_oid.name, req->r_t.target_oid.name_len);
2238}
2239
2240static void encode_request_finish(struct ceph_msg *msg)
2241{
2242 void *p = msg->front.iov_base;
2243 void *const partial_end = p + msg->front.iov_len;
2244 void *const end = p + msg->front_alloc_len;
2245
2246 if (CEPH_HAVE_FEATURE(msg->con->peer_features, RESEND_ON_SPLIT)) {
2247 /* luminous OSD -- encode features and be done */
2248 p = partial_end;
2249 ceph_encode_64(&p, msg->con->peer_features);
2250 } else {
2251 struct {
2252 char spgid[CEPH_ENCODING_START_BLK_LEN +
2253 CEPH_PGID_ENCODING_LEN + 1];
2254 __le32 hash;
2255 __le32 epoch;
2256 __le32 flags;
2257 char reqid[CEPH_ENCODING_START_BLK_LEN +
2258 sizeof(struct ceph_osd_reqid)];
2259 char trace[sizeof(struct ceph_blkin_trace_info)];
2260 __le32 client_inc;
2261 struct ceph_timespec mtime;
2262 } __packed head;
2263 struct ceph_pg pgid;
2264 void *oloc, *oid, *tail;
2265 int oloc_len, oid_len, tail_len;
2266 int len;
2267
2268 /*
2269 * Pre-luminous OSD -- reencode v8 into v4 using @head
2270 * as a temporary buffer. Encode the raw PG; the rest
2271 * is just a matter of moving oloc, oid and tail blobs
2272 * around.
2273 */
2274 memcpy(&head, p, sizeof(head));
2275 p += sizeof(head);
2276
2277 oloc = p;
2278 p += CEPH_ENCODING_START_BLK_LEN;
2279 pgid.pool = ceph_decode_64(&p);
2280 p += 4 + 4; /* preferred, key len */
2281 len = ceph_decode_32(&p);
2282 p += len; /* nspace */
2283 oloc_len = p - oloc;
2284
2285 oid = p;
2286 len = ceph_decode_32(&p);
2287 p += len;
2288 oid_len = p - oid;
2289
2290 tail = p;
2291 tail_len = partial_end - p;
2292
2293 p = msg->front.iov_base;
2294 ceph_encode_copy(&p, &head.client_inc, sizeof(head.client_inc));
2295 ceph_encode_copy(&p, &head.epoch, sizeof(head.epoch));
2296 ceph_encode_copy(&p, &head.flags, sizeof(head.flags));
2297 ceph_encode_copy(&p, &head.mtime, sizeof(head.mtime));
2298
2299 /* reassert_version */
2300 memset(p, 0, sizeof(struct ceph_eversion));
2301 p += sizeof(struct ceph_eversion);
2302
2303 BUG_ON(p >= oloc);
2304 memmove(p, oloc, oloc_len);
2305 p += oloc_len;
2306
2307 pgid.seed = le32_to_cpu(head.hash);
2308 encode_pgid(&p, &pgid); /* raw pg */
2309
2310 BUG_ON(p >= oid);
2311 memmove(p, oid, oid_len);
2312 p += oid_len;
2313
2314 /* tail -- ops, snapid, snapc, retry_attempt */
2315 BUG_ON(p >= tail);
2316 memmove(p, tail, tail_len);
2317 p += tail_len;
2318
2319 msg->hdr.version = cpu_to_le16(4); /* MOSDOp v4 */
2320 }
2321
2322 BUG_ON(p > end);
2323 msg->front.iov_len = p - msg->front.iov_base;
2324 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
2325
2326 dout("%s msg %p tid %llu %u+%u+%u v%d\n", __func__, msg,
2327 le64_to_cpu(msg->hdr.tid), le32_to_cpu(msg->hdr.front_len),
2328 le32_to_cpu(msg->hdr.middle_len), le32_to_cpu(msg->hdr.data_len),
2329 le16_to_cpu(msg->hdr.version));
2330}
2331
2332/*
2333 * @req has to be assigned a tid and registered.
2334 */
2335static void send_request(struct ceph_osd_request *req)
2336{
2337 struct ceph_osd *osd = req->r_osd;
2338
2339 verify_osd_locked(osd);
2340 WARN_ON(osd->o_osd != req->r_t.osd);
2341
2342 /* backoff? */
2343 if (should_plug_request(req))
2344 return;
2345
2346 /*
2347 * We may have a previously queued request message hanging
2348 * around. Cancel it to avoid corrupting the msgr.
2349 */
2350 if (req->r_sent)
2351 ceph_msg_revoke(req->r_request);
2352
2353 req->r_flags |= CEPH_OSD_FLAG_KNOWN_REDIR;
2354 if (req->r_attempts)
2355 req->r_flags |= CEPH_OSD_FLAG_RETRY;
2356 else
2357 WARN_ON(req->r_flags & CEPH_OSD_FLAG_RETRY);
2358
2359 encode_request_partial(req, req->r_request);
2360
2361 dout("%s req %p tid %llu to pgid %llu.%x spgid %llu.%xs%d osd%d e%u flags 0x%x attempt %d\n",
2362 __func__, req, req->r_tid, req->r_t.pgid.pool, req->r_t.pgid.seed,
2363 req->r_t.spgid.pgid.pool, req->r_t.spgid.pgid.seed,
2364 req->r_t.spgid.shard, osd->o_osd, req->r_t.epoch, req->r_flags,
2365 req->r_attempts);
2366
2367 req->r_t.paused = false;
2368 req->r_stamp = jiffies;
2369 req->r_attempts++;
2370
2371 req->r_sent = osd->o_incarnation;
2372 req->r_request->hdr.tid = cpu_to_le64(req->r_tid);
2373 ceph_con_send(&osd->o_con, ceph_msg_get(req->r_request));
2374}
2375
2376static void maybe_request_map(struct ceph_osd_client *osdc)
2377{
2378 bool continuous = false;
2379
2380 verify_osdc_locked(osdc);
2381 WARN_ON(!osdc->osdmap->epoch);
2382
2383 if (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
2384 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD) ||
2385 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR)) {
2386 dout("%s osdc %p continuous\n", __func__, osdc);
2387 continuous = true;
2388 } else {
2389 dout("%s osdc %p onetime\n", __func__, osdc);
2390 }
2391
2392 if (ceph_monc_want_map(&osdc->client->monc, CEPH_SUB_OSDMAP,
2393 osdc->osdmap->epoch + 1, continuous))
2394 ceph_monc_renew_subs(&osdc->client->monc);
2395}
2396
2397static void complete_request(struct ceph_osd_request *req, int err);
2398static void send_map_check(struct ceph_osd_request *req);
2399
2400static void __submit_request(struct ceph_osd_request *req, bool wrlocked)
2401{
2402 struct ceph_osd_client *osdc = req->r_osdc;
2403 struct ceph_osd *osd;
2404 enum calc_target_result ct_res;
2405 int err = 0;
2406 bool need_send = false;
2407 bool promoted = false;
2408
2409 WARN_ON(req->r_tid);
2410 dout("%s req %p wrlocked %d\n", __func__, req, wrlocked);
2411
2412again:
2413 ct_res = calc_target(osdc, &req->r_t, false);
2414 if (ct_res == CALC_TARGET_POOL_DNE && !wrlocked)
2415 goto promote;
2416
2417 osd = lookup_create_osd(osdc, req->r_t.osd, wrlocked);
2418 if (IS_ERR(osd)) {
2419 WARN_ON(PTR_ERR(osd) != -EAGAIN || wrlocked);
2420 goto promote;
2421 }
2422
2423 if (osdc->abort_err) {
2424 dout("req %p abort_err %d\n", req, osdc->abort_err);
2425 err = osdc->abort_err;
2426 } else if (osdc->osdmap->epoch < osdc->epoch_barrier) {
2427 dout("req %p epoch %u barrier %u\n", req, osdc->osdmap->epoch,
2428 osdc->epoch_barrier);
2429 req->r_t.paused = true;
2430 maybe_request_map(osdc);
2431 } else if ((req->r_flags & CEPH_OSD_FLAG_WRITE) &&
2432 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR)) {
2433 dout("req %p pausewr\n", req);
2434 req->r_t.paused = true;
2435 maybe_request_map(osdc);
2436 } else if ((req->r_flags & CEPH_OSD_FLAG_READ) &&
2437 ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD)) {
2438 dout("req %p pauserd\n", req);
2439 req->r_t.paused = true;
2440 maybe_request_map(osdc);
2441 } else if ((req->r_flags & CEPH_OSD_FLAG_WRITE) &&
2442 !(req->r_flags & (CEPH_OSD_FLAG_FULL_TRY |
2443 CEPH_OSD_FLAG_FULL_FORCE)) &&
2444 (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
2445 pool_full(osdc, req->r_t.base_oloc.pool))) {
2446 dout("req %p full/pool_full\n", req);
2447 if (ceph_test_opt(osdc->client, ABORT_ON_FULL)) {
2448 err = -ENOSPC;
2449 } else {
2450 if (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL))
2451 pr_warn_ratelimited("cluster is full (osdmap FULL)\n");
2452 else
2453 pr_warn_ratelimited("pool %lld is full or reached quota\n",
2454 req->r_t.base_oloc.pool);
2455 req->r_t.paused = true;
2456 maybe_request_map(osdc);
2457 }
2458 } else if (!osd_homeless(osd)) {
2459 need_send = true;
2460 } else {
2461 maybe_request_map(osdc);
2462 }
2463
2464 mutex_lock(&osd->lock);
2465 /*
2466 * Assign the tid atomically with send_request() to protect
2467 * multiple writes to the same object from racing with each
2468 * other, resulting in out of order ops on the OSDs.
2469 */
2470 req->r_tid = atomic64_inc_return(&osdc->last_tid);
2471 link_request(osd, req);
2472 if (need_send)
2473 send_request(req);
2474 else if (err)
2475 complete_request(req, err);
2476 mutex_unlock(&osd->lock);
2477
2478 if (!err && ct_res == CALC_TARGET_POOL_DNE)
2479 send_map_check(req);
2480
2481 if (promoted)
2482 downgrade_write(&osdc->lock);
2483 return;
2484
2485promote:
2486 up_read(&osdc->lock);
2487 down_write(&osdc->lock);
2488 wrlocked = true;
2489 promoted = true;
2490 goto again;
2491}
2492
2493static void account_request(struct ceph_osd_request *req)
2494{
2495 WARN_ON(req->r_flags & (CEPH_OSD_FLAG_ACK | CEPH_OSD_FLAG_ONDISK));
2496 WARN_ON(!(req->r_flags & (CEPH_OSD_FLAG_READ | CEPH_OSD_FLAG_WRITE)));
2497
2498 req->r_flags |= CEPH_OSD_FLAG_ONDISK;
2499 atomic_inc(&req->r_osdc->num_requests);
2500
2501 req->r_start_stamp = jiffies;
2502 req->r_start_latency = ktime_get();
2503}
2504
2505static void submit_request(struct ceph_osd_request *req, bool wrlocked)
2506{
2507 ceph_osdc_get_request(req);
2508 account_request(req);
2509 __submit_request(req, wrlocked);
2510}
2511
2512static void finish_request(struct ceph_osd_request *req)
2513{
2514 struct ceph_osd_client *osdc = req->r_osdc;
2515
2516 WARN_ON(lookup_request_mc(&osdc->map_checks, req->r_tid));
2517 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
2518
2519 req->r_end_latency = ktime_get();
2520
2521 if (req->r_osd) {
2522 ceph_init_sparse_read(&req->r_osd->o_sparse_read);
2523 unlink_request(req->r_osd, req);
2524 }
2525 atomic_dec(&osdc->num_requests);
2526
2527 /*
2528 * If an OSD has failed or returned and a request has been sent
2529 * twice, it's possible to get a reply and end up here while the
2530 * request message is queued for delivery. We will ignore the
2531 * reply, so not a big deal, but better to try and catch it.
2532 */
2533 ceph_msg_revoke(req->r_request);
2534 ceph_msg_revoke_incoming(req->r_reply);
2535}
2536
2537static void __complete_request(struct ceph_osd_request *req)
2538{
2539 dout("%s req %p tid %llu cb %ps result %d\n", __func__, req,
2540 req->r_tid, req->r_callback, req->r_result);
2541
2542 if (req->r_callback)
2543 req->r_callback(req);
2544 complete_all(&req->r_completion);
2545 ceph_osdc_put_request(req);
2546}
2547
2548static void complete_request_workfn(struct work_struct *work)
2549{
2550 struct ceph_osd_request *req =
2551 container_of(work, struct ceph_osd_request, r_complete_work);
2552
2553 __complete_request(req);
2554}
2555
2556/*
2557 * This is open-coded in handle_reply().
2558 */
2559static void complete_request(struct ceph_osd_request *req, int err)
2560{
2561 dout("%s req %p tid %llu err %d\n", __func__, req, req->r_tid, err);
2562
2563 req->r_result = err;
2564 finish_request(req);
2565
2566 INIT_WORK(&req->r_complete_work, complete_request_workfn);
2567 queue_work(req->r_osdc->completion_wq, &req->r_complete_work);
2568}
2569
2570static void cancel_map_check(struct ceph_osd_request *req)
2571{
2572 struct ceph_osd_client *osdc = req->r_osdc;
2573 struct ceph_osd_request *lookup_req;
2574
2575 verify_osdc_wrlocked(osdc);
2576
2577 lookup_req = lookup_request_mc(&osdc->map_checks, req->r_tid);
2578 if (!lookup_req)
2579 return;
2580
2581 WARN_ON(lookup_req != req);
2582 erase_request_mc(&osdc->map_checks, req);
2583 ceph_osdc_put_request(req);
2584}
2585
2586static void cancel_request(struct ceph_osd_request *req)
2587{
2588 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
2589
2590 cancel_map_check(req);
2591 finish_request(req);
2592 complete_all(&req->r_completion);
2593 ceph_osdc_put_request(req);
2594}
2595
2596static void abort_request(struct ceph_osd_request *req, int err)
2597{
2598 dout("%s req %p tid %llu err %d\n", __func__, req, req->r_tid, err);
2599
2600 cancel_map_check(req);
2601 complete_request(req, err);
2602}
2603
2604static int abort_fn(struct ceph_osd_request *req, void *arg)
2605{
2606 int err = *(int *)arg;
2607
2608 abort_request(req, err);
2609 return 0; /* continue iteration */
2610}
2611
2612/*
2613 * Abort all in-flight requests with @err and arrange for all future
2614 * requests to be failed immediately.
2615 */
2616void ceph_osdc_abort_requests(struct ceph_osd_client *osdc, int err)
2617{
2618 dout("%s osdc %p err %d\n", __func__, osdc, err);
2619 down_write(&osdc->lock);
2620 for_each_request(osdc, abort_fn, &err);
2621 osdc->abort_err = err;
2622 up_write(&osdc->lock);
2623}
2624EXPORT_SYMBOL(ceph_osdc_abort_requests);
2625
2626void ceph_osdc_clear_abort_err(struct ceph_osd_client *osdc)
2627{
2628 down_write(&osdc->lock);
2629 osdc->abort_err = 0;
2630 up_write(&osdc->lock);
2631}
2632EXPORT_SYMBOL(ceph_osdc_clear_abort_err);
2633
2634static void update_epoch_barrier(struct ceph_osd_client *osdc, u32 eb)
2635{
2636 if (likely(eb > osdc->epoch_barrier)) {
2637 dout("updating epoch_barrier from %u to %u\n",
2638 osdc->epoch_barrier, eb);
2639 osdc->epoch_barrier = eb;
2640 /* Request map if we're not to the barrier yet */
2641 if (eb > osdc->osdmap->epoch)
2642 maybe_request_map(osdc);
2643 }
2644}
2645
2646void ceph_osdc_update_epoch_barrier(struct ceph_osd_client *osdc, u32 eb)
2647{
2648 down_read(&osdc->lock);
2649 if (unlikely(eb > osdc->epoch_barrier)) {
2650 up_read(&osdc->lock);
2651 down_write(&osdc->lock);
2652 update_epoch_barrier(osdc, eb);
2653 up_write(&osdc->lock);
2654 } else {
2655 up_read(&osdc->lock);
2656 }
2657}
2658EXPORT_SYMBOL(ceph_osdc_update_epoch_barrier);
2659
2660/*
2661 * We can end up releasing caps as a result of abort_request().
2662 * In that case, we probably want to ensure that the cap release message
2663 * has an updated epoch barrier in it, so set the epoch barrier prior to
2664 * aborting the first request.
2665 */
2666static int abort_on_full_fn(struct ceph_osd_request *req, void *arg)
2667{
2668 struct ceph_osd_client *osdc = req->r_osdc;
2669 bool *victims = arg;
2670
2671 if ((req->r_flags & CEPH_OSD_FLAG_WRITE) &&
2672 (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
2673 pool_full(osdc, req->r_t.base_oloc.pool))) {
2674 if (!*victims) {
2675 update_epoch_barrier(osdc, osdc->osdmap->epoch);
2676 *victims = true;
2677 }
2678 abort_request(req, -ENOSPC);
2679 }
2680
2681 return 0; /* continue iteration */
2682}
2683
2684/*
2685 * Drop all pending requests that are stalled waiting on a full condition to
2686 * clear, and complete them with ENOSPC as the return code. Set the
2687 * osdc->epoch_barrier to the latest map epoch that we've seen if any were
2688 * cancelled.
2689 */
2690static void ceph_osdc_abort_on_full(struct ceph_osd_client *osdc)
2691{
2692 bool victims = false;
2693
2694 if (ceph_test_opt(osdc->client, ABORT_ON_FULL) &&
2695 (ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) || have_pool_full(osdc)))
2696 for_each_request(osdc, abort_on_full_fn, &victims);
2697}
2698
2699static void check_pool_dne(struct ceph_osd_request *req)
2700{
2701 struct ceph_osd_client *osdc = req->r_osdc;
2702 struct ceph_osdmap *map = osdc->osdmap;
2703
2704 verify_osdc_wrlocked(osdc);
2705 WARN_ON(!map->epoch);
2706
2707 if (req->r_attempts) {
2708 /*
2709 * We sent a request earlier, which means that
2710 * previously the pool existed, and now it does not
2711 * (i.e., it was deleted).
2712 */
2713 req->r_map_dne_bound = map->epoch;
2714 dout("%s req %p tid %llu pool disappeared\n", __func__, req,
2715 req->r_tid);
2716 } else {
2717 dout("%s req %p tid %llu map_dne_bound %u have %u\n", __func__,
2718 req, req->r_tid, req->r_map_dne_bound, map->epoch);
2719 }
2720
2721 if (req->r_map_dne_bound) {
2722 if (map->epoch >= req->r_map_dne_bound) {
2723 /* we had a new enough map */
2724 pr_info_ratelimited("tid %llu pool does not exist\n",
2725 req->r_tid);
2726 complete_request(req, -ENOENT);
2727 }
2728 } else {
2729 send_map_check(req);
2730 }
2731}
2732
2733static void map_check_cb(struct ceph_mon_generic_request *greq)
2734{
2735 struct ceph_osd_client *osdc = &greq->monc->client->osdc;
2736 struct ceph_osd_request *req;
2737 u64 tid = greq->private_data;
2738
2739 WARN_ON(greq->result || !greq->u.newest);
2740
2741 down_write(&osdc->lock);
2742 req = lookup_request_mc(&osdc->map_checks, tid);
2743 if (!req) {
2744 dout("%s tid %llu dne\n", __func__, tid);
2745 goto out_unlock;
2746 }
2747
2748 dout("%s req %p tid %llu map_dne_bound %u newest %llu\n", __func__,
2749 req, req->r_tid, req->r_map_dne_bound, greq->u.newest);
2750 if (!req->r_map_dne_bound)
2751 req->r_map_dne_bound = greq->u.newest;
2752 erase_request_mc(&osdc->map_checks, req);
2753 check_pool_dne(req);
2754
2755 ceph_osdc_put_request(req);
2756out_unlock:
2757 up_write(&osdc->lock);
2758}
2759
2760static void send_map_check(struct ceph_osd_request *req)
2761{
2762 struct ceph_osd_client *osdc = req->r_osdc;
2763 struct ceph_osd_request *lookup_req;
2764 int ret;
2765
2766 verify_osdc_wrlocked(osdc);
2767
2768 lookup_req = lookup_request_mc(&osdc->map_checks, req->r_tid);
2769 if (lookup_req) {
2770 WARN_ON(lookup_req != req);
2771 return;
2772 }
2773
2774 ceph_osdc_get_request(req);
2775 insert_request_mc(&osdc->map_checks, req);
2776 ret = ceph_monc_get_version_async(&osdc->client->monc, "osdmap",
2777 map_check_cb, req->r_tid);
2778 WARN_ON(ret);
2779}
2780
2781/*
2782 * lingering requests, watch/notify v2 infrastructure
2783 */
2784static void linger_release(struct kref *kref)
2785{
2786 struct ceph_osd_linger_request *lreq =
2787 container_of(kref, struct ceph_osd_linger_request, kref);
2788
2789 dout("%s lreq %p reg_req %p ping_req %p\n", __func__, lreq,
2790 lreq->reg_req, lreq->ping_req);
2791 WARN_ON(!RB_EMPTY_NODE(&lreq->node));
2792 WARN_ON(!RB_EMPTY_NODE(&lreq->osdc_node));
2793 WARN_ON(!RB_EMPTY_NODE(&lreq->mc_node));
2794 WARN_ON(!list_empty(&lreq->scan_item));
2795 WARN_ON(!list_empty(&lreq->pending_lworks));
2796 WARN_ON(lreq->osd);
2797
2798 if (lreq->request_pl)
2799 ceph_pagelist_release(lreq->request_pl);
2800 if (lreq->notify_id_pages)
2801 ceph_release_page_vector(lreq->notify_id_pages, 1);
2802
2803 ceph_osdc_put_request(lreq->reg_req);
2804 ceph_osdc_put_request(lreq->ping_req);
2805 target_destroy(&lreq->t);
2806 kfree(lreq);
2807}
2808
2809static void linger_put(struct ceph_osd_linger_request *lreq)
2810{
2811 if (lreq)
2812 kref_put(&lreq->kref, linger_release);
2813}
2814
2815static struct ceph_osd_linger_request *
2816linger_get(struct ceph_osd_linger_request *lreq)
2817{
2818 kref_get(&lreq->kref);
2819 return lreq;
2820}
2821
2822static struct ceph_osd_linger_request *
2823linger_alloc(struct ceph_osd_client *osdc)
2824{
2825 struct ceph_osd_linger_request *lreq;
2826
2827 lreq = kzalloc(sizeof(*lreq), GFP_NOIO);
2828 if (!lreq)
2829 return NULL;
2830
2831 kref_init(&lreq->kref);
2832 mutex_init(&lreq->lock);
2833 RB_CLEAR_NODE(&lreq->node);
2834 RB_CLEAR_NODE(&lreq->osdc_node);
2835 RB_CLEAR_NODE(&lreq->mc_node);
2836 INIT_LIST_HEAD(&lreq->scan_item);
2837 INIT_LIST_HEAD(&lreq->pending_lworks);
2838 init_completion(&lreq->reg_commit_wait);
2839 init_completion(&lreq->notify_finish_wait);
2840
2841 lreq->osdc = osdc;
2842 target_init(&lreq->t);
2843
2844 dout("%s lreq %p\n", __func__, lreq);
2845 return lreq;
2846}
2847
2848DEFINE_RB_INSDEL_FUNCS(linger, struct ceph_osd_linger_request, linger_id, node)
2849DEFINE_RB_FUNCS(linger_osdc, struct ceph_osd_linger_request, linger_id, osdc_node)
2850DEFINE_RB_FUNCS(linger_mc, struct ceph_osd_linger_request, linger_id, mc_node)
2851
2852/*
2853 * Create linger request <-> OSD session relation.
2854 *
2855 * @lreq has to be registered, @osd may be homeless.
2856 */
2857static void link_linger(struct ceph_osd *osd,
2858 struct ceph_osd_linger_request *lreq)
2859{
2860 verify_osd_locked(osd);
2861 WARN_ON(!lreq->linger_id || lreq->osd);
2862 dout("%s osd %p osd%d lreq %p linger_id %llu\n", __func__, osd,
2863 osd->o_osd, lreq, lreq->linger_id);
2864
2865 if (!osd_homeless(osd))
2866 __remove_osd_from_lru(osd);
2867 else
2868 atomic_inc(&osd->o_osdc->num_homeless);
2869
2870 get_osd(osd);
2871 insert_linger(&osd->o_linger_requests, lreq);
2872 lreq->osd = osd;
2873}
2874
2875static void unlink_linger(struct ceph_osd *osd,
2876 struct ceph_osd_linger_request *lreq)
2877{
2878 verify_osd_locked(osd);
2879 WARN_ON(lreq->osd != osd);
2880 dout("%s osd %p osd%d lreq %p linger_id %llu\n", __func__, osd,
2881 osd->o_osd, lreq, lreq->linger_id);
2882
2883 lreq->osd = NULL;
2884 erase_linger(&osd->o_linger_requests, lreq);
2885 put_osd(osd);
2886
2887 if (!osd_homeless(osd))
2888 maybe_move_osd_to_lru(osd);
2889 else
2890 atomic_dec(&osd->o_osdc->num_homeless);
2891}
2892
2893static bool __linger_registered(struct ceph_osd_linger_request *lreq)
2894{
2895 verify_osdc_locked(lreq->osdc);
2896
2897 return !RB_EMPTY_NODE(&lreq->osdc_node);
2898}
2899
2900static bool linger_registered(struct ceph_osd_linger_request *lreq)
2901{
2902 struct ceph_osd_client *osdc = lreq->osdc;
2903 bool registered;
2904
2905 down_read(&osdc->lock);
2906 registered = __linger_registered(lreq);
2907 up_read(&osdc->lock);
2908
2909 return registered;
2910}
2911
2912static void linger_register(struct ceph_osd_linger_request *lreq)
2913{
2914 struct ceph_osd_client *osdc = lreq->osdc;
2915
2916 verify_osdc_wrlocked(osdc);
2917 WARN_ON(lreq->linger_id);
2918
2919 linger_get(lreq);
2920 lreq->linger_id = ++osdc->last_linger_id;
2921 insert_linger_osdc(&osdc->linger_requests, lreq);
2922}
2923
2924static void linger_unregister(struct ceph_osd_linger_request *lreq)
2925{
2926 struct ceph_osd_client *osdc = lreq->osdc;
2927
2928 verify_osdc_wrlocked(osdc);
2929
2930 erase_linger_osdc(&osdc->linger_requests, lreq);
2931 linger_put(lreq);
2932}
2933
2934static void cancel_linger_request(struct ceph_osd_request *req)
2935{
2936 struct ceph_osd_linger_request *lreq = req->r_priv;
2937
2938 WARN_ON(!req->r_linger);
2939 cancel_request(req);
2940 linger_put(lreq);
2941}
2942
2943struct linger_work {
2944 struct work_struct work;
2945 struct ceph_osd_linger_request *lreq;
2946 struct list_head pending_item;
2947 unsigned long queued_stamp;
2948
2949 union {
2950 struct {
2951 u64 notify_id;
2952 u64 notifier_id;
2953 void *payload; /* points into @msg front */
2954 size_t payload_len;
2955
2956 struct ceph_msg *msg; /* for ceph_msg_put() */
2957 } notify;
2958 struct {
2959 int err;
2960 } error;
2961 };
2962};
2963
2964static struct linger_work *lwork_alloc(struct ceph_osd_linger_request *lreq,
2965 work_func_t workfn)
2966{
2967 struct linger_work *lwork;
2968
2969 lwork = kzalloc(sizeof(*lwork), GFP_NOIO);
2970 if (!lwork)
2971 return NULL;
2972
2973 INIT_WORK(&lwork->work, workfn);
2974 INIT_LIST_HEAD(&lwork->pending_item);
2975 lwork->lreq = linger_get(lreq);
2976
2977 return lwork;
2978}
2979
2980static void lwork_free(struct linger_work *lwork)
2981{
2982 struct ceph_osd_linger_request *lreq = lwork->lreq;
2983
2984 mutex_lock(&lreq->lock);
2985 list_del(&lwork->pending_item);
2986 mutex_unlock(&lreq->lock);
2987
2988 linger_put(lreq);
2989 kfree(lwork);
2990}
2991
2992static void lwork_queue(struct linger_work *lwork)
2993{
2994 struct ceph_osd_linger_request *lreq = lwork->lreq;
2995 struct ceph_osd_client *osdc = lreq->osdc;
2996
2997 verify_lreq_locked(lreq);
2998 WARN_ON(!list_empty(&lwork->pending_item));
2999
3000 lwork->queued_stamp = jiffies;
3001 list_add_tail(&lwork->pending_item, &lreq->pending_lworks);
3002 queue_work(osdc->notify_wq, &lwork->work);
3003}
3004
3005static void do_watch_notify(struct work_struct *w)
3006{
3007 struct linger_work *lwork = container_of(w, struct linger_work, work);
3008 struct ceph_osd_linger_request *lreq = lwork->lreq;
3009
3010 if (!linger_registered(lreq)) {
3011 dout("%s lreq %p not registered\n", __func__, lreq);
3012 goto out;
3013 }
3014
3015 WARN_ON(!lreq->is_watch);
3016 dout("%s lreq %p notify_id %llu notifier_id %llu payload_len %zu\n",
3017 __func__, lreq, lwork->notify.notify_id, lwork->notify.notifier_id,
3018 lwork->notify.payload_len);
3019 lreq->wcb(lreq->data, lwork->notify.notify_id, lreq->linger_id,
3020 lwork->notify.notifier_id, lwork->notify.payload,
3021 lwork->notify.payload_len);
3022
3023out:
3024 ceph_msg_put(lwork->notify.msg);
3025 lwork_free(lwork);
3026}
3027
3028static void do_watch_error(struct work_struct *w)
3029{
3030 struct linger_work *lwork = container_of(w, struct linger_work, work);
3031 struct ceph_osd_linger_request *lreq = lwork->lreq;
3032
3033 if (!linger_registered(lreq)) {
3034 dout("%s lreq %p not registered\n", __func__, lreq);
3035 goto out;
3036 }
3037
3038 dout("%s lreq %p err %d\n", __func__, lreq, lwork->error.err);
3039 lreq->errcb(lreq->data, lreq->linger_id, lwork->error.err);
3040
3041out:
3042 lwork_free(lwork);
3043}
3044
3045static void queue_watch_error(struct ceph_osd_linger_request *lreq)
3046{
3047 struct linger_work *lwork;
3048
3049 lwork = lwork_alloc(lreq, do_watch_error);
3050 if (!lwork) {
3051 pr_err("failed to allocate error-lwork\n");
3052 return;
3053 }
3054
3055 lwork->error.err = lreq->last_error;
3056 lwork_queue(lwork);
3057}
3058
3059static void linger_reg_commit_complete(struct ceph_osd_linger_request *lreq,
3060 int result)
3061{
3062 if (!completion_done(&lreq->reg_commit_wait)) {
3063 lreq->reg_commit_error = (result <= 0 ? result : 0);
3064 complete_all(&lreq->reg_commit_wait);
3065 }
3066}
3067
3068static void linger_commit_cb(struct ceph_osd_request *req)
3069{
3070 struct ceph_osd_linger_request *lreq = req->r_priv;
3071
3072 mutex_lock(&lreq->lock);
3073 if (req != lreq->reg_req) {
3074 dout("%s lreq %p linger_id %llu unknown req (%p != %p)\n",
3075 __func__, lreq, lreq->linger_id, req, lreq->reg_req);
3076 goto out;
3077 }
3078
3079 dout("%s lreq %p linger_id %llu result %d\n", __func__, lreq,
3080 lreq->linger_id, req->r_result);
3081 linger_reg_commit_complete(lreq, req->r_result);
3082 lreq->committed = true;
3083
3084 if (!lreq->is_watch) {
3085 struct ceph_osd_data *osd_data =
3086 osd_req_op_data(req, 0, notify, response_data);
3087 void *p = page_address(osd_data->pages[0]);
3088
3089 WARN_ON(req->r_ops[0].op != CEPH_OSD_OP_NOTIFY ||
3090 osd_data->type != CEPH_OSD_DATA_TYPE_PAGES);
3091
3092 /* make note of the notify_id */
3093 if (req->r_ops[0].outdata_len >= sizeof(u64)) {
3094 lreq->notify_id = ceph_decode_64(&p);
3095 dout("lreq %p notify_id %llu\n", lreq,
3096 lreq->notify_id);
3097 } else {
3098 dout("lreq %p no notify_id\n", lreq);
3099 }
3100 }
3101
3102out:
3103 mutex_unlock(&lreq->lock);
3104 linger_put(lreq);
3105}
3106
3107static int normalize_watch_error(int err)
3108{
3109 /*
3110 * Translate ENOENT -> ENOTCONN so that a delete->disconnection
3111 * notification and a failure to reconnect because we raced with
3112 * the delete appear the same to the user.
3113 */
3114 if (err == -ENOENT)
3115 err = -ENOTCONN;
3116
3117 return err;
3118}
3119
3120static void linger_reconnect_cb(struct ceph_osd_request *req)
3121{
3122 struct ceph_osd_linger_request *lreq = req->r_priv;
3123
3124 mutex_lock(&lreq->lock);
3125 if (req != lreq->reg_req) {
3126 dout("%s lreq %p linger_id %llu unknown req (%p != %p)\n",
3127 __func__, lreq, lreq->linger_id, req, lreq->reg_req);
3128 goto out;
3129 }
3130
3131 dout("%s lreq %p linger_id %llu result %d last_error %d\n", __func__,
3132 lreq, lreq->linger_id, req->r_result, lreq->last_error);
3133 if (req->r_result < 0) {
3134 if (!lreq->last_error) {
3135 lreq->last_error = normalize_watch_error(req->r_result);
3136 queue_watch_error(lreq);
3137 }
3138 }
3139
3140out:
3141 mutex_unlock(&lreq->lock);
3142 linger_put(lreq);
3143}
3144
3145static void send_linger(struct ceph_osd_linger_request *lreq)
3146{
3147 struct ceph_osd_client *osdc = lreq->osdc;
3148 struct ceph_osd_request *req;
3149 int ret;
3150
3151 verify_osdc_wrlocked(osdc);
3152 mutex_lock(&lreq->lock);
3153 dout("%s lreq %p linger_id %llu\n", __func__, lreq, lreq->linger_id);
3154
3155 if (lreq->reg_req) {
3156 if (lreq->reg_req->r_osd)
3157 cancel_linger_request(lreq->reg_req);
3158 ceph_osdc_put_request(lreq->reg_req);
3159 }
3160
3161 req = ceph_osdc_alloc_request(osdc, NULL, 1, true, GFP_NOIO);
3162 BUG_ON(!req);
3163
3164 target_copy(&req->r_t, &lreq->t);
3165 req->r_mtime = lreq->mtime;
3166
3167 if (lreq->is_watch && lreq->committed) {
3168 osd_req_op_watch_init(req, 0, CEPH_OSD_WATCH_OP_RECONNECT,
3169 lreq->linger_id, ++lreq->register_gen);
3170 dout("lreq %p reconnect register_gen %u\n", lreq,
3171 req->r_ops[0].watch.gen);
3172 req->r_callback = linger_reconnect_cb;
3173 } else {
3174 if (lreq->is_watch) {
3175 osd_req_op_watch_init(req, 0, CEPH_OSD_WATCH_OP_WATCH,
3176 lreq->linger_id, 0);
3177 } else {
3178 lreq->notify_id = 0;
3179
3180 refcount_inc(&lreq->request_pl->refcnt);
3181 osd_req_op_notify_init(req, 0, lreq->linger_id,
3182 lreq->request_pl);
3183 ceph_osd_data_pages_init(
3184 osd_req_op_data(req, 0, notify, response_data),
3185 lreq->notify_id_pages, PAGE_SIZE, 0, false, false);
3186 }
3187 dout("lreq %p register\n", lreq);
3188 req->r_callback = linger_commit_cb;
3189 }
3190
3191 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
3192 BUG_ON(ret);
3193
3194 req->r_priv = linger_get(lreq);
3195 req->r_linger = true;
3196 lreq->reg_req = req;
3197 mutex_unlock(&lreq->lock);
3198
3199 submit_request(req, true);
3200}
3201
3202static void linger_ping_cb(struct ceph_osd_request *req)
3203{
3204 struct ceph_osd_linger_request *lreq = req->r_priv;
3205
3206 mutex_lock(&lreq->lock);
3207 if (req != lreq->ping_req) {
3208 dout("%s lreq %p linger_id %llu unknown req (%p != %p)\n",
3209 __func__, lreq, lreq->linger_id, req, lreq->ping_req);
3210 goto out;
3211 }
3212
3213 dout("%s lreq %p linger_id %llu result %d ping_sent %lu last_error %d\n",
3214 __func__, lreq, lreq->linger_id, req->r_result, lreq->ping_sent,
3215 lreq->last_error);
3216 if (lreq->register_gen == req->r_ops[0].watch.gen) {
3217 if (!req->r_result) {
3218 lreq->watch_valid_thru = lreq->ping_sent;
3219 } else if (!lreq->last_error) {
3220 lreq->last_error = normalize_watch_error(req->r_result);
3221 queue_watch_error(lreq);
3222 }
3223 } else {
3224 dout("lreq %p register_gen %u ignoring old pong %u\n", lreq,
3225 lreq->register_gen, req->r_ops[0].watch.gen);
3226 }
3227
3228out:
3229 mutex_unlock(&lreq->lock);
3230 linger_put(lreq);
3231}
3232
3233static void send_linger_ping(struct ceph_osd_linger_request *lreq)
3234{
3235 struct ceph_osd_client *osdc = lreq->osdc;
3236 struct ceph_osd_request *req;
3237 int ret;
3238
3239 if (ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD)) {
3240 dout("%s PAUSERD\n", __func__);
3241 return;
3242 }
3243
3244 lreq->ping_sent = jiffies;
3245 dout("%s lreq %p linger_id %llu ping_sent %lu register_gen %u\n",
3246 __func__, lreq, lreq->linger_id, lreq->ping_sent,
3247 lreq->register_gen);
3248
3249 if (lreq->ping_req) {
3250 if (lreq->ping_req->r_osd)
3251 cancel_linger_request(lreq->ping_req);
3252 ceph_osdc_put_request(lreq->ping_req);
3253 }
3254
3255 req = ceph_osdc_alloc_request(osdc, NULL, 1, true, GFP_NOIO);
3256 BUG_ON(!req);
3257
3258 target_copy(&req->r_t, &lreq->t);
3259 osd_req_op_watch_init(req, 0, CEPH_OSD_WATCH_OP_PING, lreq->linger_id,
3260 lreq->register_gen);
3261 req->r_callback = linger_ping_cb;
3262
3263 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
3264 BUG_ON(ret);
3265
3266 req->r_priv = linger_get(lreq);
3267 req->r_linger = true;
3268 lreq->ping_req = req;
3269
3270 ceph_osdc_get_request(req);
3271 account_request(req);
3272 req->r_tid = atomic64_inc_return(&osdc->last_tid);
3273 link_request(lreq->osd, req);
3274 send_request(req);
3275}
3276
3277static void linger_submit(struct ceph_osd_linger_request *lreq)
3278{
3279 struct ceph_osd_client *osdc = lreq->osdc;
3280 struct ceph_osd *osd;
3281
3282 down_write(&osdc->lock);
3283 linger_register(lreq);
3284
3285 calc_target(osdc, &lreq->t, false);
3286 osd = lookup_create_osd(osdc, lreq->t.osd, true);
3287 link_linger(osd, lreq);
3288
3289 send_linger(lreq);
3290 up_write(&osdc->lock);
3291}
3292
3293static void cancel_linger_map_check(struct ceph_osd_linger_request *lreq)
3294{
3295 struct ceph_osd_client *osdc = lreq->osdc;
3296 struct ceph_osd_linger_request *lookup_lreq;
3297
3298 verify_osdc_wrlocked(osdc);
3299
3300 lookup_lreq = lookup_linger_mc(&osdc->linger_map_checks,
3301 lreq->linger_id);
3302 if (!lookup_lreq)
3303 return;
3304
3305 WARN_ON(lookup_lreq != lreq);
3306 erase_linger_mc(&osdc->linger_map_checks, lreq);
3307 linger_put(lreq);
3308}
3309
3310/*
3311 * @lreq has to be both registered and linked.
3312 */
3313static void __linger_cancel(struct ceph_osd_linger_request *lreq)
3314{
3315 if (lreq->ping_req && lreq->ping_req->r_osd)
3316 cancel_linger_request(lreq->ping_req);
3317 if (lreq->reg_req && lreq->reg_req->r_osd)
3318 cancel_linger_request(lreq->reg_req);
3319 cancel_linger_map_check(lreq);
3320 unlink_linger(lreq->osd, lreq);
3321 linger_unregister(lreq);
3322}
3323
3324static void linger_cancel(struct ceph_osd_linger_request *lreq)
3325{
3326 struct ceph_osd_client *osdc = lreq->osdc;
3327
3328 down_write(&osdc->lock);
3329 if (__linger_registered(lreq))
3330 __linger_cancel(lreq);
3331 up_write(&osdc->lock);
3332}
3333
3334static void send_linger_map_check(struct ceph_osd_linger_request *lreq);
3335
3336static void check_linger_pool_dne(struct ceph_osd_linger_request *lreq)
3337{
3338 struct ceph_osd_client *osdc = lreq->osdc;
3339 struct ceph_osdmap *map = osdc->osdmap;
3340
3341 verify_osdc_wrlocked(osdc);
3342 WARN_ON(!map->epoch);
3343
3344 if (lreq->register_gen) {
3345 lreq->map_dne_bound = map->epoch;
3346 dout("%s lreq %p linger_id %llu pool disappeared\n", __func__,
3347 lreq, lreq->linger_id);
3348 } else {
3349 dout("%s lreq %p linger_id %llu map_dne_bound %u have %u\n",
3350 __func__, lreq, lreq->linger_id, lreq->map_dne_bound,
3351 map->epoch);
3352 }
3353
3354 if (lreq->map_dne_bound) {
3355 if (map->epoch >= lreq->map_dne_bound) {
3356 /* we had a new enough map */
3357 pr_info("linger_id %llu pool does not exist\n",
3358 lreq->linger_id);
3359 linger_reg_commit_complete(lreq, -ENOENT);
3360 __linger_cancel(lreq);
3361 }
3362 } else {
3363 send_linger_map_check(lreq);
3364 }
3365}
3366
3367static void linger_map_check_cb(struct ceph_mon_generic_request *greq)
3368{
3369 struct ceph_osd_client *osdc = &greq->monc->client->osdc;
3370 struct ceph_osd_linger_request *lreq;
3371 u64 linger_id = greq->private_data;
3372
3373 WARN_ON(greq->result || !greq->u.newest);
3374
3375 down_write(&osdc->lock);
3376 lreq = lookup_linger_mc(&osdc->linger_map_checks, linger_id);
3377 if (!lreq) {
3378 dout("%s linger_id %llu dne\n", __func__, linger_id);
3379 goto out_unlock;
3380 }
3381
3382 dout("%s lreq %p linger_id %llu map_dne_bound %u newest %llu\n",
3383 __func__, lreq, lreq->linger_id, lreq->map_dne_bound,
3384 greq->u.newest);
3385 if (!lreq->map_dne_bound)
3386 lreq->map_dne_bound = greq->u.newest;
3387 erase_linger_mc(&osdc->linger_map_checks, lreq);
3388 check_linger_pool_dne(lreq);
3389
3390 linger_put(lreq);
3391out_unlock:
3392 up_write(&osdc->lock);
3393}
3394
3395static void send_linger_map_check(struct ceph_osd_linger_request *lreq)
3396{
3397 struct ceph_osd_client *osdc = lreq->osdc;
3398 struct ceph_osd_linger_request *lookup_lreq;
3399 int ret;
3400
3401 verify_osdc_wrlocked(osdc);
3402
3403 lookup_lreq = lookup_linger_mc(&osdc->linger_map_checks,
3404 lreq->linger_id);
3405 if (lookup_lreq) {
3406 WARN_ON(lookup_lreq != lreq);
3407 return;
3408 }
3409
3410 linger_get(lreq);
3411 insert_linger_mc(&osdc->linger_map_checks, lreq);
3412 ret = ceph_monc_get_version_async(&osdc->client->monc, "osdmap",
3413 linger_map_check_cb, lreq->linger_id);
3414 WARN_ON(ret);
3415}
3416
3417static int linger_reg_commit_wait(struct ceph_osd_linger_request *lreq)
3418{
3419 int ret;
3420
3421 dout("%s lreq %p linger_id %llu\n", __func__, lreq, lreq->linger_id);
3422 ret = wait_for_completion_killable(&lreq->reg_commit_wait);
3423 return ret ?: lreq->reg_commit_error;
3424}
3425
3426static int linger_notify_finish_wait(struct ceph_osd_linger_request *lreq,
3427 unsigned long timeout)
3428{
3429 long left;
3430
3431 dout("%s lreq %p linger_id %llu\n", __func__, lreq, lreq->linger_id);
3432 left = wait_for_completion_killable_timeout(&lreq->notify_finish_wait,
3433 ceph_timeout_jiffies(timeout));
3434 if (left <= 0)
3435 left = left ?: -ETIMEDOUT;
3436 else
3437 left = lreq->notify_finish_error; /* completed */
3438
3439 return left;
3440}
3441
3442/*
3443 * Timeout callback, called every N seconds. When 1 or more OSD
3444 * requests has been active for more than N seconds, we send a keepalive
3445 * (tag + timestamp) to its OSD to ensure any communications channel
3446 * reset is detected.
3447 */
3448static void handle_timeout(struct work_struct *work)
3449{
3450 struct ceph_osd_client *osdc =
3451 container_of(work, struct ceph_osd_client, timeout_work.work);
3452 struct ceph_options *opts = osdc->client->options;
3453 unsigned long cutoff = jiffies - opts->osd_keepalive_timeout;
3454 unsigned long expiry_cutoff = jiffies - opts->osd_request_timeout;
3455 LIST_HEAD(slow_osds);
3456 struct rb_node *n, *p;
3457
3458 dout("%s osdc %p\n", __func__, osdc);
3459 down_write(&osdc->lock);
3460
3461 /*
3462 * ping osds that are a bit slow. this ensures that if there
3463 * is a break in the TCP connection we will notice, and reopen
3464 * a connection with that osd (from the fault callback).
3465 */
3466 for (n = rb_first(&osdc->osds); n; n = rb_next(n)) {
3467 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
3468 bool found = false;
3469
3470 for (p = rb_first(&osd->o_requests); p; ) {
3471 struct ceph_osd_request *req =
3472 rb_entry(p, struct ceph_osd_request, r_node);
3473
3474 p = rb_next(p); /* abort_request() */
3475
3476 if (time_before(req->r_stamp, cutoff)) {
3477 dout(" req %p tid %llu on osd%d is laggy\n",
3478 req, req->r_tid, osd->o_osd);
3479 found = true;
3480 }
3481 if (opts->osd_request_timeout &&
3482 time_before(req->r_start_stamp, expiry_cutoff)) {
3483 pr_err_ratelimited("tid %llu on osd%d timeout\n",
3484 req->r_tid, osd->o_osd);
3485 abort_request(req, -ETIMEDOUT);
3486 }
3487 }
3488 for (p = rb_first(&osd->o_linger_requests); p; p = rb_next(p)) {
3489 struct ceph_osd_linger_request *lreq =
3490 rb_entry(p, struct ceph_osd_linger_request, node);
3491
3492 dout(" lreq %p linger_id %llu is served by osd%d\n",
3493 lreq, lreq->linger_id, osd->o_osd);
3494 found = true;
3495
3496 mutex_lock(&lreq->lock);
3497 if (lreq->is_watch && lreq->committed && !lreq->last_error)
3498 send_linger_ping(lreq);
3499 mutex_unlock(&lreq->lock);
3500 }
3501
3502 if (found)
3503 list_move_tail(&osd->o_keepalive_item, &slow_osds);
3504 }
3505
3506 if (opts->osd_request_timeout) {
3507 for (p = rb_first(&osdc->homeless_osd.o_requests); p; ) {
3508 struct ceph_osd_request *req =
3509 rb_entry(p, struct ceph_osd_request, r_node);
3510
3511 p = rb_next(p); /* abort_request() */
3512
3513 if (time_before(req->r_start_stamp, expiry_cutoff)) {
3514 pr_err_ratelimited("tid %llu on osd%d timeout\n",
3515 req->r_tid, osdc->homeless_osd.o_osd);
3516 abort_request(req, -ETIMEDOUT);
3517 }
3518 }
3519 }
3520
3521 if (atomic_read(&osdc->num_homeless) || !list_empty(&slow_osds))
3522 maybe_request_map(osdc);
3523
3524 while (!list_empty(&slow_osds)) {
3525 struct ceph_osd *osd = list_first_entry(&slow_osds,
3526 struct ceph_osd,
3527 o_keepalive_item);
3528 list_del_init(&osd->o_keepalive_item);
3529 ceph_con_keepalive(&osd->o_con);
3530 }
3531
3532 up_write(&osdc->lock);
3533 schedule_delayed_work(&osdc->timeout_work,
3534 osdc->client->options->osd_keepalive_timeout);
3535}
3536
3537static void handle_osds_timeout(struct work_struct *work)
3538{
3539 struct ceph_osd_client *osdc =
3540 container_of(work, struct ceph_osd_client,
3541 osds_timeout_work.work);
3542 unsigned long delay = osdc->client->options->osd_idle_ttl / 4;
3543 struct ceph_osd *osd, *nosd;
3544
3545 dout("%s osdc %p\n", __func__, osdc);
3546 down_write(&osdc->lock);
3547 list_for_each_entry_safe(osd, nosd, &osdc->osd_lru, o_osd_lru) {
3548 if (time_before(jiffies, osd->lru_ttl))
3549 break;
3550
3551 WARN_ON(!RB_EMPTY_ROOT(&osd->o_requests));
3552 WARN_ON(!RB_EMPTY_ROOT(&osd->o_linger_requests));
3553 close_osd(osd);
3554 }
3555
3556 up_write(&osdc->lock);
3557 schedule_delayed_work(&osdc->osds_timeout_work,
3558 round_jiffies_relative(delay));
3559}
3560
3561static int ceph_oloc_decode(void **p, void *end,
3562 struct ceph_object_locator *oloc)
3563{
3564 u8 struct_v, struct_cv;
3565 u32 len;
3566 void *struct_end;
3567 int ret = 0;
3568
3569 ceph_decode_need(p, end, 1 + 1 + 4, e_inval);
3570 struct_v = ceph_decode_8(p);
3571 struct_cv = ceph_decode_8(p);
3572 if (struct_v < 3) {
3573 pr_warn("got v %d < 3 cv %d of ceph_object_locator\n",
3574 struct_v, struct_cv);
3575 goto e_inval;
3576 }
3577 if (struct_cv > 6) {
3578 pr_warn("got v %d cv %d > 6 of ceph_object_locator\n",
3579 struct_v, struct_cv);
3580 goto e_inval;
3581 }
3582 len = ceph_decode_32(p);
3583 ceph_decode_need(p, end, len, e_inval);
3584 struct_end = *p + len;
3585
3586 oloc->pool = ceph_decode_64(p);
3587 *p += 4; /* skip preferred */
3588
3589 len = ceph_decode_32(p);
3590 if (len > 0) {
3591 pr_warn("ceph_object_locator::key is set\n");
3592 goto e_inval;
3593 }
3594
3595 if (struct_v >= 5) {
3596 bool changed = false;
3597
3598 len = ceph_decode_32(p);
3599 if (len > 0) {
3600 ceph_decode_need(p, end, len, e_inval);
3601 if (!oloc->pool_ns ||
3602 ceph_compare_string(oloc->pool_ns, *p, len))
3603 changed = true;
3604 *p += len;
3605 } else {
3606 if (oloc->pool_ns)
3607 changed = true;
3608 }
3609 if (changed) {
3610 /* redirect changes namespace */
3611 pr_warn("ceph_object_locator::nspace is changed\n");
3612 goto e_inval;
3613 }
3614 }
3615
3616 if (struct_v >= 6) {
3617 s64 hash = ceph_decode_64(p);
3618 if (hash != -1) {
3619 pr_warn("ceph_object_locator::hash is set\n");
3620 goto e_inval;
3621 }
3622 }
3623
3624 /* skip the rest */
3625 *p = struct_end;
3626out:
3627 return ret;
3628
3629e_inval:
3630 ret = -EINVAL;
3631 goto out;
3632}
3633
3634static int ceph_redirect_decode(void **p, void *end,
3635 struct ceph_request_redirect *redir)
3636{
3637 u8 struct_v, struct_cv;
3638 u32 len;
3639 void *struct_end;
3640 int ret;
3641
3642 ceph_decode_need(p, end, 1 + 1 + 4, e_inval);
3643 struct_v = ceph_decode_8(p);
3644 struct_cv = ceph_decode_8(p);
3645 if (struct_cv > 1) {
3646 pr_warn("got v %d cv %d > 1 of ceph_request_redirect\n",
3647 struct_v, struct_cv);
3648 goto e_inval;
3649 }
3650 len = ceph_decode_32(p);
3651 ceph_decode_need(p, end, len, e_inval);
3652 struct_end = *p + len;
3653
3654 ret = ceph_oloc_decode(p, end, &redir->oloc);
3655 if (ret)
3656 goto out;
3657
3658 len = ceph_decode_32(p);
3659 if (len > 0) {
3660 pr_warn("ceph_request_redirect::object_name is set\n");
3661 goto e_inval;
3662 }
3663
3664 /* skip the rest */
3665 *p = struct_end;
3666out:
3667 return ret;
3668
3669e_inval:
3670 ret = -EINVAL;
3671 goto out;
3672}
3673
3674struct MOSDOpReply {
3675 struct ceph_pg pgid;
3676 u64 flags;
3677 int result;
3678 u32 epoch;
3679 int num_ops;
3680 u32 outdata_len[CEPH_OSD_MAX_OPS];
3681 s32 rval[CEPH_OSD_MAX_OPS];
3682 int retry_attempt;
3683 struct ceph_eversion replay_version;
3684 u64 user_version;
3685 struct ceph_request_redirect redirect;
3686};
3687
3688static int decode_MOSDOpReply(const struct ceph_msg *msg, struct MOSDOpReply *m)
3689{
3690 void *p = msg->front.iov_base;
3691 void *const end = p + msg->front.iov_len;
3692 u16 version = le16_to_cpu(msg->hdr.version);
3693 struct ceph_eversion bad_replay_version;
3694 u8 decode_redir;
3695 u32 len;
3696 int ret;
3697 int i;
3698
3699 ceph_decode_32_safe(&p, end, len, e_inval);
3700 ceph_decode_need(&p, end, len, e_inval);
3701 p += len; /* skip oid */
3702
3703 ret = ceph_decode_pgid(&p, end, &m->pgid);
3704 if (ret)
3705 return ret;
3706
3707 ceph_decode_64_safe(&p, end, m->flags, e_inval);
3708 ceph_decode_32_safe(&p, end, m->result, e_inval);
3709 ceph_decode_need(&p, end, sizeof(bad_replay_version), e_inval);
3710 memcpy(&bad_replay_version, p, sizeof(bad_replay_version));
3711 p += sizeof(bad_replay_version);
3712 ceph_decode_32_safe(&p, end, m->epoch, e_inval);
3713
3714 ceph_decode_32_safe(&p, end, m->num_ops, e_inval);
3715 if (m->num_ops > ARRAY_SIZE(m->outdata_len))
3716 goto e_inval;
3717
3718 ceph_decode_need(&p, end, m->num_ops * sizeof(struct ceph_osd_op),
3719 e_inval);
3720 for (i = 0; i < m->num_ops; i++) {
3721 struct ceph_osd_op *op = p;
3722
3723 m->outdata_len[i] = le32_to_cpu(op->payload_len);
3724 p += sizeof(*op);
3725 }
3726
3727 ceph_decode_32_safe(&p, end, m->retry_attempt, e_inval);
3728 for (i = 0; i < m->num_ops; i++)
3729 ceph_decode_32_safe(&p, end, m->rval[i], e_inval);
3730
3731 if (version >= 5) {
3732 ceph_decode_need(&p, end, sizeof(m->replay_version), e_inval);
3733 memcpy(&m->replay_version, p, sizeof(m->replay_version));
3734 p += sizeof(m->replay_version);
3735 ceph_decode_64_safe(&p, end, m->user_version, e_inval);
3736 } else {
3737 m->replay_version = bad_replay_version; /* struct */
3738 m->user_version = le64_to_cpu(m->replay_version.version);
3739 }
3740
3741 if (version >= 6) {
3742 if (version >= 7)
3743 ceph_decode_8_safe(&p, end, decode_redir, e_inval);
3744 else
3745 decode_redir = 1;
3746 } else {
3747 decode_redir = 0;
3748 }
3749
3750 if (decode_redir) {
3751 ret = ceph_redirect_decode(&p, end, &m->redirect);
3752 if (ret)
3753 return ret;
3754 } else {
3755 ceph_oloc_init(&m->redirect.oloc);
3756 }
3757
3758 return 0;
3759
3760e_inval:
3761 return -EINVAL;
3762}
3763
3764/*
3765 * Handle MOSDOpReply. Set ->r_result and call the callback if it is
3766 * specified.
3767 */
3768static void handle_reply(struct ceph_osd *osd, struct ceph_msg *msg)
3769{
3770 struct ceph_osd_client *osdc = osd->o_osdc;
3771 struct ceph_osd_request *req;
3772 struct MOSDOpReply m;
3773 u64 tid = le64_to_cpu(msg->hdr.tid);
3774 u32 data_len = 0;
3775 int ret;
3776 int i;
3777
3778 dout("%s msg %p tid %llu\n", __func__, msg, tid);
3779
3780 down_read(&osdc->lock);
3781 if (!osd_registered(osd)) {
3782 dout("%s osd%d unknown\n", __func__, osd->o_osd);
3783 goto out_unlock_osdc;
3784 }
3785 WARN_ON(osd->o_osd != le64_to_cpu(msg->hdr.src.num));
3786
3787 mutex_lock(&osd->lock);
3788 req = lookup_request(&osd->o_requests, tid);
3789 if (!req) {
3790 dout("%s osd%d tid %llu unknown\n", __func__, osd->o_osd, tid);
3791 goto out_unlock_session;
3792 }
3793
3794 m.redirect.oloc.pool_ns = req->r_t.target_oloc.pool_ns;
3795 ret = decode_MOSDOpReply(msg, &m);
3796 m.redirect.oloc.pool_ns = NULL;
3797 if (ret) {
3798 pr_err("failed to decode MOSDOpReply for tid %llu: %d\n",
3799 req->r_tid, ret);
3800 ceph_msg_dump(msg);
3801 goto fail_request;
3802 }
3803 dout("%s req %p tid %llu flags 0x%llx pgid %llu.%x epoch %u attempt %d v %u'%llu uv %llu\n",
3804 __func__, req, req->r_tid, m.flags, m.pgid.pool, m.pgid.seed,
3805 m.epoch, m.retry_attempt, le32_to_cpu(m.replay_version.epoch),
3806 le64_to_cpu(m.replay_version.version), m.user_version);
3807
3808 if (m.retry_attempt >= 0) {
3809 if (m.retry_attempt != req->r_attempts - 1) {
3810 dout("req %p tid %llu retry_attempt %d != %d, ignoring\n",
3811 req, req->r_tid, m.retry_attempt,
3812 req->r_attempts - 1);
3813 goto out_unlock_session;
3814 }
3815 } else {
3816 WARN_ON(1); /* MOSDOpReply v4 is assumed */
3817 }
3818
3819 if (!ceph_oloc_empty(&m.redirect.oloc)) {
3820 dout("req %p tid %llu redirect pool %lld\n", req, req->r_tid,
3821 m.redirect.oloc.pool);
3822 unlink_request(osd, req);
3823 mutex_unlock(&osd->lock);
3824
3825 /*
3826 * Not ceph_oloc_copy() - changing pool_ns is not
3827 * supported.
3828 */
3829 req->r_t.target_oloc.pool = m.redirect.oloc.pool;
3830 req->r_flags |= CEPH_OSD_FLAG_REDIRECTED |
3831 CEPH_OSD_FLAG_IGNORE_OVERLAY |
3832 CEPH_OSD_FLAG_IGNORE_CACHE;
3833 req->r_tid = 0;
3834 __submit_request(req, false);
3835 goto out_unlock_osdc;
3836 }
3837
3838 if (m.result == -EAGAIN) {
3839 dout("req %p tid %llu EAGAIN\n", req, req->r_tid);
3840 unlink_request(osd, req);
3841 mutex_unlock(&osd->lock);
3842
3843 /*
3844 * The object is missing on the replica or not (yet)
3845 * readable. Clear pgid to force a resend to the primary
3846 * via legacy_change.
3847 */
3848 req->r_t.pgid.pool = 0;
3849 req->r_t.pgid.seed = 0;
3850 WARN_ON(!req->r_t.used_replica);
3851 req->r_flags &= ~(CEPH_OSD_FLAG_BALANCE_READS |
3852 CEPH_OSD_FLAG_LOCALIZE_READS);
3853 req->r_tid = 0;
3854 __submit_request(req, false);
3855 goto out_unlock_osdc;
3856 }
3857
3858 if (m.num_ops != req->r_num_ops) {
3859 pr_err("num_ops %d != %d for tid %llu\n", m.num_ops,
3860 req->r_num_ops, req->r_tid);
3861 goto fail_request;
3862 }
3863 for (i = 0; i < req->r_num_ops; i++) {
3864 dout(" req %p tid %llu op %d rval %d len %u\n", req,
3865 req->r_tid, i, m.rval[i], m.outdata_len[i]);
3866 req->r_ops[i].rval = m.rval[i];
3867 req->r_ops[i].outdata_len = m.outdata_len[i];
3868 data_len += m.outdata_len[i];
3869 }
3870 if (data_len != le32_to_cpu(msg->hdr.data_len)) {
3871 pr_err("sum of lens %u != %u for tid %llu\n", data_len,
3872 le32_to_cpu(msg->hdr.data_len), req->r_tid);
3873 goto fail_request;
3874 }
3875 dout("%s req %p tid %llu result %d data_len %u\n", __func__,
3876 req, req->r_tid, m.result, data_len);
3877
3878 /*
3879 * Since we only ever request ONDISK, we should only ever get
3880 * one (type of) reply back.
3881 */
3882 WARN_ON(!(m.flags & CEPH_OSD_FLAG_ONDISK));
3883 req->r_version = m.user_version;
3884 req->r_result = m.result ?: data_len;
3885 finish_request(req);
3886 mutex_unlock(&osd->lock);
3887 up_read(&osdc->lock);
3888
3889 __complete_request(req);
3890 return;
3891
3892fail_request:
3893 complete_request(req, -EIO);
3894out_unlock_session:
3895 mutex_unlock(&osd->lock);
3896out_unlock_osdc:
3897 up_read(&osdc->lock);
3898}
3899
3900static void set_pool_was_full(struct ceph_osd_client *osdc)
3901{
3902 struct rb_node *n;
3903
3904 for (n = rb_first(&osdc->osdmap->pg_pools); n; n = rb_next(n)) {
3905 struct ceph_pg_pool_info *pi =
3906 rb_entry(n, struct ceph_pg_pool_info, node);
3907
3908 pi->was_full = __pool_full(pi);
3909 }
3910}
3911
3912static bool pool_cleared_full(struct ceph_osd_client *osdc, s64 pool_id)
3913{
3914 struct ceph_pg_pool_info *pi;
3915
3916 pi = ceph_pg_pool_by_id(osdc->osdmap, pool_id);
3917 if (!pi)
3918 return false;
3919
3920 return pi->was_full && !__pool_full(pi);
3921}
3922
3923static enum calc_target_result
3924recalc_linger_target(struct ceph_osd_linger_request *lreq)
3925{
3926 struct ceph_osd_client *osdc = lreq->osdc;
3927 enum calc_target_result ct_res;
3928
3929 ct_res = calc_target(osdc, &lreq->t, true);
3930 if (ct_res == CALC_TARGET_NEED_RESEND) {
3931 struct ceph_osd *osd;
3932
3933 osd = lookup_create_osd(osdc, lreq->t.osd, true);
3934 if (osd != lreq->osd) {
3935 unlink_linger(lreq->osd, lreq);
3936 link_linger(osd, lreq);
3937 }
3938 }
3939
3940 return ct_res;
3941}
3942
3943/*
3944 * Requeue requests whose mapping to an OSD has changed.
3945 */
3946static void scan_requests(struct ceph_osd *osd,
3947 bool force_resend,
3948 bool cleared_full,
3949 bool check_pool_cleared_full,
3950 struct rb_root *need_resend,
3951 struct list_head *need_resend_linger)
3952{
3953 struct ceph_osd_client *osdc = osd->o_osdc;
3954 struct rb_node *n;
3955 bool force_resend_writes;
3956
3957 for (n = rb_first(&osd->o_linger_requests); n; ) {
3958 struct ceph_osd_linger_request *lreq =
3959 rb_entry(n, struct ceph_osd_linger_request, node);
3960 enum calc_target_result ct_res;
3961
3962 n = rb_next(n); /* recalc_linger_target() */
3963
3964 dout("%s lreq %p linger_id %llu\n", __func__, lreq,
3965 lreq->linger_id);
3966 ct_res = recalc_linger_target(lreq);
3967 switch (ct_res) {
3968 case CALC_TARGET_NO_ACTION:
3969 force_resend_writes = cleared_full ||
3970 (check_pool_cleared_full &&
3971 pool_cleared_full(osdc, lreq->t.base_oloc.pool));
3972 if (!force_resend && !force_resend_writes)
3973 break;
3974
3975 fallthrough;
3976 case CALC_TARGET_NEED_RESEND:
3977 cancel_linger_map_check(lreq);
3978 /*
3979 * scan_requests() for the previous epoch(s)
3980 * may have already added it to the list, since
3981 * it's not unlinked here.
3982 */
3983 if (list_empty(&lreq->scan_item))
3984 list_add_tail(&lreq->scan_item, need_resend_linger);
3985 break;
3986 case CALC_TARGET_POOL_DNE:
3987 list_del_init(&lreq->scan_item);
3988 check_linger_pool_dne(lreq);
3989 break;
3990 }
3991 }
3992
3993 for (n = rb_first(&osd->o_requests); n; ) {
3994 struct ceph_osd_request *req =
3995 rb_entry(n, struct ceph_osd_request, r_node);
3996 enum calc_target_result ct_res;
3997
3998 n = rb_next(n); /* unlink_request(), check_pool_dne() */
3999
4000 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
4001 ct_res = calc_target(osdc, &req->r_t, false);
4002 switch (ct_res) {
4003 case CALC_TARGET_NO_ACTION:
4004 force_resend_writes = cleared_full ||
4005 (check_pool_cleared_full &&
4006 pool_cleared_full(osdc, req->r_t.base_oloc.pool));
4007 if (!force_resend &&
4008 (!(req->r_flags & CEPH_OSD_FLAG_WRITE) ||
4009 !force_resend_writes))
4010 break;
4011
4012 fallthrough;
4013 case CALC_TARGET_NEED_RESEND:
4014 cancel_map_check(req);
4015 unlink_request(osd, req);
4016 insert_request(need_resend, req);
4017 break;
4018 case CALC_TARGET_POOL_DNE:
4019 check_pool_dne(req);
4020 break;
4021 }
4022 }
4023}
4024
4025static int handle_one_map(struct ceph_osd_client *osdc,
4026 void *p, void *end, bool incremental,
4027 struct rb_root *need_resend,
4028 struct list_head *need_resend_linger)
4029{
4030 struct ceph_osdmap *newmap;
4031 struct rb_node *n;
4032 bool skipped_map = false;
4033 bool was_full;
4034
4035 was_full = ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL);
4036 set_pool_was_full(osdc);
4037
4038 if (incremental)
4039 newmap = osdmap_apply_incremental(&p, end,
4040 ceph_msgr2(osdc->client),
4041 osdc->osdmap);
4042 else
4043 newmap = ceph_osdmap_decode(&p, end, ceph_msgr2(osdc->client));
4044 if (IS_ERR(newmap))
4045 return PTR_ERR(newmap);
4046
4047 if (newmap != osdc->osdmap) {
4048 /*
4049 * Preserve ->was_full before destroying the old map.
4050 * For pools that weren't in the old map, ->was_full
4051 * should be false.
4052 */
4053 for (n = rb_first(&newmap->pg_pools); n; n = rb_next(n)) {
4054 struct ceph_pg_pool_info *pi =
4055 rb_entry(n, struct ceph_pg_pool_info, node);
4056 struct ceph_pg_pool_info *old_pi;
4057
4058 old_pi = ceph_pg_pool_by_id(osdc->osdmap, pi->id);
4059 if (old_pi)
4060 pi->was_full = old_pi->was_full;
4061 else
4062 WARN_ON(pi->was_full);
4063 }
4064
4065 if (osdc->osdmap->epoch &&
4066 osdc->osdmap->epoch + 1 < newmap->epoch) {
4067 WARN_ON(incremental);
4068 skipped_map = true;
4069 }
4070
4071 ceph_osdmap_destroy(osdc->osdmap);
4072 osdc->osdmap = newmap;
4073 }
4074
4075 was_full &= !ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL);
4076 scan_requests(&osdc->homeless_osd, skipped_map, was_full, true,
4077 need_resend, need_resend_linger);
4078
4079 for (n = rb_first(&osdc->osds); n; ) {
4080 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
4081
4082 n = rb_next(n); /* close_osd() */
4083
4084 scan_requests(osd, skipped_map, was_full, true, need_resend,
4085 need_resend_linger);
4086 if (!ceph_osd_is_up(osdc->osdmap, osd->o_osd) ||
4087 memcmp(&osd->o_con.peer_addr,
4088 ceph_osd_addr(osdc->osdmap, osd->o_osd),
4089 sizeof(struct ceph_entity_addr)))
4090 close_osd(osd);
4091 }
4092
4093 return 0;
4094}
4095
4096static void kick_requests(struct ceph_osd_client *osdc,
4097 struct rb_root *need_resend,
4098 struct list_head *need_resend_linger)
4099{
4100 struct ceph_osd_linger_request *lreq, *nlreq;
4101 enum calc_target_result ct_res;
4102 struct rb_node *n;
4103
4104 /* make sure need_resend targets reflect latest map */
4105 for (n = rb_first(need_resend); n; ) {
4106 struct ceph_osd_request *req =
4107 rb_entry(n, struct ceph_osd_request, r_node);
4108
4109 n = rb_next(n);
4110
4111 if (req->r_t.epoch < osdc->osdmap->epoch) {
4112 ct_res = calc_target(osdc, &req->r_t, false);
4113 if (ct_res == CALC_TARGET_POOL_DNE) {
4114 erase_request(need_resend, req);
4115 check_pool_dne(req);
4116 }
4117 }
4118 }
4119
4120 for (n = rb_first(need_resend); n; ) {
4121 struct ceph_osd_request *req =
4122 rb_entry(n, struct ceph_osd_request, r_node);
4123 struct ceph_osd *osd;
4124
4125 n = rb_next(n);
4126 erase_request(need_resend, req); /* before link_request() */
4127
4128 osd = lookup_create_osd(osdc, req->r_t.osd, true);
4129 link_request(osd, req);
4130 if (!req->r_linger) {
4131 if (!osd_homeless(osd) && !req->r_t.paused)
4132 send_request(req);
4133 } else {
4134 cancel_linger_request(req);
4135 }
4136 }
4137
4138 list_for_each_entry_safe(lreq, nlreq, need_resend_linger, scan_item) {
4139 if (!osd_homeless(lreq->osd))
4140 send_linger(lreq);
4141
4142 list_del_init(&lreq->scan_item);
4143 }
4144}
4145
4146/*
4147 * Process updated osd map.
4148 *
4149 * The message contains any number of incremental and full maps, normally
4150 * indicating some sort of topology change in the cluster. Kick requests
4151 * off to different OSDs as needed.
4152 */
4153void ceph_osdc_handle_map(struct ceph_osd_client *osdc, struct ceph_msg *msg)
4154{
4155 void *p = msg->front.iov_base;
4156 void *const end = p + msg->front.iov_len;
4157 u32 nr_maps, maplen;
4158 u32 epoch;
4159 struct ceph_fsid fsid;
4160 struct rb_root need_resend = RB_ROOT;
4161 LIST_HEAD(need_resend_linger);
4162 bool handled_incremental = false;
4163 bool was_pauserd, was_pausewr;
4164 bool pauserd, pausewr;
4165 int err;
4166
4167 dout("%s have %u\n", __func__, osdc->osdmap->epoch);
4168 down_write(&osdc->lock);
4169
4170 /* verify fsid */
4171 ceph_decode_need(&p, end, sizeof(fsid), bad);
4172 ceph_decode_copy(&p, &fsid, sizeof(fsid));
4173 if (ceph_check_fsid(osdc->client, &fsid) < 0)
4174 goto bad;
4175
4176 was_pauserd = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD);
4177 was_pausewr = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR) ||
4178 ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
4179 have_pool_full(osdc);
4180
4181 /* incremental maps */
4182 ceph_decode_32_safe(&p, end, nr_maps, bad);
4183 dout(" %d inc maps\n", nr_maps);
4184 while (nr_maps > 0) {
4185 ceph_decode_need(&p, end, 2*sizeof(u32), bad);
4186 epoch = ceph_decode_32(&p);
4187 maplen = ceph_decode_32(&p);
4188 ceph_decode_need(&p, end, maplen, bad);
4189 if (osdc->osdmap->epoch &&
4190 osdc->osdmap->epoch + 1 == epoch) {
4191 dout("applying incremental map %u len %d\n",
4192 epoch, maplen);
4193 err = handle_one_map(osdc, p, p + maplen, true,
4194 &need_resend, &need_resend_linger);
4195 if (err)
4196 goto bad;
4197 handled_incremental = true;
4198 } else {
4199 dout("ignoring incremental map %u len %d\n",
4200 epoch, maplen);
4201 }
4202 p += maplen;
4203 nr_maps--;
4204 }
4205 if (handled_incremental)
4206 goto done;
4207
4208 /* full maps */
4209 ceph_decode_32_safe(&p, end, nr_maps, bad);
4210 dout(" %d full maps\n", nr_maps);
4211 while (nr_maps) {
4212 ceph_decode_need(&p, end, 2*sizeof(u32), bad);
4213 epoch = ceph_decode_32(&p);
4214 maplen = ceph_decode_32(&p);
4215 ceph_decode_need(&p, end, maplen, bad);
4216 if (nr_maps > 1) {
4217 dout("skipping non-latest full map %u len %d\n",
4218 epoch, maplen);
4219 } else if (osdc->osdmap->epoch >= epoch) {
4220 dout("skipping full map %u len %d, "
4221 "older than our %u\n", epoch, maplen,
4222 osdc->osdmap->epoch);
4223 } else {
4224 dout("taking full map %u len %d\n", epoch, maplen);
4225 err = handle_one_map(osdc, p, p + maplen, false,
4226 &need_resend, &need_resend_linger);
4227 if (err)
4228 goto bad;
4229 }
4230 p += maplen;
4231 nr_maps--;
4232 }
4233
4234done:
4235 /*
4236 * subscribe to subsequent osdmap updates if full to ensure
4237 * we find out when we are no longer full and stop returning
4238 * ENOSPC.
4239 */
4240 pauserd = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSERD);
4241 pausewr = ceph_osdmap_flag(osdc, CEPH_OSDMAP_PAUSEWR) ||
4242 ceph_osdmap_flag(osdc, CEPH_OSDMAP_FULL) ||
4243 have_pool_full(osdc);
4244 if (was_pauserd || was_pausewr || pauserd || pausewr ||
4245 osdc->osdmap->epoch < osdc->epoch_barrier)
4246 maybe_request_map(osdc);
4247
4248 kick_requests(osdc, &need_resend, &need_resend_linger);
4249
4250 ceph_osdc_abort_on_full(osdc);
4251 ceph_monc_got_map(&osdc->client->monc, CEPH_SUB_OSDMAP,
4252 osdc->osdmap->epoch);
4253 up_write(&osdc->lock);
4254 wake_up_all(&osdc->client->auth_wq);
4255 return;
4256
4257bad:
4258 pr_err("osdc handle_map corrupt msg\n");
4259 ceph_msg_dump(msg);
4260 up_write(&osdc->lock);
4261}
4262
4263/*
4264 * Resubmit requests pending on the given osd.
4265 */
4266static void kick_osd_requests(struct ceph_osd *osd)
4267{
4268 struct rb_node *n;
4269
4270 clear_backoffs(osd);
4271
4272 for (n = rb_first(&osd->o_requests); n; ) {
4273 struct ceph_osd_request *req =
4274 rb_entry(n, struct ceph_osd_request, r_node);
4275
4276 n = rb_next(n); /* cancel_linger_request() */
4277
4278 if (!req->r_linger) {
4279 if (!req->r_t.paused)
4280 send_request(req);
4281 } else {
4282 cancel_linger_request(req);
4283 }
4284 }
4285 for (n = rb_first(&osd->o_linger_requests); n; n = rb_next(n)) {
4286 struct ceph_osd_linger_request *lreq =
4287 rb_entry(n, struct ceph_osd_linger_request, node);
4288
4289 send_linger(lreq);
4290 }
4291}
4292
4293/*
4294 * If the osd connection drops, we need to resubmit all requests.
4295 */
4296static void osd_fault(struct ceph_connection *con)
4297{
4298 struct ceph_osd *osd = con->private;
4299 struct ceph_osd_client *osdc = osd->o_osdc;
4300
4301 dout("%s osd %p osd%d\n", __func__, osd, osd->o_osd);
4302
4303 down_write(&osdc->lock);
4304 if (!osd_registered(osd)) {
4305 dout("%s osd%d unknown\n", __func__, osd->o_osd);
4306 goto out_unlock;
4307 }
4308
4309 if (!reopen_osd(osd))
4310 kick_osd_requests(osd);
4311 maybe_request_map(osdc);
4312
4313out_unlock:
4314 up_write(&osdc->lock);
4315}
4316
4317struct MOSDBackoff {
4318 struct ceph_spg spgid;
4319 u32 map_epoch;
4320 u8 op;
4321 u64 id;
4322 struct ceph_hobject_id *begin;
4323 struct ceph_hobject_id *end;
4324};
4325
4326static int decode_MOSDBackoff(const struct ceph_msg *msg, struct MOSDBackoff *m)
4327{
4328 void *p = msg->front.iov_base;
4329 void *const end = p + msg->front.iov_len;
4330 u8 struct_v;
4331 u32 struct_len;
4332 int ret;
4333
4334 ret = ceph_start_decoding(&p, end, 1, "spg_t", &struct_v, &struct_len);
4335 if (ret)
4336 return ret;
4337
4338 ret = ceph_decode_pgid(&p, end, &m->spgid.pgid);
4339 if (ret)
4340 return ret;
4341
4342 ceph_decode_8_safe(&p, end, m->spgid.shard, e_inval);
4343 ceph_decode_32_safe(&p, end, m->map_epoch, e_inval);
4344 ceph_decode_8_safe(&p, end, m->op, e_inval);
4345 ceph_decode_64_safe(&p, end, m->id, e_inval);
4346
4347 m->begin = kzalloc(sizeof(*m->begin), GFP_NOIO);
4348 if (!m->begin)
4349 return -ENOMEM;
4350
4351 ret = decode_hoid(&p, end, m->begin);
4352 if (ret) {
4353 free_hoid(m->begin);
4354 return ret;
4355 }
4356
4357 m->end = kzalloc(sizeof(*m->end), GFP_NOIO);
4358 if (!m->end) {
4359 free_hoid(m->begin);
4360 return -ENOMEM;
4361 }
4362
4363 ret = decode_hoid(&p, end, m->end);
4364 if (ret) {
4365 free_hoid(m->begin);
4366 free_hoid(m->end);
4367 return ret;
4368 }
4369
4370 return 0;
4371
4372e_inval:
4373 return -EINVAL;
4374}
4375
4376static struct ceph_msg *create_backoff_message(
4377 const struct ceph_osd_backoff *backoff,
4378 u32 map_epoch)
4379{
4380 struct ceph_msg *msg;
4381 void *p, *end;
4382 int msg_size;
4383
4384 msg_size = CEPH_ENCODING_START_BLK_LEN +
4385 CEPH_PGID_ENCODING_LEN + 1; /* spgid */
4386 msg_size += 4 + 1 + 8; /* map_epoch, op, id */
4387 msg_size += CEPH_ENCODING_START_BLK_LEN +
4388 hoid_encoding_size(backoff->begin);
4389 msg_size += CEPH_ENCODING_START_BLK_LEN +
4390 hoid_encoding_size(backoff->end);
4391
4392 msg = ceph_msg_new(CEPH_MSG_OSD_BACKOFF, msg_size, GFP_NOIO, true);
4393 if (!msg)
4394 return NULL;
4395
4396 p = msg->front.iov_base;
4397 end = p + msg->front_alloc_len;
4398
4399 encode_spgid(&p, &backoff->spgid);
4400 ceph_encode_32(&p, map_epoch);
4401 ceph_encode_8(&p, CEPH_OSD_BACKOFF_OP_ACK_BLOCK);
4402 ceph_encode_64(&p, backoff->id);
4403 encode_hoid(&p, end, backoff->begin);
4404 encode_hoid(&p, end, backoff->end);
4405 BUG_ON(p != end);
4406
4407 msg->front.iov_len = p - msg->front.iov_base;
4408 msg->hdr.version = cpu_to_le16(1); /* MOSDBackoff v1 */
4409 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
4410
4411 return msg;
4412}
4413
4414static void handle_backoff_block(struct ceph_osd *osd, struct MOSDBackoff *m)
4415{
4416 struct ceph_spg_mapping *spg;
4417 struct ceph_osd_backoff *backoff;
4418 struct ceph_msg *msg;
4419
4420 dout("%s osd%d spgid %llu.%xs%d id %llu\n", __func__, osd->o_osd,
4421 m->spgid.pgid.pool, m->spgid.pgid.seed, m->spgid.shard, m->id);
4422
4423 spg = lookup_spg_mapping(&osd->o_backoff_mappings, &m->spgid);
4424 if (!spg) {
4425 spg = alloc_spg_mapping();
4426 if (!spg) {
4427 pr_err("%s failed to allocate spg\n", __func__);
4428 return;
4429 }
4430 spg->spgid = m->spgid; /* struct */
4431 insert_spg_mapping(&osd->o_backoff_mappings, spg);
4432 }
4433
4434 backoff = alloc_backoff();
4435 if (!backoff) {
4436 pr_err("%s failed to allocate backoff\n", __func__);
4437 return;
4438 }
4439 backoff->spgid = m->spgid; /* struct */
4440 backoff->id = m->id;
4441 backoff->begin = m->begin;
4442 m->begin = NULL; /* backoff now owns this */
4443 backoff->end = m->end;
4444 m->end = NULL; /* ditto */
4445
4446 insert_backoff(&spg->backoffs, backoff);
4447 insert_backoff_by_id(&osd->o_backoffs_by_id, backoff);
4448
4449 /*
4450 * Ack with original backoff's epoch so that the OSD can
4451 * discard this if there was a PG split.
4452 */
4453 msg = create_backoff_message(backoff, m->map_epoch);
4454 if (!msg) {
4455 pr_err("%s failed to allocate msg\n", __func__);
4456 return;
4457 }
4458 ceph_con_send(&osd->o_con, msg);
4459}
4460
4461static bool target_contained_by(const struct ceph_osd_request_target *t,
4462 const struct ceph_hobject_id *begin,
4463 const struct ceph_hobject_id *end)
4464{
4465 struct ceph_hobject_id hoid;
4466 int cmp;
4467
4468 hoid_fill_from_target(&hoid, t);
4469 cmp = hoid_compare(&hoid, begin);
4470 return !cmp || (cmp > 0 && hoid_compare(&hoid, end) < 0);
4471}
4472
4473static void handle_backoff_unblock(struct ceph_osd *osd,
4474 const struct MOSDBackoff *m)
4475{
4476 struct ceph_spg_mapping *spg;
4477 struct ceph_osd_backoff *backoff;
4478 struct rb_node *n;
4479
4480 dout("%s osd%d spgid %llu.%xs%d id %llu\n", __func__, osd->o_osd,
4481 m->spgid.pgid.pool, m->spgid.pgid.seed, m->spgid.shard, m->id);
4482
4483 backoff = lookup_backoff_by_id(&osd->o_backoffs_by_id, m->id);
4484 if (!backoff) {
4485 pr_err("%s osd%d spgid %llu.%xs%d id %llu backoff dne\n",
4486 __func__, osd->o_osd, m->spgid.pgid.pool,
4487 m->spgid.pgid.seed, m->spgid.shard, m->id);
4488 return;
4489 }
4490
4491 if (hoid_compare(backoff->begin, m->begin) &&
4492 hoid_compare(backoff->end, m->end)) {
4493 pr_err("%s osd%d spgid %llu.%xs%d id %llu bad range?\n",
4494 __func__, osd->o_osd, m->spgid.pgid.pool,
4495 m->spgid.pgid.seed, m->spgid.shard, m->id);
4496 /* unblock it anyway... */
4497 }
4498
4499 spg = lookup_spg_mapping(&osd->o_backoff_mappings, &backoff->spgid);
4500 BUG_ON(!spg);
4501
4502 erase_backoff(&spg->backoffs, backoff);
4503 erase_backoff_by_id(&osd->o_backoffs_by_id, backoff);
4504 free_backoff(backoff);
4505
4506 if (RB_EMPTY_ROOT(&spg->backoffs)) {
4507 erase_spg_mapping(&osd->o_backoff_mappings, spg);
4508 free_spg_mapping(spg);
4509 }
4510
4511 for (n = rb_first(&osd->o_requests); n; n = rb_next(n)) {
4512 struct ceph_osd_request *req =
4513 rb_entry(n, struct ceph_osd_request, r_node);
4514
4515 if (!ceph_spg_compare(&req->r_t.spgid, &m->spgid)) {
4516 /*
4517 * Match against @m, not @backoff -- the PG may
4518 * have split on the OSD.
4519 */
4520 if (target_contained_by(&req->r_t, m->begin, m->end)) {
4521 /*
4522 * If no other installed backoff applies,
4523 * resend.
4524 */
4525 send_request(req);
4526 }
4527 }
4528 }
4529}
4530
4531static void handle_backoff(struct ceph_osd *osd, struct ceph_msg *msg)
4532{
4533 struct ceph_osd_client *osdc = osd->o_osdc;
4534 struct MOSDBackoff m;
4535 int ret;
4536
4537 down_read(&osdc->lock);
4538 if (!osd_registered(osd)) {
4539 dout("%s osd%d unknown\n", __func__, osd->o_osd);
4540 up_read(&osdc->lock);
4541 return;
4542 }
4543 WARN_ON(osd->o_osd != le64_to_cpu(msg->hdr.src.num));
4544
4545 mutex_lock(&osd->lock);
4546 ret = decode_MOSDBackoff(msg, &m);
4547 if (ret) {
4548 pr_err("failed to decode MOSDBackoff: %d\n", ret);
4549 ceph_msg_dump(msg);
4550 goto out_unlock;
4551 }
4552
4553 switch (m.op) {
4554 case CEPH_OSD_BACKOFF_OP_BLOCK:
4555 handle_backoff_block(osd, &m);
4556 break;
4557 case CEPH_OSD_BACKOFF_OP_UNBLOCK:
4558 handle_backoff_unblock(osd, &m);
4559 break;
4560 default:
4561 pr_err("%s osd%d unknown op %d\n", __func__, osd->o_osd, m.op);
4562 }
4563
4564 free_hoid(m.begin);
4565 free_hoid(m.end);
4566
4567out_unlock:
4568 mutex_unlock(&osd->lock);
4569 up_read(&osdc->lock);
4570}
4571
4572/*
4573 * Process osd watch notifications
4574 */
4575static void handle_watch_notify(struct ceph_osd_client *osdc,
4576 struct ceph_msg *msg)
4577{
4578 void *p = msg->front.iov_base;
4579 void *const end = p + msg->front.iov_len;
4580 struct ceph_osd_linger_request *lreq;
4581 struct linger_work *lwork;
4582 u8 proto_ver, opcode;
4583 u64 cookie, notify_id;
4584 u64 notifier_id = 0;
4585 s32 return_code = 0;
4586 void *payload = NULL;
4587 u32 payload_len = 0;
4588
4589 ceph_decode_8_safe(&p, end, proto_ver, bad);
4590 ceph_decode_8_safe(&p, end, opcode, bad);
4591 ceph_decode_64_safe(&p, end, cookie, bad);
4592 p += 8; /* skip ver */
4593 ceph_decode_64_safe(&p, end, notify_id, bad);
4594
4595 if (proto_ver >= 1) {
4596 ceph_decode_32_safe(&p, end, payload_len, bad);
4597 ceph_decode_need(&p, end, payload_len, bad);
4598 payload = p;
4599 p += payload_len;
4600 }
4601
4602 if (le16_to_cpu(msg->hdr.version) >= 2)
4603 ceph_decode_32_safe(&p, end, return_code, bad);
4604
4605 if (le16_to_cpu(msg->hdr.version) >= 3)
4606 ceph_decode_64_safe(&p, end, notifier_id, bad);
4607
4608 down_read(&osdc->lock);
4609 lreq = lookup_linger_osdc(&osdc->linger_requests, cookie);
4610 if (!lreq) {
4611 dout("%s opcode %d cookie %llu dne\n", __func__, opcode,
4612 cookie);
4613 goto out_unlock_osdc;
4614 }
4615
4616 mutex_lock(&lreq->lock);
4617 dout("%s opcode %d cookie %llu lreq %p is_watch %d\n", __func__,
4618 opcode, cookie, lreq, lreq->is_watch);
4619 if (opcode == CEPH_WATCH_EVENT_DISCONNECT) {
4620 if (!lreq->last_error) {
4621 lreq->last_error = -ENOTCONN;
4622 queue_watch_error(lreq);
4623 }
4624 } else if (!lreq->is_watch) {
4625 /* CEPH_WATCH_EVENT_NOTIFY_COMPLETE */
4626 if (lreq->notify_id && lreq->notify_id != notify_id) {
4627 dout("lreq %p notify_id %llu != %llu, ignoring\n", lreq,
4628 lreq->notify_id, notify_id);
4629 } else if (!completion_done(&lreq->notify_finish_wait)) {
4630 struct ceph_msg_data *data =
4631 msg->num_data_items ? &msg->data[0] : NULL;
4632
4633 if (data) {
4634 if (lreq->preply_pages) {
4635 WARN_ON(data->type !=
4636 CEPH_MSG_DATA_PAGES);
4637 *lreq->preply_pages = data->pages;
4638 *lreq->preply_len = data->length;
4639 data->own_pages = false;
4640 }
4641 }
4642 lreq->notify_finish_error = return_code;
4643 complete_all(&lreq->notify_finish_wait);
4644 }
4645 } else {
4646 /* CEPH_WATCH_EVENT_NOTIFY */
4647 lwork = lwork_alloc(lreq, do_watch_notify);
4648 if (!lwork) {
4649 pr_err("failed to allocate notify-lwork\n");
4650 goto out_unlock_lreq;
4651 }
4652
4653 lwork->notify.notify_id = notify_id;
4654 lwork->notify.notifier_id = notifier_id;
4655 lwork->notify.payload = payload;
4656 lwork->notify.payload_len = payload_len;
4657 lwork->notify.msg = ceph_msg_get(msg);
4658 lwork_queue(lwork);
4659 }
4660
4661out_unlock_lreq:
4662 mutex_unlock(&lreq->lock);
4663out_unlock_osdc:
4664 up_read(&osdc->lock);
4665 return;
4666
4667bad:
4668 pr_err("osdc handle_watch_notify corrupt msg\n");
4669}
4670
4671/*
4672 * Register request, send initial attempt.
4673 */
4674void ceph_osdc_start_request(struct ceph_osd_client *osdc,
4675 struct ceph_osd_request *req)
4676{
4677 down_read(&osdc->lock);
4678 submit_request(req, false);
4679 up_read(&osdc->lock);
4680}
4681EXPORT_SYMBOL(ceph_osdc_start_request);
4682
4683/*
4684 * Unregister request. If @req was registered, it isn't completed:
4685 * r_result isn't set and __complete_request() isn't invoked.
4686 *
4687 * If @req wasn't registered, this call may have raced with
4688 * handle_reply(), in which case r_result would already be set and
4689 * __complete_request() would be getting invoked, possibly even
4690 * concurrently with this call.
4691 */
4692void ceph_osdc_cancel_request(struct ceph_osd_request *req)
4693{
4694 struct ceph_osd_client *osdc = req->r_osdc;
4695
4696 down_write(&osdc->lock);
4697 if (req->r_osd)
4698 cancel_request(req);
4699 up_write(&osdc->lock);
4700}
4701EXPORT_SYMBOL(ceph_osdc_cancel_request);
4702
4703/*
4704 * @timeout: in jiffies, 0 means "wait forever"
4705 */
4706static int wait_request_timeout(struct ceph_osd_request *req,
4707 unsigned long timeout)
4708{
4709 long left;
4710
4711 dout("%s req %p tid %llu\n", __func__, req, req->r_tid);
4712 left = wait_for_completion_killable_timeout(&req->r_completion,
4713 ceph_timeout_jiffies(timeout));
4714 if (left <= 0) {
4715 left = left ?: -ETIMEDOUT;
4716 ceph_osdc_cancel_request(req);
4717 } else {
4718 left = req->r_result; /* completed */
4719 }
4720
4721 return left;
4722}
4723
4724/*
4725 * wait for a request to complete
4726 */
4727int ceph_osdc_wait_request(struct ceph_osd_client *osdc,
4728 struct ceph_osd_request *req)
4729{
4730 return wait_request_timeout(req, 0);
4731}
4732EXPORT_SYMBOL(ceph_osdc_wait_request);
4733
4734/*
4735 * sync - wait for all in-flight requests to flush. avoid starvation.
4736 */
4737void ceph_osdc_sync(struct ceph_osd_client *osdc)
4738{
4739 struct rb_node *n, *p;
4740 u64 last_tid = atomic64_read(&osdc->last_tid);
4741
4742again:
4743 down_read(&osdc->lock);
4744 for (n = rb_first(&osdc->osds); n; n = rb_next(n)) {
4745 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
4746
4747 mutex_lock(&osd->lock);
4748 for (p = rb_first(&osd->o_requests); p; p = rb_next(p)) {
4749 struct ceph_osd_request *req =
4750 rb_entry(p, struct ceph_osd_request, r_node);
4751
4752 if (req->r_tid > last_tid)
4753 break;
4754
4755 if (!(req->r_flags & CEPH_OSD_FLAG_WRITE))
4756 continue;
4757
4758 ceph_osdc_get_request(req);
4759 mutex_unlock(&osd->lock);
4760 up_read(&osdc->lock);
4761 dout("%s waiting on req %p tid %llu last_tid %llu\n",
4762 __func__, req, req->r_tid, last_tid);
4763 wait_for_completion(&req->r_completion);
4764 ceph_osdc_put_request(req);
4765 goto again;
4766 }
4767
4768 mutex_unlock(&osd->lock);
4769 }
4770
4771 up_read(&osdc->lock);
4772 dout("%s done last_tid %llu\n", __func__, last_tid);
4773}
4774EXPORT_SYMBOL(ceph_osdc_sync);
4775
4776/*
4777 * Returns a handle, caller owns a ref.
4778 */
4779struct ceph_osd_linger_request *
4780ceph_osdc_watch(struct ceph_osd_client *osdc,
4781 struct ceph_object_id *oid,
4782 struct ceph_object_locator *oloc,
4783 rados_watchcb2_t wcb,
4784 rados_watcherrcb_t errcb,
4785 void *data)
4786{
4787 struct ceph_osd_linger_request *lreq;
4788 int ret;
4789
4790 lreq = linger_alloc(osdc);
4791 if (!lreq)
4792 return ERR_PTR(-ENOMEM);
4793
4794 lreq->is_watch = true;
4795 lreq->wcb = wcb;
4796 lreq->errcb = errcb;
4797 lreq->data = data;
4798 lreq->watch_valid_thru = jiffies;
4799
4800 ceph_oid_copy(&lreq->t.base_oid, oid);
4801 ceph_oloc_copy(&lreq->t.base_oloc, oloc);
4802 lreq->t.flags = CEPH_OSD_FLAG_WRITE;
4803 ktime_get_real_ts64(&lreq->mtime);
4804
4805 linger_submit(lreq);
4806 ret = linger_reg_commit_wait(lreq);
4807 if (ret) {
4808 linger_cancel(lreq);
4809 goto err_put_lreq;
4810 }
4811
4812 return lreq;
4813
4814err_put_lreq:
4815 linger_put(lreq);
4816 return ERR_PTR(ret);
4817}
4818EXPORT_SYMBOL(ceph_osdc_watch);
4819
4820/*
4821 * Releases a ref.
4822 *
4823 * Times out after mount_timeout to preserve rbd unmap behaviour
4824 * introduced in 2894e1d76974 ("rbd: timeout watch teardown on unmap
4825 * with mount_timeout").
4826 */
4827int ceph_osdc_unwatch(struct ceph_osd_client *osdc,
4828 struct ceph_osd_linger_request *lreq)
4829{
4830 struct ceph_options *opts = osdc->client->options;
4831 struct ceph_osd_request *req;
4832 int ret;
4833
4834 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
4835 if (!req)
4836 return -ENOMEM;
4837
4838 ceph_oid_copy(&req->r_base_oid, &lreq->t.base_oid);
4839 ceph_oloc_copy(&req->r_base_oloc, &lreq->t.base_oloc);
4840 req->r_flags = CEPH_OSD_FLAG_WRITE;
4841 ktime_get_real_ts64(&req->r_mtime);
4842 osd_req_op_watch_init(req, 0, CEPH_OSD_WATCH_OP_UNWATCH,
4843 lreq->linger_id, 0);
4844
4845 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
4846 if (ret)
4847 goto out_put_req;
4848
4849 ceph_osdc_start_request(osdc, req);
4850 linger_cancel(lreq);
4851 linger_put(lreq);
4852 ret = wait_request_timeout(req, opts->mount_timeout);
4853
4854out_put_req:
4855 ceph_osdc_put_request(req);
4856 return ret;
4857}
4858EXPORT_SYMBOL(ceph_osdc_unwatch);
4859
4860static int osd_req_op_notify_ack_init(struct ceph_osd_request *req, int which,
4861 u64 notify_id, u64 cookie, void *payload,
4862 u32 payload_len)
4863{
4864 struct ceph_osd_req_op *op;
4865 struct ceph_pagelist *pl;
4866 int ret;
4867
4868 op = osd_req_op_init(req, which, CEPH_OSD_OP_NOTIFY_ACK, 0);
4869
4870 pl = ceph_pagelist_alloc(GFP_NOIO);
4871 if (!pl)
4872 return -ENOMEM;
4873
4874 ret = ceph_pagelist_encode_64(pl, notify_id);
4875 ret |= ceph_pagelist_encode_64(pl, cookie);
4876 if (payload) {
4877 ret |= ceph_pagelist_encode_32(pl, payload_len);
4878 ret |= ceph_pagelist_append(pl, payload, payload_len);
4879 } else {
4880 ret |= ceph_pagelist_encode_32(pl, 0);
4881 }
4882 if (ret) {
4883 ceph_pagelist_release(pl);
4884 return -ENOMEM;
4885 }
4886
4887 ceph_osd_data_pagelist_init(&op->notify_ack.request_data, pl);
4888 op->indata_len = pl->length;
4889 return 0;
4890}
4891
4892int ceph_osdc_notify_ack(struct ceph_osd_client *osdc,
4893 struct ceph_object_id *oid,
4894 struct ceph_object_locator *oloc,
4895 u64 notify_id,
4896 u64 cookie,
4897 void *payload,
4898 u32 payload_len)
4899{
4900 struct ceph_osd_request *req;
4901 int ret;
4902
4903 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
4904 if (!req)
4905 return -ENOMEM;
4906
4907 ceph_oid_copy(&req->r_base_oid, oid);
4908 ceph_oloc_copy(&req->r_base_oloc, oloc);
4909 req->r_flags = CEPH_OSD_FLAG_READ;
4910
4911 ret = osd_req_op_notify_ack_init(req, 0, notify_id, cookie, payload,
4912 payload_len);
4913 if (ret)
4914 goto out_put_req;
4915
4916 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
4917 if (ret)
4918 goto out_put_req;
4919
4920 ceph_osdc_start_request(osdc, req);
4921 ret = ceph_osdc_wait_request(osdc, req);
4922
4923out_put_req:
4924 ceph_osdc_put_request(req);
4925 return ret;
4926}
4927EXPORT_SYMBOL(ceph_osdc_notify_ack);
4928
4929/*
4930 * @timeout: in seconds
4931 *
4932 * @preply_{pages,len} are initialized both on success and error.
4933 * The caller is responsible for:
4934 *
4935 * ceph_release_page_vector(reply_pages, calc_pages_for(0, reply_len))
4936 */
4937int ceph_osdc_notify(struct ceph_osd_client *osdc,
4938 struct ceph_object_id *oid,
4939 struct ceph_object_locator *oloc,
4940 void *payload,
4941 u32 payload_len,
4942 u32 timeout,
4943 struct page ***preply_pages,
4944 size_t *preply_len)
4945{
4946 struct ceph_osd_linger_request *lreq;
4947 int ret;
4948
4949 WARN_ON(!timeout);
4950 if (preply_pages) {
4951 *preply_pages = NULL;
4952 *preply_len = 0;
4953 }
4954
4955 lreq = linger_alloc(osdc);
4956 if (!lreq)
4957 return -ENOMEM;
4958
4959 lreq->request_pl = ceph_pagelist_alloc(GFP_NOIO);
4960 if (!lreq->request_pl) {
4961 ret = -ENOMEM;
4962 goto out_put_lreq;
4963 }
4964
4965 ret = ceph_pagelist_encode_32(lreq->request_pl, 1); /* prot_ver */
4966 ret |= ceph_pagelist_encode_32(lreq->request_pl, timeout);
4967 ret |= ceph_pagelist_encode_32(lreq->request_pl, payload_len);
4968 ret |= ceph_pagelist_append(lreq->request_pl, payload, payload_len);
4969 if (ret) {
4970 ret = -ENOMEM;
4971 goto out_put_lreq;
4972 }
4973
4974 /* for notify_id */
4975 lreq->notify_id_pages = ceph_alloc_page_vector(1, GFP_NOIO);
4976 if (IS_ERR(lreq->notify_id_pages)) {
4977 ret = PTR_ERR(lreq->notify_id_pages);
4978 lreq->notify_id_pages = NULL;
4979 goto out_put_lreq;
4980 }
4981
4982 lreq->preply_pages = preply_pages;
4983 lreq->preply_len = preply_len;
4984
4985 ceph_oid_copy(&lreq->t.base_oid, oid);
4986 ceph_oloc_copy(&lreq->t.base_oloc, oloc);
4987 lreq->t.flags = CEPH_OSD_FLAG_READ;
4988
4989 linger_submit(lreq);
4990 ret = linger_reg_commit_wait(lreq);
4991 if (!ret)
4992 ret = linger_notify_finish_wait(lreq,
4993 msecs_to_jiffies(2 * timeout * MSEC_PER_SEC));
4994 else
4995 dout("lreq %p failed to initiate notify %d\n", lreq, ret);
4996
4997 linger_cancel(lreq);
4998out_put_lreq:
4999 linger_put(lreq);
5000 return ret;
5001}
5002EXPORT_SYMBOL(ceph_osdc_notify);
5003
5004static int decode_watcher(void **p, void *end, struct ceph_watch_item *item)
5005{
5006 u8 struct_v;
5007 u32 struct_len;
5008 int ret;
5009
5010 ret = ceph_start_decoding(p, end, 2, "watch_item_t",
5011 &struct_v, &struct_len);
5012 if (ret)
5013 goto bad;
5014
5015 ret = -EINVAL;
5016 ceph_decode_copy_safe(p, end, &item->name, sizeof(item->name), bad);
5017 ceph_decode_64_safe(p, end, item->cookie, bad);
5018 ceph_decode_skip_32(p, end, bad); /* skip timeout seconds */
5019
5020 if (struct_v >= 2) {
5021 ret = ceph_decode_entity_addr(p, end, &item->addr);
5022 if (ret)
5023 goto bad;
5024 } else {
5025 ret = 0;
5026 }
5027
5028 dout("%s %s%llu cookie %llu addr %s\n", __func__,
5029 ENTITY_NAME(item->name), item->cookie,
5030 ceph_pr_addr(&item->addr));
5031bad:
5032 return ret;
5033}
5034
5035static int decode_watchers(void **p, void *end,
5036 struct ceph_watch_item **watchers,
5037 u32 *num_watchers)
5038{
5039 u8 struct_v;
5040 u32 struct_len;
5041 int i;
5042 int ret;
5043
5044 ret = ceph_start_decoding(p, end, 1, "obj_list_watch_response_t",
5045 &struct_v, &struct_len);
5046 if (ret)
5047 return ret;
5048
5049 *num_watchers = ceph_decode_32(p);
5050 *watchers = kcalloc(*num_watchers, sizeof(**watchers), GFP_NOIO);
5051 if (!*watchers)
5052 return -ENOMEM;
5053
5054 for (i = 0; i < *num_watchers; i++) {
5055 ret = decode_watcher(p, end, *watchers + i);
5056 if (ret) {
5057 kfree(*watchers);
5058 return ret;
5059 }
5060 }
5061
5062 return 0;
5063}
5064
5065/*
5066 * On success, the caller is responsible for:
5067 *
5068 * kfree(watchers);
5069 */
5070int ceph_osdc_list_watchers(struct ceph_osd_client *osdc,
5071 struct ceph_object_id *oid,
5072 struct ceph_object_locator *oloc,
5073 struct ceph_watch_item **watchers,
5074 u32 *num_watchers)
5075{
5076 struct ceph_osd_request *req;
5077 struct page **pages;
5078 int ret;
5079
5080 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
5081 if (!req)
5082 return -ENOMEM;
5083
5084 ceph_oid_copy(&req->r_base_oid, oid);
5085 ceph_oloc_copy(&req->r_base_oloc, oloc);
5086 req->r_flags = CEPH_OSD_FLAG_READ;
5087
5088 pages = ceph_alloc_page_vector(1, GFP_NOIO);
5089 if (IS_ERR(pages)) {
5090 ret = PTR_ERR(pages);
5091 goto out_put_req;
5092 }
5093
5094 osd_req_op_init(req, 0, CEPH_OSD_OP_LIST_WATCHERS, 0);
5095 ceph_osd_data_pages_init(osd_req_op_data(req, 0, list_watchers,
5096 response_data),
5097 pages, PAGE_SIZE, 0, false, true);
5098
5099 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
5100 if (ret)
5101 goto out_put_req;
5102
5103 ceph_osdc_start_request(osdc, req);
5104 ret = ceph_osdc_wait_request(osdc, req);
5105 if (ret >= 0) {
5106 void *p = page_address(pages[0]);
5107 void *const end = p + req->r_ops[0].outdata_len;
5108
5109 ret = decode_watchers(&p, end, watchers, num_watchers);
5110 }
5111
5112out_put_req:
5113 ceph_osdc_put_request(req);
5114 return ret;
5115}
5116EXPORT_SYMBOL(ceph_osdc_list_watchers);
5117
5118/*
5119 * Call all pending notify callbacks - for use after a watch is
5120 * unregistered, to make sure no more callbacks for it will be invoked
5121 */
5122void ceph_osdc_flush_notifies(struct ceph_osd_client *osdc)
5123{
5124 dout("%s osdc %p\n", __func__, osdc);
5125 flush_workqueue(osdc->notify_wq);
5126}
5127EXPORT_SYMBOL(ceph_osdc_flush_notifies);
5128
5129void ceph_osdc_maybe_request_map(struct ceph_osd_client *osdc)
5130{
5131 down_read(&osdc->lock);
5132 maybe_request_map(osdc);
5133 up_read(&osdc->lock);
5134}
5135EXPORT_SYMBOL(ceph_osdc_maybe_request_map);
5136
5137/*
5138 * Execute an OSD class method on an object.
5139 *
5140 * @flags: CEPH_OSD_FLAG_*
5141 * @resp_len: in/out param for reply length
5142 */
5143int ceph_osdc_call(struct ceph_osd_client *osdc,
5144 struct ceph_object_id *oid,
5145 struct ceph_object_locator *oloc,
5146 const char *class, const char *method,
5147 unsigned int flags,
5148 struct page *req_page, size_t req_len,
5149 struct page **resp_pages, size_t *resp_len)
5150{
5151 struct ceph_osd_request *req;
5152 int ret;
5153
5154 if (req_len > PAGE_SIZE)
5155 return -E2BIG;
5156
5157 req = ceph_osdc_alloc_request(osdc, NULL, 1, false, GFP_NOIO);
5158 if (!req)
5159 return -ENOMEM;
5160
5161 ceph_oid_copy(&req->r_base_oid, oid);
5162 ceph_oloc_copy(&req->r_base_oloc, oloc);
5163 req->r_flags = flags;
5164
5165 ret = osd_req_op_cls_init(req, 0, class, method);
5166 if (ret)
5167 goto out_put_req;
5168
5169 if (req_page)
5170 osd_req_op_cls_request_data_pages(req, 0, &req_page, req_len,
5171 0, false, false);
5172 if (resp_pages)
5173 osd_req_op_cls_response_data_pages(req, 0, resp_pages,
5174 *resp_len, 0, false, false);
5175
5176 ret = ceph_osdc_alloc_messages(req, GFP_NOIO);
5177 if (ret)
5178 goto out_put_req;
5179
5180 ceph_osdc_start_request(osdc, req);
5181 ret = ceph_osdc_wait_request(osdc, req);
5182 if (ret >= 0) {
5183 ret = req->r_ops[0].rval;
5184 if (resp_pages)
5185 *resp_len = req->r_ops[0].outdata_len;
5186 }
5187
5188out_put_req:
5189 ceph_osdc_put_request(req);
5190 return ret;
5191}
5192EXPORT_SYMBOL(ceph_osdc_call);
5193
5194/*
5195 * reset all osd connections
5196 */
5197void ceph_osdc_reopen_osds(struct ceph_osd_client *osdc)
5198{
5199 struct rb_node *n;
5200
5201 down_write(&osdc->lock);
5202 for (n = rb_first(&osdc->osds); n; ) {
5203 struct ceph_osd *osd = rb_entry(n, struct ceph_osd, o_node);
5204
5205 n = rb_next(n);
5206 if (!reopen_osd(osd))
5207 kick_osd_requests(osd);
5208 }
5209 up_write(&osdc->lock);
5210}
5211
5212/*
5213 * init, shutdown
5214 */
5215int ceph_osdc_init(struct ceph_osd_client *osdc, struct ceph_client *client)
5216{
5217 int err;
5218
5219 dout("init\n");
5220 osdc->client = client;
5221 init_rwsem(&osdc->lock);
5222 osdc->osds = RB_ROOT;
5223 INIT_LIST_HEAD(&osdc->osd_lru);
5224 spin_lock_init(&osdc->osd_lru_lock);
5225 osd_init(&osdc->homeless_osd);
5226 osdc->homeless_osd.o_osdc = osdc;
5227 osdc->homeless_osd.o_osd = CEPH_HOMELESS_OSD;
5228 osdc->last_linger_id = CEPH_LINGER_ID_START;
5229 osdc->linger_requests = RB_ROOT;
5230 osdc->map_checks = RB_ROOT;
5231 osdc->linger_map_checks = RB_ROOT;
5232 INIT_DELAYED_WORK(&osdc->timeout_work, handle_timeout);
5233 INIT_DELAYED_WORK(&osdc->osds_timeout_work, handle_osds_timeout);
5234
5235 err = -ENOMEM;
5236 osdc->osdmap = ceph_osdmap_alloc();
5237 if (!osdc->osdmap)
5238 goto out;
5239
5240 osdc->req_mempool = mempool_create_slab_pool(10,
5241 ceph_osd_request_cache);
5242 if (!osdc->req_mempool)
5243 goto out_map;
5244
5245 err = ceph_msgpool_init(&osdc->msgpool_op, CEPH_MSG_OSD_OP,
5246 PAGE_SIZE, CEPH_OSD_SLAB_OPS, 10, "osd_op");
5247 if (err < 0)
5248 goto out_mempool;
5249 err = ceph_msgpool_init(&osdc->msgpool_op_reply, CEPH_MSG_OSD_OPREPLY,
5250 PAGE_SIZE, CEPH_OSD_SLAB_OPS, 10,
5251 "osd_op_reply");
5252 if (err < 0)
5253 goto out_msgpool;
5254
5255 err = -ENOMEM;
5256 osdc->notify_wq = create_singlethread_workqueue("ceph-watch-notify");
5257 if (!osdc->notify_wq)
5258 goto out_msgpool_reply;
5259
5260 osdc->completion_wq = create_singlethread_workqueue("ceph-completion");
5261 if (!osdc->completion_wq)
5262 goto out_notify_wq;
5263
5264 schedule_delayed_work(&osdc->timeout_work,
5265 osdc->client->options->osd_keepalive_timeout);
5266 schedule_delayed_work(&osdc->osds_timeout_work,
5267 round_jiffies_relative(osdc->client->options->osd_idle_ttl));
5268
5269 return 0;
5270
5271out_notify_wq:
5272 destroy_workqueue(osdc->notify_wq);
5273out_msgpool_reply:
5274 ceph_msgpool_destroy(&osdc->msgpool_op_reply);
5275out_msgpool:
5276 ceph_msgpool_destroy(&osdc->msgpool_op);
5277out_mempool:
5278 mempool_destroy(osdc->req_mempool);
5279out_map:
5280 ceph_osdmap_destroy(osdc->osdmap);
5281out:
5282 return err;
5283}
5284
5285void ceph_osdc_stop(struct ceph_osd_client *osdc)
5286{
5287 destroy_workqueue(osdc->completion_wq);
5288 destroy_workqueue(osdc->notify_wq);
5289 cancel_delayed_work_sync(&osdc->timeout_work);
5290 cancel_delayed_work_sync(&osdc->osds_timeout_work);
5291
5292 down_write(&osdc->lock);
5293 while (!RB_EMPTY_ROOT(&osdc->osds)) {
5294 struct ceph_osd *osd = rb_entry(rb_first(&osdc->osds),
5295 struct ceph_osd, o_node);
5296 close_osd(osd);
5297 }
5298 up_write(&osdc->lock);
5299 WARN_ON(refcount_read(&osdc->homeless_osd.o_ref) != 1);
5300 osd_cleanup(&osdc->homeless_osd);
5301
5302 WARN_ON(!list_empty(&osdc->osd_lru));
5303 WARN_ON(!RB_EMPTY_ROOT(&osdc->linger_requests));
5304 WARN_ON(!RB_EMPTY_ROOT(&osdc->map_checks));
5305 WARN_ON(!RB_EMPTY_ROOT(&osdc->linger_map_checks));
5306 WARN_ON(atomic_read(&osdc->num_requests));
5307 WARN_ON(atomic_read(&osdc->num_homeless));
5308
5309 ceph_osdmap_destroy(osdc->osdmap);
5310 mempool_destroy(osdc->req_mempool);
5311 ceph_msgpool_destroy(&osdc->msgpool_op);
5312 ceph_msgpool_destroy(&osdc->msgpool_op_reply);
5313}
5314
5315int osd_req_op_copy_from_init(struct ceph_osd_request *req,
5316 u64 src_snapid, u64 src_version,
5317 struct ceph_object_id *src_oid,
5318 struct ceph_object_locator *src_oloc,
5319 u32 src_fadvise_flags,
5320 u32 dst_fadvise_flags,
5321 u32 truncate_seq, u64 truncate_size,
5322 u8 copy_from_flags)
5323{
5324 struct ceph_osd_req_op *op;
5325 struct page **pages;
5326 void *p, *end;
5327
5328 pages = ceph_alloc_page_vector(1, GFP_KERNEL);
5329 if (IS_ERR(pages))
5330 return PTR_ERR(pages);
5331
5332 op = osd_req_op_init(req, 0, CEPH_OSD_OP_COPY_FROM2,
5333 dst_fadvise_flags);
5334 op->copy_from.snapid = src_snapid;
5335 op->copy_from.src_version = src_version;
5336 op->copy_from.flags = copy_from_flags;
5337 op->copy_from.src_fadvise_flags = src_fadvise_flags;
5338
5339 p = page_address(pages[0]);
5340 end = p + PAGE_SIZE;
5341 ceph_encode_string(&p, end, src_oid->name, src_oid->name_len);
5342 encode_oloc(&p, end, src_oloc);
5343 ceph_encode_32(&p, truncate_seq);
5344 ceph_encode_64(&p, truncate_size);
5345 op->indata_len = PAGE_SIZE - (end - p);
5346
5347 ceph_osd_data_pages_init(&op->copy_from.osd_data, pages,
5348 op->indata_len, 0, false, true);
5349 return 0;
5350}
5351EXPORT_SYMBOL(osd_req_op_copy_from_init);
5352
5353int __init ceph_osdc_setup(void)
5354{
5355 size_t size = sizeof(struct ceph_osd_request) +
5356 CEPH_OSD_SLAB_OPS * sizeof(struct ceph_osd_req_op);
5357
5358 BUG_ON(ceph_osd_request_cache);
5359 ceph_osd_request_cache = kmem_cache_create("ceph_osd_request", size,
5360 0, 0, NULL);
5361
5362 return ceph_osd_request_cache ? 0 : -ENOMEM;
5363}
5364
5365void ceph_osdc_cleanup(void)
5366{
5367 BUG_ON(!ceph_osd_request_cache);
5368 kmem_cache_destroy(ceph_osd_request_cache);
5369 ceph_osd_request_cache = NULL;
5370}
5371
5372/*
5373 * handle incoming message
5374 */
5375static void osd_dispatch(struct ceph_connection *con, struct ceph_msg *msg)
5376{
5377 struct ceph_osd *osd = con->private;
5378 struct ceph_osd_client *osdc = osd->o_osdc;
5379 int type = le16_to_cpu(msg->hdr.type);
5380
5381 switch (type) {
5382 case CEPH_MSG_OSD_MAP:
5383 ceph_osdc_handle_map(osdc, msg);
5384 break;
5385 case CEPH_MSG_OSD_OPREPLY:
5386 handle_reply(osd, msg);
5387 break;
5388 case CEPH_MSG_OSD_BACKOFF:
5389 handle_backoff(osd, msg);
5390 break;
5391 case CEPH_MSG_WATCH_NOTIFY:
5392 handle_watch_notify(osdc, msg);
5393 break;
5394
5395 default:
5396 pr_err("received unknown message type %d %s\n", type,
5397 ceph_msg_type_name(type));
5398 }
5399
5400 ceph_msg_put(msg);
5401}
5402
5403/* How much sparse data was requested? */
5404static u64 sparse_data_requested(struct ceph_osd_request *req)
5405{
5406 u64 len = 0;
5407
5408 if (req->r_flags & CEPH_OSD_FLAG_READ) {
5409 int i;
5410
5411 for (i = 0; i < req->r_num_ops; ++i) {
5412 struct ceph_osd_req_op *op = &req->r_ops[i];
5413
5414 if (op->op == CEPH_OSD_OP_SPARSE_READ)
5415 len += op->extent.length;
5416 }
5417 }
5418 return len;
5419}
5420
5421/*
5422 * Lookup and return message for incoming reply. Don't try to do
5423 * anything about a larger than preallocated data portion of the
5424 * message at the moment - for now, just skip the message.
5425 */
5426static struct ceph_msg *get_reply(struct ceph_connection *con,
5427 struct ceph_msg_header *hdr,
5428 int *skip)
5429{
5430 struct ceph_osd *osd = con->private;
5431 struct ceph_osd_client *osdc = osd->o_osdc;
5432 struct ceph_msg *m = NULL;
5433 struct ceph_osd_request *req;
5434 int front_len = le32_to_cpu(hdr->front_len);
5435 int data_len = le32_to_cpu(hdr->data_len);
5436 u64 tid = le64_to_cpu(hdr->tid);
5437 u64 srlen;
5438
5439 down_read(&osdc->lock);
5440 if (!osd_registered(osd)) {
5441 dout("%s osd%d unknown, skipping\n", __func__, osd->o_osd);
5442 *skip = 1;
5443 goto out_unlock_osdc;
5444 }
5445 WARN_ON(osd->o_osd != le64_to_cpu(hdr->src.num));
5446
5447 mutex_lock(&osd->lock);
5448 req = lookup_request(&osd->o_requests, tid);
5449 if (!req) {
5450 dout("%s osd%d tid %llu unknown, skipping\n", __func__,
5451 osd->o_osd, tid);
5452 *skip = 1;
5453 goto out_unlock_session;
5454 }
5455
5456 ceph_msg_revoke_incoming(req->r_reply);
5457
5458 if (front_len > req->r_reply->front_alloc_len) {
5459 pr_warn("%s osd%d tid %llu front %d > preallocated %d\n",
5460 __func__, osd->o_osd, req->r_tid, front_len,
5461 req->r_reply->front_alloc_len);
5462 m = ceph_msg_new(CEPH_MSG_OSD_OPREPLY, front_len, GFP_NOFS,
5463 false);
5464 if (!m)
5465 goto out_unlock_session;
5466 ceph_msg_put(req->r_reply);
5467 req->r_reply = m;
5468 }
5469
5470 srlen = sparse_data_requested(req);
5471 if (!srlen && data_len > req->r_reply->data_length) {
5472 pr_warn("%s osd%d tid %llu data %d > preallocated %zu, skipping\n",
5473 __func__, osd->o_osd, req->r_tid, data_len,
5474 req->r_reply->data_length);
5475 m = NULL;
5476 *skip = 1;
5477 goto out_unlock_session;
5478 }
5479
5480 m = ceph_msg_get(req->r_reply);
5481 m->sparse_read_total = srlen;
5482
5483 dout("get_reply tid %lld %p\n", tid, m);
5484
5485out_unlock_session:
5486 mutex_unlock(&osd->lock);
5487out_unlock_osdc:
5488 up_read(&osdc->lock);
5489 return m;
5490}
5491
5492static struct ceph_msg *alloc_msg_with_page_vector(struct ceph_msg_header *hdr)
5493{
5494 struct ceph_msg *m;
5495 int type = le16_to_cpu(hdr->type);
5496 u32 front_len = le32_to_cpu(hdr->front_len);
5497 u32 data_len = le32_to_cpu(hdr->data_len);
5498
5499 m = ceph_msg_new2(type, front_len, 1, GFP_NOIO, false);
5500 if (!m)
5501 return NULL;
5502
5503 if (data_len) {
5504 struct page **pages;
5505
5506 pages = ceph_alloc_page_vector(calc_pages_for(0, data_len),
5507 GFP_NOIO);
5508 if (IS_ERR(pages)) {
5509 ceph_msg_put(m);
5510 return NULL;
5511 }
5512
5513 ceph_msg_data_add_pages(m, pages, data_len, 0, true);
5514 }
5515
5516 return m;
5517}
5518
5519static struct ceph_msg *osd_alloc_msg(struct ceph_connection *con,
5520 struct ceph_msg_header *hdr,
5521 int *skip)
5522{
5523 struct ceph_osd *osd = con->private;
5524 int type = le16_to_cpu(hdr->type);
5525
5526 *skip = 0;
5527 switch (type) {
5528 case CEPH_MSG_OSD_MAP:
5529 case CEPH_MSG_OSD_BACKOFF:
5530 case CEPH_MSG_WATCH_NOTIFY:
5531 return alloc_msg_with_page_vector(hdr);
5532 case CEPH_MSG_OSD_OPREPLY:
5533 return get_reply(con, hdr, skip);
5534 default:
5535 pr_warn("%s osd%d unknown msg type %d, skipping\n", __func__,
5536 osd->o_osd, type);
5537 *skip = 1;
5538 return NULL;
5539 }
5540}
5541
5542/*
5543 * Wrappers to refcount containing ceph_osd struct
5544 */
5545static struct ceph_connection *osd_get_con(struct ceph_connection *con)
5546{
5547 struct ceph_osd *osd = con->private;
5548 if (get_osd(osd))
5549 return con;
5550 return NULL;
5551}
5552
5553static void osd_put_con(struct ceph_connection *con)
5554{
5555 struct ceph_osd *osd = con->private;
5556 put_osd(osd);
5557}
5558
5559/*
5560 * authentication
5561 */
5562
5563/*
5564 * Note: returned pointer is the address of a structure that's
5565 * managed separately. Caller must *not* attempt to free it.
5566 */
5567static struct ceph_auth_handshake *
5568osd_get_authorizer(struct ceph_connection *con, int *proto, int force_new)
5569{
5570 struct ceph_osd *o = con->private;
5571 struct ceph_osd_client *osdc = o->o_osdc;
5572 struct ceph_auth_client *ac = osdc->client->monc.auth;
5573 struct ceph_auth_handshake *auth = &o->o_auth;
5574 int ret;
5575
5576 ret = __ceph_auth_get_authorizer(ac, auth, CEPH_ENTITY_TYPE_OSD,
5577 force_new, proto, NULL, NULL);
5578 if (ret)
5579 return ERR_PTR(ret);
5580
5581 return auth;
5582}
5583
5584static int osd_add_authorizer_challenge(struct ceph_connection *con,
5585 void *challenge_buf, int challenge_buf_len)
5586{
5587 struct ceph_osd *o = con->private;
5588 struct ceph_osd_client *osdc = o->o_osdc;
5589 struct ceph_auth_client *ac = osdc->client->monc.auth;
5590
5591 return ceph_auth_add_authorizer_challenge(ac, o->o_auth.authorizer,
5592 challenge_buf, challenge_buf_len);
5593}
5594
5595static int osd_verify_authorizer_reply(struct ceph_connection *con)
5596{
5597 struct ceph_osd *o = con->private;
5598 struct ceph_osd_client *osdc = o->o_osdc;
5599 struct ceph_auth_client *ac = osdc->client->monc.auth;
5600 struct ceph_auth_handshake *auth = &o->o_auth;
5601
5602 return ceph_auth_verify_authorizer_reply(ac, auth->authorizer,
5603 auth->authorizer_reply_buf, auth->authorizer_reply_buf_len,
5604 NULL, NULL, NULL, NULL);
5605}
5606
5607static int osd_invalidate_authorizer(struct ceph_connection *con)
5608{
5609 struct ceph_osd *o = con->private;
5610 struct ceph_osd_client *osdc = o->o_osdc;
5611 struct ceph_auth_client *ac = osdc->client->monc.auth;
5612
5613 ceph_auth_invalidate_authorizer(ac, CEPH_ENTITY_TYPE_OSD);
5614 return ceph_monc_validate_auth(&osdc->client->monc);
5615}
5616
5617static int osd_get_auth_request(struct ceph_connection *con,
5618 void *buf, int *buf_len,
5619 void **authorizer, int *authorizer_len)
5620{
5621 struct ceph_osd *o = con->private;
5622 struct ceph_auth_client *ac = o->o_osdc->client->monc.auth;
5623 struct ceph_auth_handshake *auth = &o->o_auth;
5624 int ret;
5625
5626 ret = ceph_auth_get_authorizer(ac, auth, CEPH_ENTITY_TYPE_OSD,
5627 buf, buf_len);
5628 if (ret)
5629 return ret;
5630
5631 *authorizer = auth->authorizer_buf;
5632 *authorizer_len = auth->authorizer_buf_len;
5633 return 0;
5634}
5635
5636static int osd_handle_auth_reply_more(struct ceph_connection *con,
5637 void *reply, int reply_len,
5638 void *buf, int *buf_len,
5639 void **authorizer, int *authorizer_len)
5640{
5641 struct ceph_osd *o = con->private;
5642 struct ceph_auth_client *ac = o->o_osdc->client->monc.auth;
5643 struct ceph_auth_handshake *auth = &o->o_auth;
5644 int ret;
5645
5646 ret = ceph_auth_handle_svc_reply_more(ac, auth, reply, reply_len,
5647 buf, buf_len);
5648 if (ret)
5649 return ret;
5650
5651 *authorizer = auth->authorizer_buf;
5652 *authorizer_len = auth->authorizer_buf_len;
5653 return 0;
5654}
5655
5656static int osd_handle_auth_done(struct ceph_connection *con,
5657 u64 global_id, void *reply, int reply_len,
5658 u8 *session_key, int *session_key_len,
5659 u8 *con_secret, int *con_secret_len)
5660{
5661 struct ceph_osd *o = con->private;
5662 struct ceph_auth_client *ac = o->o_osdc->client->monc.auth;
5663 struct ceph_auth_handshake *auth = &o->o_auth;
5664
5665 return ceph_auth_handle_svc_reply_done(ac, auth, reply, reply_len,
5666 session_key, session_key_len,
5667 con_secret, con_secret_len);
5668}
5669
5670static int osd_handle_auth_bad_method(struct ceph_connection *con,
5671 int used_proto, int result,
5672 const int *allowed_protos, int proto_cnt,
5673 const int *allowed_modes, int mode_cnt)
5674{
5675 struct ceph_osd *o = con->private;
5676 struct ceph_mon_client *monc = &o->o_osdc->client->monc;
5677 int ret;
5678
5679 if (ceph_auth_handle_bad_authorizer(monc->auth, CEPH_ENTITY_TYPE_OSD,
5680 used_proto, result,
5681 allowed_protos, proto_cnt,
5682 allowed_modes, mode_cnt)) {
5683 ret = ceph_monc_validate_auth(monc);
5684 if (ret)
5685 return ret;
5686 }
5687
5688 return -EACCES;
5689}
5690
5691static void osd_reencode_message(struct ceph_msg *msg)
5692{
5693 int type = le16_to_cpu(msg->hdr.type);
5694
5695 if (type == CEPH_MSG_OSD_OP)
5696 encode_request_finish(msg);
5697}
5698
5699static int osd_sign_message(struct ceph_msg *msg)
5700{
5701 struct ceph_osd *o = msg->con->private;
5702 struct ceph_auth_handshake *auth = &o->o_auth;
5703
5704 return ceph_auth_sign_message(auth, msg);
5705}
5706
5707static int osd_check_message_signature(struct ceph_msg *msg)
5708{
5709 struct ceph_osd *o = msg->con->private;
5710 struct ceph_auth_handshake *auth = &o->o_auth;
5711
5712 return ceph_auth_check_message_signature(auth, msg);
5713}
5714
5715static void advance_cursor(struct ceph_msg_data_cursor *cursor, size_t len,
5716 bool zero)
5717{
5718 while (len) {
5719 struct page *page;
5720 size_t poff, plen;
5721
5722 page = ceph_msg_data_next(cursor, &poff, &plen);
5723 if (plen > len)
5724 plen = len;
5725 if (zero)
5726 zero_user_segment(page, poff, poff + plen);
5727 len -= plen;
5728 ceph_msg_data_advance(cursor, plen);
5729 }
5730}
5731
5732static int prep_next_sparse_read(struct ceph_connection *con,
5733 struct ceph_msg_data_cursor *cursor)
5734{
5735 struct ceph_osd *o = con->private;
5736 struct ceph_sparse_read *sr = &o->o_sparse_read;
5737 struct ceph_osd_request *req;
5738 struct ceph_osd_req_op *op;
5739
5740 spin_lock(&o->o_requests_lock);
5741 req = lookup_request(&o->o_requests, le64_to_cpu(con->in_msg->hdr.tid));
5742 if (!req) {
5743 spin_unlock(&o->o_requests_lock);
5744 return -EBADR;
5745 }
5746
5747 if (o->o_sparse_op_idx < 0) {
5748 dout("%s: [%d] starting new sparse read req\n",
5749 __func__, o->o_osd);
5750 } else {
5751 u64 end;
5752
5753 op = &req->r_ops[o->o_sparse_op_idx];
5754
5755 WARN_ON_ONCE(op->extent.sparse_ext);
5756
5757 /* hand back buffer we took earlier */
5758 op->extent.sparse_ext = sr->sr_extent;
5759 sr->sr_extent = NULL;
5760 op->extent.sparse_ext_cnt = sr->sr_count;
5761 sr->sr_ext_len = 0;
5762 dout("%s: [%d] completed extent array len %d cursor->resid %zd\n",
5763 __func__, o->o_osd, op->extent.sparse_ext_cnt, cursor->resid);
5764 /* Advance to end of data for this operation */
5765 end = ceph_sparse_ext_map_end(op);
5766 if (end < sr->sr_req_len)
5767 advance_cursor(cursor, sr->sr_req_len - end, false);
5768 }
5769
5770 ceph_init_sparse_read(sr);
5771
5772 /* find next op in this request (if any) */
5773 while (++o->o_sparse_op_idx < req->r_num_ops) {
5774 op = &req->r_ops[o->o_sparse_op_idx];
5775 if (op->op == CEPH_OSD_OP_SPARSE_READ)
5776 goto found;
5777 }
5778
5779 /* reset for next sparse read request */
5780 spin_unlock(&o->o_requests_lock);
5781 o->o_sparse_op_idx = -1;
5782 return 0;
5783found:
5784 sr->sr_req_off = op->extent.offset;
5785 sr->sr_req_len = op->extent.length;
5786 sr->sr_pos = sr->sr_req_off;
5787 dout("%s: [%d] new sparse read op at idx %d 0x%llx~0x%llx\n", __func__,
5788 o->o_osd, o->o_sparse_op_idx, sr->sr_req_off, sr->sr_req_len);
5789
5790 /* hand off request's sparse extent map buffer */
5791 sr->sr_ext_len = op->extent.sparse_ext_cnt;
5792 op->extent.sparse_ext_cnt = 0;
5793 sr->sr_extent = op->extent.sparse_ext;
5794 op->extent.sparse_ext = NULL;
5795
5796 spin_unlock(&o->o_requests_lock);
5797 return 1;
5798}
5799
5800#ifdef __BIG_ENDIAN
5801static inline void convert_extent_map(struct ceph_sparse_read *sr)
5802{
5803 int i;
5804
5805 for (i = 0; i < sr->sr_count; i++) {
5806 struct ceph_sparse_extent *ext = &sr->sr_extent[i];
5807
5808 ext->off = le64_to_cpu((__force __le64)ext->off);
5809 ext->len = le64_to_cpu((__force __le64)ext->len);
5810 }
5811}
5812#else
5813static inline void convert_extent_map(struct ceph_sparse_read *sr)
5814{
5815}
5816#endif
5817
5818static int osd_sparse_read(struct ceph_connection *con,
5819 struct ceph_msg_data_cursor *cursor,
5820 char **pbuf)
5821{
5822 struct ceph_osd *o = con->private;
5823 struct ceph_sparse_read *sr = &o->o_sparse_read;
5824 u32 count = sr->sr_count;
5825 u64 eoff, elen, len = 0;
5826 int i, ret;
5827
5828 switch (sr->sr_state) {
5829 case CEPH_SPARSE_READ_HDR:
5830next_op:
5831 ret = prep_next_sparse_read(con, cursor);
5832 if (ret <= 0)
5833 return ret;
5834
5835 /* number of extents */
5836 ret = sizeof(sr->sr_count);
5837 *pbuf = (char *)&sr->sr_count;
5838 sr->sr_state = CEPH_SPARSE_READ_EXTENTS;
5839 break;
5840 case CEPH_SPARSE_READ_EXTENTS:
5841 /* Convert sr_count to host-endian */
5842 count = le32_to_cpu((__force __le32)sr->sr_count);
5843 sr->sr_count = count;
5844 dout("[%d] got %u extents\n", o->o_osd, count);
5845
5846 if (count > 0) {
5847 if (!sr->sr_extent || count > sr->sr_ext_len) {
5848 /* no extent array provided, or too short */
5849 kfree(sr->sr_extent);
5850 sr->sr_extent = kmalloc_array(count,
5851 sizeof(*sr->sr_extent),
5852 GFP_NOIO);
5853 if (!sr->sr_extent) {
5854 pr_err("%s: failed to allocate %u extents\n",
5855 __func__, count);
5856 return -ENOMEM;
5857 }
5858 sr->sr_ext_len = count;
5859 }
5860 ret = count * sizeof(*sr->sr_extent);
5861 *pbuf = (char *)sr->sr_extent;
5862 sr->sr_state = CEPH_SPARSE_READ_DATA_LEN;
5863 break;
5864 }
5865 /* No extents? Read data len */
5866 fallthrough;
5867 case CEPH_SPARSE_READ_DATA_LEN:
5868 convert_extent_map(sr);
5869 ret = sizeof(sr->sr_datalen);
5870 *pbuf = (char *)&sr->sr_datalen;
5871 sr->sr_state = CEPH_SPARSE_READ_DATA_PRE;
5872 break;
5873 case CEPH_SPARSE_READ_DATA_PRE:
5874 /* Convert sr_datalen to host-endian */
5875 sr->sr_datalen = le32_to_cpu((__force __le32)sr->sr_datalen);
5876 for (i = 0; i < count; i++)
5877 len += sr->sr_extent[i].len;
5878 if (sr->sr_datalen != len) {
5879 pr_warn_ratelimited("data len %u != extent len %llu\n",
5880 sr->sr_datalen, len);
5881 return -EREMOTEIO;
5882 }
5883 sr->sr_state = CEPH_SPARSE_READ_DATA;
5884 fallthrough;
5885 case CEPH_SPARSE_READ_DATA:
5886 if (sr->sr_index >= count) {
5887 sr->sr_state = CEPH_SPARSE_READ_HDR;
5888 goto next_op;
5889 }
5890
5891 eoff = sr->sr_extent[sr->sr_index].off;
5892 elen = sr->sr_extent[sr->sr_index].len;
5893
5894 dout("[%d] ext %d off 0x%llx len 0x%llx\n",
5895 o->o_osd, sr->sr_index, eoff, elen);
5896
5897 if (elen > INT_MAX) {
5898 dout("Sparse read extent length too long (0x%llx)\n",
5899 elen);
5900 return -EREMOTEIO;
5901 }
5902
5903 /* zero out anything from sr_pos to start of extent */
5904 if (sr->sr_pos < eoff)
5905 advance_cursor(cursor, eoff - sr->sr_pos, true);
5906
5907 /* Set position to end of extent */
5908 sr->sr_pos = eoff + elen;
5909
5910 /* send back the new length and nullify the ptr */
5911 cursor->sr_resid = elen;
5912 ret = elen;
5913 *pbuf = NULL;
5914
5915 /* Bump the array index */
5916 ++sr->sr_index;
5917 break;
5918 }
5919 return ret;
5920}
5921
5922static const struct ceph_connection_operations osd_con_ops = {
5923 .get = osd_get_con,
5924 .put = osd_put_con,
5925 .sparse_read = osd_sparse_read,
5926 .alloc_msg = osd_alloc_msg,
5927 .dispatch = osd_dispatch,
5928 .fault = osd_fault,
5929 .reencode_message = osd_reencode_message,
5930 .get_authorizer = osd_get_authorizer,
5931 .add_authorizer_challenge = osd_add_authorizer_challenge,
5932 .verify_authorizer_reply = osd_verify_authorizer_reply,
5933 .invalidate_authorizer = osd_invalidate_authorizer,
5934 .sign_message = osd_sign_message,
5935 .check_message_signature = osd_check_message_signature,
5936 .get_auth_request = osd_get_auth_request,
5937 .handle_auth_reply_more = osd_handle_auth_reply_more,
5938 .handle_auth_done = osd_handle_auth_done,
5939 .handle_auth_bad_method = osd_handle_auth_bad_method,
5940};