80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
package prox5
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
rl "github.com/yunginnanet/Rate5"
|
|
|
|
"git.tcp.direct/kayos/prox5/internal/pools"
|
|
)
|
|
|
|
// https://pkg.go.dev/github.com/yunginnanet/Rate5#Policy
|
|
var defaultUseProxyRatelimiter = rl.Policy{
|
|
Window: 55,
|
|
Burst: 55,
|
|
}
|
|
|
|
var defaultBadProxyRateLimiter = rl.Policy{
|
|
Window: 55,
|
|
Burst: 25,
|
|
}
|
|
|
|
const (
|
|
stateUnlocked uint32 = iota
|
|
stateLocked
|
|
)
|
|
|
|
// Proxy represents an individual proxy
|
|
type Proxy struct {
|
|
// Endpoint is the address:port of the proxy that we connect to
|
|
Endpoint string
|
|
// ProxiedIP is the address that we end up having when making proxied requests through this proxy
|
|
ProxiedIP string
|
|
// protocol is the version/Protocol (currently SOCKS* only) of the proxy
|
|
protocol proto
|
|
// lastValidated is the time this proxy was last verified working
|
|
lastValidated time.Time
|
|
// timesValidated is the amount of times the proxy has been validated.
|
|
timesValidated int64
|
|
// timesBad is the amount of times the proxy has been marked as bad.
|
|
timesBad int64
|
|
|
|
parent *Swamp
|
|
lock uint32
|
|
}
|
|
|
|
// UniqueKey is an implementation of the Identity interface from Rate5.
|
|
// See: https://pkg.go.dev/github.com/yunginnanet/Rate5#Identity
|
|
func (sock *Proxy) UniqueKey() string {
|
|
return sock.Endpoint
|
|
}
|
|
|
|
// GetProto retrieves the known protocol value of the Proxy.
|
|
func (sock *Proxy) GetProto() ProxyProtocol {
|
|
return sock.protocol.Get()
|
|
}
|
|
|
|
// GetProto safely retrieves the protocol value of the Proxy.
|
|
func (sock *Proxy) String() string {
|
|
tout := ""
|
|
if sock.parent.GetServerTimeoutStr() != "-1" {
|
|
tbuf := pools.CopABuffer.Get().(*strings.Builder)
|
|
tbuf.WriteString("?timeout=")
|
|
tbuf.WriteString(sock.parent.GetServerTimeoutStr())
|
|
tbuf.WriteString("s")
|
|
tout = tbuf.String()
|
|
pools.DiscardBuffer(tbuf)
|
|
}
|
|
buf := pools.CopABuffer.Get().(*strings.Builder)
|
|
buf.WriteString(sock.GetProto().String())
|
|
buf.WriteString("://")
|
|
buf.WriteString(sock.Endpoint)
|
|
if tout != "" {
|
|
buf.WriteString(tout)
|
|
}
|
|
out := buf.String()
|
|
pools.DiscardBuffer(buf)
|
|
return out
|
|
}
|