-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.go
60 lines (50 loc) · 1006 Bytes
/
main.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 main
import (
"flag"
"fmt"
"os"
"strings"
local "github.com/Si-Huan/nothing/client"
"github.com/Si-Huan/nothing/common"
server "github.com/Si-Huan/nothing/server"
)
func main() {
listCiphers := flag.Bool("listcipher", false, "List supported ciphers")
genKey := flag.Bool("genkey", false, "Generate key file")
flag.Parse()
if *genKey {
common.RSAGenKey(4096, "nothing_c")
common.RSAGenKey(4096, "nothing_s")
return
}
if *listCiphers {
println(strings.Join(common.ListCipher(), " "))
return
}
if len(config.Client) == 0 && len(config.Server) == 0 {
flag.Usage()
return
}
if len(config.Client) > 0 {
runclient()
}
if len(config.Server) > 0 {
runserver()
}
}
func runclient() {
client, err := local.NewClient(config)
if err != nil {
fmt.Println("致命错误:", err)
os.Exit(-1)
}
client.Start()
}
func runserver() {
server, err := server.NewServer(config)
if err != nil {
fmt.Println("致命错误:", err)
os.Exit(-1)
}
server.Start()
}