func ExampleNewProxy() {
server := serve2.New()
// SSH just so happens to send "SSH" and version number as the first thing
proxy := proto.NewProxy([]byte("SSH"), "tcp", "localhost:22")
server.AddHandlers(proxy)
l, err := net.Listen("tcp", ":8080")
if err != nil {
panic(err)
}
server.Serve(l)
}
func main() {
defer func() {
if err := recover(); err != nil {
logit("Panicked: %s", err)
}
}()
if len(os.Args) <= 1 {
panic("Missing configuration path")
}
path := os.Args[1]
bytes, err := ioutil.ReadFile(path)
if err != nil {
logit("Reading configuration failed")
panic(err)
}
err = json.Unmarshal(bytes, &conf)
if err != nil {
logit("Parsing configuration failed")
panic(err)
}
confReady = true
server = serve2.New()
if conf.LogStdout && conf.LogFile != "" {
panic("Unable to both log to stdout and to logfile")
}
if conf.LogStdout || conf.LogFile != "" {
if conf.LogFile != "" {
file, err := os.OpenFile(conf.LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
logit("Failed to open logfile: %s", conf.LogFile)
panic(err)
}
log.SetOutput(file)
}
logger = log.Printf
server.Logger = log.Printf
}
if conf.MaxRead != 0 {
server.BytesToCheck = conf.MaxRead
}
logit("Maximum buffer size: %d", server.BytesToCheck)
l, err := net.Listen("tcp", conf.Address)
if err != nil {
logit("Listen on [%s] failed", conf.Address)
panic(err)
}
logit("Listening on: %s", conf.Address)
for _, v := range conf.Protocols {
var (
handler serve2.ProtocolHandler
err error
)
switch v.Kind {
case "proxy":
magic, mok := v.Conf["magic"].(string)
magicSlice, sok := v.Conf["magic"].([]interface{})
if !mok && !sok {
panic("Proxy declaration is missing valid magic")
}
target, ok := v.Conf["target"].(string)
if !ok {
panic("Proxy declaration is missing valid target")
}
if mok {
handler = proto.NewProxy([]byte(magic), "tcp", target)
} else {
magics := make([][]byte, len(magicSlice))
for i := range magicSlice {
magic, ok := magicSlice[i].(string)
if !ok {
panic("magic declaration invalid")
}
magics[i] = []byte(magic)
}
handler = proto.NewMultiProxy(magics, "tcp", target)
}
case "tls":
cert, ok := v.Conf["cert"].(string)
if !ok {
panic("TLS declaration is missing valid certificate")
}
key, ok := v.Conf["key"].(string)
if !ok {
panic("TLS declaration is missing valid key")
}
var protos []string
y, ok := v.Conf["protos"].([]interface{})
if !ok {
panic("TLS protos declaration invalid")
}
for _, x := range y {
proto, ok := x.(string)
if !ok {
panic("TLS protos declaration invalid")
}
protos = append(protos, proto)
}
handler, err = proto.NewTLS(protos, cert, key)
if err != nil {
logit("TLS configuration failed")
panic(err)
}
case "tlsmatcher":
target, ok := v.Conf["target"].(string)
if !ok {
panic("TLSMatcher declaration is missing valid target")
}
var cb func(net.Conn) (net.Conn, error)
dialTLS, ok := v.Conf["dialTLS"].(bool)
if !ok || !dialTLS {
cb = func(c net.Conn) (net.Conn, error) {
return nil, utils.DialAndProxy(c, "tcp", target)
}
} else {
cb = func(c net.Conn) (net.Conn, error) {
serverName := ""
proto := ""
hints := utils.GetHints(c)
if len(hints) > 0 {
if tc, ok := hints[len(hints)-1].(*tls.Conn); ok {
cs := tc.ConnectionState()
serverName = cs.ServerName
proto = cs.NegotiatedProtocol
}
}
return nil, utils.DialAndProxyTLS(c, "tcp", target, &tls.Config{
ServerName: serverName,
NextProtos: []string{proto},
InsecureSkipVerify: true,
})
}
}
t := proto.NewTLSMatcher(cb)
var checks proto.TLSMatcherChecks
if sn, ok := v.Conf["serverNames"].([]interface{}); ok {
checks |= proto.TLSCheckServerName
t.ServerNames = make([]string, len(sn))
for i, x := range sn {
s, ok := x.(string)
if !ok {
panic("TLSMatcher serverNames declaration invalid")
}
t.ServerNames[i] = s
}
}
if np, ok := v.Conf["negotiatedProtocols"].([]interface{}); ok {
checks |= proto.TLSCheckNegotiatedProtocol
t.NegotiatedProtocols = make([]string, len(np))
for i, x := range np {
n, ok := x.(string)
if !ok {
panic("TLSMatcher negotiatedProtocols declaration invalid")
}
t.NegotiatedProtocols[i] = n
}
}
if npm, ok := v.Conf["negotiatedProtocolIsMutual"].(bool); ok {
checks |= proto.TLSCheckNegotiatedProtocolIsMutual
t.NegotiatedProtocolIsMutual = npm
}
t.Checks = checks
t.Description = fmt.Sprintf("TLSMatcher [dest: %s]", target)
handler = t
case "http":
h := httpHandler{}
msg, msgOk := v.Conf["notFoundMsg"]
filename, fileOk := v.Conf["notFoundFile"]
if fileOk && msgOk {
panic("HTTP notFoundMsg and notFoundFile declared simultaneously")
}
if !msgOk && !fileOk {
h.notFoundMsg = "<!DOCTYPE html><html><body><h1>404</h1></body></html>"
} else if msgOk {
h.notFoundMsg, msgOk = msg.(string)
if !msgOk {
panic("HTTP notFoundMsg declaration invalid")
}
} else if fileOk {
f, ok := filename.(string)
if !ok {
panic("HTTP notFoundFile declaration invalid")
}
x, err := ioutil.ReadFile(f)
if err != nil {
logit("HTTP unable to open notFoundFile")
panic(err)
}
h.notFoundMsg = string(x)
}
c, ok := v.Conf["defaultFile"]
if !ok {
h.defaultFile = "index.html"
} else {
h.defaultFile, ok = c.(string)
if !ok {
panic("HTTP defaultFile declaration invalid")
}
}
h.path, ok = v.Conf["path"].(string)
if !ok {
panic("HTTP path declaration invalid")
}
handler = proto.NewHTTP(h)
case "echo":
handler = proto.NewEcho()
case "discard":
handler = proto.NewDiscard()
default:
panic("Unknown kind: " + v.Kind)
}
if v.AsDefault {
server.DefaultProtocol = handler
} else {
server.AddHandler(handler)
}
}
if server.DefaultProtocol != nil {
logit("Default protocol set to: %v", server.DefaultProtocol)
}
server.Serve(l)
}
func main() {
defer func() {
if err := recover(); err != nil {
logit("Panicked: %s", err)
}
}()
if len(os.Args) <= 1 {
panic("Missing configuration path")
}
path := os.Args[1]
bytes, err := ioutil.ReadFile(path)
if err != nil {
logit("Reading configuration failed")
panic(err)
}
err = json.Unmarshal(bytes, &conf)
if err != nil {
logit("Parsing configuration failed")
panic(err)
}
confReady = true
server = serve2.New()
if conf.LogStdout && conf.LogFile != "" {
panic("Unable to both log to stdout and to logfile")
}
if conf.LogStdout || conf.LogFile != "" {
if conf.LogFile != "" {
file, err := os.Create(conf.LogFile)
if err != nil {
logit("Failed to open logfile: %s", conf.LogFile)
panic(err)
}
log.SetOutput(file)
}
logger = log.Printf
server.Logger = log.Printf
}
if conf.MaxRead != 0 {
server.BytesToCheck = conf.MaxRead
}
logit("Maximum buffer size: %d", server.BytesToCheck)
l, err := net.Listen("tcp", conf.Address)
if err != nil {
logit("Listen on [%s] failed", conf.Address)
panic(err)
}
logit("Listening on: %s", conf.Address)
for _, v := range conf.Protocols {
var (
handler serve2.ProtocolHandler
err error
)
switch v.Kind {
case "proxy":
magic, mok := v.Conf["magic"].(string)
magicSlice, sok := v.Conf["magic"].([]interface{})
if !mok && !sok {
panic("Proxy declaration is missing valid magic")
}
target, ok := v.Conf["target"].(string)
if !ok {
panic("Proxy declaration is missing valid target")
}
if mok {
handler = proto.NewProxy([]byte(magic), "tcp", target)
} else {
magics := make([][]byte, len(magicSlice))
for i := range magicSlice {
magic, ok := magicSlice[i].(string)
if !ok {
panic("magic declaration invalid")
}
magics[i] = []byte(magic)
}
handler = proto.NewMultiProxy(magics, "tcp", target)
}
case "tls":
cert, ok := v.Conf["cert"].(string)
if !ok {
panic("TLS declaration is missing valid certificate")
}
key, ok := v.Conf["key"].(string)
if !ok {
panic("TLS declaration is missing valid key")
}
var protos []string
y, ok := v.Conf["protos"].([]interface{})
if !ok {
panic("TLS protos declaration invalid")
}
for _, x := range y {
proto, ok := x.(string)
if !ok {
panic("TLS protos declaration invalid")
}
protos = append(protos, proto)
}
handler, err = proto.NewTLS(protos, cert, key)
if err != nil {
logit("TLS configuration failed")
panic(err)
}
case "http":
h := httpHandler{}
msg, msgOk := v.Conf["notFoundMsg"]
filename, fileOk := v.Conf["notFoundFile"]
if fileOk && msgOk {
panic("HTTP notFoundMsg and notFoundFile declared simultaneously")
}
if !msgOk && !fileOk {
h.notFoundMsg = "<!DOCTYPE html><html><body><h1>404</h1></body></html>"
} else if msgOk {
h.notFoundMsg, msgOk = msg.(string)
if !msgOk {
panic("HTTP notFoundMsg declaration invalid")
}
} else if fileOk {
f, ok := filename.(string)
if !ok {
panic("HTTP notFoundFile declaration invalid")
}
x, err := ioutil.ReadFile(f)
if err != nil {
logit("HTTP unable to open notFoundFile")
panic(err)
}
h.notFoundMsg = string(x)
}
c, ok := v.Conf["defaultFile"]
if !ok {
h.defaultFile = "index.html"
} else {
h.defaultFile, ok = c.(string)
if !ok {
panic("HTTP defaultFile declaration invalid")
}
}
h.path, ok = v.Conf["path"].(string)
if !ok {
panic("HTTP path declaration invalid")
}
handler = proto.NewHTTP(h)
case "echo":
handler = proto.NewEcho()
case "discard":
handler = proto.NewDiscard()
default:
panic("Unknown kind: " + v.Kind)
}
if v.AsDefault {
server.DefaultProtocol = handler
} else {
server.AddHandler(handler)
}
}
if server.DefaultProtocol != nil {
logit("Default protocol set to: %v", server.DefaultProtocol)
}
server.Serve(l)
}