Nội dung chính
Kiến thức cần nhớ
- TCP/IP có cơ chế gửi tin tin cậy, có nghĩa là các gói tin được gửi đi sẽ được check xem đã đến đích hay chưa.
- Trong java, đối tượng java.net.ServerSocket được sử dụng để khởi tạo một TCP server, mỗi khi có client kết nối đến thì đối tượng java.net.Socket được tạo ra để tương tác với client. Đối tượng java.net.Socket cũng được sử dụng từ phía client để kết nối và tương tác với server.
- Phương thức Socket.getOutputStream() trả về OutputStream được sử dụng để gửi dữ liệu.
- Phương thức Socket.getInputStream() trả về InputStream được sử dụng để nhận dữ liệu.
- Việc cài đặt số lần gửi phải tương ứng với số lần nhận.
TCP/IP transfer file example
Đây là một ví dụ về việc sử dụng giao thức TCP/IP để truyền file.
Tạo lớp FileInfo
Tạo lớp vn.kienthuclaptrinh.common.FileInfo giống nhau cho cả phía client và server để gửi và nhận thông tin của một file thông qua đối tượng này.
package vn.kienthuclaptrinh.common;
import java.io.Serializable;
public class FileInfo implements Serializable {
private static final long serialVersionUID = 1L;
private String destinationDirectory;
private String sourceDirectory;
private String filename;
private long fileSize;
private int piecesOfFile;
private int lastByteLength;
private byte[] dataBytes;
private String status;
// Constructor
// getter & setter
}
Cài đặt phía server
Cấu trúc TCPServer project:
File: TCPServer.java
package vn.kienthuclaptrinh.server;
import java.io.BufferedOutputStream;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import vn.kienthuclaptrinh.common.FileInfo;
public class TCPServer extends Thread {
// create serverSocket object
private ServerSocket serverSocket;
private int port = 9900;
/**
* run program
*
* @author kienthuclaptrinh.net
* @param args
*/
public static void main(String[] args) {
TCPServer tcpServer = new TCPServer();
tcpServer.open();
tcpServer.start();
}
/**
* open server
*
* @author kienthuclaptrinh.net
*/
public void open() {
try {
serverSocket = new ServerSocket(port);
System.out.println("server is open on port " + port);
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* handle event
*
* @author vietuts.vn
*/
public void run() {
while (true) {
Socket server = null;
DataInputStream inFromClient = null;
ObjectInputStream ois = null;
ObjectOutputStream oos = null;
try {
// accept connect from client and create Socket object
server = serverSocket.accept();
System.out.println("connected to "
+ server.getRemoteSocketAddress());
// get greeting from client
inFromClient = new DataInputStream(server.getInputStream());
System.out.println(inFromClient.readUTF());
// receive file info
ois = new ObjectInputStream(server.getInputStream());
FileInfo fileInfo = (FileInfo) ois.readObject();
if (fileInfo != null) {
createFile(fileInfo);
}
// confirm that file is received
oos = new ObjectOutputStream(server.getOutputStream());
fileInfo.setStatus("success");
fileInfo.setDataBytes(null);
oos.writeObject(fileInfo);
} catch (IOException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} finally {
// close all stream
closeStream(ois);
closeStream(oos);
closeStream(inFromClient);
// close session
closeSocket(server);
}
}
}
/**
* create file with fileInfo
*
* @author kienthuclaptrinh.net
* @param fileInfo
* @return file is created or not
*/
private boolean createFile(FileInfo fileInfo) {
BufferedOutputStream bos = null;
try {
if (fileInfo != null) {
File fileReceive = new File(fileInfo.getDestinationDirectory()
+ fileInfo.getFilename());
bos = new BufferedOutputStream(
new FileOutputStream(fileReceive));
// write file content
bos.write(fileInfo.getDataBytes());
bos.flush();
}
} catch (IOException e) {
e.printStackTrace();
return false;
} finally {
closeStream(bos);
}
return true;
}
/**
* close socket
*
* @author kienthuclaptrinh.net
*/
public void closeSocket(Socket socket) {
try {
if (socket != null) {
socket.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* close input stream
*
* @author kienthuclaptrinh.net
*/
public void closeStream(InputStream inputStream) {
try {
if (inputStream != null) {
inputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
/**
* close output stream
*
* @author kienthuclaptrinh.net
*/
public void closeStream(OutputStream outputStream) {
try {
if (outputStream != null) {
outputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
}
Cài đặt phía client
Cấu trúc TCPClient project:
Không giống như giao thức UDP. Giao thức TCP/IP không giới hạn dung lượng cho gói tin mỗi lần transfer (Nó phụ thuôc vào dung bộ nhớ heap). Nên bạn có thể đính kèm nội dung của file dưới dạng byte stream vào đối tượng FileInfo để transfer một thể.
File: TCPClient.java
package vn.kienthuclaptrinh.client;
import java.io.BufferedInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import vn.kienthuclaptrinh.common.FileInfo;
public class TCPClient {
// create Socket object
private Socket client;
private String serverHost = "localhost";
private int serverPort = 9900;
/**
* run program
*
* @author kienthuclaptrinh.net
* @param args
*/
public static void main(String[] args) {
String sourceFilePath = "D:\\client\\lingoes.zip";
String destinationDir = "D:\\server\\";
TCPClient tcpClient = new TCPClient();
tcpClient.connectServer();
tcpClient.sendFile(sourceFilePath, destinationDir);
tcpClient.closeSocket(tcpClient.client);
}
/**
* connect to server
*
* @author kienthuclaptrinh.net
*/
public void connectServer() {
try {
client = new Socket(serverHost, serverPort);
System.out.println("connected to server");
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* send file to server
*
* @param sourceFilePath
* @param destinationDir
*/
public void sendFile(String sourceFilePath, String destinationDir) {
DataOutputStream outToServer = null;
ObjectOutputStream oos = null;
ObjectInputStream ois = null;
try {
// make greeting
outToServer = new DataOutputStream(
client.getOutputStream());
outToServer.writeUTF("Hello from "
+ client.getLocalSocketAddress());
// get file info
FileInfo fileInfo = getFileInfo(sourceFilePath, destinationDir);
// send file
oos = new ObjectOutputStream(client.getOutputStream());
oos.writeObject(fileInfo);
// get confirmation
ois = new ObjectInputStream(client.getInputStream());
fileInfo = (FileInfo) ois.readObject();
if (fileInfo != null) {
System.out.println(fileInfo.getStatus());
}
} catch (IOException ex) {
ex.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} finally {
// close all stream
closeStream(oos);
closeStream(ois);
closeStream(outToServer);
}
}
/**
* get source file info
*
* @author kienthuclaptrinh.net
* @param sourceFilePath
* @param destinationDir
* @return FileInfo
*/
private FileInfo getFileInfo(String sourceFilePath, String destinationDir) {
FileInfo fileInfo = null;
BufferedInputStream bis = null;
try {
File sourceFile = new File(sourceFilePath);
bis = new BufferedInputStream(new FileInputStream(sourceFile));
fileInfo = new FileInfo();
byte[] fileBytes = new byte[(int) sourceFile.length()];
// get file info
bis.read(fileBytes, 0, fileBytes.length);
fileInfo.setFilename(sourceFile.getName());
fileInfo.setDataBytes(fileBytes);
fileInfo.setDestinationDirectory(destinationDir);
} catch (IOException ex) {
ex.printStackTrace();
} finally {
closeStream(bis);
}
return fileInfo;
}
/**
* close socket
*
* @author kienthuclaptrinh.net
*/
public void closeSocket(Socket socket) {
try {
if (socket != null) {
socket.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* close input stream
*
* @author kienthuclaptrinh.net
*/
public void closeStream(InputStream inputStream) {
try {
if (inputStream != null) {
inputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
/**
* close output stream
*
* @author kienthuclaptrinh.net
*/
public void closeStream(OutputStream outputStream) {
try {
if (outputStream != null) {
outputStream.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
}