This is an automated email from the ASF dual-hosted git repository. ppalaga pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/camel-quarkus.git
The following commit(s) were added to refs/heads/main by this push: new 56e63f6242 Snmp: cover snmp v3 for POLL operation #4881 56e63f6242 is described below commit 56e63f62424d7221eb8af6da0546277d88acb774 Author: JiriOndrusek <ondrusek.j...@gmail.com> AuthorDate: Thu May 4 17:25:28 2023 +0200 Snmp: cover snmp v3 for POLL operation #4881 --- .../quarkus/component/snmp/it/SnmpResource.java | 38 ++++++++++ .../camel/quarkus/component/snmp/it/SnmpRoute.java | 40 +---------- .../camel/quarkus/component/snmp/it/SnmpTest.java | 82 +++++++++------------- .../component/snmp/it/SnmpTestResource.java | 51 ++++++++++---- 4 files changed, 108 insertions(+), 103 deletions(-) diff --git a/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpResource.java b/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpResource.java index f76f9c1d13..3a0475e6c7 100644 --- a/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpResource.java +++ b/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpResource.java @@ -28,8 +28,11 @@ import jakarta.ws.rs.POST; import jakarta.ws.rs.Path; import jakarta.ws.rs.PathParam; import jakarta.ws.rs.Produces; +import jakarta.ws.rs.QueryParam; import jakarta.ws.rs.core.MediaType; import jakarta.ws.rs.core.Response; +import org.apache.camel.ConsumerTemplate; +import org.apache.camel.Exchange; import org.apache.camel.ProducerTemplate; import org.apache.camel.component.snmp.SnmpMessage; import org.eclipse.microprofile.config.inject.ConfigProperty; @@ -59,6 +62,9 @@ public class SnmpResource { @Named("snmpTrapResults") Map<String, Deque<SnmpMessage>> snmpResults; + @Inject + ConsumerTemplate consumerTemplate; + @Inject ProducerTemplate producerTemplate; @@ -87,6 +93,9 @@ public class SnmpResource { String response = pdu.stream() .flatMap(m -> m.getSnmpMessage().getVariableBindings().stream()) + //snmp may add null oid to the result - because responder supports v3 + // (see https://camel.apache.org/components/3.20.x/snmp-component.html#_the_result_of_a_poll) + .filter(vb -> !"Null".equals(vb.getVariable().toString())) .map(vb -> vb.getVariable().toString()) .collect(Collectors.joining(",")); @@ -106,6 +115,35 @@ public class SnmpResource { return Response.ok().build(); } + @Path("/poll/{version}") + @POST + @Produces(MediaType.TEXT_PLAIN) + public Response poll(@PathParam("version") int version, + @QueryParam("user") String user, + @QueryParam("securityLevel") String securityLevel, + String oid) { + String url = String.format("snmp:%s?protocol=udp&snmpVersion=%d&type=POLL&oids=%s", snmpListenAddress, version, oid); + if (user != null) { + url = url + "&securityName=" + user; + } + if (securityLevel != null) { + url = url + "&securityLevel=" + securityLevel; + } + + //Even if routeBuilder is preferred instead of consumerTemplate, consumerTemplete can be used in a case, when the component uses polling consumers by default. + // In this case: + // - only polling consumer is used by the SNMP component + // - usage of the consumerTemplate reduces a lot of requests between SNMP providers significantly, thus making the tests more stable. + Exchange e = consumerTemplate.receive(url); + + String result = e.getIn().getBody(SnmpMessage.class).getSnmpMessage().getVariableBindings().stream() + //snmp may add null oid to the result - because responder supports v3 + // (see https://camel.apache.org/components/3.20.x/snmp-component.html#_the_result_of_a_poll) + .filter(vb -> !"Null".equals(vb.getVariable().toString())) + .map(v -> v.getVariable().toString()).collect(Collectors.joining(",")); + return Response.ok(result).build(); + } + @Path("/results/{from}") @POST @Produces(MediaType.TEXT_PLAIN) diff --git a/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpRoute.java b/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpRoute.java index d6fb2f64f9..8fe4149ece 100644 --- a/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpRoute.java +++ b/integration-tests-jvm/snmp/src/main/java/org/apache/camel/quarkus/component/snmp/it/SnmpRoute.java @@ -28,7 +28,6 @@ import jakarta.inject.Singleton; import org.apache.camel.builder.RouteBuilder; import org.apache.camel.component.snmp.SnmpMessage; import org.eclipse.microprofile.config.inject.ConfigProperty; -import org.snmp4j.mp.SnmpConstants; @ApplicationScoped public class SnmpRoute extends RouteBuilder { @@ -39,12 +38,9 @@ public class SnmpRoute extends RouteBuilder { @ConfigProperty(name = TRAP_V0_PORT) int trap0Port; - @ConfigProperty(name = "SnmpRoute_trap_v1") + @ConfigProperty(name = TRAP_V1_PORT) int trap1Port; - @ConfigProperty(name = "snmpListenAddress") - String snmpListenAddress; - @Inject @Named("snmpTrapResults") Map<String, Deque<SnmpMessage>> snmpResults; @@ -58,33 +54,6 @@ public class SnmpRoute extends RouteBuilder { //TRAP consumer snmpVersion=1 from("snmp:0.0.0.0:" + trap1Port + "?protocol=udp&type=TRAP&snmpVersion=1") .process(e -> snmpResults.get("v1_trap").add(e.getIn().getBody(SnmpMessage.class))); - - //POLL consumer 2 oids, snmpVersion=0 - from("snmp://" + snmpListenAddress + "?protocol=udp&snmpVersion=0&type=POLL&oids=" + - SnmpConstants.sysLocation + "," + SnmpConstants.sysContact) - .process(e -> snmpResults.get("v0_poll2oids").add(e.getIn().getBody(SnmpMessage.class))); - //POLL consumer 2 oids, snmpVersion=1 - from("snmp://" + snmpListenAddress + "?protocol=udp&snmpVersion=1&type=POLL&oids=" + - SnmpConstants.sysLocation + "," + SnmpConstants.sysContact) - .process(e -> snmpResults.get("v1_poll2oids").add(e.getIn().getBody(SnmpMessage.class))); - - // POLL consumer starting with dot snmpVersion=0 - from("snmp://" + snmpListenAddress - + "?protocol=udp&snmpVersion=0&type=POLL&oids=.1.3.6.1.4.1.6527.3.1.2.21.2.1.50") - .process(e -> snmpResults.get("v0_pollStartingDot").add(e.getIn().getBody(SnmpMessage.class))); - //POLL consumer startingWith dot snmpVersion=1 - from("snmp://" + snmpListenAddress - + "?protocol=udp&snmpVersion=1&type=POLL&oids=.1.3.6.1.4.1.6527.3.1.2.21.2.1.50") - .process(e -> snmpResults.get("v1_pollStartingDot").add(e.getIn().getBody(SnmpMessage.class))); - - //POLL consumer snmpVersion=0 - from("snmp://" + snmpListenAddress + "?protocol=udp&snmpVersion=0&type=POLL&oids=" - + SnmpConstants.sysDescr) - .process(e -> snmpResults.get("v0_poll").add(e.getIn().getBody(SnmpMessage.class))); - //POLL consumer snmpVersion=1 - from("snmp://" + snmpListenAddress + "?protocol=udp&snmpVersion=1&type=POLL&oids=" - + SnmpConstants.sysDescr) - .process(e -> snmpResults.get("v1_poll").add(e.getIn().getBody(SnmpMessage.class))); } static class Producers { @@ -95,13 +64,6 @@ public class SnmpRoute extends RouteBuilder { Map<String, Deque<SnmpMessage>> map = new ConcurrentHashMap<>(); map.put("v0_trap", new ConcurrentLinkedDeque<>()); map.put("v1_trap", new ConcurrentLinkedDeque<>()); - map.put("v0_poll", new ConcurrentLinkedDeque<>()); - map.put("v1_poll", new ConcurrentLinkedDeque<>()); - map.put("v3_poll", new ConcurrentLinkedDeque<>()); - map.put("v0_pollStartingDot", new ConcurrentLinkedDeque<>()); - map.put("v1_pollStartingDot", new ConcurrentLinkedDeque<>()); - map.put("v0_poll2oids", new ConcurrentLinkedDeque<>()); - map.put("v1_poll2oids", new ConcurrentLinkedDeque<>()); return map; } } diff --git a/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTest.java b/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTest.java index fac78e768c..1220e8aa44 100644 --- a/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTest.java +++ b/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTest.java @@ -22,9 +22,11 @@ import java.util.stream.Stream; import io.quarkus.test.common.QuarkusTestResource; import io.quarkus.test.junit.QuarkusTest; import io.restassured.RestAssured; +import io.restassured.specification.RequestSpecification; import org.hamcrest.Matchers; import org.junit.jupiter.params.ParameterizedTest; import org.junit.jupiter.params.provider.MethodSource; +import org.junit.jupiter.params.provider.ValueSource; import org.snmp4j.mp.SnmpConstants; import org.snmp4j.smi.OID; @@ -76,38 +78,44 @@ class SnmpTest { } @ParameterizedTest - @MethodSource("supportedVersions") + @ValueSource(ints = { 0, 1, 3 }) public void testPoll(int version) throws Exception { - String resultsName = "v" + version + "_poll"; + RequestSpecification rs = RestAssured.given() + .body(POLL_OID.toString()); - await().atMost(20L, TimeUnit.SECONDS).pollDelay(100, TimeUnit.MILLISECONDS).until(() -> { - String result = RestAssured.given() - .body(SnmpConstants.sysDescr.toString()) - .post("/snmp/results/" + resultsName) - .then() - .statusCode(200) - .extract().body().asString(); + if (version == 3) { + rs.queryParam("user", "test") + .queryParam("securityLevel", 1); + } - return result - .startsWith("My POLL Printer - response #1,My POLL Printer - response #2,My POLL Printer - response #3"); - }); + rs.post("/snmp/poll/" + version) + .then() + .statusCode(200) + .body(Matchers.equalTo("My POLL Printer - response #1")); } @ParameterizedTest @MethodSource("supportedVersions") - public void testPollStartingDot(int version) throws Exception { - String resultsName = "v" + version + "_pollStartingDot"; - - await().atMost(20L, TimeUnit.SECONDS).pollDelay(100, TimeUnit.MILLISECONDS).until(() -> { - String result = RestAssured.given() - .body(DOT_OID.toString()) - .post("/snmp/results/" + resultsName) - .then() - .statusCode(200) - .extract().body().asString(); + public void testPollWith2OIDs(int version) throws Exception { + RestAssured.given() + .body(TWO_OIDS_A + "," + TWO_OIDS_B) + .post("/snmp/poll/" + version) + .then() + .statusCode(200) + .body(Matchers.anyOf( + Matchers.equalTo("My 2 OIDs A Printer - response #1,My 2 OIDs B Printer - response #1"), + Matchers.equalTo("My 2 OIDs B Printer - response #1,My 2 OIDs A Printer - response #1"))); + } - return result.startsWith("My DOT Printer - response #1,My DOT Printer - response #2,My DOT Printer - response #3"); - }); + @ParameterizedTest + @MethodSource("supportedVersions") + public void testPollStartingDot(int version) throws Exception { + RestAssured.given() + .body("." + DOT_OID) + .post("/snmp/poll/" + version) + .then() + .statusCode(200) + .body(Matchers.equalTo("My DOT Printer - response #1")); } @ParameterizedTest @@ -134,30 +142,4 @@ class SnmpTest { .body(Matchers.endsWith("2,My GET_NEXT Printer - response #3")); } - @ParameterizedTest - @MethodSource("supportedVersions") - public void testPollWith2OIDs(int version) throws Exception { - String resultsName = "v" + version + "_poll2oids"; - - await().atMost(20L, TimeUnit.SECONDS).pollDelay(100, TimeUnit.MILLISECONDS).until(() -> { - String resultOid1 = RestAssured.given() - .body(TWO_OIDS_A.toString()) - .post("/snmp/results/" + resultsName) - .then() - .statusCode(200) - .extract().body().asString(); - String resultOid2 = RestAssured.given() - .body(TWO_OIDS_B.toString()) - .post("/snmp/results/" + resultsName) - .then() - .statusCode(200) - .extract().body().asString(); - - return resultOid1.startsWith( - "My 2 OIDs A Printer - response #1,My 2 OIDs A Printer - response #2,My 2 OIDs A Printer - response #3") && - resultOid2.startsWith( - "My 2 OIDs B Printer - response #1,My 2 OIDs B Printer - response #2,My 2 OIDs B Printer - response #3"); - }); - - } } diff --git a/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTestResource.java b/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTestResource.java index 769c7824b6..c487d4acdc 100644 --- a/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTestResource.java +++ b/integration-tests-jvm/snmp/src/test/java/org/apache/camel/quarkus/component/snmp/it/SnmpTestResource.java @@ -34,8 +34,16 @@ import org.snmp4j.PDU; import org.snmp4j.PDUv1; import org.snmp4j.ScopedPDU; import org.snmp4j.Snmp; +import org.snmp4j.mp.CounterSupport; +import org.snmp4j.mp.DefaultCounterListener; +import org.snmp4j.mp.MPv3; import org.snmp4j.mp.SnmpConstants; import org.snmp4j.mp.StatusInformation; +import org.snmp4j.security.AuthSHA; +import org.snmp4j.security.SecurityModels; +import org.snmp4j.security.SecurityProtocols; +import org.snmp4j.security.USM; +import org.snmp4j.security.UsmUser; import org.snmp4j.smi.OID; import org.snmp4j.smi.OctetString; import org.snmp4j.smi.UdpAddress; @@ -51,6 +59,7 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { @Override public Map<String, String> start() { + SecurityProtocols.getInstance().addDefaultProtocols(); DefaultUdpTransportMapping udpTransportMapping; try { udpTransportMapping = new DefaultUdpTransportMapping(new UdpAddress(LOCAL_ADDRESS)); @@ -59,6 +68,20 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { TestCommandResponder responder = new TestCommandResponder(snmpResponder); snmpResponder.addCommandResponder(responder); + SecurityModels respSecModels = new SecurityModels() { + }; + + CounterSupport.getInstance().addCounterListener(new DefaultCounterListener()); + MPv3 mpv3CR = (MPv3) snmpResponder.getMessageDispatcher().getMessageProcessingModel(MPv3.ID); + mpv3CR.setLocalEngineID(MPv3.createLocalEngineID(new OctetString("responder"))); + respSecModels.addSecurityModel(new USM(SecurityProtocols.getInstance(), + new OctetString(mpv3CR.getLocalEngineID()), 0)); + mpv3CR.setSecurityModels(respSecModels); + + snmpResponder.getUSM().addUser( + new UsmUser(new OctetString("test"), AuthSHA.ID, new OctetString("changeit"), + AuthSHA.ID, new OctetString("changeit"))); + snmpResponder.listen(); } catch (Exception e) { throw new RuntimeException(e); @@ -75,6 +98,7 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { @Override public void stop() { + SecurityProtocols.setSecurityProtocols(null); if (snmpResponder != null) { try { snmpResponder.close(); @@ -115,7 +139,7 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { int numberOfSent = counts.getOrDefault(key, 0); try { - PDU response = makeResponse(++numberOfSent, SnmpConstants.version1, vbs); + PDU response = makeResponse(pdu, ++numberOfSent, SnmpConstants.version1, vbs); if (response != null) { response.setRequestID(pdu.getRequestID()); commandResponder.getMessageDispatcher().returnResponsePdu( @@ -130,18 +154,18 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { counts.put(key, numberOfSent); } - private PDU makeResponse(int counter, int version, Vector<? extends VariableBinding> vbs) { - PDU responsePDU = new PDU(); + private PDU makeResponse(PDU originalPDU, int counter, int version, Vector<? extends VariableBinding> vbs) { + PDU responsePDU = (PDU) originalPDU.clone(); responsePDU.setType(PDU.RESPONSE); responsePDU.setErrorStatus(PDU.noError); responsePDU.setErrorIndex(0); if (vbs.isEmpty()) { - VariableBinding vb = generateResponseBinding(counter, SnmpTest.PRODUCE_PDU_OID); + VariableBinding vb = generateResponseBinding(counter, version, SnmpTest.PRODUCE_PDU_OID); if (vb != null) { responsePDU.add(vb); } } else { - vbs.stream().forEach(vb -> responsePDU.add(generateResponseBinding(counter, vb.getOid()))); + vbs.stream().forEach(vb -> responsePDU.add(generateResponseBinding(counter, version, vb.getOid()))); } if (responsePDU.getVariableBindings().isEmpty()) { return null; @@ -149,7 +173,7 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { return responsePDU; } - private VariableBinding generateResponseBinding(int counter, OID oid) { + private VariableBinding generateResponseBinding(int counter, int version, OID oid) { //get next test if (SnmpTest.GET_NEXT_OID.equals(oid)) { //if counter < 2 return the same oid @@ -166,11 +190,12 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { } if (SnmpTest.POLL_OID.equals(oid)) { - if (counter < 4) { - return new VariableBinding(SnmpTest.POLL_OID, - new OctetString("My POLL Printer - response #" + counter)); + if (counter > 1) { + throw new RuntimeException( + String.format("Not expected request #%d for poll of version %d.", counter, version)); } - + return new VariableBinding(SnmpTest.POLL_OID, + new OctetString("My POLL Printer - response #" + counter)); } if (SnmpTest.PRODUCE_PDU_OID.equals(oid)) { @@ -195,10 +220,8 @@ public class SnmpTestResource implements QuarkusTestResourceLifecycleManager { } if (SnmpTest.DOT_OID.equals(oid)) { - if (counter < 4) { - return new VariableBinding(SnmpTest.DOT_OID, - new OctetString("My DOT Printer - response #" + counter)); - } + return new VariableBinding(SnmpTest.DOT_OID, + new OctetString("My DOT Printer - response #" + counter)); } return null;