defaultdatahandler.java

来自「java实现的代理程序 socket 编程实现的http 和socket 代理程」· Java 代码 · 共 91 行

package javatunnel;

import java.net.*;
import java.io.*;

/**
 * 
 * JavaTunnel
 * 
 * Copyright (C) 2002 Andr閟 Ederra
 * 
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2 of the License, or (at your option)
 * any later version.
 * 
 * This program 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 General Public License for
 * more details.
 * 
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
 * Place - Suite 330, Boston, MA 02111-1307, USA.
 */
class DefaultDataHandler extends Thread implements DataHandler {

	private Socket in;
	private Socket out;
	InputStream inStream = null;
	OutputStream outStream = null;
	private boolean terminate=false;
	
	public DefaultDataHandler(Socket in, Socket out) {

		this.in = in;
		this.out = out;
	}

	public void run() {

		try {
			inStream = in.getInputStream();
			outStream = out.getOutputStream();

			byte[] buffer = new byte[Configuration.BUFFERSIZE];
			int n = 0;
			while (n != -1) {
				n = inStream.read(buffer);
				if (terminate) {
					shutdownHandler();
					return;
				}
				if (n != -1) {
					outStream.write(buffer, 0, n);
					outStream.flush();
				}
			}
		} catch (SocketException se) {
			//do nothing
			//the tcp connection could be closed
		} catch (IOException ioe) {
			ioe.printStackTrace();
			System.exit(1);
		} finally {
			shutdownHandler();
		}

	}
	public void stopHandler() {
		terminate=true;
	}
	private void shutdownHandler() {
		//Try to close sockets cleanly
		try {
			if (inStream != null)
				inStream.close();

		} catch (IOException ioe) {
			ioe.printStackTrace();
		}
		try {
			if (inStream != null) {
				outStream.flush();
				outStream.close();
			}
		} catch (IOException ioe) {
			ioe.printStackTrace();
		}
	}
}