121 lines
2.5 KiB
Go
121 lines
2.5 KiB
Go
package request
|
|
|
|
import (
|
|
"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)
|
|
|
|
case header.ORIGIN: fmt.Printf("[origin] ")
|
|
err = r.extractOrigin(head.Values)
|
|
|
|
case header.UPGRADE: fmt.Printf("[upgrade] ")
|
|
err = r.checkUpgrade(head.Values)
|
|
|
|
case header.CONNECTION: fmt.Printf("[connection] ")
|
|
err = r.checkConnection(head.Values)
|
|
|
|
case header.WSVERSION: fmt.Printf("[sec-websocket-version] ")
|
|
err = r.checkVersion(head.Values)
|
|
|
|
case header.WSKEY: fmt.Printf("[sec-websocket-key] ")
|
|
err = r.extractKey(head.Values)
|
|
|
|
case header.WSPROTOCOL: fmt.Printf("[sec-websocket-protocol] ")
|
|
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
|
|
|
|
}
|
|
|
|
|
|
|
|
// isComplete returns whether the Upgrade Request
|
|
// is complete (no missing required item)
|
|
func (r T) isComplete() error {
|
|
|
|
/* (1) Request-Line */
|
|
if !r.first {
|
|
return fmt.Errorf("Missing HTTP Request-Line");
|
|
}
|
|
|
|
/* (2) Host */
|
|
if len(r.host) == 0 {
|
|
return fmt.Errorf("Missing 'Host' header")
|
|
}
|
|
|
|
/* (3) Origin */
|
|
if len(r.origin) == 0 {
|
|
return fmt.Errorf("Missing 'Origin' header")
|
|
}
|
|
|
|
/* (4) Connection */
|
|
if !r.hasConnection {
|
|
return fmt.Errorf("Missing 'Connection' header");
|
|
}
|
|
|
|
/* (5) Upgrade */
|
|
if !r.hasUpgrade {
|
|
return fmt.Errorf("Missing 'Upgrade' header");
|
|
}
|
|
|
|
/* (6) Sec-WebSocket-Version */
|
|
if !r.hasVersion {
|
|
return fmt.Errorf("Missing 'Sec-WebSocket-Version' header");
|
|
}
|
|
|
|
/* (7) Sec-WebSocket-Key */
|
|
if len(r.key) < 1 {
|
|
return fmt.Errorf("Missing 'Sec-WebSocket-Key' header");
|
|
}
|
|
|
|
return nil
|
|
|
|
} |