40 lines
848 B
Go
40 lines
848 B
Go
package cert
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"sync"
|
|
|
|
"github.com/Dreamacro/clash/component/trie"
|
|
)
|
|
|
|
// DomainTrieCertsStorage cache wildcard certificates
|
|
type DomainTrieCertsStorage struct {
|
|
certsCache *trie.DomainTrie[*tls.Certificate]
|
|
lock sync.RWMutex
|
|
}
|
|
|
|
// Get gets the certificate from the storage
|
|
func (c *DomainTrieCertsStorage) Get(key string) (*tls.Certificate, bool) {
|
|
c.lock.RLock()
|
|
defer c.lock.RUnlock()
|
|
|
|
ca := c.certsCache.Search(key)
|
|
if ca == nil {
|
|
return nil, false
|
|
}
|
|
return ca.Data, true
|
|
}
|
|
|
|
// Set saves the certificate to the storage
|
|
func (c *DomainTrieCertsStorage) Set(key string, cert *tls.Certificate) {
|
|
c.lock.Lock()
|
|
_ = c.certsCache.Insert(key, cert)
|
|
c.lock.Unlock()
|
|
}
|
|
|
|
func NewDomainTrieCertsStorage() *DomainTrieCertsStorage {
|
|
return &DomainTrieCertsStorage{
|
|
certsCache: trie.New[*tls.Certificate](),
|
|
}
|
|
}
|