1 files changed, 117 insertions, 0 deletions
diff --git a/src/main/java/at/gv/util/client/szr/SZRSOAPHandler.java b/src/main/java/at/gv/util/client/szr/SZRSOAPHandler.java
new file mode 100644
index 0000000..05e5004
--- /dev/null
+++ b/src/main/java/at/gv/util/client/szr/SZRSOAPHandler.java
@@ -0,0 +1,117 @@
+package at.gv.util.client.szr;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.util.Set;
+
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBException;
+import javax.xml.namespace.QName;
+import javax.xml.parsers.ParserConfigurationException;
+import javax.xml.soap.SOAPElement;
+import javax.xml.soap.SOAPEnvelope;
+import javax.xml.soap.SOAPException;
+import javax.xml.soap.SOAPFactory;
+import javax.xml.soap.SOAPHeader;
+import javax.xml.soap.SOAPMessage;
+import javax.xml.ws.handler.MessageContext;
+import javax.xml.ws.handler.soap.SOAPHandler;
+import javax.xml.ws.handler.soap.SOAPMessageContext;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.w3c.dom.Element;
+import org.xml.sax.SAXException;
+
+import at.gv.util.DOMUtils;
+import at.gv.util.MiscUtil;
+import at.gv.util.xsd.szr.pvp.ObjectFactory;
+import at.gv.util.xsd.szr.pvp.PvpTokenType;
+
+public class SZRSOAPHandler implements SOAPHandler<SOAPMessageContext> {
+
+	private static final String AUTH_NS = "http://schemas.xmlsoap.org/ws/2002/04/secext";
+  private static final String AUTH_PREFIX="wss";
+
+  private PvpTokenType pvpTokenType = null;
+  
+  private Logger log = LoggerFactory.getLogger(SZRSOAPHandler.class);
+	
+
+  public void close(MessageContext arg0) {
+  }
+
+
+  public boolean handleFault(SOAPMessageContext arg0) {
+	  return true;
+  }
+
+	public void configure(PvpTokenType pvpToken) {
+		MiscUtil.assertNotNull(pvpToken, "pvpToken");
+		this.pvpTokenType = pvpToken;
+	}
+	
+
+  public boolean handleMessage(SOAPMessageContext smc) {
+		log.trace("Initializing SZR SOAP message handler.");
+
+		boolean isOutMessage = ((Boolean) smc.get(SOAPMessageContext.MESSAGE_OUTBOUND_PROPERTY)).booleanValue();
+		log.trace("Outbound message: " + isOutMessage);
+		if (pvpTokenType == null) {
+			throw new NullPointerException("Please configure first the PVP token.");
+		}
+		
+		if (!isOutMessage) {
+			return true;
+		}
+		try {
+			SOAPMessage message = smc.getMessage();
+			SOAPEnvelope envelope = message.getSOAPPart().getEnvelope(); 
+			SOAPFactory soapFactory = SOAPFactory.newInstance();
+						
+			// Creating WS-Security header element
+			SOAPElement wsSecHeaderElm = soapFactory.createElement(
+					"Security",
+					AUTH_PREFIX,
+					AUTH_NS);
+			
+			// serialize pvp token
+			JAXBContext ctx = JAXBContext.newInstance(PvpTokenType.class);
+			ObjectFactory of = new ObjectFactory();
+			ByteArrayOutputStream bos = new ByteArrayOutputStream();
+			ctx.createMarshaller().marshal(of.createPvpToken(pvpTokenType), bos);
+			Element pvpTokenElement = DOMUtils.parseXmlNonValidating(new ByteArrayInputStream(bos.toByteArray()));
+			SOAPElement pvpToken = soapFactory.createElement(pvpTokenElement);
+			
+			// adding elements
+			wsSecHeaderElm.addChildElement(pvpToken);
+			SOAPHeader header = envelope.addHeader(); 
+      header.addChildElement(wsSecHeaderElm);
+      
+      return true;
+		} catch(SOAPException e) {
+			log.warn("Cannot add WS-Security header.", e);
+			return false;
+		} catch (JAXBException e) {
+			log.warn("Cannot add WS-Security header.", e);
+			return false;
+    } catch (ParserConfigurationException e) {
+    	log.warn("Cannot add WS-Security header.", e);
+			return false;
+    } catch (SAXException e) {
+    	log.warn("Cannot add WS-Security header.", e);
+			return false;
+    } catch (IOException e) {
+    	log.warn("Cannot add WS-Security header.", e);
+			return false;
+    }
+
+	
+  }
+
+  public Set<QName> getHeaders() {
+	  return null;
+  }
+
+}