org.minidns.iterative.ResolutionState Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of minidns-iterative-resolver Show documentation

A DNS library for Java and Android systems

There is a newer version: 1.1.1

Show newest version

/*
 * Copyright 2015-2018 the original author or authors
 *
 * This software is licensed under the Apache License, Version 2.0,
 * the GNU Lesser General Public License version 2 or later ("LGPL")
 * and the WTFPL.
 * You may choose either license to govern your use of this software only
 * upon the condition that you accept all of the terms of either
 * the Apache License 2.0, the LGPL 2.1+ or the WTFPL.
 */
package org.minidns.iterative;

import java.net.InetAddress;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;

import org.minidns.dnsmessage.DNSMessage;
import org.minidns.dnsmessage.Question;
import org.minidns.iterative.IterativeClientException.LoopDetected;
import org.minidns.iterative.IterativeClientException.MaxIterativeStepsReached;

public class ResolutionState {

    private final IterativeDNSClient recursiveDnsClient;
    private final HashMap> map = new HashMap<>();
    private int steps;

    ResolutionState(IterativeDNSClient recursiveDnsClient) {
        this.recursiveDnsClient = recursiveDnsClient;
    }

    void recurse(InetAddress address, DNSMessage query) throws LoopDetected, MaxIterativeStepsReached {
        Question question = query.getQuestion();
        if (!map.containsKey(address)) {
            map.put(address, new HashSet());
        } else if (map.get(address).contains(question)) {
            throw new IterativeClientException.LoopDetected();
        }

        if (++steps > recursiveDnsClient.maxSteps) {
            throw new IterativeClientException.MaxIterativeStepsReached(); 
        }

        boolean isNew = map.get(address).add(question);
        assert(isNew);
    }

    void decrementSteps() {
        steps--;
    }

}