mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-18 23:02:33 +00:00
55 lines
979 B
Go
55 lines
979 B
Go
|
package wintun
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"sync"
|
||
|
"sync/atomic"
|
||
|
"unsafe"
|
||
|
)
|
||
|
|
||
|
func newLazyDLL(name string, onLoad func(d *lazyDLL)) *lazyDLL {
|
||
|
return &lazyDLL{Name: name, onLoad: onLoad}
|
||
|
}
|
||
|
|
||
|
func (d *lazyDLL) NewProc(name string) *lazyProc {
|
||
|
return &lazyProc{dll: d, Name: name}
|
||
|
}
|
||
|
|
||
|
type lazyProc struct {
|
||
|
Name string
|
||
|
mu sync.Mutex
|
||
|
dll *lazyDLL
|
||
|
addr uintptr
|
||
|
}
|
||
|
|
||
|
func (p *lazyProc) Find() error {
|
||
|
if atomic.LoadPointer((*unsafe.Pointer)(unsafe.Pointer(&p.addr))) != nil {
|
||
|
return nil
|
||
|
}
|
||
|
p.mu.Lock()
|
||
|
defer p.mu.Unlock()
|
||
|
if p.addr != 0 {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
err := p.dll.Load()
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error loading %v DLL: %w", p.dll.Name, err)
|
||
|
}
|
||
|
addr, err := p.nameToAddr()
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error getting %v address: %w", p.Name, err)
|
||
|
}
|
||
|
|
||
|
atomic.StorePointer((*unsafe.Pointer)(unsafe.Pointer(&p.addr)), unsafe.Pointer(addr))
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (p *lazyProc) Addr() uintptr {
|
||
|
err := p.Find()
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
return p.addr
|
||
|
}
|