mirror of
https://github.com/AuxXxilium/linux_dsm_epyc7002.git
synced 2024-12-05 18:26:39 +07:00
9e5e10a225
Signed-off-by: David S. Miller <davem@davemloft.net>
406 lines
9.0 KiB
C
406 lines
9.0 KiB
C
/* connection-level event handling
|
|
*
|
|
* Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
|
|
* Written by David Howells (dhowells@redhat.com)
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version
|
|
* 2 of the License, or (at your option) any later version.
|
|
*/
|
|
|
|
#include <linux/module.h>
|
|
#include <linux/net.h>
|
|
#include <linux/skbuff.h>
|
|
#include <linux/errqueue.h>
|
|
#include <linux/udp.h>
|
|
#include <linux/in.h>
|
|
#include <linux/in6.h>
|
|
#include <linux/icmp.h>
|
|
#include <net/sock.h>
|
|
#include <net/af_rxrpc.h>
|
|
#include <net/ip.h>
|
|
#include "ar-internal.h"
|
|
|
|
/*
|
|
* pass a connection-level abort onto all calls on that connection
|
|
*/
|
|
static void rxrpc_abort_calls(struct rxrpc_connection *conn, int state,
|
|
u32 abort_code)
|
|
{
|
|
struct rxrpc_call *call;
|
|
struct rb_node *p;
|
|
|
|
_enter("{%d},%x", conn->debug_id, abort_code);
|
|
|
|
read_lock_bh(&conn->lock);
|
|
|
|
for (p = rb_first(&conn->calls); p; p = rb_next(p)) {
|
|
call = rb_entry(p, struct rxrpc_call, conn_node);
|
|
write_lock(&call->state_lock);
|
|
if (call->state <= RXRPC_CALL_COMPLETE) {
|
|
call->state = state;
|
|
call->abort_code = abort_code;
|
|
if (state == RXRPC_CALL_LOCALLY_ABORTED)
|
|
set_bit(RXRPC_CALL_CONN_ABORT, &call->events);
|
|
else
|
|
set_bit(RXRPC_CALL_RCVD_ABORT, &call->events);
|
|
rxrpc_queue_call(call);
|
|
}
|
|
write_unlock(&call->state_lock);
|
|
}
|
|
|
|
read_unlock_bh(&conn->lock);
|
|
_leave("");
|
|
}
|
|
|
|
/*
|
|
* generate a connection-level abort
|
|
*/
|
|
static int rxrpc_abort_connection(struct rxrpc_connection *conn,
|
|
u32 error, u32 abort_code)
|
|
{
|
|
struct rxrpc_header hdr;
|
|
struct msghdr msg;
|
|
struct kvec iov[2];
|
|
__be32 word;
|
|
size_t len;
|
|
int ret;
|
|
|
|
_enter("%d,,%u,%u", conn->debug_id, error, abort_code);
|
|
|
|
/* generate a connection-level abort */
|
|
spin_lock_bh(&conn->state_lock);
|
|
if (conn->state < RXRPC_CONN_REMOTELY_ABORTED) {
|
|
conn->state = RXRPC_CONN_LOCALLY_ABORTED;
|
|
conn->error = error;
|
|
spin_unlock_bh(&conn->state_lock);
|
|
} else {
|
|
spin_unlock_bh(&conn->state_lock);
|
|
_leave(" = 0 [already dead]");
|
|
return 0;
|
|
}
|
|
|
|
rxrpc_abort_calls(conn, RXRPC_CALL_LOCALLY_ABORTED, abort_code);
|
|
|
|
msg.msg_name = &conn->trans->peer->srx.transport.sin;
|
|
msg.msg_namelen = sizeof(conn->trans->peer->srx.transport.sin);
|
|
msg.msg_control = NULL;
|
|
msg.msg_controllen = 0;
|
|
msg.msg_flags = 0;
|
|
|
|
hdr.epoch = conn->epoch;
|
|
hdr.cid = conn->cid;
|
|
hdr.callNumber = 0;
|
|
hdr.seq = 0;
|
|
hdr.type = RXRPC_PACKET_TYPE_ABORT;
|
|
hdr.flags = conn->out_clientflag;
|
|
hdr.userStatus = 0;
|
|
hdr.securityIndex = conn->security_ix;
|
|
hdr._rsvd = 0;
|
|
hdr.serviceId = conn->service_id;
|
|
|
|
word = htonl(abort_code);
|
|
|
|
iov[0].iov_base = &hdr;
|
|
iov[0].iov_len = sizeof(hdr);
|
|
iov[1].iov_base = &word;
|
|
iov[1].iov_len = sizeof(word);
|
|
|
|
len = iov[0].iov_len + iov[1].iov_len;
|
|
|
|
hdr.serial = htonl(atomic_inc_return(&conn->serial));
|
|
_proto("Tx CONN ABORT %%%u { %d }", ntohl(hdr.serial), abort_code);
|
|
|
|
ret = kernel_sendmsg(conn->trans->local->socket, &msg, iov, 2, len);
|
|
if (ret < 0) {
|
|
_debug("sendmsg failed: %d", ret);
|
|
return -EAGAIN;
|
|
}
|
|
|
|
_leave(" = 0");
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* mark a call as being on a now-secured channel
|
|
* - must be called with softirqs disabled
|
|
*/
|
|
static void rxrpc_call_is_secure(struct rxrpc_call *call)
|
|
{
|
|
_enter("%p", call);
|
|
if (call) {
|
|
read_lock(&call->state_lock);
|
|
if (call->state < RXRPC_CALL_COMPLETE &&
|
|
!test_and_set_bit(RXRPC_CALL_SECURED, &call->events))
|
|
rxrpc_queue_call(call);
|
|
read_unlock(&call->state_lock);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* connection-level Rx packet processor
|
|
*/
|
|
static int rxrpc_process_event(struct rxrpc_connection *conn,
|
|
struct sk_buff *skb,
|
|
u32 *_abort_code)
|
|
{
|
|
struct rxrpc_skb_priv *sp = rxrpc_skb(skb);
|
|
__be32 tmp;
|
|
u32 serial;
|
|
int loop, ret;
|
|
|
|
if (conn->state >= RXRPC_CONN_REMOTELY_ABORTED) {
|
|
kleave(" = -ECONNABORTED [%u]", conn->state);
|
|
return -ECONNABORTED;
|
|
}
|
|
|
|
serial = ntohl(sp->hdr.serial);
|
|
|
|
_enter("{%d},{%u,%%%u},", conn->debug_id, sp->hdr.type, serial);
|
|
|
|
switch (sp->hdr.type) {
|
|
case RXRPC_PACKET_TYPE_ABORT:
|
|
if (skb_copy_bits(skb, 0, &tmp, sizeof(tmp)) < 0)
|
|
return -EPROTO;
|
|
_proto("Rx ABORT %%%u { ac=%d }", serial, ntohl(tmp));
|
|
|
|
conn->state = RXRPC_CONN_REMOTELY_ABORTED;
|
|
rxrpc_abort_calls(conn, RXRPC_CALL_REMOTELY_ABORTED,
|
|
ntohl(tmp));
|
|
return -ECONNABORTED;
|
|
|
|
case RXRPC_PACKET_TYPE_CHALLENGE:
|
|
if (conn->security)
|
|
return conn->security->respond_to_challenge(
|
|
conn, skb, _abort_code);
|
|
return -EPROTO;
|
|
|
|
case RXRPC_PACKET_TYPE_RESPONSE:
|
|
if (!conn->security)
|
|
return -EPROTO;
|
|
|
|
ret = conn->security->verify_response(conn, skb, _abort_code);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
ret = conn->security->init_connection_security(conn);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
conn->security->prime_packet_security(conn);
|
|
read_lock_bh(&conn->lock);
|
|
spin_lock(&conn->state_lock);
|
|
|
|
if (conn->state == RXRPC_CONN_SERVER_CHALLENGING) {
|
|
conn->state = RXRPC_CONN_SERVER;
|
|
for (loop = 0; loop < RXRPC_MAXCALLS; loop++)
|
|
rxrpc_call_is_secure(conn->channels[loop]);
|
|
}
|
|
|
|
spin_unlock(&conn->state_lock);
|
|
read_unlock_bh(&conn->lock);
|
|
return 0;
|
|
|
|
default:
|
|
_leave(" = -EPROTO [%u]", sp->hdr.type);
|
|
return -EPROTO;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* set up security and issue a challenge
|
|
*/
|
|
static void rxrpc_secure_connection(struct rxrpc_connection *conn)
|
|
{
|
|
u32 abort_code;
|
|
int ret;
|
|
|
|
_enter("{%d}", conn->debug_id);
|
|
|
|
ASSERT(conn->security_ix != 0);
|
|
|
|
if (!conn->key) {
|
|
_debug("set up security");
|
|
ret = rxrpc_init_server_conn_security(conn);
|
|
switch (ret) {
|
|
case 0:
|
|
break;
|
|
case -ENOENT:
|
|
abort_code = RX_CALL_DEAD;
|
|
goto abort;
|
|
default:
|
|
abort_code = RXKADNOAUTH;
|
|
goto abort;
|
|
}
|
|
}
|
|
|
|
ASSERT(conn->security != NULL);
|
|
|
|
if (conn->security->issue_challenge(conn) < 0) {
|
|
abort_code = RX_CALL_DEAD;
|
|
ret = -ENOMEM;
|
|
goto abort;
|
|
}
|
|
|
|
_leave("");
|
|
return;
|
|
|
|
abort:
|
|
_debug("abort %d, %d", ret, abort_code);
|
|
rxrpc_abort_connection(conn, -ret, abort_code);
|
|
_leave(" [aborted]");
|
|
}
|
|
|
|
/*
|
|
* connection-level event processor
|
|
*/
|
|
void rxrpc_process_connection(struct work_struct *work)
|
|
{
|
|
struct rxrpc_connection *conn =
|
|
container_of(work, struct rxrpc_connection, processor);
|
|
struct sk_buff *skb;
|
|
u32 abort_code = RX_PROTOCOL_ERROR;
|
|
int ret;
|
|
|
|
_enter("{%d}", conn->debug_id);
|
|
|
|
atomic_inc(&conn->usage);
|
|
|
|
if (test_and_clear_bit(RXRPC_CONN_CHALLENGE, &conn->events)) {
|
|
rxrpc_secure_connection(conn);
|
|
rxrpc_put_connection(conn);
|
|
}
|
|
|
|
/* go through the conn-level event packets, releasing the ref on this
|
|
* connection that each one has when we've finished with it */
|
|
while ((skb = skb_dequeue(&conn->rx_queue))) {
|
|
ret = rxrpc_process_event(conn, skb, &abort_code);
|
|
switch (ret) {
|
|
case -EPROTO:
|
|
case -EKEYEXPIRED:
|
|
case -EKEYREJECTED:
|
|
goto protocol_error;
|
|
case -EAGAIN:
|
|
goto requeue_and_leave;
|
|
case -ECONNABORTED:
|
|
default:
|
|
rxrpc_put_connection(conn);
|
|
rxrpc_free_skb(skb);
|
|
break;
|
|
}
|
|
}
|
|
|
|
out:
|
|
rxrpc_put_connection(conn);
|
|
_leave("");
|
|
return;
|
|
|
|
requeue_and_leave:
|
|
skb_queue_head(&conn->rx_queue, skb);
|
|
goto out;
|
|
|
|
protocol_error:
|
|
if (rxrpc_abort_connection(conn, -ret, abort_code) < 0)
|
|
goto requeue_and_leave;
|
|
rxrpc_put_connection(conn);
|
|
rxrpc_free_skb(skb);
|
|
_leave(" [EPROTO]");
|
|
goto out;
|
|
}
|
|
|
|
/*
|
|
* put a packet up for transport-level abort
|
|
*/
|
|
void rxrpc_reject_packet(struct rxrpc_local *local, struct sk_buff *skb)
|
|
{
|
|
CHECK_SLAB_OKAY(&local->usage);
|
|
|
|
if (!atomic_inc_not_zero(&local->usage)) {
|
|
printk("resurrected on reject\n");
|
|
BUG();
|
|
}
|
|
|
|
skb_queue_tail(&local->reject_queue, skb);
|
|
rxrpc_queue_work(&local->rejecter);
|
|
}
|
|
|
|
/*
|
|
* reject packets through the local endpoint
|
|
*/
|
|
void rxrpc_reject_packets(struct work_struct *work)
|
|
{
|
|
union {
|
|
struct sockaddr sa;
|
|
struct sockaddr_in sin;
|
|
} sa;
|
|
struct rxrpc_skb_priv *sp;
|
|
struct rxrpc_header hdr;
|
|
struct rxrpc_local *local;
|
|
struct sk_buff *skb;
|
|
struct msghdr msg;
|
|
struct kvec iov[2];
|
|
size_t size;
|
|
__be32 code;
|
|
|
|
local = container_of(work, struct rxrpc_local, rejecter);
|
|
rxrpc_get_local(local);
|
|
|
|
_enter("%d", local->debug_id);
|
|
|
|
iov[0].iov_base = &hdr;
|
|
iov[0].iov_len = sizeof(hdr);
|
|
iov[1].iov_base = &code;
|
|
iov[1].iov_len = sizeof(code);
|
|
size = sizeof(hdr) + sizeof(code);
|
|
|
|
msg.msg_name = &sa;
|
|
msg.msg_control = NULL;
|
|
msg.msg_controllen = 0;
|
|
msg.msg_flags = 0;
|
|
|
|
memset(&sa, 0, sizeof(sa));
|
|
sa.sa.sa_family = local->srx.transport.family;
|
|
switch (sa.sa.sa_family) {
|
|
case AF_INET:
|
|
msg.msg_namelen = sizeof(sa.sin);
|
|
break;
|
|
default:
|
|
msg.msg_namelen = 0;
|
|
break;
|
|
}
|
|
|
|
memset(&hdr, 0, sizeof(hdr));
|
|
hdr.type = RXRPC_PACKET_TYPE_ABORT;
|
|
|
|
while ((skb = skb_dequeue(&local->reject_queue))) {
|
|
sp = rxrpc_skb(skb);
|
|
switch (sa.sa.sa_family) {
|
|
case AF_INET:
|
|
sa.sin.sin_port = udp_hdr(skb)->source;
|
|
sa.sin.sin_addr.s_addr = ip_hdr(skb)->saddr;
|
|
code = htonl(skb->priority);
|
|
|
|
hdr.epoch = sp->hdr.epoch;
|
|
hdr.cid = sp->hdr.cid;
|
|
hdr.callNumber = sp->hdr.callNumber;
|
|
hdr.serviceId = sp->hdr.serviceId;
|
|
hdr.flags = sp->hdr.flags;
|
|
hdr.flags ^= RXRPC_CLIENT_INITIATED;
|
|
hdr.flags &= RXRPC_CLIENT_INITIATED;
|
|
|
|
kernel_sendmsg(local->socket, &msg, iov, 2, size);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
rxrpc_free_skb(skb);
|
|
rxrpc_put_local(local);
|
|
}
|
|
|
|
rxrpc_put_local(local);
|
|
_leave("");
|
|
}
|