From 828f83601db44d3f99f6a7389eb1d491b22ad84b Mon Sep 17 00:00:00 2001 From: Zack Scholl Date: Fri, 21 Sep 2018 21:13:46 -0700 Subject: [PATCH] fix hash file --- src/recipient/recipient.go | 6 ++++-- src/utils/hash.go | 3 ++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/src/recipient/recipient.go b/src/recipient/recipient.go index dbfb6186..616140a7 100644 --- a/src/recipient/recipient.go +++ b/src/recipient/recipient.go @@ -89,8 +89,9 @@ func receive(c *websocket.Conn, codephrase string) (err error) { return err } // await file - f, err := os.Create("out") + f, err := os.Create(fstats.Name) if err != nil { + log.Error(err) return err } bytesWritten := 0 @@ -149,8 +150,9 @@ func receive(c *websocket.Conn, codephrase string) (err error) { bar.Finish() // check hash - hash256, err := utils.HashFile("out") + hash256, err := utils.HashFile(fstats.Name) if err != nil { + log.Error(err) return err } diff --git a/src/utils/hash.go b/src/utils/hash.go index 7adbc9ca..83d512bb 100644 --- a/src/utils/hash.go +++ b/src/utils/hash.go @@ -6,8 +6,9 @@ import ( "os" ) +// HashFile returns the md5 hash of a file func HashFile(fname string) (hash256 []byte, err error) { - f, err := os.Open("file.txt") + f, err := os.Open(fname) if err != nil { return }