80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package tunmanager
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
)
|
|
|
|
import (
|
|
"github.com/eyedeekay/eeproxy/conn"
|
|
"github.com/eyedeekay/eeproxy/resolve"
|
|
"github.com/eyedeekay/go-socks5"
|
|
"github.com/eyedeekay/sam3"
|
|
)
|
|
|
|
type Manager struct {
|
|
resolver.Resolver
|
|
socks5.Config
|
|
*sam3.SAM
|
|
conns []*conn.Conn
|
|
datadir string
|
|
samhost string
|
|
samport string
|
|
samopts []string
|
|
}
|
|
|
|
func (m Manager) Serve() error {
|
|
return nil
|
|
}
|
|
|
|
func (m Manager) DialI2P(ctx context.Context, addr string) (*sam3.SAMConn, error) {
|
|
i2paddr, err := sam3.NewI2PAddrFromString(addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for i, c := range m.conns {
|
|
if i2paddr.Base32() == c.RemoteAddr().(*sam3.I2PAddr).Base32() {
|
|
log.Println("Found destination for address:", i2paddr.Base32(), "at position", i)
|
|
return c.SAMConn, nil
|
|
}
|
|
}
|
|
newconn, err := conn.NewConn(m.SAM, addr, m.datadir, m.samopts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
m.conns = append(m.conns, newconn)
|
|
log.Println("Generated destination for address:", i2paddr.Base32(), "at position", len(m.conns)-1)
|
|
return m.conns[len(m.conns)-1].SAMConn, nil
|
|
}
|
|
|
|
func (m Manager) Dial(ctx context.Context, network, addr string) (net.Conn, error) {
|
|
return m.DialI2P(ctx, addr)
|
|
}
|
|
|
|
func NewManager(samhost, samport, datadir string, samopts []string) (*Manager, error) {
|
|
return NewManagerFromOptions(
|
|
SetHost(samhost),
|
|
SetPort(samport),
|
|
SetDataDir(datadir),
|
|
SetSAMOpts(samopts),
|
|
)
|
|
}
|
|
|
|
func NewManagerFromOptions(opts ...func(*Manager) error) (*Manager, error) {
|
|
var m Manager
|
|
for _, o := range opts {
|
|
if err := o(&m); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if r, err := resolver.NewResolver(); err == nil {
|
|
m.Config = socks5.Config{
|
|
Resolver: r,
|
|
Dial: m.Dial,
|
|
}
|
|
return &m, nil
|
|
}
|
|
return nil, fmt.Errorf("Resolver creation error.")
|
|
}
|