Update the TTL field in forwarded packets

Closes: #1085

1 files changed, 18 insertions, 0 deletions

diff --git a/pkg/tcpip/network/ipv6/ipv6.go b/pkg/tcpip/network/ipv6/ipv6.go
index 9aef5234b..df86dd3b2 100644
--- a/pkg/tcpip/network/ipv6/ipv6.go
+++ b/pkg/tcpip/network/ipv6/ipv6.go
@@ -21,6 +21,7 @@
 package ipv6
 
 import (
+	"fmt"
 	"sync/atomic"
 
 	"gvisor.dev/gvisor/pkg/tcpip"
@@ -189,6 +190,23 @@ func (e *endpoint) HandlePacket(r *stack.Route, pkt tcpip.PacketBuffer) {
 	e.dispatcher.DeliverTransportPacket(r, p, pkt)
 }
 
+func (*endpoint) UpdateTTL(pkt tcpip.PacketBuffer, n uint8) *tcpip.Error {
+	headerView := pkt.Header.View()
+	h := header.IPv6(headerView)
+	if !h.IsValid(pkt.Data.Size()) {
+		fmt.Println("Malformed packet", len(h), header.IPVersion(h), pkt.Data.Size())
+		//r.Stats().IP.MalformedPacketsReceived.Increment()
+		return tcpip.ErrInvalidOptionValue
+	}
+
+	ttl := h.HopLimit()
+	if ttl <= n {
+		return tcpip.ErrTimeout
+	}
+	h.SetHopLimit(ttl - n)
+	return nil
+}
+
 // Close cleans up resources associated with the endpoint.
 func (*endpoint) Close() {}