blob: 3a99b1a908df5bbb1bec67aff0c814d2ed2f2ee5 [file] [log] [blame]
David Howells0b58b8a2016-09-02 22:39:45 +01001/* AF_RXRPC sendmsg() implementation.
2 *
3 * Copyright (C) 2007, 2016 Red Hat, Inc. All Rights Reserved.
4 * Written by David Howells (dhowells@redhat.com)
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public Licence
8 * as published by the Free Software Foundation; either version
9 * 2 of the Licence, or (at your option) any later version.
10 */
11
12#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
13
14#include <linux/net.h>
15#include <linux/gfp.h>
16#include <linux/skbuff.h>
17#include <linux/export.h>
Ingo Molnar174cd4b2017-02-02 19:15:33 +010018#include <linux/sched/signal.h>
19
David Howells0b58b8a2016-09-02 22:39:45 +010020#include <net/sock.h>
21#include <net/af_rxrpc.h>
22#include "ar-internal.h"
23
David Howells3dc20f02016-09-04 13:25:21 +010024enum rxrpc_command {
25 RXRPC_CMD_SEND_DATA, /* send data message */
26 RXRPC_CMD_SEND_ABORT, /* request abort generation */
27 RXRPC_CMD_ACCEPT, /* [server] accept incoming call */
28 RXRPC_CMD_REJECT_BUSY, /* [server] reject a call as busy */
29};
30
David Howells3ab26a62017-06-07 14:41:52 +010031struct rxrpc_send_params {
David Howellse754eba2017-06-07 12:40:03 +010032 s64 tx_total_len; /* Total Tx data length (if send data) */
David Howells3ab26a62017-06-07 14:41:52 +010033 unsigned long user_call_ID; /* User's call ID */
34 u32 abort_code; /* Abort code to Tx (if abort) */
35 enum rxrpc_command command : 8; /* The command to implement */
36 bool exclusive; /* Shared or exclusive call */
37 bool upgrade; /* If the connection is upgradeable */
38};
39
David Howells0b58b8a2016-09-02 22:39:45 +010040/*
David Howellsbc5e3a52017-10-18 11:07:31 +010041 * Wait for space to appear in the Tx queue or a signal to occur.
42 */
43static int rxrpc_wait_for_tx_window_intr(struct rxrpc_sock *rx,
44 struct rxrpc_call *call,
45 long *timeo)
46{
47 for (;;) {
48 set_current_state(TASK_INTERRUPTIBLE);
49 if (call->tx_top - call->tx_hard_ack <
50 min_t(unsigned int, call->tx_winsize,
51 call->cong_cwnd + call->cong_extra))
52 return 0;
53
54 if (call->state >= RXRPC_CALL_COMPLETE)
55 return call->error;
56
57 if (signal_pending(current))
58 return sock_intr_errno(*timeo);
59
60 trace_rxrpc_transmit(call, rxrpc_transmit_wait);
61 mutex_unlock(&call->user_mutex);
62 *timeo = schedule_timeout(*timeo);
63 if (mutex_lock_interruptible(&call->user_mutex) < 0)
64 return sock_intr_errno(*timeo);
65 }
66}
67
68/*
69 * Wait for space to appear in the Tx queue uninterruptibly, but with
70 * a timeout of 2*RTT if no progress was made and a signal occurred.
71 */
72static int rxrpc_wait_for_tx_window_nonintr(struct rxrpc_sock *rx,
73 struct rxrpc_call *call)
74{
75 rxrpc_seq_t tx_start, tx_win;
76 signed long rtt2, timeout;
77 u64 rtt;
78
79 rtt = READ_ONCE(call->peer->rtt);
80 rtt2 = nsecs_to_jiffies64(rtt) * 2;
81 if (rtt2 < 1)
82 rtt2 = 1;
83
84 timeout = rtt2;
85 tx_start = READ_ONCE(call->tx_hard_ack);
86
87 for (;;) {
88 set_current_state(TASK_UNINTERRUPTIBLE);
89
90 tx_win = READ_ONCE(call->tx_hard_ack);
91 if (call->tx_top - tx_win <
92 min_t(unsigned int, call->tx_winsize,
93 call->cong_cwnd + call->cong_extra))
94 return 0;
95
96 if (call->state >= RXRPC_CALL_COMPLETE)
97 return call->error;
98
99 if (timeout == 0 &&
100 tx_win == tx_start && signal_pending(current))
101 return -EINTR;
102
103 if (tx_win != tx_start) {
104 timeout = rtt2;
105 tx_start = tx_win;
106 }
107
108 trace_rxrpc_transmit(call, rxrpc_transmit_wait);
109 timeout = schedule_timeout(timeout);
110 }
111}
112
113/*
David Howells0b58b8a2016-09-02 22:39:45 +0100114 * wait for space to appear in the transmit/ACK window
115 * - caller holds the socket locked
116 */
117static int rxrpc_wait_for_tx_window(struct rxrpc_sock *rx,
118 struct rxrpc_call *call,
David Howellsbc5e3a52017-10-18 11:07:31 +0100119 long *timeo,
120 bool waitall)
David Howells0b58b8a2016-09-02 22:39:45 +0100121{
122 DECLARE_WAITQUEUE(myself, current);
123 int ret;
124
David Howells248f2192016-09-08 11:10:12 +0100125 _enter(",{%u,%u,%u}",
126 call->tx_hard_ack, call->tx_top, call->tx_winsize);
David Howells0b58b8a2016-09-02 22:39:45 +0100127
128 add_wait_queue(&call->waitq, &myself);
129
David Howellsbc5e3a52017-10-18 11:07:31 +0100130 if (waitall)
131 ret = rxrpc_wait_for_tx_window_nonintr(rx, call);
132 else
133 ret = rxrpc_wait_for_tx_window_intr(rx, call, timeo);
David Howells0b58b8a2016-09-02 22:39:45 +0100134
135 remove_wait_queue(&call->waitq, &myself);
136 set_current_state(TASK_RUNNING);
137 _leave(" = %d", ret);
138 return ret;
139}
140
141/*
David Howells248f2192016-09-08 11:10:12 +0100142 * Schedule an instant Tx resend.
David Howells0b58b8a2016-09-02 22:39:45 +0100143 */
David Howells248f2192016-09-08 11:10:12 +0100144static inline void rxrpc_instant_resend(struct rxrpc_call *call, int ix)
David Howells0b58b8a2016-09-02 22:39:45 +0100145{
David Howells248f2192016-09-08 11:10:12 +0100146 spin_lock_bh(&call->lock);
147
148 if (call->state < RXRPC_CALL_COMPLETE) {
149 call->rxtx_annotations[ix] = RXRPC_TX_ANNO_RETRANS;
150 if (!test_and_set_bit(RXRPC_CALL_EV_RESEND, &call->events))
David Howells0b58b8a2016-09-02 22:39:45 +0100151 rxrpc_queue_call(call);
152 }
David Howells248f2192016-09-08 11:10:12 +0100153
154 spin_unlock_bh(&call->lock);
David Howells0b58b8a2016-09-02 22:39:45 +0100155}
156
157/*
David Howellse8332512017-08-29 10:18:56 +0100158 * Notify the owner of the call that the transmit phase is ended and the last
159 * packet has been queued.
160 */
161static void rxrpc_notify_end_tx(struct rxrpc_sock *rx, struct rxrpc_call *call,
162 rxrpc_notify_end_tx_t notify_end_tx)
163{
164 if (notify_end_tx)
165 notify_end_tx(&rx->sk, call, call->user_call_ID);
166}
167
168/*
David Howells248f2192016-09-08 11:10:12 +0100169 * Queue a DATA packet for transmission, set the resend timeout and send the
170 * packet immediately
David Howells0b58b8a2016-09-02 22:39:45 +0100171 */
David Howellse8332512017-08-29 10:18:56 +0100172static void rxrpc_queue_packet(struct rxrpc_sock *rx, struct rxrpc_call *call,
173 struct sk_buff *skb, bool last,
174 rxrpc_notify_end_tx_t notify_end_tx)
David Howells0b58b8a2016-09-02 22:39:45 +0100175{
176 struct rxrpc_skb_priv *sp = rxrpc_skb(skb);
David Howells248f2192016-09-08 11:10:12 +0100177 rxrpc_seq_t seq = sp->hdr.seq;
178 int ret, ix;
David Howells70790dbe2016-09-23 12:39:22 +0100179 u8 annotation = RXRPC_TX_ANNO_UNACK;
David Howells0b58b8a2016-09-02 22:39:45 +0100180
David Howells248f2192016-09-08 11:10:12 +0100181 _net("queue skb %p [%d]", skb, seq);
David Howells0b58b8a2016-09-02 22:39:45 +0100182
David Howells248f2192016-09-08 11:10:12 +0100183 ASSERTCMP(seq, ==, call->tx_top + 1);
184
David Howellsc038a582017-08-29 10:19:01 +0100185 if (last) {
David Howells70790dbe2016-09-23 12:39:22 +0100186 annotation |= RXRPC_TX_ANNO_LAST;
David Howellsc038a582017-08-29 10:19:01 +0100187 set_bit(RXRPC_CALL_TX_LASTQ, &call->flags);
188 }
David Howells70790dbe2016-09-23 12:39:22 +0100189
David Howellsb24d2892016-09-23 13:17:33 +0100190 /* We have to set the timestamp before queueing as the retransmit
191 * algorithm can see the packet as soon as we queue it.
192 */
193 skb->tstamp = ktime_get_real();
194
David Howells248f2192016-09-08 11:10:12 +0100195 ix = seq & RXRPC_RXTX_BUFF_MASK;
David Howells71f3ca42016-09-17 10:49:14 +0100196 rxrpc_get_skb(skb, rxrpc_skb_tx_got);
David Howells70790dbe2016-09-23 12:39:22 +0100197 call->rxtx_annotations[ix] = annotation;
David Howells0b58b8a2016-09-02 22:39:45 +0100198 smp_wmb();
David Howells248f2192016-09-08 11:10:12 +0100199 call->rxtx_buffer[ix] = skb;
200 call->tx_top = seq;
David Howells70790dbe2016-09-23 12:39:22 +0100201 if (last)
David Howellsa124fe32016-09-17 10:49:13 +0100202 trace_rxrpc_transmit(call, rxrpc_transmit_queue_last);
David Howells70790dbe2016-09-23 12:39:22 +0100203 else
David Howellsa124fe32016-09-17 10:49:13 +0100204 trace_rxrpc_transmit(call, rxrpc_transmit_queue);
David Howells0b58b8a2016-09-02 22:39:45 +0100205
206 if (last || call->state == RXRPC_CALL_SERVER_ACK_REQUEST) {
207 _debug("________awaiting reply/ACK__________");
208 write_lock_bh(&call->state_lock);
209 switch (call->state) {
210 case RXRPC_CALL_CLIENT_SEND_REQUEST:
211 call->state = RXRPC_CALL_CLIENT_AWAIT_REPLY;
David Howellse8332512017-08-29 10:18:56 +0100212 rxrpc_notify_end_tx(rx, call, notify_end_tx);
David Howells0b58b8a2016-09-02 22:39:45 +0100213 break;
214 case RXRPC_CALL_SERVER_ACK_REQUEST:
215 call->state = RXRPC_CALL_SERVER_SEND_REPLY;
David Howells9749fd22016-10-06 08:11:50 +0100216 call->ack_at = call->expire_at;
217 if (call->ackr_reason == RXRPC_ACK_DELAY)
218 call->ackr_reason = 0;
219 __rxrpc_set_timer(call, rxrpc_timer_init_for_send_reply,
220 ktime_get_real());
David Howells0b58b8a2016-09-02 22:39:45 +0100221 if (!last)
222 break;
Gustavo A. R. Silvae3cf39702017-10-19 16:54:48 -0500223 /* Fall through */
David Howells0b58b8a2016-09-02 22:39:45 +0100224 case RXRPC_CALL_SERVER_SEND_REPLY:
225 call->state = RXRPC_CALL_SERVER_AWAIT_ACK;
David Howellse8332512017-08-29 10:18:56 +0100226 rxrpc_notify_end_tx(rx, call, notify_end_tx);
David Howells0b58b8a2016-09-02 22:39:45 +0100227 break;
228 default:
229 break;
230 }
231 write_unlock_bh(&call->state_lock);
232 }
233
David Howells248f2192016-09-08 11:10:12 +0100234 if (seq == 1 && rxrpc_is_client_call(call))
235 rxrpc_expose_client_call(call);
236
David Howellsa1767072016-09-29 22:37:15 +0100237 ret = rxrpc_send_data_packet(call, skb, false);
David Howells0b58b8a2016-09-02 22:39:45 +0100238 if (ret < 0) {
239 _debug("need instant resend %d", ret);
David Howells248f2192016-09-08 11:10:12 +0100240 rxrpc_instant_resend(call, ix);
David Howellsdfc3da42016-09-23 12:39:23 +0100241 } else {
David Howellsdf0adc72016-09-26 22:12:49 +0100242 ktime_t now = ktime_get_real(), resend_at;
David Howellsdfc3da42016-09-23 12:39:23 +0100243
David Howellsdf0adc72016-09-26 22:12:49 +0100244 resend_at = ktime_add_ms(now, rxrpc_resend_timeout);
David Howellsdfc3da42016-09-23 12:39:23 +0100245
David Howellsdf0adc72016-09-26 22:12:49 +0100246 if (ktime_before(resend_at, call->resend_at)) {
David Howellsdfc3da42016-09-23 12:39:23 +0100247 call->resend_at = resend_at;
David Howellsdf0adc72016-09-26 22:12:49 +0100248 rxrpc_set_timer(call, rxrpc_timer_set_for_send, now);
David Howellsdfc3da42016-09-23 12:39:23 +0100249 }
David Howells0b58b8a2016-09-02 22:39:45 +0100250 }
251
David Howells71f3ca42016-09-17 10:49:14 +0100252 rxrpc_free_skb(skb, rxrpc_skb_tx_freed);
David Howells0b58b8a2016-09-02 22:39:45 +0100253 _leave("");
254}
255
256/*
David Howells0b58b8a2016-09-02 22:39:45 +0100257 * send data through a socket
258 * - must be called in process context
David Howells540b1c42017-02-27 15:43:06 +0000259 * - The caller holds the call user access mutex, but not the socket lock.
David Howells0b58b8a2016-09-02 22:39:45 +0100260 */
261static int rxrpc_send_data(struct rxrpc_sock *rx,
262 struct rxrpc_call *call,
David Howellse8332512017-08-29 10:18:56 +0100263 struct msghdr *msg, size_t len,
264 rxrpc_notify_end_tx_t notify_end_tx)
David Howells0b58b8a2016-09-02 22:39:45 +0100265{
266 struct rxrpc_skb_priv *sp;
267 struct sk_buff *skb;
268 struct sock *sk = &rx->sk;
269 long timeo;
270 bool more;
271 int ret, copied;
272
273 timeo = sock_sndtimeo(sk, msg->msg_flags & MSG_DONTWAIT);
274
275 /* this should be in poll */
276 sk_clear_bit(SOCKWQ_ASYNC_NOSPACE, sk);
277
278 if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
279 return -EPIPE;
280
281 more = msg->msg_flags & MSG_MORE;
282
David Howellse754eba2017-06-07 12:40:03 +0100283 if (call->tx_total_len != -1) {
284 if (len > call->tx_total_len)
285 return -EMSGSIZE;
286 if (!more && len != call->tx_total_len)
287 return -EMSGSIZE;
288 }
289
David Howells0b58b8a2016-09-02 22:39:45 +0100290 skb = call->tx_pending;
291 call->tx_pending = NULL;
David Howells71f3ca42016-09-17 10:49:14 +0100292 rxrpc_see_skb(skb, rxrpc_skb_tx_seen);
David Howells0b58b8a2016-09-02 22:39:45 +0100293
294 copied = 0;
295 do {
David Howells7aa51da2016-09-22 00:29:31 +0100296 /* Check to see if there's a ping ACK to reply to. */
297 if (call->ackr_reason == RXRPC_ACK_PING_RESPONSE)
David Howellsa5af7e12016-10-06 08:11:49 +0100298 rxrpc_send_ack_packet(call, false);
David Howells7aa51da2016-09-22 00:29:31 +0100299
David Howells0b58b8a2016-09-02 22:39:45 +0100300 if (!skb) {
301 size_t size, chunk, max, space;
302
303 _debug("alloc");
304
David Howells248f2192016-09-08 11:10:12 +0100305 if (call->tx_top - call->tx_hard_ack >=
David Howells57494342016-09-24 18:05:27 +0100306 min_t(unsigned int, call->tx_winsize,
307 call->cong_cwnd + call->cong_extra)) {
David Howells0b58b8a2016-09-02 22:39:45 +0100308 ret = -EAGAIN;
309 if (msg->msg_flags & MSG_DONTWAIT)
310 goto maybe_error;
311 ret = rxrpc_wait_for_tx_window(rx, call,
David Howellsbc5e3a52017-10-18 11:07:31 +0100312 &timeo,
313 msg->msg_flags & MSG_WAITALL);
David Howells0b58b8a2016-09-02 22:39:45 +0100314 if (ret < 0)
315 goto maybe_error;
316 }
317
David Howells182f5052016-09-17 10:49:12 +0100318 max = RXRPC_JUMBO_DATALEN;
David Howells0b58b8a2016-09-02 22:39:45 +0100319 max -= call->conn->security_size;
320 max &= ~(call->conn->size_align - 1UL);
321
322 chunk = max;
323 if (chunk > msg_data_left(msg) && !more)
324 chunk = msg_data_left(msg);
325
326 space = chunk + call->conn->size_align;
327 space &= ~(call->conn->size_align - 1UL);
328
David Howells5a924b82016-09-22 00:29:31 +0100329 size = space + call->conn->security_size;
David Howells0b58b8a2016-09-02 22:39:45 +0100330
331 _debug("SIZE: %zu/%zu/%zu", chunk, space, size);
332
333 /* create a buffer that we can retain until it's ACK'd */
334 skb = sock_alloc_send_skb(
335 sk, size, msg->msg_flags & MSG_DONTWAIT, &ret);
336 if (!skb)
337 goto maybe_error;
338
David Howells71f3ca42016-09-17 10:49:14 +0100339 rxrpc_new_skb(skb, rxrpc_skb_tx_new);
David Howells0b58b8a2016-09-02 22:39:45 +0100340
341 _debug("ALLOC SEND %p", skb);
342
343 ASSERTCMP(skb->mark, ==, 0);
344
David Howells5a924b82016-09-22 00:29:31 +0100345 _debug("HS: %u", call->conn->security_size);
346 skb_reserve(skb, call->conn->security_size);
347 skb->len += call->conn->security_size;
David Howells0b58b8a2016-09-02 22:39:45 +0100348
349 sp = rxrpc_skb(skb);
350 sp->remain = chunk;
351 if (sp->remain > skb_tailroom(skb))
352 sp->remain = skb_tailroom(skb);
353
354 _net("skb: hr %d, tr %d, hl %d, rm %d",
355 skb_headroom(skb),
356 skb_tailroom(skb),
357 skb_headlen(skb),
358 sp->remain);
359
360 skb->ip_summed = CHECKSUM_UNNECESSARY;
361 }
362
363 _debug("append");
364 sp = rxrpc_skb(skb);
365
366 /* append next segment of data to the current buffer */
367 if (msg_data_left(msg) > 0) {
368 int copy = skb_tailroom(skb);
369 ASSERTCMP(copy, >, 0);
370 if (copy > msg_data_left(msg))
371 copy = msg_data_left(msg);
372 if (copy > sp->remain)
373 copy = sp->remain;
374
375 _debug("add");
376 ret = skb_add_data(skb, &msg->msg_iter, copy);
377 _debug("added");
378 if (ret < 0)
379 goto efault;
380 sp->remain -= copy;
381 skb->mark += copy;
382 copied += copy;
David Howellse754eba2017-06-07 12:40:03 +0100383 if (call->tx_total_len != -1)
384 call->tx_total_len -= copy;
David Howells0b58b8a2016-09-02 22:39:45 +0100385 }
386
David Howells0b58b8a2016-09-02 22:39:45 +0100387 /* add the packet to the send queue if it's now full */
388 if (sp->remain <= 0 ||
389 (msg_data_left(msg) == 0 && !more)) {
390 struct rxrpc_connection *conn = call->conn;
391 uint32_t seq;
392 size_t pad;
393
394 /* pad out if we're using security */
395 if (conn->security_ix) {
396 pad = conn->security_size + skb->mark;
397 pad = conn->size_align - pad;
398 pad &= conn->size_align - 1;
399 _debug("pad %zu", pad);
400 if (pad)
Johannes Bergb080db52017-06-16 14:29:19 +0200401 skb_put_zero(skb, pad);
David Howells0b58b8a2016-09-02 22:39:45 +0100402 }
403
David Howells248f2192016-09-08 11:10:12 +0100404 seq = call->tx_top + 1;
David Howells0b58b8a2016-09-02 22:39:45 +0100405
David Howells0b58b8a2016-09-02 22:39:45 +0100406 sp->hdr.seq = seq;
David Howells0b58b8a2016-09-02 22:39:45 +0100407 sp->hdr._rsvd = 0;
David Howells5a924b82016-09-22 00:29:31 +0100408 sp->hdr.flags = conn->out_clientflag;
David Howells0b58b8a2016-09-02 22:39:45 +0100409
David Howells0b58b8a2016-09-02 22:39:45 +0100410 if (msg_data_left(msg) == 0 && !more)
411 sp->hdr.flags |= RXRPC_LAST_PACKET;
David Howells248f2192016-09-08 11:10:12 +0100412 else if (call->tx_top - call->tx_hard_ack <
413 call->tx_winsize)
David Howells0b58b8a2016-09-02 22:39:45 +0100414 sp->hdr.flags |= RXRPC_MORE_PACKETS;
David Howells0b58b8a2016-09-02 22:39:45 +0100415
416 ret = conn->security->secure_packet(
David Howells5a924b82016-09-22 00:29:31 +0100417 call, skb, skb->mark, skb->head);
David Howells0b58b8a2016-09-02 22:39:45 +0100418 if (ret < 0)
419 goto out;
420
David Howellse8332512017-08-29 10:18:56 +0100421 rxrpc_queue_packet(rx, call, skb,
422 !msg_data_left(msg) && !more,
423 notify_end_tx);
David Howells0b58b8a2016-09-02 22:39:45 +0100424 skb = NULL;
425 }
David Howellsc038a582017-08-29 10:19:01 +0100426
427 /* Check for the far side aborting the call or a network error
428 * occurring. If this happens, save any packet that was under
429 * construction so that in the case of a network error, the
430 * call can be retried or redirected.
431 */
432 if (call->state == RXRPC_CALL_COMPLETE) {
433 ret = call->error;
434 goto out;
435 }
David Howells0b58b8a2016-09-02 22:39:45 +0100436 } while (msg_data_left(msg) > 0);
437
438success:
439 ret = copied;
440out:
441 call->tx_pending = skb;
442 _leave(" = %d", ret);
443 return ret;
444
David Howells0b58b8a2016-09-02 22:39:45 +0100445maybe_error:
446 if (copied)
447 goto success;
448 goto out;
449
450efault:
451 ret = -EFAULT;
452 goto out;
453}
David Howellsdf423a42016-09-02 22:39:45 +0100454
455/*
456 * extract control messages from the sendmsg() control buffer
457 */
David Howells3ab26a62017-06-07 14:41:52 +0100458static int rxrpc_sendmsg_cmsg(struct msghdr *msg, struct rxrpc_send_params *p)
David Howellsdf423a42016-09-02 22:39:45 +0100459{
460 struct cmsghdr *cmsg;
461 bool got_user_ID = false;
462 int len;
463
David Howellsdf423a42016-09-02 22:39:45 +0100464 if (msg->msg_controllen == 0)
465 return -EINVAL;
466
467 for_each_cmsghdr(cmsg, msg) {
468 if (!CMSG_OK(msg, cmsg))
469 return -EINVAL;
470
yuan linyu1ff8ceb2017-01-03 20:42:17 +0800471 len = cmsg->cmsg_len - sizeof(struct cmsghdr);
David Howellsdf423a42016-09-02 22:39:45 +0100472 _debug("CMSG %d, %d, %d",
473 cmsg->cmsg_level, cmsg->cmsg_type, len);
474
475 if (cmsg->cmsg_level != SOL_RXRPC)
476 continue;
477
478 switch (cmsg->cmsg_type) {
479 case RXRPC_USER_CALL_ID:
480 if (msg->msg_flags & MSG_CMSG_COMPAT) {
481 if (len != sizeof(u32))
482 return -EINVAL;
David Howells3ab26a62017-06-07 14:41:52 +0100483 p->user_call_ID = *(u32 *)CMSG_DATA(cmsg);
David Howellsdf423a42016-09-02 22:39:45 +0100484 } else {
485 if (len != sizeof(unsigned long))
486 return -EINVAL;
David Howells3ab26a62017-06-07 14:41:52 +0100487 p->user_call_ID = *(unsigned long *)
David Howellsdf423a42016-09-02 22:39:45 +0100488 CMSG_DATA(cmsg);
489 }
David Howellsdf423a42016-09-02 22:39:45 +0100490 got_user_ID = true;
491 break;
492
493 case RXRPC_ABORT:
David Howells3ab26a62017-06-07 14:41:52 +0100494 if (p->command != RXRPC_CMD_SEND_DATA)
David Howellsdf423a42016-09-02 22:39:45 +0100495 return -EINVAL;
David Howells3ab26a62017-06-07 14:41:52 +0100496 p->command = RXRPC_CMD_SEND_ABORT;
497 if (len != sizeof(p->abort_code))
David Howellsdf423a42016-09-02 22:39:45 +0100498 return -EINVAL;
David Howells3ab26a62017-06-07 14:41:52 +0100499 p->abort_code = *(unsigned int *)CMSG_DATA(cmsg);
500 if (p->abort_code == 0)
David Howellsdf423a42016-09-02 22:39:45 +0100501 return -EINVAL;
502 break;
503
504 case RXRPC_ACCEPT:
David Howells3ab26a62017-06-07 14:41:52 +0100505 if (p->command != RXRPC_CMD_SEND_DATA)
David Howellsdf423a42016-09-02 22:39:45 +0100506 return -EINVAL;
David Howells3ab26a62017-06-07 14:41:52 +0100507 p->command = RXRPC_CMD_ACCEPT;
David Howellsdf423a42016-09-02 22:39:45 +0100508 if (len != 0)
509 return -EINVAL;
510 break;
511
512 case RXRPC_EXCLUSIVE_CALL:
David Howells3ab26a62017-06-07 14:41:52 +0100513 p->exclusive = true;
David Howellsdf423a42016-09-02 22:39:45 +0100514 if (len != 0)
515 return -EINVAL;
516 break;
David Howells4e255722017-06-05 14:30:49 +0100517
518 case RXRPC_UPGRADE_SERVICE:
David Howells3ab26a62017-06-07 14:41:52 +0100519 p->upgrade = true;
David Howells4e255722017-06-05 14:30:49 +0100520 if (len != 0)
521 return -EINVAL;
522 break;
523
David Howellse754eba2017-06-07 12:40:03 +0100524 case RXRPC_TX_LENGTH:
525 if (p->tx_total_len != -1 || len != sizeof(__s64))
526 return -EINVAL;
527 p->tx_total_len = *(__s64 *)CMSG_DATA(cmsg);
528 if (p->tx_total_len < 0)
529 return -EINVAL;
530 break;
531
David Howellsdf423a42016-09-02 22:39:45 +0100532 default:
533 return -EINVAL;
534 }
535 }
536
537 if (!got_user_ID)
538 return -EINVAL;
David Howellse754eba2017-06-07 12:40:03 +0100539 if (p->tx_total_len != -1 && p->command != RXRPC_CMD_SEND_DATA)
540 return -EINVAL;
David Howellsdf423a42016-09-02 22:39:45 +0100541 _leave(" = 0");
542 return 0;
543}
544
545/*
David Howellsdf423a42016-09-02 22:39:45 +0100546 * Create a new client call for sendmsg().
David Howells540b1c42017-02-27 15:43:06 +0000547 * - Called with the socket lock held, which it must release.
548 * - If it returns a call, the call's lock will need releasing by the caller.
David Howellsdf423a42016-09-02 22:39:45 +0100549 */
550static struct rxrpc_call *
551rxrpc_new_client_call_for_sendmsg(struct rxrpc_sock *rx, struct msghdr *msg,
David Howells3ab26a62017-06-07 14:41:52 +0100552 struct rxrpc_send_params *p)
David Howells540b1c42017-02-27 15:43:06 +0000553 __releases(&rx->sk.sk_lock.slock)
David Howellsdf423a42016-09-02 22:39:45 +0100554{
555 struct rxrpc_conn_parameters cp;
556 struct rxrpc_call *call;
557 struct key *key;
558
559 DECLARE_SOCKADDR(struct sockaddr_rxrpc *, srx, msg->msg_name);
560
561 _enter("");
562
David Howells540b1c42017-02-27 15:43:06 +0000563 if (!msg->msg_name) {
564 release_sock(&rx->sk);
David Howellsdf423a42016-09-02 22:39:45 +0100565 return ERR_PTR(-EDESTADDRREQ);
David Howells540b1c42017-02-27 15:43:06 +0000566 }
David Howellsdf423a42016-09-02 22:39:45 +0100567
568 key = rx->key;
569 if (key && !rx->key->payload.data[0])
570 key = NULL;
571
572 memset(&cp, 0, sizeof(cp));
573 cp.local = rx->local;
574 cp.key = rx->key;
575 cp.security_level = rx->min_sec_level;
David Howells3ab26a62017-06-07 14:41:52 +0100576 cp.exclusive = rx->exclusive | p->exclusive;
577 cp.upgrade = p->upgrade;
David Howellsdf423a42016-09-02 22:39:45 +0100578 cp.service_id = srx->srx_service;
David Howellse754eba2017-06-07 12:40:03 +0100579 call = rxrpc_new_client_call(rx, &cp, srx, p->user_call_ID,
580 p->tx_total_len, GFP_KERNEL);
David Howells540b1c42017-02-27 15:43:06 +0000581 /* The socket is now unlocked */
David Howellsdf423a42016-09-02 22:39:45 +0100582
583 _leave(" = %p\n", call);
584 return call;
585}
586
587/*
588 * send a message forming part of a client call through an RxRPC socket
589 * - caller holds the socket locked
590 * - the socket may be either a client socket or a server socket
591 */
592int rxrpc_do_sendmsg(struct rxrpc_sock *rx, struct msghdr *msg, size_t len)
David Howells540b1c42017-02-27 15:43:06 +0000593 __releases(&rx->sk.sk_lock.slock)
David Howellsdf423a42016-09-02 22:39:45 +0100594{
David Howells146d8fe2017-03-04 00:01:41 +0000595 enum rxrpc_call_state state;
David Howellsdf423a42016-09-02 22:39:45 +0100596 struct rxrpc_call *call;
David Howellsdf423a42016-09-02 22:39:45 +0100597 int ret;
598
David Howells3ab26a62017-06-07 14:41:52 +0100599 struct rxrpc_send_params p = {
David Howellse754eba2017-06-07 12:40:03 +0100600 .tx_total_len = -1,
David Howells3ab26a62017-06-07 14:41:52 +0100601 .user_call_ID = 0,
602 .abort_code = 0,
603 .command = RXRPC_CMD_SEND_DATA,
604 .exclusive = false,
605 .upgrade = true,
606 };
607
David Howellsdf423a42016-09-02 22:39:45 +0100608 _enter("");
609
David Howells3ab26a62017-06-07 14:41:52 +0100610 ret = rxrpc_sendmsg_cmsg(msg, &p);
David Howellsdf423a42016-09-02 22:39:45 +0100611 if (ret < 0)
David Howells540b1c42017-02-27 15:43:06 +0000612 goto error_release_sock;
David Howellsdf423a42016-09-02 22:39:45 +0100613
David Howells3ab26a62017-06-07 14:41:52 +0100614 if (p.command == RXRPC_CMD_ACCEPT) {
David Howells540b1c42017-02-27 15:43:06 +0000615 ret = -EINVAL;
David Howellsdf423a42016-09-02 22:39:45 +0100616 if (rx->sk.sk_state != RXRPC_SERVER_LISTENING)
David Howells540b1c42017-02-27 15:43:06 +0000617 goto error_release_sock;
David Howells3ab26a62017-06-07 14:41:52 +0100618 call = rxrpc_accept_call(rx, p.user_call_ID, NULL);
David Howells540b1c42017-02-27 15:43:06 +0000619 /* The socket is now unlocked. */
David Howellsdf423a42016-09-02 22:39:45 +0100620 if (IS_ERR(call))
621 return PTR_ERR(call);
David Howells03a6c822017-11-24 10:18:40 +0000622 ret = 0;
623 goto out_put_unlock;
David Howellsdf423a42016-09-02 22:39:45 +0100624 }
625
David Howells3ab26a62017-06-07 14:41:52 +0100626 call = rxrpc_find_call_by_user_ID(rx, p.user_call_ID);
David Howellsdf423a42016-09-02 22:39:45 +0100627 if (!call) {
David Howells540b1c42017-02-27 15:43:06 +0000628 ret = -EBADSLT;
David Howells3ab26a62017-06-07 14:41:52 +0100629 if (p.command != RXRPC_CMD_SEND_DATA)
David Howells540b1c42017-02-27 15:43:06 +0000630 goto error_release_sock;
David Howells3ab26a62017-06-07 14:41:52 +0100631 call = rxrpc_new_client_call_for_sendmsg(rx, msg, &p);
David Howells540b1c42017-02-27 15:43:06 +0000632 /* The socket is now unlocked... */
David Howellsdf423a42016-09-02 22:39:45 +0100633 if (IS_ERR(call))
634 return PTR_ERR(call);
David Howells540b1c42017-02-27 15:43:06 +0000635 /* ... and we have the call lock. */
636 } else {
David Howells146d8fe2017-03-04 00:01:41 +0000637 switch (READ_ONCE(call->state)) {
638 case RXRPC_CALL_UNINITIALISED:
639 case RXRPC_CALL_CLIENT_AWAIT_CONN:
640 case RXRPC_CALL_SERVER_PREALLOC:
641 case RXRPC_CALL_SERVER_SECURING:
642 case RXRPC_CALL_SERVER_ACCEPTING:
643 ret = -EBUSY;
David Howells37411ca2017-03-02 23:48:52 +0000644 goto error_release_sock;
David Howells146d8fe2017-03-04 00:01:41 +0000645 default:
646 break;
647 }
David Howells37411ca2017-03-02 23:48:52 +0000648
David Howells540b1c42017-02-27 15:43:06 +0000649 ret = mutex_lock_interruptible(&call->user_mutex);
650 release_sock(&rx->sk);
651 if (ret < 0) {
652 ret = -ERESTARTSYS;
653 goto error_put;
654 }
David Howellse754eba2017-06-07 12:40:03 +0100655
656 if (p.tx_total_len != -1) {
657 ret = -EINVAL;
658 if (call->tx_total_len != -1 ||
659 call->tx_pending ||
660 call->tx_top != 0)
661 goto error_put;
662 call->tx_total_len = p.tx_total_len;
663 }
David Howellsdf423a42016-09-02 22:39:45 +0100664 }
665
David Howells146d8fe2017-03-04 00:01:41 +0000666 state = READ_ONCE(call->state);
David Howellsdf423a42016-09-02 22:39:45 +0100667 _debug("CALL %d USR %lx ST %d on CONN %p",
David Howells146d8fe2017-03-04 00:01:41 +0000668 call->debug_id, call->user_call_ID, state, call->conn);
David Howellsdf423a42016-09-02 22:39:45 +0100669
David Howells146d8fe2017-03-04 00:01:41 +0000670 if (state >= RXRPC_CALL_COMPLETE) {
David Howellsdf423a42016-09-02 22:39:45 +0100671 /* it's too late for this call */
672 ret = -ESHUTDOWN;
David Howells3ab26a62017-06-07 14:41:52 +0100673 } else if (p.command == RXRPC_CMD_SEND_ABORT) {
David Howellsdf423a42016-09-02 22:39:45 +0100674 ret = 0;
David Howells3ab26a62017-06-07 14:41:52 +0100675 if (rxrpc_abort_call("CMD", call, 0, p.abort_code, -ECONNABORTED))
David Howells26cb02a2016-10-06 08:11:49 +0100676 ret = rxrpc_send_abort_packet(call);
David Howells3ab26a62017-06-07 14:41:52 +0100677 } else if (p.command != RXRPC_CMD_SEND_DATA) {
David Howellsdf423a42016-09-02 22:39:45 +0100678 ret = -EINVAL;
679 } else if (rxrpc_is_client_call(call) &&
David Howells146d8fe2017-03-04 00:01:41 +0000680 state != RXRPC_CALL_CLIENT_SEND_REQUEST) {
David Howellsdf423a42016-09-02 22:39:45 +0100681 /* request phase complete for this client call */
682 ret = -EPROTO;
683 } else if (rxrpc_is_service_call(call) &&
David Howells146d8fe2017-03-04 00:01:41 +0000684 state != RXRPC_CALL_SERVER_ACK_REQUEST &&
685 state != RXRPC_CALL_SERVER_SEND_REPLY) {
David Howellsdf423a42016-09-02 22:39:45 +0100686 /* Reply phase not begun or not complete for service call. */
687 ret = -EPROTO;
688 } else {
David Howellse8332512017-08-29 10:18:56 +0100689 ret = rxrpc_send_data(rx, call, msg, len, NULL);
David Howellsdf423a42016-09-02 22:39:45 +0100690 }
691
David Howells03a6c822017-11-24 10:18:40 +0000692out_put_unlock:
David Howells540b1c42017-02-27 15:43:06 +0000693 mutex_unlock(&call->user_mutex);
694error_put:
David Howellsfff724292016-09-07 14:34:21 +0100695 rxrpc_put_call(call, rxrpc_call_put);
David Howellsdf423a42016-09-02 22:39:45 +0100696 _leave(" = %d", ret);
697 return ret;
David Howells540b1c42017-02-27 15:43:06 +0000698
699error_release_sock:
700 release_sock(&rx->sk);
701 return ret;
David Howellsdf423a42016-09-02 22:39:45 +0100702}
703
704/**
705 * rxrpc_kernel_send_data - Allow a kernel service to send data on a call
706 * @sock: The socket the call is on
707 * @call: The call to send data through
708 * @msg: The data to send
709 * @len: The amount of data to send
David Howellse8332512017-08-29 10:18:56 +0100710 * @notify_end_tx: Notification that the last packet is queued.
David Howellsdf423a42016-09-02 22:39:45 +0100711 *
712 * Allow a kernel service to send data on a call. The call must be in an state
713 * appropriate to sending data. No control data should be supplied in @msg,
714 * nor should an address be supplied. MSG_MORE should be flagged if there's
715 * more data to come, otherwise this data will end the transmission phase.
716 */
717int rxrpc_kernel_send_data(struct socket *sock, struct rxrpc_call *call,
David Howellse8332512017-08-29 10:18:56 +0100718 struct msghdr *msg, size_t len,
719 rxrpc_notify_end_tx_t notify_end_tx)
David Howellsdf423a42016-09-02 22:39:45 +0100720{
721 int ret;
722
723 _enter("{%d,%s},", call->debug_id, rxrpc_call_states[call->state]);
724
725 ASSERTCMP(msg->msg_name, ==, NULL);
726 ASSERTCMP(msg->msg_control, ==, NULL);
727
David Howells540b1c42017-02-27 15:43:06 +0000728 mutex_lock(&call->user_mutex);
David Howellsdf423a42016-09-02 22:39:45 +0100729
730 _debug("CALL %d USR %lx ST %d on CONN %p",
731 call->debug_id, call->user_call_ID, call->state, call->conn);
732
David Howells146d8fe2017-03-04 00:01:41 +0000733 switch (READ_ONCE(call->state)) {
734 case RXRPC_CALL_CLIENT_SEND_REQUEST:
735 case RXRPC_CALL_SERVER_ACK_REQUEST:
736 case RXRPC_CALL_SERVER_SEND_REPLY:
David Howellse8332512017-08-29 10:18:56 +0100737 ret = rxrpc_send_data(rxrpc_sk(sock->sk), call, msg, len,
738 notify_end_tx);
David Howells146d8fe2017-03-04 00:01:41 +0000739 break;
740 case RXRPC_CALL_COMPLETE:
David Howells6fc166d2017-03-09 08:10:32 +0000741 read_lock_bh(&call->state_lock);
David Howellsbd2db2d2017-08-29 10:18:43 +0100742 ret = call->error;
David Howells6fc166d2017-03-09 08:10:32 +0000743 read_unlock_bh(&call->state_lock);
David Howells146d8fe2017-03-04 00:01:41 +0000744 break;
745 default:
David Howellsfb46f6e2017-04-06 10:12:00 +0100746 /* Request phase complete for this client call */
747 trace_rxrpc_rx_eproto(call, 0, tracepoint_string("late_send"));
David Howells146d8fe2017-03-04 00:01:41 +0000748 ret = -EPROTO;
749 break;
David Howellsdf423a42016-09-02 22:39:45 +0100750 }
751
David Howells540b1c42017-02-27 15:43:06 +0000752 mutex_unlock(&call->user_mutex);
David Howellsdf423a42016-09-02 22:39:45 +0100753 _leave(" = %d", ret);
754 return ret;
755}
756EXPORT_SYMBOL(rxrpc_kernel_send_data);
757
758/**
759 * rxrpc_kernel_abort_call - Allow a kernel service to abort a call
760 * @sock: The socket the call is on
761 * @call: The call to be aborted
762 * @abort_code: The abort code to stick into the ABORT packet
David Howells5a429762016-09-06 22:19:51 +0100763 * @error: Local error value
764 * @why: 3-char string indicating why.
David Howellsdf423a42016-09-02 22:39:45 +0100765 *
David Howells84a4c092017-04-06 10:11:59 +0100766 * Allow a kernel service to abort a call, if it's still in an abortable state
767 * and return true if the call was aborted, false if it was already complete.
David Howellsdf423a42016-09-02 22:39:45 +0100768 */
David Howells84a4c092017-04-06 10:11:59 +0100769bool rxrpc_kernel_abort_call(struct socket *sock, struct rxrpc_call *call,
David Howells5a429762016-09-06 22:19:51 +0100770 u32 abort_code, int error, const char *why)
David Howellsdf423a42016-09-02 22:39:45 +0100771{
David Howells84a4c092017-04-06 10:11:59 +0100772 bool aborted;
773
David Howells5a429762016-09-06 22:19:51 +0100774 _enter("{%d},%d,%d,%s", call->debug_id, abort_code, error, why);
David Howellsdf423a42016-09-02 22:39:45 +0100775
David Howells540b1c42017-02-27 15:43:06 +0000776 mutex_lock(&call->user_mutex);
David Howellsdf423a42016-09-02 22:39:45 +0100777
David Howells84a4c092017-04-06 10:11:59 +0100778 aborted = rxrpc_abort_call(why, call, 0, abort_code, error);
779 if (aborted)
David Howells26cb02a2016-10-06 08:11:49 +0100780 rxrpc_send_abort_packet(call);
David Howellsdf423a42016-09-02 22:39:45 +0100781
David Howells540b1c42017-02-27 15:43:06 +0000782 mutex_unlock(&call->user_mutex);
David Howells84a4c092017-04-06 10:11:59 +0100783 return aborted;
David Howellsdf423a42016-09-02 22:39:45 +0100784}
David Howellsdf423a42016-09-02 22:39:45 +0100785EXPORT_SYMBOL(rxrpc_kernel_abort_call);
David Howellse754eba2017-06-07 12:40:03 +0100786
787/**
788 * rxrpc_kernel_set_tx_length - Set the total Tx length on a call
789 * @sock: The socket the call is on
790 * @call: The call to be informed
791 * @tx_total_len: The amount of data to be transmitted for this call
792 *
793 * Allow a kernel service to set the total transmit length on a call. This
794 * allows buffer-to-packet encrypt-and-copy to be performed.
795 *
796 * This function is primarily for use for setting the reply length since the
797 * request length can be set when beginning the call.
798 */
799void rxrpc_kernel_set_tx_length(struct socket *sock, struct rxrpc_call *call,
800 s64 tx_total_len)
801{
802 WARN_ON(call->tx_total_len != -1);
803 call->tx_total_len = tx_total_len;
804}
805EXPORT_SYMBOL(rxrpc_kernel_set_tx_length);