Search in sources :

Example 1 with IncludeUrl

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

the class DiscoveryConfigurationLocationMigratorOfflineTest method testRemoveAttribute.

@Test
public void testRemoveAttribute() throws Exception {
    final DiscoveryConfigurationLocationMigratorOffline task = new DiscoveryConfigurationLocationMigratorOffline();
    task.preExecute();
    task.execute();
    task.postExecute();
    final File configFile = new File(m_tempFolder.getRoot(), "etc/discovery-configuration.xml");
    final DiscoveryConfiguration discoveryConfiguration = JaxbUtils.unmarshal(DiscoveryConfiguration.class, new FileReader(configFile));
    Assert.assertNotNull(discoveryConfiguration);
    Assert.assertEquals(3, discoveryConfiguration.getIncludeRanges().size());
    Assert.assertEquals(3, discoveryConfiguration.getExcludeRanges().size());
    Assert.assertEquals(3, discoveryConfiguration.getSpecifics().size());
    Assert.assertEquals(3, discoveryConfiguration.getIncludeUrls().size());
    int pittsboroLocation = 0;
    int oldDefaultLocation = 0;
    int newDefaultLocation = 0;
    int nullLocation = 0;
    for (IncludeRange e : discoveryConfiguration.getIncludeRanges()) {
        if (DiscoveryConfigurationLocationMigratorOffline.NEW_DEFAULT_LOCATION.equals(e.getLocation().orElse(null))) {
            newDefaultLocation++;
        }
        if (DiscoveryConfigurationLocationMigratorOffline.OLD_DEFAULT_LOCATION.equals(e.getLocation().orElse(null))) {
            oldDefaultLocation++;
        }
        if (!e.getLocation().isPresent()) {
            nullLocation++;
        }
        if ("pittsboro".equals(e.getLocation().orElse(null))) {
            pittsboroLocation++;
        }
    }
    Assert.assertEquals(1, newDefaultLocation);
    Assert.assertEquals(1, pittsboroLocation);
    Assert.assertEquals(1, nullLocation);
    Assert.assertEquals(0, oldDefaultLocation);
    pittsboroLocation = 0;
    oldDefaultLocation = 0;
    newDefaultLocation = 0;
    nullLocation = 0;
    for (Specific e : discoveryConfiguration.getSpecifics()) {
        if (DiscoveryConfigurationLocationMigratorOffline.NEW_DEFAULT_LOCATION.equals(e.getLocation().orElse(null))) {
            newDefaultLocation++;
        }
        if (DiscoveryConfigurationLocationMigratorOffline.OLD_DEFAULT_LOCATION.equals(e.getLocation().orElse(null))) {
            oldDefaultLocation++;
        }
        if (!e.getLocation().isPresent()) {
            nullLocation++;
        }
        if ("pittsboro".equals(e.getLocation().orElse(null))) {
            pittsboroLocation++;
        }
    }
    Assert.assertEquals(1, newDefaultLocation);
    Assert.assertEquals(1, pittsboroLocation);
    Assert.assertEquals(1, nullLocation);
    Assert.assertEquals(0, oldDefaultLocation);
    pittsboroLocation = 0;
    oldDefaultLocation = 0;
    newDefaultLocation = 0;
    nullLocation = 0;
    for (IncludeUrl e : discoveryConfiguration.getIncludeUrls()) {
        if (DiscoveryConfigurationLocationMigratorOffline.NEW_DEFAULT_LOCATION.equals(e.getLocation().orElse(null))) {
            newDefaultLocation++;
        }
        if (DiscoveryConfigurationLocationMigratorOffline.OLD_DEFAULT_LOCATION.equals(e.getLocation().orElse(null))) {
            oldDefaultLocation++;
        }
        if (!e.getLocation().isPresent()) {
            nullLocation++;
        }
        if ("pittsboro".equals(e.getLocation().orElse(null))) {
            pittsboroLocation++;
        }
    }
    Assert.assertEquals(1, newDefaultLocation);
    Assert.assertEquals(1, pittsboroLocation);
    Assert.assertEquals(1, nullLocation);
    Assert.assertEquals(0, oldDefaultLocation);
}
Also used : IncludeRange(org.opennms.netmgt.config.discovery.IncludeRange) IncludeUrl(org.opennms.netmgt.config.discovery.IncludeUrl) DiscoveryConfiguration(org.opennms.netmgt.config.discovery.DiscoveryConfiguration) FileReader(java.io.FileReader) Specific(org.opennms.netmgt.config.discovery.Specific) File(java.io.File) Test(org.junit.Test)

Example 2 with IncludeUrl

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

the class DiscoveryConfigFactory method getURLSpecifics.

/**
 * <p>getURLSpecifics</p>
 *
 * @return a {@link java.util.List} object.
 */
@Override
public List<IPPollAddress> getURLSpecifics() {
    final List<IPPollAddress> specifics = new LinkedList<IPPollAddress>();
    getReadLock().lock();
    try {
        Long defaultTimeout = getConfiguration().getTimeout().orElse(DEFAULT_TIMEOUT);
        Integer defaultRetries = getConfiguration().getRetries().orElse(DEFAULT_RETRIES);
        for (final IncludeUrl url : getConfiguration().getIncludeUrls()) {
            long timeout = url.getTimeout().orElse(defaultTimeout);
            int retries = url.getRetries().orElse(defaultRetries);
            addToSpecificsFromURL(specifics, url.getUrl().orElse(null), url.getForeignSource().orElse(null), url.getLocation().orElse(null), timeout, retries);
        }
        return specifics;
    } finally {
        getReadLock().unlock();
    }
}
Also used : IPPollAddress(org.opennms.netmgt.model.discovery.IPPollAddress) IncludeUrl(org.opennms.netmgt.config.discovery.IncludeUrl) LinkedList(java.util.LinkedList)

