add logging output to seqs.ControlBlock

soypat · Dec 13, 2023 · 95290f8 · 95290f8
1 parent 918d558
commit 95290f8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/control_user.go b/control_user.go
@@ -183,3 +183,8 @@ func (tcb *ControlBlock) MaxInFlightData() Size {
 func (tcb *ControlBlock) SetRecvWindow(wnd Size) {
 	tcb.rcv.WND = wnd
 }
+
+// SetLogger sets the logger to be used by the ControlBlock.
+func (tcb *ControlBlock) SetLogger(log *slog.Logger) {
+	tcb.log = log
+}
diff --git a/stacks/socket_tcp.go b/stacks/socket_tcp.go
@@ -153,6 +153,7 @@ func (sock *TCPSocket) open(state seqs.State, localPortNum uint16, iss seqs.Valu
 	if err != nil {
 		return err
 	}
+	sock.scb.SetLogger(sock.stack.logger)
 	sock.remoteMAC = remoteMAC
 	sock.remote = remoteAddr
 	sock.localPort = localPortNum