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

Broken

This commit is contained in:
Zack Scholl 2017-10-21 16:07:44 -06:00
parent 3bb5f43c07
commit 23e19b009c
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
connect.go
View File

@ -387,6 +387,10 @@ func (c *Connection) receiveFile(id int, connection net.Conn) error {
fileSizeInt, _ := strconv.Atoi(fileDataString)
chunkSize := int64(fileSizeInt)
logger.Debugf("chunk size: %d", chunkSize)
if chunkSize == 0 {
logger.Debug(fileSizeBuffer)
return errors.New("chunk size is empty!")
}
os.Remove(c.File.Name + ".enc." + strconv.Itoa(id))
log.Debug("Making " + c.File.Name + ".enc." + strconv.Itoa(id))
@ -448,7 +452,10 @@ func (c *Connection) sendFile(id int, connection net.Conn) error {
return err
}
logger.Debugf("sending chunk size: %d", fi.Size())
connection.Write([]byte(fillString(strconv.FormatInt(int64(fi.Size()), 10), 10)))
_, err = connection.Write([]byte(fillString(strconv.FormatInt(int64(fi.Size()), 10), 10)))
if err != nil {
return errors.Wrap(err, "Problem sending chunk data: ")
}
// show the progress
if !c.Debug {