From 7545dc2d5641922d6a7a05e11cc36211fa539798 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Mon, 21 Aug 2023 14:30:38 +0800
Subject: [PATCH] Fix darwin monitor socket leak

---
 monitor_darwin.go | 29 ++++++++++++++++++++++-------
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/monitor_darwin.go b/monitor_darwin.go
index 9455ac3..938bb14 100644
--- a/monitor_darwin.go
+++ b/monitor_darwin.go
@@ -17,14 +17,19 @@ import (
 )
 
 type networkUpdateMonitor struct {
-	access      sync.Mutex
-	callbacks   list.List[NetworkUpdateCallback]
-	routeSocket int
-	logger      logger.Logger
+	access          sync.Mutex
+	callbacks       list.List[NetworkUpdateCallback]
+	routeSocketFile *os.File
+	closeOnce       sync.Once
+	done            chan struct{}
+	logger          logger.Logger
 }
 
 func NewNetworkUpdateMonitor(logger logger.Logger) (NetworkUpdateMonitor, error) {
-	return &networkUpdateMonitor{logger: logger}, nil
+	return &networkUpdateMonitor{
+		logger: logger,
+		done:   make(chan struct{}),
+	}, nil
 }
 
 func (m *networkUpdateMonitor) Start() error {
@@ -34,6 +39,11 @@ func (m *networkUpdateMonitor) Start() error {
 
 func (m *networkUpdateMonitor) loopUpdate() {
 	for {
+		select {
+		case <-m.done:
+			return
+		case <-time.After(time.Second):
+		}
 		err := m.loopUpdate0()
 		if err != nil {
 			m.logger.Error("listen network update: ", err)
@@ -47,7 +57,9 @@ func (m *networkUpdateMonitor) loopUpdate0() error {
 	if err != nil {
 		return err
 	}
-	m.loopUpdate1(os.NewFile(uintptr(routeSocket), "route"))
+	routeSocketFile := os.NewFile(uintptr(routeSocket), "route")
+	m.routeSocketFile = routeSocketFile
+	m.loopUpdate1(routeSocketFile)
 	return nil
 }
 
@@ -73,7 +85,10 @@ func (m *networkUpdateMonitor) loopUpdate1(routeSocketFile *os.File) {
 }
 
 func (m *networkUpdateMonitor) Close() error {
-	return unix.Close(m.routeSocket)
+	m.closeOnce.Do(func() {
+		close(m.done)
+	})
+	return nil
 }
 
 func (m *defaultInterfaceMonitor) checkUpdate() error {