init

go-gost/x/internal/net/http/http.go

@@ -0,0 +1,72 @@
+package http
+
+import (
+	"bytes"
+	"io"
+	"net"
+	"net/http"
+	"strings"
+)
+
+func GetClientIP(req *http.Request) net.IP {
+	if req == nil {
+		return nil
+	}
+	// cloudflare CDN
+	sip := req.Header.Get("CF-Connecting-IP")
+	if sip == "" {
+		ss := strings.Split(req.Header.Get("X-Forwarded-For"), ",")
+		if len(ss) > 0 && ss[0] != "" {
+			sip = ss[0]
+		}
+	}
+	if sip == "" {
+		sip = req.Header.Get("X-Real-Ip")
+	}
+
+	return net.ParseIP(sip)
+}
+
+type Body struct {
+	r          io.ReadCloser
+	buf        bytes.Buffer
+	length     int64
+	recordSize int
+}
+
+func NewBody(r io.ReadCloser, maxRecordSize int) *Body {
+	p := &Body{
+		r:          r,
+		recordSize: maxRecordSize,
+	}
+
+	return p
+}
+
+func (p *Body) Read(b []byte) (n int, err error) {
+	n, err = p.r.Read(b)
+	p.length += int64(n)
+
+	if p.recordSize > 0 {
+		b = b[:n]
+		if n > p.recordSize {
+			b = b[:p.recordSize]
+		}
+		p.buf.Write(b)
+		p.recordSize -= n
+	}
+
+	return
+}
+
+func (p *Body) Close() error {
+	return p.r.Close()
+}
+
+func (p *Body) Content() []byte {
+	return p.buf.Bytes()
+}
+
+func (p *Body) Length() int64 {
+	return p.length
+}