add Secret::set for c api

1 files changed, 17 insertions, 0 deletions

diff --git a/src/bls.cpp b/src/bls.cpp
index 03f7d16..3dfab45 100644
--- a/src/bls.cpp
+++ b/src/bls.cpp
@@ -87,6 +87,18 @@ struct Wrap {
 	size_t size() const { return pv->size(); }
 };
 
+template<class T, class G>
+struct WrapPointer {
+	const T *const *v;
+	size_t k;
+	WrapPointer(const T *const *v, size_t k) : v(v), k(k) {}
+	const G& operator[](size_t i) const
+	{
+		return v[i]->self_->get();
+	}
+	size_t size() const { return k; }
+};
+
 struct Polynomial {
 	FrVec c; // f[x] = sum_{i=0}^{k-1} c[i] x^i
 	void init(const Fr& s, int k)
@@ -426,6 +438,11 @@ void SecretKey::set(const SecretKeyVec& msk, const Id& id)
 	Wrap<SecretKey, Fr> w(msk);
 	evalPoly(self_->s, id.self_->v, w);
 }
+void SecretKey::set(const SecretKey *const *msk, size_t k, const Id& id)
+{
+	WrapPointer<SecretKey, Fr> w(msk, k);
+	evalPoly(self_->s, id.self_->v, w);
+}
 
 void SecretKey::recover(const SecretKeyVec& secVec, const IdVec& idVec)
 {