51 lines
790 B
Go
51 lines
790 B
Go
package registry
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/go-gost/gost/pkg/bypass"
|
|
)
|
|
|
|
var (
|
|
bypassReg = &bypassRegistry{}
|
|
)
|
|
|
|
func Bypass() *bypassRegistry {
|
|
return bypassReg
|
|
}
|
|
|
|
type bypassRegistry struct {
|
|
m sync.Map
|
|
}
|
|
|
|
func (r *bypassRegistry) Register(name string, bypass bypass.Bypass) error {
|
|
if _, loaded := r.m.LoadOrStore(name, bypass); loaded {
|
|
return ErrDup
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *bypassRegistry) Unregister(name string) {
|
|
r.m.Delete(name)
|
|
}
|
|
|
|
func (r *bypassRegistry) Get(name string) bypass.Bypass {
|
|
if _, ok := r.m.Load(name); !ok {
|
|
return nil
|
|
}
|
|
return &bypassWrapper{name: name}
|
|
}
|
|
|
|
type bypassWrapper struct {
|
|
name string
|
|
}
|
|
|
|
func (w *bypassWrapper) Contains(addr string) bool {
|
|
bp := bypassReg.Get(w.name)
|
|
if bp == nil {
|
|
return false
|
|
}
|
|
return bp.Contains(addr)
|
|
}
|