From 38601da6d16200479623d19a96a3e3821b3d9935 Mon Sep 17 00:00:00 2001
From: fumiama <s862105088@gmail.com>
Date: Fri, 31 Dec 2021 00:04:39 +0800
Subject: [PATCH] add ipv4 total len in wu

---
 waterutil/tun_ipv4.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/waterutil/tun_ipv4.go b/waterutil/tun_ipv4.go
index 6e249e5..c0aba4c 100644
--- a/waterutil/tun_ipv4.go
+++ b/waterutil/tun_ipv4.go
@@ -1,6 +1,7 @@
 package waterutil
 
 import (
+	"encoding/binary"
 	"net"
 )
 
@@ -12,6 +13,10 @@ func IPv4ECN(packet []byte) byte {
 	return packet[1] & 0x03
 }
 
+func IPv4TotalLength(packet []byte) uint16 {
+	return binary.BigEndian.Uint16(packet[2:4])
+}
+
 func IPv4Identification(packet []byte) [2]byte {
 	return [2]byte{packet[4], packet[5]}
 }