Skip to content

Commit

Permalink
Repair totally broken STARTTLS and TLS listeners
Browse files Browse the repository at this point in the history
  • Loading branch information
decke committed Dec 13, 2018
1 parent a70b148 commit b55f556
Showing 1 changed file with 31 additions and 7 deletions.
38 changes: 31 additions & 7 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -59,10 +59,9 @@ func main() {
}

if strings.Index(listeners[i], "://") == -1 {
;
} else if strings.HasPrefix(listeners[i], "tls://") || strings.HasPrefix(listeners[i], "starttls://") {

listener = strings.TrimPrefix(listener, "tls://")
log.Printf("Listen on %s ...\n", listener)
go server.ListenAndServe(listener)
} else if strings.HasPrefix(listeners[i], "starttls://") {
listener = strings.TrimPrefix(listener, "starttls://")

if *localCert == "" || *localKey == "" {
Expand All @@ -74,16 +73,41 @@ func main() {
log.Fatal(err)
}

server.TLSConfig = &tls.Config {
Certificates: [] tls.Certificate{cert},
}
server.ForceTLS = *localForceTLS

log.Printf("Listen on %s (STARTSSL) ...\n", listener)
lsnr, err := net.Listen("tcp", listener)
defer lsnr.Close()

go server.Serve(lsnr)
} else if strings.HasPrefix(listeners[i], "tls://") {

listener = strings.TrimPrefix(listener, "tls://")

if *localCert == "" || *localKey == "" {
log.Fatal("TLS certificate/key not defined in config")
}

cert, err := tls.LoadX509KeyPair(*localCert, *localKey)
if err != nil {
log.Fatal(err)
}

server.TLSConfig = &tls.Config {
Certificates: [] tls.Certificate{cert},
}

log.Printf("Listen on %s (TLS) ...\n", listener)
lsnr, err := tls.Listen("tcp", listener, server.TLSConfig)
defer lsnr.Close()

go server.Serve(lsnr)
} else {
log.Fatal("Unknown protocol in listener ", listener)
}

log.Printf("Listen on %s ...\n", listener)
go server.ListenAndServe(listener)
}

for true {
Expand Down

0 comments on commit b55f556

Please sign in to comment.