forked from TrueCloudLab/frostfs-node
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package object
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
"strings"
|
|
|
|
internalclient "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/client"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key"
|
|
commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/checksum"
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const getRangeHashSaltFlag = "salt"
|
|
|
|
const (
|
|
hashSha256 = "sha256"
|
|
hashTz = "tz"
|
|
rangeSep = ":"
|
|
)
|
|
|
|
var objectHashCmd = &cobra.Command{
|
|
Use: "hash",
|
|
Short: "Get object hash",
|
|
Long: "Get object hash",
|
|
Run: getObjectHash,
|
|
}
|
|
|
|
func initObjectHashCmd() {
|
|
commonflags.Init(objectHashCmd)
|
|
initFlagSession(objectHashCmd, "RANGEHASH")
|
|
|
|
flags := objectHashCmd.Flags()
|
|
|
|
flags.String(commonflags.CIDFlag, "", commonflags.CIDFlagUsage)
|
|
_ = objectHashCmd.MarkFlagRequired(commonflags.CIDFlag)
|
|
|
|
flags.String(commonflags.OIDFlag, "", commonflags.OIDFlagUsage)
|
|
_ = objectHashCmd.MarkFlagRequired(commonflags.OIDFlag)
|
|
|
|
flags.String("range", "", "Range to take hash from in the form offset1:length1,...")
|
|
flags.String("type", hashSha256, "Hash type. Either 'sha256' or 'tz'")
|
|
flags.String(getRangeHashSaltFlag, "", "Salt in hex format")
|
|
}
|
|
|
|
func getObjectHash(cmd *cobra.Command, _ []string) {
|
|
var cnr cid.ID
|
|
var obj oid.ID
|
|
|
|
objAddr := readObjectAddress(cmd, &cnr, &obj)
|
|
|
|
ranges, err := getRangeList(cmd)
|
|
commonCmd.ExitOnErr(cmd, "", err)
|
|
typ, err := getHashType(cmd)
|
|
commonCmd.ExitOnErr(cmd, "", err)
|
|
|
|
strSalt := strings.TrimPrefix(cmd.Flag(getRangeHashSaltFlag).Value.String(), "0x")
|
|
|
|
salt, err := hex.DecodeString(strSalt)
|
|
commonCmd.ExitOnErr(cmd, "could not decode salt: %w", err)
|
|
|
|
pk := key.GetOrGenerate(cmd)
|
|
cli := internalclient.GetSDKClientByFlag(cmd, pk, commonflags.RPC)
|
|
|
|
tz := typ == hashTz
|
|
fullHash := len(ranges) == 0
|
|
if fullHash {
|
|
var headPrm internalclient.HeadObjectPrm
|
|
headPrm.SetClient(cli)
|
|
Prepare(cmd, &headPrm)
|
|
headPrm.SetAddress(objAddr)
|
|
|
|
// get hash of full payload through HEAD (may be user can do it through dedicated command?)
|
|
res, err := internalclient.HeadObject(cmd.Context(), headPrm)
|
|
commonCmd.ExitOnErr(cmd, "rpc error: %w", err)
|
|
|
|
var cs checksum.Checksum
|
|
var csSet bool
|
|
|
|
if tz {
|
|
cs, csSet = res.Header().PayloadHomomorphicHash()
|
|
} else {
|
|
cs, csSet = res.Header().PayloadChecksum()
|
|
}
|
|
|
|
if csSet {
|
|
cmd.Println(hex.EncodeToString(cs.Value()))
|
|
} else {
|
|
cmd.Println("Missing checksum in object header.")
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
var hashPrm internalclient.HashPayloadRangesPrm
|
|
hashPrm.SetClient(cli)
|
|
Prepare(cmd, &hashPrm)
|
|
readSession(cmd, &hashPrm, pk, cnr, obj)
|
|
hashPrm.SetAddress(objAddr)
|
|
hashPrm.SetSalt(salt)
|
|
hashPrm.SetRanges(ranges)
|
|
|
|
if tz {
|
|
hashPrm.TZ()
|
|
}
|
|
|
|
res, err := internalclient.HashPayloadRanges(cmd.Context(), hashPrm)
|
|
commonCmd.ExitOnErr(cmd, "rpc error: %w", err)
|
|
|
|
hs := res.HashList()
|
|
|
|
for i := range hs {
|
|
cmd.Printf("Offset=%d (Length=%d)\t: %s\n", ranges[i].GetOffset(), ranges[i].GetLength(),
|
|
hex.EncodeToString(hs[i]))
|
|
}
|
|
}
|
|
|
|
func getHashType(cmd *cobra.Command) (string, error) {
|
|
rawType := cmd.Flag("type").Value.String()
|
|
switch typ := strings.ToLower(rawType); typ {
|
|
case hashSha256, hashTz:
|
|
return typ, nil
|
|
default:
|
|
return "", fmt.Errorf("invalid hash type: %s", typ)
|
|
}
|
|
}
|