131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package request
|
|
|
|
import (
|
|
"strings"
|
|
"strconv"
|
|
"fmt"
|
|
"git.xdrm.io/gws/internal/http/upgrade/request/parser/header"
|
|
)
|
|
|
|
|
|
// parseHeader parses any http request line
|
|
// (header and request-line)
|
|
func (r *T) parseHeader(b []byte) error {
|
|
|
|
/* (1) First line -> GET {uri} HTTP/{version}
|
|
---------------------------------------------------------*/
|
|
if !r.first {
|
|
|
|
err := r.request.Parse(b)
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("Error while parsing first line: %s", err)
|
|
}
|
|
|
|
r.first = true
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
/* (2) Other lines -> Header-Name: Header-Value
|
|
---------------------------------------------------------*/
|
|
/* (1) Try to parse header */
|
|
head, err := header.Parse(b)
|
|
if err != nil {
|
|
return fmt.Errorf("Error parsing header: %s", err)
|
|
}
|
|
|
|
/* (2) Manage header */
|
|
switch head.Name {
|
|
|
|
case header.HOST: fmt.Printf("[host] ")
|
|
err = r.extractHostPort(head.Values[0])
|
|
|
|
case header.UPGRADE: fmt.Printf("[upgrade] ")
|
|
err = r.checkUpgrade(head.Values)
|
|
|
|
case header.CONNECTION: fmt.Printf("[connection] ")
|
|
err = r.checkConnection(head.Values)
|
|
|
|
case header.WSKEY: fmt.Printf("[sec-websocket-key] ")
|
|
case header.ORIGIN: fmt.Printf("[origin] ")
|
|
case header.WSPROTOCOL: fmt.Printf("[sec-websocket-protocol] ")
|
|
case header.WSVERSION: fmt.Printf("[sec-websocket-version] ")
|
|
default:
|
|
return nil
|
|
}
|
|
|
|
if err != nil { return err }
|
|
|
|
|
|
for i, v := range head.Values {
|
|
if i == 0 {
|
|
fmt.Printf("[ '%s'", v)
|
|
} else {
|
|
fmt.Printf(", '%s'", v)
|
|
}
|
|
}
|
|
fmt.Printf(" ]\n");
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
// checkHost checks and extracts the Host header
|
|
func (r *T) extractHostPort(b []byte) error {
|
|
|
|
split := strings.Split(string(b), ":")
|
|
|
|
r.host = split[0]
|
|
|
|
// no port
|
|
if len(split) < 2 {
|
|
return nil
|
|
}
|
|
|
|
// extract port
|
|
readPort, err := strconv.ParseUint(split[1], 10, 16)
|
|
if err != nil {
|
|
return fmt.Errorf("Cannot read port number '%s'", split[1])
|
|
}
|
|
|
|
r.port = uint16(readPort)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
// checkConnection checks the 'Connection' header
|
|
// it MUST contain 'Upgrade'
|
|
func (r T) checkConnection(bb [][]byte) error {
|
|
|
|
for _, b := range bb {
|
|
|
|
if strings.ToLower( string(b) ) == "upgrade" {
|
|
return nil
|
|
}
|
|
|
|
}
|
|
|
|
return fmt.Errorf("Connection header must be 'Upgrade'")
|
|
|
|
}
|
|
|
|
// checkUpgrade checks the 'Upgrade' header
|
|
// it MUST be 'websocket'
|
|
func (r T) checkUpgrade(bb [][]byte) error {
|
|
|
|
if len(bb) != 1 {
|
|
return fmt.Errorf("Upgrade header must have only 1 element")
|
|
}
|
|
|
|
if strings.ToLower( string(bb[0]) ) == "websocket" {
|
|
return nil
|
|
}
|
|
|
|
return fmt.Errorf("Upgrade header must be 'websocket', got '%s'", bb[0])
|
|
|
|
} |