49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"golang.org/x/net/proxy"
|
|
|
|
mcnet "git.konjactw.dev/falloutBot/go-mc/net"
|
|
|
|
"git.konjactw.dev/patyhank/minego/pkg/bot"
|
|
)
|
|
|
|
// createSOCKS5Dialer 建立 SOCKS5 dialer
|
|
func socks5(proxyConfig *bot.ProxyConfig) (mcnet.MCDialer, error) {
|
|
var auth *proxy.Auth
|
|
if proxyConfig.Username != "" || proxyConfig.Password != "" {
|
|
auth = &proxy.Auth{
|
|
User: proxyConfig.Username,
|
|
Password: proxyConfig.Password,
|
|
}
|
|
}
|
|
|
|
dialer, err := proxy.SOCKS5("tcp", proxyConfig.Host, auth, proxy.Direct)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create SOCKS5 proxy dialer: %w", err)
|
|
}
|
|
|
|
return &socks5MCDialer{
|
|
dialer: dialer,
|
|
}, nil
|
|
}
|
|
|
|
// socks5MCDialer 實作 mcnet.MCDialer 介面，使用 SOCKS5 proxy
|
|
type socks5MCDialer struct {
|
|
dialer proxy.Dialer
|
|
}
|
|
|
|
func (d *socks5MCDialer) DialMCContext(ctx context.Context, address string) (*mcnet.Conn, error) {
|
|
// 使用 SOCKS5 proxy 建立連線
|
|
conn, err := d.dialer.Dial("tcp", address)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to dial through SOCKS5 proxy: %w", err)
|
|
}
|
|
|
|
// 將 net.Conn 包裝成 mcnet.Conn
|
|
return mcnet.WrapConn(conn), nil
|
|
}
|