六狼论坛

 找回密码
 立即注册

QQ登录

只需一步,快速开始

新浪微博账号登陆

只需一步,快速开始

搜索
查看: 50|回复: 0

如何把Socket作为平台 发送和接收文件

[复制链接]

升级  77%

131

主题

131

主题

131

主题

举人

Rank: 3Rank: 3

积分
431
 楼主| 发表于 2013-1-15 03:00:28 | 显示全部楼层 |阅读模式
这是一个简单的包含发送端和接收端的例子。发送端向接收端发送文件名和文件内容,接收端将收到的文件保存在磁盘上。接收端可以同时接收多个发送端传来的文件,但没有处理文件同名的情况。
这个例子中设计了一个简单的协议。发送的内容是这样的:
文件名长度(4字节)—文件名—文件内容长度(4字节)—文件内容。
接收端也按照这个结构进行解析。建议先看 Client 类,再看 Server 类。
view plaincopy to clipboardprint?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);    // 输出文件名       }   }   
<div class="links">»
您需要登录后才可以回帖 登录 | 立即注册 新浪微博账号登陆

本版积分规则

快速回复 返回顶部 返回列表