// Test that we remove a stream from the session if both sides half-close
func TestRemoveAfterHalfClose(t *testing.T) {
local, remote := newFakeConnPair()
remote.Discard()
s := NewSession(local, NewStream, false)
trans := frame.NewBasicTransport(remote)
// open stream
str, err := s.Open()
if err != nil {
t.Fatalf("failed to open stream")
}
// half close remote side (true means half-close)
fData := frame.NewWStreamData()
if err := fData.Set(str.Id(), []byte{0x1, 0x2, 0x3}, true); err != nil {
t.Fatalf("Failed to set data frame")
}
if err := trans.WriteFrame(fData); err != nil {
t.Fatalf("Failed to write data frame")
}
// half-close local side
str.HalfClose([]byte{0xFF, 0xFE, 0xFD, 0xFC})
// yield so the stream can process
time.Sleep(0)
// verify stream is removed
if stream, ok := s.(*Session).streams.Get(str.Id()); ok {
t.Fatalf("Expected stream %d to be removed after both sides half-closed, but found: %v!", str.Id(), stream)
}
}
func TestWrongServerParity(t *testing.T) {
t.Parallel()
local, remote := newFakeConnPair()
// true for a client session
s := NewSession(local, fakeStreamFactory, true)
// don't need the remote output
remote.Discard()
// 300 is even, and only servers send even stream ids
f := frame.NewWStreamSyn()
f.Set(301, 0, 0, false)
// send the frame into the session
trans := frame.NewBasicTransport(remote)
trans.WriteFrame(f)
// wait for failure
code, err, _ := s.Wait()
if code != frame.ProtocolError {
t.Errorf("Session not terminated with protocol error. Got %d, expected %d. Session error: %v", code, frame.ProtocolError, err)
}
if !local.closed {
t.Errorf("Session transport not closed after protocol failure.")
}
}
func TestSynLowId(t *testing.T) {
t.Parallel()
local, remote := newFakeConnPair()
// don't need the remote output
remote.Discard()
// true for a client session
s := NewSession(local, fakeStreamFactory, true)
// Start a stream
f := frame.NewWStreamSyn()
f.Set(302, 0, 0, false)
// send the frame into the session
trans := frame.NewBasicTransport(remote)
trans.WriteFrame(f)
// accept it
s.Accept()
// Start a closed stream at a lower id number
f.Set(300, 0, 0, false)
// send the frame into the session
trans.WriteFrame(f)
code, err, _ := s.Wait()
if code != frame.ProtocolError {
t.Errorf("Session not terminated with protocol error, got %d expected %d. Error: %v", code, frame.ProtocolError, err)
}
}
func NewSession(conn net.Conn, newStream streamFactory, isClient bool) ISession {
sess := &Session{
conn: conn,
transport: frame.NewBasicTransport(conn),
streams: NewConcurrentStreamMap(),
local: halfState{lastId: firstStream},
remote: halfState{lastId: firstStream},
syn: frame.NewWStreamSyn(),
diebit: 0,
defaultWindowSize: defaultWindowSize,
accept: make(chan stream, defaultAcceptQueueDepth),
newStream: newStream,
dead: make(chan deadReason, 1), // don't block die() if there is no Wait call
}
if isClient {
sess.isLocal = sess.isClient
sess.local.lastId += 1
} else {
sess.isLocal = sess.isServer
sess.remote.lastId += 1
}
go sess.reader()
return sess
}
func TestDataAfterRst(t *testing.T) {
local, remote := newFakeConnPair()
_ = NewSession(local, NewStream, false)
trans := frame.NewBasicTransport(remote)
// make sure that we get an RST STREAM_CLOSED
done := make(chan int)
go func() {
defer func() { done <- 1 }()
f, err := trans.ReadFrame()
if err != nil {
t.Errorf("Failed to read frame sent from session: %v", err)
return
}
fr, ok := f.(*frame.RStreamRst)
if !ok {
t.Errorf("Frame is not STREAM_RST: %v", f)
return
}
if fr.ErrorCode() != frame.StreamClosed {
t.Errorf("Error code on STREAM_RST is not STREAM_CLOSED. Got %d, expected %d", fr.ErrorCode(), frame.StreamClosed)
return
}
}()
fSyn := frame.NewWStreamSyn()
if err := fSyn.Set(301, 0, 0, false); err != nil {
t.Fatalf("Failed to make syn frame: %v", err)
}
if err := trans.WriteFrame(fSyn); err != nil {
t.Fatalf("Failed to send syn: %v", err)
}
fRst := frame.NewWStreamRst()
if err := fRst.Set(301, frame.Cancel); err != nil {
t.Fatal("Failed to make rst frame: %v", err)
}
if err := trans.WriteFrame(fRst); err != nil {
t.Fatalf("Failed to write rst frame: %v", err)
}
fData := frame.NewWStreamData()
if err := fData.Set(301, []byte{0xa, 0xFF}, false); err != nil {
t.Fatalf("Failed to set data frame")
}
trans.WriteFrame(fData)
<-done
}
func TestFlowControlError(t *testing.T) {
local, remote := newFakeConnPair()
s := NewSession(local, NewStream, false)
s.(*Session).defaultWindowSize = 10
trans := frame.NewBasicTransport(remote)
// make sure that we get an RST FLOW_CONTROL_ERROR
done := make(chan int)
go func() {
defer func() { done <- 1 }()
f, err := trans.ReadFrame()
if err != nil {
t.Errorf("Failed to read frame sent from session: %v", err)
return
}
fr, ok := f.(*frame.RStreamRst)
if !ok {
t.Errorf("Frame is not STREAM_RST: %v", f)
return
}
if fr.ErrorCode() != frame.FlowControlError {
t.Errorf("Error code on STREAM_RST is not FLOW_CONTROL_ERROR. Got %d, expected %d", fr.ErrorCode(), frame.FlowControlError)
return
}
}()
fSyn := frame.NewWStreamSyn()
if err := fSyn.Set(301, 0, 0, false); err != nil {
t.Fatalf("Failed to make syn frame: %v", err)
}
if err := trans.WriteFrame(fSyn); err != nil {
t.Fatalf("Failed to send syn: %v", err)
}
fData := frame.NewWStreamData()
if err := fData.Set(301, make([]byte, 11), false); err != nil {
t.Fatalf("Failed to set data frame")
}
trans.WriteFrame(fData)
<-done
}
func TestTolerateLateFrameAfterRst(t *testing.T) {
local, remote := newFakeConnPair()
s := NewSession(local, NewStream, false)
trans := frame.NewBasicTransport(remote)
// make sure that we don't get any error on a late frame
done := make(chan int)
go func() {
defer func() { done <- 1 }()
// read syn
trans.ReadFrame()
// read rst
trans.ReadFrame()
// should block
if f, err := trans.ReadFrame(); err != nil {
t.Errorf("Error reading frame: %v", err)
} else {
t.Errorf("Got frame that we shouldn't have read: %v. Type: %v", f, f.Type())
}
}()
str, err := s.Open()
if err != nil {
t.Fatalf("failed to open stream")
}
str.(*Stream).resetWith(frame.Cancel, fmt.Errorf("cancel"))
fData := frame.NewWStreamData()
if err := fData.Set(str.Id(), []byte{0x1, 0x2, 0x3}, false); err != nil {
t.Fatalf("Failed to set data frame")
}
trans.WriteFrame(fData)
select {
case <-done:
t.Fatalf("Stream sent response to late DATA frame")
case <-time.After(1 * time.Second):
// ok
}
}
func TestAcceptStream(t *testing.T) {
t.Parallel()
local, remote := newFakeConnPair()
// don't need the remote output
remote.Discard()
// true for a client session
s := NewSession(local, NewStream, true)
defer s.Close()
f := frame.NewWStreamSyn()
f.Set(300, 0, 0, false)
// send the frame into the session
trans := frame.NewBasicTransport(remote)
trans.WriteFrame(f)
done := make(chan int)
go func() {
defer func() { done <- 1 }()
// wait for accept
str, err := s.Accept()
if err != nil {
t.Errorf("Error accepting stream: %v", err)
return
}
if str.Id() != frame.StreamId(300) {
t.Errorf("Stream has wrong id. Expected %d, got %d", str.Id(), 300)
}
}()
select {
case <-time.After(time.Second):
t.Fatalf("Timed out!")
case <-done:
}
}
func TestSendHalfCloseWithZeroWindow(t *testing.T) {
t.Parallel()
local, remote := newFakeConnPair()
s := NewSession(local, NewStream, false)
s.(*Session).defaultWindowSize = 10
go func() {
trans := frame.NewBasicTransport(remote)
f, err := trans.ReadFrame()
if err != nil {
t.Errorf("Failed to read next frame: %v", err)
return
}
_, ok := f.(*frame.RStreamSyn)
if !ok {
t.Errorf("Wrong frame type. Got %v, expected %v", f.Type(), frame.TypeStreamSyn)
return
}
f, err = trans.ReadFrame()
if err != nil {
t.Errorf("Failed to read next frame: %v", err)
return
}
fr, ok := f.(*frame.RStreamData)
if !ok {
t.Errorf("Wrong frame type. Got %v, expected %v", f.Type(), frame.TypeStreamData)
return
}
if fr.Length() != 10 {
t.Errorf("Wrong data length. Got %v, expected %d", fr.Length(), 10)
return
}
n, err := io.CopyN(ioutil.Discard, fr.Reader(), 10)
if n != 10 {
t.Errorf("Wrong read size. Got %d, expected %d", n, 10)
return
}
f, err = trans.ReadFrame()
if err != nil {
t.Errorf("Failed to read next frame: %v", err)
return
}
fr, ok = f.(*frame.RStreamData)
if !ok {
t.Errorf("Wrong frame type. Got %v, expected %v", f.Type(), frame.TypeStreamData)
return
}
if !fr.Fin() {
t.Errorf("Wrong frame flags. Expected fin flag to be set.")
return
}
trans.ReadFrame()
}()
str, err := s.Open()
if err != nil {
t.Fatalf("Failed to open stream: %v", err)
}
_, err = str.Write(make([]byte, 10))
if err != nil {
t.Fatalf("Failed to write data: %v", err)
}
_, err = str.HalfClose([]byte{})
if err != nil {
t.Fatalf("Failed to half-close with an empty buffer")
}
}