added state manager with tests

2 files changed, 9 insertions, 7 deletions

diff --git a/types/types.go b/types/types.go
index 405c825..9ca7db8 100644
--- a/types/types.go
+++ b/types/types.go
@@ -2,11 +2,10 @@ package types
 
 import (
 	"crypto"
-	"fmt"
-	"strings"
-
 	"crypto/ed25519"
 	"crypto/sha256"
+	"fmt"
+	"strings"
 )
 
 const (
@@ -138,7 +137,10 @@ func (th *TreeHead) Sign(signer crypto.Signer) (*SignedTreeHead, error) {
 }
 
 // Verify verifies the tree head signature using the log's signature scheme
-func (th *TreeHead) Verify(pub crypto.PublicKey) error { // TODO
+func (th *TreeHead) Verify(vk *[VerificationKeySize]byte, sig *[SignatureSize]byte) error {
+	if !ed25519.Verify(ed25519.PublicKey(vk[:]), th.Marshal(), sig[:]) {
+		return fmt.Errorf("invalid tree head signature")
+	}
 	return nil
 }
 
diff --git a/types/types_test.go b/types/types_test.go
index 22a03d4..da89c59 100644
--- a/types/types_test.go
+++ b/types/types_test.go
@@ -52,7 +52,7 @@ func TestEndpointPath(t *testing.T) {
 	}
 }
 
-func TestTreeHeadSign(t *testing.T) {}
-func TestTreeHeadVerify(t *testing.T) {}
-func TestInclusionProofVerify(t *testing.T) {}
+func TestTreeHeadSign(t *testing.T)           {}
+func TestTreeHeadVerify(t *testing.T)         {}
+func TestInclusionProofVerify(t *testing.T)   {}
 func TestConsistencyProofVerify(t *testing.T) {}