NetworkProxyServerConnection.java

/**
Copyright 2013 Luciano Zu project Ardulink http://www.ardulink.org/

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

@author Luciano Zu
*/

package org.zu.ardulink.connection.proxy;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.List;
import java.util.concurrent.TimeUnit;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.zu.ardulink.Link;
import org.zu.ardulink.RawDataListener;
import org.zu.ardulink.protocol.IProtocol;

/**
 * [ardulinktitle] [ardulinkversion]
 * @author Luciano Zu project Ardulink http://www.ardulink.org/
 * 
 * [adsense]
 */
public class NetworkProxyServerConnection implements Runnable, NetworkProxyMessages, RawDataListener {

	private static final Logger logger = LoggerFactory.getLogger(NetworkProxyServerConnection.class);

	private Link link;
	private Socket socket;
	
	private BufferedReader bufferedReader;
	private InputStream inputStream;
	private PrintWriter printWriter;
	private OutputStream outputStream;

	private boolean closed;
	private boolean handshakeComplete;
		
	public NetworkProxyServerConnection(Socket socket) {
		super();
		this.socket = socket;
		this.link = null;
	}

	@Override
	public void run() {
		try {
			outputStream = socket.getOutputStream();
			printWriter = new PrintWriter(outputStream, true);
			
			inputStream = socket.getInputStream();
			bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
			
			String inputLine = bufferedReader.readLine();
			while (!closed && !handshakeComplete) {
				processInput(inputLine);
				if(!closed && !handshakeComplete) {
					inputLine = bufferedReader.readLine();
				}
            }
			if(!closed) {
				int dataReceived = bufferedReader.read();
				while (dataReceived != -1) {
					String data = String.valueOf((char)dataReceived);
					// System.out.print(data);
					writeSerial(data);
					dataReceived = bufferedReader.read();
	            }			
			}
		} catch (IOException e) {
		}
		finally {
			logger.info("{} connection closed.", socket.getRemoteSocketAddress());
			closed = true;
			if(link != null) {
				link.removeRawDataListener(this);
				NetworkProxyServer.disconnect(link.getName());
			}
			try {
				socket.close();
			} catch (IOException socketClosingExceptionTry) {
			}
			link = null;
			socket = null;
			bufferedReader = null;
			inputStream = null;
			printWriter = null;
			outputStream = null;
			closed = true;
		}
	}

	private void processInput(String inputLine) throws IOException {
		if(inputLine.equals(STOP_SERVER_CMD)) {
			logger.info("Stop request received.");
			NetworkProxyServer.stop();
			closed = true;
		} else if(inputLine.equals(GET_PORT_LIST_CMD)) {
			List<String> portList = getPortList();
			if(portList == null || portList.isEmpty()) {
				printWriter.println(NUMBER_OF_PORTS + "0");
			} else {
				printWriter.println(NUMBER_OF_PORTS + portList.size());
				for (String port : portList) {
					printWriter.println(port);
				}
			}
			printWriter.flush();
		} else if(inputLine.equals(CONNECT_CMD)) {
			String portName = bufferedReader.readLine();
			Integer baudRate = new Integer(bufferedReader.readLine());
			boolean connected = connect(portName, baudRate);
			try { // wait a wile to avoid messages with connection not valid.
				TimeUnit.SECONDS.sleep(1);
			} catch (InterruptedException e) {
				throw new RuntimeException(e);
			}
			if(connected) {
				printWriter.println(OK);
				printWriter.flush();
				handshakeComplete = true;
				link.addRawDataListener(this);
			} else {
				printWriter.println(KO);
				printWriter.flush();
			}
			
		}
	}

	private List<String> getPortList() {
		// return Link.getDefaultInstance().getPortList();
		return NetworkProxyServer.getPortList();
	}

	private boolean writeSerial(String message) {
		boolean retvalue = false;
		synchronized (link) {
			retvalue = link.writeSerial(message);
		}
		return retvalue;
	}

	private boolean connect(String portName, int baudRate) {
		
		link = NetworkProxyServer.connect(portName, baudRate);
		boolean isConnected = link.isConnected();
		return isConnected;
	}

	@Override
	public void parseInput(String id, int numBytes, int[] message) {
		for(int i = 0; i < numBytes; i++) {
			try {
				outputStream.write(message[i]);
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		try {
			outputStream.write(IProtocol.DEFAULT_INCOMING_MESSAGE_DIVIDER);
			outputStream.flush();
		} catch (IOException e) {
			e.printStackTrace();
		}

	}
}