mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-10-26 07:44:59 +03:00
Hamed
8eb58a4a14
The access-log scanner and the cross-node merge stored the same IP list in opposite orders, so neither saw a row as unchanged, and the merge re-read every preloaded row and re-upserted every attribution row each sync cycle. All writers now share one canonical encoder (timestamp, then IP ascending), the first CAS is seeded from the preload, and identical attribution rows are skipped. A create race between the scanner and node sync now merges into the winning row instead of dropping the loser's IPs, which on PostgreSQL also aborted the scanner's whole scan transaction. Closes #6927
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
package job
|
|
|
|
import (
|
|
"net/netip"
|
|
"slices"
|
|
"strings"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
|
)
|
|
|
|
// An address that matches is neither counted towards a client's IP limit nor
|
|
// banned: counting it would still cut the shared network it protects (#5378).
|
|
type ipLimitAllowlist struct {
|
|
prefixes []netip.Prefix
|
|
addrs []netip.Addr
|
|
}
|
|
|
|
// Comma-separated, each entry a CIDR or a bare address. Unparseable entries are
|
|
// skipped: the validator uses these same rules, so only a hand-edited DB differs.
|
|
func parseIpLimitAllowlist(raw string) ipLimitAllowlist {
|
|
var list ipLimitAllowlist
|
|
for field := range strings.SplitSeq(raw, ",") {
|
|
field = strings.TrimSpace(field)
|
|
if field == "" {
|
|
continue
|
|
}
|
|
if prefix, err := netip.ParsePrefix(field); err == nil {
|
|
// Unmapped: contains() unmaps the queried address, and Prefix.Contains
|
|
// is false whenever the bit lengths disagree.
|
|
if addr := prefix.Addr(); addr.Is4In6() {
|
|
if p4, perr := addr.Unmap().Prefix(prefix.Bits() - 96); perr == nil {
|
|
prefix = p4
|
|
}
|
|
}
|
|
list.prefixes = append(list.prefixes, prefix.Masked())
|
|
continue
|
|
}
|
|
if addr, err := netip.ParseAddr(field); err == nil {
|
|
list.addrs = append(list.addrs, addr.Unmap())
|
|
}
|
|
}
|
|
return list
|
|
}
|
|
|
|
func (l ipLimitAllowlist) empty() bool {
|
|
return len(l.prefixes) == 0 && len(l.addrs) == 0
|
|
}
|
|
|
|
func (l ipLimitAllowlist) contains(ip string) bool {
|
|
if l.empty() {
|
|
return false
|
|
}
|
|
addr, err := netip.ParseAddr(strings.TrimSpace(ip))
|
|
if err != nil {
|
|
return false
|
|
}
|
|
addr = addr.Unmap()
|
|
if slices.Contains(l.addrs, addr) {
|
|
return true
|
|
}
|
|
for _, prefix := range l.prefixes {
|
|
if prefix.Contains(addr) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// split separates the entries an allowlist protects from the ones the limit
|
|
// still applies to, preserving the caller's ordering in both.
|
|
func (l ipLimitAllowlist) split(entries []model.ClientIpEntry) (limited, allowed []model.ClientIpEntry) {
|
|
if l.empty() {
|
|
return entries, nil
|
|
}
|
|
limited = make([]model.ClientIpEntry, 0, len(entries))
|
|
for _, entry := range entries {
|
|
if l.contains(entry.IP) {
|
|
allowed = append(allowed, entry)
|
|
continue
|
|
}
|
|
limited = append(limited, entry)
|
|
}
|
|
return limited, allowed
|
|
}
|