X-Git-Url: https://git.tcpdump.org/tcpdump/blobdiff_plain/20eab47a4d7d6acb7ef916f61f7db9ffb3d51ff9..refs/pull/1034/head:/print-ipoib.c

diff --git a/print-ipoib.c b/print-ipoib.c
index c0ebf033..3d431235 100644
--- a/print-ipoib.c
+++ b/print-ipoib.c
@@ -29,10 +29,7 @@
 #include <config.h>
 #endif
 
-#include <netdissect-stdinc.h>
-
-#include <stdio.h>
-#include <pcap.h>
+#include "netdissect-stdinc.h"
 
 #include "netdissect.h"
 #include "extract.h"
@@ -75,19 +72,27 @@ ipoib_print(netdissect_options *ndo, const u_char *p, u_int length, u_int caplen
 	u_int orig_length;
 	u_short ether_type;
 
-	if (caplen < IPOIB_HDRLEN || length < IPOIB_HDRLEN) {
+	if (caplen < IPOIB_HDRLEN) {
+		nd_print_trunc(ndo);
+		ndo->ndo_ll_hdr_len += caplen;
+		return;
+	}
+
+	if (length < IPOIB_HDRLEN) {
 		nd_print_trunc(ndo);
+		ndo->ndo_ll_hdr_len += length;
 		return;
 	}
 
 	if (ndo->ndo_eflag) {
-		ND_PRINT("IPOIB");
+		nd_print_protocol_caps(ndo);
 		if (print_encap_header != NULL)
 			(*print_encap_header)(encap_header_arg);
 		ipoib_hdr_print(ndo, p, length);
 	}
 	orig_length = length;
 
+	ndo->ndo_ll_hdr_len += IPOIB_HDRLEN;
 	length -= IPOIB_HDRLEN;
 	caplen -= IPOIB_HDRLEN;
 	ether_type = GET_BE_U_2(p + 40);
@@ -112,11 +117,9 @@ ipoib_print(netdissect_options *ndo, const u_char *p, u_int length, u_int caplen
  * 'h->len' is the length of the packet off the wire, and 'h->caplen'
  * is the number of bytes actually captured.
  */
-u_int
+void
 ipoib_if_print(netdissect_options *ndo, const struct pcap_pkthdr *h, const u_char *p)
 {
 	ndo->ndo_protocol = "ipoib";
 	ipoib_print(ndo, p, h->len, h->caplen, NULL, NULL);
-
-	return (IPOIB_HDRLEN);
 }