Cleanup daens and veth for graceful termination

cmd/run.go

@@ -268,10 +268,11 @@ loop:
 			break loop
 		}
 	}
+	defer os.Remove(PidFilePath)
+	defer control.GetDaeNetns().Close()
 	if e := c.Close(); e != nil {
 		return fmt.Errorf("close control plane: %w", e)
 	}
-	_ = os.Remove(PidFilePath)
 	return nil
 }
 

control/netns_utils.go

@@ -38,6 +38,10 @@ func init() {
 	daeNetns = &DaeNetns{}
 }
 
+func GetDaeNetns() *DaeNetns {
+	return daeNetns
+}
+
 func (ns *DaeNetns) Setup() (err error) {
 	if ns.setupDone.Load() {
 		return
@@ -55,6 +59,12 @@ func (ns *DaeNetns) Setup() (err error) {
 	return nil
 }
 
+func (ns *DaeNetns) Close() (err error) {
+	DeleteNamedNetns(NsName)
+	DeleteLink(HostVethName)
+	return
+}
+
 func (ns *DaeNetns) With(f func() error) (err error) {
 	if err = daeNetns.Setup(); err != nil {
 		return fmt.Errorf("failed to setup dae netns: %v", err)

control/udp.go

@@ -95,7 +95,7 @@ func sendPkt(data []byte, from netip.AddrPort, realTo, to netip.AddrPort, lConn
 			WithField("to", to).
 			WithField("realTo", realTo).
 			Trace("Port in use, fallback to use netns.")
-		err = daeNetns.With(func() (err error) {
+		err = GetDaeNetns().With(func() (err error) {
 			uConn, _, err = DefaultAnyfromPool.GetOrCreate(from.String(), AnyfromTimeout)
 			return err
 		})