mirror of
https://gitlab.com/pulsechaincom/go-pulse.git
synced 2024-12-25 21:07:17 +00:00
289b30715d
This commit converts the dependency management from Godeps to the vendor folder, also switching the tool from godep to trash. Since the upstream tool lacks a few features proposed via a few PRs, until those PRs are merged in (if), use github.com/karalabe/trash. You can update dependencies via trash --update. All dependencies have been updated to their latest version. Parts of the build system are reworked to drop old notions of Godeps and invocation of the go vet command so that it doesn't run against the vendor folder, as that will just blow up during vetting. The conversion drops OpenCL (and hence GPU mining support) from ethash and our codebase. The short reasoning is that there's noone to maintain and having opencl libs in our deps messes up builds as go install ./... tries to build them, failing with unsatisfied link errors for the C OpenCL deps. golang.org/x/net/context is not vendored in. We expect it to be fetched by the user (i.e. using go get). To keep ci.go builds reproducible the package is "vendored" in build/_vendor.
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package httpu
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"regexp"
|
|
)
|
|
|
|
const (
|
|
DefaultMaxMessageBytes = 2048
|
|
)
|
|
|
|
var (
|
|
trailingWhitespaceRx = regexp.MustCompile(" +\r\n")
|
|
crlf = []byte("\r\n")
|
|
)
|
|
|
|
// Handler is the interface by which received HTTPU messages are passed to
|
|
// handling code.
|
|
type Handler interface {
|
|
// ServeMessage is called for each HTTPU message received. peerAddr contains
|
|
// the address that the message was received from.
|
|
ServeMessage(r *http.Request)
|
|
}
|
|
|
|
// HandlerFunc is a function-to-Handler adapter.
|
|
type HandlerFunc func(r *http.Request)
|
|
|
|
func (f HandlerFunc) ServeMessage(r *http.Request) {
|
|
f(r)
|
|
}
|
|
|
|
// A Server defines parameters for running an HTTPU server.
|
|
type Server struct {
|
|
Addr string // UDP address to listen on
|
|
Multicast bool // Should listen for multicast?
|
|
Interface *net.Interface // Network interface to listen on for multicast, nil for default multicast interface
|
|
Handler Handler // handler to invoke
|
|
MaxMessageBytes int // maximum number of bytes to read from a packet, DefaultMaxMessageBytes if 0
|
|
}
|
|
|
|
// ListenAndServe listens on the UDP network address srv.Addr. If srv.Multicast
|
|
// is true, then a multicast UDP listener will be used on srv.Interface (or
|
|
// default interface if nil).
|
|
func (srv *Server) ListenAndServe() error {
|
|
var err error
|
|
|
|
var addr *net.UDPAddr
|
|
if addr, err = net.ResolveUDPAddr("udp", srv.Addr); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var conn net.PacketConn
|
|
if srv.Multicast {
|
|
if conn, err = net.ListenMulticastUDP("udp", srv.Interface, addr); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if conn, err = net.ListenUDP("udp", addr); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return srv.Serve(conn)
|
|
}
|
|
|
|
// Serve messages received on the given packet listener to the srv.Handler.
|
|
func (srv *Server) Serve(l net.PacketConn) error {
|
|
maxMessageBytes := DefaultMaxMessageBytes
|
|
if srv.MaxMessageBytes != 0 {
|
|
maxMessageBytes = srv.MaxMessageBytes
|
|
}
|
|
for {
|
|
buf := make([]byte, maxMessageBytes)
|
|
n, peerAddr, err := l.ReadFrom(buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
buf = buf[:n]
|
|
|
|
go func(buf []byte, peerAddr net.Addr) {
|
|
// At least one router's UPnP implementation has added a trailing space
|
|
// after "HTTP/1.1" - trim it.
|
|
buf = trailingWhitespaceRx.ReplaceAllLiteral(buf, crlf)
|
|
|
|
req, err := http.ReadRequest(bufio.NewReader(bytes.NewBuffer(buf)))
|
|
if err != nil {
|
|
log.Printf("httpu: Failed to parse request: %v", err)
|
|
return
|
|
}
|
|
req.RemoteAddr = peerAddr.String()
|
|
srv.Handler.ServeMessage(req)
|
|
// No need to call req.Body.Close - underlying reader is bytes.Buffer.
|
|
}(buf, peerAddr)
|
|
}
|
|
}
|
|
|
|
// Serve messages received on the given packet listener to the given handler.
|
|
func Serve(l net.PacketConn, handler Handler) error {
|
|
srv := Server{
|
|
Handler: handler,
|
|
MaxMessageBytes: DefaultMaxMessageBytes,
|
|
}
|
|
return srv.Serve(l)
|
|
}
|