func TestMaxRxSizeSet(t *testing.T) {
srep, err := rep.NewSocket()
if err != nil {
t.Errorf("Failed to make REP: %v", err)
return
}
defer srep.Close()
err = srep.SetOption(mangos.OptionMaxRecvSize, 100)
if err != nil {
t.Errorf("Failed SetOption: %v", err)
return
}
v, err := srep.GetOption(mangos.OptionMaxRecvSize)
if err != nil {
t.Errorf("Failed GetOption: %v", err)
return
}
if val, ok := v.(int); !ok {
t.Errorf("Returned value not type int")
} else if val != 100 {
t.Errorf("Returned value %d not %d", val, 100)
}
}
func StartNanomsgJsonServer(port string) {
var server mangos.Socket
var err error
if server, err = rep.NewSocket(); err != nil {
log.Fatal(err)
}
server.AddTransport(tcp.NewTransport())
server.Listen("tcp://0.0.0.0:" + port)
for {
msg, _ := server.Recv()
var body Request
var response []byte
json.Unmarshal(msg, &body)
if response, err = json.Marshal(Response{Method: body.Method, PayloadLength: len(body.Payload)}); err != nil {
log.Fatal(err)
}
server.Send(response)
}
}
// NewSocket allocates a new Socket. The Socket is the handle used to
// access the underlying library.
func NewSocket(d Domain, p Protocol) (*Socket, error) {
var s Socket
var err error
s.proto = p
s.dom = d
switch p {
case PUB:
s.sock, err = pub.NewSocket()
case SUB:
s.sock, err = sub.NewSocket()
case PUSH:
s.sock, err = push.NewSocket()
case PULL:
s.sock, err = pull.NewSocket()
case REQ:
s.sock, err = req.NewSocket()
case REP:
s.sock, err = rep.NewSocket()
case SURVEYOR:
s.sock, err = surveyor.NewSocket()
case RESPONDENT:
s.sock, err = respondent.NewSocket()
case PAIR:
s.sock, err = pair.NewSocket()
case BUS:
s.sock, err = bus.NewSocket()
default:
err = mangos.ErrBadProto
}
if err != nil {
return nil, err
}
switch d {
case AF_SP:
case AF_SP_RAW:
err = s.sock.SetOption(mangos.OptionRaw, true)
default:
err = errBadDomain
}
if err != nil {
s.sock.Close()
return nil, err
}
// Compat mode sockets should timeout on send if we don't have any pipes
if err = s.sock.SetOption(mangos.OptionWriteQLen, 0); err != nil {
s.sock.Close()
return nil, err
}
s.rto = -1
s.sto = -1
all.AddTransports(s.sock)
return &s, nil
}
func TestWebsockPath(t *testing.T) {
sockReq, _ := req.NewSocket()
sockRep, _ := rep.NewSocket()
tran := NewTransport()
l, e := tran.NewListener("ws://127.0.0.1:3335/mysock", sockReq)
if e != nil {
t.Errorf("Failed new Listener: %v", e)
return
}
d, e := tran.NewDialer("ws://127.0.0.1:3335/boguspath", sockRep)
if e != nil {
t.Errorf("Failed new Dialer: %v", e)
return
}
if e = l.Listen(); e != nil {
t.Errorf("Listen failed")
return
}
defer l.Close()
p, e := d.Dial()
if p != nil {
defer p.Close()
}
if e == nil {
t.Errorf("Dial passed, when should not have!")
return
}
t.Logf("Got expected error %v", e)
}
func server(url string, nworkers int) {
var sock mangos.Socket
var err error
var wg sync.WaitGroup
rand.Seed(time.Now().UnixNano())
if sock, err = rep.NewSocket(); err != nil {
die("can't get new rep socket: %s", err)
}
if err = sock.SetOption(mangos.OptionRaw, true); err != nil {
die("can't set raw mode: %s", err)
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Listen(url); err != nil {
die("can't listen on rep socket: %s", err.Error())
}
wg.Add(nworkers)
fmt.Printf("Starting %d workers\n", nworkers)
for id := 0; id < nworkers; id++ {
go func(id int) {
defer wg.Done()
serverWorker(sock, id)
}(id)
}
wg.Wait()
}
func StartNanomsgProtoServer(port string) {
var server mangos.Socket
var err error
if server, err = rep.NewSocket(); err != nil {
log.Fatal(err)
}
server.AddTransport(tcp.NewTransport())
server.Listen("tcp://0.0.0.0:" + port)
for {
msg, _ := server.Recv()
var body pb.Request
proto.Unmarshal(msg, &body)
data, _ := proto.Marshal(&pb.Response{
Method: body.Method,
PayloadLength: int64(len(body.Payload)),
})
server.Send(data)
}
}
func TestWebsockMux(t *testing.T) {
sockReq, _ := req.NewSocket()
sockRep, _ := rep.NewSocket()
tran := NewTransport()
l, e := tran.NewListener("ws://127.0.0.1:3336/mysock", sockReq)
if e != nil {
t.Errorf("Failed new Listener: %v", e)
return
}
muxi, e := l.GetOption(OptionWebSocketMux)
if e != nil {
t.Errorf("Failed get mux: %v", e)
}
mux := muxi.(*http.ServeMux)
mux.HandleFunc("/bogus", bogusHandler)
d, e := tran.NewDialer("ws://127.0.0.1:3336/bogus", sockRep)
if e != nil {
t.Errorf("Failed new Dialer: %v", e)
return
}
if e = l.Listen(); e != nil {
t.Errorf("Listen failed")
return
}
defer l.Close()
p, e := d.Dial()
if p != nil {
defer p.Close()
}
if e == nil {
t.Errorf("Dial passed, when should not have!")
return
}
t.Logf("Got expected error %v", e)
// Now let's try to use http client.
resp, err := http.Get("http://127.0.0.1:3336/bogus")
if err != nil {
t.Errorf("Get of boguspath failed: %v", err)
return
}
if resp.StatusCode != 200 {
t.Errorf("Response code wrong: %d", resp.StatusCode)
return
}
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
t.Errorf("ReadAll Failed: %v", err)
return
}
if string(body) != bogusstr {
t.Errorf("Results mismatch: %s != %s", string(body), bogusstr)
}
t.Logf("Got body: %s", string(body))
}
func node0(url string) {
var sock mangos.Socket
var err error
var msg []byte
if sock, err = rep.NewSocket(); err != nil {
die("can't get new rep socket: %s", err)
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Listen(url); err != nil {
die("can't listen on rep socket: %s", err.Error())
}
for {
// Could also use sock.RecvMsg to get header
msg, err = sock.Recv()
if string(msg) == "DATE" { // no need to terminate
fmt.Println("NODE0: RECEIVED DATE REQUEST")
d := date()
fmt.Printf("NODE0: SENDING DATE %s\n", d)
err = sock.Send([]byte(d))
if err != nil {
die("can't send reply: %s", err.Error())
}
}
}
}
// NewDaemon creates and returns a new Daemon from the provided Config. An
// error is returned if the Daemon cannot be created.
func NewDaemon(config *Config) (*Daemon, error) {
rep, err := rep.NewSocket()
if err != nil {
return nil, err
}
rep.AddTransport(tcp.NewTransport())
return &Daemon{rep, nil, []*publisher{}, []*subscriber{}, config}, nil
}
func (rt *repTest) Init(t *testing.T, addr string) bool {
var err error
if rt.Sock, err = rep.NewSocket(); err != nil {
rt.Errorf("NewSocket(): %v", err)
return false
}
return rt.T.Init(t, addr)
}
func NewTranTest(tran mangos.Transport, addr string) *TranTest {
tt := &TranTest{addr: addr, tran: tran}
if strings.HasPrefix(tt.addr, "tls+tcp://") || strings.HasPrefix(tt.addr, "wss://") {
tt.cliCfg, _ = GetTlsConfig(false)
tt.srvCfg, _ = GetTlsConfig(true)
}
tt.sockRep, _ = rep.NewSocket()
tt.sockReq, _ = req.NewSocket()
return tt
}
func TestTtlInvalidTooBig(t *testing.T) {
srep, err := rep.NewSocket()
if err != nil {
t.Errorf("Failed to make REP: %v", err)
return
}
defer srep.Close()
err = srep.SetOption(mangos.OptionTtl, 256)
switch err {
case mangos.ErrBadValue: // expected result
case nil:
t.Errorf("Negative test fail, permitted too large TTL")
default:
t.Errorf("Negative test fail (256), wrong error %v")
}
}
func TestMaxRxSizeInvalidNotInt(t *testing.T) {
srep, err := rep.NewSocket()
if err != nil {
t.Errorf("Failed to make REP: %v", err)
return
}
defer srep.Close()
err = srep.SetOption(mangos.OptionMaxRecvSize, "garbage")
switch err {
case mangos.ErrBadValue: // expected result
case nil:
t.Errorf("Negative test fail, permitted non-int value")
default:
t.Errorf("Negative test fail (garbage), wrong error %v")
}
}
func newCmdServer(addr string) (mangos.Socket, error) {
log.Println("alauncher: creating cmd server...")
socket, err := rep.NewSocket()
if err != nil {
return nil, err
}
if err := socket.SetOption(mangos.OptionRecvDeadline, time.Second*3); err != nil {
socket.Close()
return nil, err
}
//socket.AddTransport(ipc.NewTransport())
socket.AddTransport(tcp.NewTransport())
if err := socket.Listen(addr); err != nil {
socket.Close()
return nil, err
}
return socket, nil
}
func (receiver *SPReceiver) Receive(address string) error {
var sock mangos.Socket
var err error
if sock, err = rep.NewSocket(); err != nil {
return err
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err := sock.Listen(address); err != nil {
return err
}
for {
if _, err := sock.Recv(); err != nil {
return err
}
}
return nil
}
func TestDeviceReqRep(t *testing.T) {
s1, err := req.NewSocket()
if err != nil {
t.Errorf("Failed to open S1: %v", err)
return
}
defer s1.Close()
s2, err := rep.NewSocket()
if err != nil {
t.Errorf("Failed to open S2: %v", err)
return
}
defer s2.Close()
switch err := mangos.Device(s1, s2); err {
case nil:
t.Logf("Matching req/rep ok!")
return
default:
t.Errorf("Got unexpected err: %v", err)
return
}
}
func TestTtlDrop(t *testing.T) {
nhop := 3
srep := make([]mangos.Socket, 0, nhop)
sreq := make([]mangos.Socket, 0, nhop)
inp := inproc.NewTransport()
for i := 0; i < nhop; i++ {
s, err := rep.NewSocket()
if err != nil {
t.Errorf("Failed to make REP: %v", err)
return
}
defer s.Close()
s.AddTransport(inp)
err = s.Listen(AddrTestInp + fmt.Sprintf("HOP%d", i))
if err != nil {
t.Errorf("Failed listen: %v", err)
return
}
err = s.SetOption(mangos.OptionRaw, true)
if err != nil {
t.Errorf("Failed set raw mode: %v", err)
return
}
srep = append(srep, s)
}
for i := 0; i < nhop; i++ {
s, err := req.NewSocket()
if err != nil {
t.Errorf("Failed to make REQ: %v", err)
return
}
defer s.Close()
s.AddTransport(inp)
err = s.Dial(AddrTestInp + fmt.Sprintf("HOP%d", i))
if err != nil {
t.Errorf("Failed dial: %v", err)
return
}
sreq = append(sreq, s)
}
// Now make the device chain
for i := 0; i < nhop-1; i++ {
err := mangos.Device(srep[i], sreq[i+1])
if err != nil {
t.Errorf("Device failed: %v", err)
return
}
}
// At this point, we can issue requests on sreq[0], and read them from
// srep[nhop-1].
rq := sreq[0]
rp := srep[nhop-1]
err := rp.SetOption(mangos.OptionRecvDeadline, time.Millisecond*20)
if err != nil {
t.Errorf("Failed set recv deadline")
return
}
if err = rq.Send([]byte("GOOD")); err != nil {
t.Errorf("Failed first send: %v", err)
return
}
v, err := rp.Recv()
if err != nil {
t.Errorf("Failed first recv: %v", err)
return
} else if !bytes.Equal(v, []byte("GOOD")) {
t.Errorf("Got wrong message: %v", v)
return
} else {
t.Logf("Got good message: %v", v)
}
// Now try setting the option
err = rp.SetOption(mangos.OptionTtl, nhop-1)
if err != nil {
t.Errorf("Failed set TTL: %v", err)
return
}
if err = rq.Send([]byte("DROP")); err != nil {
t.Errorf("Failed send drop: %v", err)
return
}
v, err = rp.Recv()
switch err {
case mangos.ErrRecvTimeout: // expected
t.Logf("TTL honored")
case nil:
t.Errorf("Message not dropped: %v", v)
default:
t.Errorf("Got unexpected error: %v", err)
}
}
// This test verifies that we can use stock http server instances with
// our own websocket handler.
func TestWebsockHandler(t *testing.T) {
sockReq, _ := req.NewSocket()
sockRep, _ := rep.NewSocket()
tran := NewTransport()
l, e := tran.NewListener("ws://127.0.0.1:3337/mysock", sockReq)
if e != nil {
t.Errorf("Failed new Listener: %v", e)
return
}
hi, e := l.GetOption(OptionWebSocketHandler)
if e != nil {
t.Errorf("Failed get WebSocketHandler: %v", e)
}
handler := hi.(http.Handler)
mux := http.NewServeMux()
mux.HandleFunc("/bogus", bogusHandler)
mux.Handle("/mysock", handler)
// Note that we are *counting* on this to die gracefully when our
// program exits. There appears to be no way to shutdown http
// instances gracefully.
go http.ListenAndServe("127.0.0.1:3337", mux)
d, e := tran.NewDialer("ws://127.0.0.1:3337/bogus", sockRep)
if e != nil {
t.Errorf("Failed new Dialer: %v", e)
return
}
defer l.Close()
p, e := d.Dial()
if p != nil {
defer p.Close()
}
if e == nil {
t.Errorf("Dial passed, when should not have!")
return
}
t.Logf("Got expected error %v", e)
// Now let's try to use http client.
resp, err := http.Get("http://127.0.0.1:3337/bogus")
if err != nil {
t.Errorf("Get of boguspath failed: %v", err)
return
}
if resp.StatusCode != 200 {
t.Errorf("Response code wrong: %d", resp.StatusCode)
return
}
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
t.Errorf("ReadAll Failed: %v", err)
return
}
if string(body) != bogusstr {
t.Errorf("Results mismatch: %s != %s", string(body), bogusstr)
}
t.Logf("Got body: %s", string(body))
}
func benchmarkReq(t *testing.B, url string, size int) {
if strings.HasPrefix(url, "ipc://") && runtime.GOOS == "windows" {
t.Skip("IPC not supported on Windows")
return
}
srvopts := make(map[string]interface{})
cliopts := make(map[string]interface{})
if strings.HasPrefix(url, "wss://") || strings.HasPrefix(url, "tls+tcp://") {
srvopts[mangos.OptionTLSConfig] = srvCfg
cliopts[mangos.OptionTLSConfig] = cliCfg
}
srvrdy := make(chan struct{})
srvsock, err := rep.NewSocket()
if err != nil || srvsock == nil {
t.Errorf("Failed creating server socket: %v", err)
return
}
defer srvsock.Close()
all.AddTransports(srvsock)
clisock, err := req.NewSocket()
if err != nil || clisock == nil {
t.Errorf("Failed creating client socket: %v", err)
return
}
defer clisock.Close()
all.AddTransports(clisock)
go func() {
var err error
var msg *mangos.Message
if err = srvsock.ListenOptions(url, srvopts); err != nil {
t.Errorf("Server listen failed: %v", err)
return
}
close(srvrdy)
// echo server
for {
if msg, err = srvsock.RecvMsg(); err != nil {
return
}
if err = srvsock.SendMsg(msg); err != nil {
t.Errorf("Server send failed: %v", err)
return
}
}
}()
if err = clisock.DialOptions(url, cliopts); err != nil {
t.Errorf("Client dial failed: %v", err)
return
}
<-srvrdy
time.Sleep(time.Millisecond * 1000)
t.ResetTimer()
msg := make([]byte, size)
for i := 0; i < t.N; i++ {
if err = clisock.Send(msg); err != nil {
t.Errorf("Client send failed: %v", err)
return
}
if msg, err = clisock.Recv(); err != nil {
t.Errorf("Client receive failed: %v", err)
return
}
}
if size > 128 {
t.SetBytes(int64(size))
}
t.StopTimer()
}
func TestPortHook(t *testing.T) {
t.Logf("Testing Add Hook")
srvtest := &hooktest{allow: true, t: t}
clitest := &hooktest{allow: true, t: t}
addr := AddrTestTCP
srvtest.expect = []hookinfo{{
action: mangos.PortActionAdd,
addr: addr,
server: true,
isopen: true,
}, {
action: mangos.PortActionRemove,
addr: addr,
server: true,
isopen: false,
}}
clitest.expect = []hookinfo{{
action: mangos.PortActionAdd,
addr: addr,
server: false,
isopen: true,
}, {
action: mangos.PortActionRemove,
addr: addr,
server: false,
isopen: false,
}}
sockreq, err := req.NewSocket()
if err != nil {
t.Errorf("NewSocket failed: %v", err)
return
}
defer sockreq.Close()
sockreq.AddTransport(tcp.NewTransport())
if sockreq.SetPortHook(clitest.Hook) != nil {
t.Errorf("SetPortHook result not nil!")
return
}
d, err := sockreq.NewDialer(addr, nil)
if err != nil {
t.Errorf("NewDialer failed: %v", err)
return
}
sockrep, err := rep.NewSocket()
if err != nil {
t.Errorf("NewSocket failed: %v", err)
return
}
defer sockrep.Close()
sockrep.AddTransport(tcp.NewTransport())
if sockrep.SetPortHook(srvtest.Hook) != nil {
t.Errorf("SetPortHook result not nil!")
return
}
l, err := sockrep.NewListener(addr, nil)
if err != nil {
t.Errorf("NewListener failed: %v", err)
return
}
if err := l.Listen(); err != nil {
t.Errorf("Listen failed: %v", err)
return
}
if err := d.Dial(); err != nil {
t.Errorf("Dial failed: %v", err)
return
}
// wait a second for connection to establish
// could also issue a req/rep...
t.Logf("Waiting a bit...")
time.Sleep(100 * time.Millisecond)
d.Close()
l.Close()
// shut down the server
sockrep.Close()
sockreq.Close()
clitest.Lock()
defer clitest.Unlock()
srvtest.Lock()
defer srvtest.Unlock()
for i, info := range clitest.expect {
t.Logf("Exp C[%d]: %s", i, info.String())
}
for i, info := range clitest.calls {
t.Logf("Got C[%d]: %s", i, info.String())
}
for i, info := range srvtest.expect {
t.Logf("Exp S[%d]: %s", i, info.String())
}
for i, info := range srvtest.calls {
t.Logf("Got S[%d]: %s", i, info.String())
}
if len(srvtest.calls) != len(srvtest.expect) {
t.Errorf("Server got wrong # calls, %d != %d", len(srvtest.calls), len(srvtest.expect))
return
}
for i := range srvtest.calls {
if srvtest.calls[i].String() != srvtest.expect[i].String() {
t.Errorf("Server hook %d wrong: %s != %s", i)
}
}
if len(clitest.calls) != len(clitest.expect) {
t.Errorf("Client got wrong # calls, %d != %d", len(clitest.calls), len(clitest.expect))
return
}
for i := range clitest.calls {
if clitest.calls[i].String() != clitest.expect[i].String() {
t.Errorf("Server hook %d wrong: %s != %s", i)
}
}
}
// limitations under the License.
package tcp
import (
"bytes"
"testing"
"time"
"github.com/gdamore/mangos"
"github.com/gdamore/mangos/protocol/rep"
"github.com/gdamore/mangos/protocol/req"
)
var tran = NewTransport()
var sockRep, _ = rep.NewSocket()
var sockReq, _ = req.NewSocket()
func TestTCPListenAndAccept(t *testing.T) {
addr := "tcp://127.0.0.1:3333"
t.Logf("Establishing accepter")
l, err := tran.NewListener(addr, sockRep)
if err != nil {
t.Errorf("NewListener failed: %v", err)
return
}
defer l.Close()
if err = l.Listen(); err != nil {
t.Errorf("Listen failed: %v", err)
return
}
func testMaxRx(t *testing.T, addr string, tran mangos.Transport) {
maxrx := 100
rp, err := rep.NewSocket()
if err != nil {
t.Errorf("Failed to make REP: %v", err)
return
}
defer rp.Close()
rp.AddTransport(tran)
// Now try setting the option
err = rp.SetOption(mangos.OptionMaxRecvSize, maxrx)
if err != nil {
t.Errorf("Failed set MaxRecvSize: %v", err)
return
}
// At this point, we can issue requests on rq, and read them from rp.
if err = rp.SetOption(mangos.OptionRecvDeadline, time.Millisecond*20); err != nil {
t.Errorf("Failed set recv deadline")
return
}
if err = rp.Listen(addr); err != nil {
t.Errorf("Failed listen: %v", err)
return
}
rq, err := req.NewSocket()
if err != nil {
t.Errorf("Failed to make REQ: %v", err)
return
}
defer rq.Close()
rq.AddTransport(tran)
if err = rq.Dial(addr); err != nil {
t.Errorf("Failed dial: %v", err)
return
}
time.Sleep(time.Millisecond * 10)
msg1 := make([]byte, maxrx+1)
msg2 := make([]byte, maxrx+1)
for i := 0; i < len(msg1); i++ {
msg1[i] = 'A' + byte(i%26)
msg2[i] = 'A' + byte(i%26)
}
// NB: maxrx *includes* the header.
if err = rq.Send(msg1[:maxrx-8]); err != nil {
t.Errorf("Failed first send: %v", err)
return
}
v, err := rp.Recv()
if err != nil {
t.Errorf("Failed first recv: %v", err)
return
} else if !bytes.Equal(v, msg2[:maxrx-8]) {
t.Errorf("Got wrong message: %v", v)
return
} else {
t.Logf("Got good message")
}
if err = rq.Send(msg1); err != nil {
t.Errorf("Failed send drop: %v", err)
return
}
v, err = rp.Recv()
switch err {
case mangos.ErrRecvTimeout: // expected
t.Logf("MaxRx honored")
case nil:
t.Errorf("Message not dropped: %v", v)
default:
t.Errorf("Got unexpected error: %v", err)
}
}