libceph: fully initialize connection in con_init()
[linux-2.6-block.git] / net / ceph / mon_client.c
... / ...
CommitLineData
1#include <linux/ceph/ceph_debug.h>
2
3#include <linux/module.h>
4#include <linux/types.h>
5#include <linux/slab.h>
6#include <linux/random.h>
7#include <linux/sched.h>
8
9#include <linux/ceph/mon_client.h>
10#include <linux/ceph/libceph.h>
11#include <linux/ceph/debugfs.h>
12#include <linux/ceph/decode.h>
13#include <linux/ceph/auth.h>
14
15/*
16 * Interact with Ceph monitor cluster. Handle requests for new map
17 * versions, and periodically resend as needed. Also implement
18 * statfs() and umount().
19 *
20 * A small cluster of Ceph "monitors" are responsible for managing critical
21 * cluster configuration and state information. An odd number (e.g., 3, 5)
22 * of cmon daemons use a modified version of the Paxos part-time parliament
23 * algorithm to manage the MDS map (mds cluster membership), OSD map, and
24 * list of clients who have mounted the file system.
25 *
26 * We maintain an open, active session with a monitor at all times in order to
27 * receive timely MDSMap updates. We periodically send a keepalive byte on the
28 * TCP socket to ensure we detect a failure. If the connection does break, we
29 * randomly hunt for a new monitor. Once the connection is reestablished, we
30 * resend any outstanding requests.
31 */
32
33static const struct ceph_connection_operations mon_con_ops;
34
35static int __validate_auth(struct ceph_mon_client *monc);
36
37/*
38 * Decode a monmap blob (e.g., during mount).
39 */
40struct ceph_monmap *ceph_monmap_decode(void *p, void *end)
41{
42 struct ceph_monmap *m = NULL;
43 int i, err = -EINVAL;
44 struct ceph_fsid fsid;
45 u32 epoch, num_mon;
46 u16 version;
47 u32 len;
48
49 ceph_decode_32_safe(&p, end, len, bad);
50 ceph_decode_need(&p, end, len, bad);
51
52 dout("monmap_decode %p %p len %d\n", p, end, (int)(end-p));
53
54 ceph_decode_16_safe(&p, end, version, bad);
55
56 ceph_decode_need(&p, end, sizeof(fsid) + 2*sizeof(u32), bad);
57 ceph_decode_copy(&p, &fsid, sizeof(fsid));
58 epoch = ceph_decode_32(&p);
59
60 num_mon = ceph_decode_32(&p);
61 ceph_decode_need(&p, end, num_mon*sizeof(m->mon_inst[0]), bad);
62
63 if (num_mon >= CEPH_MAX_MON)
64 goto bad;
65 m = kmalloc(sizeof(*m) + sizeof(m->mon_inst[0])*num_mon, GFP_NOFS);
66 if (m == NULL)
67 return ERR_PTR(-ENOMEM);
68 m->fsid = fsid;
69 m->epoch = epoch;
70 m->num_mon = num_mon;
71 ceph_decode_copy(&p, m->mon_inst, num_mon*sizeof(m->mon_inst[0]));
72 for (i = 0; i < num_mon; i++)
73 ceph_decode_addr(&m->mon_inst[i].addr);
74
75 dout("monmap_decode epoch %d, num_mon %d\n", m->epoch,
76 m->num_mon);
77 for (i = 0; i < m->num_mon; i++)
78 dout("monmap_decode mon%d is %s\n", i,
79 ceph_pr_addr(&m->mon_inst[i].addr.in_addr));
80 return m;
81
82bad:
83 dout("monmap_decode failed with %d\n", err);
84 kfree(m);
85 return ERR_PTR(err);
86}
87
88/*
89 * return true if *addr is included in the monmap.
90 */
91int ceph_monmap_contains(struct ceph_monmap *m, struct ceph_entity_addr *addr)
92{
93 int i;
94
95 for (i = 0; i < m->num_mon; i++)
96 if (memcmp(addr, &m->mon_inst[i].addr, sizeof(*addr)) == 0)
97 return 1;
98 return 0;
99}
100
101/*
102 * Send an auth request.
103 */
104static void __send_prepared_auth_request(struct ceph_mon_client *monc, int len)
105{
106 monc->pending_auth = 1;
107 monc->m_auth->front.iov_len = len;
108 monc->m_auth->hdr.front_len = cpu_to_le32(len);
109 ceph_con_revoke(&monc->con, monc->m_auth);
110 ceph_msg_get(monc->m_auth); /* keep our ref */
111 ceph_con_send(&monc->con, monc->m_auth);
112}
113
114/*
115 * Close monitor session, if any.
116 */
117static void __close_session(struct ceph_mon_client *monc)
118{
119 dout("__close_session closing mon%d\n", monc->cur_mon);
120 ceph_con_revoke(&monc->con, monc->m_auth);
121 ceph_con_close(&monc->con);
122 monc->con.private = NULL;
123 monc->cur_mon = -1;
124 monc->pending_auth = 0;
125 ceph_auth_reset(monc->auth);
126}
127
128/*
129 * Open a session with a (new) monitor.
130 */
131static int __open_session(struct ceph_mon_client *monc)
132{
133 char r;
134 int ret;
135
136 if (monc->cur_mon < 0) {
137 get_random_bytes(&r, 1);
138 monc->cur_mon = r % monc->monmap->num_mon;
139 dout("open_session num=%d r=%d -> mon%d\n",
140 monc->monmap->num_mon, r, monc->cur_mon);
141 monc->sub_sent = 0;
142 monc->sub_renew_after = jiffies; /* i.e., expired */
143 monc->want_next_osdmap = !!monc->want_next_osdmap;
144
145 ceph_con_init(&monc->con, monc, &mon_con_ops,
146 &monc->client->msgr,
147 CEPH_ENTITY_TYPE_MON, monc->cur_mon);
148
149 dout("open_session mon%d opening\n", monc->cur_mon);
150 ceph_con_open(&monc->con,
151 &monc->monmap->mon_inst[monc->cur_mon].addr);
152
153 /* initiatiate authentication handshake */
154 ret = ceph_auth_build_hello(monc->auth,
155 monc->m_auth->front.iov_base,
156 monc->m_auth->front_max);
157 __send_prepared_auth_request(monc, ret);
158 } else {
159 dout("open_session mon%d already open\n", monc->cur_mon);
160 }
161 return 0;
162}
163
164static bool __sub_expired(struct ceph_mon_client *monc)
165{
166 return time_after_eq(jiffies, monc->sub_renew_after);
167}
168
169/*
170 * Reschedule delayed work timer.
171 */
172static void __schedule_delayed(struct ceph_mon_client *monc)
173{
174 unsigned delay;
175
176 if (monc->cur_mon < 0 || __sub_expired(monc))
177 delay = 10 * HZ;
178 else
179 delay = 20 * HZ;
180 dout("__schedule_delayed after %u\n", delay);
181 schedule_delayed_work(&monc->delayed_work, delay);
182}
183
184/*
185 * Send subscribe request for mdsmap and/or osdmap.
186 */
187static void __send_subscribe(struct ceph_mon_client *monc)
188{
189 dout("__send_subscribe sub_sent=%u exp=%u want_osd=%d\n",
190 (unsigned)monc->sub_sent, __sub_expired(monc),
191 monc->want_next_osdmap);
192 if ((__sub_expired(monc) && !monc->sub_sent) ||
193 monc->want_next_osdmap == 1) {
194 struct ceph_msg *msg = monc->m_subscribe;
195 struct ceph_mon_subscribe_item *i;
196 void *p, *end;
197 int num;
198
199 p = msg->front.iov_base;
200 end = p + msg->front_max;
201
202 num = 1 + !!monc->want_next_osdmap + !!monc->want_mdsmap;
203 ceph_encode_32(&p, num);
204
205 if (monc->want_next_osdmap) {
206 dout("__send_subscribe to 'osdmap' %u\n",
207 (unsigned)monc->have_osdmap);
208 ceph_encode_string(&p, end, "osdmap", 6);
209 i = p;
210 i->have = cpu_to_le64(monc->have_osdmap);
211 i->onetime = 1;
212 p += sizeof(*i);
213 monc->want_next_osdmap = 2; /* requested */
214 }
215 if (monc->want_mdsmap) {
216 dout("__send_subscribe to 'mdsmap' %u+\n",
217 (unsigned)monc->have_mdsmap);
218 ceph_encode_string(&p, end, "mdsmap", 6);
219 i = p;
220 i->have = cpu_to_le64(monc->have_mdsmap);
221 i->onetime = 0;
222 p += sizeof(*i);
223 }
224 ceph_encode_string(&p, end, "monmap", 6);
225 i = p;
226 i->have = 0;
227 i->onetime = 0;
228 p += sizeof(*i);
229
230 msg->front.iov_len = p - msg->front.iov_base;
231 msg->hdr.front_len = cpu_to_le32(msg->front.iov_len);
232 ceph_con_revoke(&monc->con, msg);
233 ceph_con_send(&monc->con, ceph_msg_get(msg));
234
235 monc->sub_sent = jiffies | 1; /* never 0 */
236 }
237}
238
239static void handle_subscribe_ack(struct ceph_mon_client *monc,
240 struct ceph_msg *msg)
241{
242 unsigned seconds;
243 struct ceph_mon_subscribe_ack *h = msg->front.iov_base;
244
245 if (msg->front.iov_len < sizeof(*h))
246 goto bad;
247 seconds = le32_to_cpu(h->duration);
248
249 mutex_lock(&monc->mutex);
250 if (monc->hunting) {
251 pr_info("mon%d %s session established\n",
252 monc->cur_mon,
253 ceph_pr_addr(&monc->con.peer_addr.in_addr));
254 monc->hunting = false;
255 }
256 dout("handle_subscribe_ack after %d seconds\n", seconds);
257 monc->sub_renew_after = monc->sub_sent + (seconds >> 1)*HZ - 1;
258 monc->sub_sent = 0;
259 mutex_unlock(&monc->mutex);
260 return;
261bad:
262 pr_err("got corrupt subscribe-ack msg\n");
263 ceph_msg_dump(msg);
264}
265
266/*
267 * Keep track of which maps we have
268 */
269int ceph_monc_got_mdsmap(struct ceph_mon_client *monc, u32 got)
270{
271 mutex_lock(&monc->mutex);
272 monc->have_mdsmap = got;
273 mutex_unlock(&monc->mutex);
274 return 0;
275}
276EXPORT_SYMBOL(ceph_monc_got_mdsmap);
277
278int ceph_monc_got_osdmap(struct ceph_mon_client *monc, u32 got)
279{
280 mutex_lock(&monc->mutex);
281 monc->have_osdmap = got;
282 monc->want_next_osdmap = 0;
283 mutex_unlock(&monc->mutex);
284 return 0;
285}
286
287/*
288 * Register interest in the next osdmap
289 */
290void ceph_monc_request_next_osdmap(struct ceph_mon_client *monc)
291{
292 dout("request_next_osdmap have %u\n", monc->have_osdmap);
293 mutex_lock(&monc->mutex);
294 if (!monc->want_next_osdmap)
295 monc->want_next_osdmap = 1;
296 if (monc->want_next_osdmap < 2)
297 __send_subscribe(monc);
298 mutex_unlock(&monc->mutex);
299}
300
301/*
302 *
303 */
304int ceph_monc_open_session(struct ceph_mon_client *monc)
305{
306 mutex_lock(&monc->mutex);
307 __open_session(monc);
308 __schedule_delayed(monc);
309 mutex_unlock(&monc->mutex);
310 return 0;
311}
312EXPORT_SYMBOL(ceph_monc_open_session);
313
314/*
315 * The monitor responds with mount ack indicate mount success. The
316 * included client ticket allows the client to talk to MDSs and OSDs.
317 */
318static void ceph_monc_handle_map(struct ceph_mon_client *monc,
319 struct ceph_msg *msg)
320{
321 struct ceph_client *client = monc->client;
322 struct ceph_monmap *monmap = NULL, *old = monc->monmap;
323 void *p, *end;
324
325 mutex_lock(&monc->mutex);
326
327 dout("handle_monmap\n");
328 p = msg->front.iov_base;
329 end = p + msg->front.iov_len;
330
331 monmap = ceph_monmap_decode(p, end);
332 if (IS_ERR(monmap)) {
333 pr_err("problem decoding monmap, %d\n",
334 (int)PTR_ERR(monmap));
335 goto out;
336 }
337
338 if (ceph_check_fsid(monc->client, &monmap->fsid) < 0) {
339 kfree(monmap);
340 goto out;
341 }
342
343 client->monc.monmap = monmap;
344 kfree(old);
345
346 if (!client->have_fsid) {
347 client->have_fsid = true;
348 mutex_unlock(&monc->mutex);
349 /*
350 * do debugfs initialization without mutex to avoid
351 * creating a locking dependency
352 */
353 ceph_debugfs_client_init(client);
354 goto out_unlocked;
355 }
356out:
357 mutex_unlock(&monc->mutex);
358out_unlocked:
359 wake_up_all(&client->auth_wq);
360}
361
362/*
363 * generic requests (e.g., statfs, poolop)
364 */
365static struct ceph_mon_generic_request *__lookup_generic_req(
366 struct ceph_mon_client *monc, u64 tid)
367{
368 struct ceph_mon_generic_request *req;
369 struct rb_node *n = monc->generic_request_tree.rb_node;
370
371 while (n) {
372 req = rb_entry(n, struct ceph_mon_generic_request, node);
373 if (tid < req->tid)
374 n = n->rb_left;
375 else if (tid > req->tid)
376 n = n->rb_right;
377 else
378 return req;
379 }
380 return NULL;
381}
382
383static void __insert_generic_request(struct ceph_mon_client *monc,
384 struct ceph_mon_generic_request *new)
385{
386 struct rb_node **p = &monc->generic_request_tree.rb_node;
387 struct rb_node *parent = NULL;
388 struct ceph_mon_generic_request *req = NULL;
389
390 while (*p) {
391 parent = *p;
392 req = rb_entry(parent, struct ceph_mon_generic_request, node);
393 if (new->tid < req->tid)
394 p = &(*p)->rb_left;
395 else if (new->tid > req->tid)
396 p = &(*p)->rb_right;
397 else
398 BUG();
399 }
400
401 rb_link_node(&new->node, parent, p);
402 rb_insert_color(&new->node, &monc->generic_request_tree);
403}
404
405static void release_generic_request(struct kref *kref)
406{
407 struct ceph_mon_generic_request *req =
408 container_of(kref, struct ceph_mon_generic_request, kref);
409
410 if (req->reply)
411 ceph_msg_put(req->reply);
412 if (req->request)
413 ceph_msg_put(req->request);
414
415 kfree(req);
416}
417
418static void put_generic_request(struct ceph_mon_generic_request *req)
419{
420 kref_put(&req->kref, release_generic_request);
421}
422
423static void get_generic_request(struct ceph_mon_generic_request *req)
424{
425 kref_get(&req->kref);
426}
427
428static struct ceph_msg *get_generic_reply(struct ceph_connection *con,
429 struct ceph_msg_header *hdr,
430 int *skip)
431{
432 struct ceph_mon_client *monc = con->private;
433 struct ceph_mon_generic_request *req;
434 u64 tid = le64_to_cpu(hdr->tid);
435 struct ceph_msg *m;
436
437 mutex_lock(&monc->mutex);
438 req = __lookup_generic_req(monc, tid);
439 if (!req) {
440 dout("get_generic_reply %lld dne\n", tid);
441 *skip = 1;
442 m = NULL;
443 } else {
444 dout("get_generic_reply %lld got %p\n", tid, req->reply);
445 m = ceph_msg_get(req->reply);
446 /*
447 * we don't need to track the connection reading into
448 * this reply because we only have one open connection
449 * at a time, ever.
450 */
451 }
452 mutex_unlock(&monc->mutex);
453 return m;
454}
455
456static int do_generic_request(struct ceph_mon_client *monc,
457 struct ceph_mon_generic_request *req)
458{
459 int err;
460
461 /* register request */
462 mutex_lock(&monc->mutex);
463 req->tid = ++monc->last_tid;
464 req->request->hdr.tid = cpu_to_le64(req->tid);
465 __insert_generic_request(monc, req);
466 monc->num_generic_requests++;
467 ceph_con_send(&monc->con, ceph_msg_get(req->request));
468 mutex_unlock(&monc->mutex);
469
470 err = wait_for_completion_interruptible(&req->completion);
471
472 mutex_lock(&monc->mutex);
473 rb_erase(&req->node, &monc->generic_request_tree);
474 monc->num_generic_requests--;
475 mutex_unlock(&monc->mutex);
476
477 if (!err)
478 err = req->result;
479 return err;
480}
481
482/*
483 * statfs
484 */
485static void handle_statfs_reply(struct ceph_mon_client *monc,
486 struct ceph_msg *msg)
487{
488 struct ceph_mon_generic_request *req;
489 struct ceph_mon_statfs_reply *reply = msg->front.iov_base;
490 u64 tid = le64_to_cpu(msg->hdr.tid);
491
492 if (msg->front.iov_len != sizeof(*reply))
493 goto bad;
494 dout("handle_statfs_reply %p tid %llu\n", msg, tid);
495
496 mutex_lock(&monc->mutex);
497 req = __lookup_generic_req(monc, tid);
498 if (req) {
499 *(struct ceph_statfs *)req->buf = reply->st;
500 req->result = 0;
501 get_generic_request(req);
502 }
503 mutex_unlock(&monc->mutex);
504 if (req) {
505 complete_all(&req->completion);
506 put_generic_request(req);
507 }
508 return;
509
510bad:
511 pr_err("corrupt generic reply, tid %llu\n", tid);
512 ceph_msg_dump(msg);
513}
514
515/*
516 * Do a synchronous statfs().
517 */
518int ceph_monc_do_statfs(struct ceph_mon_client *monc, struct ceph_statfs *buf)
519{
520 struct ceph_mon_generic_request *req;
521 struct ceph_mon_statfs *h;
522 int err;
523
524 req = kzalloc(sizeof(*req), GFP_NOFS);
525 if (!req)
526 return -ENOMEM;
527
528 kref_init(&req->kref);
529 req->buf = buf;
530 req->buf_len = sizeof(*buf);
531 init_completion(&req->completion);
532
533 err = -ENOMEM;
534 req->request = ceph_msg_new(CEPH_MSG_STATFS, sizeof(*h), GFP_NOFS,
535 true);
536 if (!req->request)
537 goto out;
538 req->reply = ceph_msg_new(CEPH_MSG_STATFS_REPLY, 1024, GFP_NOFS,
539 true);
540 if (!req->reply)
541 goto out;
542
543 /* fill out request */
544 h = req->request->front.iov_base;
545 h->monhdr.have_version = 0;
546 h->monhdr.session_mon = cpu_to_le16(-1);
547 h->monhdr.session_mon_tid = 0;
548 h->fsid = monc->monmap->fsid;
549
550 err = do_generic_request(monc, req);
551
552out:
553 kref_put(&req->kref, release_generic_request);
554 return err;
555}
556EXPORT_SYMBOL(ceph_monc_do_statfs);
557
558/*
559 * pool ops
560 */
561static int get_poolop_reply_buf(const char *src, size_t src_len,
562 char *dst, size_t dst_len)
563{
564 u32 buf_len;
565
566 if (src_len != sizeof(u32) + dst_len)
567 return -EINVAL;
568
569 buf_len = le32_to_cpu(*(u32 *)src);
570 if (buf_len != dst_len)
571 return -EINVAL;
572
573 memcpy(dst, src + sizeof(u32), dst_len);
574 return 0;
575}
576
577static void handle_poolop_reply(struct ceph_mon_client *monc,
578 struct ceph_msg *msg)
579{
580 struct ceph_mon_generic_request *req;
581 struct ceph_mon_poolop_reply *reply = msg->front.iov_base;
582 u64 tid = le64_to_cpu(msg->hdr.tid);
583
584 if (msg->front.iov_len < sizeof(*reply))
585 goto bad;
586 dout("handle_poolop_reply %p tid %llu\n", msg, tid);
587
588 mutex_lock(&monc->mutex);
589 req = __lookup_generic_req(monc, tid);
590 if (req) {
591 if (req->buf_len &&
592 get_poolop_reply_buf(msg->front.iov_base + sizeof(*reply),
593 msg->front.iov_len - sizeof(*reply),
594 req->buf, req->buf_len) < 0) {
595 mutex_unlock(&monc->mutex);
596 goto bad;
597 }
598 req->result = le32_to_cpu(reply->reply_code);
599 get_generic_request(req);
600 }
601 mutex_unlock(&monc->mutex);
602 if (req) {
603 complete(&req->completion);
604 put_generic_request(req);
605 }
606 return;
607
608bad:
609 pr_err("corrupt generic reply, tid %llu\n", tid);
610 ceph_msg_dump(msg);
611}
612
613/*
614 * Do a synchronous pool op.
615 */
616int ceph_monc_do_poolop(struct ceph_mon_client *monc, u32 op,
617 u32 pool, u64 snapid,
618 char *buf, int len)
619{
620 struct ceph_mon_generic_request *req;
621 struct ceph_mon_poolop *h;
622 int err;
623
624 req = kzalloc(sizeof(*req), GFP_NOFS);
625 if (!req)
626 return -ENOMEM;
627
628 kref_init(&req->kref);
629 req->buf = buf;
630 req->buf_len = len;
631 init_completion(&req->completion);
632
633 err = -ENOMEM;
634 req->request = ceph_msg_new(CEPH_MSG_POOLOP, sizeof(*h), GFP_NOFS,
635 true);
636 if (!req->request)
637 goto out;
638 req->reply = ceph_msg_new(CEPH_MSG_POOLOP_REPLY, 1024, GFP_NOFS,
639 true);
640 if (!req->reply)
641 goto out;
642
643 /* fill out request */
644 req->request->hdr.version = cpu_to_le16(2);
645 h = req->request->front.iov_base;
646 h->monhdr.have_version = 0;
647 h->monhdr.session_mon = cpu_to_le16(-1);
648 h->monhdr.session_mon_tid = 0;
649 h->fsid = monc->monmap->fsid;
650 h->pool = cpu_to_le32(pool);
651 h->op = cpu_to_le32(op);
652 h->auid = 0;
653 h->snapid = cpu_to_le64(snapid);
654 h->name_len = 0;
655
656 err = do_generic_request(monc, req);
657
658out:
659 kref_put(&req->kref, release_generic_request);
660 return err;
661}
662
663int ceph_monc_create_snapid(struct ceph_mon_client *monc,
664 u32 pool, u64 *snapid)
665{
666 return ceph_monc_do_poolop(monc, POOL_OP_CREATE_UNMANAGED_SNAP,
667 pool, 0, (char *)snapid, sizeof(*snapid));
668
669}
670EXPORT_SYMBOL(ceph_monc_create_snapid);
671
672int ceph_monc_delete_snapid(struct ceph_mon_client *monc,
673 u32 pool, u64 snapid)
674{
675 return ceph_monc_do_poolop(monc, POOL_OP_CREATE_UNMANAGED_SNAP,
676 pool, snapid, 0, 0);
677
678}
679
680/*
681 * Resend pending generic requests.
682 */
683static void __resend_generic_request(struct ceph_mon_client *monc)
684{
685 struct ceph_mon_generic_request *req;
686 struct rb_node *p;
687
688 for (p = rb_first(&monc->generic_request_tree); p; p = rb_next(p)) {
689 req = rb_entry(p, struct ceph_mon_generic_request, node);
690 ceph_con_revoke(&monc->con, req->request);
691 ceph_con_send(&monc->con, ceph_msg_get(req->request));
692 }
693}
694
695/*
696 * Delayed work. If we haven't mounted yet, retry. Otherwise,
697 * renew/retry subscription as needed (in case it is timing out, or we
698 * got an ENOMEM). And keep the monitor connection alive.
699 */
700static void delayed_work(struct work_struct *work)
701{
702 struct ceph_mon_client *monc =
703 container_of(work, struct ceph_mon_client, delayed_work.work);
704
705 dout("monc delayed_work\n");
706 mutex_lock(&monc->mutex);
707 if (monc->hunting) {
708 __close_session(monc);
709 __open_session(monc); /* continue hunting */
710 } else {
711 ceph_con_keepalive(&monc->con);
712
713 __validate_auth(monc);
714
715 if (monc->auth->ops->is_authenticated(monc->auth))
716 __send_subscribe(monc);
717 }
718 __schedule_delayed(monc);
719 mutex_unlock(&monc->mutex);
720}
721
722/*
723 * On startup, we build a temporary monmap populated with the IPs
724 * provided by mount(2).
725 */
726static int build_initial_monmap(struct ceph_mon_client *monc)
727{
728 struct ceph_options *opt = monc->client->options;
729 struct ceph_entity_addr *mon_addr = opt->mon_addr;
730 int num_mon = opt->num_mon;
731 int i;
732
733 /* build initial monmap */
734 monc->monmap = kzalloc(sizeof(*monc->monmap) +
735 num_mon*sizeof(monc->monmap->mon_inst[0]),
736 GFP_KERNEL);
737 if (!monc->monmap)
738 return -ENOMEM;
739 for (i = 0; i < num_mon; i++) {
740 monc->monmap->mon_inst[i].addr = mon_addr[i];
741 monc->monmap->mon_inst[i].addr.nonce = 0;
742 monc->monmap->mon_inst[i].name.type =
743 CEPH_ENTITY_TYPE_MON;
744 monc->monmap->mon_inst[i].name.num = cpu_to_le64(i);
745 }
746 monc->monmap->num_mon = num_mon;
747 monc->have_fsid = false;
748 return 0;
749}
750
751int ceph_monc_init(struct ceph_mon_client *monc, struct ceph_client *cl)
752{
753 int err = 0;
754
755 dout("init\n");
756 memset(monc, 0, sizeof(*monc));
757 monc->client = cl;
758 monc->monmap = NULL;
759 mutex_init(&monc->mutex);
760
761 err = build_initial_monmap(monc);
762 if (err)
763 goto out;
764
765 /* connection */
766 /* authentication */
767 monc->auth = ceph_auth_init(cl->options->name,
768 cl->options->key);
769 if (IS_ERR(monc->auth)) {
770 err = PTR_ERR(monc->auth);
771 goto out_monmap;
772 }
773 monc->auth->want_keys =
774 CEPH_ENTITY_TYPE_AUTH | CEPH_ENTITY_TYPE_MON |
775 CEPH_ENTITY_TYPE_OSD | CEPH_ENTITY_TYPE_MDS;
776
777 /* msgs */
778 err = -ENOMEM;
779 monc->m_subscribe_ack = ceph_msg_new(CEPH_MSG_MON_SUBSCRIBE_ACK,
780 sizeof(struct ceph_mon_subscribe_ack),
781 GFP_NOFS, true);
782 if (!monc->m_subscribe_ack)
783 goto out_auth;
784
785 monc->m_subscribe = ceph_msg_new(CEPH_MSG_MON_SUBSCRIBE, 96, GFP_NOFS,
786 true);
787 if (!monc->m_subscribe)
788 goto out_subscribe_ack;
789
790 monc->m_auth_reply = ceph_msg_new(CEPH_MSG_AUTH_REPLY, 4096, GFP_NOFS,
791 true);
792 if (!monc->m_auth_reply)
793 goto out_subscribe;
794
795 monc->m_auth = ceph_msg_new(CEPH_MSG_AUTH, 4096, GFP_NOFS, true);
796 monc->pending_auth = 0;
797 if (!monc->m_auth)
798 goto out_auth_reply;
799
800 monc->cur_mon = -1;
801 monc->hunting = true;
802 monc->sub_renew_after = jiffies;
803 monc->sub_sent = 0;
804
805 INIT_DELAYED_WORK(&monc->delayed_work, delayed_work);
806 monc->generic_request_tree = RB_ROOT;
807 monc->num_generic_requests = 0;
808 monc->last_tid = 0;
809
810 monc->have_mdsmap = 0;
811 monc->have_osdmap = 0;
812 monc->want_next_osdmap = 1;
813 return 0;
814
815out_auth_reply:
816 ceph_msg_put(monc->m_auth_reply);
817out_subscribe:
818 ceph_msg_put(monc->m_subscribe);
819out_subscribe_ack:
820 ceph_msg_put(monc->m_subscribe_ack);
821out_auth:
822 ceph_auth_destroy(monc->auth);
823out_monmap:
824 kfree(monc->monmap);
825out:
826 return err;
827}
828EXPORT_SYMBOL(ceph_monc_init);
829
830void ceph_monc_stop(struct ceph_mon_client *monc)
831{
832 dout("stop\n");
833 cancel_delayed_work_sync(&monc->delayed_work);
834
835 mutex_lock(&monc->mutex);
836 __close_session(monc);
837
838 mutex_unlock(&monc->mutex);
839
840 ceph_auth_destroy(monc->auth);
841
842 ceph_msg_put(monc->m_auth);
843 ceph_msg_put(monc->m_auth_reply);
844 ceph_msg_put(monc->m_subscribe);
845 ceph_msg_put(monc->m_subscribe_ack);
846
847 kfree(monc->monmap);
848}
849EXPORT_SYMBOL(ceph_monc_stop);
850
851static void handle_auth_reply(struct ceph_mon_client *monc,
852 struct ceph_msg *msg)
853{
854 int ret;
855 int was_auth = 0;
856
857 mutex_lock(&monc->mutex);
858 if (monc->auth->ops)
859 was_auth = monc->auth->ops->is_authenticated(monc->auth);
860 monc->pending_auth = 0;
861 ret = ceph_handle_auth_reply(monc->auth, msg->front.iov_base,
862 msg->front.iov_len,
863 monc->m_auth->front.iov_base,
864 monc->m_auth->front_max);
865 if (ret < 0) {
866 monc->client->auth_err = ret;
867 wake_up_all(&monc->client->auth_wq);
868 } else if (ret > 0) {
869 __send_prepared_auth_request(monc, ret);
870 } else if (!was_auth && monc->auth->ops->is_authenticated(monc->auth)) {
871 dout("authenticated, starting session\n");
872
873 monc->client->msgr.inst.name.type = CEPH_ENTITY_TYPE_CLIENT;
874 monc->client->msgr.inst.name.num =
875 cpu_to_le64(monc->auth->global_id);
876
877 __send_subscribe(monc);
878 __resend_generic_request(monc);
879 }
880 mutex_unlock(&monc->mutex);
881}
882
883static int __validate_auth(struct ceph_mon_client *monc)
884{
885 int ret;
886
887 if (monc->pending_auth)
888 return 0;
889
890 ret = ceph_build_auth(monc->auth, monc->m_auth->front.iov_base,
891 monc->m_auth->front_max);
892 if (ret <= 0)
893 return ret; /* either an error, or no need to authenticate */
894 __send_prepared_auth_request(monc, ret);
895 return 0;
896}
897
898int ceph_monc_validate_auth(struct ceph_mon_client *monc)
899{
900 int ret;
901
902 mutex_lock(&monc->mutex);
903 ret = __validate_auth(monc);
904 mutex_unlock(&monc->mutex);
905 return ret;
906}
907EXPORT_SYMBOL(ceph_monc_validate_auth);
908
909/*
910 * handle incoming message
911 */
912static void dispatch(struct ceph_connection *con, struct ceph_msg *msg)
913{
914 struct ceph_mon_client *monc = con->private;
915 int type = le16_to_cpu(msg->hdr.type);
916
917 if (!monc)
918 return;
919
920 switch (type) {
921 case CEPH_MSG_AUTH_REPLY:
922 handle_auth_reply(monc, msg);
923 break;
924
925 case CEPH_MSG_MON_SUBSCRIBE_ACK:
926 handle_subscribe_ack(monc, msg);
927 break;
928
929 case CEPH_MSG_STATFS_REPLY:
930 handle_statfs_reply(monc, msg);
931 break;
932
933 case CEPH_MSG_POOLOP_REPLY:
934 handle_poolop_reply(monc, msg);
935 break;
936
937 case CEPH_MSG_MON_MAP:
938 ceph_monc_handle_map(monc, msg);
939 break;
940
941 case CEPH_MSG_OSD_MAP:
942 ceph_osdc_handle_map(&monc->client->osdc, msg);
943 break;
944
945 default:
946 /* can the chained handler handle it? */
947 if (monc->client->extra_mon_dispatch &&
948 monc->client->extra_mon_dispatch(monc->client, msg) == 0)
949 break;
950
951 pr_err("received unknown message type %d %s\n", type,
952 ceph_msg_type_name(type));
953 }
954 ceph_msg_put(msg);
955}
956
957/*
958 * Allocate memory for incoming message
959 */
960static struct ceph_msg *mon_alloc_msg(struct ceph_connection *con,
961 struct ceph_msg_header *hdr,
962 int *skip)
963{
964 struct ceph_mon_client *monc = con->private;
965 int type = le16_to_cpu(hdr->type);
966 int front_len = le32_to_cpu(hdr->front_len);
967 struct ceph_msg *m = NULL;
968
969 *skip = 0;
970
971 switch (type) {
972 case CEPH_MSG_MON_SUBSCRIBE_ACK:
973 m = ceph_msg_get(monc->m_subscribe_ack);
974 break;
975 case CEPH_MSG_POOLOP_REPLY:
976 case CEPH_MSG_STATFS_REPLY:
977 return get_generic_reply(con, hdr, skip);
978 case CEPH_MSG_AUTH_REPLY:
979 m = ceph_msg_get(monc->m_auth_reply);
980 break;
981 case CEPH_MSG_MON_MAP:
982 case CEPH_MSG_MDS_MAP:
983 case CEPH_MSG_OSD_MAP:
984 m = ceph_msg_new(type, front_len, GFP_NOFS, false);
985 break;
986 }
987
988 if (!m) {
989 pr_info("alloc_msg unknown type %d\n", type);
990 *skip = 1;
991 }
992 return m;
993}
994
995/*
996 * If the monitor connection resets, pick a new monitor and resubmit
997 * any pending requests.
998 */
999static void mon_fault(struct ceph_connection *con)
1000{
1001 struct ceph_mon_client *monc = con->private;
1002
1003 if (!monc)
1004 return;
1005
1006 dout("mon_fault\n");
1007 mutex_lock(&monc->mutex);
1008 if (!con->private)
1009 goto out;
1010
1011 if (!monc->hunting)
1012 pr_info("mon%d %s session lost, "
1013 "hunting for new mon\n", monc->cur_mon,
1014 ceph_pr_addr(&monc->con.peer_addr.in_addr));
1015
1016 __close_session(monc);
1017 if (!monc->hunting) {
1018 /* start hunting */
1019 monc->hunting = true;
1020 __open_session(monc);
1021 } else {
1022 /* already hunting, let's wait a bit */
1023 __schedule_delayed(monc);
1024 }
1025out:
1026 mutex_unlock(&monc->mutex);
1027}
1028
1029/*
1030 * We can ignore refcounting on the connection struct, as all references
1031 * will come from the messenger workqueue, which is drained prior to
1032 * mon_client destruction.
1033 */
1034static struct ceph_connection *con_get(struct ceph_connection *con)
1035{
1036 return con;
1037}
1038
1039static void con_put(struct ceph_connection *con)
1040{
1041}
1042
1043static const struct ceph_connection_operations mon_con_ops = {
1044 .get = con_get,
1045 .put = con_put,
1046 .dispatch = dispatch,
1047 .fault = mon_fault,
1048 .alloc_msg = mon_alloc_msg,
1049};