svn commit: r304356 - in stable/11/sys/dev/ntb: . if_ntb

[Alexander Motin]

Author: mav
Date: Thu Aug 18 09:25:10 2016
New Revision: 304356
URL: https://svnweb.freebsd.org/changeset/base/304356

Log:
  MFC r302495: Improve memory allocation errors handling on receive.

Modified:
  stable/11/sys/dev/ntb/if_ntb/if_ntb.c
  stable/11/sys/dev/ntb/ntb_transport.c
Directory Properties:
  stable/11/   (props changed)

Modified: stable/11/sys/dev/ntb/if_ntb/if_ntb.c
==============================================================================
--- stable/11/sys/dev/ntb/if_ntb/if_ntb.c	Thu Aug 18 09:24:36 2016	(r304355)
+++ stable/11/sys/dev/ntb/if_ntb/if_ntb.c	Thu Aug 18 09:25:10 2016	(r304356)
@@ -241,7 +241,12 @@ ntb_net_rx_handler(struct ntb_transport_
 	struct mbuf *m = data;
 	struct ifnet *ifp = qp_data;
 
-	CTR0(KTR_NTB, "RX: rx handler");
+	CTR1(KTR_NTB, "RX: rx handler (%d)", len);
+	if (len < 0) {
+		if_inc_counter(ifp, IFCOUNTER_IERRORS, 1);
+		return;
+	}
+
 	m->m_pkthdr.csum_flags = CSUM_IP_CHECKED | CSUM_IP_VALID;
 	(*ifp->if_input)(ifp, m);
 }

Modified: stable/11/sys/dev/ntb/ntb_transport.c
==============================================================================
--- stable/11/sys/dev/ntb/ntb_transport.c	Thu Aug 18 09:24:36 2016	(r304355)
+++ stable/11/sys/dev/ntb/ntb_transport.c	Thu Aug 18 09:25:10 2016	(r304356)
@@ -877,6 +877,8 @@ ntb_memcpy_rx(struct ntb_transport_qp *q
 	CTR2(KTR_NTB, "RX: copying %d bytes from offset %p", len, offset);
 
 	entry->buf = (void *)m_devget(offset, len, 0, ifp, NULL);
+	if (entry->buf == NULL)
+		entry->len = -ENOMEM;
 
 	/* Ensure that the data is globally visible before clearing the flag */
 	wmb();