mirror of
https://github.com/wader/fq.git
synced 2024-11-30 09:58:13 +03:00
e3ae1440c9
Feels less cluttered, easier to read and more consistent. Still keep tovalue, tobytes etc that are more basic functions this only renamed format related functions. Also there is an exceptin for to/fromjson as it comes from jq. Also fixes lots of spelling errors while reading thru.
81 lines
1.3 KiB
Go
81 lines
1.3 KiB
Go
package crypto
|
|
|
|
import (
|
|
"crypto/md5"
|
|
//nolint: gosec
|
|
"crypto/sha1"
|
|
"crypto/sha256"
|
|
"crypto/sha512"
|
|
"embed"
|
|
"fmt"
|
|
"hash"
|
|
"io"
|
|
|
|
"github.com/wader/fq/pkg/bitio"
|
|
"github.com/wader/fq/pkg/interp"
|
|
|
|
//nolint: staticcheck
|
|
"golang.org/x/crypto/md4"
|
|
"golang.org/x/crypto/sha3"
|
|
)
|
|
|
|
//go:embed hash.jq
|
|
var hashFS embed.FS
|
|
|
|
func init() {
|
|
interp.RegisterFunc1("_to_hash", toHash)
|
|
interp.RegisterFS(hashFS)
|
|
}
|
|
|
|
func hashFn(s string) hash.Hash {
|
|
switch s {
|
|
case "md4":
|
|
return md4.New()
|
|
case "md5":
|
|
return md5.New()
|
|
case "sha1":
|
|
return sha1.New()
|
|
case "sha256":
|
|
return sha256.New()
|
|
case "sha512":
|
|
return sha512.New()
|
|
case "sha3_224":
|
|
return sha3.New224()
|
|
case "sha3_256":
|
|
return sha3.New256()
|
|
case "sha3_384":
|
|
return sha3.New384()
|
|
case "sha3_512":
|
|
return sha3.New512()
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
type toHashOpts struct {
|
|
Name string
|
|
}
|
|
|
|
func toHash(_ *interp.Interp, c any, opts toHashOpts) any {
|
|
inBR, err := interp.ToBitReader(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
h := hashFn(opts.Name)
|
|
if h == nil {
|
|
return fmt.Errorf("unknown hash function %s", opts.Name)
|
|
}
|
|
if _, err := io.Copy(h, bitio.NewIOReader(inBR)); err != nil {
|
|
return err
|
|
}
|
|
|
|
outBR := bitio.NewBitReader(h.Sum(nil), -1)
|
|
|
|
bb, err := interp.NewBinaryFromBitReader(outBR, 8, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return bb
|
|
}
|