fix endpoint nil pointer

MetaCubeX · Apr 3, 2024 · 036e5a0 · 036e5a0
1 parent 0223b8b
commit 036e5a0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/stack_gvisor.go b/stack_gvisor.go
@@ -153,7 +153,9 @@ func (t *GVisor) Start() error {
 }
 
 func (t *GVisor) Close() error {
-	t.endpoint.Attach(nil)
+	if t.endpoint != nil {
+		t.endpoint.Attach(nil)
+	}
 	t.stack.Close()
 	for _, endpoint := range t.stack.CleanupEndpoints() {
 		endpoint.Abort()

diff --git a/stack_mixed.go b/stack_mixed.go
@@ -261,7 +261,9 @@ func (m *Mixed) packetLoop() {
 }
 
 func (m *Mixed) Close() error {
-	m.endpoint.Attach(nil)
+	if m.endpoint != nil {
+		m.endpoint.Attach(nil)
+	}
 	m.stack.Close()
 	for _, endpoint := range m.stack.CleanupEndpoints() {
 		endpoint.Abort()