sync

sepc.go

@@ -0,0 +1,131 @@
+package requests
+
+import (
+	"bytes"
+	"encoding/hex"
+	"errors"
+	"log"
+	"strings"
+
+	"github.com/gospider007/http2"
+	"github.com/gospider007/ja3"
+	"github.com/gospider007/tools"
+)
+
+func (obj *Spec) Map() map[string]any {
+	results := map[string]any{
+		"orderHeaders": obj.OrderHeaders,
+		"raw":          obj.String(),
+	}
+	return results
+}
+func (obj *Spec) Hex() string {
+	return tools.Hex(obj.raw)
+}
+func (obj *Spec) Bytes() []byte {
+	return obj.raw
+}
+func (obj *Spec) String() string {
+	return tools.BytesToString(obj.raw)
+}
+
+type Spec struct {
+	OrderHeaders []string
+	raw          []byte
+}
+
+func ParseSpec(raw []byte) (*Spec, error) {
+	i := bytes.Index(raw, []byte("\r\n\r\n"))
+	if i == -1 {
+		return nil, errors.New("not found \\r\\n")
+	}
+	rawContent := raw[:i]
+	orderHeaders := []string{}
+	for i, line := range bytes.Split(rawContent, []byte("\r\n")) {
+		if i == 0 {
+			continue
+		}
+		ols := bytes.Split(line, []byte(": "))
+		if len(ols) < 2 {
+			return nil, errors.New("not found header")
+		}
+		orderHeaders = append(orderHeaders, string(ols[0]))
+	}
+	return &Spec{
+		raw:          raw,
+		OrderHeaders: orderHeaders,
+	}, nil
+}
+
+type GospiderSpec struct {
+	TLSSpec *ja3.Spec
+	H1Spec  *Spec
+	H2Spec  *http2.Spec
+}
+
+func ParseGospiderSpec(value string) (*GospiderSpec, error) {
+	specs := strings.Split(value, "@")
+	spec := new(GospiderSpec)
+	if len(specs) != 3 {
+		return nil, errors.New("spec format error")
+	}
+	if specs[0] != "" {
+		b, err := hex.DecodeString(specs[0])
+		if err != nil {
+			return nil, err
+		}
+		if spec.TLSSpec, err = ja3.ParseSpec(b); err != nil {
+			return nil, err
+		}
+		log.Print("发送请求：", spec.TLSSpec.CipherSuites)
+	}
+	if specs[1] != "" {
+		b, err := hex.DecodeString(specs[1])
+		if err != nil {
+			return nil, err
+		}
+		if spec.H1Spec, err = ParseSpec(b); err != nil {
+			return nil, err
+		}
+	}
+	if specs[2] != "" {
+		b, err := hex.DecodeString(specs[2])
+		if err != nil {
+			return nil, err
+		}
+		if spec.H2Spec, err = http2.ParseSpec(b); err != nil {
+			return nil, err
+		}
+	}
+	return spec, nil
+}
+
+func (obj *RequestOption) initSpec() error {
+	if obj.Spec == "" {
+		return nil
+	}
+	gospiderSpec, err := ParseGospiderSpec(obj.Spec)
+	if err != nil {
+		return err
+	}
+	obj.gospiderSpec = gospiderSpec
+	if gospiderSpec.H1Spec != nil {
+		if obj.orderHeaders == nil {
+			orderData := NewOrderData()
+			for _, key := range gospiderSpec.H1Spec.OrderHeaders {
+				orderData.Add(key, nil)
+			}
+			obj.orderHeaders = orderData
+		}
+	}
+	if gospiderSpec.H2Spec != nil {
+		if obj.orderHeaders == nil {
+			orderData := NewOrderData()
+			for _, key := range gospiderSpec.H2Spec.OrderHeaders {
+				orderData.Add(key, nil)
+			}
+			obj.orderHeaders = orderData
+		}
+	}
+	return nil
+}