/*

 * JBoss, Home of Professional Open Source

 * Copyright 2011, Red Hat, Inc. and individual contributors

 * by the @authors tag. See the copyright.txt in the distribution for a

 * full listing of individual contributors.

 *

 * This is free software; you can redistribute it and/or modify it

 * under the terms of the GNU Lesser General Public License as

 * published by the Free Software Foundation; either version 2.1 of

 * the License, or (at your option) any later version.

 *

 * This software is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this software; if not, write to the Free

 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA

 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.

 */



package org.mobicents.protocols.ss7.map.functional;



import java.io.FileOutputStream;

import java.io.IOException;



import org.mobicents.protocols.ss7.mtp.Mtp3TransferPrimitive;

import org.mobicents.protocols.ss7.mtp.Mtp3UserPartBaseImpl;

import org.mobicents.protocols.ss7.sccp.SccpProvider;

import org.mobicents.protocols.ss7.sccp.impl.RemoteSignalingPointCode;

import org.mobicents.protocols.ss7.sccp.impl.RemoteSubSystem;

import org.mobicents.protocols.ss7.sccp.impl.SccpResource;

import org.mobicents.protocols.ss7.sccp.impl.SccpStackImpl;

import org.mobicents.protocols.ss7.sccp.impl.router.Router;



/**

 * @author amit bhayani

 * 

 */

public abstract class SccpHarness {



	protected SccpStackImpl sccpStack1 = new SccpStackImpl();

	protected SccpProvider sccpProvider1 = sccpStack1.getSccpProvider();



	protected SccpStackImpl sccpStack2 = new SccpStackImpl();

	protected SccpProvider sccpProvider2 = sccpStack2.getSccpProvider();



//	private ConcurrentLinkedQueue<byte[]> data1 = new ConcurrentLinkedQueue<byte[]>();

//	private ConcurrentLinkedQueue<byte[]> data2 = new ConcurrentLinkedQueue<byte[]>();



	private Mtp3UserPartImpl mtp3UserPart1 = new Mtp3UserPartImpl();

	private Mtp3UserPartImpl mtp3UserPart2 = new Mtp3UserPartImpl();



	protected Router router1 = new Router();

	protected Router router2 = new Router();



	protected SccpResource resource1 = new SccpResource();

	protected SccpResource resource2 = new SccpResource();



	

	/**

	 * 

	 */

	public SccpHarness() {

		mtp3UserPart1.setOtherPart(mtp3UserPart2);

		mtp3UserPart2.setOtherPart(mtp3UserPart1);

	}



	private void setUpStack1() {

		try {

			router1.start();

			router1.getRules().clear();

			router1.getPrimaryAddresses().clear();

			router1.getBackupAddresses().clear();

		} catch (Exception e) {

			// ignore

		}



		resource1.start();

		resource1.getRemoteSpcs().clear();

		resource1.getRemoteSsns().clear();



		resource1.getRemoteSpcs().put(1, new RemoteSignalingPointCode(2, 0, 0));

		resource1.getRemoteSsns().put(1, new RemoteSubSystem(2, 8, 0));



		sccpStack1.setRouter(router1);

		sccpStack1.setSccpResource(resource1);

		sccpStack1.setLocalSpc(1);

		sccpStack1.setNi(2);

		sccpStack1.setMtp3UserPart(mtp3UserPart1);

		sccpStack1.start();

	}



	private void setUpStack2() {

		try {

			router2.start();

			router2.getRules().clear();

			router2.getPrimaryAddresses().clear();

			router2.getBackupAddresses().clear();

		} catch (Exception e) {

			// ignore

		}



		resource2.start();

		resource2.getRemoteSpcs().clear();

		resource2.getRemoteSsns().clear();



		resource2.getRemoteSpcs().put(1, new RemoteSignalingPointCode(1, 0, 0));

		resource2.getRemoteSsns().put(1, new RemoteSubSystem(1, 8, 0));



		sccpStack2.setRouter(router2);

		sccpStack2.setSccpResource(resource2);

		sccpStack2.setLocalSpc(2);

		sccpStack2.setNi(2);

		sccpStack2.setMtp3UserPart(mtp3UserPart2);

		sccpStack2.start();

	}



