Brendan LE GLAUNEC
5849898283
Unit tests functional and coverage back to 100% Add more routes to dictionary, add more credentials, add default port 5554, rename cameradar logs ENV variable, improve unit test readability, remove tmp file
124 lines
2.2 KiB
Go
124 lines
2.2 KiB
Go
// Copyright 2013 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package ssh
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
type server struct {
|
|
*ServerConn
|
|
chans <-chan NewChannel
|
|
}
|
|
|
|
func newServer(c net.Conn, conf *ServerConfig) (*server, error) {
|
|
sconn, chans, reqs, err := NewServerConn(c, conf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go DiscardRequests(reqs)
|
|
return &server{sconn, chans}, nil
|
|
}
|
|
|
|
func (s *server) Accept() (NewChannel, error) {
|
|
n, ok := <-s.chans
|
|
if !ok {
|
|
return nil, io.EOF
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
func sshPipe() (Conn, *server, error) {
|
|
c1, c2, err := netPipe()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
clientConf := ClientConfig{
|
|
User: "user",
|
|
HostKeyCallback: InsecureIgnoreHostKey(),
|
|
}
|
|
serverConf := ServerConfig{
|
|
NoClientAuth: true,
|
|
}
|
|
serverConf.AddHostKey(testSigners["ecdsa"])
|
|
done := make(chan *server, 1)
|
|
go func() {
|
|
server, err := newServer(c2, &serverConf)
|
|
if err != nil {
|
|
done <- nil
|
|
}
|
|
done <- server
|
|
}()
|
|
|
|
client, _, reqs, err := NewClientConn(c1, "", &clientConf)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
server := <-done
|
|
if server == nil {
|
|
return nil, nil, errors.New("server handshake failed.")
|
|
}
|
|
go DiscardRequests(reqs)
|
|
|
|
return client, server, nil
|
|
}
|
|
|
|
func BenchmarkEndToEnd(b *testing.B) {
|
|
b.StopTimer()
|
|
|
|
client, server, err := sshPipe()
|
|
if err != nil {
|
|
b.Fatalf("sshPipe: %v", err)
|
|
}
|
|
|
|
defer client.Close()
|
|
defer server.Close()
|
|
|
|
size := (1 << 20)
|
|
input := make([]byte, size)
|
|
output := make([]byte, size)
|
|
b.SetBytes(int64(size))
|
|
done := make(chan int, 1)
|
|
|
|
go func() {
|
|
newCh, err := server.Accept()
|
|
if err != nil {
|
|
b.Fatalf("Client: %v", err)
|
|
}
|
|
ch, incoming, err := newCh.Accept()
|
|
go DiscardRequests(incoming)
|
|
for i := 0; i < b.N; i++ {
|
|
if _, err := io.ReadFull(ch, output); err != nil {
|
|
b.Fatalf("ReadFull: %v", err)
|
|
}
|
|
}
|
|
ch.Close()
|
|
done <- 1
|
|
}()
|
|
|
|
ch, in, err := client.OpenChannel("speed", nil)
|
|
if err != nil {
|
|
b.Fatalf("OpenChannel: %v", err)
|
|
}
|
|
go DiscardRequests(in)
|
|
|
|
b.ResetTimer()
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
if _, err := ch.Write(input); err != nil {
|
|
b.Fatalf("WriteFull: %v", err)
|
|
}
|
|
}
|
|
ch.Close()
|
|
b.StopTimer()
|
|
|
|
<-done
|
|
}
|