Example 3 with IncludeUrl

use of org.opennms.netmgt.config.discovery.IncludeUrl 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);
}
Also used : IncludeRange(org.opennms.netmgt.config.discovery.IncludeRange) IncludeUrl(org.opennms.netmgt.config.discovery.IncludeUrl) HttpSession(javax.servlet.http.HttpSession) DiscoveryConfiguration(org.opennms.netmgt.config.discovery.DiscoveryConfiguration) Specific(org.opennms.netmgt.config.discovery.Specific) RequestDispatcher(javax.servlet.RequestDispatcher) WebApplicationContext(org.springframework.web.context.WebApplicationContext) DiscoveryTaskExecutor(org.opennms.netmgt.discovery.DiscoveryTaskExecutor) ServletException(javax.servlet.ServletException) ServiceRegistry(org.opennms.core.soa.ServiceRegistry) ExcludeRange(org.opennms.netmgt.config.discovery.ExcludeRange)

Example 4 with IncludeUrl

use of org.opennms.netmgt.config.discovery.IncludeUrl 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));
}
Also used : IncludeRange(org.opennms.netmgt.config.discovery.IncludeRange) IncludeUrl(org.opennms.netmgt.config.discovery.IncludeUrl) DiscoveryConfiguration(org.opennms.netmgt.config.discovery.DiscoveryConfiguration) Specific(org.opennms.netmgt.config.discovery.Specific) ExcludeRange(org.opennms.netmgt.config.discovery.ExcludeRange) Test(org.junit.Test)

Example 5 with IncludeUrl

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

the class DiscoveryConfigFactory method getForeignSource.

@Override
public String getForeignSource(InetAddress address) {
    getReadLock().lock();
    try {
        LOG.debug("Looking for matching foreign source specific IP or IP range with address: {}...", address);
        List<Specific> specificCollection = getConfiguration().getSpecifics();
        for (Specific specific : specificCollection) {
            String ipAddr = specific.getAddress();
            if (ipAddr.equals(InetAddressUtils.str(address))) {
                String foreignSource = specific.getForeignSource().orElse(null);
                LOG.debug("Matched foreign source {} matching address: {} against specific {}.", foreignSource, address, ipAddr);
                return foreignSource;
            }
        }
        final byte[] laddr = address.getAddress();
        List<IncludeRange> includeRangeCollection = getConfiguration().getIncludeRanges();
        for (IncludeRange range : includeRangeCollection) {
            if (InetAddressUtils.isInetAddressInRange(laddr, range.getBegin(), range.getEnd())) {
                String foreignSource = range.getForeignSource().orElse(null);
                LOG.debug("Found foreign source {} with address {} in the range begin: {} and end: {}.", foreignSource, address, range.getBegin(), range.getEnd());
                return foreignSource;
            }
        }
        List<IncludeUrl> includeUrlCollection = getConfiguration().getIncludeUrls();
        for (IncludeUrl includeUrl : includeUrlCollection) {
            String ipAddr = includeUrl.getUrl().orElse("");
            if (ipAddr.equals(InetAddressUtils.str(address))) {
                String foreignSource = includeUrl.getForeignSource().orElse(null);
                LOG.debug("Matched foreign source {} matching address: {} in specified URL.", foreignSource, address);
                return foreignSource;
            }
        }
        return getConfiguration().getForeignSource().orElse(null);
    } finally {
        getReadLock().unlock();
    }
}
Also used : IncludeRange(org.opennms.netmgt.config.discovery.IncludeRange) IncludeUrl(org.opennms.netmgt.config.discovery.IncludeUrl) Specific(org.opennms.netmgt.config.discovery.Specific)

Aggregations

IncludeUrl (org.opennms.netmgt.config.discovery.IncludeUrl)7 IncludeRange (org.opennms.netmgt.config.discovery.IncludeRange)6 Specific (org.opennms.netmgt.config.discovery.Specific)6 DiscoveryConfiguration (org.opennms.netmgt.config.discovery.DiscoveryConfiguration)5 ExcludeRange (org.opennms.netmgt.config.discovery.ExcludeRange)4 RequestDispatcher (javax.servlet.RequestDispatcher)2 ServletException (javax.servlet.ServletException)2 HttpSession (javax.servlet.http.HttpSession)2 Test (org.junit.Test)2 File (java.io.File)1 FileReader (java.io.FileReader)1 StringWriter (java.io.StringWriter)1 LinkedList (java.util.LinkedList)1 ServiceRegistry (org.opennms.core.soa.ServiceRegistry)1 DiscoveryConfigFactory (org.opennms.netmgt.config.DiscoveryConfigFactory)1 DiscoveryTaskExecutor (org.opennms.netmgt.discovery.DiscoveryTaskExecutor)1 EventProxy (org.opennms.netmgt.events.api.EventProxy)1 IPPollAddress (org.opennms.netmgt.model.discovery.IPPollAddress)1 EventBuilder (org.opennms.netmgt.model.events.EventBuilder)1 WebApplicationContext (org.springframework.web.context.WebApplicationContext)1