Examples with DiscoveryConfiguration org.opennms.netmgt.config.discovery.DiscoveryConfiguration used on opensource projects

Example 6 with DiscoveryConfiguration

use of org.opennms.netmgt.config.discovery.DiscoveryConfiguration in project opennms by OpenNMS.

the class RangeChunkerTest method testCoalesceConsecutiveSpecifics.

@Test
public void testCoalesceConsecutiveSpecifics() {
    DiscoveryConfiguration config = new DiscoveryConfiguration();
    for (int i = 0; i < 5; i++) {
        Specific specific = new Specific();
        specific.setAddress("10.0.0." + i);
        specific.setForeignSource("ABC");
        specific.setLocation("123");
        specific.setRetries(1);
        specific.setTimeout(1000l);
        config.addSpecific(specific);
    }
    Map<String, List<DiscoveryJob>> jobs = new RangeChunker(ipAddressFilter).chunk(config);
    printJobs(jobs);
    // The specifics have been combined into one range
    assertEquals(1, jobs.size());
    assertEquals(1, jobs.get("123").get(0).getRanges().size());
    IPPollRange range = jobs.get("123").get(0).getRanges().iterator().next();
    assertEquals("10.0.0.0", new IPAddress(range.getAddressRange().getBegin()).toString());
    assertEquals("10.0.0.4", new IPAddress(range.getAddressRange().getEnd()).toString());
}

Example 7 with DiscoveryConfiguration

use of org.opennms.netmgt.config.discovery.DiscoveryConfiguration in project opennms by OpenNMS.

the class DiscoveryIntegrationIT method testDiscovery.

@Test
public void testDiscovery() throws Exception {
    // Add a range of localhost IP addresses to ping
    IncludeRange range = new IncludeRange();
    range.setBegin("127.0.5.1");
    range.setEnd("127.0.5.254");
    range.setTimeout(5000l);
    range.setRetries(0);
    range.setLocation(CUSTOM_LOCATION);
    DiscoveryConfiguration config = m_discoveryConfig.getConfiguration();
    // Start immediately
    config.setInitialSleepTime(0l);
    // Discover 255 address ~= 10 seconds
    config.setPacketsPerSecond(25.5);
    // Add a discovery range to the config
    config.clearIncludeRanges();
    config.addIncludeRange(range);
    // Don't actually save the config or we'll overwrite the
    // opennms-base-assembly XML file
    // m_discoveryConfig.saveConfiguration(config);
    // m_discoveryConfig.reload();
    // Anticipate newSuspect events for all of the addresses
    EventAnticipator anticipator = m_eventIpcManager.getEventAnticipator();
    StreamSupport.stream(m_discoveryConfig.getConfiguredAddresses().spliterator(), false).forEach(addr -> {
        System.out.println("ANTICIPATING: " + str(addr.getAddress()));
        Event event = new Event();
        event.setUei(EventConstants.NEW_SUSPECT_INTERFACE_EVENT_UEI);
        event.setInterfaceAddress(addr.getAddress());
        anticipator.anticipateEvent(event);
    });
    // Don't re-init Discovery or it will reload the
    // DiscoveryConfigFactory and erase our changes to
    // the config
    // m_discovery.init();
    m_discovery.start();
    anticipator.waitForAnticipated(120000);
    anticipator.verifyAnticipated();
    anticipator.getAnticipatedEventsReceived().stream().forEach(eachEvent -> {
        Assert.assertNotNull(eachEvent.getParm("location"));
        Assert.assertEquals(CUSTOM_LOCATION, eachEvent.getParm("location").getValue().getContent());
    });
    m_discovery.stop();
}

Example 8 with DiscoveryConfiguration

use of org.opennms.netmgt.config.discovery.DiscoveryConfiguration in project opennms by OpenNMS.

the class DiscoveryScanServlet method doPost.

