xref: /linux/net/ceph/messenger_v1.c (revision 8bd9238e511d02831022ff0270865c54ccc482d6)
1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/ceph/ceph_debug.h>
3 
4 #include <linux/bvec.h>
5 #include <linux/crc32c.h>
6 #include <linux/net.h>
7 #include <linux/socket.h>
8 #include <net/sock.h>
9 
10 #include <linux/ceph/ceph_features.h>
11 #include <linux/ceph/decode.h>
12 #include <linux/ceph/libceph.h>
13 #include <linux/ceph/messenger.h>
14 
15 /* static tag bytes (protocol control messages) */
16 static char tag_msg = CEPH_MSGR_TAG_MSG;
17 static char tag_ack = CEPH_MSGR_TAG_ACK;
18 static char tag_keepalive = CEPH_MSGR_TAG_KEEPALIVE;
19 static char tag_keepalive2 = CEPH_MSGR_TAG_KEEPALIVE2;
20 
21 /*
22  * If @buf is NULL, discard up to @len bytes.
23  */
ceph_tcp_recvmsg(struct socket * sock,void * buf,size_t len)24 static int ceph_tcp_recvmsg(struct socket *sock, void *buf, size_t len)
25 {
26 	struct kvec iov = {buf, len};
27 	struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL };
28 	int r;
29 
30 	if (!buf)
31 		msg.msg_flags |= MSG_TRUNC;
32 
33 	iov_iter_kvec(&msg.msg_iter, ITER_DEST, &iov, 1, len);
34 	r = sock_recvmsg(sock, &msg, msg.msg_flags);
35 	if (r == -EAGAIN)
36 		r = 0;
37 	return r;
38 }
39 
ceph_tcp_recvpage(struct socket * sock,struct page * page,int page_offset,size_t length)40 static int ceph_tcp_recvpage(struct socket *sock, struct page *page,
41 		     int page_offset, size_t length)
42 {
43 	struct bio_vec bvec;
44 	struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL };
45 	int r;
46 
47 	BUG_ON(page_offset + length > PAGE_SIZE);
48 	bvec_set_page(&bvec, page, length, page_offset);
49 	iov_iter_bvec(&msg.msg_iter, ITER_DEST, &bvec, 1, length);
50 	r = sock_recvmsg(sock, &msg, msg.msg_flags);
51 	if (r == -EAGAIN)
52 		r = 0;
53 	return r;
54 }
55 
56 /*
57  * write something.  @more is true if caller will be sending more data
58  * shortly.
59  */
ceph_tcp_sendmsg(struct socket * sock,struct kvec * iov,size_t kvlen,size_t len,bool more)60 static int ceph_tcp_sendmsg(struct socket *sock, struct kvec *iov,
61 			    size_t kvlen, size_t len, bool more)
62 {
63 	struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL };
64 	int r;
65 
66 	if (more)
67 		msg.msg_flags |= MSG_MORE;
68 	else
69 		msg.msg_flags |= MSG_EOR;  /* superfluous, but what the hell */
70 
71 	r = kernel_sendmsg(sock, &msg, iov, kvlen, len);
72 	if (r == -EAGAIN)
73 		r = 0;
74 	return r;
75 }
76 
77 /*
78  * @more: MSG_MORE or 0.
79  */
ceph_tcp_sendpage(struct socket * sock,struct page * page,int offset,size_t size,int more)80 static int ceph_tcp_sendpage(struct socket *sock, struct page *page,
81 			     int offset, size_t size, int more)
82 {
83 	struct msghdr msg = {
84 		.msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL | more,
85 	};
86 	struct bio_vec bvec;
87 	int ret;
88 
89 	/*
90 	 * MSG_SPLICE_PAGES cannot properly handle pages with page_count == 0,
91 	 * we need to fall back to sendmsg if that's the case.
92 	 *
93 	 * Same goes for slab pages: skb_can_coalesce() allows
94 	 * coalescing neighboring slab objects into a single frag which
95 	 * triggers one of hardened usercopy checks.
96 	 */
97 	if (sendpage_ok(page))
98 		msg.msg_flags |= MSG_SPLICE_PAGES;
99 
100 	bvec_set_page(&bvec, page, size, offset);
101 	iov_iter_bvec(&msg.msg_iter, ITER_SOURCE, &bvec, 1, size);
102 
103 	ret = sock_sendmsg(sock, &msg);
104 	if (ret == -EAGAIN)
105 		ret = 0;
106 
107 	return ret;
108 }
109 
con_out_kvec_reset(struct ceph_connection * con)110 static void con_out_kvec_reset(struct ceph_connection *con)
111 {
112 	BUG_ON(con->v1.out_skip);
113 
114 	con->v1.out_kvec_left = 0;
115 	con->v1.out_kvec_bytes = 0;
116 	con->v1.out_kvec_cur = &con->v1.out_kvec[0];
117 }
118 
con_out_kvec_add(struct ceph_connection * con,size_t size,void * data)119 static void con_out_kvec_add(struct ceph_connection *con,
120 				size_t size, void *data)
121 {
122 	int index = con->v1.out_kvec_left;
123 
124 	BUG_ON(con->v1.out_skip);
125 	BUG_ON(index >= ARRAY_SIZE(con->v1.out_kvec));
126 
127 	con->v1.out_kvec[index].iov_len = size;
128 	con->v1.out_kvec[index].iov_base = data;
129 	con->v1.out_kvec_left++;
130 	con->v1.out_kvec_bytes += size;
131 }
132 
133 /*
134  * Chop off a kvec from the end.  Return residual number of bytes for
135  * that kvec, i.e. how many bytes would have been written if the kvec
136  * hadn't been nuked.
137  */
con_out_kvec_skip(struct ceph_connection * con)138 static int con_out_kvec_skip(struct ceph_connection *con)
139 {
140 	int skip = 0;
141 
142 	if (con->v1.out_kvec_bytes > 0) {
143 		skip = con->v1.out_kvec_cur[con->v1.out_kvec_left - 1].iov_len;
144 		BUG_ON(con->v1.out_kvec_bytes < skip);
145 		BUG_ON(!con->v1.out_kvec_left);
146 		con->v1.out_kvec_bytes -= skip;
147 		con->v1.out_kvec_left--;
148 	}
149 
150 	return skip;
151 }
152 
sizeof_footer(struct ceph_connection * con)153 static size_t sizeof_footer(struct ceph_connection *con)
154 {
155 	return (con->peer_features & CEPH_FEATURE_MSG_AUTH) ?
156 	    sizeof(struct ceph_msg_footer) :
157 	    sizeof(struct ceph_msg_footer_old);
158 }
159 
prepare_message_data(struct ceph_msg * msg,u32 data_len)160 static void prepare_message_data(struct ceph_msg *msg, u32 data_len)
161 {
162 	/* Initialize data cursor if it's not a sparse read */
163 	u64 len = msg->sparse_read_total ? : data_len;
164 
165 	ceph_msg_data_cursor_init(&msg->cursor, msg, len);
166 }
167 
168 /*
169  * Prepare footer for currently outgoing message, and finish things
170  * off.  Assumes out_kvec* are already valid.. we just add on to the end.
171  */
prepare_write_message_footer(struct ceph_connection * con,struct ceph_msg * m)172 static void prepare_write_message_footer(struct ceph_connection *con,
173 					 struct ceph_msg *m)
174 {
175 	m->footer.flags |= CEPH_MSG_FOOTER_COMPLETE;
176 
177 	dout("prepare_write_message_footer %p\n", con);
178 	con_out_kvec_add(con, sizeof_footer(con), &m->footer);
179 	if (con->peer_features & CEPH_FEATURE_MSG_AUTH) {
180 		if (con->ops->sign_message)
181 			con->ops->sign_message(m);
182 		else
183 			m->footer.sig = 0;
184 	} else {
185 		m->old_footer.flags = m->footer.flags;
186 	}
187 	con->v1.out_more = m->more_to_follow;
188 	con->v1.out_msg_done = true;
189 }
190 
191 /*
192  * Prepare headers for the next outgoing message.
193  */
prepare_write_message(struct ceph_connection * con,struct ceph_msg * m)194 static void prepare_write_message(struct ceph_connection *con,
195 				  struct ceph_msg *m)
196 {
197 	u32 crc;
198 
199 	con_out_kvec_reset(con);
200 	con->v1.out_msg_done = false;
201 
202 	/* Sneak an ack in there first?  If we can get it into the same
203 	 * TCP packet that's a good thing. */
204 	if (con->in_seq > con->in_seq_acked) {
205 		con->in_seq_acked = con->in_seq;
206 		con_out_kvec_add(con, sizeof (tag_ack), &tag_ack);
207 		con->v1.out_temp_ack = cpu_to_le64(con->in_seq_acked);
208 		con_out_kvec_add(con, sizeof(con->v1.out_temp_ack),
209 			&con->v1.out_temp_ack);
210 	}
211 
212 	dout("prepare_write_message %p seq %lld type %d len %d+%d+%zd\n",
213 	     m, con->out_seq, le16_to_cpu(m->hdr.type),
214 	     le32_to_cpu(m->hdr.front_len), le32_to_cpu(m->hdr.middle_len),
215 	     m->data_length);
216 	WARN_ON(m->front.iov_len != le32_to_cpu(m->hdr.front_len));
217 	WARN_ON(m->data_length != le32_to_cpu(m->hdr.data_len));
218 
219 	/* tag + hdr + front + middle */
220 	con_out_kvec_add(con, sizeof (tag_msg), &tag_msg);
221 	con_out_kvec_add(con, sizeof(con->v1.out_hdr), &con->v1.out_hdr);
222 	con_out_kvec_add(con, m->front.iov_len, m->front.iov_base);
223 
224 	if (m->middle)
225 		con_out_kvec_add(con, m->middle->vec.iov_len,
226 			m->middle->vec.iov_base);
227 
228 	/* fill in hdr crc and finalize hdr */
229 	crc = crc32c(0, &m->hdr, offsetof(struct ceph_msg_header, crc));
230 	m->hdr.crc = cpu_to_le32(crc);
231 	memcpy(&con->v1.out_hdr, &m->hdr, sizeof(con->v1.out_hdr));
232 
233 	/* fill in front and middle crc, footer */
234 	crc = crc32c(0, m->front.iov_base, m->front.iov_len);
235 	m->footer.front_crc = cpu_to_le32(crc);
236 	if (m->middle) {
237 		crc = crc32c(0, m->middle->vec.iov_base,
238 				m->middle->vec.iov_len);
239 		m->footer.middle_crc = cpu_to_le32(crc);
240 	} else
241 		m->footer.middle_crc = 0;
242 	dout("%s front_crc %u middle_crc %u\n", __func__,
243 	     le32_to_cpu(m->footer.front_crc),
244 	     le32_to_cpu(m->footer.middle_crc));
245 	m->footer.flags = 0;
246 
247 	/* is there a data payload? */
248 	m->footer.data_crc = 0;
249 	if (m->data_length) {
250 		prepare_message_data(m, m->data_length);
251 		con->v1.out_more = 1;  /* data + footer will follow */
252 	} else {
253 		/* no, queue up footer too and be done */
254 		prepare_write_message_footer(con, m);
255 	}
256 
257 	ceph_con_flag_set(con, CEPH_CON_F_WRITE_PENDING);
258 }
259 
260 /*
261  * Prepare an ack.
262  */
prepare_write_ack(struct ceph_connection * con)263 static void prepare_write_ack(struct ceph_connection *con)
264 {
265 	dout("prepare_write_ack %p %llu -> %llu\n", con,
266 	     con->in_seq_acked, con->in_seq);
267 	con->in_seq_acked = con->in_seq;
268 
269 	con_out_kvec_reset(con);
270 
271 	con_out_kvec_add(con, sizeof (tag_ack), &tag_ack);
272 
273 	con->v1.out_temp_ack = cpu_to_le64(con->in_seq_acked);
274 	con_out_kvec_add(con, sizeof(con->v1.out_temp_ack),
275 			 &con->v1.out_temp_ack);
276 
277 	con->v1.out_more = 1;  /* more will follow.. eventually.. */
278 	ceph_con_flag_set(con, CEPH_CON_F_WRITE_PENDING);
279 }
280 
281 /*
282  * Prepare to share the seq during handshake
283  */
prepare_write_seq(struct ceph_connection * con)284 static void prepare_write_seq(struct ceph_connection *con)
285 {
286 	dout("prepare_write_seq %p %llu -> %llu\n", con,
287 	     con->in_seq_acked, con->in_seq);
288 	con->in_seq_acked = con->in_seq;
289 
290 	con_out_kvec_reset(con);
291 
292 	con->v1.out_temp_ack = cpu_to_le64(con->in_seq_acked);
293 	con_out_kvec_add(con, sizeof(con->v1.out_temp_ack),
294 			 &con->v1.out_temp_ack);
295 
296 	ceph_con_flag_set(con, CEPH_CON_F_WRITE_PENDING);
297 }
298 
299 /*
300  * Prepare to write keepalive byte.
301  */
prepare_write_keepalive(struct ceph_connection * con)302 static void prepare_write_keepalive(struct ceph_connection *con)
303 {
304 	dout("prepare_write_keepalive %p\n", con);
305 	con_out_kvec_reset(con);
306 	if (con->peer_features & CEPH_FEATURE_MSGR_KEEPALIVE2) {
307 		struct timespec64 now;
308 
309 		ktime_get_real_ts64(&now);
310 		con_out_kvec_add(con, sizeof(tag_keepalive2), &tag_keepalive2);
311 		ceph_encode_timespec64(&con->v1.out_temp_keepalive2, &now);
312 		con_out_kvec_add(con, sizeof(con->v1.out_temp_keepalive2),
313 				 &con->v1.out_temp_keepalive2);
314 	} else {
315 		con_out_kvec_add(con, sizeof(tag_keepalive), &tag_keepalive);
316 	}
317 	ceph_con_flag_set(con, CEPH_CON_F_WRITE_PENDING);
318 }
319 
320 /*
321  * Connection negotiation.
322  */
323 
get_connect_authorizer(struct ceph_connection * con)324 static int get_connect_authorizer(struct ceph_connection *con)
325 {
326 	struct ceph_auth_handshake *auth;
327 	int auth_proto;
328 
329 	if (!con->ops->get_authorizer) {
330 		con->v1.auth = NULL;
331 		con->v1.out_connect.authorizer_protocol = CEPH_AUTH_UNKNOWN;
332 		con->v1.out_connect.authorizer_len = 0;
333 		return 0;
334 	}
335 
336 	auth = con->ops->get_authorizer(con, &auth_proto, con->v1.auth_retry);
337 	if (IS_ERR(auth))
338 		return PTR_ERR(auth);
339 
340 	con->v1.auth = auth;
341 	con->v1.out_connect.authorizer_protocol = cpu_to_le32(auth_proto);
342 	con->v1.out_connect.authorizer_len =
343 		cpu_to_le32(auth->authorizer_buf_len);
344 	return 0;
345 }
346 
347 /*
348  * We connected to a peer and are saying hello.
349  */
prepare_write_banner(struct ceph_connection * con)350 static void prepare_write_banner(struct ceph_connection *con)
351 {
352 	con_out_kvec_add(con, strlen(CEPH_BANNER), CEPH_BANNER);
353 	con_out_kvec_add(con, sizeof (con->msgr->my_enc_addr),
354 					&con->msgr->my_enc_addr);
355 
356 	con->v1.out_more = 0;
357 	ceph_con_flag_set(con, CEPH_CON_F_WRITE_PENDING);
358 }
359 
__prepare_write_connect(struct ceph_connection * con)360 static void __prepare_write_connect(struct ceph_connection *con)
361 {
362 	con_out_kvec_add(con, sizeof(con->v1.out_connect),
363 			 &con->v1.out_connect);
364 	if (con->v1.auth)
365 		con_out_kvec_add(con, con->v1.auth->authorizer_buf_len,
366 				 con->v1.auth->authorizer_buf);
367 
368 	con->v1.out_more = 0;
369 	ceph_con_flag_set(con, CEPH_CON_F_WRITE_PENDING);
370 }
371 
prepare_write_connect(struct ceph_connection * con)372 static int prepare_write_connect(struct ceph_connection *con)
373 {
374 	unsigned int global_seq = ceph_get_global_seq(con->msgr, 0);
375 	int proto;
376 	int ret;
377 
378 	switch (con->peer_name.type) {
379 	case CEPH_ENTITY_TYPE_MON:
380 		proto = CEPH_MONC_PROTOCOL;
381 		break;
382 	case CEPH_ENTITY_TYPE_OSD:
383 		proto = CEPH_OSDC_PROTOCOL;
384 		break;
385 	case CEPH_ENTITY_TYPE_MDS:
386 		proto = CEPH_MDSC_PROTOCOL;
387 		break;
388 	default:
389 		BUG();
390 	}
391 
392 	dout("prepare_write_connect %p cseq=%d gseq=%d proto=%d\n", con,
393 	     con->v1.connect_seq, global_seq, proto);
394 
395 	con->v1.out_connect.features =
396 		cpu_to_le64(from_msgr(con->msgr)->supported_features);
397 	con->v1.out_connect.host_type = cpu_to_le32(CEPH_ENTITY_TYPE_CLIENT);
398 	con->v1.out_connect.connect_seq = cpu_to_le32(con->v1.connect_seq);
399 	con->v1.out_connect.global_seq = cpu_to_le32(global_seq);
400 	con->v1.out_connect.protocol_version = cpu_to_le32(proto);
401 	con->v1.out_connect.flags = 0;
402 
403 	ret = get_connect_authorizer(con);
404 	if (ret)
405 		return ret;
406 
407 	__prepare_write_connect(con);
408 	return 0;
409 }
410 
411 /*
412  * write as much of pending kvecs to the socket as we can.
413  *  1 -> done
414  *  0 -> socket full, but more to do
415  * <0 -> error
416  */
write_partial_kvec(struct ceph_connection * con)417 static int write_partial_kvec(struct ceph_connection *con)
418 {
419 	int ret;
420 
421 	dout("write_partial_kvec %p %d left\n", con, con->v1.out_kvec_bytes);
422 	while (con->v1.out_kvec_bytes > 0) {
423 		ret = ceph_tcp_sendmsg(con->sock, con->v1.out_kvec_cur,
424 				       con->v1.out_kvec_left,
425 				       con->v1.out_kvec_bytes,
426 				       con->v1.out_more);
427 		if (ret <= 0)
428 			goto out;
429 		con->v1.out_kvec_bytes -= ret;
430 		if (!con->v1.out_kvec_bytes)
431 			break;            /* done */
432 
433 		/* account for full iov entries consumed */
434 		while (ret >= con->v1.out_kvec_cur->iov_len) {
435 			BUG_ON(!con->v1.out_kvec_left);
436 			ret -= con->v1.out_kvec_cur->iov_len;
437 			con->v1.out_kvec_cur++;
438 			con->v1.out_kvec_left--;
439 		}
440 		/* and for a partially-consumed entry */
441 		if (ret) {
442 			con->v1.out_kvec_cur->iov_len -= ret;
443 			con->v1.out_kvec_cur->iov_base += ret;
444 		}
445 	}
446 	con->v1.out_kvec_left = 0;
447 	ret = 1;
448 out:
449 	dout("write_partial_kvec %p %d left in %d kvecs ret = %d\n", con,
450 	     con->v1.out_kvec_bytes, con->v1.out_kvec_left, ret);
451 	return ret;  /* done! */
452 }
453 
454 /*
455  * Write as much message data payload as we can.  If we finish, queue
456  * up the footer.
457  *  1 -> done, footer is now queued in out_kvec[].
458  *  0 -> socket full, but more to do
459  * <0 -> error
460  */
write_partial_message_data(struct ceph_connection * con,struct ceph_msg * msg)461 static int write_partial_message_data(struct ceph_connection *con,
462 				      struct ceph_msg *msg)
463 {
464 	struct ceph_msg_data_cursor *cursor = &msg->cursor;
465 	bool do_datacrc = !ceph_test_opt(from_msgr(con->msgr), NOCRC);
466 	u32 crc;
467 
468 	dout("%s %p msg %p\n", __func__, con, msg);
469 
470 	if (!msg->num_data_items)
471 		return -EINVAL;
472 
473 	/*
474 	 * Iterate through each page that contains data to be
475 	 * written, and send as much as possible for each.
476 	 *
477 	 * If we are calculating the data crc (the default), we will
478 	 * need to map the page.  If we have no pages, they have
479 	 * been revoked, so use the zero page.
480 	 */
481 	crc = do_datacrc ? le32_to_cpu(msg->footer.data_crc) : 0;
482 	while (cursor->total_resid) {
483 		struct page *page;
484 		size_t page_offset;
485 		size_t length;
486 		int ret;
487 
488 		if (!cursor->resid) {
489 			ceph_msg_data_advance(cursor, 0);
490 			continue;
491 		}
492 
493 		page = ceph_msg_data_next(cursor, &page_offset, &length);
494 		ret = ceph_tcp_sendpage(con->sock, page, page_offset, length,
495 					MSG_MORE);
496 		if (ret <= 0) {
497 			if (do_datacrc)
498 				msg->footer.data_crc = cpu_to_le32(crc);
499 
500 			return ret;
501 		}
502 		if (do_datacrc && cursor->need_crc)
503 			crc = ceph_crc32c_page(crc, page, page_offset, length);
504 		ceph_msg_data_advance(cursor, (size_t)ret);
505 	}
506 
507 	dout("%s %p msg %p done\n", __func__, con, msg);
508 
509 	/* prepare and queue up footer, too */
510 	if (do_datacrc)
511 		msg->footer.data_crc = cpu_to_le32(crc);
512 	else
513 		msg->footer.flags |= CEPH_MSG_FOOTER_NOCRC;
514 	con_out_kvec_reset(con);
515 	prepare_write_message_footer(con, msg);
516 
517 	return 1;	/* must return > 0 to indicate success */
518 }
519 
520 /*
521  * write some zeros
522  */
write_partial_skip(struct ceph_connection * con)523 static int write_partial_skip(struct ceph_connection *con)
524 {
525 	int ret;
526 
527 	dout("%s %p %d left\n", __func__, con, con->v1.out_skip);
528 	while (con->v1.out_skip > 0) {
529 		size_t size = min(con->v1.out_skip, (int)PAGE_SIZE);
530 
531 		ret = ceph_tcp_sendpage(con->sock, ceph_zero_page, 0, size,
532 					MSG_MORE);
533 		if (ret <= 0)
534 			goto out;
535 		con->v1.out_skip -= ret;
536 	}
537 	ret = 1;
538 out:
539 	return ret;
540 }
541 
542 /*
543  * Prepare to read connection handshake, or an ack.
544  */
prepare_read_banner(struct ceph_connection * con)545 static void prepare_read_banner(struct ceph_connection *con)
546 {
547 	dout("prepare_read_banner %p\n", con);
548 	con->v1.in_base_pos = 0;
549 }
550 
prepare_read_connect(struct ceph_connection * con)551 static void prepare_read_connect(struct ceph_connection *con)
552 {
553 	dout("prepare_read_connect %p\n", con);
554 	con->v1.in_base_pos = 0;
555 }
556 
prepare_read_ack(struct ceph_connection * con)557 static void prepare_read_ack(struct ceph_connection *con)
558 {
559 	dout("prepare_read_ack %p\n", con);
560 	con->v1.in_base_pos = 0;
561 }
562 
prepare_read_seq(struct ceph_connection * con)563 static void prepare_read_seq(struct ceph_connection *con)
564 {
565 	dout("prepare_read_seq %p\n", con);
566 	con->v1.in_base_pos = 0;
567 	con->v1.in_tag = CEPH_MSGR_TAG_SEQ;
568 }
569 
prepare_read_tag(struct ceph_connection * con)570 static void prepare_read_tag(struct ceph_connection *con)
571 {
572 	dout("prepare_read_tag %p\n", con);
573 	con->v1.in_base_pos = 0;
574 	con->v1.in_tag = CEPH_MSGR_TAG_READY;
575 }
576 
prepare_read_keepalive_ack(struct ceph_connection * con)577 static void prepare_read_keepalive_ack(struct ceph_connection *con)
578 {
579 	dout("prepare_read_keepalive_ack %p\n", con);
580 	con->v1.in_base_pos = 0;
581 }
582 
583 /*
584  * Prepare to read a message.
585  */
prepare_read_message(struct ceph_connection * con)586 static int prepare_read_message(struct ceph_connection *con)
587 {
588 	dout("prepare_read_message %p\n", con);
589 	BUG_ON(con->in_msg != NULL);
590 	con->v1.in_base_pos = 0;
591 	con->in_front_crc = con->in_middle_crc = con->in_data_crc = 0;
592 	return 0;
593 }
594 
read_partial(struct ceph_connection * con,int end,int size,void * object)595 static int read_partial(struct ceph_connection *con,
596 			int end, int size, void *object)
597 {
598 	while (con->v1.in_base_pos < end) {
599 		int left = end - con->v1.in_base_pos;
600 		int have = size - left;
601 		int ret = ceph_tcp_recvmsg(con->sock, object + have, left);
602 		if (ret <= 0)
603 			return ret;
604 		con->v1.in_base_pos += ret;
605 	}
606 	return 1;
607 }
608 
609 /*
610  * Read all or part of the connect-side handshake on a new connection
611  */
read_partial_banner(struct ceph_connection * con)612 static int read_partial_banner(struct ceph_connection *con)
613 {
614 	int size;
615 	int end;
616 	int ret;
617 
618 	dout("read_partial_banner %p at %d\n", con, con->v1.in_base_pos);
619 
620 	/* peer's banner */
621 	size = strlen(CEPH_BANNER);
622 	end = size;
623 	ret = read_partial(con, end, size, con->v1.in_banner);
624 	if (ret <= 0)
625 		goto out;
626 
627 	size = sizeof(con->v1.actual_peer_addr);
628 	end += size;
629 	ret = read_partial(con, end, size, &con->v1.actual_peer_addr);
630 	if (ret <= 0)
631 		goto out;
632 	ceph_decode_banner_addr(&con->v1.actual_peer_addr);
633 
634 	size = sizeof(con->v1.peer_addr_for_me);
635 	end += size;
636 	ret = read_partial(con, end, size, &con->v1.peer_addr_for_me);
637 	if (ret <= 0)
638 		goto out;
639 	ceph_decode_banner_addr(&con->v1.peer_addr_for_me);
640 
641 out:
642 	return ret;
643 }
644 
read_partial_connect(struct ceph_connection * con)645 static int read_partial_connect(struct ceph_connection *con)
646 {
647 	int size;
648 	int end;
649 	int ret;
650 
651 	dout("read_partial_connect %p at %d\n", con, con->v1.in_base_pos);
652 
653 	size = sizeof(con->v1.in_reply);
654 	end = size;
655 	ret = read_partial(con, end, size, &con->v1.in_reply);
656 	if (ret <= 0)
657 		goto out;
658 
659 	if (con->v1.auth) {
660 		size = le32_to_cpu(con->v1.in_reply.authorizer_len);
661 		if (size > con->v1.auth->authorizer_reply_buf_len) {
662 			pr_err("authorizer reply too big: %d > %zu\n", size,
663 			       con->v1.auth->authorizer_reply_buf_len);
664 			ret = -EINVAL;
665 			goto out;
666 		}
667 
668 		end += size;
669 		ret = read_partial(con, end, size,
670 				   con->v1.auth->authorizer_reply_buf);
671 		if (ret <= 0)
672 			goto out;
673 	}
674 
675 	dout("read_partial_connect %p tag %d, con_seq = %u, g_seq = %u\n",
676 	     con, con->v1.in_reply.tag,
677 	     le32_to_cpu(con->v1.in_reply.connect_seq),
678 	     le32_to_cpu(con->v1.in_reply.global_seq));
679 out:
680 	return ret;
681 }
682 
683 /*
684  * Verify the hello banner looks okay.
685  */
verify_hello(struct ceph_connection * con)686 static int verify_hello(struct ceph_connection *con)
687 {
688 	if (memcmp(con->v1.in_banner, CEPH_BANNER, strlen(CEPH_BANNER))) {
689 		pr_err("connect to %s got bad banner\n",
690 		       ceph_pr_addr(&con->peer_addr));
691 		con->error_msg = "protocol error, bad banner";
692 		return -1;
693 	}
694 	return 0;
695 }
696 
process_banner(struct ceph_connection * con)697 static int process_banner(struct ceph_connection *con)
698 {
699 	struct ceph_entity_addr *my_addr = &con->msgr->inst.addr;
700 
701 	dout("process_banner on %p\n", con);
702 
703 	if (verify_hello(con) < 0)
704 		return -1;
705 
706 	/*
707 	 * Make sure the other end is who we wanted.  note that the other
708 	 * end may not yet know their ip address, so if it's 0.0.0.0, give
709 	 * them the benefit of the doubt.
710 	 */
711 	if (memcmp(&con->peer_addr, &con->v1.actual_peer_addr,
712 		   sizeof(con->peer_addr)) != 0 &&
713 	    !(ceph_addr_is_blank(&con->v1.actual_peer_addr) &&
714 	      con->v1.actual_peer_addr.nonce == con->peer_addr.nonce)) {
715 		pr_warn("wrong peer, want %s/%u, got %s/%u\n",
716 			ceph_pr_addr(&con->peer_addr),
717 			le32_to_cpu(con->peer_addr.nonce),
718 			ceph_pr_addr(&con->v1.actual_peer_addr),
719 			le32_to_cpu(con->v1.actual_peer_addr.nonce));
720 		con->error_msg = "wrong peer at address";
721 		return -1;
722 	}
723 
724 	/*
725 	 * did we learn our address?
726 	 */
727 	if (ceph_addr_is_blank(my_addr)) {
728 		memcpy(&my_addr->in_addr,
729 		       &con->v1.peer_addr_for_me.in_addr,
730 		       sizeof(con->v1.peer_addr_for_me.in_addr));
731 		ceph_addr_set_port(my_addr, 0);
732 		ceph_encode_my_addr(con->msgr);
733 		dout("process_banner learned my addr is %s\n",
734 		     ceph_pr_addr(my_addr));
735 	}
736 
737 	return 0;
738 }
739 
process_connect(struct ceph_connection * con)740 static int process_connect(struct ceph_connection *con)
741 {
742 	u64 sup_feat = from_msgr(con->msgr)->supported_features;
743 	u64 req_feat = from_msgr(con->msgr)->required_features;
744 	u64 server_feat = le64_to_cpu(con->v1.in_reply.features);
745 	int ret;
746 
747 	dout("process_connect on %p tag %d\n", con, con->v1.in_tag);
748 
749 	if (con->v1.auth) {
750 		int len = le32_to_cpu(con->v1.in_reply.authorizer_len);
751 
752 		/*
753 		 * Any connection that defines ->get_authorizer()
754 		 * should also define ->add_authorizer_challenge() and
755 		 * ->verify_authorizer_reply().
756 		 *
757 		 * See get_connect_authorizer().
758 		 */
759 		if (con->v1.in_reply.tag ==
760 				CEPH_MSGR_TAG_CHALLENGE_AUTHORIZER) {
761 			ret = con->ops->add_authorizer_challenge(
762 				con, con->v1.auth->authorizer_reply_buf, len);
763 			if (ret < 0)
764 				return ret;
765 
766 			con_out_kvec_reset(con);
767 			__prepare_write_connect(con);
768 			prepare_read_connect(con);
769 			return 0;
770 		}
771 
772 		if (len) {
773 			ret = con->ops->verify_authorizer_reply(con);
774 			if (ret < 0) {
775 				con->error_msg = "bad authorize reply";
776 				return ret;
777 			}
778 		}
779 	}
780 
781 	switch (con->v1.in_reply.tag) {
782 	case CEPH_MSGR_TAG_FEATURES:
783 		pr_err("%s%lld %s feature set mismatch,"
784 		       " my %llx < server's %llx, missing %llx\n",
785 		       ENTITY_NAME(con->peer_name),
786 		       ceph_pr_addr(&con->peer_addr),
787 		       sup_feat, server_feat, server_feat & ~sup_feat);
788 		con->error_msg = "missing required protocol features";
789 		return -1;
790 
791 	case CEPH_MSGR_TAG_BADPROTOVER:
792 		pr_err("%s%lld %s protocol version mismatch,"
793 		       " my %d != server's %d\n",
794 		       ENTITY_NAME(con->peer_name),
795 		       ceph_pr_addr(&con->peer_addr),
796 		       le32_to_cpu(con->v1.out_connect.protocol_version),
797 		       le32_to_cpu(con->v1.in_reply.protocol_version));
798 		con->error_msg = "protocol version mismatch";
799 		return -1;
800 
801 	case CEPH_MSGR_TAG_BADAUTHORIZER:
802 		con->v1.auth_retry++;
803 		dout("process_connect %p got BADAUTHORIZER attempt %d\n", con,
804 		     con->v1.auth_retry);
805 		if (con->v1.auth_retry == 2) {
806 			con->error_msg = "connect authorization failure";
807 			return -1;
808 		}
809 		con_out_kvec_reset(con);
810 		ret = prepare_write_connect(con);
811 		if (ret < 0)
812 			return ret;
813 		prepare_read_connect(con);
814 		break;
815 
816 	case CEPH_MSGR_TAG_RESETSESSION:
817 		/*
818 		 * If we connected with a large connect_seq but the peer
819 		 * has no record of a session with us (no connection, or
820 		 * connect_seq == 0), they will send RESETSESION to indicate
821 		 * that they must have reset their session, and may have
822 		 * dropped messages.
823 		 */
824 		dout("process_connect got RESET peer seq %u\n",
825 		     le32_to_cpu(con->v1.in_reply.connect_seq));
826 		pr_info("%s%lld %s session reset\n",
827 			ENTITY_NAME(con->peer_name),
828 			ceph_pr_addr(&con->peer_addr));
829 		ceph_con_reset_session(con);
830 		con_out_kvec_reset(con);
831 		ret = prepare_write_connect(con);
832 		if (ret < 0)
833 			return ret;
834 		prepare_read_connect(con);
835 
836 		/* Tell ceph about it. */
837 		mutex_unlock(&con->mutex);
838 		if (con->ops->peer_reset)
839 			con->ops->peer_reset(con);
840 		mutex_lock(&con->mutex);
841 		if (con->state != CEPH_CON_S_V1_CONNECT_MSG)
842 			return -EAGAIN;
843 		break;
844 
845 	case CEPH_MSGR_TAG_RETRY_SESSION:
846 		/*
847 		 * If we sent a smaller connect_seq than the peer has, try
848 		 * again with a larger value.
849 		 */
850 		dout("process_connect got RETRY_SESSION my seq %u, peer %u\n",
851 		     le32_to_cpu(con->v1.out_connect.connect_seq),
852 		     le32_to_cpu(con->v1.in_reply.connect_seq));
853 		con->v1.connect_seq = le32_to_cpu(con->v1.in_reply.connect_seq);
854 		con_out_kvec_reset(con);
855 		ret = prepare_write_connect(con);
856 		if (ret < 0)
857 			return ret;
858 		prepare_read_connect(con);
859 		break;
860 
861 	case CEPH_MSGR_TAG_RETRY_GLOBAL:
862 		/*
863 		 * If we sent a smaller global_seq than the peer has, try
864 		 * again with a larger value.
865 		 */
866 		dout("process_connect got RETRY_GLOBAL my %u peer_gseq %u\n",
867 		     con->v1.peer_global_seq,
868 		     le32_to_cpu(con->v1.in_reply.global_seq));
869 		ceph_get_global_seq(con->msgr,
870 				    le32_to_cpu(con->v1.in_reply.global_seq));
871 		con_out_kvec_reset(con);
872 		ret = prepare_write_connect(con);
873 		if (ret < 0)
874 			return ret;
875 		prepare_read_connect(con);
876 		break;
877 
878 	case CEPH_MSGR_TAG_SEQ:
879 	case CEPH_MSGR_TAG_READY:
880 		if (req_feat & ~server_feat) {
881 			pr_err("%s%lld %s protocol feature mismatch,"
882 			       " my required %llx > server's %llx, need %llx\n",
883 			       ENTITY_NAME(con->peer_name),
884 			       ceph_pr_addr(&con->peer_addr),
885 			       req_feat, server_feat, req_feat & ~server_feat);
886 			con->error_msg = "missing required protocol features";
887 			return -1;
888 		}
889 
890 		WARN_ON(con->state != CEPH_CON_S_V1_CONNECT_MSG);
891 		con->state = CEPH_CON_S_OPEN;
892 		con->v1.auth_retry = 0;    /* we authenticated; clear flag */
893 		con->v1.peer_global_seq =
894 			le32_to_cpu(con->v1.in_reply.global_seq);
895 		con->v1.connect_seq++;
896 		con->peer_features = server_feat;
897 		dout("process_connect got READY gseq %d cseq %d (%d)\n",
898 		     con->v1.peer_global_seq,
899 		     le32_to_cpu(con->v1.in_reply.connect_seq),
900 		     con->v1.connect_seq);
901 		WARN_ON(con->v1.connect_seq !=
902 			le32_to_cpu(con->v1.in_reply.connect_seq));
903 
904 		if (con->v1.in_reply.flags & CEPH_MSG_CONNECT_LOSSY)
905 			ceph_con_flag_set(con, CEPH_CON_F_LOSSYTX);
906 
907 		con->delay = 0;      /* reset backoff memory */
908 
909 		if (con->v1.in_reply.tag == CEPH_MSGR_TAG_SEQ) {
910 			prepare_write_seq(con);
911 			prepare_read_seq(con);
912 		} else {
913 			prepare_read_tag(con);
914 		}
915 		break;
916 
917 	case CEPH_MSGR_TAG_WAIT:
918 		/*
919 		 * If there is a connection race (we are opening
920 		 * connections to each other), one of us may just have
921 		 * to WAIT.  This shouldn't happen if we are the
922 		 * client.
923 		 */
924 		con->error_msg = "protocol error, got WAIT as client";
925 		return -1;
926 
927 	default:
928 		con->error_msg = "protocol error, garbage tag during connect";
929 		return -1;
930 	}
931 	return 0;
932 }
933 
934 /*
935  * read (part of) an ack
936  */
read_partial_ack(struct ceph_connection * con)937 static int read_partial_ack(struct ceph_connection *con)
938 {
939 	int size = sizeof(con->v1.in_temp_ack);
940 	int end = size;
941 
942 	return read_partial(con, end, size, &con->v1.in_temp_ack);
943 }
944 
945 /*
946  * We can finally discard anything that's been acked.
947  */
process_ack(struct ceph_connection * con)948 static void process_ack(struct ceph_connection *con)
949 {
950 	u64 ack = le64_to_cpu(con->v1.in_temp_ack);
951 
952 	if (con->v1.in_tag == CEPH_MSGR_TAG_ACK)
953 		ceph_con_discard_sent(con, ack);
954 	else
955 		ceph_con_discard_requeued(con, ack);
956 
957 	prepare_read_tag(con);
958 }
959 
read_partial_message_chunk(struct ceph_connection * con,struct kvec * section,unsigned int sec_len,u32 * crc)960 static int read_partial_message_chunk(struct ceph_connection *con,
961 				      struct kvec *section,
962 				      unsigned int sec_len, u32 *crc)
963 {
964 	int ret, left;
965 
966 	BUG_ON(!section);
967 
968 	while (section->iov_len < sec_len) {
969 		BUG_ON(section->iov_base == NULL);
970 		left = sec_len - section->iov_len;
971 		ret = ceph_tcp_recvmsg(con->sock, (char *)section->iov_base +
972 				       section->iov_len, left);
973 		if (ret <= 0)
974 			return ret;
975 		section->iov_len += ret;
976 	}
977 	if (section->iov_len == sec_len)
978 		*crc = crc32c(*crc, section->iov_base, section->iov_len);
979 
980 	return 1;
981 }
982 
read_partial_message_section(struct ceph_connection * con,struct kvec * section,unsigned int sec_len,u32 * crc)983 static inline int read_partial_message_section(struct ceph_connection *con,
984 					       struct kvec *section,
985 					       unsigned int sec_len, u32 *crc)
986 {
987 	*crc = 0;
988 	return read_partial_message_chunk(con, section, sec_len, crc);
989 }
990 
read_partial_sparse_msg_extent(struct ceph_connection * con,u32 * crc)991 static int read_partial_sparse_msg_extent(struct ceph_connection *con, u32 *crc)
992 {
993 	struct ceph_msg_data_cursor *cursor = &con->in_msg->cursor;
994 	bool do_bounce = ceph_test_opt(from_msgr(con->msgr), RXBOUNCE);
995 
996 	if (do_bounce && unlikely(!con->bounce_page)) {
997 		con->bounce_page = alloc_page(GFP_NOIO);
998 		if (!con->bounce_page) {
999 			pr_err("failed to allocate bounce page\n");
1000 			return -ENOMEM;
1001 		}
1002 	}
1003 
1004 	while (cursor->sr_resid > 0) {
1005 		struct page *page, *rpage;
1006 		size_t off, len;
1007 		int ret;
1008 
1009 		page = ceph_msg_data_next(cursor, &off, &len);
1010 		rpage = do_bounce ? con->bounce_page : page;
1011 
1012 		/* clamp to what remains in extent */
1013 		len = min_t(int, len, cursor->sr_resid);
1014 		ret = ceph_tcp_recvpage(con->sock, rpage, (int)off, len);
1015 		if (ret <= 0)
1016 			return ret;
1017 		*crc = ceph_crc32c_page(*crc, rpage, off, ret);
1018 		ceph_msg_data_advance(cursor, (size_t)ret);
1019 		cursor->sr_resid -= ret;
1020 		if (do_bounce)
1021 			memcpy_page(page, off, rpage, off, ret);
1022 	}
1023 	return 1;
1024 }
1025 
read_partial_sparse_msg_data(struct ceph_connection * con)1026 static int read_partial_sparse_msg_data(struct ceph_connection *con)
1027 {
1028 	struct ceph_msg_data_cursor *cursor = &con->in_msg->cursor;
1029 	bool do_datacrc = !ceph_test_opt(from_msgr(con->msgr), NOCRC);
1030 	u32 crc = 0;
1031 	int ret = 1;
1032 
1033 	if (do_datacrc)
1034 		crc = con->in_data_crc;
1035 
1036 	while (cursor->total_resid) {
1037 		if (con->v1.in_sr_kvec.iov_base)
1038 			ret = read_partial_message_chunk(con,
1039 							 &con->v1.in_sr_kvec,
1040 							 con->v1.in_sr_len,
1041 							 &crc);
1042 		else if (cursor->sr_resid > 0)
1043 			ret = read_partial_sparse_msg_extent(con, &crc);
1044 		if (ret <= 0)
1045 			break;
1046 
1047 		memset(&con->v1.in_sr_kvec, 0, sizeof(con->v1.in_sr_kvec));
1048 		ret = con->ops->sparse_read(con, cursor,
1049 				(char **)&con->v1.in_sr_kvec.iov_base);
1050 		if (ret <= 0) {
1051 			ret = ret ? ret : 1;  /* must return > 0 to indicate success */
1052 			break;
1053 		}
1054 		con->v1.in_sr_len = ret;
1055 	}
1056 
1057 	if (do_datacrc)
1058 		con->in_data_crc = crc;
1059 
1060 	return ret;
1061 }
1062 
read_partial_msg_data(struct ceph_connection * con)1063 static int read_partial_msg_data(struct ceph_connection *con)
1064 {
1065 	struct ceph_msg_data_cursor *cursor = &con->in_msg->cursor;
1066 	bool do_datacrc = !ceph_test_opt(from_msgr(con->msgr), NOCRC);
1067 	struct page *page;
1068 	size_t page_offset;
1069 	size_t length;
1070 	u32 crc = 0;
1071 	int ret;
1072 
1073 	if (do_datacrc)
1074 		crc = con->in_data_crc;
1075 	while (cursor->total_resid) {
1076 		if (!cursor->resid) {
1077 			ceph_msg_data_advance(cursor, 0);
1078 			continue;
1079 		}
1080 
1081 		page = ceph_msg_data_next(cursor, &page_offset, &length);
1082 		ret = ceph_tcp_recvpage(con->sock, page, page_offset, length);
1083 		if (ret <= 0) {
1084 			if (do_datacrc)
1085 				con->in_data_crc = crc;
1086 
1087 			return ret;
1088 		}
1089 
1090 		if (do_datacrc)
1091 			crc = ceph_crc32c_page(crc, page, page_offset, ret);
1092 		ceph_msg_data_advance(cursor, (size_t)ret);
1093 	}
1094 	if (do_datacrc)
1095 		con->in_data_crc = crc;
1096 
1097 	return 1;	/* must return > 0 to indicate success */
1098 }
1099 
read_partial_msg_data_bounce(struct ceph_connection * con)1100 static int read_partial_msg_data_bounce(struct ceph_connection *con)
1101 {
1102 	struct ceph_msg_data_cursor *cursor = &con->in_msg->cursor;
1103 	struct page *page;
1104 	size_t off, len;
1105 	u32 crc;
1106 	int ret;
1107 
1108 	if (unlikely(!con->bounce_page)) {
1109 		con->bounce_page = alloc_page(GFP_NOIO);
1110 		if (!con->bounce_page) {
1111 			pr_err("failed to allocate bounce page\n");
1112 			return -ENOMEM;
1113 		}
1114 	}
1115 
1116 	crc = con->in_data_crc;
1117 	while (cursor->total_resid) {
1118 		if (!cursor->resid) {
1119 			ceph_msg_data_advance(cursor, 0);
1120 			continue;
1121 		}
1122 
1123 		page = ceph_msg_data_next(cursor, &off, &len);
1124 		ret = ceph_tcp_recvpage(con->sock, con->bounce_page, 0, len);
1125 		if (ret <= 0) {
1126 			con->in_data_crc = crc;
1127 			return ret;
1128 		}
1129 
1130 		crc = crc32c(crc, page_address(con->bounce_page), ret);
1131 		memcpy_to_page(page, off, page_address(con->bounce_page), ret);
1132 
1133 		ceph_msg_data_advance(cursor, ret);
1134 	}
1135 	con->in_data_crc = crc;
1136 
1137 	return 1;	/* must return > 0 to indicate success */
1138 }
1139 
1140 /*
1141  * read (part of) a message.
1142  */
read_partial_message(struct ceph_connection * con)1143 static int read_partial_message(struct ceph_connection *con)
1144 {
1145 	struct ceph_msg *m = con->in_msg;
1146 	int size;
1147 	int end;
1148 	int ret;
1149 	unsigned int front_len, middle_len, data_len;
1150 	bool do_datacrc = !ceph_test_opt(from_msgr(con->msgr), NOCRC);
1151 	bool need_sign = (con->peer_features & CEPH_FEATURE_MSG_AUTH);
1152 	u64 seq;
1153 	u32 crc;
1154 
1155 	dout("read_partial_message con %p msg %p\n", con, m);
1156 
1157 	/* header */
1158 	size = sizeof(con->v1.in_hdr);
1159 	end = size;
1160 	ret = read_partial(con, end, size, &con->v1.in_hdr);
1161 	if (ret <= 0)
1162 		return ret;
1163 
1164 	crc = crc32c(0, &con->v1.in_hdr, offsetof(struct ceph_msg_header, crc));
1165 	if (cpu_to_le32(crc) != con->v1.in_hdr.crc) {
1166 		pr_err("read_partial_message bad hdr crc %u != expected %u\n",
1167 		       crc, con->v1.in_hdr.crc);
1168 		return -EBADMSG;
1169 	}
1170 
1171 	front_len = le32_to_cpu(con->v1.in_hdr.front_len);
1172 	if (front_len > CEPH_MSG_MAX_FRONT_LEN)
1173 		return -EIO;
1174 	middle_len = le32_to_cpu(con->v1.in_hdr.middle_len);
1175 	if (middle_len > CEPH_MSG_MAX_MIDDLE_LEN)
1176 		return -EIO;
1177 	data_len = le32_to_cpu(con->v1.in_hdr.data_len);
1178 	if (data_len > CEPH_MSG_MAX_DATA_LEN)
1179 		return -EIO;
1180 
1181 	/* verify seq# */
1182 	seq = le64_to_cpu(con->v1.in_hdr.seq);
1183 	if ((s64)seq - (s64)con->in_seq < 1) {
1184 		pr_info("skipping %s%lld %s seq %lld expected %lld\n",
1185 			ENTITY_NAME(con->peer_name),
1186 			ceph_pr_addr(&con->peer_addr),
1187 			seq, con->in_seq + 1);
1188 		con->v1.in_base_pos = -front_len - middle_len - data_len -
1189 				      sizeof_footer(con);
1190 		con->v1.in_tag = CEPH_MSGR_TAG_READY;
1191 		return 1;
1192 	} else if ((s64)seq - (s64)con->in_seq > 1) {
1193 		pr_err("read_partial_message bad seq %lld expected %lld\n",
1194 		       seq, con->in_seq + 1);
1195 		con->error_msg = "bad message sequence # for incoming message";
1196 		return -EBADE;
1197 	}
1198 
1199 	/* allocate message? */
1200 	if (!con->in_msg) {
1201 		int skip = 0;
1202 
1203 		dout("got hdr type %d front %d data %d\n", con->v1.in_hdr.type,
1204 		     front_len, data_len);
1205 		ret = ceph_con_in_msg_alloc(con, &con->v1.in_hdr, &skip);
1206 		if (ret < 0)
1207 			return ret;
1208 
1209 		BUG_ON((!con->in_msg) ^ skip);
1210 		if (skip) {
1211 			/* skip this message */
1212 			dout("alloc_msg said skip message\n");
1213 			con->v1.in_base_pos = -front_len - middle_len -
1214 					      data_len - sizeof_footer(con);
1215 			con->v1.in_tag = CEPH_MSGR_TAG_READY;
1216 			con->in_seq++;
1217 			return 1;
1218 		}
1219 
1220 		BUG_ON(!con->in_msg);
1221 		BUG_ON(con->in_msg->con != con);
1222 		m = con->in_msg;
1223 		m->front.iov_len = 0;    /* haven't read it yet */
1224 		if (m->middle)
1225 			m->middle->vec.iov_len = 0;
1226 
1227 		/* prepare for data payload, if any */
1228 
1229 		if (data_len)
1230 			prepare_message_data(con->in_msg, data_len);
1231 	}
1232 
1233 	/* front */
1234 	ret = read_partial_message_section(con, &m->front, front_len,
1235 					   &con->in_front_crc);
1236 	if (ret <= 0)
1237 		return ret;
1238 
1239 	/* middle */
1240 	if (m->middle) {
1241 		ret = read_partial_message_section(con, &m->middle->vec,
1242 						   middle_len,
1243 						   &con->in_middle_crc);
1244 		if (ret <= 0)
1245 			return ret;
1246 	}
1247 
1248 	/* (page) data */
1249 	if (data_len) {
1250 		if (!m->num_data_items)
1251 			return -EIO;
1252 
1253 		if (m->sparse_read_total)
1254 			ret = read_partial_sparse_msg_data(con);
1255 		else if (ceph_test_opt(from_msgr(con->msgr), RXBOUNCE))
1256 			ret = read_partial_msg_data_bounce(con);
1257 		else
1258 			ret = read_partial_msg_data(con);
1259 		if (ret <= 0)
1260 			return ret;
1261 	}
1262 
1263 	/* footer */
1264 	size = sizeof_footer(con);
1265 	end += size;
1266 	ret = read_partial(con, end, size, &m->footer);
1267 	if (ret <= 0)
1268 		return ret;
1269 
1270 	if (!need_sign) {
1271 		m->footer.flags = m->old_footer.flags;
1272 		m->footer.sig = 0;
1273 	}
1274 
1275 	dout("read_partial_message got msg %p %d (%u) + %d (%u) + %d (%u)\n",
1276 	     m, front_len, m->footer.front_crc, middle_len,
1277 	     m->footer.middle_crc, data_len, m->footer.data_crc);
1278 
1279 	/* crc ok? */
1280 	if (con->in_front_crc != le32_to_cpu(m->footer.front_crc)) {
1281 		pr_err("read_partial_message %p front crc %u != exp. %u\n",
1282 		       m, con->in_front_crc, m->footer.front_crc);
1283 		return -EBADMSG;
1284 	}
1285 	if (con->in_middle_crc != le32_to_cpu(m->footer.middle_crc)) {
1286 		pr_err("read_partial_message %p middle crc %u != exp %u\n",
1287 		       m, con->in_middle_crc, m->footer.middle_crc);
1288 		return -EBADMSG;
1289 	}
1290 	if (do_datacrc &&
1291 	    (m->footer.flags & CEPH_MSG_FOOTER_NOCRC) == 0 &&
1292 	    con->in_data_crc != le32_to_cpu(m->footer.data_crc)) {
1293 		pr_err("read_partial_message %p data crc %u != exp. %u\n", m,
1294 		       con->in_data_crc, le32_to_cpu(m->footer.data_crc));
1295 		return -EBADMSG;
1296 	}
1297 
1298 	if (need_sign && con->ops->check_message_signature &&
1299 	    con->ops->check_message_signature(m)) {
1300 		pr_err("read_partial_message %p signature check failed\n", m);
1301 		return -EBADMSG;
1302 	}
1303 
1304 	return 1; /* done! */
1305 }
1306 
read_keepalive_ack(struct ceph_connection * con)1307 static int read_keepalive_ack(struct ceph_connection *con)
1308 {
1309 	struct ceph_timespec ceph_ts;
1310 	size_t size = sizeof(ceph_ts);
1311 	int ret = read_partial(con, size, size, &ceph_ts);
1312 	if (ret <= 0)
1313 		return ret;
1314 	ceph_decode_timespec64(&con->last_keepalive_ack, &ceph_ts);
1315 	prepare_read_tag(con);
1316 	return 1;
1317 }
1318 
1319 /*
1320  * Read what we can from the socket.
1321  */
ceph_con_v1_try_read(struct ceph_connection * con)1322 int ceph_con_v1_try_read(struct ceph_connection *con)
1323 {
1324 	int ret = -1;
1325 
1326 more:
1327 	dout("try_read start %p state %d\n", con, con->state);
1328 	if (con->state != CEPH_CON_S_V1_BANNER &&
1329 	    con->state != CEPH_CON_S_V1_CONNECT_MSG &&
1330 	    con->state != CEPH_CON_S_OPEN)
1331 		return 0;
1332 
1333 	BUG_ON(!con->sock);
1334 
1335 	dout("try_read tag %d in_base_pos %d\n", con->v1.in_tag,
1336 	     con->v1.in_base_pos);
1337 
1338 	if (con->state == CEPH_CON_S_V1_BANNER) {
1339 		ret = read_partial_banner(con);
1340 		if (ret <= 0)
1341 			goto out;
1342 		ret = process_banner(con);
1343 		if (ret < 0)
1344 			goto out;
1345 
1346 		con->state = CEPH_CON_S_V1_CONNECT_MSG;
1347 
1348 		/*
1349 		 * Received banner is good, exchange connection info.
1350 		 * Do not reset out_kvec, as sending our banner raced
1351 		 * with receiving peer banner after connect completed.
1352 		 */
1353 		ret = prepare_write_connect(con);
1354 		if (ret < 0)
1355 			goto out;
1356 		prepare_read_connect(con);
1357 
1358 		/* Send connection info before awaiting response */
1359 		goto out;
1360 	}
1361 
1362 	if (con->state == CEPH_CON_S_V1_CONNECT_MSG) {
1363 		ret = read_partial_connect(con);
1364 		if (ret <= 0)
1365 			goto out;
1366 		ret = process_connect(con);
1367 		if (ret < 0)
1368 			goto out;
1369 		goto more;
1370 	}
1371 
1372 	WARN_ON(con->state != CEPH_CON_S_OPEN);
1373 
1374 	if (con->v1.in_base_pos < 0) {
1375 		/*
1376 		 * skipping + discarding content.
1377 		 */
1378 		ret = ceph_tcp_recvmsg(con->sock, NULL, -con->v1.in_base_pos);
1379 		if (ret <= 0)
1380 			goto out;
1381 		dout("skipped %d / %d bytes\n", ret, -con->v1.in_base_pos);
1382 		con->v1.in_base_pos += ret;
1383 		if (con->v1.in_base_pos)
1384 			goto more;
1385 	}
1386 	if (con->v1.in_tag == CEPH_MSGR_TAG_READY) {
1387 		/*
1388 		 * what's next?
1389 		 */
1390 		ret = ceph_tcp_recvmsg(con->sock, &con->v1.in_tag, 1);
1391 		if (ret <= 0)
1392 			goto out;
1393 		dout("try_read got tag %d\n", con->v1.in_tag);
1394 		switch (con->v1.in_tag) {
1395 		case CEPH_MSGR_TAG_MSG:
1396 			prepare_read_message(con);
1397 			break;
1398 		case CEPH_MSGR_TAG_ACK:
1399 			prepare_read_ack(con);
1400 			break;
1401 		case CEPH_MSGR_TAG_KEEPALIVE2_ACK:
1402 			prepare_read_keepalive_ack(con);
1403 			break;
1404 		case CEPH_MSGR_TAG_CLOSE:
1405 			ceph_con_close_socket(con);
1406 			con->state = CEPH_CON_S_CLOSED;
1407 			goto out;
1408 		default:
1409 			goto bad_tag;
1410 		}
1411 	}
1412 	if (con->v1.in_tag == CEPH_MSGR_TAG_MSG) {
1413 		ret = read_partial_message(con);
1414 		if (ret <= 0) {
1415 			switch (ret) {
1416 			case -EBADMSG:
1417 				con->error_msg = "bad crc/signature";
1418 				fallthrough;
1419 			case -EBADE:
1420 				ret = -EIO;
1421 				break;
1422 			case -EIO:
1423 				con->error_msg = "io error";
1424 				break;
1425 			}
1426 			goto out;
1427 		}
1428 		if (con->v1.in_tag == CEPH_MSGR_TAG_READY)
1429 			goto more;
1430 		ceph_con_process_message(con);
1431 		if (con->state == CEPH_CON_S_OPEN)
1432 			prepare_read_tag(con);
1433 		goto more;
1434 	}
1435 	if (con->v1.in_tag == CEPH_MSGR_TAG_ACK ||
1436 	    con->v1.in_tag == CEPH_MSGR_TAG_SEQ) {
1437 		/*
1438 		 * the final handshake seq exchange is semantically
1439 		 * equivalent to an ACK
1440 		 */
1441 		ret = read_partial_ack(con);
1442 		if (ret <= 0)
1443 			goto out;
1444 		process_ack(con);
1445 		goto more;
1446 	}
1447 	if (con->v1.in_tag == CEPH_MSGR_TAG_KEEPALIVE2_ACK) {
1448 		ret = read_keepalive_ack(con);
1449 		if (ret <= 0)
1450 			goto out;
1451 		goto more;
1452 	}
1453 
1454 out:
1455 	dout("try_read done on %p ret %d\n", con, ret);
1456 	return ret;
1457 
1458 bad_tag:
1459 	pr_err("try_read bad tag %d\n", con->v1.in_tag);
1460 	con->error_msg = "protocol error, garbage tag";
1461 	ret = -1;
1462 	goto out;
1463 }
1464 
1465 /*
1466  * Write something to the socket.  Called in a worker thread when the
1467  * socket appears to be writeable and we have something ready to send.
1468  */
ceph_con_v1_try_write(struct ceph_connection * con)1469 int ceph_con_v1_try_write(struct ceph_connection *con)
1470 {
1471 	struct ceph_msg *msg;
1472 	int ret = 1;
1473 
1474 	dout("try_write start %p state %d\n", con, con->state);
1475 	if (con->state != CEPH_CON_S_PREOPEN &&
1476 	    con->state != CEPH_CON_S_V1_BANNER &&
1477 	    con->state != CEPH_CON_S_V1_CONNECT_MSG &&
1478 	    con->state != CEPH_CON_S_OPEN)
1479 		return 0;
1480 
1481 	/* open the socket first? */
1482 	if (con->state == CEPH_CON_S_PREOPEN) {
1483 		BUG_ON(con->sock);
1484 		con->state = CEPH_CON_S_V1_BANNER;
1485 
1486 		con_out_kvec_reset(con);
1487 		prepare_write_banner(con);
1488 		prepare_read_banner(con);
1489 
1490 		BUG_ON(con->in_msg);
1491 		con->v1.in_tag = CEPH_MSGR_TAG_READY;
1492 		dout("try_write initiating connect on %p new state %d\n",
1493 		     con, con->state);
1494 		ret = ceph_tcp_connect(con);
1495 		if (ret < 0) {
1496 			con->error_msg = "connect error";
1497 			goto out;
1498 		}
1499 	}
1500 
1501 more:
1502 	dout("try_write out_kvec_bytes %d\n", con->v1.out_kvec_bytes);
1503 	BUG_ON(!con->sock);
1504 
1505 	/* kvec data queued? */
1506 	if (con->v1.out_kvec_left) {
1507 		ret = write_partial_kvec(con);
1508 		if (ret <= 0)
1509 			goto out;
1510 	}
1511 	if (con->v1.out_skip) {
1512 		ret = write_partial_skip(con);
1513 		if (ret <= 0)
1514 			goto out;
1515 	}
1516 
1517 	/* msg pages? */
1518 	msg = con->out_msg;
1519 	if (msg) {
1520 		if (con->v1.out_msg_done) {
1521 			ceph_msg_put(msg);
1522 			con->out_msg = NULL;   /* we're done with this one */
1523 			goto do_next;
1524 		}
1525 
1526 		ret = write_partial_message_data(con, msg);
1527 		if (ret == 1)
1528 			goto more;  /* we need to send the footer, too! */
1529 		if (ret == 0)
1530 			goto out;
1531 		if (ret < 0) {
1532 			dout("try_write write_partial_message_data err %d\n",
1533 			     ret);
1534 			goto out;
1535 		}
1536 	}
1537 
1538 do_next:
1539 	if (con->state == CEPH_CON_S_OPEN) {
1540 		if (ceph_con_flag_test_and_clear(con,
1541 				CEPH_CON_F_KEEPALIVE_PENDING)) {
1542 			prepare_write_keepalive(con);
1543 			goto more;
1544 		}
1545 		/* is anything else pending? */
1546 		if ((msg = ceph_con_get_out_msg(con)) != NULL) {
1547 			prepare_write_message(con, msg);
1548 			goto more;
1549 		}
1550 		if (con->in_seq > con->in_seq_acked) {
1551 			prepare_write_ack(con);
1552 			goto more;
1553 		}
1554 	}
1555 
1556 	/* Nothing to do! */
1557 	ceph_con_flag_clear(con, CEPH_CON_F_WRITE_PENDING);
1558 	dout("try_write nothing else to write.\n");
1559 	ret = 0;
1560 out:
1561 	dout("try_write done on %p ret %d\n", con, ret);
1562 	return ret;
1563 }
1564 
ceph_con_v1_revoke(struct ceph_connection * con,struct ceph_msg * msg)1565 void ceph_con_v1_revoke(struct ceph_connection *con, struct ceph_msg *msg)
1566 {
1567 	WARN_ON(con->v1.out_skip);
1568 	/* footer */
1569 	if (con->v1.out_msg_done) {
1570 		con->v1.out_skip += con_out_kvec_skip(con);
1571 	} else {
1572 		WARN_ON(!msg->data_length);
1573 		con->v1.out_skip += sizeof_footer(con);
1574 	}
1575 	/* data, middle, front */
1576 	if (msg->data_length)
1577 		con->v1.out_skip += msg->cursor.total_resid;
1578 	if (msg->middle)
1579 		con->v1.out_skip += con_out_kvec_skip(con);
1580 	con->v1.out_skip += con_out_kvec_skip(con);
1581 
1582 	dout("%s con %p out_kvec_bytes %d out_skip %d\n", __func__, con,
1583 	     con->v1.out_kvec_bytes, con->v1.out_skip);
1584 }
1585 
ceph_con_v1_revoke_incoming(struct ceph_connection * con)1586 void ceph_con_v1_revoke_incoming(struct ceph_connection *con)
1587 {
1588 	unsigned int front_len = le32_to_cpu(con->v1.in_hdr.front_len);
1589 	unsigned int middle_len = le32_to_cpu(con->v1.in_hdr.middle_len);
1590 	unsigned int data_len = le32_to_cpu(con->v1.in_hdr.data_len);
1591 
1592 	/* skip rest of message */
1593 	con->v1.in_base_pos = con->v1.in_base_pos -
1594 			sizeof(struct ceph_msg_header) -
1595 			front_len -
1596 			middle_len -
1597 			data_len -
1598 			sizeof(struct ceph_msg_footer);
1599 
1600 	con->v1.in_tag = CEPH_MSGR_TAG_READY;
1601 	con->in_seq++;
1602 
1603 	dout("%s con %p in_base_pos %d\n", __func__, con, con->v1.in_base_pos);
1604 }
1605 
ceph_con_v1_opened(struct ceph_connection * con)1606 bool ceph_con_v1_opened(struct ceph_connection *con)
1607 {
1608 	return con->v1.connect_seq;
1609 }
1610 
ceph_con_v1_reset_session(struct ceph_connection * con)1611 void ceph_con_v1_reset_session(struct ceph_connection *con)
1612 {
1613 	con->v1.connect_seq = 0;
1614 	con->v1.peer_global_seq = 0;
1615 }
1616 
ceph_con_v1_reset_protocol(struct ceph_connection * con)1617 void ceph_con_v1_reset_protocol(struct ceph_connection *con)
1618 {
1619 	con->v1.out_skip = 0;
1620 }
1621