Java Socket编程 - 基于TCP方式的二进制文件传输

package com.gloomyfish.socket.tutorial.filetransfer;
 
import java.io.BufferedOutputStream;
import java.io.BufferedWriter;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.net.Socket;
 
import javax.swing.SwingWorker;
 
public class FileReceiveTask extends SwingWorker<Integer, Object> {
  private Socket _mSocket;
  public FileReceiveTask(Socket client) {
    this._mSocket = client;
  }
 
  @Override
  protected Integer doInBackground() throws Exception {
    // get file meta information
    DataInputStream input = new DataInputStream(_mSocket.getInputStream());
    String fileName = input.readUTF();
    int fileLength = (int)input.readLong(); // number of total bytes
    File file = new File("C:\\Users\\fish\\Downloads" + File.separator + fileName);
    BufferedOutputStream output = new BufferedOutputStream(new FileOutputStream(file));
    System.out.println("Received File Name = " + fileName);
    System.out.println("Received File size = " + fileLength/1024 + "KB");
    
    // start to receive the content of the file and write them
    byte[] content = new byte[2048];
    int offset = 0;
    int numReadBytes = 0;
    while(offset < fileLength && (numReadBytes = input.read(content)) > 0) {
      output.write(content, 0, numReadBytes);
      float precent = 100.0f * ((float)offset)/((float)fileLength);
      setProgress((int)precent);
      offset += numReadBytes;
    }
    System.out.println("numReadBytes = " + numReadBytes);
    if(offset < fileLength) {
      numReadBytes = input.read(content);
      System.out.println("numReadBytes = " + numReadBytes);
      System.out.println("File content error at server side");
    } else {
      System.out.println("File Receive Task has done correctly");
    }
    setProgress(100);
    
    // tell client to close the socket now, we already receive the file successfully!!
      BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(_mSocket.getOutputStream()));
      bufferedWriter.write("DONE\r\n");
      bufferedWriter.flush();
      
      // close the file and socket
    output.close();
    _mSocket.close();
    return 100;
  }
 
}

package com.gloomyfish.socket.tutorial.filetransfer;
 
import java.awt.BorderLayout;
import java.awt.FlowLayout;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import java.io.File;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
 
import javax.swing.BorderFactory;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JProgressBar;
import javax.swing.JTextField;
/**
 * 我一般写英文注释，偶尔我也会写中文注释，只是觉得写英文
 * 注释跟代码比较统一，无他。
 * 
 * @Date 2012-11-30
 * @author fish
 *
 */
public class FileTransferClient extends JFrame implements ActionListener {
  /**
   * 
   */
  private static final long serialVersionUID = 1L;
  public final static String SEND_CMD = "Send File";
  public final static int MINIMUM = 0;
  public final static int MAXIMUM = 100;
  // public final static String CONNECT_CMD = "Connect";
  private JButton sendFileBtn;
  private JTextField serverField;
  private JTextField portField;
  private JProgressBar bar;
  
  public FileTransferClient() {
    super("File Transfer Client");
    initComponents();
  }
 
  private void initComponents() {
    getContentPane().setLayout(new BorderLayout());
    JPanel progressPanel = new JPanel();
    progressPanel.setLayout(new BoxLayout(progressPanel, BoxLayout.Y_AXIS));
    bar = new JProgressBar();
    progressPanel.add(bar);
    bar.setMinimum(MINIMUM);
    bar.setMaximum(MAXIMUM);
    JPanel serverSettingPanel = new JPanel();
    serverSettingPanel.setLayout(new GridLayout(2,2,5,5));
    serverSettingPanel.setBorder(BorderFactory.createTitledBorder("Server Setting"));
    serverField = new JTextField();
    portField = new JTextField();
    serverSettingPanel.add(new JLabel("Server IP/Host:"));
    serverSettingPanel.add(serverField);
    serverSettingPanel.add(new JLabel("Server Port:"));
    serverSettingPanel.add(portField);
    
    sendFileBtn = new JButton(SEND_CMD);
    JPanel btnPanel = new JPanel();
    btnPanel.setLayout(new FlowLayout(FlowLayout.RIGHT));
    btnPanel.add(sendFileBtn);
    getContentPane().add(serverSettingPanel, BorderLayout.NORTH);
    getContentPane().add(btnPanel, BorderLayout.SOUTH);
    getContentPane().add(progressPanel, BorderLayout.CENTER);
    sendFileBtn.addActionListener(this);
  }
 
  @Override
  public void actionPerformed(ActionEvent e) {
    String command = e.getActionCommand();
    if(command.equals(SEND_CMD)) {
      if(checkNull()) {
        JOptionPane.showMessageDialog(this, "Please enter server host and port in order to set up the connection!");
        return;
      }
      JFileChooser chooser = new JFileChooser();
      int status = chooser.showOpenDialog(null);
      if (status == JFileChooser.APPROVE_OPTION) {
        File f = chooser.getSelectedFile();
        SocketAddress address = new InetSocketAddress(getServer(), getPort());
        FileTransferTask task = new FileTransferTask(f, address, this);
        bar.setValue(0);
        task.addPropertyChangeListener(new PropertyChangeListener() {
                public void propertyChange(PropertyChangeEvent evt) {
                    if ("progress".equals(evt.getPropertyName())) {
                      bar.setValue((Integer) evt.getNewValue());
                    }
                }
            });
        task.execute(); // 异步task执行
      }
    } else {
      // do nothing
    }
  }
  
  public void showSuccess() {
    bar.setValue(100);
    JOptionPane.showMessageDialog(this, "file send successfully!");
  }
  
  public String getServer() {
    return serverField.getText().trim();
  }
  
  public int getPort() {
    return Integer.parseInt(portField.getText().trim());
  }
  /**
   * make sure the UI already have some correct input information here!!!
   * @return
   */
  private boolean checkNull() {
    String serverName = serverField.getText();
    String port = portField.getText();
    if(serverName == null || serverName.length() == 0 || port == null || port.length() == 0) {
      return true;
    }
    
    try {
      Integer.parseInt(port); // try to parse it as server port number , validation code.
    } catch(NumberFormatException ne) {
      ne.printStackTrace();
      return true;
    }
    return false;
  }
  
  public static void main(String[] args) {
    FileTransferClient client = new FileTransferClient();
    client.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    client.setSize(400, 400);
    client.setResizable(false);
    // client.pack();
    client.setVisible(true);
  }
 
}

package com.gloomyfish.socket.tutorial.filetransfer;
/**
 * this is very simple file transfer protocol over TCP socket
 * @date 2012-12-01
 * @author zhigang jia
 *
 */
public class FileDataPackage {
 
  private int dataLength; // 数据包中数据长度，两个字节
  private byte[] databuff; // 数据包中数据，meici最大不超过2048字节
  
  public final static byte[] EOF = new byte[]{'E', 'O','F'};
  
  public FileDataPackage() {
    dataLength = 0;
    databuff = new byte[2048];
  }
  
  public byte[] getPackageData() {
    byte[] pData = new byte[dataLength];
    // end comment
    System.arraycopy(databuff, 0, pData, 0, dataLength);
    return pData;
  }
  
  public void setData(byte[] data, int bsize) {
    dataLength = bsize;
    for(int i=0; i<databuff.length; i++) {
      if(i<bsize) {
        databuff[i] = data[i];
      } else {
        databuff[i] = ' ';
      }
    }
  }
}