socket收发简单示例

这是一个简单的包含发送端和接收端的例子。发送端向接收端发送文件名和文件内容，接收端将收到的文件保存在磁盘上。接收端可以同时接收多个发送端传来的文件，但没有处理文件同名的情况。 


这个例子中设计了一个简单的协议。发送的内容是这样的：


文件名长度（4字节）—文件名—文件内容长度（4字节）—文件内容。 

接收端也按照这个结构进行解析。建议先看 Client 类，再看 Server 类。

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * 简单的文件发送与接收示例
 */
public class FileTrasmission {

	// 程序入口
	public static void main(String[] args) throws Exception {
		int port = 7788;
		new Server(port, "c:\\save\\").start();
		new Client().sendFile("127.0.0.1", port, "c:\\迷失在康熙末年.txt");
	}
}

/**
 * 接收端。可同时接收多个发送端发来的文件。但如果发来的文件是同名的话那就乱了。
 */
class Server {

	private int listenPort;

	private String savePath;

	/**
	 * 构造方法
	 * 
	 * @param listenPort
	 *            侦听端口
	 * @param savePath
	 *            接收的文件要保存的路径
	 * 
	 * @throws IOException
	 *             如果创建保存路径失败
	 */
	Server(int listenPort, String savePath) throws IOException {
		this.listenPort = listenPort;
		this.savePath = savePath;

		File file = new File(savePath);
		if (!file.exists() && !file.mkdirs()) {
			throw new IOException("无法创建文件夹 " + savePath);
		}
	}

	// 开始侦听
	public void start() {
		new ListenThread().start();
	}

	// 网上抄来的，将字节转成 int。b 长度不得小于 4，且只会取前 4 位。
	public static int b2i(byte[] b) {
		int value = 0;
		for (int i = 0; i < 4; i++) {
			int shift = (4 - 1 - i) * 8;
			value += (b[i] & 0x000000FF) << shift;
		}
		return value;
	}

	/**
	 * 侦听线程
	 */
	private class ListenThread extends Thread {

		@Override
		public void run() {
			try {
				ServerSocket server = new ServerSocket(listenPort);

				// 开始循环
				while (true) {
					Socket socket = server.accept();
					new HandleThread(socket).start();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

	/**
	 * 读取流并保存文件的线程
	 */
	private class HandleThread extends Thread {

		private Socket socket;

		private HandleThread(Socket socket) {
			this.socket = socket;
		}

		@Override
		public void run() {
			try {
				InputStream is = socket.getInputStream();
				readAndSave(is);
			} catch (IOException e) {
				e.printStackTrace();
			} finally {
				try {
					socket.close();
				} catch (IOException e) {
					// nothing to do
				}
			}
		}

		// 从流中读取内容并保存
		private void readAndSave(InputStream is) throws IOException {
			String filename = getFileName(is);
			int file_len = readInteger(is);
			System.out.println("接收文件：" + filename + "，长度：" + file_len);

			readAndSave0(is, savePath + filename, file_len);

			System.out.println("文件保存成功（" + file_len + "字节）。");
		}

		private void readAndSave0(InputStream is, String path, int file_len)
				throws IOException {
			FileOutputStream os = getFileOS(path);
			readAndWrite(is, os, file_len);
			os.close();
		}

		// 边读边写，直到读取 size 个字节
		private void readAndWrite(InputStream is, FileOutputStream os, int size)
				throws IOException {
			byte[] buffer = new byte[4096];
			int count = 0;
			while (count < size) {
				int n = is.read(buffer);
				// 这里没有考虑 n = -1 的情况
				os.write(buffer, 0, n);
				count += n;
			}
		}

		// 读取文件名
		private String getFileName(InputStream is) throws IOException {
			int name_len = readInteger(is);
			byte[] result = new byte[name_len];
			is.read(result);
			return new String(result);
		}

		// 读取一个数字
		private int readInteger(InputStream is) throws IOException {
			byte[] bytes = new byte[4];
			is.read(bytes);
			return b2i(bytes);
		}

		// 创建文件并返回输出流
		private FileOutputStream getFileOS(String path) throws IOException {
			File file = new File(path);
			if (!file.exists()) {
				file.createNewFile();
			}

			return new FileOutputStream(file);
		}
	}

}

/**
 * 发送端
 */
class Client {

	// 网上抄来的，将 int 转成字节
	public static byte[] i2b(int i) {
		return new byte[] { (byte) ((i >> 24) & 0xFF),
				(byte) ((i >> 16) & 0xFF), (byte) ((i >> 8) & 0xFF),
				(byte) (i & 0xFF) };
	}

	/**
	 * 发送文件。文件大小不能大于 {@link Integer#MAX_VALUE}
	 * 
	 * @param hostname
	 *            接收端主机名或 IP 地址
	 * @param port
	 *            接收端端口号
	 * @param filepath
	 *            文件路径
	 * 
	 * @throws IOException
	 *             如果读取文件或发送失败
	 */
	public void sendFile(String hostname, int port, String filepath)
			throws IOException {
		File file = new File(filepath);
		FileInputStream is = new FileInputStream(filepath);

		Socket socket = new Socket(hostname, port);
		OutputStream os = socket.getOutputStream();

		try {
			int length = (int) file.length();
			System.out.println("发送文件：" + file.getName() + "，长度：" + length);

			// 发送文件名和文件内容
			writeFileName(file, os);
			writeFileContent(is, os, length);
		} finally {
			os.close();
			is.close();
		}
	}

	// 输出文件内容
	private void writeFileContent(InputStream is, OutputStream os, int length)
			throws IOException {
		// 输出文件长度
		os.write(i2b(length));

		// 输出文件内容
		byte[] buffer = new byte[4096];
		int size;
		while ((size = is.read(buffer)) != -1) {
			os.write(buffer, 0, size);
		}
	}

	// 输出文件名
	private void writeFileName(File file, OutputStream os) throws IOException {
		byte[] fn_bytes = file.getName().getBytes();

		os.write(i2b(fn_bytes.length)); // 输出文件名长度
		os.write(fn_bytes); // 输出文件名
	}
}