@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    LOG.info("Loading Discovery configuration.");
    HttpSession sess = request.getSession(true);
    DiscoveryConfiguration config = (DiscoveryConfiguration) sess.getAttribute(ATTRIBUTE_DISCOVERY_CONFIGURATION);
    if (config == null) {
        config = new DiscoveryConfiguration();
    }
    // load current general settings
    config = GeneralSettingsLoader.load(request, config);
    String action = request.getParameter("action");
    LOG.debug("action: {}", action);
    // add a Specific
    if (action.equals(addSpecificAction)) {
        LOG.debug("Adding Specific");
        String ipAddr = request.getParameter("specificipaddress");
        String timeout = request.getParameter("specifictimeout");
        String retries = request.getParameter("specificretries");
        String foreignSource = request.getParameter("specificforeignsource");
        String location = request.getParameter("specificlocation");
        Specific newSpecific = new Specific();
        newSpecific.setAddress(ipAddr);
        if (timeout != null && !"".equals(timeout.trim()) && !timeout.equals(String.valueOf(config.getTimeout().orElse(null)))) {
            newSpecific.setTimeout(WebSecurityUtils.safeParseLong(timeout));
        }
        if (retries != null && !"".equals(retries.trim()) && !retries.equals(String.valueOf(config.getRetries().orElse(null)))) {
            newSpecific.setRetries(WebSecurityUtils.safeParseInt(retries));
        }
        if (foreignSource != null && !"".equals(foreignSource.trim()) && !foreignSource.equals(config.getForeignSource().orElse(null))) {
            newSpecific.setForeignSource(foreignSource);
        }
        if (location != null && !"".equals(location.trim()) && !location.equals(config.getLocation().orElse(MonitoringLocationDao.DEFAULT_MONITORING_LOCATION_ID))) {
            newSpecific.setLocation(location);
        }
        config.addSpecific(newSpecific);
    }
    // remove 'Specific' from configuration
    if (action.equals(removeSpecificAction)) {
        LOG.debug("Removing Specific");
        String specificIndex = request.getParameter("index");
        int index = WebSecurityUtils.safeParseInt(specificIndex);
        final int index1 = index;
        Specific spec = config.getSpecifics().get(index1);
        boolean result = config.removeSpecific(spec);
        LOG.debug("Removing Specific result = {}", result);
    }
    // add an 'Include Range'
    if (action.equals(addIncludeRangeAction)) {
        LOG.debug("Adding Include Range");
        String ipAddrBase = request.getParameter("irbase");
        String ipAddrEnd = request.getParameter("irend");
        String timeout = request.getParameter("irtimeout");
        String retries = request.getParameter("irretries");
        String foreignSource = request.getParameter("irforeignsource");
        String location = request.getParameter("irlocation");
        IncludeRange newIR = new IncludeRange();
        newIR.setBegin(ipAddrBase);
        newIR.setEnd(ipAddrEnd);
        if (timeout != null && !"".equals(timeout.trim()) && !timeout.equals(String.valueOf(config.getTimeout().orElse(null)))) {
            newIR.setTimeout(WebSecurityUtils.safeParseLong(timeout));
        }
        if (retries != null && !"".equals(retries.trim()) && !retries.equals(String.valueOf(config.getRetries().orElse(null)))) {
            newIR.setRetries(WebSecurityUtils.safeParseInt(retries));
        }
        if (foreignSource != null && !"".equals(foreignSource.trim()) && !foreignSource.equals(config.getForeignSource().orElse(null))) {
            newIR.setForeignSource(foreignSource);
        }
        if (location != null && !"".equals(location.trim()) && !location.equals(config.getLocation().orElse(MonitoringLocationDao.DEFAULT_MONITORING_LOCATION_ID))) {
            newIR.setLocation(location);
        }
        config.addIncludeRange(newIR);
    }
    // remove 'Include Range' from configuration
    if (action.equals(removeIncludeRangeAction)) {
        LOG.debug("Removing Include Range");
        String specificIndex = request.getParameter("index");
        int index = WebSecurityUtils.safeParseInt(specificIndex);
        final int index1 = index;
        IncludeRange ir = config.getIncludeRanges().get(index1);
        boolean result = config.removeIncludeRange(ir);
        LOG.debug("Removing Include Range result = {}", result);
    }
    // add an 'Include URL'
    if (action.equals(addIncludeUrlAction)) {
        LOG.debug("Adding Include URL");
        String url = request.getParameter("iuurl");
        String timeout = request.getParameter("iutimeout");
        String retries = request.getParameter("iuretries");
        String foreignSource = request.getParameter("iuforeignsource");
        String location = request.getParameter("iulocation");
        IncludeUrl iu = new IncludeUrl();
        iu.setUrl(url);
        if (timeout != null && !"".equals(timeout.trim()) && !timeout.equals(String.valueOf(config.getTimeout().orElse(null)))) {
            iu.setTimeout(WebSecurityUtils.safeParseLong(timeout));
        }
        if (retries != null && !"".equals(retries.trim()) && !retries.equals(String.valueOf(config.getRetries().orElse(null)))) {
            iu.setRetries(WebSecurityUtils.safeParseInt(retries));
        }
        if (foreignSource != null && !"".equals(foreignSource.trim()) && !foreignSource.equals(config.getForeignSource().orElse(null))) {
            iu.setForeignSource(foreignSource);
        }
        if (location != null && !"".equals(location.trim()) && !location.equals(config.getLocation().orElse(MonitoringLocationDao.DEFAULT_MONITORING_LOCATION_ID))) {
            iu.setLocation(location);
        }
        config.addIncludeUrl(iu);
    }
    // remove 'Include URL' from configuration
    if (action.equals(removeIncludeUrlAction)) {
        LOG.debug("Removing Include URL");
        String specificIndex = request.getParameter("index");
        int index = WebSecurityUtils.safeParseInt(specificIndex);
        final int index1 = index;
        IncludeUrl iu = config.getIncludeUrls().get(index1);
        boolean result = config.removeIncludeUrl(iu);
        LOG.debug("Removing Include URL result = {}", result);
    }
    // add an 'Exclude Range'
    if (action.equals(addExcludeRangeAction)) {
        LOG.debug("Adding Exclude Range");
        String ipAddrBegin = request.getParameter("erbegin");
        String ipAddrEnd = request.getParameter("erend");
        ExcludeRange newER = new ExcludeRange();
        newER.setBegin(ipAddrBegin);
        newER.setEnd(ipAddrEnd);
        config.addExcludeRange(newER);
    }
    // remove 'Exclude Range' from configuration
    if (action.equals(removeExcludeRangeAction)) {
        LOG.debug("Removing Exclude Range");
        String specificIndex = request.getParameter("index");
        int index = WebSecurityUtils.safeParseInt(specificIndex);
        final int index1 = index;
        ExcludeRange er = config.getExcludeRanges().get(index1);
        boolean result = config.removeExcludeRange(er);
        LOG.debug("Removing Exclude Range result = {}", result);
    }
    // Submit the discovery job
    if (action.equals(saveAndRestartAction)) {
        try {
            WebApplicationContext beanFactory = WebApplicationContextUtils.getRequiredWebApplicationContext(getServletContext());
            // Fetch the ServiceRegistry
            ServiceRegistry registry = beanFactory.getBean(ServiceRegistry.class);
            // Use it to look up a DiscoveryTaskExecutor service
            DiscoveryTaskExecutor executor = registry.findProvider(DiscoveryTaskExecutor.class);
            // If the service exists...
            if (executor != null) {
                // Submit the job to the discovery queue
                executor.handleDiscoveryTask(config);
            } else {
                LOG.warn("No DiscoveryTaskExecutor service is available");
            }
        } catch (Throwable ex) {
            LOG.error("Error while submitting task", ex);
            throw new ServletException(ex);
        }
        // TODO: Send an event here when the scan is started? Or do it on the Camel side?
        /*
        	EventProxy proxy = null;
        	try {
    			proxy = Util.createEventProxy();
    		} catch (Throwable me) {
    			LOG.error(me.getMessage());
    		}

    		EventBuilder bldr = new EventBuilder(EventConstants.DISCOVERYCONFIG_CHANGED_EVENT_UEI, "ActionDiscoveryServlet");
    		bldr.setHost("host");

            try {
            	proxy.send(bldr.getEvent());
            } catch (Throwable me) {
    			LOG.error(me.getMessage());
    		}

            LOG.info("Restart Discovery requested!");
            */
        sess.removeAttribute(ATTRIBUTE_DISCOVERY_CONFIGURATION);
        response.sendRedirect(Util.calculateUrlBase(request, "admin/discovery/scan-done.jsp"));
        return;
    }
    sess.setAttribute(ATTRIBUTE_DISCOVERY_CONFIGURATION, config);
    RequestDispatcher dispatcher = this.getServletContext().getRequestDispatcher("/admin/discovery/edit-scan.jsp");
    dispatcher.forward(request, response);
}

