certusone
/
dc4bc
mirror of https://github.com/certusone/dc4bc.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

WIP

This commit is contained in:
programmer10110 2020-11-02 12:56:40 +03:00
parent e27c67b86a
commit a928ccc79f
8 changed files with 42 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
airgapped/bls.go
View File

@ -3,7 +3,6 @@ package airgapped
import (
"encoding/json"
"fmt"
"github.com/corestario/kyber/pairing"
"github.com/corestario/kyber/sign/bls"
@ -111,6 +110,7 @@ func (am *Machine) reconstructThresholdSignature(o *client.Operation) error {
Data: payload.SrcPayload,
Signature: reconstructedSignature,
DKGRoundID: o.DKGIdentifier,
SigningID: payload.SigningId,
}
respBz, err := json.Marshal(response)
if err != nil {

4
client/client.go
View File

@ -294,8 +294,8 @@ func (c *BaseClient) GetSignatures(dkgID string) (map[string][]types.Reconstruct
}
//GetSignatureByDataHash returns a list of reconstructed signatures of the signed data broadcasted by users
func (c *BaseClient) GetSignatureByDataHash(dkgID, sigID string) ([]types.ReconstructedSignature, error) {
return c.state.GetSignatureByDataHash(dkgID, sigID)
func (c *BaseClient) GetSignatureByID(dkgID, sigID string) ([]types.ReconstructedSignature, error) {
return c.state.GetSignatureByID(dkgID, sigID)
}
// getOperationJSON returns a specific JSON-encoded operation

2
client/flow_test.go
View File

@ -143,7 +143,7 @@ func RemoveContents(dir, mask string) error {
func TestFullFlow(t *testing.T) {
_ = RemoveContents("/tmp", "dc4bc_*")
defer func() { _ = RemoveContents("/tmp", "dc4bc_*") }()
//defer func() { _ = RemoveContents("/tmp", "dc4bc_*") }()
var numNodes = 4
var threshold = 2

6
client/http_server.go
View File

@ -70,7 +70,7 @@ func (c *BaseClient) StartHTTPServer(listenAddr string) error {
mux.HandleFunc("/getOperationQRPath", c.getOperationQRPathHandler)
mux.HandleFunc("/getSignatures", c.getSignaturesHandler)
mux.HandleFunc("/getSignatureByDataHash", c.getSignatureByDataHashHandler)
mux.HandleFunc("/getSignatureByID", c.getSignatureByIDHandler)
mux.HandleFunc("/getOperationQR", c.getOperationQRToBodyHandler)
mux.HandleFunc("/handleProcessedOperationJSON", c.handleJSONOperationHandler)
@ -237,13 +237,13 @@ func (c *BaseClient) getSignaturesHandler(w http.ResponseWriter, r *http.Request
successResponse(w, signatures)
}
func (c *BaseClient) getSignatureByDataHashHandler(w http.ResponseWriter, r *http.Request) {
func (c *BaseClient) getSignatureByIDHandler(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
errorResponse(w, http.StatusBadRequest, "Wrong HTTP method")
return
}
signature, err := c.GetSignatureByDataHash(r.URL.Query().Get("dkgID"), r.URL.Query().Get("hash"))
signature, err := c.GetSignatureByID(r.URL.Query().Get("dkgID"), r.URL.Query().Get("hash"))
if err != nil {
errorResponse(w, http.StatusInternalServerError, fmt.Sprintf("failed to get signature: %v", err))
return

13
client/state.go
View File

@ -1,9 +1,7 @@
package client
import (
"crypto/md5"
"encoding/binary"
"encoding/hex"
"encoding/json"
"errors"
"fmt"
@ -39,7 +37,7 @@ type State interface {
GetOperationByID(operationID string) (*types.Operation, error)
SaveSignature(signature types.ReconstructedSignature) error
GetSignatureByDataHash(dkgID, signatureID string) ([]types.ReconstructedSignature, error)
GetSignatureByID(dkgID, signatureID string) ([]types.ReconstructedSignature, error)
GetSignatures(dkgID string) (map[string][]types.ReconstructedSignature, error)
}
@ -311,7 +309,7 @@ func (s *LevelDBState) GetSignatures(dkgID string) (map[string][]types.Reconstru
return s.getSignatures(dkgID)
}
func (s *LevelDBState) GetSignatureByDataHash(dkgID, signatureID string) ([]types.ReconstructedSignature, error) {
func (s *LevelDBState) GetSignatureByID(dkgID, signatureID string) ([]types.ReconstructedSignature, error) {
s.Lock()
defer s.Unlock()
@ -340,12 +338,9 @@ func (s *LevelDBState) SaveSignature(signature types.ReconstructedSignature) err
signatures = make(map[string][]types.ReconstructedSignature)
}
dataHash := md5.Sum(signature.Data)
dataHashString := hex.EncodeToString(dataHash[:])
sig := signatures[dataHashString]
sig := signatures[signature.SigningID]
sig = append(sig, signature)
signatures[dataHashString] = sig
signatures[signature.SigningID] = sig
signaturesJSON, err := json.Marshal(signatures)
if err != nil {

1
client/types/types.go
View File

@ -23,6 +23,7 @@ const (
)
type ReconstructedSignature struct {
SigningID string
Data []byte
Signature []byte
Participant string

29
cmd/dc4bc_cli/main.go
View File

@ -63,6 +63,7 @@ func main() {
getOffsetCommand(),
getFSMStatusCommand(),
getFSMListCommand(),
getSignatureDataCommand(),
)
if err := rootCmd.Execute(); err != nil {
log.Fatalf("Failed to execute root command: %v", err)
@ -121,6 +122,7 @@ func getOperationsCommand() *cobra.Command {
}
msgHash := md5.Sum(payload.SrcPayload)
fmt.Printf("Hash of the message to sign - %s\n", hex.EncodeToString(msgHash[:]))
fmt.Printf("Signing ID: %s", payload.SigningId)
}
fmt.Println("-----------------------------------------------------")
}
@ -198,7 +200,7 @@ func getSignatureRequest(host string, dkgID, dataHash string) (*SignatureRespons
func getSignatureCommand() *cobra.Command {
return &cobra.Command{
Use: "get_signature [dkgID] [hash_of_the_signed_data]",
Use: "get_signature [dkgID] [signing_id]",
Args: cobra.ExactArgs(2),
Short: "returns a list of reconstructed signatures of the signed data broadcasted by users",
RunE: func(cmd *cobra.Command, args []string) error {
@ -223,6 +225,31 @@ func getSignatureCommand() *cobra.Command {
}
}
func getSignatureDataCommand() *cobra.Command {
return &cobra.Command{
Use: "get_signature_data [dkgID] [signing_id]",
Args: cobra.ExactArgs(2),
Short: "returns a data which was signed",
RunE: func(cmd *cobra.Command, args []string) error {
listenAddr, err := cmd.Flags().GetString(flagListenAddr)
if err != nil {
return fmt.Errorf("failed to read configuration: %v", err)
}
signatures, err := getSignatureRequest(listenAddr, args[0], args[1])
if err != nil {
return fmt.Errorf("failed to get signatures: %w", err)
}
if signatures.ErrorMessage != "" {
return fmt.Errorf("failed to get signatures: %s", signatures.ErrorMessage)
}
if len(signatures.Result) > 0 {
fmt.Println(string(signatures.Result[0].Data))
}
return nil
},
}
}
func getOperationRequest(host string, operationID string) (*OperationResponse, error) {
resp, err := http.Get(fmt.Sprintf("http://%s/getOperation?operationID=%s", host, operationID))
if err != nil {

3
cmd/dc4bc_d/main.go
View File

@ -182,7 +182,8 @@ func startClientCommand() *cobra.Command {
log.Fatalf("Failed to init state client: %v", err)
}
stg, err := storage.NewKafkaStorage(ctx, cfg.StorageDBDSN, cfg.StorageTopic)
stg, err := storage.NewFileStorage("/tmp/dc4bc_storage")
//stg, err := storage.NewKafkaStorage(ctx, cfg.StorageDBDSN, cfg.StorageTopic)
if err != nil {
log.Fatalf("Failed to init storage client: %v", err)
}