ca.uhn.hl7v2.hoh.llp.HohLlpReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hapi-hl7overhttp Show documentation
There is a newer version: 2.5.1
package ca.uhn.hl7v2.hoh.llp;

import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;

import ca.uhn.hl7v2.hoh.api.DecodeException;
import ca.uhn.hl7v2.hoh.api.IAuthorizationServerCallback;
import ca.uhn.hl7v2.hoh.encoder.AbstractHl7OverHttpDecoder;
import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpRequestDecoder;
import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpResponseDecoder;
import ca.uhn.hl7v2.hoh.encoder.NoMessageReceivedException;
import ca.uhn.hl7v2.hoh.sign.SignatureVerificationException;
import ca.uhn.hl7v2.hoh.util.HTTPUtils;
import ca.uhn.hl7v2.hoh.util.ServerRoleEnum;
import ca.uhn.hl7v2.llp.HL7Reader;
import ca.uhn.hl7v2.llp.LLPException;

class HohLlpReader implements HL7Reader {
	private static final org.slf4j.Logger ourLog = org.slf4j.LoggerFactory.getLogger(HohLlpReader.class);

	private InputStream myInputStream;
	private Hl7OverHttpLowerLayerProtocol myProtocol;
	private HohLlpWriter myWriter;

	/**
	 * Constructor
	 */
	public HohLlpReader(Hl7OverHttpLowerLayerProtocol theProtocol) {
		myProtocol = theProtocol;
	}

	/**
	 * {@inheritDoc}
	 */
	public void close() throws IOException {
		myInputStream.close();
	}

	public String getMessage() throws LLPException, IOException {
		AbstractHl7OverHttpDecoder decoder;

		if (myProtocol.getRole() == ServerRoleEnum.CLIENT) {
			decoder = new Hl7OverHttpResponseDecoder();
		} else {
			Hl7OverHttpRequestDecoder requestDecoder = new Hl7OverHttpRequestDecoder();
			requestDecoder.setAuthorizationCallback(myProtocol.getAuthorizationServerCallback());
			decoder = requestDecoder;
			decoder.setSigner(myProtocol.getSigner());
		}

		
		try {
			decoder.readHeadersAndContentsFromInputStreamAndDecode(myInputStream);
		} catch (DecodeException e) {
			if (myProtocol.getRole() == ServerRoleEnum.CLIENT) {
				throw new LLPException("Failed to process response", e);
			} else {
				ourLog.info("Failed to read contents", e);
				HTTPUtils.write400BadRequest(myWriter.getOutputStream(), e.getMessage());

				myInputStream.close();
				myWriter.getOutputStream().close();

				throw new LLPException("Failed to read message", e);
			}
		} catch (NoMessageReceivedException e) {
			return null;
		} catch (SignatureVerificationException e) {
			throw new LLPException("Failed to verify message signature", e);
		}
		
		if (myProtocol.getRole() == ServerRoleEnum.SERVER) {
			Charset charset = decoder.getCharset();
			myWriter.setCharsetForNextMessage(charset);
			
			IAuthorizationServerCallback authorizationCallback = myProtocol.getAuthorizationServerCallback();
			if (authorizationCallback != null) {
				boolean auth = authorizationCallback.authorize(decoder.getPath(), decoder.getUsername(), decoder.getPassword());
				if (!auth) {
					HTTPUtils.write401Unauthorized(myWriter.getOutputStream());
					throw new LLPException("Authorization at URI[" + decoder.getPath() + "] failed for user[" + decoder.getUsername() + "]");
				}
			}
		}

		return decoder.getMessage();

	}

	public void setInputStream(InputStream theInputStream) throws IOException {
		myInputStream = theInputStream;
	}

	void setWriter(HohLlpWriter theWriter) {
		myWriter = theWriter;
	}

}