1 /* RxRPC packet transmission 2 * 3 * Copyright (C) 2007 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 License 8 * as published by the Free Software Foundation; either version 9 * 2 of the License, 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> 18 #include <net/sock.h> 19 #include <net/af_rxrpc.h> 20 #include "ar-internal.h" 21 22 struct rxrpc_ack_buffer { 23 struct rxrpc_wire_header whdr; 24 struct rxrpc_ackpacket ack; 25 u8 acks[255]; 26 u8 pad[3]; 27 struct rxrpc_ackinfo ackinfo; 28 }; 29 30 struct rxrpc_abort_buffer { 31 struct rxrpc_wire_header whdr; 32 __be32 abort_code; 33 }; 34 35 /* 36 * Fill out an ACK packet. 37 */ 38 static size_t rxrpc_fill_out_ack(struct rxrpc_call *call, 39 struct rxrpc_ack_buffer *pkt, 40 rxrpc_seq_t *_hard_ack, 41 rxrpc_seq_t *_top, 42 u8 reason) 43 { 44 rxrpc_serial_t serial; 45 rxrpc_seq_t hard_ack, top, seq; 46 int ix; 47 u32 mtu, jmax; 48 u8 *ackp = pkt->acks; 49 50 /* Barrier against rxrpc_input_data(). */ 51 serial = call->ackr_serial; 52 hard_ack = READ_ONCE(call->rx_hard_ack); 53 top = smp_load_acquire(&call->rx_top); 54 *_hard_ack = hard_ack; 55 *_top = top; 56 57 pkt->ack.bufferSpace = htons(8); 58 pkt->ack.maxSkew = htons(call->ackr_skew); 59 pkt->ack.firstPacket = htonl(hard_ack + 1); 60 pkt->ack.previousPacket = htonl(call->ackr_prev_seq); 61 pkt->ack.serial = htonl(serial); 62 pkt->ack.reason = reason; 63 pkt->ack.nAcks = top - hard_ack; 64 65 if (reason == RXRPC_ACK_PING) 66 pkt->whdr.flags |= RXRPC_REQUEST_ACK; 67 68 if (after(top, hard_ack)) { 69 seq = hard_ack + 1; 70 do { 71 ix = seq & RXRPC_RXTX_BUFF_MASK; 72 if (call->rxtx_buffer[ix]) 73 *ackp++ = RXRPC_ACK_TYPE_ACK; 74 else 75 *ackp++ = RXRPC_ACK_TYPE_NACK; 76 seq++; 77 } while (before_eq(seq, top)); 78 } 79 80 mtu = call->conn->params.peer->if_mtu; 81 mtu -= call->conn->params.peer->hdrsize; 82 jmax = (call->nr_jumbo_bad > 3) ? 1 : rxrpc_rx_jumbo_max; 83 pkt->ackinfo.rxMTU = htonl(rxrpc_rx_mtu); 84 pkt->ackinfo.maxMTU = htonl(mtu); 85 pkt->ackinfo.rwind = htonl(call->rx_winsize); 86 pkt->ackinfo.jumbo_max = htonl(jmax); 87 88 *ackp++ = 0; 89 *ackp++ = 0; 90 *ackp++ = 0; 91 return top - hard_ack + 3; 92 } 93 94 /* 95 * Send an ACK call packet. 96 */ 97 int rxrpc_send_ack_packet(struct rxrpc_call *call, bool ping) 98 { 99 struct rxrpc_connection *conn = NULL; 100 struct rxrpc_ack_buffer *pkt; 101 struct msghdr msg; 102 struct kvec iov[2]; 103 rxrpc_serial_t serial; 104 rxrpc_seq_t hard_ack, top; 105 size_t len, n; 106 int ret; 107 u8 reason; 108 109 spin_lock_bh(&call->lock); 110 if (call->conn) 111 conn = rxrpc_get_connection_maybe(call->conn); 112 spin_unlock_bh(&call->lock); 113 if (!conn) 114 return -ECONNRESET; 115 116 pkt = kzalloc(sizeof(*pkt), GFP_KERNEL); 117 if (!pkt) { 118 rxrpc_put_connection(conn); 119 return -ENOMEM; 120 } 121 122 msg.msg_name = &call->peer->srx.transport; 123 msg.msg_namelen = call->peer->srx.transport_len; 124 msg.msg_control = NULL; 125 msg.msg_controllen = 0; 126 msg.msg_flags = 0; 127 128 pkt->whdr.epoch = htonl(conn->proto.epoch); 129 pkt->whdr.cid = htonl(call->cid); 130 pkt->whdr.callNumber = htonl(call->call_id); 131 pkt->whdr.seq = 0; 132 pkt->whdr.type = RXRPC_PACKET_TYPE_ACK; 133 pkt->whdr.flags = RXRPC_SLOW_START_OK | conn->out_clientflag; 134 pkt->whdr.userStatus = 0; 135 pkt->whdr.securityIndex = call->security_ix; 136 pkt->whdr._rsvd = 0; 137 pkt->whdr.serviceId = htons(call->service_id); 138 139 spin_lock_bh(&call->lock); 140 if (ping) { 141 reason = RXRPC_ACK_PING; 142 } else { 143 reason = call->ackr_reason; 144 if (!call->ackr_reason) { 145 spin_unlock_bh(&call->lock); 146 ret = 0; 147 goto out; 148 } 149 call->ackr_reason = 0; 150 } 151 n = rxrpc_fill_out_ack(call, pkt, &hard_ack, &top, reason); 152 153 spin_unlock_bh(&call->lock); 154 155 iov[0].iov_base = pkt; 156 iov[0].iov_len = sizeof(pkt->whdr) + sizeof(pkt->ack) + n; 157 iov[1].iov_base = &pkt->ackinfo; 158 iov[1].iov_len = sizeof(pkt->ackinfo); 159 len = iov[0].iov_len + iov[1].iov_len; 160 161 serial = atomic_inc_return(&conn->serial); 162 pkt->whdr.serial = htonl(serial); 163 trace_rxrpc_tx_ack(call, serial, 164 ntohl(pkt->ack.firstPacket), 165 ntohl(pkt->ack.serial), 166 pkt->ack.reason, pkt->ack.nAcks); 167 168 if (ping) { 169 call->ping_serial = serial; 170 smp_wmb(); 171 /* We need to stick a time in before we send the packet in case 172 * the reply gets back before kernel_sendmsg() completes - but 173 * asking UDP to send the packet can take a relatively long 174 * time, so we update the time after, on the assumption that 175 * the packet transmission is more likely to happen towards the 176 * end of the kernel_sendmsg() call. 177 */ 178 call->ping_time = ktime_get_real(); 179 set_bit(RXRPC_CALL_PINGING, &call->flags); 180 trace_rxrpc_rtt_tx(call, rxrpc_rtt_tx_ping, serial); 181 } 182 183 ret = kernel_sendmsg(conn->params.local->socket, &msg, iov, 2, len); 184 if (ping) 185 call->ping_time = ktime_get_real(); 186 187 if (call->state < RXRPC_CALL_COMPLETE) { 188 if (ret < 0) { 189 if (ping) 190 clear_bit(RXRPC_CALL_PINGING, &call->flags); 191 rxrpc_propose_ACK(call, pkt->ack.reason, 192 ntohs(pkt->ack.maxSkew), 193 ntohl(pkt->ack.serial), 194 true, true, 195 rxrpc_propose_ack_retry_tx); 196 } else { 197 spin_lock_bh(&call->lock); 198 if (after(hard_ack, call->ackr_consumed)) 199 call->ackr_consumed = hard_ack; 200 if (after(top, call->ackr_seen)) 201 call->ackr_seen = top; 202 spin_unlock_bh(&call->lock); 203 } 204 } 205 206 out: 207 rxrpc_put_connection(conn); 208 kfree(pkt); 209 return ret; 210 } 211 212 /* 213 * Send an ABORT call packet. 214 */ 215 int rxrpc_send_abort_packet(struct rxrpc_call *call) 216 { 217 struct rxrpc_connection *conn = NULL; 218 struct rxrpc_abort_buffer pkt; 219 struct msghdr msg; 220 struct kvec iov[1]; 221 rxrpc_serial_t serial; 222 int ret; 223 224 spin_lock_bh(&call->lock); 225 if (call->conn) 226 conn = rxrpc_get_connection_maybe(call->conn); 227 spin_unlock_bh(&call->lock); 228 if (!conn) 229 return -ECONNRESET; 230 231 msg.msg_name = &call->peer->srx.transport; 232 msg.msg_namelen = call->peer->srx.transport_len; 233 msg.msg_control = NULL; 234 msg.msg_controllen = 0; 235 msg.msg_flags = 0; 236 237 pkt.whdr.epoch = htonl(conn->proto.epoch); 238 pkt.whdr.cid = htonl(call->cid); 239 pkt.whdr.callNumber = htonl(call->call_id); 240 pkt.whdr.seq = 0; 241 pkt.whdr.type = RXRPC_PACKET_TYPE_ABORT; 242 pkt.whdr.flags = conn->out_clientflag; 243 pkt.whdr.userStatus = 0; 244 pkt.whdr.securityIndex = call->security_ix; 245 pkt.whdr._rsvd = 0; 246 pkt.whdr.serviceId = htons(call->service_id); 247 pkt.abort_code = htonl(call->abort_code); 248 249 iov[0].iov_base = &pkt; 250 iov[0].iov_len = sizeof(pkt); 251 252 serial = atomic_inc_return(&conn->serial); 253 pkt.whdr.serial = htonl(serial); 254 255 ret = kernel_sendmsg(conn->params.local->socket, 256 &msg, iov, 1, sizeof(pkt)); 257 258 rxrpc_put_connection(conn); 259 return ret; 260 } 261 262 /* 263 * send a packet through the transport endpoint 264 */ 265 int rxrpc_send_data_packet(struct rxrpc_call *call, struct sk_buff *skb, 266 bool retrans) 267 { 268 struct rxrpc_connection *conn = call->conn; 269 struct rxrpc_wire_header whdr; 270 struct rxrpc_skb_priv *sp = rxrpc_skb(skb); 271 struct msghdr msg; 272 struct kvec iov[2]; 273 rxrpc_serial_t serial; 274 size_t len; 275 bool lost = false; 276 int ret, opt; 277 278 _enter(",{%d}", skb->len); 279 280 /* Each transmission of a Tx packet needs a new serial number */ 281 serial = atomic_inc_return(&conn->serial); 282 283 whdr.epoch = htonl(conn->proto.epoch); 284 whdr.cid = htonl(call->cid); 285 whdr.callNumber = htonl(call->call_id); 286 whdr.seq = htonl(sp->hdr.seq); 287 whdr.serial = htonl(serial); 288 whdr.type = RXRPC_PACKET_TYPE_DATA; 289 whdr.flags = sp->hdr.flags; 290 whdr.userStatus = 0; 291 whdr.securityIndex = call->security_ix; 292 whdr._rsvd = htons(sp->hdr._rsvd); 293 whdr.serviceId = htons(call->service_id); 294 295 if (test_bit(RXRPC_CONN_PROBING_FOR_UPGRADE, &conn->flags) && 296 sp->hdr.seq == 1) 297 whdr.userStatus = RXRPC_USERSTATUS_SERVICE_UPGRADE; 298 299 iov[0].iov_base = &whdr; 300 iov[0].iov_len = sizeof(whdr); 301 iov[1].iov_base = skb->head; 302 iov[1].iov_len = skb->len; 303 len = iov[0].iov_len + iov[1].iov_len; 304 305 msg.msg_name = &call->peer->srx.transport; 306 msg.msg_namelen = call->peer->srx.transport_len; 307 msg.msg_control = NULL; 308 msg.msg_controllen = 0; 309 msg.msg_flags = 0; 310 311 /* If our RTT cache needs working on, request an ACK. Also request 312 * ACKs if a DATA packet appears to have been lost. 313 */ 314 if (!(sp->hdr.flags & RXRPC_LAST_PACKET) && 315 (retrans || 316 call->cong_mode == RXRPC_CALL_SLOW_START || 317 (call->peer->rtt_usage < 3 && sp->hdr.seq & 1) || 318 ktime_before(ktime_add_ms(call->peer->rtt_last_req, 1000), 319 ktime_get_real()))) 320 whdr.flags |= RXRPC_REQUEST_ACK; 321 322 if (IS_ENABLED(CONFIG_AF_RXRPC_INJECT_LOSS)) { 323 static int lose; 324 if ((lose++ & 7) == 7) { 325 ret = 0; 326 lost = true; 327 goto done; 328 } 329 } 330 331 _proto("Tx DATA %%%u { #%u }", serial, sp->hdr.seq); 332 333 /* send the packet with the don't fragment bit set if we currently 334 * think it's small enough */ 335 if (iov[1].iov_len >= call->peer->maxdata) 336 goto send_fragmentable; 337 338 down_read(&conn->params.local->defrag_sem); 339 /* send the packet by UDP 340 * - returns -EMSGSIZE if UDP would have to fragment the packet 341 * to go out of the interface 342 * - in which case, we'll have processed the ICMP error 343 * message and update the peer record 344 */ 345 ret = kernel_sendmsg(conn->params.local->socket, &msg, iov, 2, len); 346 347 up_read(&conn->params.local->defrag_sem); 348 if (ret == -EMSGSIZE) 349 goto send_fragmentable; 350 351 done: 352 trace_rxrpc_tx_data(call, sp->hdr.seq, serial, whdr.flags, 353 retrans, lost); 354 if (ret >= 0) { 355 ktime_t now = ktime_get_real(); 356 skb->tstamp = now; 357 smp_wmb(); 358 sp->hdr.serial = serial; 359 if (whdr.flags & RXRPC_REQUEST_ACK) { 360 call->peer->rtt_last_req = now; 361 trace_rxrpc_rtt_tx(call, rxrpc_rtt_tx_data, serial); 362 } 363 } 364 _leave(" = %d [%u]", ret, call->peer->maxdata); 365 return ret; 366 367 send_fragmentable: 368 /* attempt to send this message with fragmentation enabled */ 369 _debug("send fragment"); 370 371 down_write(&conn->params.local->defrag_sem); 372 373 switch (conn->params.local->srx.transport.family) { 374 case AF_INET: 375 opt = IP_PMTUDISC_DONT; 376 ret = kernel_setsockopt(conn->params.local->socket, 377 SOL_IP, IP_MTU_DISCOVER, 378 (char *)&opt, sizeof(opt)); 379 if (ret == 0) { 380 ret = kernel_sendmsg(conn->params.local->socket, &msg, 381 iov, 2, len); 382 383 opt = IP_PMTUDISC_DO; 384 kernel_setsockopt(conn->params.local->socket, SOL_IP, 385 IP_MTU_DISCOVER, 386 (char *)&opt, sizeof(opt)); 387 } 388 break; 389 390 #ifdef CONFIG_AF_RXRPC_IPV6 391 case AF_INET6: 392 opt = IPV6_PMTUDISC_DONT; 393 ret = kernel_setsockopt(conn->params.local->socket, 394 SOL_IPV6, IPV6_MTU_DISCOVER, 395 (char *)&opt, sizeof(opt)); 396 if (ret == 0) { 397 ret = kernel_sendmsg(conn->params.local->socket, &msg, 398 iov, 1, iov[0].iov_len); 399 400 opt = IPV6_PMTUDISC_DO; 401 kernel_setsockopt(conn->params.local->socket, 402 SOL_IPV6, IPV6_MTU_DISCOVER, 403 (char *)&opt, sizeof(opt)); 404 } 405 break; 406 #endif 407 } 408 409 up_write(&conn->params.local->defrag_sem); 410 goto done; 411 } 412 413 /* 414 * reject packets through the local endpoint 415 */ 416 void rxrpc_reject_packets(struct rxrpc_local *local) 417 { 418 struct sockaddr_rxrpc srx; 419 struct rxrpc_skb_priv *sp; 420 struct rxrpc_wire_header whdr; 421 struct sk_buff *skb; 422 struct msghdr msg; 423 struct kvec iov[2]; 424 size_t size; 425 __be32 code; 426 427 _enter("%d", local->debug_id); 428 429 iov[0].iov_base = &whdr; 430 iov[0].iov_len = sizeof(whdr); 431 iov[1].iov_base = &code; 432 iov[1].iov_len = sizeof(code); 433 size = sizeof(whdr) + sizeof(code); 434 435 msg.msg_name = &srx.transport; 436 msg.msg_control = NULL; 437 msg.msg_controllen = 0; 438 msg.msg_flags = 0; 439 440 memset(&whdr, 0, sizeof(whdr)); 441 whdr.type = RXRPC_PACKET_TYPE_ABORT; 442 443 while ((skb = skb_dequeue(&local->reject_queue))) { 444 rxrpc_see_skb(skb, rxrpc_skb_rx_seen); 445 sp = rxrpc_skb(skb); 446 447 if (rxrpc_extract_addr_from_skb(local, &srx, skb) == 0) { 448 msg.msg_namelen = srx.transport_len; 449 450 code = htonl(skb->priority); 451 452 whdr.epoch = htonl(sp->hdr.epoch); 453 whdr.cid = htonl(sp->hdr.cid); 454 whdr.callNumber = htonl(sp->hdr.callNumber); 455 whdr.serviceId = htons(sp->hdr.serviceId); 456 whdr.flags = sp->hdr.flags; 457 whdr.flags ^= RXRPC_CLIENT_INITIATED; 458 whdr.flags &= RXRPC_CLIENT_INITIATED; 459 460 kernel_sendmsg(local->socket, &msg, iov, 2, size); 461 } 462 463 rxrpc_free_skb(skb, rxrpc_skb_rx_freed); 464 } 465 466 _leave(""); 467 } 468