/
tls_server.go
49 lines (42 loc) · 1.01 KB
/
tls_server.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
package secrpc
import (
"crypto/tls"
"net"
"net/rpc"
"net/rpc/jsonrpc"
)
type SecureServer struct {
conn net.Listener
rpcServ *rpc.Server
}
func NewSecureServer(addr, certFile, keyFile string) (*SecureServer, error) {
cert, err := tls.LoadX509KeyPair(certFile, keyFile)
if err != nil {
return nil, err
}
s := &SecureServer{rpcServ: rpc.NewServer()}
s.conn, err = tls.Listen("tcp", addr, &tls.Config{
Certificates: []tls.Certificate{cert},
CipherSuites: []uint16{
tls.TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256,
tls.TLS_ECDHE_ECDSA_WITH_AES_128_GCM_SHA256,
},
MinVersion: tls.VersionTLS12,
PreferServerCipherSuites: true,
})
if err != nil {
return nil, err
}
return s, nil
}
func (s *SecureServer) RegisterName(name string, rcvr interface{}) error {
return s.rpcServ.RegisterName(name, rcvr)
}
func (s *SecureServer) Serve() error {
if conn, err := s.conn.Accept(); err == nil {
go s.rpcServ.ServeCodec(jsonrpc.NewServerCodec(conn))
} else {
return err
}
return nil
}