All files / src/resolvers/objects SimpleTargetResolver.ts

90% Statements 18/20
83.33% Branches 10/12
100% Functions 4/4
94.74% Lines 18/19

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 473x           3x   6x 6x   6x       4x 2x 1x     1x 1x 1x         2x         1x       2x 2x 1x   1x 1x          
import * as net from 'net';
import { FlowTarget, FlowRuleGroup, ResolvedFlowTarget } from "../../FlowDefinitions";
import { LoggerFactory } from "../../logger-factory";
import { Logger } from "../../logger-type";
import { TargetResolver } from "./TargetResolver";
 
export class SimpleTargetResolver implements TargetResolver {
    logger: Logger;
    IP_V4_CIDR_REGEX = /^([0-9]{1,3}\.){3}[0-9]{1,3}(\/([0-9]|[1-2][0-9]|3[0-2]))?$/;
    IP_V6_CIDR_REGEX = /^s*((([0-9A-Fa-f]{1,4}:){7}([0-9A-Fa-f]{1,4}|:))|(([0-9A-Fa-f]{1,4}:){6}(:[0-9A-Fa-f]{1,4}|((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3})|:))|(([0-9A-Fa-f]{1,4}:){5}(((:[0-9A-Fa-f]{1,4}){1,2})|:((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3})|:))|(([0-9A-Fa-f]{1,4}:){4}(((:[0-9A-Fa-f]{1,4}){1,3})|((:[0-9A-Fa-f]{1,4})?:((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){3}(((:[0-9A-Fa-f]{1,4}){1,4})|((:[0-9A-Fa-f]{1,4}){0,2}:((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){2}(((:[0-9A-Fa-f]{1,4}){1,5})|((:[0-9A-Fa-f]{1,4}){0,3}:((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){1}(((:[0-9A-Fa-f]{1,4}){1,6})|((:[0-9A-Fa-f]{1,4}){0,4}:((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3}))|:))|(:(((:[0-9A-Fa-f]{1,4}){1,7})|((:[0-9A-Fa-f]{1,4}){0,5}:((25[0-5]|2[0-4]d|1dd|[1-9]?d)(.(25[0-5]|2[0-4]d|1dd|[1-9]?d)){3}))|:)))(%.+)?s*(\/([0-9]|[1-9][0-9]|1[0-1][0-9]|12[0-8]))?$/;
    constructor(loggerFactory: LoggerFactory) {
        this.logger = loggerFactory.getLogger('SimpleObjectResolver');
    }
 
    canResolve(target: FlowTarget): boolean {
        if (target.type === 'Address') {
            if (this.isCidr(target.value)) {
                return true;
            }
            //https://nodejs.org/api/net.html#net_net_isip_input , where 0 invalid, 4 ipv4, 6 ipv6
            else Eif (net.isIP(target.value) === 0) {
                this.logger.error('Invalid ip address', target);
                return false;
            } else {
                return true;
            }
        } else {
            return false;
        }
    }
 
    async resolve(target: FlowTarget, ruleGroup?: FlowRuleGroup): Promise<ResolvedFlowTarget> {
        return { ...target, addresses: [target.value] }
    }
 
    isCidr(inputCidr: string) {
        Iif (!inputCidr) { return false }
        if (inputCidr.match(this.IP_V4_CIDR_REGEX) || inputCidr.match(this.IP_V6_CIDR_REGEX)) {
            return true;
        } else {
            this.logger.info('Invalid cidr', inputCidr);
            return false;
        }
 
    };
 
}