go-gemini/request.go

92 lines
2.4 KiB
Go
Raw Normal View History

2020-10-24 13:15:32 -06:00
package gemini
2020-10-21 15:07:28 -06:00
import (
"bufio"
"crypto/tls"
"io"
2020-10-21 15:07:28 -06:00
"net/url"
)
2021-02-14 13:50:41 -07:00
// A Request represents a Gemini request received by a server or to be sent
// by a client.
2020-10-21 15:07:28 -06:00
type Request struct {
2021-02-23 16:45:58 -07:00
// URL specifies the URL being requested.
2020-10-21 15:07:28 -06:00
URL *url.URL
2021-02-14 17:02:34 -07:00
// For client requests, Host optionally specifies the server to
// connect to. It may be of the form "host" or "host:port".
2021-02-14 17:02:34 -07:00
// If empty, the value of URL.Host is used.
// For international domain names, Host may be in Punycode or
// Unicode form. Use golang.org/x/net/idna to convert it to
// either format if needed.
2021-02-23 16:45:58 -07:00
// This field is ignored by the Gemini server.
2020-10-21 15:07:28 -06:00
Host string
// For client requests, Certificate optionally specifies the
// TLS certificate to present to the other side of the connection.
// This field is ignored by the Gemini server.
2020-10-21 15:07:28 -06:00
Certificate *tls.Certificate
}
2021-02-14 17:02:34 -07:00
// NewRequest returns a new request.
// The returned Request is suitable for use with Client.Do.
//
// Callers should be careful that the URL query is properly escaped.
// See the documentation for QueryEscape for more information.
2020-10-21 15:07:28 -06:00
func NewRequest(rawurl string) (*Request, error) {
u, err := url.Parse(rawurl)
if err != nil {
return nil, err
}
return &Request{URL: u}, nil
2020-10-21 15:07:28 -06:00
}
2021-02-14 13:50:41 -07:00
// ReadRequest reads and parses an incoming request from r.
//
// ReadRequest is a low-level function and should only be used
// for specialized applications; most code should use the Server
// to read requests and handle them via the Handler interface.
func ReadRequest(r io.Reader) (*Request, error) {
// Read URL
r = io.LimitReader(r, 1026)
2021-02-14 13:50:41 -07:00
br := bufio.NewReaderSize(r, 1026)
rawurl, err := br.ReadString('\r')
if err != nil {
return nil, err
}
// Read terminating line feed
if b, err := br.ReadByte(); err != nil {
return nil, err
} else if b != '\n' {
return nil, ErrInvalidRequest
}
// Trim carriage return
rawurl = rawurl[:len(rawurl)-1]
// Validate URL
if len(rawurl) > 1024 {
return nil, ErrInvalidRequest
}
u, err := url.Parse(rawurl)
if err != nil {
return nil, err
}
return &Request{URL: u}, nil
}
2021-02-14 13:50:41 -07:00
// Write writes a Gemini request in wire format.
// This method consults the request URL only.
2021-02-18 19:58:35 -07:00
func (r *Request) Write(w io.Writer) error {
bw := bufio.NewWriterSize(w, 1026)
url := r.URL.String()
if len(url) > 1024 {
return ErrInvalidRequest
}
2021-02-18 19:58:35 -07:00
if _, err := bw.WriteString(url); err != nil {
2020-10-21 15:07:28 -06:00
return err
}
2021-02-18 19:58:35 -07:00
if _, err := bw.Write(crlf); err != nil {
2020-10-21 15:07:28 -06:00
return err
}
2021-02-18 19:58:35 -07:00
return bw.Flush()
2020-10-21 15:07:28 -06:00
}