arch/soju
1
0
Fork 0
Code Issues Pull requests 1 Actions Packages Projects Releases Wiki Activity

cmd/soju: add withDefaultPort helper

Browse source
This commit is contained in:
Simon Ser 2024-01-08 18:22:39 +01:00
parent 4c818d28cc
commit 866845ea95
1 changed files with 17 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
cmd/soju/main.go
View file

@ -165,16 +165,13 @@ func main() {
if tlsCfg == nil {
log.Fatalf("failed to listen on %q: missing TLS configuration", listen)
}
host := u.Host
if _, _, err := net.SplitHostPort(host); err != nil {
host = host + ":6697"
}
addr := withDefaultPort(u.Host, "6697")
ircsTLSCfg := tlsCfg.Clone()
ircsTLSCfg.NextProtos = []string{"irc"}
lc := net.ListenConfig{
KeepAlive: downstreamKeepAlive,
}
l, err := lc.Listen(context.Background(), "tcp", host)
l, err := lc.Listen(context.Background(), "tcp", addr)
if err != nil {
log.Fatalf("failed to start TLS listener on %q: %v", listen, err)
}
@ -186,14 +183,11 @@ func main() {
}
}()
case "irc+insecure":
host := u.Host
if _, _, err := net.SplitHostPort(host); err != nil {
host = host + ":6667"
}
addr := withDefaultPort(u.Host, "6667")
lc := net.ListenConfig{
KeepAlive: downstreamKeepAlive,
}
ln, err := lc.Listen(context.Background(), "tcp", host)
ln, err := lc.Listen(context.Background(), "tcp", addr)
if err != nil {
log.Fatalf("failed to start listener on %q: %v", listen, err)
}
@ -237,12 +231,8 @@ func main() {
if tlsCfg == nil {
log.Fatalf("failed to listen on %q: missing TLS configuration", listen)
}
addr := u.Host
if _, _, err := net.SplitHostPort(addr); err != nil {
addr = addr + ":https"
}
httpSrv := http.Server{
Addr: addr,
Addr: withDefaultPort(u.Host, "https"),
TLSConfig: tlsCfg,
Handler: srv,
}
@ -252,12 +242,8 @@ func main() {
}
}()
case "ws+insecure":
addr := u.Host
if _, _, err := net.SplitHostPort(addr); err != nil {
addr = addr + ":http"
}
httpSrv := http.Server{
Addr: addr,
Addr: withDefaultPort(u.Host, "http"),
Handler: srv,
}
go func() {
@ -270,11 +256,8 @@ func main() {
srv.Identd = identd.New()
}
host := u.Host
if _, _, err := net.SplitHostPort(host); err != nil {
host = host + ":113"
}
ln, err := net.Listen("tcp", host)
addr := withDefaultPort(u.Host, "113")
ln, err := net.Listen("tcp", addr)
if err != nil {
log.Fatalf("failed to start listener on %q: %v", listen, err)
}
@ -339,12 +322,8 @@ func main() {
if tlsCfg == nil {
log.Fatalf("failed to listen on %q: missing TLS configuration", listen)
}
addr := u.Host
if _, _, err := net.SplitHostPort(addr); err != nil {
addr = addr + ":https"
}
httpSrv := http.Server{
Addr: addr,
Addr: withDefaultPort(u.Host, "https"),
TLSConfig: tlsCfg,
Handler: httpMux,
}
@ -354,12 +333,8 @@ func main() {
}
}()
case "http+insecure":
addr := u.Host
if _, _, err := net.SplitHostPort(addr); err != nil {
addr = addr + ":http"
}
httpSrv := http.Server{
Addr: addr,
Addr: withDefaultPort(u.Host, "http"),
Handler: httpMux,
}
go func() {
@ -420,3 +395,10 @@ func proxyProtoListener(ln net.Listener, srv *soju.Server) net.Listener {
ReadHeaderTimeout: 5 * time.Second,
}
}
func withDefaultPort(addr, port string) string {
if _, _, err := net.SplitHostPort(addr); err != nil {
addr += ":" + port
}
return addr
}