💾 Archived View for source.community › ckaznocha › gemini › blob › main › request.go captured on 2024-05-26 at 14:51:57. Gemini links have been rewritten to link to archived content

-=-=-=-=-=-=-

                                                         .
,-. ,-. . . ,-. ,-. ,-.    ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | |   |   |-'    |   | | | | | | | | | | | | | |  | |
`-' `-' `-^ '   `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
                                                             /|
                                                            `-'

ckaznocha / gemini

git clone https://source.community/ckaznocha/gemini.git

Branches

Log

Tree

/request.go (main)

↑ /

blob

View raw contents of /request.go (main)

────╮
   1│ package gemini
   2│ 
   3│ import (
   4│ 	"bufio"
   5│ 	"crypto/x509/pkix"
   6│ 	"fmt"
   7│ 	"net/textproto"
   8│ 	"reflect"
   9│ 	"sync"
  10│ 	"unsafe"
  11│ )
  12│ 
  13│ const maxRequestLength = 1024
  14│ 
  15│ // Request is a Gemini request.
  16│ type Request struct {
  17│ 	Subject    *pkix.Name
  18│ 	URI        *URI
  19│ 	RemoteAddr string
  20│ }
  21│ 
  22│ var requestPool = sync.Pool{
  23│ 	New: func() interface{} {
  24│ 		b := make([]byte, 0, maxRequestLength)
  25│ 
  26│ 		return &b
  27│ 	},
  28│ }
  29│ 
  30│ // ReadRequest reads and parses a Gemini request from buffer. It returns an
  31│ // error if the request could not be read or was  malformed.
  32│ func ReadRequest(b *bufio.Reader) (*Request, error) {
  33│ 	u, ok := requestPool.Get().(*[]byte)
  34│ 	if !ok {
  35│ 		return nil, fmt.Errorf("%w: %s", ErrRequestRead, "request pool returned non-byte slice")
  36│ 	}
  37│ 
  38│ 	defer requestPool.Put(u)
  39│ 
  40│ 	sh := (*reflect.SliceHeader)(unsafe.Pointer(u)) //nolint:gosec // This is safe.
  41│ 	sh.Len = 0
  42│ 
  43│ 	cr := false
  44│ 
  45│ 	for {
  46│ 		b, err := b.ReadByte()
  47│ 		if err != nil {
  48│ 			return nil, fmt.Errorf("%w: %s", ErrRequestRead, err)
  49│ 		}
  50│ 
  51│ 		if b == '\r' {
  52│ 			cr = true
  53│ 
  54│ 			continue
  55│ 		}
  56│ 
  57│ 		if cr && b == '\n' {
  58│ 			break
  59│ 		}
  60│ 
  61│ 		cr = false
  62│ 
  63│ 		*u = append(*u, b)
  64│ 		if len(*u) > maxRequestLength {
  65│ 			return nil, ErrMaxRequestLengthExceeded
  66│ 		}
  67│ 	}
  68│ 
  69│ 	*u = textproto.TrimBytes(*u)
  70│ 
  71│ 	uri, err := ParseRequestURI(*(*string)(unsafe.Pointer(u))) //nolint:gosec // This is safe.
  72│ 	if err != nil {
  73│ 		return nil, err
  74│ 	}
  75│ 
  76│ 	return &Request{URI: uri}, nil
  77│ }
────╯

· · ·

info@source.community