aboutsummaryrefslogtreecommitdiff
path: root/cmd/sigsum-debug/leaf/inclusion/inclusion.go
blob: e70b3eb1234133be6bb13b270855b94a6f38041f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package inclusion

import (
	"bytes"
	"fmt"

	"git.sigsum.org/sigsum-go/internal/fmtio"
	"git.sigsum.org/sigsum-go/pkg/types"
)

func Main(args []string, optLeafHash, optRootHash string, optTreeSize uint64) error {
	if len(args) != 0 {
		return fmt.Errorf("trailing arguments: %v", args)
	}
	b, err := fmtio.BytesFromStdin()
	if err != nil {
		return fmt.Errorf("read: %w", err)
	}
	var proof types.InclusionProof
	if err := proof.FromASCII(bytes.NewBuffer(b), optTreeSize); err != nil {
		return fmt.Errorf("parse proof: %w", err)
	}
	leafHash, err := fmtio.HashFromHex(optLeafHash)
	if err != nil {
		return fmt.Errorf("parse leaf hash: %w", err)
	}
	rootHash, err := fmtio.HashFromHex(optRootHash)
	if err != nil {
		return fmt.Errorf("parse root hash: %w", err)
	}
	if err := proof.Verify(&leafHash, &rootHash); err != nil {
		return fmt.Errorf("verify: %w", err)
	}
	return nil
}