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);
}
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();
}
}
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);
}
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));
}
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();
}
}
Aggregations