fix missing upstream timer shutdown

src/main/java/com/siemens/pki/lightweightcmpra/main/RA.java

@@ -111,14 +111,15 @@ public static void stopAllRas() {
 
     private DownstreamInterface downstreamInterface;
     private String configFile;
+    private Map<CertProfileBodyTypeTupel, UpstreamInterface> upstreamInterfaceMap;
 
     private RA(final String actConfigFile) throws Exception {
         configFile = actConfigFile;
 
         try {
             final ConfigurationImpl configuration = YamlConfigLoader.loadConfig(configFile, ConfigurationImpl.class);
             final DeferredSupplier<CmpRaInterface> raHolder = new DeferredSupplier<>();
-            final Map<CertProfileBodyTypeTupel, UpstreamInterface> upstreamInterfaceMap = new HashMap<>();
+            upstreamInterfaceMap = new HashMap<>();
             final UpstreamExchange upstreamExchange = (request, certProfile, bodyTypeOfFirstRequest) -> {
                 final CertProfileBodyTypeTupel key = new CertProfileBodyTypeTupel(certProfile, bodyTypeOfFirstRequest);
                 UpstreamInterface upstreamInterface = upstreamInterfaceMap.get(key);
@@ -146,6 +147,7 @@ private void stop() {
         if (downstreamInterface != null) {
             downstreamInterface.stop();
         }
+        upstreamInterfaceMap.values().forEach(UpstreamInterface::stop);
         System.out.println("RA configured with " + configFile + " stopped");
     }
 }

src/main/java/com/siemens/pki/lightweightcmpra/upstream/UpstreamInterface.java

@@ -36,4 +36,6 @@ interface AsyncResponseHandler {
      *            the callback
      */
     void setDelayedResponseHandler(AsyncResponseHandler asyncResponseHandler);
+
+    void stop();
 }

src/main/java/com/siemens/pki/lightweightcmpra/upstream/offline/CmpFileOfflineClient.java

@@ -45,6 +45,7 @@ public class CmpFileOfflineClient implements UpstreamInterface {
 
     private final File outputDirectory;
     private AsyncResponseHandler asyncResponseHandler;
+    private TimerTask timerTask;
 
     /**
      *
@@ -66,14 +67,14 @@ public CmpFileOfflineClient(final OfflineFileClientConfig config) throws IOExcep
         }
         final long pollInterval = config.getInputDirectoryPollcycle() * 1000L;
         final Timer pollTimer = new Timer(true);
-        final TimerTask task = new TimerTask() {
+        timerTask = new TimerTask() {
 
             @Override
             public void run() {
                 pollInputDirectory();
             }
         };
-        pollTimer.schedule(task, new Date(System.currentTimeMillis() + pollInterval), pollInterval);
+        pollTimer.schedule(timerTask, new Date(System.currentTimeMillis() + pollInterval), pollInterval);
     }
 
     @Override
@@ -118,4 +119,9 @@ private void pollInputDirectory() {
             }
         }
     }
+
+    @Override
+    public void stop() {
+        timerTask.cancel();
+    }
 }

src/main/java/com/siemens/pki/lightweightcmpra/upstream/online/CmpCoapClient.java

@@ -53,4 +53,9 @@ public byte[] apply(final byte[] message, final String certProfile) {
     public void setDelayedResponseHandler(final AsyncResponseHandler asyncResponseHandler) {
         // no async response expected
     }
+
+    @Override
+    public void stop() {
+        client.shutdown();
+    }
 }

src/main/java/com/siemens/pki/lightweightcmpra/upstream/online/CmpHttpClient.java

@@ -107,4 +107,10 @@ public byte[] apply(final byte[] message, final String certProfile) {
     public void setDelayedResponseHandler(final AsyncResponseHandler asyncResponseHandler) {
         // no async response expected
     }
+
+    @Override
+    public void stop() {
+        // nothing to do
+
+    }
 }

src/test/java/com/siemens/pki/lightweightcmpclient/test/TestDelayedEnrollmentAndRevocation.java

@@ -0,0 +1,65 @@
+/*
+ *  Copyright (c) 2023 Siemens AG
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License"); you may
+ *  not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ *  SPDX-License-Identifier: Apache-2.0
+ */
+package com.siemens.pki.lightweightcmpclient.test;
+
+import com.siemens.pki.lightweightcmpra.main.RA;
+import com.siemens.pki.lightweightcmpra.test.framework.TestUtils;
+import java.security.GeneralSecurityException;
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+public class TestDelayedEnrollmentAndRevocation extends EnrollmentTestcaseBase {
+
+    private static final String UPSTREAM_DIR = "./target/CmpTest/Upstream_REV";
+    private static final String DOWNSTREAM_DIR = "./target/CmpTest/Downstream_REV";
+
+    @BeforeClass
+    public static void setupRas() throws GeneralSecurityException, InterruptedException, Exception {
+        TestUtils.createDirectories(DOWNSTREAM_DIR, UPSTREAM_DIR);
+        initTestbed("DelayedEnrollmentRaTestConfig.yaml", "DelayedEnrollmentLraTestConfig.yaml");
+    }
+
+    @AfterClass
+    public static void stopAllRas() {
+        RA.stopAllRas();
+        TestUtils.removeDirectories(DOWNSTREAM_DIR, UPSTREAM_DIR);
+    }
+
+    @After
+    public void cleanUpDelayedEnrollmentDirs() {
+        TestUtils.deleteAllFilesIn(DOWNSTREAM_DIR, UPSTREAM_DIR);
+    }
+
+    @Before
+    public void setUpDelayedEnrollmentDirs() throws Exception {
+        TestUtils.deleteAllFilesIn(DOWNSTREAM_DIR, UPSTREAM_DIR);
+    }
+
+    @Test(timeout = 100000L)
+    public void testCrWithPolling() throws Exception {
+        enrollWithConfig("DelayedClientEnrollmentConfigWithHttpAndSignature.yaml");
+    }
+
+    @Test(timeout = 100000L)
+    public void testRrWithPolling() throws Exception {
+        revokeWithConfigAndCert("DelayedClientEnrollmentConfigWithHttpAndSignature.yaml");
+    }
+}

src/test/java/com/siemens/pki/lightweightcmpclient/test/TestGeneralMessagesWithPolling.java

@@ -31,8 +31,8 @@
 
 public class TestGeneralMessagesWithPolling extends CmpTestcaseBase {
 
-    private static final String UPSTREAM_DIR = "./target/CmpTest/GenUpstream";
-    private static final String DOWNSTREAM_DIR = "./target/CmpTest/GenDownstream";
+    private static final String UPSTREAM_DIR = "./target/CmpTest/GenUpstream_GEN";
+    private static final String DOWNSTREAM_DIR = "./target/CmpTest/GenDownstream_GEN";
 
     @AfterClass
     public static void cleanUpDirsAnRas() {
@@ -56,7 +56,7 @@ public void cleanDirectories() {
      *
      * @throws Exception
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testCrlUpdateRetrieval() throws Exception {
         final String cmdArgs = "--configfile " + "DelayedClientGeneralMessagesWithHttpAndSignature.yaml" + " "
                 + "--getCrls ./target/CmpTest/Results/CRLs.crl " + "--issuer CN=distributionPoint ";
@@ -70,7 +70,7 @@ public void testCrlUpdateRetrieval() throws Exception {
      *
      * @throws Exception
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testCrlUpdateRetrievalWithOldCrl() throws Exception {
         final String cmdArgs = "--configfile " + "DelayedClientGeneralMessagesWithHttpAndSignature.yaml" + " "
                 + "--getCrls ./target/CmpTest/Results/CRLs.crl "
@@ -83,7 +83,7 @@ public void testCrlUpdateRetrievalWithOldCrl() throws Exception {
     /*
      * Get CA certificates
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testGetCaCerts() throws Exception {
         final String cmdArgs = "--configfile " + "DelayedClientGeneralMessagesWithHttpAndSignature.yaml" + " "
                 + "--getCaCertificates ./target/CmpTest/Results/Certificates.cer ";
@@ -95,7 +95,7 @@ public void testGetCaCerts() throws Exception {
     /*
      * Get certificate request template
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testGetCertificateRequestTemplate() throws Exception {
 
         final String cmdArgs = "--configfile " + "DelayedClientGeneralMessagesWithHttpAndSignature.yaml" + " "
@@ -108,7 +108,7 @@ public void testGetCertificateRequestTemplate() throws Exception {
     /*
      * Get root CA certificate update
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testGetRootCaKeyUpdateInfo() throws Exception {
         final String cmdArgs = "--configfile " + "DelayedClientGeneralMessagesWithHttpAndSignature.yaml" + " "
                 + "--getRootCaCertificateUpdate " + "--NewWithNew ./target/CmpTest/Results/NewWithNew.cer "

src/test/java/com/siemens/pki/lightweightcmpra/test/DelayedEnrollmentTescaseBase.java

@@ -34,6 +34,7 @@
 import org.bouncycastle.asn1.crmf.CertTemplateBuilder;
 import org.bouncycastle.asn1.x500.X500Name;
 import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo;
+import org.junit.After;
 import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
@@ -42,6 +43,9 @@
 
 public class DelayedEnrollmentTescaseBase extends EnrollmentTestcaseBase {
 
+    private static final String UPSTREAM_DIR = "./target/CmpTest/Upstream_REV";
+    private static final String DOWNSTREAM_DIR = "./target/CmpTest/Downstream_REV";
+
     private static final Logger LOGGER = LoggerFactory.getLogger(DelayedEnrollmentTescaseBase.class);
 
     protected static EnrollmentResult executeDelayedCertificateRequest(
@@ -96,20 +100,25 @@ protected static EnrollmentResult executeDelayedCertificateRequest(
 
     @AfterClass
     public static void removeDirs() {
-        TestUtils.removeDirectories("./target/CmpTest/Downstream", "./target/CmpTest/Upstream");
+        TestUtils.removeDirectories(DOWNSTREAM_DIR, UPSTREAM_DIR);
     }
 
     @BeforeClass
     public static void setUpDirsAndRas() throws Exception {
-        TestUtils.createDirectories("./target/CmpTest/Downstream", "./target/CmpTest/Upstream");
+        TestUtils.createDirectories(DOWNSTREAM_DIR, UPSTREAM_DIR);
         initTestbed(
                 "http://localhost:6003/delayedlra",
                 "DelayedEnrollmentRaTestConfig.yaml",
                 "DelayedEnrollmentLraTestConfig.yaml");
     }
 
     @Before
-    public void clearDirs() {
-        TestUtils.deleteAllFilesIn("./target/CmpTest/Downstream", "./target/CmpTest/Upstream");
+    public void clearDirsBefore() {
+        TestUtils.deleteAllFilesIn(DOWNSTREAM_DIR, UPSTREAM_DIR);
+    }
+
+    @After
+    public void clearDirsAfter() {
+        TestUtils.deleteAllFilesIn(DOWNSTREAM_DIR, UPSTREAM_DIR);
     }
 }

src/test/java/com/siemens/pki/lightweightcmpra/test/TestCrWithNested.java

@@ -35,7 +35,7 @@ public static void setUpRas() throws Exception {
      *
      * @throws Exception
      */
-    @Test(timeout = 10000000L)
+    @Test(timeout = 100000L)
     public void testCrWithNested() throws Exception {
         executeCrmfCertificateRequest(
                 PKIBody.TYPE_CERT_REQ,

src/test/java/com/siemens/pki/lightweightcmpra/test/TestCrWithPolling.java

@@ -27,7 +27,7 @@ public class TestCrWithPolling extends DelayedEnrollmentTescaseBase {
      *
      * @throws Exception
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testCrWithPolling() throws Exception {
         executeDelayedCertificateRequest(
                 PKIBody.TYPE_CERT_REQ,

src/test/java/com/siemens/pki/lightweightcmpra/test/TestGeneralMessagesWithPolling.java

@@ -54,25 +54,25 @@
 import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
-import org.junit.Ignore;
 import org.junit.Test;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-@Ignore
 public class TestGeneralMessagesWithPolling extends CmpTestcaseBase {
 
+    private static final String UPSTREAM_DIR = "./target/CmpTest/GenUpstream_GEN";
+    private static final String DOWNSTREAM_DIR = "./target/CmpTest/GenDownstream_GEN";
     private static final Logger LOGGER = LoggerFactory.getLogger(TestGeneralMessagesWithPolling.class);
 
     @AfterClass
     public static void removeDirs() {
         RA.stopAllRas();
-        TestUtils.removeDirectories("./target/CmpTest/GenDownstream", "./target/CmpTest/GenUpstream");
+        TestUtils.removeDirectories(DOWNSTREAM_DIR, UPSTREAM_DIR);
     }
 
     @BeforeClass
     public static void setUpDirsAndRas() throws Exception {
-        TestUtils.createDirectories("./target/CmpTest/GenDownstream", "./target/CmpTest/GenUpstream");
+        TestUtils.createDirectories(DOWNSTREAM_DIR, UPSTREAM_DIR);
         initTestbed(
                 "http://localhost:6006/delayedsupportlra",
                 "DelayedSupportMessagesRaTestConfig.yaml",
@@ -81,15 +81,15 @@ public static void setUpDirsAndRas() throws Exception {
 
     @Before
     public void clearDirs() {
-        TestUtils.deleteAllFilesIn("./target/CmpTest/GenDownstream", "./target/CmpTest/GenUpstream");
+        TestUtils.deleteAllFilesIn(DOWNSTREAM_DIR, UPSTREAM_DIR);
     }
 
     /**
      * CRL Update Retrieval
      *
      * @throws Exception
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testCrlUpdateRetrieval() throws Exception {
         final Function<PKIMessage, PKIMessage> eeCmpClient = getEeCmpClient();
         final ASN1ObjectIdentifier statusListOid = new ASN1ObjectIdentifier("1.3.6.1.5.5.7.4.22");
@@ -132,7 +132,7 @@ null, new GeneralNames(new GeneralName(new X500Name("CN=distributionPoint")))),
     /*
      * Get CA certificates
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testGetCaCerts() throws Exception {
         final Function<PKIMessage, PKIMessage> eeCmpClient = getEeCmpClient();
         final ASN1ObjectIdentifier getCaCertOid = new ASN1ObjectIdentifier("1.3.6.1.5.5.7.4.17");
@@ -166,7 +166,7 @@ public void testGetCaCerts() throws Exception {
     /*
      * Get certificate request template
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testGetCertificateRequestTemplate() throws Exception {
         final Function<PKIMessage, PKIMessage> eeCmpClient = getEeCmpClient();
         final ASN1ObjectIdentifier getCaCertOid = new ASN1ObjectIdentifier("1.3.6.1.5.5.7.4.19");
@@ -215,7 +215,7 @@ public void testGetCertificateRequestTemplate() throws Exception {
     /*
      * Get root CA certificate update
      */
-    @Test
+    @Test(timeout = 100000L)
     public void testGetRootCaKeyUpdateInfo() throws Exception {
         final Function<PKIMessage, PKIMessage> eeCmpClient = getEeCmpClient();
         final ASN1ObjectIdentifier getCaCertOid = new ASN1ObjectIdentifier("1.3.6.1.5.5.7.4.20");

src/test/java/com/siemens/pki/lightweightcmpra/test/TestRrWithPolling.java

@@ -39,7 +39,7 @@ public class TestRrWithPolling extends DelayedEnrollmentTescaseBase {
      *
      * @throws Exception
      */
-    @Test(timeout = 60000L)
+    @Test(timeout = 100000L)
     public void testRrWithPolling() throws Exception {
         final EnrollmentResult certificateToRevoke = executeDelayedCertificateRequest(
                 PKIBody.TYPE_CERT_REQ,

src/test/java/com/siemens/pki/lightweightcmpra/test/config/DelayedEnrollmentLraTestConfig.yaml

@@ -8,8 +8,8 @@ DownstreamInterface:
 UpstreamInterface:
   - OfflineFileClient:
       certProfile: ~
-      inputDirectory: "./target/CmpTest/Downstream"
-      outputDirectory: "./target/CmpTest/Upstream"
+      inputDirectory: "./target/CmpTest/Downstream_REV"
+      outputDirectory: "./target/CmpTest/Upstream_REV"
 
 UpstreamConfiguration:
   - VerificationContext:
@@ -53,6 +53,6 @@ RetryAfterTimeInSeconds:
   - value: 1
 
 DownstreamTimeout:
-  - value: 20
+  - value: 100
 
 

src/test/java/com/siemens/pki/lightweightcmpra/test/config/DelayedEnrollmentRaTestConfig.yaml

@@ -7,8 +7,8 @@ UpstreamInterface:
 
 DownstreamInterface:
   OfflineFileServer:
-    inputDirectory: "./target/CmpTest/Upstream"
-    outputDirectory: "./target/CmpTest/Downstream"
+    inputDirectory: "./target/CmpTest/Upstream_REV"
+    outputDirectory: "./target/CmpTest/Downstream_REV"
 
 UpstreamConfiguration:
   - VerificationContext:
@@ -40,4 +40,4 @@ ForceRaVerifyOnUpstream:
   - value: false
 
 DownstreamTimeout:
-  - value: 20
+  - value: 100