sing-box/common/dialer/conntrack/track.go

package conntrack

import (
	"io"
	"sync"

	"github.com/sagernet/sing/common"
	"github.com/sagernet/sing/common/x/list"
)

var (
	connAccess     sync.RWMutex
	openConnection list.List[*ConnEntry]
)

type ConnEntry struct {
	Conn  io.Closer
	Stack []byte
}

func Count() int {
	return openConnection.Len()
}

func List() []*ConnEntry {
	connAccess.RLock()
	defer connAccess.RUnlock()
	connList := make([]*ConnEntry, 0, openConnection.Len())
	for element := openConnection.Front(); element != nil; element = element.Next() {
		connList = append(connList, element.Value)
	}
	return connList
}

func Close() {
	connAccess.Lock()
	defer connAccess.Unlock()
	for element := openConnection.Front(); element != nil; element = element.Next() {
		common.Close(element.Value.Conn)
		element.Value = nil
	}
	openConnection = list.List[*ConnEntry]{}
}