Minh Doan
6 years ago
commit
210f9f7ba9
@ -1,14 +1,100 @@ |
||||
#!/usr/bin/env bash |
||||
#!/bin/bash |
||||
# this script is used to generate the binary of benchmark/txgen |
||||
# TODO: add error and parameter checking |
||||
|
||||
declare -A SRC |
||||
SRC[benchmark]=benchmark.go |
||||
SRC[txgen]=client/txgen/main.go |
||||
|
||||
BINDIR=bin |
||||
BUCKET=unique-bucket-bin |
||||
GOOS=linux |
||||
GOARCH=amd64 |
||||
env GOOS=$GOOS GOARCH=$GOARCH go build -o bin/benchmark benchmark.go |
||||
env GOOS=$GOOS GOARCH=$GOARCH go build -o bin/txgen client/txgen/main.go |
||||
FOLDER= |
||||
|
||||
function usage |
||||
{ |
||||
ME=$(basename $0) |
||||
cat<<EOF |
||||
|
||||
Usage: $ME [OPTIONS] ACTION |
||||
|
||||
OPTIONS: |
||||
-h print this help message |
||||
-p profile aws profile name |
||||
-a arch set build arch (default: $GOARCH) |
||||
-o os set build OS (default: $GOOS, windows is supported) |
||||
-b bucket set the upload bucket name (default: $BUCKET) |
||||
-f folder set the upload folder name in the bucket (default: $FOLDER) |
||||
|
||||
ACTION: |
||||
build build binaries only (default action) |
||||
upload upload binaries to s3 |
||||
|
||||
EXAMPLES: |
||||
|
||||
# build linux binaries only by default |
||||
$ME |
||||
|
||||
# build windows binaries |
||||
$ME -o windows |
||||
|
||||
# upload binaries to my s3 bucket, 0908 folder |
||||
$ME -b mybucket -f 0908 upload |
||||
|
||||
EOF |
||||
exit 1 |
||||
} |
||||
|
||||
function build_only |
||||
{ |
||||
VERSION=$(git rev-list --all --count) |
||||
COMMIT=$(git describe --always --long --dirty) |
||||
BUILTAT=$(date +%FT%T%z) |
||||
BUILTBY=${USER}@ |
||||
|
||||
for bin in "${!SRC[@]}"; do |
||||
env GOOS=$GOOS GOARCH=$GOARCH go build -ldflags="-X main.version=v${VERSION} -X main.commit=${COMMIT} -X main.builtAt=${BUILTAT} -X main.builtBy=${BUILTBY}" -o $BINDIR/$bin ${SRC[$bin]} |
||||
$BINDIR/$bin -version |
||||
done |
||||
|
||||
md5sum $BINDIR/* > $BINDIR/md5sum.txt |
||||
} |
||||
|
||||
function upload |
||||
{ |
||||
AWSCLI=aws |
||||
|
||||
if [ -n "$PROFILE" ]; then |
||||
AWSCLI+=" --profile $PROFILE" |
||||
fi |
||||
|
||||
for bin in "${!SRC[@]}"; do |
||||
[ -e $BINDIR/$bin ] && $AWSCLI s3 cp $BINDIR/$bin s3://${BUCKET}$FOLDER/$bin --acl public-read |
||||
done |
||||
[ -e $BINDIR/md5sum.txt ] && $AWSCLI s3 cp $BINDIR/md5sum.txt s3://${BUCKET}$FOLDER/md5sum.txt --acl public-read |
||||
} |
||||
|
||||
################################ MAIN FUNCTION ############################## |
||||
while getopts "hp:a:o:b:f:" option; do |
||||
case $option in |
||||
h) usage ;; |
||||
p) PROFILE=$OPTARG ;; |
||||
a) GOARCH=$OPTARG ;; |
||||
o) GOOS=$OPTARG ;; |
||||
b) BUCKET=$OPTARG/ ;; |
||||
f) FOLDER=$OPTARG ;; |
||||
esac |
||||
done |
||||
|
||||
mkdir -p $BINDIR |
||||
|
||||
shift $(($OPTIND-1)) |
||||
|
||||
AWSCLI=aws |
||||
if [ "$1" != "" ]; then |
||||
AWSCLI+=" --profile $1" |
||||
fi |
||||
ACTION=${1:-build} |
||||
|
||||
$AWSCLI s3 cp bin/benchmark s3://unique-bucket-bin/benchmark --acl public-read-write |
||||
$AWSCLI s3 cp bin/txgen s3://unique-bucket-bin/txgen --acl public-read-write |
||||
case "$ACTION" in |
||||
"build") build_only ;; |
||||
"upload") upload ;; |
||||
*) usage ;; |
||||
esac |
||||
|
||||
@ -1,20 +0,0 @@ |
||||
package identitychain |
||||
|
||||
import ( |
||||
"fmt" |
||||
"os" |
||||
"testing" |
||||
|
||||
"github.com/simple-rules/harmony-benchmark/p2p" |
||||
) |
||||
|
||||
func TestIDCFormed(test *testing.T) { |
||||
peer := p2p.Peer{Ip: "127.0.0.1", Port: "8080"} |
||||
IDC := New(peer) |
||||
if IDC == nil { |
||||
fmt.Println("IDC not formed.") |
||||
os.Exit(1) |
||||
} |
||||
} |
||||
|
||||
//TODO Mock netconnection to test whether identitychain is listening.
|
||||
@ -1,43 +0,0 @@ |
||||
package identitymanage |
||||
|
||||
// import (
|
||||
// "bytes"
|
||||
// "encoding/binary"
|
||||
// "log"
|
||||
|
||||
// "github.com/dedis/kyber"
|
||||
// )
|
||||
|
||||
// // Consensus data containing all info related to one round of consensus process
|
||||
// type Identity struct {
|
||||
// priKey kyber.Scalar
|
||||
// pubKey kyber.Point
|
||||
// Log log.Logger
|
||||
// }
|
||||
|
||||
// // Construct the response message to send to leader (assumption the consensus data is already verified)
|
||||
// func (identity *Identity) registerIdentity(msgType proto_consensus.MessageType, response kyber.Scalar) []byte {
|
||||
// buffer := bytes.NewBuffer([]byte{})
|
||||
|
||||
// // 4 byte consensus id
|
||||
// fourBytes := make([]byte, 4)
|
||||
// binary.BigEndian.PutUint32(fourBytes, consensus.consensusId)
|
||||
// buffer.Write(fourBytes)
|
||||
|
||||
// // 32 byte block hash
|
||||
// buffer.Write(consensus.blockHash[:32])
|
||||
|
||||
// // 2 byte validator id
|
||||
// twoBytes := make([]byte, 2)
|
||||
// binary.BigEndian.PutUint16(twoBytes, consensus.nodeId)
|
||||
// buffer.Write(twoBytes)
|
||||
|
||||
// // 32 byte of response
|
||||
// response.MarshalTo(buffer)
|
||||
|
||||
// // 64 byte of signature on previous data
|
||||
// signature := consensus.signMessage(buffer.Bytes())
|
||||
// buffer.Write(signature)
|
||||
|
||||
// return proto_identity.ConstructIdentityMessage(msgType, buffer.Bytes())
|
||||
// }
|
||||
@ -0,0 +1,21 @@ |
||||
MIT License |
||||
|
||||
Copyright (c) 2018 Bas Westerbaan |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in all |
||||
copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
||||
SOFTWARE. |
||||
@ -0,0 +1,57 @@ |
||||
go-pow |
||||
====== |
||||
|
||||
`go-pow` is a simple Go package to add (asymmetric) *Proof of Work* to your service. |
||||
|
||||
To create a Proof-of-Work request (with difficulty 5), use `pow.NewRequest`: |
||||
|
||||
```go |
||||
req := pow.NewRequest(5, someRandomNonce) |
||||
``` |
||||
|
||||
This returns a string like `sha2bday-5-c29tZSByYW5kb20gbm9uY2U`, |
||||
which can be passed on to the client. |
||||
The client fulfils the proof of work by running `pow.Fulfil`: |
||||
|
||||
```go |
||||
proof, _ := pow.Fulfil(req, []byte("some bound data")) |
||||
``` |
||||
|
||||
The client returns the proof (in this case `AAAAAAAAAAMAAAAAAAAADgAAAAAAAAAb`) |
||||
to the server, which can check it is indeed a valid proof of work, by running: |
||||
|
||||
|
||||
``` go |
||||
ok, _ := pow.Check(req, proof, []byte("some bound data")) |
||||
``` |
||||
|
||||
Notes |
||||
----- |
||||
1. There should be at least sufficient randomness in either the `nonce` passed to |
||||
`NewRequest` or the `data` passed to `Fulfil` and `Check`. |
||||
Thus it is fine to use the same bound `data` for every client, if every client |
||||
get a different `nonce` in its proof-of-work request. |
||||
It is also fine to use the same `nonce` in the proof-of-work request, |
||||
if every client is (by the encapsulating protocol) forced to use |
||||
different bound `data`. |
||||
2. The work to fulfil a request scales exponentially in the difficulty parameter. |
||||
The work to check it proof is correct remains constant: |
||||
|
||||
``` |
||||
Check on Difficulty=5 500000 2544 ns/op |
||||
Check on Difficulty=10 500000 2561 ns/op |
||||
Check on Difficulty=15 500000 2549 ns/op |
||||
Check on Difficulty=20 500000 2525 ns/op |
||||
Fulfil on Difficulty=5 100000 15725 ns/op |
||||
Fulfil on Difficulty=10 30000 46808 ns/op |
||||
Fulfil on Difficulty=15 2000 955606 ns/op |
||||
Fulfil on Difficulty=20 200 6887722 ns/op |
||||
``` |
||||
|
||||
To do |
||||
----- |
||||
|
||||
- Support for [equihash](https://www.cryptolux.org/index.php/Equihash) would be nice. |
||||
- Port to Python, Java, Javascript, ... |
||||
- Parallelize. |
||||
|
||||
@ -0,0 +1,131 @@ |
||||
// Create and fulfill proof of work requests.
|
||||
package pow |
||||
|
||||
import ( |
||||
"encoding/base64" |
||||
"fmt" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
type Algorithm string |
||||
|
||||
const ( |
||||
Sha2BDay Algorithm = "sha2bday" |
||||
) |
||||
|
||||
// Represents a proof-of-work request.
|
||||
type Request struct { |
||||
|
||||
// The requested algorithm
|
||||
Alg Algorithm |
||||
|
||||
// The requested difficulty
|
||||
Difficulty uint32 |
||||
|
||||
// Nonce to diversify the request
|
||||
Nonce []byte |
||||
} |
||||
|
||||
// Represents a completed proof-of-work
|
||||
type Proof struct { |
||||
buf []byte |
||||
} |
||||
|
||||
// Convenience function to create a new sha3bday proof-of-work request
|
||||
// as a string
|
||||
func NewRequest(difficulty uint32, nonce []byte) string { |
||||
req := Request{ |
||||
Difficulty: difficulty, |
||||
Nonce: nonce, |
||||
Alg: Sha2BDay, |
||||
} |
||||
s, _ := req.MarshalText() |
||||
return string(s) |
||||
} |
||||
|
||||
func (proof Proof) MarshalText() ([]byte, error) { |
||||
return []byte(base64.RawStdEncoding.EncodeToString(proof.buf)), nil |
||||
} |
||||
|
||||
func (proof *Proof) UnmarshalText(buf []byte) error { |
||||
var err error |
||||
proof.buf, err = base64.RawStdEncoding.DecodeString(string(buf)) |
||||
return err |
||||
} |
||||
|
||||
func (req Request) MarshalText() ([]byte, error) { |
||||
return []byte(fmt.Sprintf("%s-%d-%s", |
||||
req.Alg, |
||||
req.Difficulty, |
||||
string(base64.RawStdEncoding.EncodeToString(req.Nonce)))), nil |
||||
} |
||||
|
||||
func (req *Request) UnmarshalText(buf []byte) error { |
||||
bits := strings.SplitN(string(buf), "-", 3) |
||||
if len(bits) != 3 { |
||||
return fmt.Errorf("There should be two dashes in a PoW request") |
||||
} |
||||
alg := Algorithm(bits[0]) |
||||
if alg != Sha2BDay { |
||||
return fmt.Errorf("%s: unsupported algorithm", bits[0]) |
||||
} |
||||
req.Alg = alg |
||||
diff, err := strconv.Atoi(bits[1]) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
req.Difficulty = uint32(diff) |
||||
req.Nonce, err = base64.RawStdEncoding.DecodeString(bits[2]) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
// Convenience function to check whether a proof of work is fulfilled
|
||||
func Check(request, proof string, data []byte) (bool, error) { |
||||
var req Request |
||||
var prf Proof |
||||
err := req.UnmarshalText([]byte(request)) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
err = prf.UnmarshalText([]byte(proof)) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
return prf.Check(req, data), nil |
||||
} |
||||
|
||||
// Fulfil the proof-of-work request.
|
||||
func (req *Request) Fulfil(data []byte) Proof { |
||||
switch req.Alg { |
||||
case Sha2BDay: |
||||
return Proof{fulfilSha2BDay(req.Nonce, req.Difficulty, data)} |
||||
default: |
||||
panic("No such algorithm") |
||||
} |
||||
} |
||||
|
||||
// Convenience function to fulfil the proof of work request
|
||||
func Fulfil(request string, data []byte) (string, error) { |
||||
var req Request |
||||
err := req.UnmarshalText([]byte(request)) |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
proof := req.Fulfil(data) |
||||
s, _ := proof.MarshalText() |
||||
return string(s), nil |
||||
} |
||||
|
||||
// Check whether the proof is ok
|
||||
func (proof *Proof) Check(req Request, data []byte) bool { |
||||
switch req.Alg { |
||||
case Sha2BDay: |
||||
return checkSha2BDay(proof.buf, req.Nonce, data, req.Difficulty) |
||||
default: |
||||
panic("No such algorithm") |
||||
} |
||||
} |
||||
@ -0,0 +1,56 @@ |
||||
package pow |
||||
|
||||
import ( |
||||
"testing" |
||||
) |
||||
|
||||
func TestSha2BDay(t *testing.T) { |
||||
nonce := []byte{1, 2, 3, 4, 5} |
||||
data := []byte{2, 2, 3, 4, 5} |
||||
r := NewRequest(5, nonce) |
||||
proof, err := Fulfil(r, data) |
||||
if err != nil { |
||||
t.Fatalf("Fulfil: %v", err) |
||||
} |
||||
ok, err := Check(r, proof, data) |
||||
if err != nil { |
||||
t.Fatalf("Check: %v", err) |
||||
} |
||||
if !ok { |
||||
t.Fatalf("Proof of work should be ok") |
||||
} |
||||
ok, err = Check(r, proof, nonce) |
||||
if err != nil { |
||||
t.Fatalf("Check: %v", err) |
||||
} |
||||
if ok { |
||||
t.Fatalf("Proof of work should not be ok") |
||||
} |
||||
} |
||||
|
||||
func BenchmarkCheck5(b *testing.B) { benchmarkCheck(5, b) } |
||||
func BenchmarkCheck10(b *testing.B) { benchmarkCheck(10, b) } |
||||
func BenchmarkCheck15(b *testing.B) { benchmarkCheck(15, b) } |
||||
func BenchmarkCheck20(b *testing.B) { benchmarkCheck(20, b) } |
||||
|
||||
func benchmarkCheck(diff uint32, b *testing.B) { |
||||
req := NewRequest(diff, []byte{1, 2, 3, 4, 5}) |
||||
prf, _ := Fulfil(req, []byte{6, 7, 8, 9}) |
||||
b.ResetTimer() |
||||
for n := 0; n < b.N; n++ { |
||||
Check(req, prf, []byte{6, 7, 8, 9}) |
||||
} |
||||
} |
||||
|
||||
func BenchmarkFulfil5(b *testing.B) { benchmarkFulfil(5, b) } |
||||
func BenchmarkFulfil10(b *testing.B) { benchmarkFulfil(10, b) } |
||||
func BenchmarkFulfil15(b *testing.B) { benchmarkFulfil(15, b) } |
||||
func BenchmarkFulfil20(b *testing.B) { benchmarkFulfil(20, b) } |
||||
|
||||
func benchmarkFulfil(diff uint32, b *testing.B) { |
||||
req := NewRequest(diff, []byte{1, 2, 3, 4, 5}) |
||||
b.ResetTimer() |
||||
for n := 0; n < b.N; n++ { |
||||
Fulfil(req, []byte{6, 7, 8, 9}) |
||||
} |
||||
} |
||||
@ -0,0 +1,25 @@ |
||||
package pow_test |
||||
|
||||
import ( |
||||
"fmt" // imported as pow
|
||||
|
||||
"github.com/simple-rules/harmony-benchmark/pow" |
||||
) |
||||
|
||||
func Example() { |
||||
// Create a proof of work request with difficulty 5
|
||||
req := pow.NewRequest(5, []byte("some random nonce")) |
||||
fmt.Printf("req: %s\n", req) |
||||
|
||||
// Fulfil the proof of work
|
||||
proof, _ := pow.Fulfil(req, []byte("some bound data")) |
||||
fmt.Printf("proof: %s\n", proof) |
||||
|
||||
// Check if the proof is correct
|
||||
ok, _ := pow.Check(req, proof, []byte("some bound data")) |
||||
fmt.Printf("check: %v", ok) |
||||
|
||||
// Output: req: sha2bday-5-c29tZSByYW5kb20gbm9uY2U
|
||||
// proof: AAAAAAAAAAMAAAAAAAAADgAAAAAAAAAb
|
||||
// check: true
|
||||
} |
||||
@ -1,81 +0,0 @@ |
||||
package pow |
||||
|
||||
import ( |
||||
"bytes" |
||||
"crypto/sha256" |
||||
"encoding/binary" |
||||
"fmt" |
||||
"math" |
||||
"math/big" |
||||
) |
||||
|
||||
var ( |
||||
maxNonce = math.MaxUint32 |
||||
) |
||||
|
||||
const targetBits = 24 |
||||
|
||||
// ProofOfWork represents a proof-of-work
|
||||
type ProofOfWork struct { |
||||
Challenge uint32 |
||||
target *big.Int |
||||
FinalNonce uint32 |
||||
} |
||||
|
||||
// NewProofOfWork builds and returns a ProofOfWork
|
||||
func NewProofOfWork(c uint32) *ProofOfWork { |
||||
target := big.NewInt(1) |
||||
target.Lsh(target, uint(256-targetBits)) |
||||
|
||||
pow := &ProofOfWork{Challenge: c, target: target, FinalNonce: 0} |
||||
|
||||
return pow |
||||
} |
||||
|
||||
func (pow *ProofOfWork) prepareData(nonce uint32) []byte { |
||||
challenge := make([]byte, 4) |
||||
binary.LittleEndian.PutUint32(challenge, pow.Challenge) |
||||
nonceB := make([]byte, 4) |
||||
binary.LittleEndian.PutUint32(nonceB, nonce) |
||||
data := bytes.Join( |
||||
[][]byte{ |
||||
challenge, |
||||
nonceB, |
||||
}, |
||||
[]byte{}, |
||||
) |
||||
return data |
||||
} |
||||
|
||||
// Run performs a proof-of-work
|
||||
func (pow *ProofOfWork) Run() int { |
||||
var hashInt big.Int |
||||
var hash [32]byte |
||||
nonce := 0 |
||||
for nonce < maxNonce { |
||||
data := pow.prepareData(uint32(nonce)) |
||||
|
||||
hash = sha256.Sum256(data) |
||||
fmt.Printf("\r%x", hash) |
||||
hashInt.SetBytes(hash[:]) |
||||
|
||||
if hashInt.Cmp(pow.target) == -1 { |
||||
pow.FinalNonce = uint32(nonce) |
||||
break |
||||
} else { |
||||
nonce++ |
||||
} |
||||
} |
||||
fmt.Print("\n\n") |
||||
return nonce |
||||
} |
||||
|
||||
// Validate validates block's PoW
|
||||
func (pow *ProofOfWork) Validate(nonce uint32) bool { |
||||
var hashInt big.Int |
||||
data := pow.prepareData(nonce) |
||||
hash := sha256.Sum256(data) |
||||
hashInt.SetBytes(hash[:]) |
||||
isValid := hashInt.Cmp(pow.target) == -1 |
||||
return isValid |
||||
} |
||||
@ -0,0 +1,78 @@ |
||||
package pow |
||||
|
||||
import ( |
||||
"bytes" |
||||
"crypto/sha256" |
||||
"encoding/binary" |
||||
) |
||||
|
||||
func checkSha2BDay(proof []byte, nonce, data []byte, diff uint32) bool { |
||||
if len(proof) != 24 { |
||||
return false |
||||
} |
||||
prefix1 := proof[:8] |
||||
prefix2 := proof[8:16] |
||||
prefix3 := proof[16:] |
||||
if bytes.Equal(prefix1, prefix2) || bytes.Equal(prefix2, prefix3) || |
||||
bytes.Equal(prefix1, prefix3) { |
||||
return false |
||||
} |
||||
resBuf := make([]byte, 32) |
||||
h := sha256.New() |
||||
h.Write(prefix1) |
||||
h.Write(data) |
||||
h.Write(nonce) |
||||
h.Sum(resBuf[:0]) |
||||
res1 := binary.BigEndian.Uint64(resBuf) & ((1 << diff) - 1) |
||||
h.Reset() |
||||
h.Write(prefix2) |
||||
h.Write(data) |
||||
h.Write(nonce) |
||||
h.Sum(resBuf[:0]) |
||||
res2 := binary.BigEndian.Uint64(resBuf) & ((1 << diff) - 1) |
||||
h.Reset() |
||||
h.Write(prefix3) |
||||
h.Write(data) |
||||
h.Write(nonce) |
||||
h.Sum(resBuf[:0]) |
||||
res3 := binary.BigEndian.Uint64(resBuf) & ((1 << diff) - 1) |
||||
return res1 == res2 && res2 == res3 |
||||
} |
||||
|
||||
func fulfilSha2BDay(nonce []byte, diff uint32, data []byte) []byte { |
||||
// TODO make multithreaded if the difficulty is high enough.
|
||||
// For light proof-of-work requests, the overhead of parallelizing is
|
||||
// not worth it.
|
||||
type Pair struct { |
||||
First, Second uint64 |
||||
} |
||||
var i uint64 = 1 |
||||
prefix := make([]byte, 8) |
||||
resBuf := make([]byte, 32) |
||||
lut := make(map[uint64]Pair) |
||||
h := sha256.New() |
||||
for { |
||||
binary.BigEndian.PutUint64(prefix, i) |
||||
h.Write(prefix) |
||||
h.Write(data) |
||||
h.Write(nonce) |
||||
h.Sum(resBuf[:0]) |
||||
res := binary.BigEndian.Uint64(resBuf) & ((1 << diff) - 1) |
||||
pair, ok := lut[res] |
||||
if ok { |
||||
if pair.Second != 0 { |
||||
ret := make([]byte, 24) |
||||
binary.BigEndian.PutUint64(ret, pair.First) |
||||
binary.BigEndian.PutUint64(ret[8:], pair.Second) |
||||
copy(ret[16:], prefix) |
||||
return ret |
||||
} |
||||
|
||||
lut[res] = Pair{First: pair.First, Second: i} |
||||
} else { |
||||
lut[res] = Pair{First: i} |
||||
} |
||||
h.Reset() |
||||
i++ |
||||
} |
||||
} |
||||
@ -1,21 +1,30 @@ |
||||
package main |
||||
|
||||
import "fmt" |
||||
import ( |
||||
"flag" |
||||
"fmt" |
||||
"time" |
||||
|
||||
// import (
|
||||
// "flag"
|
||||
|
||||
// "github.com/simple-rules/harmony-benchmark/p2p"
|
||||
// "github.com/simple-rules/harmony-benchmark/waitnode"
|
||||
// )
|
||||
"github.com/simple-rules/harmony-benchmark/node" |
||||
"github.com/simple-rules/harmony-benchmark/p2p" |
||||
) |
||||
|
||||
func main() { |
||||
fmt.Println("hello") |
||||
// ip := flag.String("ip", "127.0.0.0", "IP of the node")
|
||||
// port := flag.String("port", "8080", "port of the node")
|
||||
// flag.Parse()
|
||||
// peer := p2p.Peer{Ip: *ip, Port: *port}
|
||||
// idcpeer := p2p.Peer{Ip: "localhost", Port: "9000"} //Hardcoded here.
|
||||
// node := waitnode.New(peer)
|
||||
// node.ConnectIdentityChain(idcpeer)
|
||||
ip := flag.String("ip", "localhost", "IP of the node") |
||||
port := flag.String("port", "8080", "port of the node") |
||||
flag.Parse() |
||||
|
||||
i := 0 |
||||
peer := p2p.Peer{Ip: *ip, Port: *port} |
||||
fmt.Println("Now onto node i", i) |
||||
idcpeer := p2p.Peer{Ip: "localhost", Port: "9000"} //Hardcoded here.
|
||||
node := node.NewWaitNode(peer, idcpeer) |
||||
go func() { |
||||
node.ConnectIdentityChain() |
||||
}() |
||||
node.StartServer(*port) |
||||
|
||||
time.Sleep(5 * time.Second) |
||||
//}
|
||||
|
||||
} |
||||
|
||||
@ -1,82 +0,0 @@ |
||||
package waitnode |
||||
|
||||
import ( |
||||
"bytes" |
||||
"crypto/sha256" |
||||
"fmt" |
||||
"math" |
||||
"math/big" |
||||
) |
||||
|
||||
var ( |
||||
maxNonce = math.MaxInt64 |
||||
) |
||||
|
||||
const targetBits = 24 |
||||
|
||||
// ProofOfWork represents a proof-of-work |
||||
type ProofOfWork struct { |
||||
Challenge int32 |
||||
target *big.Int |
||||
} |
||||
|
||||
// NewProofOfWork builds and returns a ProofOfWork |
||||
func NewProofOfWork(c int32) *ProofOfWork { |
||||
target := big.NewInt(1) |
||||
target.Lsh(target, uint(256-targetBits)) |
||||
|
||||
pow := &ProofOfWork{c, target} |
||||
|
||||
return pow |
||||
} |
||||
|
||||
func (pow *ProofOfWork) prepareData(nonce int) []byte { |
||||
data := bytes.Join( |
||||
[][]byte{ |
||||
pow.Challenge., |
||||
IntToHex(int64(targetBits)), |
||||
IntToHex(int64(nonce)), |
||||
}, |
||||
[]byte{}, |
||||
) |
||||
|
||||
return data |
||||
} |
||||
|
||||
// Run performs a proof-of-work |
||||
func (pow *ProofOfWork) Run() (int, []byte) { |
||||
var hashInt big.Int |
||||
var hash [32]byte |
||||
nonce := 0 |
||||
|
||||
fmt.Printf("Mining the block containing \"%s\"\n", pow.block.Data) |
||||
for nonce < maxNonce { |
||||
data := pow.prepareData(nonce) |
||||
|
||||
hash = sha256.Sum256(data) |
||||
fmt.Printf("\r%x", hash) |
||||
hashInt.SetBytes(hash[:]) |
||||
|
||||
if hashInt.Cmp(pow.target) == -1 { |
||||
break |
||||
} else { |
||||
nonce++ |
||||
} |
||||
} |
||||
fmt.Print("\n\n") |
||||
|
||||
return nonce, hash[:] |
||||
} |
||||
|
||||
// Validate validates block's PoW |
||||
func (pow *ProofOfWork) Validate() bool { |
||||
var hashInt big.Int |
||||
|
||||
data := pow.prepareData(pow.challenge.Nonce) |
||||
hash := sha256.Sum256(data) |
||||
hashInt.SetBytes(hash[:]) |
||||
|
||||
isValid := hashInt.Cmp(pow.target) == -1 |
||||
|
||||
return isValid |
||||
} |
||||
@ -1,82 +0,0 @@ |
||||
package waitnode |
||||
|
||||
import ( |
||||
"bytes" |
||||
"crypto/sha256" |
||||
"encoding/gob" |
||||
"log" |
||||
|
||||
"github.com/simple-rules/harmony-benchmark/p2p" |
||||
"github.com/simple-rules/harmony-benchmark/utils" |
||||
) |
||||
|
||||
//WaitNode is for nodes waiting to join consensus
|
||||
type WaitNode struct { |
||||
Peer p2p.Peer |
||||
ID uint16 |
||||
SeedPeers p2p.Peer |
||||
} |
||||
|
||||
// StartServer a server and process the request by a handler.
|
||||
func (node *WaitNode) StartServer() { |
||||
log.Printf("Starting waitnode on server %s and port %s", node.Peer.Ip, node.Peer.Port) |
||||
} |
||||
|
||||
// //ConnectIdentityChain connects to identity chain
|
||||
// func (node *WaitNode) ConnectIdentityChain(peer p2p.Peer) {
|
||||
// pow := NewProofOfWork(10)
|
||||
// nonce := pow.Run()
|
||||
// if pow.FinalNonce != uint32(nonce) {
|
||||
// fmt.Println("Something wrong with POW")
|
||||
// }
|
||||
// p2p.SendMessage(peer, identity.ConstructIdentityMessage(identity.REGISTER, node.SerializeWaitNode()))
|
||||
// }
|
||||
|
||||
//Constructs node-id by hashing the IP.
|
||||
func calculateHash(num string) []byte { |
||||
var hashes [][]byte |
||||
hashes = append(hashes, utils.ConvertFixedDataIntoByteArray(num)) |
||||
hash := sha256.Sum256(bytes.Join(hashes, []byte{})) |
||||
return hash[:] |
||||
} |
||||
|
||||
// //SerializePOW serializes the node
|
||||
// func SerializePOW(pow ProofOfWork) []byte {
|
||||
// var result bytes.Buffer
|
||||
// encoder := gob.NewEncoder(&pow)
|
||||
// err := encoder.Encode(pow)
|
||||
// if err != nil {
|
||||
// log.Panic(err.Error())
|
||||
// }
|
||||
// return pow.Bytes()
|
||||
// }
|
||||
|
||||
//SerializeWaitNode serializes the node
|
||||
func (node *WaitNode) SerializeWaitNode() []byte { |
||||
var result bytes.Buffer |
||||
encoder := gob.NewEncoder(&result) |
||||
err := encoder.Encode(node) |
||||
if err != nil { |
||||
log.Panic(err.Error()) |
||||
} |
||||
return result.Bytes() |
||||
} |
||||
|
||||
// DeserializeWaitNode deserializes the node
|
||||
func DeserializeWaitNode(d []byte) *WaitNode { |
||||
var wn WaitNode |
||||
decoder := gob.NewDecoder(bytes.NewReader(d)) |
||||
err := decoder.Decode(&wn) |
||||
if err != nil { |
||||
log.Panic(err) |
||||
} |
||||
return &wn |
||||
} |
||||
|
||||
// New Create a new Node
|
||||
func New(Peer p2p.Peer) *WaitNode { |
||||
node := WaitNode{} |
||||
node.Peer = Peer |
||||
node.ID = utils.GetUniqueIdFromPeer(Peer) |
||||
return &node |
||||
} |
||||
@ -1,18 +0,0 @@ |
||||
package waitnode |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"github.com/simple-rules/harmony-benchmark/p2p" |
||||
) |
||||
|
||||
func TestNewNode(test *testing.T) { |
||||
p := p2p.Peer{Ip: "127.0.0.1", Port: "8080"} |
||||
wn := New(p) |
||||
b := wn.SerializeWaitNode() |
||||
wnd := DeserializeWaitNode(b) |
||||
if *wn != *wnd { |
||||
test.Error("Serialization is not working") |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue