105 lines
3.3 KiB
Go
105 lines
3.3 KiB
Go
package graphvent
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
"crypto/rand"
|
|
"crypto/ecdsa"
|
|
)
|
|
|
|
func TestNodeDB(t *testing.T) {
|
|
ctx := logTestContext(t, []string{})
|
|
node_type := NodeType("test")
|
|
err := ctx.RegisterNodeType(node_type, []ExtType{GroupExtType})
|
|
fatalErr(t, err)
|
|
|
|
node := NewNode(ctx, nil, node_type, 10, nil, NewGroupExt(nil))
|
|
|
|
ctx.Nodes = NodeMap{}
|
|
_, err = ctx.GetNode(node.ID)
|
|
fatalErr(t, err)
|
|
}
|
|
|
|
func TestNodeRead(t *testing.T) {
|
|
ctx := logTestContext(t, []string{})
|
|
node_type := NodeType("TEST")
|
|
err := ctx.RegisterNodeType(node_type, []ExtType{ACLExtType, GroupExtType, ECDHExtType})
|
|
fatalErr(t, err)
|
|
|
|
n1_key, err := ecdsa.GenerateKey(ctx.ECDSA, rand.Reader)
|
|
fatalErr(t, err)
|
|
n2_key, err := ecdsa.GenerateKey(ctx.ECDSA, rand.Reader)
|
|
fatalErr(t, err)
|
|
|
|
n1_id := KeyID(&n1_key.PublicKey)
|
|
n2_id := KeyID(&n2_key.PublicKey)
|
|
|
|
ctx.Log.Logf("test", "N1: %s", n1_id)
|
|
ctx.Log.Logf("test", "N2: %s", n2_id)
|
|
|
|
n2_policy := NewPerNodePolicy(map[NodeID]Actions{
|
|
n1_id: Actions{MakeAction(ReadResultSignalType, "+")},
|
|
})
|
|
n2_listener := NewListenerExt(10)
|
|
n2 := NewNode(ctx, n2_key, node_type, 10, nil, NewACLExt(n2_policy), NewGroupExt(nil), NewECDHExt(), n2_listener)
|
|
|
|
n1_policy := NewPerNodePolicy(map[NodeID]Actions{
|
|
n2_id: Actions{MakeAction(ReadSignalType, "+")},
|
|
})
|
|
n1 := NewNode(ctx, n1_key, node_type, 10, nil, NewACLExt(n1_policy), NewGroupExt(nil), NewECDHExt())
|
|
|
|
ctx.Send(n2.ID, n1.ID, NewReadSignal(map[ExtType][]string{
|
|
GroupExtType: []string{"members"},
|
|
}))
|
|
|
|
res, err := WaitForSignal(ctx, n2_listener, 10*time.Millisecond, ReadResultSignalType, func(sig ReadResultSignal) bool {
|
|
return true
|
|
})
|
|
fatalErr(t, err)
|
|
ctx.Log.Logf("test", "READ_RESULT: %+v", res)
|
|
}
|
|
|
|
func TestECDH(t *testing.T) {
|
|
ctx := logTestContext(t, []string{"test", "ecdh", "policy"})
|
|
|
|
node_type := NodeType("TEST")
|
|
err := ctx.RegisterNodeType(node_type, []ExtType{ACLExtType, ECDHExtType})
|
|
fatalErr(t, err)
|
|
|
|
n1_listener := NewListenerExt(10)
|
|
ecdh_policy := NewAllNodesPolicy(Actions{MakeAction(ECDHSignalType, "+"), MakeAction(ECDHProxySignalType, "+")})
|
|
n1 := NewNode(ctx, nil, node_type, 10, nil, NewACLExt(ecdh_policy), NewECDHExt(), n1_listener)
|
|
n2 := NewNode(ctx, nil, node_type, 10, nil, NewACLExt(ecdh_policy), NewECDHExt())
|
|
n3_listener := NewListenerExt(10)
|
|
n3_policy := NewPerNodePolicy(NodeActions{n1.ID: Actions{MakeAction(StopSignalType)}})
|
|
n3 := NewNode(ctx, nil, node_type, 10, nil, NewACLExt(ecdh_policy, n3_policy), NewECDHExt(), n3_listener)
|
|
|
|
ctx.Log.Logf("test", "N1: %s", n1.ID)
|
|
ctx.Log.Logf("test", "N2: %s", n2.ID)
|
|
|
|
|
|
ecdh_req, n1_ec, err := NewECDHReqSignal(ctx, n1)
|
|
ecdh_ext, err := GetExt[*ECDHExt](n1)
|
|
fatalErr(t, err)
|
|
ecdh_ext.ECDHStates[n2.ID] = ECDHState{
|
|
ECKey: n1_ec,
|
|
SharedSecret: nil,
|
|
}
|
|
fatalErr(t, err)
|
|
ctx.Log.Logf("test", "N1_EC: %+v", n1_ec)
|
|
err = ctx.Send(n1.ID, n2.ID, ecdh_req)
|
|
fatalErr(t, err)
|
|
|
|
_, err = WaitForSignal(ctx, n1_listener, 100*time.Millisecond, ECDHSignalType, func(sig ECDHSignal) bool {
|
|
return sig.Str == "resp"
|
|
})
|
|
fatalErr(t, err)
|
|
time.Sleep(10*time.Millisecond)
|
|
|
|
ecdh_sig, err := NewECDHProxySignal(n1.ID, n3.ID, NewDirectSignal(StopSignalType), ecdh_ext.ECDHStates[n2.ID].SharedSecret)
|
|
fatalErr(t, err)
|
|
|
|
err = ctx.Send(n1.ID, n2.ID, ecdh_sig)
|
|
fatalErr(t, err)
|
|
}
|