	private void tearDownStack1() {

		router1.getRules().clear();

		router1.getPrimaryAddresses().clear();

		router1.getBackupAddresses().clear();

		router1.stop();



		resource1.getRemoteSpcs().clear();

		resource1.getRemoteSsns().clear();

		resource1.stop();



		sccpStack1.stop();



	}



	private void tearDownStack2() {

		router1.getRules().clear();

		router1.getPrimaryAddresses().clear();

		router1.getBackupAddresses().clear();

		router1.stop();



		resource1.getRemoteSpcs().clear();

		resource1.getRemoteSsns().clear();

		resource1.stop();



		sccpStack1.stop();



	}



	public void setUp() {

		this.setUpStack1();

		this.setUpStack2();

	}



	public void tearDown() {

		this.tearDownStack1();

		this.tearDownStack2();

	}

	

	/**

	 * After this method invoking all MTP traffic will be save into the file "MsgLog.txt"

	 * file format:

	 * [message][message]...[message]

	 * [message] ::= { byte-length low byte, byte-length high byte, byte[] message }

	 */

	public void saveTrafficInFile() {

		((Mtp3UserPartImpl) this.mtp3UserPart1).saveTrafficInFile = true;

		((Mtp3UserPartImpl) this.mtp3UserPart2).saveTrafficInFile = true;

		

		try {

			FileOutputStream fs = new FileOutputStream("MsgLog.txt", false);

			fs.close();

		} catch (Exception e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

	}



	private class Mtp3UserPartImpl extends Mtp3UserPartBaseImpl {



//		private ConcurrentLinkedQueue<byte[]> readFrom;

//		private ConcurrentLinkedQueue<byte[]> writeTo;

		

		private Mtp3UserPartImpl otherPart;

		

		private boolean saveTrafficInFile = false;



		Mtp3UserPartImpl() {

			try {

				this.start();

			} catch (Exception e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

		}

		

		public void setOtherPart(Mtp3UserPartImpl otherPart) {

			this.otherPart = otherPart;

		}



		@Override

		public void sendMessage(Mtp3TransferPrimitive msg) throws IOException {

			if (saveTrafficInFile) {

				try {

					byte[] txData = msg.encodeMtp3();

					FileOutputStream fs = new FileOutputStream("MsgLog.txt", true);

					int ln = txData.length;

					fs.write(ln & 0xFF);

					fs.write(ln >> 8);

					fs.write(txData);

					fs.close();

				} catch (Exception e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				}

			}



			this.otherPart.sendTransferMessageToLocalUser(msg, msg.getSls());

		}



//		@Override

//		public int read(ByteBuffer b) throws IOException {

//			int dataRead = 0;

//			while (!this.readFrom.isEmpty()) {

//				byte[] rxData = this.readFrom.poll();

//				System.out.println("Read " + Arrays.toString(rxData));

//				dataRead = dataRead + rxData.length;

//				b.put(rxData);

//			}

//			return dataRead;

//		}

//

//		@Override

//		public int write(ByteBuffer b) throws IOException {

//			int dataAdded = 0;

//			if (b.hasRemaining()) {

//				byte[] txData = new byte[b.limit() - b.position()];

//				b.get(txData);

//				dataAdded = dataAdded + txData.length;

//				System.out.println("Write " + Arrays.toString(txData));

//				this.writeTo.add(txData);

//				

//				if (saveTrafficInFile) {

//					try {

//						FileOutputStream fs = new FileOutputStream("MsgLog.txt", true);

//						int ln = txData.length;

//						fs.write(ln & 0xFF);

//						fs.write(ln >> 8);

//						fs.write(txData);

//						fs.close();

//					} catch (Exception e) {

//						// TODO Auto-generated catch block

//						e.printStackTrace();

//					}

//				}

//			}

//			return dataAdded;

//		}



//		@Override

//		public void execute() throws IOException {

//			//We dont use this

//			

//			try {

//				//FIXME: this has to be changed to some MTP change!!!

//				Thread.sleep(100);

//			} catch( Exception e ) {

//			}

//		}





	}



}