schollz/croc
1
0
Fork 0
mirror of https://github.com/schollz/croc.git synced 2024-11-28 01:16:10 +03:00
Code Issues Projects Releases Wiki Activity

use sent name

Browse Source
This commit is contained in:
Zack Scholl 2018-09-21 21:55:07 -07:00
parent 83f199b213
commit f9ed7553af
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/filestats.go
View File

@ -7,5 +7,5 @@ type FileStats struct {
Size int64
ModTime time.Time
IsDir bool
DirName string
SentName string
}

4
src/recipient/recipient.go
View File

@ -89,7 +89,7 @@ func receive(c *websocket.Conn, codephrase string) (err error) {
return err
}
// await file
f, err := os.Create(fstats.Name)
f, err := os.Create(fstats.SentName)
if err != nil {
log.Error(err)
return err
@ -150,7 +150,7 @@ func receive(c *websocket.Conn, codephrase string) (err error) {
bar.Finish()
// check hash
hash256, err := utils.HashFile(fstats.Name)
hash256, err := utils.HashFile(fstats.SentName)
if err != nil {
log.Error(err)
return err

7
src/sender/sender.go
View File

@ -49,16 +49,17 @@ func send(c *websocket.Conn, fname string, codephrase string) (err error) {
if err != nil {
return err
}
fstats := models.FileStats{fstat.Name(), fstat.Size(), fstat.ModTime(), fstat.IsDir(), ""}
// get stats
fstats := models.FileStats{fstat.Name(), fstat.Size(), fstat.ModTime(), fstat.IsDir(), fstats.Name()}
if fstats.IsDir {
// zip the directory
fstats.DirName, err = zipper.ZipFile(fname, true)
fstats.SentName, err = zipper.ZipFile(fname, true)
if err != nil {
return
}
f.Close()
// reopen file
f, err = os.Open(fstats.DirName)
f, err = os.Open(fstats.SentName)
if err != nil {
return
}