init

2026-10-13 12:46:36 +08:00 · 2025-06-17 12:17:33 +08:00
parent 4a6b2c8e0f
commit 9fa968d07c
648 changed files with 64968 additions and 1621 deletions
@@ -0,0 +1,110 @@
+package recorder
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"time"
+
+	"github.com/go-gost/core/recorder"
+)
+
+const (
+	RecorderServiceHandler       = "recorder.service.handler"
+	RecorderServiceHandlerSerial = "recorder.service.handler.serial"
+	RecorderServiceHandlerTunnel = "recorder.service.handler.tunnel"
+)
+
+type HTTPRequestRecorderObject struct {
+	ContentLength int64       `json:"contentLength"`
+	Header        http.Header `json:"header"`
+	Body          []byte      `json:"body"`
+}
+
+type HTTPResponseRecorderObject struct {
+	ContentLength int64       `json:"contentLength"`
+	Header        http.Header `json:"header"`
+	Body          []byte      `json:"body"`
+}
+
+type HTTPRecorderObject struct {
+	Host       string                     `json:"host"`
+	Method     string                     `json:"method"`
+	Proto      string                     `json:"proto"`
+	Scheme     string                     `json:"scheme"`
+	URI        string                     `json:"uri"`
+	StatusCode int                        `json:"statusCode"`
+	Request    HTTPRequestRecorderObject  `json:"request"`
+	Response   HTTPResponseRecorderObject `json:"response"`
+}
+
+type WebsocketRecorderObject struct {
+	From    string `json:"from"`
+	Fin     bool   `json:"fin"`
+	Rsv1    bool   `json:"rsv1"`
+	Rsv2    bool   `json:"rsv2"`
+	Rsv3    bool   `json:"rsv3"`
+	OpCode  int    `json:"opcode"`
+	Masked  bool   `json:"masked"`
+	MaskKey uint32 `json:"maskKey"`
+	Length  int64  `json:"length"`
+	Payload []byte `json:"payload"`
+}
+
+type TLSRecorderObject struct {
+	ServerName        string `json:"serverName"`
+	CipherSuite       string `json:"cipherSuite"`
+	CompressionMethod uint8  `json:"compressionMethod"`
+	Proto             string `json:"proto"`
+	Version           string `json:"version"`
+	ClientHello       string `json:"clientHello"`
+	ServerHello       string `json:"serverHello"`
+}
+
+type DNSRecorderObject struct {
+	ID       int    `json:"id"`
+	Name     string `json:"name"`
+	Class    string `json:"class"`
+	Type     string `json:"type"`
+	Question string `json:"question"`
+	Answer   string `json:"answer"`
+	Cached   bool   `json:"cached"`
+}
+
+type HandlerRecorderObject struct {
+	Node        string                   `json:"node,omitempty"`
+	Service     string                   `json:"service"`
+	Network     string                   `json:"network"`
+	RemoteAddr  string                   `json:"remote"`
+	LocalAddr   string                   `json:"local"`
+	Host        string                   `json:"host"`
+	Dst         string                   `json:"dst"`
+	Proto       string                   `json:"proto,omitempty"`
+	ClientIP    string                   `json:"clientIP"`
+	ClientID    string                   `json:"clientID,omitempty"`
+	HTTP        *HTTPRecorderObject      `json:"http,omitempty"`
+	Websocket   *WebsocketRecorderObject `json:"websocket,omitempty"`
+	TLS         *TLSRecorderObject       `json:"tls,omitempty"`
+	DNS         *DNSRecorderObject       `json:"dns,omitempty"`
+	Route       string                   `json:"route,omitempty"`
+	InputBytes  uint64                   `json:"inputBytes"`
+	OutputBytes uint64                   `json:"outputBytes"`
+	Redirect    string                   `json:"redirect,omitempty"`
+	Err         string                   `json:"err,omitempty"`
+	SID         string                   `json:"sid"`
+	Duration    time.Duration            `json:"duration"`
+	Time        time.Time                `json:"time"`
+}
+
+func (p *HandlerRecorderObject) Record(ctx context.Context, r recorder.Recorder) error {
+	if p == nil || r == nil || p.Time.IsZero() {
+		return nil
+	}
+
+	data, err := json.Marshal(p)
+	if err != nil {
+		return err
+	}
+
+	return r.Record(ctx, data)
+}