mirror of https://gitee.com/godoos/godoos.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.1 KiB
61 lines
1.1 KiB
package files
|
|
|
|
import (
|
|
"fmt"
|
|
"godo/libs"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func Test_WriteFile(t *testing.T) {
|
|
filePath := "test.txt"
|
|
content := "hello world"
|
|
configPwd := "b854f92a0bb8462ad75239152081c12b"
|
|
file, err := os.Create(filePath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer file.Close()
|
|
_, err = file.WriteString(fmt.Sprintf("@%s@", configPwd))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
encryData, err := libs.EncryptData([]byte(content), []byte(configPwd))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = file.Write(encryData)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func Test_ReadFile(t *testing.T) {
|
|
filePath := "test.txt"
|
|
pwd := "b854f92a0bb8462ad75239152081c12b"
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer file.Close()
|
|
data, err := io.ReadAll(file)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if pwd == "" {
|
|
fmt.Println(string(data))
|
|
return
|
|
}
|
|
realPwd := data[1:33]
|
|
if pwd != string(realPwd) {
|
|
t.Error("密码错误")
|
|
return
|
|
}
|
|
content := data[34:]
|
|
decryData, err := libs.DecryptData(content, []byte(pwd))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
fmt.Println(string(decryData))
|
|
}
|
|
|