forked from mdlayher/goat
/
listener.go
60 lines (51 loc) · 1.48 KB
/
listener.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package goat
import (
"crypto/tls"
"log"
"net"
"os"
"strconv"
)
// Listen and handle HTTP (TCP) connections
func listenHTTP(sendChan chan bool, recvChan chan bool) {
// Listen on specified TCP port
l, err := net.Listen("tcp", ":"+strconv.Itoa(static.Config.Port))
if err != nil {
log.Println("Cannot start HTTP server, exiting now.")
panic(err)
}
// Send listener to handler
go handleHTTP(l, sendChan, recvChan)
}
// Listen and handle HTTPS (SSL over TCP) connections
func listenHTTPS(sendChan chan bool, recvChan chan bool) {
// Load certificate and key
cert, err := tls.LoadX509KeyPair(static.Config.SSL.Certificate, static.Config.SSL.Key)
if err != nil {
log.Println("Cannot load HTTPS X509 key pair, exiting now.")
panic(err)
}
// SSL configuration
sslConfig := tls.Config{
Certificates: []tls.Certificate{cert},
}
// Listen on specified SSL port
l, err := tls.Listen("tcp", ":"+strconv.Itoa(static.Config.SSL.Port), &sslConfig)
if err != nil {
log.Println("Cannot start HTTPS server, exiting now.")
panic(err)
}
// Send listener to handler
go handleHTTP(l, sendChan, recvChan)
}
// Listen on specified UDP port, accept and handle connections
func listenUDP(sendChan chan bool, recvChan chan bool) {
// Listen on specified UDP port
addr, err := net.ResolveUDPAddr("udp", ":"+strconv.Itoa(static.Config.Port))
l, err := net.ListenUDP("udp", addr)
if err != nil {
log.Println(err.Error())
}
// Send listener to handler
go handleUDP(l, sendChan, recvChan)
}