service_matcher-go/http/http.go

package http

import (
	"strings"

	csm "git.loafle.net/commons/service_matcher-go"
)

type HTTPMatcher struct {
	csm.Matchers
	meta csm.Metadata
}

func (h *HTTPMatcher) Key() string {
	return "HTTP"
}

func (h *HTTPMatcher) Name() string {
	name := "HTTP"
	if v, ok := h.meta["server"]; ok {
		name = name + " (" + v + ")"
	}
	return name
}

func (h *HTTPMatcher) Meta() csm.Metadata {
	return h.meta
}

func (h *HTTPMatcher) IsPrePacket() bool {
	return false
}

func (h *HTTPMatcher) HasResponse(index int) bool {
	return true
}

func (h *HTTPMatcher) IsError(info csm.MatchInfo, index int, packet *csm.Packet) bool {
	return false
}

func (h *HTTPMatcher) Match(info csm.MatchInfo, index int, packet *csm.Packet) error {

	if packet == nil || packet.Buffer == nil || packet.Len == 0 {
		return csm.NoPacketReceivedError()
	}

	str := string(packet.Buffer)
	elems := strings.Split(str, "\r\n")

	if len(elems) <= 0 || 9 > len(elems[0]) {
		return csm.NotMatchedError()
	}

	protocol := (elems[0])[:8]

	if !strings.HasPrefix(protocol, "HTTP/") {
		return csm.NotMatchedError()
	}

	serverName := ""
	for _, valueStr := range elems {
		tempElems := strings.Split(valueStr, ":")

		if 0 == strings.Compare(tempElems[0], "Server") {
			serverName = strings.TrimSpace(tempElems[1])
			break
		}
	}

	h.meta["protocol_version"] = protocol
	if len(serverName) > 0 {
		h.meta["server"] = serverName
	}

	return nil
}

func NewMatcher() csm.Matcher {

	m := &HTTPMatcher{}
	m.meta = csm.NewMetadata()

	reqStr := "GET / HTTP/1.1\r\n\r\n"
	byte := make([]byte, len(reqStr))
	copy(byte[:], reqStr)

	m.AddPacket(csm.NewPacket(byte, len(reqStr)))

	return m
}