Feat: support reject ip from adguard filter

This commit is contained in:
SukkaW
2025-06-20 16:30:07 +08:00
parent 172e4efd8a
commit 515f262042
5 changed files with 116 additions and 101 deletions

View File

@@ -28,7 +28,7 @@ export function processFilterRulesWithPreload(
) {
const downloadPromise = fetchAssets(filterRulesUrl, fallbackUrls);
return (span: Span) => span.traceChildAsync<Record<'whiteDomains' | 'whiteDomainSuffixes' | 'blackDomains' | 'blackDomainSuffixes', string[]>>(`process filter rules: ${filterRulesUrl}`, async (span) => {
return (span: Span) => span.traceChildAsync<Record<'whiteDomains' | 'whiteDomainSuffixes' | 'blackDomains' | 'blackDomainSuffixes' | 'blackIPs', string[]>>(`process filter rules: ${filterRulesUrl}`, async (span) => {
const filterRules = await span.traceChildPromise('download', downloadPromise);
const whiteDomains = new Set<string>();
@@ -39,6 +39,8 @@ export function processFilterRulesWithPreload(
const warningMessages: string[] = [];
const blackIPs: string[] = [];
const MUTABLE_PARSE_LINE_RESULT: [string, ParseType] = ['', ParseType.NotParsed];
/**
* @param {string} line
@@ -78,6 +80,9 @@ export function processFilterRulesWithPreload(
case ParseType.ErrorMessage:
warningMessages.push(hostname);
break;
case ParseType.BlackIP:
blackIPs.push(hostname);
break;
default:
break;
}
@@ -107,7 +112,8 @@ export function processFilterRulesWithPreload(
whiteDomains: Array.from(whiteDomains),
whiteDomainSuffixes: Array.from(whiteDomainSuffixes),
blackDomains: Array.from(blackDomains),
blackDomainSuffixes: Array.from(blackDomainSuffixes)
blackDomainSuffixes: Array.from(blackDomainSuffixes),
blackIPs
};
});
}

View File

@@ -1,6 +1,7 @@
import type { Span } from '../../trace';
import { HostnameSmolTrie } from '../trie';
import { not, nullthrow } from 'foxts/guard';
import { fastIpVersion } from '../misc';
import type { MaybePromise } from '../misc';
import type { BaseWriteStrategy } from '../writing-strategy/base';
import { merge as mergeCidr } from 'fast-cidr-tools';
@@ -225,6 +226,27 @@ export class FileOutput {
return ip + '/128';
};
bulkAddAnyCIDR(cidrs: string[], noResolve = false) {
const list4 = noResolve ? this.ipcidrNoResolve : this.ipcidr;
const list6 = noResolve ? this.ipcidr6NoResolve : this.ipcidr6;
for (let i = 0, len = cidrs.length; i < len; i++) {
let cidr = cidrs[i];
const version = fastIpVersion(cidr);
if (version === 0) {
continue; // skip invalid IPs
}
cidr = FileOutput.ipToCidr(cidr, version);
if (version === 4) {
list4.add(cidr);
} else /* if (version === 6) */ {
list6.add(cidr);
}
}
return this;
}
bulkAddCIDR4(cidrs: string[]) {
for (let i = 0, len = cidrs.length; i < len; i++) {
this.ipcidr.add(FileOutput.ipToCidr(cidrs[i], 4));