Introduce the m_length() function which will return the accumulated

length of an mbuf-chain and optionally a pointer to the last mbuf.

1 files changed, 19 insertions, 6 deletions

diff --git a/sys/kern/uipc_mbuf.c b/sys/kern/uipc_mbuf.c
index 6d66245..009b812 100644
--- a/sys/kern/uipc_mbuf.c
+++ b/sys/kern/uipc_mbuf.c
@@ -714,13 +714,26 @@ m_print(const struct mbuf *m)
 int
 m_fixhdr(struct mbuf *m0)
 {
-	struct mbuf *m = m0;
-	int len = 0;
+	int len;
 
-	while (m) {
+	len = m_length(m0, NULL);
+	m0->m_pkthdr.len = len;
+	return (len);
+}
+
+int
+m_length(struct mbuf *m0, struct mbuf **last)
+{
+	struct mbuf *m;
+	int len;
+
+	len = 0;
+	for (m = m0; m != NULL; m = m->m_next) {
 		len += m->m_len;
-		m = m->m_next;
+		if (m->m_next == NULL)
+			break;
 	}
-	m0->m_pkthdr.len = len;
-	return len;
+	if (last != NULL)
+		*last = m;
+	return (len);
 }