Set the value of Triple to $10^{27}$

packages/client/libclient-py/quickmpc/proto/common_types/common_types_pb2.pyi

@@ -220,3 +220,21 @@ class Schema(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "type", b"type"]) -> None: ...
 
 global___Schema = Schema
+
+@typing_extensions.final
+class BigIntByte(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SGN_FIELD_NUMBER: builtins.int
+    ABS_BYTE_FIELD_NUMBER: builtins.int
+    sgn: builtins.bool
+    abs_byte: builtins.bytes
+    def __init__(
+        self,
+        *,
+        sgn: builtins.bool = ...,
+        abs_byte: builtins.bytes = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["abs_byte", b"abs_byte", "sgn", b"sgn"]) -> None: ...
+
+global___BigIntByte = BigIntByte

packages/client/libclient-py/quickmpc/share/share.py

@@ -1,5 +1,8 @@
+import inspect
+import threading
 from dataclasses import dataclass
-from decimal import Decimal
+from decimal import Decimal, getcontext
+from functools import wraps
 from typing import (Any, Callable, ClassVar, List,
                     Optional, Sequence, Tuple, Union)
 
@@ -16,6 +19,47 @@
 logger = get_logger()
 
 
+# getcontext().precがthreadごとに初期化されてしまうため，
+# theread毎に初期化し返せるようできるようにフラグを管理する
+_local_dict = threading.local()
+# Triple cのshareが23byteで約10^{54}なので, 余裕を持って収まるように100にしておく
+_decimal_prec = 100
+
+
+def set_decimal_prec():
+    try:
+        _local_dict.is_initialized
+    except Exception:
+        getcontext().prec = _decimal_prec
+        _local_dict.is_initialized = True
+
+
+set_decimal_prec()
+
+
+def _verify_decimal_prec(f):
+    """メソッドにprecのverifyを適用させるデコレータ"""
+    @wraps(f)
+    def _wrapper(*args, **kw):
+        set_decimal_prec()
+        if getcontext().prec != _decimal_prec:
+            raise RuntimeError(
+                f"Decimal context prec is must be {_decimal_prec}.")
+        return f(*args, **kw)
+    return _wrapper
+
+
+def _verify_decimal_prec_all(cls):
+    """クラスの全てのメソッドにprecのverifyを適用させるデコレータ"""
+    for name, fn in inspect.getmembers(cls):
+        if name.startswith('__'):
+            continue
+        if callable(getattr(cls, name)):
+            setattr(cls, name, _verify_decimal_prec(fn))
+    return cls
+
+
+@_verify_decimal_prec_all
 @dataclass(frozen=True)
 class Share:
     __share_random_range: ClassVar[Tuple[Decimal, Decimal]] =\

packages/client/libclient-py/tests/unit_tests/share/test_share_recons.py

@@ -1,11 +1,11 @@
 import math
-from decimal import Decimal
+from decimal import Decimal, getcontext
 
 import numpy as np
 import pytest
 
 from quickmpc.exception import ArgumentError
-from quickmpc.share.share import Share
+from quickmpc.share.share import _decimal_prec, Share
 
 
 class TestQMPC:
@@ -174,3 +174,23 @@ def test_sharize_recons(self, secrets: list):
             secrets_2: list = Share.recons(conved)
             secrets_2 = Share.convert_type(secrets_2)
             assert (np.allclose(secrets, secrets_2))
+
+    @pytest.mark.parametrize(
+        ("secrets", "prec"),
+        [
+            (1, 28),
+            (1, 280),
+            ([1], 28),
+            ([1], 280),
+            ([[1]], 28),
+            ([[1]], 280),
+        ]
+    )
+    def test_recons_decimal_prec_error(self, secrets, prec):
+        """ 異なるprecが指定されているときにエラーが出るかTest """
+        getcontext().prec = prec
+        with pytest.raises(RuntimeError):
+            Share.recons(secrets)
+
+        # Global値をいじっているので元の値に戻しておく
+        getcontext().prec = _decimal_prec

packages/client/libclient-py/tests/unit_tests/share/test_share_sharize.py

@@ -1,12 +1,12 @@
 import math
-from decimal import Decimal
+from decimal import Decimal, getcontext
 from typing import List
 
 import numpy as np
 import pytest
 
 from quickmpc.exception import ArgumentError
-from quickmpc.share.share import Share
+from quickmpc.share.share import _decimal_prec, Share
 
 
 def sharize_params(secrets=[1], party_size=3):
@@ -154,3 +154,23 @@ def test_sharize_errorhandring(self, args, expected_exception):
         """ 異常値を与えてエラーが出るかTest """
         with pytest.raises(expected_exception):
             Share.sharize(*args)
+
+    @pytest.mark.parametrize(
+        ("args", "prec"),
+        [
+            (sharize_params(secrets=1), 28),
+            (sharize_params(secrets=1), 280),
+            (sharize_params(secrets=[1]), 28),
+            (sharize_params(secrets=[1]), 280),
+            (sharize_params(secrets=[[1]]), 28),
+            (sharize_params(secrets=[[1]]), 280),
+        ]
+    )
+    def test_sharize_decimal_prec_error(self, args, prec):
+        """ 異なるprecが指定されているときにエラーが出るかTest """
+        getcontext().prec = prec
+        with pytest.raises(RuntimeError):
+            Share.sharize(*args)
+
+        # Global値をいじっているので元の値に戻しておく
+        getcontext().prec = _decimal_prec

packages/server/beaver_triple_service/go.mod

@@ -2,9 +2,13 @@ module github.com/acompany-develop/QuickMPC/packages/server/beaver_triple_servic
 
 go 1.18
 
-replace github.com/acompany-develop/QuickMPC/proto/engine_to_bts => ./../proto/engine_to_bts
+replace (
+	github.com/acompany-develop/QuickMPC/proto/common_types => ./../proto/common_types
+	github.com/acompany-develop/QuickMPC/proto/engine_to_bts => ./../proto/engine_to_bts
+)
 
 require (
+	github.com/acompany-develop/QuickMPC/proto/common_types v0.0.0-00010101000000-000000000000
 	github.com/acompany-develop/QuickMPC/proto/engine_to_bts v0.0.0-00010101000000-000000000000
 	github.com/golang-jwt/jwt/v4 v4.4.2
 	github.com/google/uuid v1.3.0

packages/server/beaver_triple_service/server/engine_to_bts/e2b.go

@@ -61,7 +61,7 @@ func (s *server) GetTriples(ctx context.Context, in *pb.GetRequest) (*pb.GetTrip
 	if err != nil {
 		return nil, err
 	}
-	logger.Infof("jobId: %d, partyId: %d\n", in.GetJobId(), partyId,)
+	logger.Infof("jobId: %d, partyId: %d\n", in.GetJobId(), partyId)
 
 	triples, err := tg.GetTriples(claims, in.GetJobId(), partyId, in.GetAmount(), in.GetRequestId())
 	if err != nil {

packages/server/beaver_triple_service/server/engine_to_bts/e2b_test.go

@@ -3,14 +3,16 @@ package e2bserver
 import (
 	"context"
 	"fmt"
+	"math/big"
 	"os"
 	"sync"
 	"testing"
 
 	jwt_types "github.com/acompany-develop/QuickMPC/packages/server/beaver_triple_service/jwt"
-	ts "github.com/acompany-develop/QuickMPC/packages/server/beaver_triple_service/triple_store"
 	rbs "github.com/acompany-develop/QuickMPC/packages/server/beaver_triple_service/rand_bit_store"
+	ts "github.com/acompany-develop/QuickMPC/packages/server/beaver_triple_service/triple_store"
 	utils "github.com/acompany-develop/QuickMPC/packages/server/beaver_triple_service/utils"
+	pb_types "github.com/acompany-develop/QuickMPC/proto/common_types"
 	pb "github.com/acompany-develop/QuickMPC/proto/engine_to_bts"
 
 	"google.golang.org/grpc/metadata"
@@ -133,16 +135,28 @@ func testGetTriplesByJobId(t *testing.T, client pb.EngineToBtsClient, amount uin
 	})
 }
 
+func convertToBigInt(b *pb_types.BigIntByte) *big.Int {
+	var ret *big.Int = big.NewInt(0)
+	bytes := b.AbsByte
+	ret.SetBytes(bytes)
+	if b.Sgn {
+		ret.Neg(ret)
+	}
+	return ret
+}
+
 func testValidityOfTriples(t *testing.T) {
 	for _, triples := range DbTripleTest.Triples {
 		for i := 0; i < len(triples[1]); i++ {
-			aShareSum, bShareSum, cShareSum := int64(0), int64(0), int64(0)
+			aShareSum, bShareSum, cShareSum := big.NewInt(0), big.NewInt(0), big.NewInt(0)
 			for partyId := uint32(1); partyId <= uint32(len(triples)); partyId++ {
-				aShareSum += triples[partyId][i].A
-				bShareSum += triples[partyId][i].B
-				cShareSum += triples[partyId][i].C
+				aShareSum.Add(aShareSum, convertToBigInt(triples[partyId][i].A))
+				bShareSum.Add(bShareSum, convertToBigInt(triples[partyId][i].B))
+				cShareSum.Add(cShareSum, convertToBigInt(triples[partyId][i].C))
 			}
-			if aShareSum*bShareSum != cShareSum {
+			ab := big.NewInt(0)
+			ab.Mul(aShareSum, bShareSum)
+			if ab.Cmp(cShareSum) != 0 {
 				t.Fatal("a*b != c")
 			}
 		}
@@ -263,8 +277,10 @@ func TestGetRandBits_1_10000(t *testing.T) { testGetRandBits(t, 1, 10000) }
 
 func TestGetRandBits_100_1(t *testing.T)   { testGetRandBits(t, 100, 1) }
 func TestGetRandBits_100_100(t *testing.T) { testGetRandBits(t, 100, 100) }
+
 // func TestGetRandBits_100_10000(t *testing.T) { testGetRandBits(t, 100, 10000) }
 
 func TestGetRandBits_10000_1(t *testing.T)   { testGetRandBits(t, 10000, 1) }
 func TestGetRandBits_10000_100(t *testing.T) { testGetRandBits(t, 10000, 100) }
-// func TestGetRandBits_10000_10000(t *testing.T) { testGetRandBits(t, 10000, 10000) }
+
+// func TestGetRandBits_10000_10000(t *testing.T) { testGetRandBits(t, 10000, 10000) }