Example 9 with DiscoveryConfiguration

use of org.opennms.netmgt.config.discovery.DiscoveryConfiguration in project opennms by OpenNMS.

the class RangeChunkerTest method testRangeChunker.

@Test
public void testRangeChunker() throws Exception {
    DiscoveryConfiguration config = new DiscoveryConfiguration();
    IncludeRange range = new IncludeRange();
    range.setBegin("10.1.0.1");
    range.setEnd("10.1.4.254");
    range.setForeignSource("Routers");
    range.setLocation("Raleigh");
    range.setRetries(2);
    range.setTimeout(500l);
    IncludeUrl url = new IncludeUrl();
    url.setUrl("file:src/test/resources/included_ip_addresses");
    url.setForeignSource("Switches");
    url.setLocation("Durham");
    url.setRetries(1);
    url.setTimeout(5000l);
    ExcludeRange excludes = new ExcludeRange();
    excludes.setBegin("10.1.3.0");
    excludes.setEnd("10.1.5.0");
    ExcludeRange excludes2 = new ExcludeRange();
    excludes2.setBegin("10.1.1.100");
    excludes2.setEnd("10.1.1.150");
    Specific specific = new Specific();
    specific.setAddress("10.1.3.5");
    specific.setForeignSource("Gateway");
    specific.setLocation("Co-lo");
    specific.setRetries(5);
    specific.setTimeout(1000l);
    Specific specific2 = new Specific();
    specific2.setAddress("10.1.2.5");
    specific2.setForeignSource("Routers");
    specific2.setLocation("Raleigh");
    specific2.setRetries(5);
    specific2.setTimeout(1000l);
    config.addExcludeRange(excludes);
    config.addExcludeRange(excludes2);
    config.addIncludeRange(range);
    config.addIncludeUrl(url);
    config.addSpecific(specific);
    config.addSpecific(specific2);
    config.setChunkSize(100);
    config.setForeignSource("Datacenter");
    config.setLocation("RDU");
    printJobs(new RangeChunker(ipAddressFilter).chunk(config));
}

Example 10 with DiscoveryConfiguration

use of org.opennms.netmgt.config.discovery.DiscoveryConfiguration in project opennms by OpenNMS.

the class RangeChunkerTest method testConsecutiveSpecificsWithDifferentRetries.

@Test
public void testConsecutiveSpecificsWithDifferentRetries() {
    DiscoveryConfiguration config = new DiscoveryConfiguration();
    for (int i = 0; i < 5; i++) {
        Specific specific = new Specific();
        specific.setAddress("10.0.0." + i);
        specific.setForeignSource("ABC");
        specific.setLocation("123");
        specific.setRetries(i);
        specific.setTimeout(1000l);
        config.addSpecific(specific);
    }
    Map<String, List<DiscoveryJob>> jobs = new RangeChunker(ipAddressFilter).chunk(config);
    printJobs(jobs);
    assertEquals(1, jobs.get("123").size());
    assertEquals(5, jobs.get("123").get(0).getRanges().size());
}