assume show mode when file ends in .hdn
This commit is contained in:
parent
9ebd945d4e
commit
594f1a8b6b
13
main.go
13
main.go
|
@ -9,11 +9,13 @@ import (
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
|
"regexp"
|
||||||
"syscall"
|
"syscall"
|
||||||
"golang.org/x/crypto/ssh/terminal"
|
"golang.org/x/crypto/ssh/terminal"
|
||||||
)
|
)
|
||||||
|
|
||||||
var debug bool = false
|
var debug bool = false
|
||||||
|
var FileName string
|
||||||
|
|
||||||
func checkErr(err error) {
|
func checkErr(err error) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -69,6 +71,11 @@ func data2file(filename string, data []byte) {
|
||||||
|
|
||||||
func getMode() string {
|
func getMode() string {
|
||||||
var mode string
|
var mode string
|
||||||
|
//fmt.Printf("from getMode, fn = %s \n",FileName)
|
||||||
|
match, _ := regexp.Match(`.*\.hdn$`,[]byte(FileName))
|
||||||
|
if match {
|
||||||
|
return "s"
|
||||||
|
}
|
||||||
fmt.Print("mode: ")
|
fmt.Print("mode: ")
|
||||||
fmt.Fscan(os.Stdin, &mode)
|
fmt.Fscan(os.Stdin, &mode)
|
||||||
if mode == "q" {os.Exit(1)}
|
if mode == "q" {os.Exit(1)}
|
||||||
|
@ -120,7 +127,8 @@ func checkPass() string {
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
fn := readFn()
|
FileName = readFn()
|
||||||
|
fn := FileName
|
||||||
fmt.Println(fn)
|
fmt.Println(fn)
|
||||||
mode := getMode()
|
mode := getMode()
|
||||||
switch mode {
|
switch mode {
|
||||||
|
@ -130,6 +138,9 @@ func main() {
|
||||||
data := file2data(fn)
|
data := file2data(fn)
|
||||||
secret := encrypt(data,key)
|
secret := encrypt(data,key)
|
||||||
secretfn := fn + ".hdn"
|
secretfn := fn + ".hdn"
|
||||||
|
clearfn := fn + ".origin"
|
||||||
|
err := os.Rename(fn,clearfn)
|
||||||
|
checkErr(err)
|
||||||
data2file(secretfn,secret)
|
data2file(secretfn,secret)
|
||||||
case "s":
|
case "s":
|
||||||
fmt.Print("password: ")
|
fmt.Print("password: ")
|
||||||
|
|
Loading…
Reference in New Issue