Better logging for simple TLS workflows

Author: jurajsomorovsky

Runnable/src/main/java/de/rub/nds/tlsattacker/Main.java

@@ -46,6 +46,8 @@
 import de.rub.nds.tlsattacker.tls.config.GeneralConfig;
 import de.rub.nds.tlsattacker.tls.config.WorkflowTraceSerializer;
 import de.rub.nds.tlsattacker.tls.exceptions.ConfigurationException;
+import de.rub.nds.tlsattacker.tls.exceptions.WorkflowExecutionException;
+import de.rub.nds.tlsattacker.tls.util.LogLevel;
 import de.rub.nds.tlsattacker.tls.workflow.TlsContext;
 import de.rub.nds.tlsattacker.tls.workflow.WorkflowExecutor;
 import de.rub.nds.tlsattacker.transport.TransportHandler;
@@ -54,13 +56,17 @@
 import java.io.IOException;
 import java.util.List;
 import javax.xml.bind.JAXBException;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 
 /**
  * 
  * @author Juraj Somorovsky <juraj.somorovsky@rub.de>
  */
 public class Main {
 
+    private static final Logger LOGGER = LogManager.getLogger(Main.class);
+
     public static void main(String[] args) throws Exception {
 
 	GeneralConfig generalConfig = new GeneralConfig();
@@ -172,7 +178,13 @@ private static void startSimpleTls(GeneralConfig generalConfig, CommandConfig co
 	TlsContext tlsContext = configHandler.initializeTlsContext(config);
 	WorkflowExecutor workflowExecutor = configHandler.initializeWorkflowExecutor(transportHandler, tlsContext);
 
-	workflowExecutor.executeWorkflow();
+	try {
+	    workflowExecutor.executeWorkflow();
+	} catch (WorkflowExecutionException ex) {
+	    LOGGER.info(ex.getLocalizedMessage(), ex);
+	    LOGGER.log(LogLevel.CONSOLE_OUTPUT,
+		    "The TLS protocol flow was not executed completely, follow the debug messages for more information.");
+	}
 
 	transportHandler.closeConnection();