63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"git.coolaj86.com/coolaj86/go-telebitd/mplexer"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
r := &Router{
|
||
|
|
secret: os.Getenv("SECRET"),
|
||
|
|
}
|
||
|
|
m := &mplexer.MultiplexLocal{
|
||
|
|
Relay: os.Getenv("RELAY"),
|
||
|
|
SortingHat: r,
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := context.Background()
|
||
|
|
|
||
|
|
// TODO more m.ListenAndServe(mux) style?
|
||
|
|
m.ListenAndServe(ctx)
|
||
|
|
}
|
||
|
|
|
||
|
|
type Router struct {
|
||
|
|
secret string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Router) Authz() (string, error) {
|
||
|
|
return r.secret, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// this function is very client-specific logic
|
||
|
|
func (r *Router) LookupTarget(paddr packer.Addr) (net.Conn, error) {
|
||
|
|
//if target := LookupPort(paddr.Servername()); nil != target { }
|
||
|
|
if target := r.LookupServername(paddr.Port()); nil != target {
|
||
|
|
tconn, err := net.Dial(target.Network(), target.Hostname())
|
||
|
|
if nil != err {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
/*
|
||
|
|
// TODO for http proxy
|
||
|
|
return mplexer.TargetOptions {
|
||
|
|
Hostname // default localhost
|
||
|
|
Termination // default TLS
|
||
|
|
XFWD // default... no?
|
||
|
|
Port // default 0
|
||
|
|
Conn // should be dialed beforehand
|
||
|
|
}, nil
|
||
|
|
*/
|
||
|
|
return tconn, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Router) LookupServername(servername string) mplexer.Addr {
|
||
|
|
return &mplexer.NewAddr(
|
||
|
|
mplexer.HTTPS,
|
||
|
|
mplexer.TCP, // TCP -> termination.None? / Plain?
|
||
|
|
"localhost",
|
||
|
|
3000,
|
||
|
|
)
|
||
|
|
}
|