aboutsummaryrefslogtreecommitdiff
path: root/test_msgnet_tls
diff options
context:
space:
mode:
Diffstat (limited to 'test_msgnet_tls')
-rw-r--r--test_msgnet_tls/main.go221
1 files changed, 116 insertions, 105 deletions
diff --git a/test_msgnet_tls/main.go b/test_msgnet_tls/main.go
index 1b434b2..a53b491 100644
--- a/test_msgnet_tls/main.go
+++ b/test_msgnet_tls/main.go
@@ -1,6 +1,5 @@
package main
-// #cgo CFLAGS: -I${SRCDIR}/../salticidae/include/
// #include <stdlib.h>
// #include "salticidae/network.h"
// void onTerm(int sig, void *);
@@ -11,159 +10,171 @@ package main
import "C"
import (
- "os"
- "fmt"
- "unsafe"
- "github.com/Determinant/salticidae-go"
+ "fmt"
+ "github.com/Determinant/salticidae-go"
+ "os"
+ "unsafe"
)
const (
- MSG_OPCODE_HELLO salticidae.Opcode = iota
- MSG_OPCODE_ACK
+ MSG_OPCODE_HELLO salticidae.Opcode = iota
+ MSG_OPCODE_ACK
)
func msgHelloSerialize(name string, text string) salticidae.Msg {
- serialized := salticidae.NewDataStream(true)
- serialized.PutU32(salticidae.ToLittleEndianU32(uint32(len(name))))
- serialized.PutData([]byte(name))
- serialized.PutData([]byte(text))
- return salticidae.NewMsgMovedFromByteArray(
- MSG_OPCODE_HELLO, salticidae.NewByteArrayMovedFromDataStream(serialized, true), true)
+ serialized := salticidae.NewDataStream(true)
+ serialized.PutU32(salticidae.ToLittleEndianU32(uint32(len(name))))
+ serialized.PutData([]byte(name))
+ serialized.PutData([]byte(text))
+ return salticidae.NewMsgMovedFromByteArray(
+ MSG_OPCODE_HELLO, salticidae.NewByteArrayMovedFromDataStream(serialized, true), true)
}
func msgHelloUnserialize(msg salticidae.Msg) (name string, text string) {
- p := msg.GetPayloadByMove()
- succ := true
- length := salticidae.FromLittleEndianU32(p.GetU32(&succ))
- t := p.GetDataInPlace(int(length)); name = string(t.Get()); t.Release()
- t = p.GetDataInPlace(p.Size()); text = string(t.Get()); t.Release()
- return
+ p := msg.GetPayloadByMove()
+ succ := true
+ length := salticidae.FromLittleEndianU32(p.GetU32(&succ))
+ t := p.GetDataInPlace(int(length))
+ name = string(t.Get())
+ t.Release()
+ t = p.GetDataInPlace(p.Size())
+ text = string(t.Get())
+ t.Release()
+ return
}
func msgAckSerialize() salticidae.Msg {
- return salticidae.NewMsgMovedFromByteArray(MSG_OPCODE_ACK, salticidae.NewByteArray(true), true)
+ return salticidae.NewMsgMovedFromByteArray(MSG_OPCODE_ACK, salticidae.NewByteArray(true), true)
}
func checkError(err *salticidae.Error) {
- if err.GetCode() != 0 {
- fmt.Printf("error during a sync call: %s\n", salticidae.StrError(err.GetCode()))
- os.Exit(1)
- }
+ if err.GetCode() != 0 {
+ fmt.Printf("error during a sync call: %s\n", salticidae.StrError(err.GetCode()))
+ os.Exit(1)
+ }
}
type MyNet struct {
- net salticidae.MsgNetwork
- name string
- peerCert salticidae.UInt256
- cname *C.char
+ net salticidae.MsgNetwork
+ name string
+ peerCert salticidae.UInt256
+ cname *C.char
}
var (
- alice, bob MyNet
- ec salticidae.EventContext
+ alice, bob MyNet
+ ec salticidae.EventContext
)
//export onTerm
func onTerm(_ C.int, _ unsafe.Pointer) {
- ec.Stop()
+ ec.Stop()
}
//export onReceiveHello
func onReceiveHello(_msg *C.struct_msg_t, _conn *C.struct_msgnetwork_conn_t, userdata unsafe.Pointer) {
- msg := salticidae.MsgFromC(salticidae.CMsg(_msg))
- conn := salticidae.MsgNetworkConnFromC(salticidae.CMsgNetworkConn(_conn))
- net := conn.GetNet()
- name, text := msgHelloUnserialize(msg)
- myName := C.GoString((*C.char)(userdata))
- fmt.Printf("[%s] %s says %s\n", myName, name, text)
- ack := msgAckSerialize()
- net.SendMsg(ack, conn)
+ msg := salticidae.MsgFromC(salticidae.CMsg(_msg))
+ conn := salticidae.MsgNetworkConnFromC(salticidae.CMsgNetworkConn(_conn))
+ net := conn.GetNet()
+ name, text := msgHelloUnserialize(msg)
+ myName := C.GoString((*C.char)(userdata))
+ fmt.Printf("[%s] %s says %s\n", myName, name, text)
+ ack := msgAckSerialize()
+ net.SendMsg(ack, conn)
}
//export onReceiveAck
func onReceiveAck(_ *C.struct_msg_t, _conn *C.struct_msgnetwork_conn_t, userdata unsafe.Pointer) {
- myName := C.GoString((*C.char)(userdata))
- fmt.Printf("[%s] the peer knows\n", myName)
+ myName := C.GoString((*C.char)(userdata))
+ fmt.Printf("[%s] the peer knows\n", myName)
}
//export connHandler
func connHandler(_conn *C.struct_msgnetwork_conn_t, connected C.bool, userdata unsafe.Pointer) C.bool {
- conn := salticidae.MsgNetworkConnFromC(salticidae.CMsgNetworkConn(_conn))
- net := conn.GetNet()
- myName := C.GoString((*C.char)(userdata))
- n := alice
- if myName == "bob" { n = bob }
- res := true
- if connected {
- certHash := conn.GetPeerCert().GetDer(true).GetHash(true)
- res = certHash.IsEq(n.peerCert)
- if conn.GetMode() == salticidae.CONN_MODE_ACTIVE {
- fmt.Printf("[%s] connected, sending hello.\n", myName)
- hello := msgHelloSerialize(myName, "Hello there!")
- n.net.SendMsg(hello, conn)
- } else {
- status := "fail"
- if res { status = "ok" }
- fmt.Printf("[%s] accepted, waiting for greetings.\n" +
- "The peer certificate fingerprint is %s (%s)\n",
- myName, certHash.GetHex(), status)
- }
- } else {
- fmt.Printf("[%s] disconnected, retrying.\n", myName)
- net.Connect(conn.GetAddr())
- }
- return C.bool(res)
+ conn := salticidae.MsgNetworkConnFromC(salticidae.CMsgNetworkConn(_conn))
+ net := conn.GetNet()
+ myName := C.GoString((*C.char)(userdata))
+ n := alice
+ if myName == "bob" {
+ n = bob
+ }
+ res := true
+ if connected {
+ certHash := conn.GetPeerCert().GetDer(true).GetHash(true)
+ res = certHash.IsEq(n.peerCert)
+ if conn.GetMode() == salticidae.CONN_MODE_ACTIVE {
+ fmt.Printf("[%s] connected, sending hello.\n", myName)
+ hello := msgHelloSerialize(myName, "Hello there!")
+ n.net.SendMsg(hello, conn)
+ } else {
+ status := "fail"
+ if res {
+ status = "ok"
+ }
+ fmt.Printf("[%s] accepted, waiting for greetings.\n"+
+ "The peer certificate fingerprint is %s (%s)\n",
+ myName, certHash.GetHex(), status)
+ }
+ } else {
+ fmt.Printf("[%s] disconnected, retrying.\n", myName)
+ net.Connect(conn.GetAddr())
+ }
+ return C.bool(res)
}
//export errorHandler
func errorHandler(_err *C.struct_SalticidaeCError, fatal C.bool, _ unsafe.Pointer) {
- err := (*salticidae.Error)(unsafe.Pointer(_err))
- s := "recoverable"
- if fatal { s = "fatal" }
- fmt.Printf("Captured %s error during an async call: %s\n", s, salticidae.StrError(err.GetCode()))
+ err := (*salticidae.Error)(unsafe.Pointer(_err))
+ s := "recoverable"
+ if fatal {
+ s = "fatal"
+ }
+ fmt.Printf("Captured %s error during an async call: %s\n", s, salticidae.StrError(err.GetCode()))
}
func genMyNet(ec salticidae.EventContext,
- name string, peerCert string,
- myAddr salticidae.NetAddr, otherAddr salticidae.NetAddr) MyNet {
- err := salticidae.NewError()
- netconfig := salticidae.NewMsgNetworkConfig()
- netconfig.EnableTLS(true)
- netconfig.TLSKeyFile(name + ".pem")
- netconfig.TLSCertFile(name + ".pem")
- net := salticidae.NewMsgNetwork(ec, netconfig, &err); checkError(&err)
- _peerCert := salticidae.NewUInt256FromByteArray(salticidae.NewByteArrayFromHex(peerCert))
- n := MyNet { net: net, name: name, peerCert: _peerCert, cname: C.CString(name) }
- cname := unsafe.Pointer(n.cname)
- n.net.RegHandler(MSG_OPCODE_HELLO, salticidae.MsgNetworkMsgCallback(C.onReceiveHello), cname)
- n.net.RegHandler(MSG_OPCODE_ACK, salticidae.MsgNetworkMsgCallback(C.onReceiveAck), cname)
- n.net.RegConnHandler(salticidae.MsgNetworkConnCallback(C.connHandler), cname)
- n.net.RegErrorHandler(salticidae.MsgNetworkErrorCallback(C.errorHandler), cname)
-
- n.net.Start()
- n.net.Listen(myAddr, &err); checkError(&err)
- n.net.Connect(otherAddr)
- return n
+ name string, peerCert string,
+ myAddr salticidae.NetAddr, otherAddr salticidae.NetAddr) MyNet {
+ err := salticidae.NewError()
+ netconfig := salticidae.NewMsgNetworkConfig()
+ netconfig.EnableTLS(true)
+ netconfig.TLSKeyFile(name + ".pem")
+ netconfig.TLSCertFile(name + ".pem")
+ net := salticidae.NewMsgNetwork(ec, netconfig, &err)
+ checkError(&err)
+ _peerCert := salticidae.NewUInt256FromByteArray(salticidae.NewByteArrayFromHex(peerCert))
+ n := MyNet{net: net, name: name, peerCert: _peerCert, cname: C.CString(name)}
+ cname := unsafe.Pointer(n.cname)
+ n.net.RegHandler(MSG_OPCODE_HELLO, salticidae.MsgNetworkMsgCallback(C.onReceiveHello), cname)
+ n.net.RegHandler(MSG_OPCODE_ACK, salticidae.MsgNetworkMsgCallback(C.onReceiveAck), cname)
+ n.net.RegConnHandler(salticidae.MsgNetworkConnCallback(C.connHandler), cname)
+ n.net.RegErrorHandler(salticidae.MsgNetworkErrorCallback(C.errorHandler), cname)
+
+ n.net.Start()
+ n.net.Listen(myAddr, &err)
+ checkError(&err)
+ n.net.Connect(otherAddr)
+ return n
}
func main() {
- ec = salticidae.NewEventContext()
- err := salticidae.NewError()
+ ec = salticidae.NewEventContext()
+ err := salticidae.NewError()
- aliceAddr := salticidae.NewNetAddrFromIPPortString("127.0.0.1:12345", true, &err)
- bobAddr := salticidae.NewNetAddrFromIPPortString("127.0.0.1:12346", true, &err)
+ aliceAddr := salticidae.NewNetAddrFromIPPortString("127.0.0.1:12345", true, &err)
+ bobAddr := salticidae.NewNetAddrFromIPPortString("127.0.0.1:12346", true, &err)
- alice = genMyNet(ec, "alice", "ed5a9a8c7429dcb235a88244bc69d43d16b35008ce49736b27aaa3042a674043", aliceAddr, bobAddr)
- bob = genMyNet(ec, "bob", "ef3bea4e72f4d0e85da7643545312e2ff6dded5e176560bdffb1e53b1cef4896", bobAddr, aliceAddr)
+ alice = genMyNet(ec, "alice", "ed5a9a8c7429dcb235a88244bc69d43d16b35008ce49736b27aaa3042a674043", aliceAddr, bobAddr)
+ bob = genMyNet(ec, "bob", "ef3bea4e72f4d0e85da7643545312e2ff6dded5e176560bdffb1e53b1cef4896", bobAddr, aliceAddr)
- ev_int := salticidae.NewSigEvent(ec, salticidae.SigEventCallback(C.onTerm), nil)
- ev_int.Add(salticidae.SIGINT)
- ev_term := salticidae.NewSigEvent(ec, salticidae.SigEventCallback(C.onTerm), nil)
- ev_term.Add(salticidae.SIGTERM)
+ ev_int := salticidae.NewSigEvent(ec, salticidae.SigEventCallback(C.onTerm), nil)
+ ev_int.Add(salticidae.SIGINT)
+ ev_term := salticidae.NewSigEvent(ec, salticidae.SigEventCallback(C.onTerm), nil)
+ ev_term.Add(salticidae.SIGTERM)
- ec.Dispatch()
- alice.net.Stop()
- bob.net.Stop()
- C.free(unsafe.Pointer(alice.cname))
- C.free(unsafe.Pointer(bob.cname))
+ ec.Dispatch()
+ alice.net.Stop()
+ bob.net.Stop()
+ C.free(unsafe.Pointer(alice.cname))
+ C.free(unsafe.Pointer(bob.cname))
}
Powered by cgit, Themed by Ted Yin.