1 files changed, 259 insertions, 0 deletions

diff --git a/src/org/uic/barcode/Encoder.java b/src/org/uic/barcode/Encoder.java
new file mode 100644
index 0000000..301b7c5
--- /dev/null
+++ b/src/org/uic/barcode/Encoder.java
@@ -0,0 +1,259 @@
+package org.uic.barcode;

+

+import java.io.IOException;

+import java.security.PrivateKey;

+import java.security.PublicKey;

+

+import org.uic.barcode.dynamicFrame.Constants;

+import org.uic.barcode.dynamicFrame.DataType;

+import org.uic.barcode.dynamicFrame.DynamicFrame;

+import org.uic.barcode.dynamicFrame.Level1DataType;

+import org.uic.barcode.dynamicFrame.Level2DataType;

+import org.uic.barcode.dynamicFrame.SequenceOfDataType;

+import org.uic.barcode.staticFrame.StaticFrame;

+import org.uic.barcode.staticFrame.UFLEXDataRecord;

+import org.uic.barcode.staticFrame.UHEADDataRecord;

+import org.uic.barcode.staticFrame.UTLAYDataRecord;

+import org.uic.barcode.staticHeader.ticketLayoutBarcode.TicketLayout;

+import org.uic.ticket.EncodingFormatException;

+import org.uic.ticket.UicRailTicketCoder;

+import org.uic.ticket.api.spec.IUicRailTicket;

+

+import net.gcdc.asn1.datatypesimpl.OctetString;

+

+

+/**

+ * The Class Encoder.

+ * 

+ * signing and encoding of UIC bar codes 

+ * 

+ * 

+ */

