-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
0 parents
commit 29952b5
Showing
8 changed files
with
412 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
.idea/ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,116 @@ | ||
package cryptor | ||
|
||
import ( | ||
"crypto/hmac" | ||
"crypto/sha256" | ||
"github.com/gogf/gf/v2/encoding/gbase64" | ||
"github.com/gogf/gf/v2/errors/gcode" | ||
"github.com/gogf/gf/v2/errors/gerror" | ||
"github.com/gogf/gf/v2/frame/g" | ||
"github.com/wsy9981999/goframe-extends/json" | ||
"hash" | ||
"sync" | ||
) | ||
|
||
const ( | ||
value = "value" | ||
iv = "iv" | ||
mac = "mac" | ||
) | ||
|
||
type payload struct { | ||
Value []byte | ||
IV []byte | ||
Mac []byte | ||
} | ||
|
||
var hashInstance hash.Hash | ||
var once = sync.Once{} | ||
|
||
func (p *payload) UnmarshalJSON(bytes []byte) (err error) { | ||
|
||
var m g.MapStrStr | ||
if err = json.Unmarshal(bytes, &m); err != nil { | ||
return err | ||
} | ||
for _, v := range []string{iv, value, mac} { | ||
val, exists := m[v] | ||
if !exists || len(v) == 0 { | ||
return gerror.NewCode(gcode.CodeMissingParameter) | ||
} | ||
switch v { | ||
case iv: | ||
p.IV, err = gbase64.DecodeString(val) | ||
if err != nil { | ||
return err | ||
} | ||
case value: | ||
p.Value, err = gbase64.DecodeString(val) | ||
if err != nil { | ||
return err | ||
} | ||
case mac: | ||
p.Mac, err = gbase64.DecodeString(val) | ||
if err != nil { | ||
return err | ||
} | ||
} | ||
} | ||
|
||
return nil | ||
} | ||
|
||
func (p *payload) String() string { | ||
marshalJSON, err := p.MarshalJSON() | ||
if err != nil { | ||
return "" | ||
} | ||
return gbase64.EncodeToString(marshalJSON) | ||
} | ||
|
||
func (p *payload) MarshalJSON() ([]byte, error) { | ||
var s = g.MapStrStr{} | ||
s[value] = gbase64.EncodeToString(p.Value) | ||
s[mac] = gbase64.EncodeToString(p.Mac) | ||
s[iv] = gbase64.EncodeToString(p.IV) | ||
return json.Marshal(s) | ||
} | ||
|
||
func (p *payload) check(key []byte) bool { | ||
newMac := calcMac(p.Value, p.IV, key) | ||
return hmac.Equal(newMac, p.Mac) | ||
|
||
} | ||
|
||
func newPayload(value, iv, key []byte) *payload { | ||
return &payload{ | ||
Value: value, | ||
IV: iv, | ||
Mac: calcMac(value, iv, key), | ||
} | ||
} | ||
|
||
func calcMac(value, iv, key []byte) []byte { | ||
once.Do(func() { | ||
hashInstance = hmac.New(sha256.New, key) | ||
}) | ||
defer hashInstance.Reset() | ||
|
||
hashInstance.Write(value) | ||
hashInstance.Write(iv) | ||
return hashInstance.Sum(nil) | ||
} | ||
func parse(value string, key []byte) (*payload, error) { | ||
decodeString, err := gbase64.DecodeString(value) | ||
if err != nil { | ||
return nil, err | ||
} | ||
var s *payload | ||
if err = json.Unmarshal(decodeString, &s); err != nil { | ||
return nil, err | ||
} | ||
if !s.check(key) { | ||
return nil, gerror.NewCode(gcode.CodeInvalidParameter) | ||
} | ||
return s, nil | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,95 @@ | ||
package cryptor | ||
|
||
import ( | ||
"bytes" | ||
"crypto/aes" | ||
"crypto/cipher" | ||
"errors" | ||
"fmt" | ||
"github.com/gogf/gf/v2/errors/gcode" | ||
"github.com/gogf/gf/v2/errors/gerror" | ||
"github.com/gogf/gf/v2/util/grand" | ||
) | ||
|
||
type Cryptor struct { | ||
cipher cipher.Block | ||
k []byte | ||
encryptor cipher.BlockMode | ||
decryptor cipher.BlockMode | ||
} | ||
|
||
func NewCryptor(key []byte) *Cryptor { | ||
c, err := aes.NewCipher(key) | ||
if err != nil { | ||
return nil | ||
} | ||
|
||
return &Cryptor{cipher: c, k: key} | ||
} | ||
|
||
func (receiver *Cryptor) Encrypt(plaintext []byte) string { | ||
ivValue := grand.B(16) | ||
plainText := PKCS7Padding(plaintext, receiver.cipher.BlockSize()) | ||
encrypter := cipher.NewCBCEncrypter(receiver.cipher, ivValue) | ||
cipherText := make([]byte, len(plainText)) | ||
encrypter.CryptBlocks(cipherText, plainText) | ||
return newPayload(cipherText, ivValue, receiver.k).String() | ||
} | ||
func (receiver *Cryptor) EncryptStr(plaintext string) string { | ||
return receiver.Encrypt([]byte(plaintext)) | ||
} | ||
|
||
func (receiver *Cryptor) Decrypt(payload string) ([]byte, error) { | ||
p, err := parse(payload, receiver.k) | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
blockSize := receiver.cipher.BlockSize() | ||
if len(p.Value) < blockSize { | ||
return nil, gerror.NewCode(gcode.CodeInvalidParameter, "cipherText too short") | ||
} | ||
|
||
if len(p.Value)%blockSize != 0 { | ||
return nil, gerror.NewCode(gcode.CodeInvalidParameter, "cipherText is not a multiple of the block size") | ||
} | ||
blockModel := cipher.NewCBCDecrypter(receiver.cipher, p.IV) | ||
plainText := make([]byte, len(p.Value)) | ||
blockModel.CryptBlocks(plainText, p.Value) | ||
plainText, e := PKCS7UnPadding(plainText, blockSize) | ||
if e != nil { | ||
return nil, e | ||
} | ||
return plainText, nil | ||
} | ||
func PKCS7Padding(src []byte, blockSize int) []byte { | ||
padding := blockSize - len(src)%blockSize | ||
padtext := bytes.Repeat([]byte{byte(padding)}, padding) | ||
return append(src, padtext...) | ||
} | ||
|
||
// PKCS7UnPadding removes PKCS#7 padding from the source byte slice based on the given block size. | ||
func PKCS7UnPadding(src []byte, blockSize int) ([]byte, error) { | ||
length := len(src) | ||
if blockSize <= 0 { | ||
return nil, errors.New(fmt.Sprintf("invalid blockSize: %d", blockSize)) | ||
} | ||
|
||
if length%blockSize != 0 || length == 0 { | ||
return nil, errors.New("invalid data len") | ||
} | ||
|
||
unpadding := int(src[length-1]) | ||
if unpadding > blockSize || unpadding == 0 { | ||
return nil, errors.New("invalid unpadding") | ||
} | ||
|
||
padding := src[length-unpadding:] | ||
for i := 0; i < unpadding; i++ { | ||
if padding[i] != byte(unpadding) { | ||
return nil, errors.New("invalid padding") | ||
} | ||
} | ||
|
||
return src[:(length - unpadding)], nil | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
module github.com/wsy9981999/goframe-extends | ||
|
||
go 1.22 | ||
|
||
require ( | ||
github.com/bytedance/sonic v1.11.7 | ||
github.com/gogf/gf/v2 v2.7.1 | ||
golang.org/x/crypto v0.23.0 | ||
) | ||
|
||
require ( | ||
github.com/BurntSushi/toml v1.3.2 // indirect | ||
github.com/bytedance/sonic/loader v0.1.1 // indirect | ||
github.com/clbanning/mxj/v2 v2.7.0 // indirect | ||
github.com/cloudwego/base64x v0.1.4 // indirect | ||
github.com/cloudwego/iasm v0.2.0 // indirect | ||
github.com/fatih/color v1.16.0 // indirect | ||
github.com/fsnotify/fsnotify v1.7.0 // indirect | ||
github.com/go-logr/logr v1.2.3 // indirect | ||
github.com/go-logr/stdr v1.2.2 // indirect | ||
github.com/gorilla/websocket v1.5.1 // indirect | ||
github.com/grokify/html-strip-tags-go v0.1.0 // indirect | ||
github.com/klauspost/cpuid/v2 v2.2.7 // indirect | ||
github.com/magiconair/properties v1.8.7 // indirect | ||
github.com/mattn/go-colorable v0.1.13 // indirect | ||
github.com/mattn/go-isatty v0.0.20 // indirect | ||
github.com/mattn/go-runewidth v0.0.9 // indirect | ||
github.com/olekukonko/tablewriter v0.0.5 // indirect | ||
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect | ||
go.opentelemetry.io/otel v1.14.0 // indirect | ||
go.opentelemetry.io/otel/sdk v1.14.0 // indirect | ||
go.opentelemetry.io/otel/trace v1.14.0 // indirect | ||
golang.org/x/arch v0.8.0 // indirect | ||
golang.org/x/net v0.24.0 // indirect | ||
golang.org/x/sys v0.20.0 // indirect | ||
golang.org/x/text v0.15.0 // indirect | ||
gopkg.in/yaml.v3 v3.0.1 // indirect | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,87 @@ | ||
github.com/BurntSushi/toml v1.3.2 h1:o7IhLm0Msx3BaB+n3Ag7L8EVlByGnpq14C4YWiu/gL8= | ||
github.com/BurntSushi/toml v1.3.2/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ= | ||
github.com/bytedance/sonic v1.11.7 h1:k/l9p1hZpNIMJSk37wL9ltkcpqLfIho1vYthi4xT2t4= | ||
github.com/bytedance/sonic v1.11.7/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4= | ||
github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM= | ||
github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= | ||
github.com/clbanning/mxj/v2 v2.7.0 h1:WA/La7UGCanFe5NpHF0Q3DNtnCsVoxbPKuyBNHWRyME= | ||
github.com/clbanning/mxj/v2 v2.7.0/go.mod h1:hNiWqW14h+kc+MdF9C6/YoRfjEJoR3ou6tn/Qo+ve2s= | ||
github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y= | ||
github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= | ||
github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg= | ||
github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= | ||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | ||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||
github.com/fatih/color v1.16.0 h1:zmkK9Ngbjj+K0yRhTVONQh1p/HknKYSlNT+vZCzyokM= | ||
github.com/fatih/color v1.16.0/go.mod h1:fL2Sau1YI5c0pdGEVCbKQbLXB6edEj1ZgiY4NijnWvE= | ||
github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA= | ||
github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM= | ||
github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= | ||
github.com/go-logr/logr v1.2.3 h1:2DntVwHkVopvECVRSlL5PSo9eG+cAkDCuckLubN+rq0= | ||
github.com/go-logr/logr v1.2.3/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= | ||
github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= | ||
github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= | ||
github.com/gogf/gf/v2 v2.7.1 h1:Ukp7vzwh6VKnivEEx/xiMc61dL1HVZqCCHl//3GBRxc= | ||
github.com/gogf/gf/v2 v2.7.1/go.mod h1:3oyGjyLHtSSo8kQ57Nj1TPdUNc0e2HS0A2J+KkXoW+I= | ||
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38= | ||
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | ||
github.com/gorilla/websocket v1.5.1 h1:gmztn0JnHVt9JZquRuzLw3g4wouNVzKL15iLr/zn/QY= | ||
github.com/gorilla/websocket v1.5.1/go.mod h1:x3kM2JMyaluk02fnUJpQuwD2dCS5NDG2ZHL0uE0tcaY= | ||
github.com/grokify/html-strip-tags-go v0.1.0 h1:03UrQLjAny8xci+R+qjCce/MYnpNXCtgzltlQbOBae4= | ||
github.com/grokify/html-strip-tags-go v0.1.0/go.mod h1:ZdzgfHEzAfz9X6Xe5eBLVblWIxXfYSQ40S/VKrAOGpc= | ||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= | ||
github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM= | ||
github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= | ||
github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= | ||
github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY= | ||
github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0= | ||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= | ||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= | ||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= | ||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= | ||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= | ||
github.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0= | ||
github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI= | ||
github.com/olekukonko/tablewriter v0.0.5 h1:P2Ga83D34wi1o9J6Wh1mRuqd4mF/x/lgBS7N7AbDhec= | ||
github.com/olekukonko/tablewriter v0.0.5/go.mod h1:hPp6KlRPjbx+hW8ykQs1w3UBbZlj6HuIJcUGPhkA7kY= | ||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= | ||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= | ||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= | ||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= | ||
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= | ||
github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8= | ||
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= | ||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= | ||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= | ||
go.opentelemetry.io/otel v1.14.0 h1:/79Huy8wbf5DnIPhemGB+zEPVwnN6fuQybr/SRXa6hM= | ||
go.opentelemetry.io/otel v1.14.0/go.mod h1:o4buv+dJzx8rohcUeRmWUZhqupFvzWis188WlggnNeU= | ||
go.opentelemetry.io/otel/sdk v1.14.0 h1:PDCppFRDq8A1jL9v6KMI6dYesaq+DFcDZvjsoGvxGzY= | ||
go.opentelemetry.io/otel/sdk v1.14.0/go.mod h1:bwIC5TjrNG6QDCHNWvW4HLHtUQ4I+VQDsnjhvyZCALM= | ||
go.opentelemetry.io/otel/trace v1.14.0 h1:wp2Mmvj41tDsyAJXiWDWpfNsOiIyd38fy85pyKcFq/M= | ||
go.opentelemetry.io/otel/trace v1.14.0/go.mod h1:8avnQLK+CG77yNLUae4ea2JDQ6iT+gozhnZjy/rw9G8= | ||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= | ||
golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc= | ||
golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys= | ||
golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI= | ||
golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8= | ||
golang.org/x/net v0.24.0 h1:1PcaxkF854Fu3+lvBIx5SYn9wRlBzzcnHZSiaFFAb0w= | ||
golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8= | ||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||
golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y= | ||
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||
golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk= | ||
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= | ||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= | ||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= | ||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||
nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= | ||
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,49 @@ | ||
package hashing | ||
|
||
import ( | ||
bcrypt "golang.org/x/crypto/bcrypt" | ||
) | ||
|
||
type Hashing struct { | ||
Cost int | ||
} | ||
|
||
func NewHashing(cost ...int) *Hashing { | ||
_cost := bcrypt.DefaultCost | ||
if len(cost) > 0 { | ||
_cost = cost[0] | ||
} | ||
return &Hashing{Cost: _cost} | ||
} | ||
|
||
func (receiver *Hashing) Make(pwd []byte, cost ...int) (string, error) { | ||
_cost := receiver.Cost | ||
if len(cost) > 0 { | ||
_cost = cost[0] | ||
} | ||
password, err := bcrypt.GenerateFromPassword(pwd, _cost) | ||
if err != nil { | ||
return "", err | ||
} | ||
return string(password), nil | ||
} | ||
func (receiver *Hashing) MustMake(pwd []byte, cost ...int) string { | ||
s, err := receiver.Make(pwd, cost...) | ||
if err != nil { | ||
panic(err) | ||
} | ||
return s | ||
|
||
} | ||
func (receiver *Hashing) MakeStr(pwd string) (string, error) { | ||
return receiver.Make([]byte(pwd)) | ||
} | ||
func (receiver *Hashing) MustMakeStr(pwd string) string { | ||
return receiver.MustMake([]byte(pwd)) | ||
} | ||
func (receiver *Hashing) Check(hashed, clearText []byte) bool { | ||
return bcrypt.CompareHashAndPassword(hashed, clearText) == nil | ||
} | ||
func (receiver *Hashing) CheckStr(hashed, clearText string) bool { | ||
return receiver.Check([]byte(hashed), []byte(clearText)) | ||
} |
Oops, something went wrong.