#include <sys/param.h>
#include <sys/systm.h>
#include <sys/sysctl.h>
#include <sys/mbuf.h>
#include <sys/mcache.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <net/if_types.h>
#include <net/route.h>
#include <netinet/in.h>
#include <netinet/in_systm.h>
#include <net/if.h>
#include <netinet/ip.h>
#include <netinet/ip_var.h>
#include <netinet/in_var.h>
#include <netinet/tcp.h>
#include <netinet/tcp_seq.h>
#include <netinet/tcpip.h>
#include <netinet/tcp_var.h>
#include <netinet/tcp_lro.h>
#include <netinet/lro_ext.h>
#include <kern/locks.h>
unsigned int lrocount = 0;
unsigned int lro_seq_outoforder = 0;
unsigned int lro_seq_mismatch = 0;
unsigned int lro_eject_req = 0;
unsigned int lro_flushes = 0;
unsigned int lro_single_flushes = 0;
unsigned int lro_double_flushes = 0;
unsigned int lro_good_flushes = 0;
unsigned int coalesc_sz = LRO_MX_COALESCE_PKTS;
SYSCTL_INT(_net_inet_tcp, OID_AUTO, lro_sz, CTLFLAG_RW | CTLFLAG_LOCKED,
&coalesc_sz, 0, "Max coalescing size");
unsigned int coalesc_time = LRO_MX_TIME_TO_BUFFER;
SYSCTL_INT(_net_inet_tcp, OID_AUTO, lro_time, CTLFLAG_RW | CTLFLAG_LOCKED,
&coalesc_time, 0, "Max coalescing time");
struct lro_flow lro_flow_list[TCP_LRO_NUM_FLOWS];
char lro_flow_map[TCP_LRO_FLOW_MAP];
static lck_attr_t *tcp_lro_mtx_attr = NULL;
static lck_grp_t *tcp_lro_mtx_grp = NULL;
static lck_grp_attr_t *tcp_lro_mtx_grp_attr = NULL;
decl_lck_mtx_data( ,tcp_lro_lock);
unsigned int lro_byte_count = 0;
uint64_t lro_deadline = 0;
uint32_t lro_timer_set = 0;
u_int32_t lro_pkt_count = 0;
thread_call_t tcp_lro_timer;
extern u_int32_t kipf_count;
static void tcp_lro_timer_proc(void*, void*);
static void lro_update_stats(struct mbuf*);
static void lro_update_flush_stats(struct mbuf *);
static void tcp_lro_flush_flows(void);
static void tcp_lro_sched_timer(uint64_t);
static void lro_proto_input(struct mbuf *);
static struct mbuf *lro_tcp_xsum_validate(struct mbuf*, struct ipovly *,
struct tcphdr*);
static struct mbuf *tcp_lro_process_pkt(struct mbuf*, struct ip*, struct tcphdr*,
int);
void
tcp_lro_init(void)
{
int i;
bzero(lro_flow_list, sizeof (struct lro_flow) * TCP_LRO_NUM_FLOWS);
for (i = 0; i < TCP_LRO_FLOW_MAP; i++) {
lro_flow_map[i] = TCP_LRO_FLOW_UNINIT;
}
tcp_lro_mtx_grp_attr = lck_grp_attr_alloc_init();
tcp_lro_mtx_grp = lck_grp_alloc_init("tcplro", tcp_lro_mtx_grp_attr);
tcp_lro_mtx_attr = lck_attr_alloc_init();
lck_mtx_init(&tcp_lro_lock, tcp_lro_mtx_grp, tcp_lro_mtx_attr);
tcp_lro_timer = thread_call_allocate(tcp_lro_timer_proc, NULL);
if (tcp_lro_timer == NULL) {
panic_plain("%s: unable to allocate lro timer", __func__);
}
return;
}
static int
tcp_lro_matching_tuple(struct ip* ip_hdr, struct tcphdr *tcp_hdr, int *hash,
int *flow_id )
{
struct lro_flow *flow;
tcp_seq seqnum;
unsigned int off = 0;
int payload_len = 0;
*hash = LRO_HASH(ip_hdr->ip_src.s_addr, ip_hdr->ip_dst.s_addr,
tcp_hdr->th_sport, tcp_hdr->th_dport, (TCP_LRO_FLOW_MAP - 1));
*flow_id = lro_flow_map[*hash];
if (*flow_id == TCP_LRO_FLOW_NOTFOUND) {
return TCP_LRO_NAN;
}
seqnum = tcp_hdr->th_seq;
off = tcp_hdr->th_off << 2;
payload_len = ip_hdr->ip_len - off;
flow = &lro_flow_list[*flow_id];
if ((flow->lr_faddr.s_addr == ip_hdr->ip_src.s_addr) &&
(flow->lr_laddr.s_addr == ip_hdr->ip_dst.s_addr) &&
(flow->lr_fport == tcp_hdr->th_sport) &&
(flow->lr_lport == tcp_hdr->th_dport)) {
if (flow->lr_tcphdr == NULL) {
if (ntohl(seqnum) == flow->lr_seq) {
return TCP_LRO_COALESCE;
}
if (lrodebug >= 4) {
printf("%s: seqnum = %x, lr_seq = %x\n",
__func__, ntohl(seqnum), flow->lr_seq);
}
lro_seq_mismatch++;
if (SEQ_GT(ntohl(seqnum), flow->lr_seq)) {
lro_seq_outoforder++;
flow->lr_flags |= LRO_EJECT_REQ;
}
return TCP_LRO_NAN;
}
if (flow->lr_flags & LRO_EJECT_REQ) {
if (lrodebug)
printf("%s: eject. \n", __func__);
return TCP_LRO_EJECT_FLOW;
}
if (SEQ_GT(tcp_hdr->th_ack, flow->lr_tcphdr->th_ack)) {
if (lrodebug) {
printf("%s: th_ack = %x flow_ack = %x \n",
__func__, tcp_hdr->th_ack,
flow->lr_tcphdr->th_ack);
}
return TCP_LRO_EJECT_FLOW;
}
if (ntohl(seqnum) == (ntohl(lro_flow_list[*flow_id].lr_tcphdr->th_seq) + lro_flow_list[*flow_id].lr_len)) {
return TCP_LRO_COALESCE;
} else {
flow->lr_flags |= LRO_EJECT_REQ;
return TCP_LRO_EJECT_FLOW;
}
}
if (lrodebug) printf("tcp_lro_matching_tuple: collision \n");
return TCP_LRO_COLLISION;
}
static void
tcp_lro_init_flow(int flow_id, struct ip* ip_hdr, struct tcphdr *tcp_hdr,
int hash, u_int32_t timestamp, int payload_len)
{
struct lro_flow *flow = NULL;
flow = &lro_flow_list[flow_id];
flow->lr_hash_map = hash;
flow->lr_faddr.s_addr = ip_hdr->ip_src.s_addr;
flow->lr_laddr.s_addr = ip_hdr->ip_dst.s_addr;
flow->lr_fport = tcp_hdr->th_sport;
flow->lr_lport = tcp_hdr->th_dport;
lro_flow_map[hash] = flow_id;
flow->lr_timestamp = timestamp;
flow->lr_seq = ntohl(tcp_hdr->th_seq) + payload_len;
flow->lr_flags = 0;
return;
}
static void
tcp_lro_coalesce(int flow_id, struct mbuf *lro_mb, struct tcphdr *tcphdr,
int payload_len, int drop_hdrlen, struct tcpopt *topt,
u_int32_t* tsval, u_int32_t* tsecr, int thflags)
{
struct lro_flow *flow = NULL;
struct mbuf *last;
struct ip *ip = NULL;
flow = &lro_flow_list[flow_id];
if (flow->lr_mhead) {
if (lrodebug)
printf("%s: lr_mhead %x %d \n", __func__, flow->lr_seq,
payload_len);
m_adj(lro_mb, drop_hdrlen);
last = flow->lr_mtail;
while (last->m_next != NULL) {
last = last->m_next;
}
last->m_next = lro_mb;
flow->lr_mtail = lro_mb;
ip = mtod(flow->lr_mhead, struct ip *);
ip->ip_len += lro_mb->m_pkthdr.len;
flow->lr_mhead->m_pkthdr.len += lro_mb->m_pkthdr.len;
if (flow->lr_len == 0) {
panic_plain("%s: Inconsistent LRO flow state", __func__);
}
flow->lr_len += payload_len;
flow->lr_seq += payload_len;
flow->lr_mhead->m_pkthdr.aux_flags |= MAUXF_SW_LRO_PKT;
flow->lr_mhead->m_pkthdr.lro_npkts++;
if (flow->lr_mhead->m_pkthdr.lro_pktlen <
lro_mb->m_pkthdr.lro_pktlen) {
flow->lr_mhead->m_pkthdr.lro_pktlen =
lro_mb->m_pkthdr.lro_pktlen;
}
if (topt->to_flags & TOF_TS) {
if ((flow->lr_tsval) &&
(TSTMP_GT(topt->to_tsval, ntohl(*(flow->lr_tsval))))) {
*(flow->lr_tsval) = htonl(topt->to_tsval);
}
if ((flow->lr_tsecr) &&
(topt->to_tsecr != 0) &&
(TSTMP_GT(topt->to_tsecr, ntohl(*(flow->lr_tsecr))))) {
if (lrodebug >= 2) {
printf("%s: instantaneous RTT = %d \n", __func__,
topt->to_tsecr - ntohl(*(flow->lr_tsecr)));
}
*(flow->lr_tsecr) = htonl(topt->to_tsecr);
}
}
if (thflags) {
flow->lr_tcphdr->th_flags |= thflags;
}
flow->lr_tcphdr->th_win = tcphdr->th_win;
} else {
if (lro_mb) {
flow->lr_mhead = flow->lr_mtail = lro_mb;
flow->lr_mhead->m_pkthdr.aux_flags |= MAUXF_SW_LRO_PKT;
flow->lr_tcphdr = tcphdr;
if ((topt) && (topt->to_flags & TOF_TS)) {
ASSERT(tsval != NULL);
ASSERT(tsecr != NULL);
flow->lr_tsval = tsval;
flow->lr_tsecr = tsecr;
}
flow->lr_len = payload_len;
flow->lr_timestamp = tcp_now;
tcp_lro_sched_timer(0);
}
flow->lr_seq = ntohl(tcphdr->th_seq) + payload_len;
}
if (lro_mb) {
tcpstat.tcps_coalesced_pack++;
}
return;
}
static struct mbuf *
tcp_lro_eject_flow(int flow_id)
{
struct mbuf *mb = NULL;
mb = lro_flow_list[flow_id].lr_mhead;
ASSERT(lro_flow_map[lro_flow_list[flow_id].lr_hash_map] == flow_id);
lro_flow_map[lro_flow_list[flow_id].lr_hash_map] = TCP_LRO_FLOW_UNINIT;
bzero(&lro_flow_list[flow_id], sizeof(struct lro_flow));
return mb;
}
static struct mbuf*
tcp_lro_eject_coalesced_pkt(int flow_id)
{
struct mbuf *mb = NULL;
mb = lro_flow_list[flow_id].lr_mhead;
lro_flow_list[flow_id].lr_mhead =
lro_flow_list[flow_id].lr_mtail = NULL;
lro_flow_list[flow_id].lr_tcphdr = NULL;
return mb;
}
static struct mbuf*
tcp_lro_insert_flow(struct mbuf *lro_mb, struct ip *ip_hdr,
struct tcphdr *tcp_hdr, int payload_len,
int drop_hdrlen, int hash, struct tcpopt *topt,
u_int32_t *tsval, u_int32_t *tsecr)
{
int i;
int slot_available = 0;
int candidate_flow = 0;
u_int32_t oldest_timestamp;
struct mbuf *mb = NULL;
int collision = 0;
oldest_timestamp = tcp_now;
if (lro_flow_map[hash] != TCP_LRO_FLOW_UNINIT) {
if (lrodebug) {
collision = 1;
}
candidate_flow = lro_flow_map[hash];
tcpstat.tcps_flowtbl_collision++;
goto kick_flow;
}
for (i = 0; i < TCP_LRO_NUM_FLOWS; i++) {
if (lro_flow_list[i].lr_mhead == NULL) {
candidate_flow = i;
slot_available = 1;
break;
}
if (oldest_timestamp >= lro_flow_list[i].lr_timestamp) {
candidate_flow = i;
oldest_timestamp = lro_flow_list[i].lr_timestamp;
}
}
if (!slot_available) {
tcpstat.tcps_flowtbl_full++;
kick_flow:
mb = tcp_lro_eject_flow(candidate_flow);
if (lrodebug) {
if (!slot_available) {
printf("%s: slot unavailable.\n",__func__);
}
if (collision) {
printf("%s: collision.\n",__func__);
}
}
} else {
candidate_flow = i;
}
tcp_lro_init_flow(candidate_flow, ip_hdr, tcp_hdr, hash,
tcp_now, payload_len);
tcp_lro_coalesce(candidate_flow, lro_mb, tcp_hdr, payload_len,
drop_hdrlen, topt, tsval, tsecr, 0);
return mb;
}
struct mbuf*
tcp_lro_process_pkt(struct mbuf *lro_mb, struct ip *ip_hdr,
struct tcphdr *tcp_hdr, int drop_hdrlen)
{
int flow_id = TCP_LRO_FLOW_UNINIT;
int hash;
unsigned int off = 0;
int eject_flow = 0;
int optlen;
int retval = 0;
struct mbuf *mb = NULL;
int payload_len = 0;
u_char *optp = NULL;
int thflags = 0;
struct tcpopt to;
int ret_response = TCP_LRO_CONSUMED;
int coalesced = 0, tcpflags = 0, unknown_tcpopts = 0;
u_int8_t ecn;
if (lro_mb->m_len < (int32_t)sizeof (struct tcpiphdr)) {
if ((lro_mb = m_pullup(lro_mb, sizeof(struct tcpiphdr))) == 0) {
tcpstat.tcps_rcvshort++;
m_freem(lro_mb);
if (lrodebug) {
printf("tcp_lro_process_pkt:mbuf too short.\n");
}
return NULL;
}
}
if ((lro_mb = lro_tcp_xsum_validate(lro_mb,
(struct ipovly*)ip_hdr, tcp_hdr)) == NULL) {
if (lrodebug) {
printf("tcp_lro_process_pkt: TCP xsum failed.\n");
}
return NULL;
}
lro_pkt_count++;
lro_mb->m_pkthdr.aux_flags |= MAUXF_SW_LRO_DID_CSUM;
off = tcp_hdr->th_off << 2;
optlen = off - sizeof (struct tcphdr);
payload_len = ip_hdr->ip_len - off;
optp = (u_char *)(tcp_hdr + 1);
if ((optlen == TCPOLEN_TSTAMP_APPA ||
(optlen > TCPOLEN_TSTAMP_APPA &&
optp[TCPOLEN_TSTAMP_APPA] == TCPOPT_EOL)) &&
*(u_int32_t *)optp == htonl(TCPOPT_TSTAMP_HDR) &&
(tcp_hdr->th_flags & TH_SYN) == 0) {
to.to_flags |= TOF_TS;
to.to_tsval = ntohl(*(u_int32_t *)(void *)(optp + 4));
to.to_tsecr = ntohl(*(u_int32_t *)(void *)(optp + 8));
} else {
to.to_flags = to.to_tsecr = 0;
eject_flow = 1;
}
thflags = tcp_hdr->th_flags;
if (thflags & (TH_SYN | TH_URG | TH_ECE | TH_CWR | TH_PUSH | TH_RST | TH_FIN)) {
eject_flow = tcpflags = 1;
}
if (optlen && !((optlen == TCPOLEN_TSTAMP_APPA) &&
(to.to_flags & TOF_TS))) {
eject_flow = unknown_tcpopts = 1;
}
if (payload_len <= LRO_MIN_COALESC_SZ) {
eject_flow = 1;
}
ecn = ip_hdr->ip_tos & IPTOS_ECN_MASK;
if (ecn == IPTOS_ECN_CE) {
if (lrodebug) {
printf("%s: ECE bits set.\n", __func__);
}
eject_flow = 1;
}
lck_mtx_lock_spin(&tcp_lro_lock);
retval = tcp_lro_matching_tuple(ip_hdr, tcp_hdr, &hash, &flow_id);
switch (retval) {
case TCP_LRO_NAN:
lck_mtx_unlock(&tcp_lro_lock);
ret_response = TCP_LRO_FLOW_NOTFOUND;
break;
case TCP_LRO_COALESCE:
if ((payload_len != 0) && (unknown_tcpopts == 0) &&
(tcpflags == 0) && (ecn == 0) && (to.to_flags & TOF_TS)) {
tcp_lro_coalesce(flow_id, lro_mb, tcp_hdr, payload_len,
drop_hdrlen, &to,
(to.to_flags & TOF_TS) ? (u_int32_t *)(void *)(optp + 4) : NULL,
(to.to_flags & TOF_TS) ? (u_int32_t *)(void *)(optp + 8) : NULL,
thflags);
if (lrodebug >= 2) {
printf("tcp_lro_process_pkt: coalesce len = %d. flow_id = %d payload_len = %d drop_hdrlen = %d optlen = %d lport = %d seqnum = %x.\n",
lro_flow_list[flow_id].lr_len, flow_id,
payload_len, drop_hdrlen, optlen,
ntohs(lro_flow_list[flow_id].lr_lport),
ntohl(tcp_hdr->th_seq));
}
if (lro_flow_list[flow_id].lr_mhead->m_pkthdr.lro_npkts >= coalesc_sz) {
eject_flow = 1;
}
coalesced = 1;
}
if (eject_flow) {
mb = tcp_lro_eject_coalesced_pkt(flow_id);
lro_flow_list[flow_id].lr_seq = ntohl(tcp_hdr->th_seq) +
payload_len;
lck_mtx_unlock(&tcp_lro_lock);
if (mb) {
lro_proto_input(mb);
}
if (!coalesced) {
if (lrodebug >= 2) {
printf("%s: pkt payload_len = %d \n", __func__, payload_len);
}
lro_proto_input(lro_mb);
}
} else {
lck_mtx_unlock(&tcp_lro_lock);
}
break;
case TCP_LRO_EJECT_FLOW:
mb = tcp_lro_eject_coalesced_pkt(flow_id);
lck_mtx_unlock(&tcp_lro_lock);
if (mb) {
if (lrodebug)
printf("tcp_lro_process_pkt eject_flow, len = %d\n", mb->m_pkthdr.len);
lro_proto_input(mb);
}
lro_proto_input(lro_mb);
break;
case TCP_LRO_COLLISION:
lck_mtx_unlock(&tcp_lro_lock);
ret_response = TCP_LRO_FLOW_NOTFOUND;
break;
default:
lck_mtx_unlock(&tcp_lro_lock);
panic_plain("%s: unrecognized type %d", __func__, retval);
break;
}
if (ret_response == TCP_LRO_FLOW_NOTFOUND) {
lro_proto_input(lro_mb);
}
return NULL;
}
static void
tcp_lro_timer_proc(void *arg1, void *arg2)
{
#pragma unused(arg1, arg2)
lck_mtx_lock_spin(&tcp_lro_lock);
lro_timer_set = 0;
lck_mtx_unlock(&tcp_lro_lock);
tcp_lro_flush_flows();
}
static void
tcp_lro_flush_flows(void)
{
int i = 0;
struct mbuf *mb;
struct lro_flow *flow;
int active_flows = 0;
int outstanding_flows = 0;
int tcpclock_updated = 0;
lck_mtx_lock(&tcp_lro_lock);
while (i < TCP_LRO_NUM_FLOWS) {
flow = &lro_flow_list[i];
if (flow->lr_mhead != NULL) {
active_flows++;
if (!tcpclock_updated) {
calculate_tcp_clock();
tcpclock_updated = 1;
}
if (((tcp_now - flow->lr_timestamp) >= coalesc_time) ||
(flow->lr_mhead->m_pkthdr.lro_npkts >=
coalesc_sz)) {
if (lrodebug >= 2)
printf("tcp_lro_flush_flows: len =%d n_pkts = %d %d %d \n",
flow->lr_len,
flow->lr_mhead->m_pkthdr.lro_npkts,
flow->lr_timestamp, tcp_now);
mb = tcp_lro_eject_flow(i);
if (mb) {
lck_mtx_unlock(&tcp_lro_lock);
lro_update_flush_stats(mb);
lro_proto_input(mb);
lck_mtx_lock(&tcp_lro_lock);
}
} else {
tcp_lro_sched_timer(0);
outstanding_flows++;
if (lrodebug >= 2) {
printf("tcp_lro_flush_flows: did not flush flow of len =%d deadline = %x timestamp = %x \n",
flow->lr_len, tcp_now, flow->lr_timestamp);
}
}
}
if (flow->lr_flags & LRO_EJECT_REQ) {
mb = tcp_lro_eject_flow(i);
if (mb) {
lck_mtx_unlock(&tcp_lro_lock);
lro_proto_input(mb);
lro_eject_req++;
lck_mtx_lock(&tcp_lro_lock);
}
}
i++;
}
lck_mtx_unlock(&tcp_lro_lock);
#if 0
if (lrocount == 900) {
printf("%s: %d %d %d %d oo: %d mismatch: %d ej_req: %d coll: %d \n",
__func__,
tcpstat.tcps_coalesced_pack,
tcpstat.tcps_lro_twopack,
tcpstat.tcps_lro_multpack,
tcpstat.tcps_lro_largepack,
lro_seq_outoforder,
lro_seq_mismatch,
lro_eject_req,
tcpstat.tcps_flowtbl_collision);
printf("%s: all: %d single: %d double: %d good: %d \n",
__func__, lro_flushes, lro_single_flushes,
lro_double_flushes, lro_good_flushes);
lrocount = 0;
} else {
lrocount++;
}
if ((lrodebug >= 2) && (active_flows > 1)) {
printf("lro_flush_flows: active_flows = %d \n", active_flows);
}
#endif
}
static void
tcp_lro_sched_timer(uint64_t hint)
{
if (lro_timer_set) {
return;
}
lro_timer_set = 1;
if (!hint) {
clock_interval_to_deadline(coalesc_time,
(NSEC_PER_SEC / TCP_RETRANSHZ), &lro_deadline);
} else {
clock_interval_to_deadline(hint, NSEC_PER_SEC / TCP_RETRANSHZ,
&lro_deadline);
}
thread_call_enter_delayed(tcp_lro_timer, lro_deadline);
}
struct mbuf*
tcp_lro(struct mbuf *m, unsigned int hlen)
{
struct ip *ip_hdr;
unsigned int tlen;
struct tcphdr * tcp_hdr = NULL;
unsigned int off = 0;
if (kipf_count != 0)
return m;
if ((m->m_pkthdr.rcvif->if_type == IFT_CELLULAR) ||
(m->m_pkthdr.rcvif->if_type == IFT_LOOP)) {
return m;
}
ip_hdr = mtod(m, struct ip*);
if (ip_hdr->ip_p != IPPROTO_TCP) {
return m;
}
if (m->m_len < (int32_t) sizeof (struct tcpiphdr)) {
if (lrodebug) printf("tcp_lro m_pullup \n");
if ((m = m_pullup(m, sizeof (struct tcpiphdr))) == 0) {
tcpstat.tcps_rcvshort++;
if (lrodebug) {
printf("ip_lro: rcvshort.\n");
}
return NULL;
}
}
tcp_hdr = (struct tcphdr *)((caddr_t)ip_hdr + hlen);
tlen = ip_hdr->ip_len ; m->m_pkthdr.lro_pktlen = tlen;
m->m_pkthdr.lro_npkts = 1;
off = tcp_hdr->th_off << 2;
if (off < sizeof (struct tcphdr) || off > tlen) {
tcpstat.tcps_rcvbadoff++;
if (lrodebug) {
printf("ip_lro: TCP off greater than TCP header.\n");
}
return m;
}
return (tcp_lro_process_pkt(m, ip_hdr, tcp_hdr, hlen + off));
}
static void
lro_proto_input(struct mbuf *m)
{
struct ip* ip_hdr = mtod(m, struct ip*);
if (lrodebug >= 3) {
printf("lro_proto_input: ip_len = %d \n",
ip_hdr->ip_len);
}
lro_update_stats(m);
ip_proto_dispatch_in_wrapper(m, ip_hdr->ip_hl << 2, ip_hdr->ip_p);
}
static struct mbuf *
lro_tcp_xsum_validate(struct mbuf *m, struct ipovly *ipov, struct tcphdr * th)
{
struct ip* ip = (struct ip*)ipov;
int tlen = ip->ip_len;
int len;
struct ifnet *ifp = ((m->m_flags & M_PKTHDR) && m->m_pkthdr.rcvif != NULL) ?
m->m_pkthdr.rcvif: NULL;
MBUF_STRICT_DATA_ALIGNMENT_CHECK_32(m);
if (m->m_pkthdr.csum_flags & CSUM_DATA_VALID) {
if (m->m_pkthdr.csum_flags & CSUM_TCP_SUM16) {
u_short pseudo;
char b[9];
bcopy(ipov->ih_x1, b, sizeof (ipov->ih_x1));
bzero(ipov->ih_x1, sizeof (ipov->ih_x1));
ipov->ih_len = (u_short)tlen;
#if BYTE_ORDER != BIG_ENDIAN
HTONS(ipov->ih_len);
#endif
pseudo = in_cksum(m, sizeof (struct ip));
bcopy(b, ipov->ih_x1, sizeof (ipov->ih_x1));
th->th_sum = in_addword(pseudo, (m->m_pkthdr.csum_data & 0xFFFF));
} else {
if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR)
th->th_sum = m->m_pkthdr.csum_data;
else
th->th_sum = in_pseudo(ip->ip_src.s_addr,
ip->ip_dst.s_addr, htonl(m->m_pkthdr.csum_data +
ip->ip_len + IPPROTO_TCP));
}
th->th_sum ^= 0xffff;
} else {
char b[9];
bcopy(ipov->ih_x1, b, sizeof (ipov->ih_x1));
bzero(ipov->ih_x1, sizeof (ipov->ih_x1));
ipov->ih_len = (u_short)tlen;
#if BYTE_ORDER != BIG_ENDIAN
HTONS(ipov->ih_len);
#endif
len = sizeof (struct ip) + tlen;
th->th_sum = in_cksum(m, len);
bcopy(b, ipov->ih_x1, sizeof (ipov->ih_x1));
tcp_in_cksum_stats(len);
}
if (th->th_sum) {
tcpstat.tcps_rcvbadsum++;
if (ifp != NULL && ifp->if_tcp_stat != NULL) {
atomic_add_64(&ifp->if_tcp_stat->badformat, 1);
}
if (lrodebug)
printf("lro_tcp_xsum_validate: bad xsum and drop m = %p.\n",m);
m_freem(m);
return NULL;
}
#if BYTE_ORDER != BIG_ENDIAN
NTOHS(ipov->ih_len);
#endif
return m;
}
int
tcp_start_coalescing(struct ip *ip_hdr, struct tcphdr *tcp_hdr, int tlen)
{
int hash;
int flow_id;
struct mbuf *eject_mb;
struct lro_flow *lf;
hash = LRO_HASH(ip_hdr->ip_src.s_addr, ip_hdr->ip_dst.s_addr,
tcp_hdr->th_sport, tcp_hdr->th_dport,
(TCP_LRO_FLOW_MAP - 1));
lck_mtx_lock_spin(&tcp_lro_lock);
flow_id = lro_flow_map[hash];
if (flow_id != TCP_LRO_FLOW_NOTFOUND) {
lf = &lro_flow_list[flow_id];
if ((lf->lr_faddr.s_addr == ip_hdr->ip_src.s_addr) &&
(lf->lr_laddr.s_addr == ip_hdr->ip_dst.s_addr) &&
(lf->lr_fport == tcp_hdr->th_sport) &&
(lf->lr_lport == tcp_hdr->th_dport)) {
if ((lf->lr_tcphdr == NULL) &&
(lf->lr_seq != (tcp_hdr->th_seq + tlen))) {
lf->lr_seq = tcp_hdr->th_seq + tlen;
}
lf->lr_flags &= ~LRO_EJECT_REQ;
}
lck_mtx_unlock(&tcp_lro_lock);
return 0;
}
HTONL(tcp_hdr->th_seq);
HTONL(tcp_hdr->th_ack);
eject_mb =
tcp_lro_insert_flow(NULL, ip_hdr, tcp_hdr, tlen, 0, hash,
NULL, NULL, NULL);
lck_mtx_unlock(&tcp_lro_lock);
NTOHL(tcp_hdr->th_seq);
NTOHL(tcp_hdr->th_ack);
if (lrodebug >= 3) {
printf("%s: src = %x dst = %x sport = %d dport = %d seq %x \n",
__func__, ip_hdr->ip_src.s_addr, ip_hdr->ip_dst.s_addr,
tcp_hdr->th_sport, tcp_hdr->th_dport, tcp_hdr->th_seq);
}
ASSERT(eject_mb == NULL);
return 0;
}
int
tcp_lro_remove_state(struct in_addr saddr, struct in_addr daddr,
unsigned short sport, unsigned short dport)
{
int hash, flow_id;
struct lro_flow *lf;
hash = LRO_HASH(daddr.s_addr, saddr.s_addr, dport, sport,
(TCP_LRO_FLOW_MAP - 1));
lck_mtx_lock_spin(&tcp_lro_lock);
flow_id = lro_flow_map[hash];
if (flow_id == TCP_LRO_FLOW_UNINIT) {
lck_mtx_unlock(&tcp_lro_lock);
return 0;
}
lf = &lro_flow_list[flow_id];
if ((lf->lr_faddr.s_addr == daddr.s_addr) &&
(lf->lr_laddr.s_addr == saddr.s_addr) &&
(lf->lr_fport == dport) &&
(lf->lr_lport == sport)) {
if (lrodebug) {
printf("%s: %x %x\n", __func__,
lf->lr_flags, lf->lr_seq);
}
lf->lr_flags |= LRO_EJECT_REQ;
}
lck_mtx_unlock(&tcp_lro_lock);
return 0;
}
void
tcp_update_lro_seq(__uint32_t rcv_nxt, struct in_addr saddr, struct in_addr daddr,
unsigned short sport, unsigned short dport)
{
int hash, flow_id;
struct lro_flow *lf;
hash = LRO_HASH(daddr.s_addr, saddr.s_addr, dport, sport,
(TCP_LRO_FLOW_MAP - 1));
lck_mtx_lock_spin(&tcp_lro_lock);
flow_id = lro_flow_map[hash];
if (flow_id == TCP_LRO_FLOW_UNINIT) {
lck_mtx_unlock(&tcp_lro_lock);
return;
}
lf = &lro_flow_list[flow_id];
if ((lf->lr_faddr.s_addr == daddr.s_addr) &&
(lf->lr_laddr.s_addr == saddr.s_addr) &&
(lf->lr_fport == dport) &&
(lf->lr_lport == sport) &&
(lf->lr_tcphdr == NULL)) {
lf->lr_seq = (tcp_seq)rcv_nxt;
}
lck_mtx_unlock(&tcp_lro_lock);
return;
}
static void
lro_update_stats(struct mbuf *m)
{
switch(m->m_pkthdr.lro_npkts) {
case 0:
case 1:
break;
case 2:
tcpstat.tcps_lro_twopack++;
break;
case 3:
case 4:
tcpstat.tcps_lro_multpack++;
break;
default:
tcpstat.tcps_lro_largepack++;
break;
}
return;
}
static void
lro_update_flush_stats(struct mbuf *m)
{
lro_flushes++;
switch(m->m_pkthdr.lro_npkts) {
case 0: ASSERT(0);
case 1: lro_single_flushes++;
break;
case 2: lro_double_flushes++;
break;
default: lro_good_flushes++;
break;
}
return;
}