+public class Encoder {

+	

+	/** The dynamic frame. */

+	private DynamicFrame dynamicFrame = null;

+

+	/** The static frame. */

+	private StaticFrame staticFrame = null;

+	

+

+	/** The UIC bar code type classic. */

+	public static String UIC_BARCODE_TYPE_CLASSIC = "UIC_CLASSIC";

+	

+	/** The UIC bar code type DOSIPAS. */

+	public static String UIC_BARCODE_TYPE_DOSIPAS = "UIC_DOSIPAS";	

+	

+	/**

+	 * Instantiates a new encoder.

+	 *

+	 * @param ticket the ticket

+	 * @param layout the layout

+	 * @param barcodeType the bar code type

+	 * @param version the version of the bar code 

+	 * @param fcbVersion the fcb version

+	 * @throws IOException Signals that an I/O exception has occurred.

+	 * @throws EncodingFormatException the encoding format exception

+	 */

+	public Encoder(IUicRailTicket ticket, TicketLayout layout, String barcodeType, int version, int fcbVersion) throws IOException, EncodingFormatException {

+		

+		if (barcodeType == UIC_BARCODE_TYPE_CLASSIC) {

+

+			staticFrame = new StaticFrame();

+			staticFrame.setVersion(version);

+			

+			if (layout != null) {

+						

+				UHEADDataRecord head = new UHEADDataRecord();

+				head.setVersionId("01");

+				staticFrame.setHeaderRecord(head);				

+				

+				UTLAYDataRecord tlay = new UTLAYDataRecord();

+				tlay.setLayout(layout);

+				tlay.setVersionId("01");

+				staticFrame.setuTlay(tlay);

+			}		

+			

+			if (ticket != null) {

+				

+				UFLEXDataRecord flex = new UFLEXDataRecord();

+				flex.setTicket(ticket);

+				flex.setVersionId(String.format("%02d",fcbVersion));

+				staticFrame.setuFlex(flex);

+			}

+			

+			

+		} else if (barcodeType == UIC_BARCODE_TYPE_DOSIPAS) {

+			

+			dynamicFrame = new DynamicFrame();

+			dynamicFrame.setLevel2SignedData(new Level2DataType());

+			dynamicFrame.getLevel2SignedData().setLevel1Data(new Level1DataType());

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setData(new SequenceOfDataType());

+			

+			if (ticket != null) {

+				

+				if (version == 1) {

+					dynamicFrame.setFormat("U1");

+				}

+				

+				DataType ticketData = new DataType();

+				

+				UicRailTicketCoder uicTicketCoder = new UicRailTicketCoder();

+				ticketData.setFormat(Constants.DATA_TYPE_FCB_VERSION_1);

+				ticketData.setData(new OctetString(uicTicketCoder.encode(ticket, fcbVersion)));

+				dynamicFrame.getLevel2SignedData().getLevel1Data().getData().add(ticketData);

+				

+			}

+		}

+	}

+	

+	

+	

+	/**

+	 * Signing level 2 of a dynamic bar code

+	 *     

+	 * @param key the key

+	 * @throws Exception the exception

+	 */

+	public void signLevel2(PrivateKey key) throws Exception {

+		if (dynamicFrame != null) {

+			dynamicFrame.signLevel2(key);

+		} 

+	}

+

+	/**

+	 * Sets the level 1 algorithm Is.

+	 *

+	 * @param level1SigningAlg the level 1 signing algorithm (OID)

+	 * @param level1KeyAlg the level 1 key algorithm (OID)

+	 */

+	public void setLevel1Algs(String level1SigningAlg, String level1KeyAlg) {

+		if (dynamicFrame != null) {

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setLevel1SigningAlg(level1SigningAlg);

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setLevel1KeyAlg(level1KeyAlg);	

+		}

+	}

+	

+	/**

+	 * Sets the level 2 algorithm Is.

+	 *

+	 * @param level2SigningAlg the level 2 signing algorithm (OID)

+	 * @param level2KeyAlg the level 2 key algorithm (OID)

+	 * @param publicKey the public key of the level 2 signature

+	 */

+	public void setLevel2Algs(String level2SigningAlg, String level2KeyAlg, PublicKey publicKey) {

+		if (dynamicFrame != null) {

+			if (dynamicFrame.getLevel2SignedData() == null) {

+				dynamicFrame.setLevel2SignedData(new Level2DataType());

+			}

+			if (dynamicFrame.getLevel2SignedData().getLevel1Data() == null) {

+				dynamicFrame.getLevel2SignedData().setLevel1Data(new Level1DataType());	

+			}

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setLevel2SigningAlg(level2SigningAlg);

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setLevel2KeyAlg(level2KeyAlg);

+			if (publicKey != null) {

+				dynamicFrame.getLevel2SignedData().getLevel1Data().setLevel2publicKey(new OctetString(publicKey.getEncoded()));

+			}

+		}

+	}

+	

+	public void setLevel2Data(DataType level2data) {

+		if (dynamicFrame != null) {

+			if (dynamicFrame.getLevel2SignedData() == null) {

+				dynamicFrame.setLevel2SignedData(new Level2DataType());

+			}		

+			dynamicFrame.getLevel2SignedData().setLevel2Data(level2data);

+		}

+	}

+	

+	public DataType getLevel2Data() {

+		if (dynamicFrame != null && dynamicFrame.getLevel2SignedData() != null) {

+			return dynamicFrame.getLevel2SignedData().getLevel2Data();

+		}

+		return null;

+	}

+	

+	/**

+	 * Sign level 1 of a dynamic bar code or a static bar code.

+	 *

+	 * @param securityProvider the security provider

+	 * @param key the key

+	 * @param signingAlg the signing algorithm (OID)

+	 * @param keyId the key id

+	 * @throws Exception the exception

+	 */

+	public void signLevel1(String securityProvider,PrivateKey key,String signingAlg, String keyId) throws Exception {

+		if (dynamicFrame != null) {

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setSecurityProvider(securityProvider);

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setLevel1SigningAlg(signingAlg);

+			dynamicFrame.getLevel2SignedData().getLevel1Data().setKeyId(Long.parseLong(keyId));

+			dynamicFrame.getLevel2SignedData().signLevel1(key);

+		} else if (staticFrame != null) {

+			staticFrame.setSignatureKey(keyId);

+			staticFrame.setSecurityProvider(securityProvider);

+			if (staticFrame.getHeaderRecord()!= null && staticFrame.getHeaderRecord().getIssuer() == null) {

+				staticFrame.getHeaderRecord().setIssuer(securityProvider);

+			}

+			staticFrame.signByAlgorithmOID(key,signingAlg);

+		}

+	}

+	

+	/**

+	 * Sets the static header parameter.

+	 *

+	 * @param ticketId the ticket id

+	 * @param language the language

+	 */

+	public void setStaticHeaderParams(String ticketId,String language) {

+	  if (staticFrame != null && staticFrame.getHeaderRecord()!= null) {

+		staticFrame.getHeaderRecord().setIdentifier(ticketId);

+		staticFrame.getHeaderRecord().setLanguage(language);

+	  }

+	}

+

+	/**

+	 * Gets the dynamic frame.

+	 *

+	 * @return the dynamic frame

+	 */

+	public DynamicFrame getDynamicFrame() {

+		return dynamicFrame;

+	}

+	

+	

+	

+	/**

+	 * Gets the static frame.

+	 *

+	 * @return the static frame

+	 */

+	public StaticFrame getStaticFrame() {

+		return staticFrame;

+	}

+

+

+

+	/**

+	 * Encodes the signed bar code data 

+	 *

+	 * @return the byte[]

+	 * @throws IOException Signals that an I/O exception has occurred.

+	 * @throws Exception the exception

+	 */

+	public byte[] encode() throws IOException, Exception {

+		if (dynamicFrame != null) {

+			return dynamicFrame.encode();

+		} else if (staticFrame != null) {

+			return staticFrame.encode();

+		}

+		return null;

+	}

+

+

+

+

+	

+	

+	

+

+}