Reveal: add error to return signature, for upcomming Y calculation errors

2 files changed, 6 insertions, 6 deletions

diff --git a/nizk/stage1.go b/nizk/stage1.go
index fe75afd..9e87846 100644
--- a/nizk/stage1.go
+++ b/nizk/stage1.go
@@ -68,7 +68,7 @@ func (s *Stage) commit(lost bool, Xs ...*Point) *StageCommitment {
 	return s.com
 }
 
-func (s *Stage) RevealStage1(Xs ...*Point) (rev *StageReveal, pr *Stage1Proof) {
+func (s *Stage) RevealStage1(Xs ...*Point) (rev *StageReveal, pr *Stage1Proof, e error) {
 	var ε [2][4]*Point
 	var r1, r2, ρ1, ρ2, ω *Scalar
 	for _, s := range []**Scalar{&r1, &r2, &ρ1, &ρ2, &ω} {
@@ -139,7 +139,7 @@ func (s *Stage) RevealStage1(Xs ...*Point) (rev *StageReveal, pr *Stage1Proof) {
 
 	s.rev = rev
 	s.prf1 = pr
-	return rev, pr
+	return rev, pr, e
 }
 
 func (c *Commitment) VerifyStage1(sc *StageCommitment, r *StageReveal, p *Stage1Proof) bool {
diff --git a/nizk/stage1_test.go b/nizk/stage1_test.go
index 8fc8970..d4e68bf 100644
--- a/nizk/stage1_test.go
+++ b/nizk/stage1_test.go
@@ -15,8 +15,8 @@ func TestStage1Simple(t *testing.T) {
 
 	s1, c1 := b1.CommitStage1()
 	s2, c2 := b2.CommitStage1()
-	r1, pr1 := s1.RevealStage1() // Note: no Xs.
-	r2, pr2 := s2.RevealStage1() // Note: no Xs
+	r1, pr1, _ := s1.RevealStage1() // Note: no Xs.
+	r2, pr2, _ := s2.RevealStage1() // Note: no Xs.
 	if !bc1.VerifyStage1(c1, r1, pr1) {
 		t.Fatal("Could not verify st1 with c1 and pr1, plus=true case")
 	}
@@ -42,8 +42,8 @@ func TestStage1FromScalars(t *testing.T) {
 
 	s1, c1 := b1.CommitStage1FromScalars(r, x)
 	s2, c2 := b2.CommitStage1FromScalars(x, r)
-	r1, pr1 := s1.RevealStage1() // Note: no Xs
-	r2, pr2 := s2.RevealStage1() // Note: no Xs
+	r1, pr1, _ := s1.RevealStage1() // Note: no Xs
+	r2, pr2, _ := s2.RevealStage1() // Note: no Xs
 	if !bc1.VerifyStage1(c1, r1, pr1) {
 		t.Fatal("Could not verify st1 with c1 and pr1, plus=true case")
 	}