mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-10 20:50:25 +08:00
增加一些注释、测试等
This commit is contained in:
@@ -49,7 +49,7 @@ func (this *IPListManager) Start() {
|
|||||||
remotelogs.Println("IP_LIST_MANAGER", err.Error())
|
remotelogs.Println("IP_LIST_MANAGER", err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
ticker := time.NewTicker(60 * time.Second) // TODO 未来改成可以手动触发IP变更事件
|
ticker := time.NewTicker(60 * time.Second)
|
||||||
events.On(events.EventQuit, func() {
|
events.On(events.EventQuit, func() {
|
||||||
ticker.Stop()
|
ticker.Stop()
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -853,6 +853,7 @@ func (this *HTTPRequest) requestServerPort() int {
|
|||||||
|
|
||||||
// 设置代理相关头部信息
|
// 设置代理相关头部信息
|
||||||
// 参考:https://tools.ietf.org/html/rfc7239
|
// 参考:https://tools.ietf.org/html/rfc7239
|
||||||
|
// TODO X-Forwarded-* 系列做成可选项,避免有些源站屏蔽此项
|
||||||
func (this *HTTPRequest) setForwardHeaders(header http.Header) {
|
func (this *HTTPRequest) setForwardHeaders(header http.Header) {
|
||||||
if this.RawReq.Header.Get("Connection") == "close" {
|
if this.RawReq.Header.Get("Connection") == "close" {
|
||||||
this.RawReq.Header.Set("Connection", "keep-alive")
|
this.RawReq.Header.Set("Connection", "keep-alive")
|
||||||
|
|||||||
@@ -84,6 +84,7 @@ func (this *HTTPRequest) checkWAFRequest(firewallPolicy *firewallconfigs.HTTPFir
|
|||||||
countryId := iplibrary.SharedCountryManager.Lookup(result.Country)
|
countryId := iplibrary.SharedCountryManager.Lookup(result.Country)
|
||||||
if countryId > 0 && lists.ContainsInt64(regionConfig.DenyCountryIds, countryId) {
|
if countryId > 0 && lists.ContainsInt64(regionConfig.DenyCountryIds, countryId) {
|
||||||
// TODO 可以配置对封禁的处理方式等
|
// TODO 可以配置对封禁的处理方式等
|
||||||
|
// TODO 需要记录日志信息
|
||||||
this.writer.WriteHeader(http.StatusForbidden)
|
this.writer.WriteHeader(http.StatusForbidden)
|
||||||
this.writer.Close()
|
this.writer.Close()
|
||||||
|
|
||||||
@@ -99,6 +100,7 @@ func (this *HTTPRequest) checkWAFRequest(firewallPolicy *firewallconfigs.HTTPFir
|
|||||||
provinceId := iplibrary.SharedProvinceManager.Lookup(result.Province)
|
provinceId := iplibrary.SharedProvinceManager.Lookup(result.Province)
|
||||||
if provinceId > 0 && lists.ContainsInt64(regionConfig.DenyProvinceIds, provinceId) {
|
if provinceId > 0 && lists.ContainsInt64(regionConfig.DenyProvinceIds, provinceId) {
|
||||||
// TODO 可以配置对封禁的处理方式等
|
// TODO 可以配置对封禁的处理方式等
|
||||||
|
// TODO 需要记录日志信息
|
||||||
this.writer.WriteHeader(http.StatusForbidden)
|
this.writer.WriteHeader(http.StatusForbidden)
|
||||||
this.writer.Close()
|
this.writer.Close()
|
||||||
|
|
||||||
|
|||||||
9
internal/utils/ip_test.go
Normal file
9
internal/utils/ip_test.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestIP2Long(t *testing.T) {
|
||||||
|
t.Log(IP2Long("0.0.0.0"))
|
||||||
|
t.Log(IP2Long("1.0.0.0"))
|
||||||
|
t.Log(IP2Long("0.0.0.0.0"))
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user