Fixed tape, and random input/tape/tree gen
This commit is contained in:
Vlad Zamfir
commit
b860c79a3d
|
|
@ -0,0 +1,366 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"math"
|
||||
"math/big"
|
||||
"encoding/hex"
|
||||
"github.com/obscuren/sha3"
|
||||
"testing"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
//For use in benchmarking
|
||||
const tree_depth = 5
|
||||
const tape_width = 32 //int(math.Pow(2,tree_depth)) < would be nice, but go won't recognize as const
|
||||
const tape_depth = 100
|
||||
|
||||
//Number of operations that are drawn from to form the tape and the tree
|
||||
const num_ops = 9
|
||||
//This is the number of times the PoW algorithm is run
|
||||
const sample_size = 100
|
||||
|
||||
func Bytes2Hex(d []byte) string {
|
||||
return hex.EncodeToString(d)
|
||||
}
|
||||
|
||||
func Hex2Bytes(str string) []byte {
|
||||
h, _ := hex.DecodeString(str)
|
||||
return h
|
||||
}
|
||||
|
||||
func plus(z *big.Int, x *big.Int, y *big.Int) *big.Int {
|
||||
var lim big.Int
|
||||
lim.Exp(big.NewInt(2),big.NewInt(256), big.NewInt(0))
|
||||
z.Add(x,y)
|
||||
return z.Mod(z,&lim)
|
||||
|
||||
}
|
||||
|
||||
func times(z *big.Int, x *big.Int, y *big.Int) *big.Int {
|
||||
var lim, x1, y1 big.Int
|
||||
lim.Exp(big.NewInt(2),big.NewInt(256), big.NewInt(0))
|
||||
x1.Set(x)
|
||||
y1.Set(y)
|
||||
z.Mul(x,y)
|
||||
z.Mod(z,&lim)
|
||||
return z
|
||||
}
|
||||
|
||||
func mod(z *big.Int, x *big.Int, y *big.Int) *big.Int {
|
||||
|
||||
if (x.Cmp(big.NewInt(0)) == 0 || y.Cmp(big.NewInt(0)) == 0) {
|
||||
return big.NewInt(0)
|
||||
}
|
||||
if x.Cmp(y) == -1 { //if x < y
|
||||
z.Mod(y,x)
|
||||
} else if x.Cmp(y) == 1 { //if x > y
|
||||
z.Mod(x,y)
|
||||
}
|
||||
return z
|
||||
}
|
||||
|
||||
func xor(z *big.Int, x *big.Int, y *big.Int) *big.Int {return z.Xor(x,y)}
|
||||
|
||||
func nxor(z *big.Int, x *big.Int, y *big.Int) *big.Int {
|
||||
z.Xor(x,y)
|
||||
return z.Not(z)
|
||||
}
|
||||
|
||||
func and(z *big.Int, x *big.Int, y *big.Int) *big.Int {return z.And(x,y)}
|
||||
|
||||
func not(z *big.Int, x *big.Int, y *big.Int) *big.Int {
|
||||
_ = y
|
||||
return z.Not(x)
|
||||
}
|
||||
|
||||
func or(z *big.Int, x *big.Int, y *big.Int) *big.Int {return z.Or(x,y)}
|
||||
|
||||
func rshift(z *big.Int, x *big.Int, y *big.Int) *big.Int {
|
||||
var lim big.Int
|
||||
lim.Exp(big.NewInt(2),big.NewInt(256), big.NewInt(0))
|
||||
z.Rsh(x,7)
|
||||
return z.Mod(z,&lim)
|
||||
}
|
||||
|
||||
|
||||
func GetOp(i int) func(z *big.Int, x *big.Int, y *big.Int) *big.Int{
|
||||
switch i {
|
||||
case 0:
|
||||
return plus
|
||||
case 1:
|
||||
return xor
|
||||
case 2:
|
||||
return not
|
||||
case 3:
|
||||
return times
|
||||
case 4:
|
||||
return mod
|
||||
case 5:
|
||||
return rshift
|
||||
case 6:
|
||||
return or
|
||||
case 7:
|
||||
return and
|
||||
case 8:
|
||||
return nxor
|
||||
}
|
||||
return plus
|
||||
}
|
||||
|
||||
//the tapelink is used to t
|
||||
type Tapelink struct {
|
||||
I int64
|
||||
J int64
|
||||
op func(z *big.Int, x *big.Int, y *big.Int) *big.Int
|
||||
}
|
||||
|
||||
func Sha3Bin(data []byte) []byte {
|
||||
d := sha3.NewKeccak256()
|
||||
d.Write(data)
|
||||
return d.Sum(nil)
|
||||
}
|
||||
|
||||
func BenchmarkSha3Bin(b *testing.B){
|
||||
for i:= 0; i < b.N; i++ {
|
||||
Sha3Bin([]byte(string(i)))
|
||||
}
|
||||
}
|
||||
|
||||
//generates a tape of operations that is w*d long
|
||||
func gen_tape(seed int64, w int, d int) []Tapelink {
|
||||
var X = big.NewInt(0)
|
||||
var Y = big.NewInt(0)
|
||||
Y.Exp(big.NewInt(2),big.NewInt(80), nil)
|
||||
var M = big.NewInt(0)
|
||||
var T []Tapelink
|
||||
for i := 0; i < w*d; i++{
|
||||
// add empty link to tape
|
||||
T = append(T, *new(Tapelink))
|
||||
|
||||
// generate new entropy as needed
|
||||
if (int64(X.Cmp(Y)) == -1) {X.SetBytes(Sha3Bin([]byte(strconv.FormatInt(seed + int64(i), 10))))}
|
||||
|
||||
// Pick random index I
|
||||
T[i].I = M.Mod(X,big.NewInt(int64(w))).Int64()
|
||||
M = big.NewInt(int64(w))
|
||||
X = X.Div(X,M)
|
||||
|
||||
// Pick random index J
|
||||
mm := big.NewInt(M.Mod(X,big.NewInt(int64(w-1))).Int64() + int64(1) + T[i].I)
|
||||
T[i].J = M.Mod(mm, big.NewInt(int64(w))).Int64()
|
||||
M = big.NewInt(int64(w-1))
|
||||
X = X.Div(X,M)
|
||||
|
||||
// Pick random operation
|
||||
T[i].op = GetOp(int(M.Mod(X, big.NewInt(int64(num_ops))).Int64()))
|
||||
M = big.NewInt(int64(num_ops))
|
||||
X = X.Div(X,M)
|
||||
}
|
||||
return T
|
||||
}
|
||||
|
||||
func BenchmarkGen_tape(b *testing.B){
|
||||
var X big.Int
|
||||
var s int64
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
b.StopTimer()
|
||||
X.SetBytes(Sha3Bin([]byte(string(i))))
|
||||
s = X.Int64()
|
||||
b.StartTimer()
|
||||
gen_tape(s, tape_width, tape_depth)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func gen_inputs(seed int64, w int) []big.Int {
|
||||
var A []big.Int
|
||||
for i := 0; i < w; i++ {
|
||||
A = append(A, *new(big.Int))
|
||||
if (i % 256 == 0) {
|
||||
A[i].SetBytes(Sha3Bin([]byte(strconv.FormatInt(seed + int64(i), 10))))
|
||||
} else {
|
||||
A[i].Lsh(&A[i-1], 1)
|
||||
}
|
||||
}
|
||||
return A
|
||||
}
|
||||
|
||||
func BenchmarkGen_inputs(b *testing.B){
|
||||
var X big.Int
|
||||
var s int64
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
b.StopTimer()
|
||||
X.SetBytes(Sha3Bin([]byte(string(i))))
|
||||
s = X.Int64()
|
||||
b.StartTimer()
|
||||
gen_inputs(s, tape_width)
|
||||
}
|
||||
}
|
||||
|
||||
//this changes the inputs as it goes through a tape with d links
|
||||
func run_tape(tape []Tapelink, inputs []big.Int, d int) {
|
||||
var X *big.Int
|
||||
X = big.NewInt(0)
|
||||
for i := 0; i < d; i++ {
|
||||
X = tape[i].op(X, &inputs[tape[i].I], &inputs[tape[i].J])
|
||||
inputs[tape[i].I].Set(X)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkRun_tape(b *testing.B){
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
b.StopTimer()
|
||||
T := gen_tape(int64(i), tape_width, tape_depth)
|
||||
I := gen_inputs(int64(i), tape_width)
|
||||
b.StartTimer()
|
||||
run_tape(T,I,tape_width*tape_depth)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
//returns a 2^d - 1 length tape of operations (no I or J's in the Tapelinks)
|
||||
func gen_tree(seed int64, d int) []Tapelink {
|
||||
M := big.NewInt(0) // dummy variable
|
||||
X := big.NewInt(0) // entropy variable
|
||||
Y := big.NewInt(0) // entropy buffer size
|
||||
Y.Exp(big.NewInt(2),big.NewInt(80), nil)
|
||||
var T []Tapelink // the tree will be stored here
|
||||
|
||||
for i := 0; i < int(math.Pow(2, float64(d))) - 1; i++{
|
||||
|
||||
T = append(T, *new(Tapelink))
|
||||
|
||||
//giving it more entropy, if X < 2^32
|
||||
if (X.Cmp(Y) == -1) {X.SetBytes(Sha3Bin([]byte(strconv.FormatInt(seed + int64(i),10))))}
|
||||
|
||||
//filling the tape with random ops
|
||||
T[i].op = GetOp(int(M.Mod(X, big.NewInt(num_ops)).Int64()))
|
||||
M = big.NewInt(num_ops)
|
||||
X = X.Div(X,M)
|
||||
}
|
||||
return T
|
||||
}
|
||||
|
||||
func BenchmarkGen_tree(b *testing.B) {
|
||||
var X big.Int
|
||||
var s int64
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
b.StopTimer()
|
||||
X.SetBytes(Sha3Bin([]byte(string(i))))
|
||||
s = X.Int64()
|
||||
b.StartTimer()
|
||||
gen_tree(s, tree_depth)
|
||||
}
|
||||
}
|
||||
|
||||
//there should be 2^d inputs and 2^d - 1 links in the tape. not complying is an unhandled exception
|
||||
func run_tree(inputs []big.Int, tree []Tapelink, d int) *big.Int {
|
||||
X := big.NewInt(0)
|
||||
counter := 0
|
||||
for j := 0; j < d; j++ {
|
||||
for i := 0; i < int(math.Pow(2,float64(d - j - 1))); i++ {
|
||||
X = tree[counter].op(X, &inputs[2*i], &inputs[2*i + 1])
|
||||
inputs[i].Set(X)
|
||||
counter += 1
|
||||
}
|
||||
}
|
||||
|
||||
return &inputs[0]
|
||||
}
|
||||
|
||||
func BenchmarkRun_tree(b *testing.B) {
|
||||
var X big.Int
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
b.StopTimer()
|
||||
var inputs []big.Int
|
||||
for j := 0; j < tape_width; j++ {
|
||||
X.SetBytes(Sha3Bin([]byte(string(j + i*tape_width))))
|
||||
inputs = append(inputs, X)
|
||||
}
|
||||
tree := gen_tree(X.Int64(), tree_depth)
|
||||
b.StartTimer()
|
||||
run_tree(inputs, tree, tree_depth)
|
||||
}
|
||||
}
|
||||
|
||||
func sha_cap(s []string, n int) []byte{
|
||||
var feed string
|
||||
feed = ""
|
||||
for i := 0; i < n; i++ {
|
||||
feed += s[i]
|
||||
}
|
||||
return Sha3Bin([]byte(feed))
|
||||
}
|
||||
|
||||
|
||||
func BenchmarkSha_cap(b *testing.B){
|
||||
var X big.Int
|
||||
var s []string
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
b.StopTimer()
|
||||
X.SetBytes(Sha3Bin([]byte(string(i))))
|
||||
s = append(s, X.String())
|
||||
b.StartTimer()
|
||||
sha_cap(s,1)
|
||||
}
|
||||
}
|
||||
|
||||
func main(){
|
||||
var seed int64
|
||||
var sample []big.Int
|
||||
|
||||
|
||||
seed = int64(13300331)
|
||||
|
||||
for i := 0; i < sample_size; i++ {
|
||||
|
||||
Tape := gen_tape(seed, tape_width, tape_depth)
|
||||
Tree := gen_tree(seed, tree_depth)
|
||||
|
||||
seed += int64(i*i)
|
||||
if i%10 == 0 {
|
||||
fmt.Printf("i: %d\n",i)
|
||||
}
|
||||
I := gen_inputs(seed, tape_width)
|
||||
run_tape(Tape, I, tape_depth*tape_width)
|
||||
|
||||
|
||||
output := *run_tree(I, Tree, tree_depth)
|
||||
if output.Cmp(big.NewInt(0)) == 0 {
|
||||
fmt.Printf("We have a zero from the tree, form operation: \n")
|
||||
fmt.Println(Tree[len(Tree)-1])
|
||||
|
||||
}
|
||||
|
||||
|
||||
var blockhashes []string
|
||||
blockhashes = append(blockhashes, output.String())
|
||||
|
||||
sample = append(sample, output)
|
||||
H := sha_cap(blockhashes, 1)
|
||||
|
||||
output.SetBytes(H)
|
||||
//sample = append(sample, output)
|
||||
}
|
||||
|
||||
var collisions int = 0
|
||||
for i := 0; i < sample_size; i++ {
|
||||
//fmt.Println(sample[i])
|
||||
for j:=0; j < i; j++ {
|
||||
//if sample[i] == sample[j] {
|
||||
if sample[i].Cmp(&sample[j]) == 0 {
|
||||
collisions += 1
|
||||
//fmt.Printf("collision on i, j: %d, %d", i, j)
|
||||
//fmt.Println(sample[i])
|
||||
}
|
||||
}
|
||||
}
|
||||
fmt.Printf("number of outputs with same value: %d out of %d\n", (1+int(math.Pow(float64(1+8*collisions),0.5)))/2, sample_size)
|
||||
}
|
||||
Loading…
Reference in New Issue