mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-11-10 02:53:12 +00:00
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package fakeip
|
|
|
|
import (
|
|
"net/netip"
|
|
|
|
"github.com/sagernet/sing-box/adapter"
|
|
"github.com/sagernet/sing/common/cache"
|
|
"github.com/sagernet/sing/common/logger"
|
|
)
|
|
|
|
var _ adapter.FakeIPStorage = (*MemoryStorage)(nil)
|
|
|
|
type MemoryStorage struct {
|
|
metadata *adapter.FakeIPMetadata
|
|
domainCache *cache.LruCache[netip.Addr, string]
|
|
}
|
|
|
|
func NewMemoryStorage() *MemoryStorage {
|
|
return &MemoryStorage{
|
|
domainCache: cache.New[netip.Addr, string](),
|
|
}
|
|
}
|
|
|
|
func (s *MemoryStorage) FakeIPMetadata() *adapter.FakeIPMetadata {
|
|
return s.metadata
|
|
}
|
|
|
|
func (s *MemoryStorage) FakeIPSaveMetadata(metadata *adapter.FakeIPMetadata) error {
|
|
s.metadata = metadata
|
|
return nil
|
|
}
|
|
|
|
func (s *MemoryStorage) FakeIPStore(address netip.Addr, domain string) error {
|
|
s.domainCache.Store(address, domain)
|
|
return nil
|
|
}
|
|
|
|
func (s *MemoryStorage) FakeIPStoreAsync(address netip.Addr, domain string, logger logger.Logger) {
|
|
s.domainCache.Store(address, domain)
|
|
}
|
|
|
|
func (s *MemoryStorage) FakeIPLoad(address netip.Addr) (string, bool) {
|
|
return s.domainCache.Load(address)
|
|
}
|
|
|
|
func (s *MemoryStorage) FakeIPReset() error {
|
|
s.domainCache = cache.New[netip.Addr, string]()
|
|
return nil
|
|
}
|