EdgeNode/internal/nodes/http_request_user_agent.go

// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .

package nodes

import (
	"net/http"
)

func (this *HTTPRequest) doCheckUserAgent() (shouldStop bool) {
	if this.web.UserAgent == nil || !this.web.UserAgent.IsOn {
		return
	}

	const cacheSeconds = "3600" // 时间不能过长，防止修改设置后长期无法生效

	if !this.web.UserAgent.AllowRequest(this.RawReq) {
		this.tags = append(this.tags, "userAgentCheck")
		this.writer.Header().Set("Cache-Control", "max-age="+cacheSeconds)
		this.writeCode(http.StatusForbidden, "The User-Agent has been blocked.", "当前访问已被UA名单拦截。")
		return true
	}

	return
}