71ed6b9dc63a83954134826a2406ffdeb9970c04
[linux-2.6-microblaze.git] / net / rxrpc / conn_event.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* connection-level event handling
3  *
4  * Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
5  * Written by David Howells (dhowells@redhat.com)
6  */
7
8 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
9
10 #include <linux/module.h>
11 #include <linux/net.h>
12 #include <linux/skbuff.h>
13 #include <linux/errqueue.h>
14 #include <net/sock.h>
15 #include <net/af_rxrpc.h>
16 #include <net/ip.h>
17 #include "ar-internal.h"
18
19 /*
20  * Retransmit terminal ACK or ABORT of the previous call.
21  */
22 static void rxrpc_conn_retransmit_call(struct rxrpc_connection *conn,
23                                        struct sk_buff *skb,
24                                        unsigned int channel)
25 {
26         struct rxrpc_skb_priv *sp = skb ? rxrpc_skb(skb) : NULL;
27         struct rxrpc_channel *chan;
28         struct msghdr msg;
29         struct kvec iov[3];
30         struct {
31                 struct rxrpc_wire_header whdr;
32                 union {
33                         __be32 abort_code;
34                         struct rxrpc_ackpacket ack;
35                 };
36         } __attribute__((packed)) pkt;
37         struct rxrpc_ackinfo ack_info;
38         size_t len;
39         int ret, ioc;
40         u32 serial, mtu, call_id, padding;
41
42         _enter("%d", conn->debug_id);
43
44         chan = &conn->channels[channel];
45
46         /* If the last call got moved on whilst we were waiting to run, just
47          * ignore this packet.
48          */
49         call_id = READ_ONCE(chan->last_call);
50         /* Sync with __rxrpc_disconnect_call() */
51         smp_rmb();
52         if (skb && call_id != sp->hdr.callNumber)
53                 return;
54
55         msg.msg_name    = &conn->peer->srx.transport;
56         msg.msg_namelen = conn->peer->srx.transport_len;
57         msg.msg_control = NULL;
58         msg.msg_controllen = 0;
59         msg.msg_flags   = 0;
60
61         iov[0].iov_base = &pkt;
62         iov[0].iov_len  = sizeof(pkt.whdr);
63         iov[1].iov_base = &padding;
64         iov[1].iov_len  = 3;
65         iov[2].iov_base = &ack_info;
66         iov[2].iov_len  = sizeof(ack_info);
67
68         pkt.whdr.epoch          = htonl(conn->proto.epoch);
69         pkt.whdr.cid            = htonl(conn->proto.cid | channel);
70         pkt.whdr.callNumber     = htonl(call_id);
71         pkt.whdr.seq            = 0;
72         pkt.whdr.type           = chan->last_type;
73         pkt.whdr.flags          = conn->out_clientflag;
74         pkt.whdr.userStatus     = 0;
75         pkt.whdr.securityIndex  = conn->security_ix;
76         pkt.whdr._rsvd          = 0;
77         pkt.whdr.serviceId      = htons(conn->service_id);
78
79         len = sizeof(pkt.whdr);
80         switch (chan->last_type) {
81         case RXRPC_PACKET_TYPE_ABORT:
82                 pkt.abort_code  = htonl(chan->last_abort);
83                 iov[0].iov_len += sizeof(pkt.abort_code);
84                 len += sizeof(pkt.abort_code);
85                 ioc = 1;
86                 break;
87
88         case RXRPC_PACKET_TYPE_ACK:
89                 mtu = conn->peer->if_mtu;
90                 mtu -= conn->peer->hdrsize;
91                 pkt.ack.bufferSpace     = 0;
92                 pkt.ack.maxSkew         = htons(skb ? skb->priority : 0);
93                 pkt.ack.firstPacket     = htonl(chan->last_seq + 1);
94                 pkt.ack.previousPacket  = htonl(chan->last_seq);
95                 pkt.ack.serial          = htonl(skb ? sp->hdr.serial : 0);
96                 pkt.ack.reason          = skb ? RXRPC_ACK_DUPLICATE : RXRPC_ACK_IDLE;
97                 pkt.ack.nAcks           = 0;
98                 ack_info.rxMTU          = htonl(rxrpc_rx_mtu);
99                 ack_info.maxMTU         = htonl(mtu);
100                 ack_info.rwind          = htonl(rxrpc_rx_window_size);
101                 ack_info.jumbo_max      = htonl(rxrpc_rx_jumbo_max);
102                 pkt.whdr.flags          |= RXRPC_SLOW_START_OK;
103                 padding                 = 0;
104                 iov[0].iov_len += sizeof(pkt.ack);
105                 len += sizeof(pkt.ack) + 3 + sizeof(ack_info);
106                 ioc = 3;
107                 break;
108
109         default:
110                 return;
111         }
112
113         /* Resync with __rxrpc_disconnect_call() and check that the last call
114          * didn't get advanced whilst we were filling out the packets.
115          */
116         smp_rmb();
117         if (READ_ONCE(chan->last_call) != call_id)
118                 return;
119
120         serial = atomic_inc_return(&conn->serial);
121         pkt.whdr.serial = htonl(serial);
122
123         switch (chan->last_type) {
124         case RXRPC_PACKET_TYPE_ABORT:
125                 break;
126         case RXRPC_PACKET_TYPE_ACK:
127                 trace_rxrpc_tx_ack(chan->call_debug_id, serial,
128                                    ntohl(pkt.ack.firstPacket),
129                                    ntohl(pkt.ack.serial),
130                                    pkt.ack.reason, 0);
131                 break;
132         }
133
134         ret = kernel_sendmsg(conn->local->socket, &msg, iov, ioc, len);
135         conn->peer->last_tx_at = ktime_get_seconds();
136         if (ret < 0)
137                 trace_rxrpc_tx_fail(chan->call_debug_id, serial, ret,
138                                     rxrpc_tx_point_call_final_resend);
139         else
140                 trace_rxrpc_tx_packet(chan->call_debug_id, &pkt.whdr,
141                                       rxrpc_tx_point_call_final_resend);
142
143         _leave("");
144 }
145
146 /*
147  * pass a connection-level abort onto all calls on that connection
148  */
149 static void rxrpc_abort_calls(struct rxrpc_connection *conn,
150                               enum rxrpc_call_completion compl,
151                               rxrpc_serial_t serial)
152 {
153         struct rxrpc_call *call;
154         int i;
155
156         _enter("{%d},%x", conn->debug_id, conn->abort_code);
157
158         spin_lock(&conn->bundle->channel_lock);
159
160         for (i = 0; i < RXRPC_MAXCALLS; i++) {
161                 call = rcu_dereference_protected(
162                         conn->channels[i].call,
163                         lockdep_is_held(&conn->bundle->channel_lock));
164                 if (call) {
165                         if (compl == RXRPC_CALL_LOCALLY_ABORTED)
166                                 trace_rxrpc_abort(call->debug_id,
167                                                   "CON", call->cid,
168                                                   call->call_id, 0,
169                                                   conn->abort_code,
170                                                   conn->error);
171                         else
172                                 trace_rxrpc_rx_abort(call, serial,
173                                                      conn->abort_code);
174                         rxrpc_set_call_completion(call, compl,
175                                                   conn->abort_code,
176                                                   conn->error);
177                 }
178         }
179
180         spin_unlock(&conn->bundle->channel_lock);
181         _leave("");
182 }
183
184 /*
185  * generate a connection-level abort
186  */
187 static int rxrpc_abort_connection(struct rxrpc_connection *conn,
188                                   int error, u32 abort_code)
189 {
190         struct rxrpc_wire_header whdr;
191         struct msghdr msg;
192         struct kvec iov[2];
193         __be32 word;
194         size_t len;
195         u32 serial;
196         int ret;
197
198         _enter("%d,,%u,%u", conn->debug_id, error, abort_code);
199
200         /* generate a connection-level abort */
201         spin_lock_bh(&conn->state_lock);
202         if (conn->state >= RXRPC_CONN_REMOTELY_ABORTED) {
203                 spin_unlock_bh(&conn->state_lock);
204                 _leave(" = 0 [already dead]");
205                 return 0;
206         }
207
208         conn->error = error;
209         conn->abort_code = abort_code;
210         conn->state = RXRPC_CONN_LOCALLY_ABORTED;
211         set_bit(RXRPC_CONN_DONT_REUSE, &conn->flags);
212         spin_unlock_bh(&conn->state_lock);
213
214         msg.msg_name    = &conn->peer->srx.transport;
215         msg.msg_namelen = conn->peer->srx.transport_len;
216         msg.msg_control = NULL;
217         msg.msg_controllen = 0;
218         msg.msg_flags   = 0;
219
220         whdr.epoch      = htonl(conn->proto.epoch);
221         whdr.cid        = htonl(conn->proto.cid);
222         whdr.callNumber = 0;
223         whdr.seq        = 0;
224         whdr.type       = RXRPC_PACKET_TYPE_ABORT;
225         whdr.flags      = conn->out_clientflag;
226         whdr.userStatus = 0;
227         whdr.securityIndex = conn->security_ix;
228         whdr._rsvd      = 0;
229         whdr.serviceId  = htons(conn->service_id);
230
231         word            = htonl(conn->abort_code);
232
233         iov[0].iov_base = &whdr;
234         iov[0].iov_len  = sizeof(whdr);
235         iov[1].iov_base = &word;
236         iov[1].iov_len  = sizeof(word);
237
238         len = iov[0].iov_len + iov[1].iov_len;
239
240         serial = atomic_inc_return(&conn->serial);
241         rxrpc_abort_calls(conn, RXRPC_CALL_LOCALLY_ABORTED, serial);
242         whdr.serial = htonl(serial);
243
244         ret = kernel_sendmsg(conn->local->socket, &msg, iov, 2, len);
245         if (ret < 0) {
246                 trace_rxrpc_tx_fail(conn->debug_id, serial, ret,
247                                     rxrpc_tx_point_conn_abort);
248                 _debug("sendmsg failed: %d", ret);
249                 return -EAGAIN;
250         }
251
252         trace_rxrpc_tx_packet(conn->debug_id, &whdr, rxrpc_tx_point_conn_abort);
253
254         conn->peer->last_tx_at = ktime_get_seconds();
255
256         _leave(" = 0");
257         return 0;
258 }
259
260 /*
261  * mark a call as being on a now-secured channel
262  * - must be called with BH's disabled.
263  */
264 static void rxrpc_call_is_secure(struct rxrpc_call *call)
265 {
266         _enter("%p", call);
267         if (call) {
268                 write_lock_bh(&call->state_lock);
269                 if (call->state == RXRPC_CALL_SERVER_SECURING) {
270                         call->state = RXRPC_CALL_SERVER_RECV_REQUEST;
271                         rxrpc_notify_socket(call);
272                 }
273                 write_unlock_bh(&call->state_lock);
274         }
275 }
276
277 /*
278  * connection-level Rx packet processor
279  */
280 static int rxrpc_process_event(struct rxrpc_connection *conn,
281                                struct sk_buff *skb,
282                                u32 *_abort_code)
283 {
284         struct rxrpc_skb_priv *sp = rxrpc_skb(skb);
285         __be32 wtmp;
286         u32 abort_code;
287         int loop, ret;
288
289         if (conn->state >= RXRPC_CONN_REMOTELY_ABORTED) {
290                 _leave(" = -ECONNABORTED [%u]", conn->state);
291                 return -ECONNABORTED;
292         }
293
294         _enter("{%d},{%u,%%%u},", conn->debug_id, sp->hdr.type, sp->hdr.serial);
295
296         switch (sp->hdr.type) {
297         case RXRPC_PACKET_TYPE_DATA:
298         case RXRPC_PACKET_TYPE_ACK:
299                 rxrpc_conn_retransmit_call(conn, skb,
300                                            sp->hdr.cid & RXRPC_CHANNELMASK);
301                 return 0;
302
303         case RXRPC_PACKET_TYPE_BUSY:
304                 /* Just ignore BUSY packets for now. */
305                 return 0;
306
307         case RXRPC_PACKET_TYPE_ABORT:
308                 if (skb_copy_bits(skb, sizeof(struct rxrpc_wire_header),
309                                   &wtmp, sizeof(wtmp)) < 0) {
310                         trace_rxrpc_rx_eproto(NULL, sp->hdr.serial,
311                                               tracepoint_string("bad_abort"));
312                         return -EPROTO;
313                 }
314                 abort_code = ntohl(wtmp);
315
316                 conn->error = -ECONNABORTED;
317                 conn->abort_code = abort_code;
318                 conn->state = RXRPC_CONN_REMOTELY_ABORTED;
319                 set_bit(RXRPC_CONN_DONT_REUSE, &conn->flags);
320                 rxrpc_abort_calls(conn, RXRPC_CALL_REMOTELY_ABORTED, sp->hdr.serial);
321                 return -ECONNABORTED;
322
323         case RXRPC_PACKET_TYPE_CHALLENGE:
324                 return conn->security->respond_to_challenge(conn, skb,
325                                                             _abort_code);
326
327         case RXRPC_PACKET_TYPE_RESPONSE:
328                 ret = conn->security->verify_response(conn, skb, _abort_code);
329                 if (ret < 0)
330                         return ret;
331
332                 ret = conn->security->init_connection_security(
333                         conn, conn->key->payload.data[0]);
334                 if (ret < 0)
335                         return ret;
336
337                 spin_lock(&conn->bundle->channel_lock);
338                 spin_lock_bh(&conn->state_lock);
339
340                 if (conn->state == RXRPC_CONN_SERVICE_CHALLENGING) {
341                         conn->state = RXRPC_CONN_SERVICE;
342                         spin_unlock_bh(&conn->state_lock);
343                         for (loop = 0; loop < RXRPC_MAXCALLS; loop++)
344                                 rxrpc_call_is_secure(
345                                         rcu_dereference_protected(
346                                                 conn->channels[loop].call,
347                                                 lockdep_is_held(&conn->bundle->channel_lock)));
348                 } else {
349                         spin_unlock_bh(&conn->state_lock);
350                 }
351
352                 spin_unlock(&conn->bundle->channel_lock);
353                 return 0;
354
355         default:
356                 trace_rxrpc_rx_eproto(NULL, sp->hdr.serial,
357                                       tracepoint_string("bad_conn_pkt"));
358                 return -EPROTO;
359         }
360 }
361
362 /*
363  * set up security and issue a challenge
364  */
365 static void rxrpc_secure_connection(struct rxrpc_connection *conn)
366 {
367         u32 abort_code;
368         int ret;
369
370         _enter("{%d}", conn->debug_id);
371
372         ASSERT(conn->security_ix != 0);
373
374         if (conn->security->issue_challenge(conn) < 0) {
375                 abort_code = RX_CALL_DEAD;
376                 ret = -ENOMEM;
377                 goto abort;
378         }
379
380         _leave("");
381         return;
382
383 abort:
384         _debug("abort %d, %d", ret, abort_code);
385         rxrpc_abort_connection(conn, ret, abort_code);
386         _leave(" [aborted]");
387 }
388
389 /*
390  * Process delayed final ACKs that we haven't subsumed into a subsequent call.
391  */
392 void rxrpc_process_delayed_final_acks(struct rxrpc_connection *conn, bool force)
393 {
394         unsigned long j = jiffies, next_j;
395         unsigned int channel;
396         bool set;
397
398 again:
399         next_j = j + LONG_MAX;
400         set = false;
401         for (channel = 0; channel < RXRPC_MAXCALLS; channel++) {
402                 struct rxrpc_channel *chan = &conn->channels[channel];
403                 unsigned long ack_at;
404
405                 if (!test_bit(RXRPC_CONN_FINAL_ACK_0 + channel, &conn->flags))
406                         continue;
407
408                 smp_rmb(); /* vs rxrpc_disconnect_client_call */
409                 ack_at = READ_ONCE(chan->final_ack_at);
410
411                 if (time_before(j, ack_at) && !force) {
412                         if (time_before(ack_at, next_j)) {
413                                 next_j = ack_at;
414                                 set = true;
415                         }
416                         continue;
417                 }
418
419                 if (test_and_clear_bit(RXRPC_CONN_FINAL_ACK_0 + channel,
420                                        &conn->flags))
421                         rxrpc_conn_retransmit_call(conn, NULL, channel);
422         }
423
424         j = jiffies;
425         if (time_before_eq(next_j, j))
426                 goto again;
427         if (set)
428                 rxrpc_reduce_conn_timer(conn, next_j);
429 }
430
431 /*
432  * connection-level event processor
433  */
434 static void rxrpc_do_process_connection(struct rxrpc_connection *conn)
435 {
436         struct sk_buff *skb;
437         u32 abort_code = RX_PROTOCOL_ERROR;
438         int ret;
439
440         if (test_and_clear_bit(RXRPC_CONN_EV_CHALLENGE, &conn->events))
441                 rxrpc_secure_connection(conn);
442
443         /* Process delayed ACKs whose time has come. */
444         if (conn->flags & RXRPC_CONN_FINAL_ACK_MASK)
445                 rxrpc_process_delayed_final_acks(conn, false);
446
447         /* go through the conn-level event packets, releasing the ref on this
448          * connection that each one has when we've finished with it */
449         while ((skb = skb_dequeue(&conn->rx_queue))) {
450                 rxrpc_see_skb(skb, rxrpc_skb_seen);
451                 ret = rxrpc_process_event(conn, skb, &abort_code);
452                 switch (ret) {
453                 case -EPROTO:
454                 case -EKEYEXPIRED:
455                 case -EKEYREJECTED:
456                         goto protocol_error;
457                 case -ENOMEM:
458                 case -EAGAIN:
459                         goto requeue_and_leave;
460                 case -ECONNABORTED:
461                 default:
462                         rxrpc_free_skb(skb, rxrpc_skb_freed);
463                         break;
464                 }
465         }
466
467         return;
468
469 requeue_and_leave:
470         skb_queue_head(&conn->rx_queue, skb);
471         return;
472
473 protocol_error:
474         if (rxrpc_abort_connection(conn, ret, abort_code) < 0)
475                 goto requeue_and_leave;
476         rxrpc_free_skb(skb, rxrpc_skb_freed);
477         return;
478 }
479
480 void rxrpc_process_connection(struct work_struct *work)
481 {
482         struct rxrpc_connection *conn =
483                 container_of(work, struct rxrpc_connection, processor);
484
485         rxrpc_see_connection(conn);
486
487         if (__rxrpc_use_local(conn->local)) {
488                 rxrpc_do_process_connection(conn);
489                 rxrpc_unuse_local(conn->local);
490         }
491
492         rxrpc_put_connection(conn);
493         _leave("");
494         return;
495 }