github
/
nps
mirror of https://github.com/ehang-io/nps
1
0
Fork 0
Code Issues Releases Wiki Activity

change exit to error when the vkey is not right

pull/298/head
刘河 2019-12-08 15:52:38 +08:00
parent f938b2fdd1
commit a602b11270
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
client/client.go
View File

@ -47,7 +47,7 @@ func (s *TRPClient) Start() {
retry: retry:
c, err := NewConn(s.bridgeConnType, s.vKey, s.svrAddr, common.WORK_MAIN, s.proxyUrl) c, err := NewConn(s.bridgeConnType, s.vKey, s.svrAddr, common.WORK_MAIN, s.proxyUrl)
if err != nil { if err != nil {
logs.Error("The connection server failed and will be reconnected in five seconds") logs.Error("The connection server failed and will be reconnected in five seconds, error", err.Error())
time.Sleep(time.Second * 5) time.Sleep(time.Second * 5)
goto retry goto retry
} }

3
client/control.go
View File

@ -241,8 +241,7 @@ func NewConn(tp string, vkey string, server string, connType string, proxyUrl st
if s, err := c.ReadFlag(); err != nil { if s, err := c.ReadFlag(); err != nil {
return nil, err return nil, err
} else if s == common.VERIFY_EER { } else if s == common.VERIFY_EER {
logs.Error("Validation key %s incorrect", vkey) return nil, errors.New(fmt.Sprintf("Validation key %s incorrect", vkey))
os.Exit(0)
} }
if _, err := c.Write([]byte(connType)); err != nil { if _, err := c.Write([]byte(connType)); err != nil {
return nil, err return nil, err