blob: aca75237bbcf83eb1d440bcdf1e4d3b702cff0a1 [file] [log] [blame]
// SPDX-License-Identifier: GPL-2.0-only
/*
* Copyright (c) 2018 Chelsio Communications, Inc.
*
* Written by: Atul Gupta (atul.gupta@chelsio.com)
*/
#include <linux/module.h>
#include <linux/list.h>
#include <linux/workqueue.h>
#include <linux/skbuff.h>
#include <linux/timer.h>
#include <linux/notifier.h>
#include <linux/inetdevice.h>
#include <linux/ip.h>
#include <linux/tcp.h>
#include <linux/sched/signal.h>
#include <linux/kallsyms.h>
#include <linux/kprobes.h>
#include <linux/if_vlan.h>
#include <net/inet_common.h>
#include <net/tcp.h>
#include <net/dst.h>
#include <net/tls.h>
#include "chtls.h"
#include "chtls_cm.h"
/*
* State transitions and actions for close. Note that if we are in SYN_SENT
* we remain in that state as we cannot control a connection while it's in
* SYN_SENT; such connections are allowed to establish and are then aborted.
*/
static unsigned char new_state[16] = {
/* current state: new state: action: */
/* (Invalid) */ TCP_CLOSE,
/* TCP_ESTABLISHED */ TCP_FIN_WAIT1 | TCP_ACTION_FIN,
/* TCP_SYN_SENT */ TCP_SYN_SENT,
/* TCP_SYN_RECV */ TCP_FIN_WAIT1 | TCP_ACTION_FIN,
/* TCP_FIN_WAIT1 */ TCP_FIN_WAIT1,
/* TCP_FIN_WAIT2 */ TCP_FIN_WAIT2,
/* TCP_TIME_WAIT */ TCP_CLOSE,
/* TCP_CLOSE */ TCP_CLOSE,
/* TCP_CLOSE_WAIT */ TCP_LAST_ACK | TCP_ACTION_FIN,
/* TCP_LAST_ACK */ TCP_LAST_ACK,
/* TCP_LISTEN */ TCP_CLOSE,
/* TCP_CLOSING */ TCP_CLOSING,
};
static struct chtls_sock *chtls_sock_create(struct chtls_dev *cdev)
{
struct chtls_sock *csk = kzalloc(sizeof(*csk), GFP_ATOMIC);
if (!csk)
return NULL;
csk->txdata_skb_cache = alloc_skb(TXDATA_SKB_LEN, GFP_ATOMIC);
if (!csk->txdata_skb_cache) {
kfree(csk);
return NULL;
}
kref_init(&csk->kref);
csk->cdev = cdev;
skb_queue_head_init(&csk->txq);
csk->wr_skb_head = NULL;
csk->wr_skb_tail = NULL;
csk->mss = MAX_MSS;
csk->tlshws.ofld = 1;
csk->tlshws.txkey = -1;
csk->tlshws.rxkey = -1;
csk->tlshws.mfs = TLS_MFS;
skb_queue_head_init(&csk->tlshws.sk_recv_queue);
return csk;
}
static void chtls_sock_release(struct kref *ref)
{
struct chtls_sock *csk =
container_of(ref, struct chtls_sock, kref);
kfree(csk);
}
static struct net_device *chtls_ipv4_netdev(struct chtls_dev *cdev,
struct sock *sk)
{
struct net_device *ndev = cdev->ports[0];
if (likely(!inet_sk(sk)->inet_rcv_saddr))
return ndev;
ndev = ip_dev_find(&init_net, inet_sk(sk)->inet_rcv_saddr);
if (!ndev)
return NULL;
if (is_vlan_dev(ndev))
return vlan_dev_real_dev(ndev);
return ndev;
}
static void assign_rxopt(struct sock *sk, unsigned int opt)
{
const struct chtls_dev *cdev;
struct chtls_sock *csk;
struct tcp_sock *tp;
csk = rcu_dereference_sk_user_data(sk);
tp = tcp_sk(sk);
cdev = csk->cdev;
tp->tcp_header_len = sizeof(struct tcphdr);
tp->rx_opt.mss_clamp = cdev->mtus[TCPOPT_MSS_G(opt)] - 40;
tp->mss_cache = tp->rx_opt.mss_clamp;
tp->rx_opt.tstamp_ok = TCPOPT_TSTAMP_G(opt);
tp->rx_opt.snd_wscale = TCPOPT_SACK_G(opt);
tp->rx_opt.wscale_ok = TCPOPT_WSCALE_OK_G(opt);
SND_WSCALE(tp) = TCPOPT_SND_WSCALE_G(opt);
if (!tp->rx_opt.wscale_ok)
tp->rx_opt.rcv_wscale = 0;
if (tp->rx_opt.tstamp_ok) {
tp->tcp_header_len += TCPOLEN_TSTAMP_ALIGNED;
tp->rx_opt.mss_clamp -= TCPOLEN_TSTAMP_ALIGNED;
} else if (csk->opt2 & TSTAMPS_EN_F) {
csk->opt2 &= ~TSTAMPS_EN_F;
csk->mtu_idx = TCPOPT_MSS_G(opt);
}
}
static void chtls_purge_receive_queue(struct sock *sk)
{
struct sk_buff *skb;
while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
skb_dst_set(skb, (void *)NULL);
kfree_skb(skb);
}
}
static void chtls_purge_write_queue(struct sock *sk)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
struct sk_buff *skb;
while ((skb = __skb_dequeue(&csk->txq))) {
sk->sk_wmem_queued -= skb->truesize;
__kfree_skb(skb);
}
}
static void chtls_purge_recv_queue(struct sock *sk)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
struct chtls_hws *tlsk = &csk->tlshws;
struct sk_buff *skb;
while ((skb = __skb_dequeue(&tlsk->sk_recv_queue)) != NULL) {
skb_dst_set(skb, NULL);
kfree_skb(skb);
}
}
static void abort_arp_failure(void *handle, struct sk_buff *skb)
{
struct cpl_abort_req *req = cplhdr(skb);
struct chtls_dev *cdev;
cdev = (struct chtls_dev *)handle;
req->cmd = CPL_ABORT_NO_RST;
cxgb4_ofld_send(cdev->lldi->ports[0], skb);
}
static struct sk_buff *alloc_ctrl_skb(struct sk_buff *skb, int len)
{
if (likely(skb && !skb_shared(skb) && !skb_cloned(skb))) {
__skb_trim(skb, 0);
refcount_add(2, &skb->users);
} else {
skb = alloc_skb(len, GFP_KERNEL | __GFP_NOFAIL);
}
return skb;
}
static void chtls_send_abort(struct sock *sk, int mode, struct sk_buff *skb)
{
struct cpl_abort_req *req;
struct chtls_sock *csk;
struct tcp_sock *tp;
csk = rcu_dereference_sk_user_data(sk);
tp = tcp_sk(sk);
if (!skb)
skb = alloc_ctrl_skb(csk->txdata_skb_cache, sizeof(*req));
req = (struct cpl_abort_req *)skb_put(skb, sizeof(*req));
INIT_TP_WR_CPL(req, CPL_ABORT_REQ, csk->tid);
skb_set_queue_mapping(skb, (csk->txq_idx << 1) | CPL_PRIORITY_DATA);
req->rsvd0 = htonl(tp->snd_nxt);
req->rsvd1 = !csk_flag_nochk(csk, CSK_TX_DATA_SENT);
req->cmd = mode;
t4_set_arp_err_handler(skb, csk->cdev, abort_arp_failure);
send_or_defer(sk, tp, skb, mode == CPL_ABORT_SEND_RST);
}
static void chtls_send_reset(struct sock *sk, int mode, struct sk_buff *skb)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
if (unlikely(csk_flag_nochk(csk, CSK_ABORT_SHUTDOWN) ||
!csk->cdev)) {
if (sk->sk_state == TCP_SYN_RECV)
csk_set_flag(csk, CSK_RST_ABORTED);
goto out;
}
if (!csk_flag_nochk(csk, CSK_TX_DATA_SENT)) {
struct tcp_sock *tp = tcp_sk(sk);
if (send_tx_flowc_wr(sk, 0, tp->snd_nxt, tp->rcv_nxt) < 0)
WARN_ONCE(1, "send tx flowc error");
csk_set_flag(csk, CSK_TX_DATA_SENT);
}
csk_set_flag(csk, CSK_ABORT_RPL_PENDING);
chtls_purge_write_queue(sk);
csk_set_flag(csk, CSK_ABORT_SHUTDOWN);
if (sk->sk_state != TCP_SYN_RECV)
chtls_send_abort(sk, mode, skb);
else
goto out;
return;
out:
kfree_skb(skb);
}
static void release_tcp_port(struct sock *sk)
{
if (inet_csk(sk)->icsk_bind_hash)
inet_put_port(sk);
}
static void tcp_uncork(struct sock *sk)
{
struct tcp_sock *tp = tcp_sk(sk);
if (tp->nonagle & TCP_NAGLE_CORK) {
tp->nonagle &= ~TCP_NAGLE_CORK;
chtls_tcp_push(sk, 0);
}
}
static void chtls_close_conn(struct sock *sk)
{
struct cpl_close_con_req *req;
struct chtls_sock *csk;
struct sk_buff *skb;
unsigned int tid;
unsigned int len;
len = roundup(sizeof(struct cpl_close_con_req), 16);
csk = rcu_dereference_sk_user_data(sk);
tid = csk->tid;
skb = alloc_skb(len, GFP_KERNEL | __GFP_NOFAIL);
req = (struct cpl_close_con_req *)__skb_put(skb, len);
memset(req, 0, len);
req->wr.wr_hi = htonl(FW_WR_OP_V(FW_TP_WR) |
FW_WR_IMMDLEN_V(sizeof(*req) -
sizeof(req->wr)));
req->wr.wr_mid = htonl(FW_WR_LEN16_V(DIV_ROUND_UP(sizeof(*req), 16)) |
FW_WR_FLOWID_V(tid));
OPCODE_TID(req) = htonl(MK_OPCODE_TID(CPL_CLOSE_CON_REQ, tid));
tcp_uncork(sk);
skb_entail(sk, skb, ULPCB_FLAG_NO_HDR | ULPCB_FLAG_NO_APPEND);
if (sk->sk_state != TCP_SYN_SENT)
chtls_push_frames(csk, 1);
}
/*
* Perform a state transition during close and return the actions indicated
* for the transition. Do not make this function inline, the main reason
* it exists at all is to avoid multiple inlining of tcp_set_state.
*/
static int make_close_transition(struct sock *sk)
{
int next = (int)new_state[sk->sk_state];
tcp_set_state(sk, next & TCP_STATE_MASK);
return next & TCP_ACTION_FIN;
}
void chtls_close(struct sock *sk, long timeout)
{
int data_lost, prev_state;
struct chtls_sock *csk;
csk = rcu_dereference_sk_user_data(sk);
lock_sock(sk);
sk->sk_shutdown |= SHUTDOWN_MASK;
data_lost = skb_queue_len(&sk->sk_receive_queue);
data_lost |= skb_queue_len(&csk->tlshws.sk_recv_queue);
chtls_purge_recv_queue(sk);
chtls_purge_receive_queue(sk);
if (sk->sk_state == TCP_CLOSE) {
goto wait;
} else if (data_lost || sk->sk_state == TCP_SYN_SENT) {
chtls_send_reset(sk, CPL_ABORT_SEND_RST, NULL);
release_tcp_port(sk);
goto unlock;
} else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
sk->sk_prot->disconnect(sk, 0);
} else if (make_close_transition(sk)) {
chtls_close_conn(sk);
}
wait:
if (timeout)
sk_stream_wait_close(sk, timeout);
unlock:
prev_state = sk->sk_state;
sock_hold(sk);
sock_orphan(sk);
release_sock(sk);
local_bh_disable();
bh_lock_sock(sk);
if (prev_state != TCP_CLOSE && sk->sk_state == TCP_CLOSE)
goto out;
if (sk->sk_state == TCP_FIN_WAIT2 && tcp_sk(sk)->linger2 < 0 &&
!csk_flag(sk, CSK_ABORT_SHUTDOWN)) {
struct sk_buff *skb;
skb = alloc_skb(sizeof(struct cpl_abort_req), GFP_ATOMIC);
if (skb)
chtls_send_reset(sk, CPL_ABORT_SEND_RST, skb);
}
if (sk->sk_state == TCP_CLOSE)
inet_csk_destroy_sock(sk);
out:
bh_unlock_sock(sk);
local_bh_enable();
sock_put(sk);
}
/*
* Wait until a socket enters on of the given states.
*/
static int wait_for_states(struct sock *sk, unsigned int states)
{
DECLARE_WAITQUEUE(wait, current);
struct socket_wq _sk_wq;
long current_timeo;
int err = 0;
current_timeo = 200;
/*
* We want this to work even when there's no associated struct socket.
* In that case we provide a temporary wait_queue_head_t.
*/
if (!sk->sk_wq) {
init_waitqueue_head(&_sk_wq.wait);
_sk_wq.fasync_list = NULL;
init_rcu_head_on_stack(&_sk_wq.rcu);
RCU_INIT_POINTER(sk->sk_wq, &_sk_wq);
}
add_wait_queue(sk_sleep(sk), &wait);
while (!sk_in_state(sk, states)) {
if (!current_timeo) {
err = -EBUSY;
break;
}
if (signal_pending(current)) {
err = sock_intr_errno(current_timeo);
break;
}
set_current_state(TASK_UNINTERRUPTIBLE);
release_sock(sk);
if (!sk_in_state(sk, states))
current_timeo = schedule_timeout(current_timeo);
__set_current_state(TASK_RUNNING);
lock_sock(sk);
}
remove_wait_queue(sk_sleep(sk), &wait);
if (rcu_dereference(sk->sk_wq) == &_sk_wq)
sk->sk_wq = NULL;
return err;
}
int chtls_disconnect(struct sock *sk, int flags)
{
struct tcp_sock *tp;
int err;
tp = tcp_sk(sk);
chtls_purge_recv_queue(sk);
chtls_purge_receive_queue(sk);
chtls_purge_write_queue(sk);
if (sk->sk_state != TCP_CLOSE) {
sk->sk_err = ECONNRESET;
chtls_send_reset(sk, CPL_ABORT_SEND_RST, NULL);
err = wait_for_states(sk, TCPF_CLOSE);
if (err)
return err;
}
chtls_purge_recv_queue(sk);
chtls_purge_receive_queue(sk);
tp->max_window = 0xFFFF << (tp->rx_opt.snd_wscale);
return tcp_disconnect(sk, flags);
}
#define SHUTDOWN_ELIGIBLE_STATE (TCPF_ESTABLISHED | \
TCPF_SYN_RECV | TCPF_CLOSE_WAIT)
void chtls_shutdown(struct sock *sk, int how)
{
if ((how & SEND_SHUTDOWN) &&
sk_in_state(sk, SHUTDOWN_ELIGIBLE_STATE) &&
make_close_transition(sk))
chtls_close_conn(sk);
}
void chtls_destroy_sock(struct sock *sk)
{
struct chtls_sock *csk;
csk = rcu_dereference_sk_user_data(sk);
chtls_purge_recv_queue(sk);
csk->ulp_mode = ULP_MODE_NONE;
chtls_purge_write_queue(sk);
free_tls_keyid(sk);
kref_put(&csk->kref, chtls_sock_release);
sk->sk_prot = &tcp_prot;
sk->sk_prot->destroy(sk);
}
static void reset_listen_child(struct sock *child)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(child);
struct sk_buff *skb;
skb = alloc_ctrl_skb(csk->txdata_skb_cache,
sizeof(struct cpl_abort_req));
chtls_send_reset(child, CPL_ABORT_SEND_RST, skb);
sock_orphan(child);
INC_ORPHAN_COUNT(child);
if (child->sk_state == TCP_CLOSE)
inet_csk_destroy_sock(child);
}
static void chtls_disconnect_acceptq(struct sock *listen_sk)
{
struct request_sock **pprev;
pprev = ACCEPT_QUEUE(listen_sk);
while (*pprev) {
struct request_sock *req = *pprev;
if (req->rsk_ops == &chtls_rsk_ops) {
struct sock *child = req->sk;
*pprev = req->dl_next;
sk_acceptq_removed(listen_sk);
reqsk_put(req);
sock_hold(child);
local_bh_disable();
bh_lock_sock(child);
release_tcp_port(child);
reset_listen_child(child);
bh_unlock_sock(child);
local_bh_enable();
sock_put(child);
} else {
pprev = &req->dl_next;
}
}
}
static int listen_hashfn(const struct sock *sk)
{
return ((unsigned long)sk >> 10) & (LISTEN_INFO_HASH_SIZE - 1);
}
static struct listen_info *listen_hash_add(struct chtls_dev *cdev,
struct sock *sk,
unsigned int stid)
{
struct listen_info *p = kmalloc(sizeof(*p), GFP_KERNEL);
if (p) {
int key = listen_hashfn(sk);
p->sk = sk;
p->stid = stid;
spin_lock(&cdev->listen_lock);
p->next = cdev->listen_hash_tab[key];
cdev->listen_hash_tab[key] = p;
spin_unlock(&cdev->listen_lock);
}
return p;
}
static int listen_hash_find(struct chtls_dev *cdev,
struct sock *sk)
{
struct listen_info *p;
int stid = -1;
int key;
key = listen_hashfn(sk);
spin_lock(&cdev->listen_lock);
for (p = cdev->listen_hash_tab[key]; p; p = p->next)
if (p->sk == sk) {
stid = p->stid;
break;
}
spin_unlock(&cdev->listen_lock);
return stid;
}
static int listen_hash_del(struct chtls_dev *cdev,
struct sock *sk)
{
struct listen_info *p, **prev;
int stid = -1;
int key;
key = listen_hashfn(sk);
prev = &cdev->listen_hash_tab[key];
spin_lock(&cdev->listen_lock);
for (p = *prev; p; prev = &p->next, p = p->next)
if (p->sk == sk) {
stid = p->stid;
*prev = p->next;
kfree(p);
break;
}
spin_unlock(&cdev->listen_lock);
return stid;
}
static void cleanup_syn_rcv_conn(struct sock *child, struct sock *parent)
{
struct request_sock *req;
struct chtls_sock *csk;
csk = rcu_dereference_sk_user_data(child);
req = csk->passive_reap_next;
reqsk_queue_removed(&inet_csk(parent)->icsk_accept_queue, req);
__skb_unlink((struct sk_buff *)&csk->synq, &csk->listen_ctx->synq);
chtls_reqsk_free(req);
csk->passive_reap_next = NULL;
}
static void chtls_reset_synq(struct listen_ctx *listen_ctx)
{
struct sock *listen_sk = listen_ctx->lsk;
while (!skb_queue_empty(&listen_ctx->synq)) {
struct chtls_sock *csk =
container_of((struct synq *)__skb_dequeue
(&listen_ctx->synq), struct chtls_sock, synq);
struct sock *child = csk->sk;
cleanup_syn_rcv_conn(child, listen_sk);
sock_hold(child);
local_bh_disable();
bh_lock_sock(child);
release_tcp_port(child);
reset_listen_child(child);
bh_unlock_sock(child);
local_bh_enable();
sock_put(child);
}
}
int chtls_listen_start(struct chtls_dev *cdev, struct sock *sk)
{
struct net_device *ndev;
struct listen_ctx *ctx;
struct adapter *adap;
struct port_info *pi;
int stid;
int ret;
if (sk->sk_family != PF_INET)
return -EAGAIN;
rcu_read_lock();
ndev = chtls_ipv4_netdev(cdev, sk);
rcu_read_unlock();
if (!ndev)
return -EBADF;
pi = netdev_priv(ndev);
adap = pi->adapter;
if (!(adap->flags & CXGB4_FULL_INIT_DONE))
return -EBADF;
if (listen_hash_find(cdev, sk) >= 0) /* already have it */
return -EADDRINUSE;
ctx = kmalloc(sizeof(*ctx), GFP_KERNEL);
if (!ctx)
return -ENOMEM;
__module_get(THIS_MODULE);
ctx->lsk = sk;
ctx->cdev = cdev;
ctx->state = T4_LISTEN_START_PENDING;
skb_queue_head_init(&ctx->synq);
stid = cxgb4_alloc_stid(cdev->tids, sk->sk_family, ctx);
if (stid < 0)
goto free_ctx;
sock_hold(sk);
if (!listen_hash_add(cdev, sk, stid))
goto free_stid;
ret = cxgb4_create_server(ndev, stid,
inet_sk(sk)->inet_rcv_saddr,
inet_sk(sk)->inet_sport, 0,
cdev->lldi->rxq_ids[0]);
if (ret > 0)
ret = net_xmit_errno(ret);
if (ret)
goto del_hash;
return 0;
del_hash:
listen_hash_del(cdev, sk);
free_stid:
cxgb4_free_stid(cdev->tids, stid, sk->sk_family);
sock_put(sk);
free_ctx:
kfree(ctx);
module_put(THIS_MODULE);
return -EBADF;
}
void chtls_listen_stop(struct chtls_dev *cdev, struct sock *sk)
{
struct listen_ctx *listen_ctx;
int stid;
stid = listen_hash_del(cdev, sk);
if (stid < 0)
return;
listen_ctx = (struct listen_ctx *)lookup_stid(cdev->tids, stid);
chtls_reset_synq(listen_ctx);
cxgb4_remove_server(cdev->lldi->ports[0], stid,
cdev->lldi->rxq_ids[0], 0);
chtls_disconnect_acceptq(sk);
}
static int chtls_pass_open_rpl(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_pass_open_rpl *rpl = cplhdr(skb) + RSS_HDR;
unsigned int stid = GET_TID(rpl);
struct listen_ctx *listen_ctx;
listen_ctx = (struct listen_ctx *)lookup_stid(cdev->tids, stid);
if (!listen_ctx)
return CPL_RET_BUF_DONE;
if (listen_ctx->state == T4_LISTEN_START_PENDING) {
listen_ctx->state = T4_LISTEN_STARTED;
return CPL_RET_BUF_DONE;
}
if (rpl->status != CPL_ERR_NONE) {
pr_info("Unexpected PASS_OPEN_RPL status %u for STID %u\n",
rpl->status, stid);
return CPL_RET_BUF_DONE;
}
cxgb4_free_stid(cdev->tids, stid, listen_ctx->lsk->sk_family);
sock_put(listen_ctx->lsk);
kfree(listen_ctx);
module_put(THIS_MODULE);
return 0;
}
static int chtls_close_listsrv_rpl(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_close_listsvr_rpl *rpl = cplhdr(skb) + RSS_HDR;
struct listen_ctx *listen_ctx;
unsigned int stid;
void *data;
stid = GET_TID(rpl);
data = lookup_stid(cdev->tids, stid);
listen_ctx = (struct listen_ctx *)data;
if (rpl->status != CPL_ERR_NONE) {
pr_info("Unexpected CLOSE_LISTSRV_RPL status %u for STID %u\n",
rpl->status, stid);
return CPL_RET_BUF_DONE;
}
cxgb4_free_stid(cdev->tids, stid, listen_ctx->lsk->sk_family);
sock_put(listen_ctx->lsk);
kfree(listen_ctx);
module_put(THIS_MODULE);
return 0;
}
static void chtls_release_resources(struct sock *sk)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
struct chtls_dev *cdev = csk->cdev;
unsigned int tid = csk->tid;
struct tid_info *tids;
if (!cdev)
return;
tids = cdev->tids;
kfree_skb(csk->txdata_skb_cache);
csk->txdata_skb_cache = NULL;
if (csk->l2t_entry) {
cxgb4_l2t_release(csk->l2t_entry);
csk->l2t_entry = NULL;
}
cxgb4_remove_tid(tids, csk->port_id, tid, sk->sk_family);
sock_put(sk);
}
static void chtls_conn_done(struct sock *sk)
{
if (sock_flag(sk, SOCK_DEAD))
chtls_purge_receive_queue(sk);
sk_wakeup_sleepers(sk, 0);
tcp_done(sk);
}
static void do_abort_syn_rcv(struct sock *child, struct sock *parent)
{
/*
* If the server is still open we clean up the child connection,
* otherwise the server already did the clean up as it was purging
* its SYN queue and the skb was just sitting in its backlog.
*/
if (likely(parent->sk_state == TCP_LISTEN)) {
cleanup_syn_rcv_conn(child, parent);
/* Without the below call to sock_orphan,
* we leak the socket resource with syn_flood test
* as inet_csk_destroy_sock will not be called
* in tcp_done since SOCK_DEAD flag is not set.
* Kernel handles this differently where new socket is
* created only after 3 way handshake is done.
*/
sock_orphan(child);
percpu_counter_inc((child)->sk_prot->orphan_count);
chtls_release_resources(child);
chtls_conn_done(child);
} else {
if (csk_flag(child, CSK_RST_ABORTED)) {
chtls_release_resources(child);
chtls_conn_done(child);
}
}
}
static void pass_open_abort(struct sock *child, struct sock *parent,
struct sk_buff *skb)
{
do_abort_syn_rcv(child, parent);
kfree_skb(skb);
}
static void bl_pass_open_abort(struct sock *lsk, struct sk_buff *skb)
{
pass_open_abort(skb->sk, lsk, skb);
}
static void chtls_pass_open_arp_failure(struct sock *sk,
struct sk_buff *skb)
{
const struct request_sock *oreq;
struct chtls_sock *csk;
struct chtls_dev *cdev;
struct sock *parent;
void *data;
csk = rcu_dereference_sk_user_data(sk);
cdev = csk->cdev;
/*
* If the connection is being aborted due to the parent listening
* socket going away there's nothing to do, the ABORT_REQ will close
* the connection.
*/
if (csk_flag(sk, CSK_ABORT_RPL_PENDING)) {
kfree_skb(skb);
return;
}
oreq = csk->passive_reap_next;
data = lookup_stid(cdev->tids, oreq->ts_recent);
parent = ((struct listen_ctx *)data)->lsk;
bh_lock_sock(parent);
if (!sock_owned_by_user(parent)) {
pass_open_abort(sk, parent, skb);
} else {
BLOG_SKB_CB(skb)->backlog_rcv = bl_pass_open_abort;
__sk_add_backlog(parent, skb);
}
bh_unlock_sock(parent);
}
static void chtls_accept_rpl_arp_failure(void *handle,
struct sk_buff *skb)
{
struct sock *sk = (struct sock *)handle;
sock_hold(sk);
process_cpl_msg(chtls_pass_open_arp_failure, sk, skb);
sock_put(sk);
}
static unsigned int chtls_select_mss(const struct chtls_sock *csk,
unsigned int pmtu,
struct cpl_pass_accept_req *req)
{
struct chtls_dev *cdev;
struct dst_entry *dst;
unsigned int tcpoptsz;
unsigned int iphdrsz;
unsigned int mtu_idx;
struct tcp_sock *tp;
unsigned int mss;
struct sock *sk;
mss = ntohs(req->tcpopt.mss);
sk = csk->sk;
dst = __sk_dst_get(sk);
cdev = csk->cdev;
tp = tcp_sk(sk);
tcpoptsz = 0;
iphdrsz = sizeof(struct iphdr) + sizeof(struct tcphdr);
if (req->tcpopt.tstamp)
tcpoptsz += round_up(TCPOLEN_TIMESTAMP, 4);
tp->advmss = dst_metric_advmss(dst);
if (USER_MSS(tp) && tp->advmss > USER_MSS(tp))
tp->advmss = USER_MSS(tp);
if (tp->advmss > pmtu - iphdrsz)
tp->advmss = pmtu - iphdrsz;
if (mss && tp->advmss > mss)
tp->advmss = mss;
tp->advmss = cxgb4_best_aligned_mtu(cdev->lldi->mtus,
iphdrsz + tcpoptsz,
tp->advmss - tcpoptsz,
8, &mtu_idx);
tp->advmss -= iphdrsz;
inet_csk(sk)->icsk_pmtu_cookie = pmtu;
return mtu_idx;
}
static unsigned int select_rcv_wscale(int space, int wscale_ok, int win_clamp)
{
int wscale = 0;
if (space > MAX_RCV_WND)
space = MAX_RCV_WND;
if (win_clamp && win_clamp < space)
space = win_clamp;
if (wscale_ok) {
while (wscale < 14 && (65535 << wscale) < space)
wscale++;
}
return wscale;
}
static void chtls_pass_accept_rpl(struct sk_buff *skb,
struct cpl_pass_accept_req *req,
unsigned int tid)
{
struct cpl_t5_pass_accept_rpl *rpl5;
struct cxgb4_lld_info *lldi;
const struct tcphdr *tcph;
const struct tcp_sock *tp;
struct chtls_sock *csk;
unsigned int len;
struct sock *sk;
u32 opt2, hlen;
u64 opt0;
sk = skb->sk;
tp = tcp_sk(sk);
csk = sk->sk_user_data;
csk->tid = tid;
lldi = csk->cdev->lldi;
len = roundup(sizeof(*rpl5), 16);
rpl5 = __skb_put_zero(skb, len);
INIT_TP_WR(rpl5, tid);
OPCODE_TID(rpl5) = cpu_to_be32(MK_OPCODE_TID(CPL_PASS_ACCEPT_RPL,
csk->tid));
csk->mtu_idx = chtls_select_mss(csk, dst_mtu(__sk_dst_get(sk)),
req);
opt0 = TCAM_BYPASS_F |
WND_SCALE_V(RCV_WSCALE(tp)) |
MSS_IDX_V(csk->mtu_idx) |
L2T_IDX_V(csk->l2t_entry->idx) |
NAGLE_V(!(tp->nonagle & TCP_NAGLE_OFF)) |
TX_CHAN_V(csk->tx_chan) |
SMAC_SEL_V(csk->smac_idx) |
DSCP_V(csk->tos >> 2) |
ULP_MODE_V(ULP_MODE_TLS) |
RCV_BUFSIZ_V(min(tp->rcv_wnd >> 10, RCV_BUFSIZ_M));
opt2 = RX_CHANNEL_V(0) |
RSS_QUEUE_VALID_F | RSS_QUEUE_V(csk->rss_qid);
if (!is_t5(lldi->adapter_type))
opt2 |= RX_FC_DISABLE_F;
if (req->tcpopt.tstamp)
opt2 |= TSTAMPS_EN_F;
if (req->tcpopt.sack)
opt2 |= SACK_EN_F;
hlen = ntohl(req->hdr_len);
tcph = (struct tcphdr *)((u8 *)(req + 1) +
T6_ETH_HDR_LEN_G(hlen) + T6_IP_HDR_LEN_G(hlen));
if (tcph->ece && tcph->cwr)
opt2 |= CCTRL_ECN_V(1);
opt2 |= CONG_CNTRL_V(CONG_ALG_NEWRENO);
opt2 |= T5_ISS_F;
opt2 |= T5_OPT_2_VALID_F;
rpl5->opt0 = cpu_to_be64(opt0);
rpl5->opt2 = cpu_to_be32(opt2);
rpl5->iss = cpu_to_be32((prandom_u32() & ~7UL) - 1);
set_wr_txq(skb, CPL_PRIORITY_SETUP, csk->port_id);
t4_set_arp_err_handler(skb, sk, chtls_accept_rpl_arp_failure);
cxgb4_l2t_send(csk->egress_dev, skb, csk->l2t_entry);
}
static void inet_inherit_port(struct inet_hashinfo *hash_info,
struct sock *lsk, struct sock *newsk)
{
local_bh_disable();
__inet_inherit_port(lsk, newsk);
local_bh_enable();
}
static int chtls_backlog_rcv(struct sock *sk, struct sk_buff *skb)
{
if (skb->protocol) {
kfree_skb(skb);
return 0;
}
BLOG_SKB_CB(skb)->backlog_rcv(sk, skb);
return 0;
}
static void chtls_set_tcp_window(struct chtls_sock *csk)
{
struct net_device *ndev = csk->egress_dev;
struct port_info *pi = netdev_priv(ndev);
unsigned int linkspeed;
u8 scale;
linkspeed = pi->link_cfg.speed;
scale = linkspeed / SPEED_10000;
#define CHTLS_10G_RCVWIN (256 * 1024)
csk->rcv_win = CHTLS_10G_RCVWIN;
if (scale)
csk->rcv_win *= scale;
#define CHTLS_10G_SNDWIN (256 * 1024)
csk->snd_win = CHTLS_10G_SNDWIN;
if (scale)
csk->snd_win *= scale;
}
static struct sock *chtls_recv_sock(struct sock *lsk,
struct request_sock *oreq,
void *network_hdr,
const struct cpl_pass_accept_req *req,
struct chtls_dev *cdev)
{
struct inet_sock *newinet;
const struct iphdr *iph;
struct tls_context *ctx;
struct net_device *ndev;
struct chtls_sock *csk;
struct dst_entry *dst;
struct neighbour *n;
struct tcp_sock *tp;
struct sock *newsk;
u16 port_id;
int rxq_idx;
int step;
iph = (const struct iphdr *)network_hdr;
newsk = tcp_create_openreq_child(lsk, oreq, cdev->askb);
if (!newsk)
goto free_oreq;
dst = inet_csk_route_child_sock(lsk, newsk, oreq);
if (!dst)
goto free_sk;
n = dst_neigh_lookup(dst, &iph->saddr);
if (!n)
goto free_sk;
ndev = n->dev;
if (!ndev)
goto free_dst;
port_id = cxgb4_port_idx(ndev);
csk = chtls_sock_create(cdev);
if (!csk)
goto free_dst;
csk->l2t_entry = cxgb4_l2t_get(cdev->lldi->l2t, n, ndev, 0);
if (!csk->l2t_entry)
goto free_csk;
newsk->sk_user_data = csk;
newsk->sk_backlog_rcv = chtls_backlog_rcv;
tp = tcp_sk(newsk);
newinet = inet_sk(newsk);
newinet->inet_daddr = iph->saddr;
newinet->inet_rcv_saddr = iph->daddr;
newinet->inet_saddr = iph->daddr;
oreq->ts_recent = PASS_OPEN_TID_G(ntohl(req->tos_stid));
sk_setup_caps(newsk, dst);
ctx = tls_get_ctx(lsk);
newsk->sk_destruct = ctx->sk_destruct;
csk->sk = newsk;
csk->passive_reap_next = oreq;
csk->tx_chan = cxgb4_port_chan(ndev);
csk->port_id = port_id;
csk->egress_dev = ndev;
csk->tos = PASS_OPEN_TOS_G(ntohl(req->tos_stid));
chtls_set_tcp_window(csk);
tp->rcv_wnd = csk->rcv_win;
csk->sndbuf = csk->snd_win;
csk->ulp_mode = ULP_MODE_TLS;
step = cdev->lldi->nrxq / cdev->lldi->nchan;
csk->rss_qid = cdev->lldi->rxq_ids[port_id * step];
rxq_idx = port_id * step;
csk->txq_idx = (rxq_idx < cdev->lldi->ntxq) ? rxq_idx :
port_id * step;
csk->sndbuf = newsk->sk_sndbuf;
csk->smac_idx = ((struct port_info *)netdev_priv(ndev))->smt_idx;
RCV_WSCALE(tp) = select_rcv_wscale(tcp_full_space(newsk),
sock_net(newsk)->
ipv4.sysctl_tcp_window_scaling,
tp->window_clamp);
neigh_release(n);
inet_inherit_port(&tcp_hashinfo, lsk, newsk);
csk_set_flag(csk, CSK_CONN_INLINE);
bh_unlock_sock(newsk); /* tcp_create_openreq_child ->sk_clone_lock */
return newsk;
free_csk:
chtls_sock_release(&csk->kref);
free_dst:
dst_release(dst);
free_sk:
inet_csk_prepare_forced_close(newsk);
tcp_done(newsk);
free_oreq:
chtls_reqsk_free(oreq);
return NULL;
}
/*
* Populate a TID_RELEASE WR. The skb must be already propely sized.
*/
static void mk_tid_release(struct sk_buff *skb,
unsigned int chan, unsigned int tid)
{
struct cpl_tid_release *req;
unsigned int len;
len = roundup(sizeof(struct cpl_tid_release), 16);
req = (struct cpl_tid_release *)__skb_put(skb, len);
memset(req, 0, len);
set_wr_txq(skb, CPL_PRIORITY_SETUP, chan);
INIT_TP_WR_CPL(req, CPL_TID_RELEASE, tid);
}
static int chtls_get_module(struct sock *sk)
{
struct inet_connection_sock *icsk = inet_csk(sk);
if (!try_module_get(icsk->icsk_ulp_ops->owner))
return -1;
return 0;
}
static void chtls_pass_accept_request(struct sock *sk,
struct sk_buff *skb)
{
struct cpl_t5_pass_accept_rpl *rpl;
struct cpl_pass_accept_req *req;
struct listen_ctx *listen_ctx;
struct vlan_ethhdr *vlan_eh;
struct request_sock *oreq;
struct sk_buff *reply_skb;
struct chtls_sock *csk;
struct chtls_dev *cdev;
struct tcphdr *tcph;
struct sock *newsk;
struct ethhdr *eh;
struct iphdr *iph;
void *network_hdr;
unsigned int stid;
unsigned int len;
unsigned int tid;
bool th_ecn, ect;
__u8 ip_dsfield; /* IPv4 tos or IPv6 dsfield */
u16 eth_hdr_len;
bool ecn_ok;
req = cplhdr(skb) + RSS_HDR;
tid = GET_TID(req);
cdev = BLOG_SKB_CB(skb)->cdev;
newsk = lookup_tid(cdev->tids, tid);
stid = PASS_OPEN_TID_G(ntohl(req->tos_stid));
if (newsk) {
pr_info("tid (%d) already in use\n", tid);
return;
}
len = roundup(sizeof(*rpl), 16);
reply_skb = alloc_skb(len, GFP_ATOMIC);
if (!reply_skb) {
cxgb4_remove_tid(cdev->tids, 0, tid, sk->sk_family);
kfree_skb(skb);
return;
}
if (sk->sk_state != TCP_LISTEN)
goto reject;
if (inet_csk_reqsk_queue_is_full(sk))
goto reject;
if (sk_acceptq_is_full(sk))
goto reject;
oreq = inet_reqsk_alloc(&chtls_rsk_ops, sk, true);
if (!oreq)
goto reject;
oreq->rsk_rcv_wnd = 0;
oreq->rsk_window_clamp = 0;
oreq->cookie_ts = 0;
oreq->mss = 0;
oreq->ts_recent = 0;
eth_hdr_len = T6_ETH_HDR_LEN_G(ntohl(req->hdr_len));
if (eth_hdr_len == ETH_HLEN) {
eh = (struct ethhdr *)(req + 1);
iph = (struct iphdr *)(eh + 1);
network_hdr = (void *)(eh + 1);
} else {
vlan_eh = (struct vlan_ethhdr *)(req + 1);
iph = (struct iphdr *)(vlan_eh + 1);
network_hdr = (void *)(vlan_eh + 1);
}
if (iph->version != 0x4)
goto free_oreq;
tcph = (struct tcphdr *)(iph + 1);
skb_set_network_header(skb, (void *)iph - (void *)req);
tcp_rsk(oreq)->tfo_listener = false;
tcp_rsk(oreq)->rcv_isn = ntohl(tcph->seq);
chtls_set_req_port(oreq, tcph->source, tcph->dest);
chtls_set_req_addr(oreq, iph->daddr, iph->saddr);
ip_dsfield = ipv4_get_dsfield(iph);
if (req->tcpopt.wsf <= 14 &&
sock_net(sk)->ipv4.sysctl_tcp_window_scaling) {
inet_rsk(oreq)->wscale_ok = 1;
inet_rsk(oreq)->snd_wscale = req->tcpopt.wsf;
}
inet_rsk(oreq)->ir_iif = sk->sk_bound_dev_if;
th_ecn = tcph->ece && tcph->cwr;
if (th_ecn) {
ect = !INET_ECN_is_not_ect(ip_dsfield);
ecn_ok = sock_net(sk)->ipv4.sysctl_tcp_ecn;
if ((!ect && ecn_ok) || tcp_ca_needs_ecn(sk))
inet_rsk(oreq)->ecn_ok = 1;
}
newsk = chtls_recv_sock(sk, oreq, network_hdr, req, cdev);
if (!newsk)
goto reject;
if (chtls_get_module(newsk))
goto reject;
inet_csk_reqsk_queue_added(sk);
reply_skb->sk = newsk;
chtls_install_cpl_ops(newsk);
cxgb4_insert_tid(cdev->tids, newsk, tid, newsk->sk_family);
csk = rcu_dereference_sk_user_data(newsk);
listen_ctx = (struct listen_ctx *)lookup_stid(cdev->tids, stid);
csk->listen_ctx = listen_ctx;
__skb_queue_tail(&listen_ctx->synq, (struct sk_buff *)&csk->synq);
chtls_pass_accept_rpl(reply_skb, req, tid);
kfree_skb(skb);
return;
free_oreq:
chtls_reqsk_free(oreq);
reject:
mk_tid_release(reply_skb, 0, tid);
cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
kfree_skb(skb);
}
/*
* Handle a CPL_PASS_ACCEPT_REQ message.
*/
static int chtls_pass_accept_req(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_pass_accept_req *req = cplhdr(skb) + RSS_HDR;
struct listen_ctx *ctx;
unsigned int stid;
unsigned int tid;
struct sock *lsk;
void *data;
stid = PASS_OPEN_TID_G(ntohl(req->tos_stid));
tid = GET_TID(req);
data = lookup_stid(cdev->tids, stid);
if (!data)
return 1;
ctx = (struct listen_ctx *)data;
lsk = ctx->lsk;
if (unlikely(tid >= cdev->tids->ntids)) {
pr_info("passive open TID %u too large\n", tid);
return 1;
}
BLOG_SKB_CB(skb)->cdev = cdev;
process_cpl_msg(chtls_pass_accept_request, lsk, skb);
return 0;
}
/*
* Completes some final bits of initialization for just established connections
* and changes their state to TCP_ESTABLISHED.
*
* snd_isn here is the ISN after the SYN, i.e., the true ISN + 1.
*/
static void make_established(struct sock *sk, u32 snd_isn, unsigned int opt)
{
struct tcp_sock *tp = tcp_sk(sk);
tp->pushed_seq = snd_isn;
tp->write_seq = snd_isn;
tp->snd_nxt = snd_isn;
tp->snd_una = snd_isn;
inet_sk(sk)->inet_id = prandom_u32();
assign_rxopt(sk, opt);
if (tp->rcv_wnd > (RCV_BUFSIZ_M << 10))
tp->rcv_wup -= tp->rcv_wnd - (RCV_BUFSIZ_M << 10);
smp_mb();
tcp_set_state(sk, TCP_ESTABLISHED);
}
static void chtls_abort_conn(struct sock *sk, struct sk_buff *skb)
{
struct sk_buff *abort_skb;
abort_skb = alloc_skb(sizeof(struct cpl_abort_req), GFP_ATOMIC);
if (abort_skb)
chtls_send_reset(sk, CPL_ABORT_SEND_RST, abort_skb);
}
static struct sock *reap_list;
static DEFINE_SPINLOCK(reap_list_lock);
/*
* Process the reap list.
*/
DECLARE_TASK_FUNC(process_reap_list, task_param)
{
spin_lock_bh(&reap_list_lock);
while (reap_list) {
struct sock *sk = reap_list;
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
reap_list = csk->passive_reap_next;
csk->passive_reap_next = NULL;
spin_unlock(&reap_list_lock);
sock_hold(sk);
bh_lock_sock(sk);
chtls_abort_conn(sk, NULL);
sock_orphan(sk);
if (sk->sk_state == TCP_CLOSE)
inet_csk_destroy_sock(sk);
bh_unlock_sock(sk);
sock_put(sk);
spin_lock(&reap_list_lock);
}
spin_unlock_bh(&reap_list_lock);
}
static DECLARE_WORK(reap_task, process_reap_list);
static void add_to_reap_list(struct sock *sk)
{
struct chtls_sock *csk = sk->sk_user_data;
local_bh_disable();
bh_lock_sock(sk);
release_tcp_port(sk); /* release the port immediately */
spin_lock(&reap_list_lock);
csk->passive_reap_next = reap_list;
reap_list = sk;
if (!csk->passive_reap_next)
schedule_work(&reap_task);
spin_unlock(&reap_list_lock);
bh_unlock_sock(sk);
local_bh_enable();
}
static void add_pass_open_to_parent(struct sock *child, struct sock *lsk,
struct chtls_dev *cdev)
{
struct request_sock *oreq;
struct chtls_sock *csk;
if (lsk->sk_state != TCP_LISTEN)
return;
csk = child->sk_user_data;
oreq = csk->passive_reap_next;
csk->passive_reap_next = NULL;
reqsk_queue_removed(&inet_csk(lsk)->icsk_accept_queue, oreq);
__skb_unlink((struct sk_buff *)&csk->synq, &csk->listen_ctx->synq);
if (sk_acceptq_is_full(lsk)) {
chtls_reqsk_free(oreq);
add_to_reap_list(child);
} else {
refcount_set(&oreq->rsk_refcnt, 1);
inet_csk_reqsk_queue_add(lsk, oreq, child);
lsk->sk_data_ready(lsk);
}
}
static void bl_add_pass_open_to_parent(struct sock *lsk, struct sk_buff *skb)
{
struct sock *child = skb->sk;
skb->sk = NULL;
add_pass_open_to_parent(child, lsk, BLOG_SKB_CB(skb)->cdev);
kfree_skb(skb);
}
static int chtls_pass_establish(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_pass_establish *req = cplhdr(skb) + RSS_HDR;
struct chtls_sock *csk;
struct sock *lsk, *sk;
unsigned int hwtid;
hwtid = GET_TID(req);
sk = lookup_tid(cdev->tids, hwtid);
if (!sk)
return (CPL_RET_UNKNOWN_TID | CPL_RET_BUF_DONE);
bh_lock_sock(sk);
if (unlikely(sock_owned_by_user(sk))) {
kfree_skb(skb);
} else {
unsigned int stid;
void *data;
csk = sk->sk_user_data;
csk->wr_max_credits = 64;
csk->wr_credits = 64;
csk->wr_unacked = 0;
make_established(sk, ntohl(req->snd_isn), ntohs(req->tcp_opt));
stid = PASS_OPEN_TID_G(ntohl(req->tos_stid));
sk->sk_state_change(sk);
if (unlikely(sk->sk_socket))
sk_wake_async(sk, 0, POLL_OUT);
data = lookup_stid(cdev->tids, stid);
lsk = ((struct listen_ctx *)data)->lsk;
bh_lock_sock(lsk);
if (unlikely(skb_queue_empty(&csk->listen_ctx->synq))) {
/* removed from synq */
bh_unlock_sock(lsk);
kfree_skb(skb);
goto unlock;
}
if (likely(!sock_owned_by_user(lsk))) {
kfree_skb(skb);
add_pass_open_to_parent(sk, lsk, cdev);
} else {
skb->sk = sk;
BLOG_SKB_CB(skb)->cdev = cdev;
BLOG_SKB_CB(skb)->backlog_rcv =
bl_add_pass_open_to_parent;
__sk_add_backlog(lsk, skb);
}
bh_unlock_sock(lsk);
}
unlock:
bh_unlock_sock(sk);
return 0;
}
/*
* Handle receipt of an urgent pointer.
*/
static void handle_urg_ptr(struct sock *sk, u32 urg_seq)
{
struct tcp_sock *tp = tcp_sk(sk);
urg_seq--;
if (tp->urg_data && !after(urg_seq, tp->urg_seq))
return; /* duplicate pointer */
sk_send_sigurg(sk);
if (tp->urg_seq == tp->copied_seq && tp->urg_data &&
!sock_flag(sk, SOCK_URGINLINE) &&
tp->copied_seq != tp->rcv_nxt) {
struct sk_buff *skb = skb_peek(&sk->sk_receive_queue);
tp->copied_seq++;
if (skb && tp->copied_seq - ULP_SKB_CB(skb)->seq >= skb->len)
chtls_free_skb(sk, skb);
}
tp->urg_data = TCP_URG_NOTYET;
tp->urg_seq = urg_seq;
}
static void check_sk_callbacks(struct chtls_sock *csk)
{
struct sock *sk = csk->sk;
if (unlikely(sk->sk_user_data &&
!csk_flag_nochk(csk, CSK_CALLBACKS_CHKD)))
csk_set_flag(csk, CSK_CALLBACKS_CHKD);
}
/*
* Handles Rx data that arrives in a state where the socket isn't accepting
* new data.
*/
static void handle_excess_rx(struct sock *sk, struct sk_buff *skb)
{
if (!csk_flag(sk, CSK_ABORT_SHUTDOWN))
chtls_abort_conn(sk, skb);
kfree_skb(skb);
}
static void chtls_recv_data(struct sock *sk, struct sk_buff *skb)
{
struct cpl_rx_data *hdr = cplhdr(skb) + RSS_HDR;
struct chtls_sock *csk;
struct tcp_sock *tp;
csk = rcu_dereference_sk_user_data(sk);
tp = tcp_sk(sk);
if (unlikely(sk->sk_shutdown & RCV_SHUTDOWN)) {
handle_excess_rx(sk, skb);
return;
}
ULP_SKB_CB(skb)->seq = ntohl(hdr->seq);
ULP_SKB_CB(skb)->psh = hdr->psh;
skb_ulp_mode(skb) = ULP_MODE_NONE;
skb_reset_transport_header(skb);
__skb_pull(skb, sizeof(*hdr) + RSS_HDR);
if (!skb->data_len)
__skb_trim(skb, ntohs(hdr->len));
if (unlikely(hdr->urg))
handle_urg_ptr(sk, tp->rcv_nxt + ntohs(hdr->urg));
if (unlikely(tp->urg_data == TCP_URG_NOTYET &&
tp->urg_seq - tp->rcv_nxt < skb->len))
tp->urg_data = TCP_URG_VALID |
skb->data[tp->urg_seq - tp->rcv_nxt];
if (unlikely(hdr->dack_mode != csk->delack_mode)) {
csk->delack_mode = hdr->dack_mode;
csk->delack_seq = tp->rcv_nxt;
}
tcp_hdr(skb)->fin = 0;
tp->rcv_nxt += skb->len;
__skb_queue_tail(&sk->sk_receive_queue, skb);
if (!sock_flag(sk, SOCK_DEAD)) {
check_sk_callbacks(csk);
sk->sk_data_ready(sk);
}
}
static int chtls_rx_data(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_rx_data *req = cplhdr(skb) + RSS_HDR;
unsigned int hwtid = GET_TID(req);
struct sock *sk;
sk = lookup_tid(cdev->tids, hwtid);
if (unlikely(!sk)) {
pr_err("can't find conn. for hwtid %u.\n", hwtid);
return -EINVAL;
}
skb_dst_set(skb, NULL);
process_cpl_msg(chtls_recv_data, sk, skb);
return 0;
}
static void chtls_recv_pdu(struct sock *sk, struct sk_buff *skb)
{
struct cpl_tls_data *hdr = cplhdr(skb);
struct chtls_sock *csk;
struct chtls_hws *tlsk;
struct tcp_sock *tp;
csk = rcu_dereference_sk_user_data(sk);
tlsk = &csk->tlshws;
tp = tcp_sk(sk);
if (unlikely(sk->sk_shutdown & RCV_SHUTDOWN)) {
handle_excess_rx(sk, skb);
return;
}
ULP_SKB_CB(skb)->seq = ntohl(hdr->seq);
ULP_SKB_CB(skb)->flags = 0;
skb_ulp_mode(skb) = ULP_MODE_TLS;
skb_reset_transport_header(skb);
__skb_pull(skb, sizeof(*hdr));
if (!skb->data_len)
__skb_trim(skb,
CPL_TLS_DATA_LENGTH_G(ntohl(hdr->length_pkd)));
if (unlikely(tp->urg_data == TCP_URG_NOTYET && tp->urg_seq -
tp->rcv_nxt < skb->len))
tp->urg_data = TCP_URG_VALID |
skb->data[tp->urg_seq - tp->rcv_nxt];
tcp_hdr(skb)->fin = 0;
tlsk->pldlen = CPL_TLS_DATA_LENGTH_G(ntohl(hdr->length_pkd));
__skb_queue_tail(&tlsk->sk_recv_queue, skb);
}
static int chtls_rx_pdu(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_tls_data *req = cplhdr(skb);
unsigned int hwtid = GET_TID(req);
struct sock *sk;
sk = lookup_tid(cdev->tids, hwtid);
if (unlikely(!sk)) {
pr_err("can't find conn. for hwtid %u.\n", hwtid);
return -EINVAL;
}
skb_dst_set(skb, NULL);
process_cpl_msg(chtls_recv_pdu, sk, skb);
return 0;
}
static void chtls_set_hdrlen(struct sk_buff *skb, unsigned int nlen)
{
struct tlsrx_cmp_hdr *tls_cmp_hdr = cplhdr(skb);
skb->hdr_len = ntohs((__force __be16)tls_cmp_hdr->length);
tls_cmp_hdr->length = ntohs((__force __be16)nlen);
}
static void chtls_rx_hdr(struct sock *sk, struct sk_buff *skb)
{
struct tlsrx_cmp_hdr *tls_hdr_pkt;
struct cpl_rx_tls_cmp *cmp_cpl;
struct sk_buff *skb_rec;
struct chtls_sock *csk;
struct chtls_hws *tlsk;
struct tcp_sock *tp;
cmp_cpl = cplhdr(skb);
csk = rcu_dereference_sk_user_data(sk);
tlsk = &csk->tlshws;
tp = tcp_sk(sk);
ULP_SKB_CB(skb)->seq = ntohl(cmp_cpl->seq);
ULP_SKB_CB(skb)->flags = 0;
skb_reset_transport_header(skb);
__skb_pull(skb, sizeof(*cmp_cpl));
tls_hdr_pkt = (struct tlsrx_cmp_hdr *)skb->data;
if (tls_hdr_pkt->res_to_mac_error & TLSRX_HDR_PKT_ERROR_M)
tls_hdr_pkt->type = CONTENT_TYPE_ERROR;
if (!skb->data_len)
__skb_trim(skb, TLS_HEADER_LENGTH);
tp->rcv_nxt +=
CPL_RX_TLS_CMP_PDULENGTH_G(ntohl(cmp_cpl->pdulength_length));
ULP_SKB_CB(skb)->flags |= ULPCB_FLAG_TLS_HDR;
skb_rec = __skb_dequeue(&tlsk->sk_recv_queue);
if (!skb_rec) {
__skb_queue_tail(&sk->sk_receive_queue, skb);
} else {
chtls_set_hdrlen(skb, tlsk->pldlen);
tlsk->pldlen = 0;
__skb_queue_tail(&sk->sk_receive_queue, skb);
__skb_queue_tail(&sk->sk_receive_queue, skb_rec);
}
if (!sock_flag(sk, SOCK_DEAD)) {
check_sk_callbacks(csk);
sk->sk_data_ready(sk);
}
}
static int chtls_rx_cmp(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_rx_tls_cmp *req = cplhdr(skb);
unsigned int hwtid = GET_TID(req);
struct sock *sk;
sk = lookup_tid(cdev->tids, hwtid);
if (unlikely(!sk)) {
pr_err("can't find conn. for hwtid %u.\n", hwtid);
return -EINVAL;
}
skb_dst_set(skb, NULL);
process_cpl_msg(chtls_rx_hdr, sk, skb);
return 0;
}
static void chtls_timewait(struct sock *sk)
{
struct tcp_sock *tp = tcp_sk(sk);
tp->rcv_nxt++;
tp->rx_opt.ts_recent_stamp = ktime_get_seconds();
tp->srtt_us = 0;
tcp_time_wait(sk, TCP_TIME_WAIT, 0);
}
static void chtls_peer_close(struct sock *sk, struct sk_buff *skb)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
sk->sk_shutdown |= RCV_SHUTDOWN;
sock_set_flag(sk, SOCK_DONE);
switch (sk->sk_state) {
case TCP_SYN_RECV:
case TCP_ESTABLISHED:
tcp_set_state(sk, TCP_CLOSE_WAIT);
break;
case TCP_FIN_WAIT1:
tcp_set_state(sk, TCP_CLOSING);
break;
case TCP_FIN_WAIT2:
chtls_release_resources(sk);
if (csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING))
chtls_conn_done(sk);
else
chtls_timewait(sk);
break;
default:
pr_info("cpl_peer_close in bad state %d\n", sk->sk_state);
}
if (!sock_flag(sk, SOCK_DEAD)) {
sk->sk_state_change(sk);
/* Do not send POLL_HUP for half duplex close. */
if ((sk->sk_shutdown & SEND_SHUTDOWN) ||
sk->sk_state == TCP_CLOSE)
sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_HUP);
else
sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
}
}
static void chtls_close_con_rpl(struct sock *sk, struct sk_buff *skb)
{
struct cpl_close_con_rpl *rpl = cplhdr(skb) + RSS_HDR;
struct chtls_sock *csk;
struct tcp_sock *tp;
csk = rcu_dereference_sk_user_data(sk);
tp = tcp_sk(sk);
tp->snd_una = ntohl(rpl->snd_nxt) - 1; /* exclude FIN */
switch (sk->sk_state) {
case TCP_CLOSING:
chtls_release_resources(sk);
if (csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING))
chtls_conn_done(sk);
else
chtls_timewait(sk);
break;
case TCP_LAST_ACK:
chtls_release_resources(sk);
chtls_conn_done(sk);
break;
case TCP_FIN_WAIT1:
tcp_set_state(sk, TCP_FIN_WAIT2);
sk->sk_shutdown |= SEND_SHUTDOWN;
if (!sock_flag(sk, SOCK_DEAD))
sk->sk_state_change(sk);
else if (tcp_sk(sk)->linger2 < 0 &&
!csk_flag_nochk(csk, CSK_ABORT_SHUTDOWN))
chtls_abort_conn(sk, skb);
break;
default:
pr_info("close_con_rpl in bad state %d\n", sk->sk_state);
}
kfree_skb(skb);
}
static struct sk_buff *get_cpl_skb(struct sk_buff *skb,
size_t len, gfp_t gfp)
{
if (likely(!skb_is_nonlinear(skb) && !skb_cloned(skb))) {
WARN_ONCE(skb->len < len, "skb alloc error");
__skb_trim(skb, len);
skb_get(skb);
} else {
skb = alloc_skb(len, gfp);
if (skb)
__skb_put(skb, len);
}
return skb;
}
static void set_abort_rpl_wr(struct sk_buff *skb, unsigned int tid,
int cmd)
{
struct cpl_abort_rpl *rpl = cplhdr(skb);
INIT_TP_WR_CPL(rpl, CPL_ABORT_RPL, tid);
rpl->cmd = cmd;
}
static void send_defer_abort_rpl(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_abort_req_rss *req = cplhdr(skb);
struct sk_buff *reply_skb;
reply_skb = alloc_skb(sizeof(struct cpl_abort_rpl),
GFP_KERNEL | __GFP_NOFAIL);
__skb_put(reply_skb, sizeof(struct cpl_abort_rpl));
set_abort_rpl_wr(reply_skb, GET_TID(req),
(req->status & CPL_ABORT_NO_RST));
set_wr_txq(reply_skb, CPL_PRIORITY_DATA, req->status >> 1);
cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
kfree_skb(skb);
}
static void send_abort_rpl(struct sock *sk, struct sk_buff *skb,
struct chtls_dev *cdev, int status, int queue)
{
struct cpl_abort_req_rss *req = cplhdr(skb);
struct sk_buff *reply_skb;
struct chtls_sock *csk;
csk = rcu_dereference_sk_user_data(sk);
reply_skb = alloc_skb(sizeof(struct cpl_abort_rpl),
GFP_KERNEL);
if (!reply_skb) {
req->status = (queue << 1);
send_defer_abort_rpl(cdev, skb);
return;
}
set_abort_rpl_wr(reply_skb, GET_TID(req), status);
kfree_skb(skb);
set_wr_txq(reply_skb, CPL_PRIORITY_DATA, queue);
if (csk_conn_inline(csk)) {
struct l2t_entry *e = csk->l2t_entry;
if (e && sk->sk_state != TCP_SYN_RECV) {
cxgb4_l2t_send(csk->egress_dev, reply_skb, e);
return;
}
}
cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
}
/*
* Add an skb to the deferred skb queue for processing from process context.
*/
static void t4_defer_reply(struct sk_buff *skb, struct chtls_dev *cdev,
defer_handler_t handler)
{
DEFERRED_SKB_CB(skb)->handler = handler;
spin_lock_bh(&cdev->deferq.lock);
__skb_queue_tail(&cdev->deferq, skb);
if (skb_queue_len(&cdev->deferq) == 1)
schedule_work(&cdev->deferq_task);
spin_unlock_bh(&cdev->deferq.lock);
}
static void chtls_send_abort_rpl(struct sock *sk, struct sk_buff *skb,
struct chtls_dev *cdev,
int status, int queue)
{
struct cpl_abort_req_rss *req = cplhdr(skb) + RSS_HDR;
struct sk_buff *reply_skb;
struct chtls_sock *csk;
unsigned int tid;
csk = rcu_dereference_sk_user_data(sk);
tid = GET_TID(req);
reply_skb = get_cpl_skb(skb, sizeof(struct cpl_abort_rpl), gfp_any());
if (!reply_skb) {
req->status = (queue << 1) | status;
t4_defer_reply(skb, cdev, send_defer_abort_rpl);
return;
}
set_abort_rpl_wr(reply_skb, tid, status);
set_wr_txq(reply_skb, CPL_PRIORITY_DATA, queue);
if (csk_conn_inline(csk)) {
struct l2t_entry *e = csk->l2t_entry;
if (e && sk->sk_state != TCP_SYN_RECV) {
cxgb4_l2t_send(csk->egress_dev, reply_skb, e);
return;
}
}
cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
kfree_skb(skb);
}
/*
* This is run from a listener's backlog to abort a child connection in
* SYN_RCV state (i.e., one on the listener's SYN queue).
*/
static void bl_abort_syn_rcv(struct sock *lsk, struct sk_buff *skb)
{
struct chtls_sock *csk;
struct sock *child;
int queue;
child = skb->sk;
csk = rcu_dereference_sk_user_data(child);
queue = csk->txq_idx;
skb->sk = NULL;
do_abort_syn_rcv(child, lsk);
send_abort_rpl(child, skb, BLOG_SKB_CB(skb)->cdev,
CPL_ABORT_NO_RST, queue);
}
static int abort_syn_rcv(struct sock *sk, struct sk_buff *skb)
{
const struct request_sock *oreq;
struct listen_ctx *listen_ctx;
struct chtls_sock *csk;
struct chtls_dev *cdev;
struct sock *psk;
void *ctx;
csk = sk->sk_user_data;
oreq = csk->passive_reap_next;
cdev = csk->cdev;
if (!oreq)
return -1;
ctx = lookup_stid(cdev->tids, oreq->ts_recent);
if (!ctx)
return -1;
listen_ctx = (struct listen_ctx *)ctx;
psk = listen_ctx->lsk;
bh_lock_sock(psk);
if (!sock_owned_by_user(psk)) {
int queue = csk->txq_idx;
do_abort_syn_rcv(sk, psk);
send_abort_rpl(sk, skb, cdev, CPL_ABORT_NO_RST, queue);
} else {
skb->sk = sk;
BLOG_SKB_CB(skb)->backlog_rcv = bl_abort_syn_rcv;
__sk_add_backlog(psk, skb);
}
bh_unlock_sock(psk);
return 0;
}
static void chtls_abort_req_rss(struct sock *sk, struct sk_buff *skb)
{
const struct cpl_abort_req_rss *req = cplhdr(skb) + RSS_HDR;
struct chtls_sock *csk = sk->sk_user_data;
int rst_status = CPL_ABORT_NO_RST;
int queue = csk->txq_idx;
if (is_neg_adv(req->status)) {
if (sk->sk_state == TCP_SYN_RECV)
chtls_set_tcb_tflag(sk, 0, 0);
kfree_skb(skb);
return;
}
csk_reset_flag(csk, CSK_ABORT_REQ_RCVD);
if (!csk_flag_nochk(csk, CSK_ABORT_SHUTDOWN) &&
!csk_flag_nochk(csk, CSK_TX_DATA_SENT)) {
struct tcp_sock *tp = tcp_sk(sk);
if (send_tx_flowc_wr(sk, 0, tp->snd_nxt, tp->rcv_nxt) < 0)
WARN_ONCE(1, "send_tx_flowc error");
csk_set_flag(csk, CSK_TX_DATA_SENT);
}
csk_set_flag(csk, CSK_ABORT_SHUTDOWN);
if (!csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING)) {
sk->sk_err = ETIMEDOUT;
if (!sock_flag(sk, SOCK_DEAD))
sk->sk_error_report(sk);
if (sk->sk_state == TCP_SYN_RECV && !abort_syn_rcv(sk, skb))
return;
chtls_release_resources(sk);
chtls_conn_done(sk);
}
chtls_send_abort_rpl(sk, skb, csk->cdev, rst_status, queue);
}
static void chtls_abort_rpl_rss(struct sock *sk, struct sk_buff *skb)
{
struct cpl_abort_rpl_rss *rpl = cplhdr(skb) + RSS_HDR;
struct chtls_sock *csk;
struct chtls_dev *cdev;
csk = rcu_dereference_sk_user_data(sk);
cdev = csk->cdev;
if (csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING)) {
csk_reset_flag(csk, CSK_ABORT_RPL_PENDING);
if (!csk_flag_nochk(csk, CSK_ABORT_REQ_RCVD)) {
if (sk->sk_state == TCP_SYN_SENT) {
cxgb4_remove_tid(cdev->tids,
csk->port_id,
GET_TID(rpl),
sk->sk_family);
sock_put(sk);
}
chtls_release_resources(sk);
chtls_conn_done(sk);
}
}
kfree_skb(skb);
}
static int chtls_conn_cpl(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_peer_close *req = cplhdr(skb) + RSS_HDR;
void (*fn)(struct sock *sk, struct sk_buff *skb);
unsigned int hwtid = GET_TID(req);
struct sock *sk;
u8 opcode;
opcode = ((const struct rss_header *)cplhdr(skb))->opcode;
sk = lookup_tid(cdev->tids, hwtid);
if (!sk)
goto rel_skb;
switch (opcode) {
case CPL_PEER_CLOSE:
fn = chtls_peer_close;
break;
case CPL_CLOSE_CON_RPL:
fn = chtls_close_con_rpl;
break;
case CPL_ABORT_REQ_RSS:
fn = chtls_abort_req_rss;
break;
case CPL_ABORT_RPL_RSS:
fn = chtls_abort_rpl_rss;
break;
default:
goto rel_skb;
}
process_cpl_msg(fn, sk, skb);
return 0;
rel_skb:
kfree_skb(skb);
return 0;
}
static struct sk_buff *dequeue_wr(struct sock *sk)
{
struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
struct sk_buff *skb = csk->wr_skb_head;
if (likely(skb)) {
/* Don't bother clearing the tail */
csk->wr_skb_head = WR_SKB_CB(skb)->next_wr;
WR_SKB_CB(skb)->next_wr = NULL;
}
return skb;
}
static void chtls_rx_ack(struct sock *sk, struct sk_buff *skb)
{
struct cpl_fw4_ack *hdr = cplhdr(skb) + RSS_HDR;
struct chtls_sock *csk = sk->sk_user_data;
struct tcp_sock *tp = tcp_sk(sk);
u32 credits = hdr->credits;
u32 snd_una;
snd_una = ntohl(hdr->snd_una);
csk->wr_credits += credits;
if (csk->wr_unacked > csk->wr_max_credits - csk->wr_credits)
csk->wr_unacked = csk->wr_max_credits - csk->wr_credits;
while (credits) {
struct sk_buff *pskb = csk->wr_skb_head;
u32 csum;
if (unlikely(!pskb)) {
if (csk->wr_nondata)
csk->wr_nondata -= credits;
break;
}
csum = (__force u32)pskb->csum;
if (unlikely(credits < csum)) {
pskb->csum = (__force __wsum)(csum - credits);
break;
}
dequeue_wr(sk);
credits -= csum;
kfree_skb(pskb);
}
if (hdr->seq_vld & CPL_FW4_ACK_FLAGS_SEQVAL) {
if (unlikely(before(snd_una, tp->snd_una))) {
kfree_skb(skb);
return;
}
if (tp->snd_una != snd_una) {
tp->snd_una = snd_una;
tp->rcv_tstamp = tcp_time_stamp(tp);
if (tp->snd_una == tp->snd_nxt &&
!csk_flag_nochk(csk, CSK_TX_FAILOVER))
csk_reset_flag(csk, CSK_TX_WAIT_IDLE);
}
}
if (hdr->seq_vld & CPL_FW4_ACK_FLAGS_CH) {
unsigned int fclen16 = roundup(failover_flowc_wr_len, 16);
csk->wr_credits -= fclen16;
csk_reset_flag(csk, CSK_TX_WAIT_IDLE);
csk_reset_flag(csk, CSK_TX_FAILOVER);
}
if (skb_queue_len(&csk->txq) && chtls_push_frames(csk, 0))
sk->sk_write_space(sk);
kfree_skb(skb);
}
static int chtls_wr_ack(struct chtls_dev *cdev, struct sk_buff *skb)
{
struct cpl_fw4_ack *rpl = cplhdr(skb) + RSS_HDR;
unsigned int hwtid = GET_TID(rpl);
struct sock *sk;
sk = lookup_tid(cdev->tids, hwtid);
if (unlikely(!sk)) {
pr_err("can't find conn. for hwtid %u.\n", hwtid);
return -EINVAL;
}
process_cpl_msg(chtls_rx_ack, sk, skb);
return 0;
}
chtls_handler_func chtls_handlers[NUM_CPL_CMDS] = {
[CPL_PASS_OPEN_RPL] = chtls_pass_open_rpl,
[CPL_CLOSE_LISTSRV_RPL] = chtls_close_listsrv_rpl,
[CPL_PASS_ACCEPT_REQ] = chtls_pass_accept_req,
[CPL_PASS_ESTABLISH] = chtls_pass_establish,
[CPL_RX_DATA] = chtls_rx_data,
[CPL_TLS_DATA] = chtls_rx_pdu,
[CPL_RX_TLS_CMP] = chtls_rx_cmp,
[CPL_PEER_CLOSE] = chtls_conn_cpl,
[CPL_CLOSE_CON_RPL] = chtls_conn_cpl,
[CPL_ABORT_REQ_RSS] = chtls_conn_cpl,
[CPL_ABORT_RPL_RSS] = chtls_conn_cpl,
[CPL_FW4_ACK] = chtls_wr_ack,
};