ed
/
hns
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

read filename from arg instead of directly loading data

This commit is contained in:
ed 2019-12-18 11:43:51 +01:00
parent 0be30e3222
commit c21833fcf0
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@ -71,28 +71,28 @@ func getMode() string {
return mode return mode
} }
func checkArg() []byte { func readFn() string {
var data []byte var filename string
arg := os.Args[1:] arg := os.Args[1:]
if len(arg) != 1 { if len(arg) != 1 {
fmt.Println("hns only takes file name as arg") fmt.Println("hns only takes file name as arg")
os.Exit(1) os.Exit(1)
} }
filename := arg[0] fn := arg[0]
if _, err := os.Stat(filename); err == nil { if _, err := os.Stat(fn); err == nil {
data = file2data(filename) filename = fn
} else { } else {
fmt.Println("No such file") fmt.Println("No such file")
os.Exit(1) os.Exit(1)
} }
return data return filename
} }
func main() { func main() {
data := checkArg() fn := readFn()
fmt.Println(fn)
mode := getMode() mode := getMode()
fmt.Println(mode) fmt.Println(mode)
fmt.Println(len(data))
/* /*
var passwd string = "ThisAnnPassphrase" var passwd string = "ThisAnnPassphrase"
fmt.Println(passwd) fmt.Println(passwd)