Robot
/
croc
mirror of https://github.com/schollz/croc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

reduce complexity

This commit is contained in:
Zack Scholl 2019-09-20 09:45:42 -07:00
parent 24abf4fdd8
commit 07c20e2cde
1 changed files with 31 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/croc/croc.go
View File

@ -479,7 +479,7 @@ func (c *Client) transfer(options TransferOptions) (err error) {
return
}
func (c *Client) processMessageFileInfo(m message.Message) (err error) {
func (c *Client) processMessageFileInfo(m message.Message) (done bool, err error) {
var senderInfo SenderInfo
err = json.Unmarshal(m.Bytes, &senderInfo)
if err != nil {
@ -575,23 +575,7 @@ func (c *Client) procesMesssagePake(m message.Message) (err error) {
return
}
func (c *Client) processMessage(payload []byte) (done bool, err error) {
m, err := message.Decode(c.Key, payload)
if err != nil {
return
}
switch m.Type {
case "finished":
err = message.Send(c.conn[0], c.Key, message.Message{
Type: "finished",
})
done = true
c.SuccessfulTransfer = true
return
case "pake":
err = c.procesMesssagePake(m)
case "salt":
func (c *Client) processMessageSalt(m message.Message) (done bool, err error) {
log.Debug("received salt")
if !c.Options.IsSender {
log.Debug("sending salt back")
@ -616,13 +600,34 @@ func (c *Client) processMessage(payload []byte) (done bool, err error) {
}
log.Debugf("connected as %s -> %s", c.ExternalIP, c.ExternalIPConnected)
c.Step1ChannelSecured = true
return
}
func (c *Client) processMessage(payload []byte) (done bool, err error) {
m, err := message.Decode(c.Key, payload)
if err != nil {
return
}
switch m.Type {
case "finished":
err = message.Send(c.conn[0], c.Key, message.Message{
Type: "finished",
})
done = true
c.SuccessfulTransfer = true
return
case "pake":
err = c.procesMesssagePake(m)
case "salt":
done, err = c.processMessageSalt(m)
case "error":
// c.spinner.Stop()
fmt.Print("\r")
err = fmt.Errorf("peer error: %s", m.Message)
return true, err
case "fileinfo":
err = c.processMessageFileInfo(m)
done, err = c.processMessageFileInfo(m)
case "recipientready":
var remoteFile RemoteFileRequest
err = json.Unmarshal(m.Bytes, &remoteFile)