From 8d5460926065473b15f2b2dc460ee268a01f65b9 Mon Sep 17 00:00:00 2001
From: MITSUNARI Shigeo <herumi@nifty.com>
Date: Mon, 13 May 2019 16:46:25 +0900
Subject: [PATCH] [she] add isZero

---
 ffi/python/she.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/ffi/python/she.py b/ffi/python/she.py
index 6e309fa..406501b 100644
--- a/ffi/python/she.py
+++ b/ffi/python/she.py
@@ -186,6 +186,14 @@ class SecretKey(Structure):
 		if ret != 0:
 			raise RuntimeError("dec")
 		return m.value
+	def isZero(self, c):
+		if isinstance(c, CipherTextG1):
+			return lib.sheIsZeroG1(byref(self.v), byref(c.v)) == 1
+		elif isinstance(c, CipherTextG2):
+			return lib.sheIsZeroG2(byref(self.v), byref(c.v)) == 1
+		elif isinstance(c, CipherTextGT):
+			return lib.sheIsZeroGT(byref(self.v), byref(c.v)) == 1
+		raise RuntimeError("dec")
 
 def neg(c):
 	ret = -1
@@ -307,6 +315,13 @@ if __name__ == '__main__':
 	# mul G1 and G2
 	if sec.dec(mul(c11, c21)) != m11 * m21: print("err8")
 
+	if not sec.isZero(pub.encG1(0)): print("err-zero11")
+	if sec.isZero(pub.encG1(3)): print("err-zero12")
+	if not sec.isZero(pub.encG2(0)): print("err-zero21")
+	if sec.isZero(pub.encG2(3)): print("err-zero22")
+	if not sec.isZero(pub.encGT(0)): print("err-zero31")
+	if sec.isZero(pub.encGT(3)): print("err-zero32")
+
 	# use precomputedPublicKey for performance
 	ppub = pub.createPrecomputedPublicKey()
 	c1 = ppub.encG1(m11)