serve_test.go

// +build !race

package gig

import (
	"crypto/tls"
	"io"
	"net"
	"syscall"
	"testing"
	"time"

	"github.com/matryer/is"
)

type errorListener struct {
	errs []error
}

func (l *errorListener) Accept() (c net.Conn, err error) {
	if len(l.errs) == 0 {
		return nil, io.EOF
	}

	err = l.errs[0]
	l.errs = l.errs[1:]

	return
}

func (l *errorListener) Close() error {
	return nil
}

func (l *errorListener) Addr() net.Addr {
	return &FakeAddr{}
}

func TestServe_NetError(t *testing.T) {
	is := is.New(t)

	ln := &errorListener{[]error{
		&net.OpError{
			Op:  "accept",
			Err: syscall.EMFILE,
		}}}
	g := New()
	g.listener = ln
	err := g.serve()
	is.Equal(io.EOF, err)
}

func TestServe(t *testing.T) {
	is := is.New(t)
	g := New()

	go func() {
		_ = g.Run("127.0.0.1:0", "_fixture/certs/cert.pem", "_fixture/certs/key.pem")
	}()
	time.Sleep(200 * time.Millisecond)

	addr := g.listener.Addr().String()
	conn, err := tls.Dial("tcp", addr, &tls.Config{InsecureSkipVerify: true})
	is.NoErr(err)
	_, err = conn.Write([]byte("/test\r\n"))
	is.NoErr(err)

	buf := make([]byte, 15)
	n, err := conn.Read(buf)
	is.NoErr(err)

	is.Equal("51 Not Found\r\n\x00", string(buf))
	is.Equal(14, n)

	g.Close()
}

func TestServe_SlowClient_Read(t *testing.T) {
	is := is.New(t)

	g := New()
	g.ReadTimeout = 1 * time.Millisecond

	go func() {
		_ = g.Run("127.0.0.1:0", "_fixture/certs/cert.pem", "_fixture/certs/key.pem")
	}()
	time.Sleep(200 * time.Millisecond)

	addr := g.listener.Addr().String()
	conn, err := tls.Dial("tcp", addr, &tls.Config{InsecureSkipVerify: true})
	is.NoErr(err)

	time.Sleep(200 * time.Millisecond) // client sleeps before sending request

	_, err = conn.Write([]byte("/test\r\n"))

	is.True(err != nil)

	g.Close()
}

func TestServe_SlowClient_Write(t *testing.T) {
	is := is.New(t)

	g := New()
	g.WriteTimeout = 1 * time.Millisecond

	go func() {
		err := g.Run("127.0.0.1:0", "_fixture/certs/cert.pem", "_fixture/certs/key.pem")
		if err != ErrServerClosed { // Prevent the test to fail after closing the servers
			is.NoErr(err)
		}
	}()
	time.Sleep(200 * time.Millisecond)

	addr := g.listener.Addr().String()
	conn, err := tls.Dial("tcp", addr, &tls.Config{InsecureSkipVerify: true})
	is.NoErr(err)
	_, err = conn.Write([]byte("/test\r\n"))
	is.NoErr(err)

	conn.Close() // client closes connection before reading response

	g.Close()
}

func TestServe_Overflow(t *testing.T) {
	is := is.New(t)

	g := New()

	go func() {
		_ = g.Run("127.0.0.1:0", "_fixture/certs/cert.pem", "_fixture/certs/key.pem")
	}()
	time.Sleep(200 * time.Millisecond)

	addr := g.listener.Addr().String()
	conn, err := tls.Dial("tcp", addr, &tls.Config{InsecureSkipVerify: true})
	is.NoErr(err)

	request := make([]byte, 2000)
	_, _ = conn.Write(request)

	buf := make([]byte, 23)
	n, err := conn.Read(buf)
	is.NoErr(err)

	is.Equal("59 Request too long!\r\n\x00", string(buf))
	is.Equal(22, n)

	g.Close()
}

func TestServe_NotGemini(t *testing.T) {
	is := is.New(t)
	g := New()

	go func() {
		_ = g.Run("127.0.0.1:0", "_fixture/certs/cert.pem", "_fixture/certs/key.pem")
	}()
	time.Sleep(200 * time.Millisecond)

	addr := g.listener.Addr().String()
	conn, err := tls.Dial("tcp", addr, &tls.Config{InsecureSkipVerify: true})
	is.NoErr(err)

	_, err = conn.Write([]byte("http://google.com\r\n"))
	is.NoErr(err)

	buf := make([]byte, 40)
	n, err := conn.Read(buf)
	is.NoErr(err)

	is.Equal("59 No proxying to non-Gemini content!\r\n\x00", string(buf))
	is.Equal(39, n)

	g.Close()
}

func TestServe_NotURL(t *testing.T) {
	is := is.New(t)
	g := New()

	go func() {
		_ = g.Run("127.0.0.1:0", "_fixture/certs/cert.pem", "_fixture/certs/key.pem")
	}()
	time.Sleep(200 * time.Millisecond)

	addr := g.listener.Addr().String()
	conn, err := tls.Dial("tcp", addr, &tls.Config{InsecureSkipVerify: true})
	is.NoErr(err)

	_, err = conn.Write([]byte("::::::\r\n"))
	is.NoErr(err)

	buf := make([]byte, 24)
	n, err := conn.Read(buf)
	is.NoErr(err)

	is.Equal("59 Error parsing URL!\r\n\x00", string(buf))
	is.Equal(23, n)

	g.Close()
}