Socket文件传输实例

Socket文件传输实例

2008/03/25 11:29Server端：

package com.server;

import java.io.*;

import https://www.360docs.net/doc/3e4806026.html,.*;

public class ServerTest {

int port = 8821;

void start() {

Socket s = null;

try {

ServerSocket ss = new ServerSocket(port);

while (true) {

// chose the file

String filePath = "D://test/testSend/test.txt";

File fi = new File(filePath);

System.out.println("the length of the file is :"

+ (int) fi.length());

s = ss.accept();

System.out.println("setup socket connection");

DataInputStream dis = new DataInputStream(

new BufferedInputStream(s.getInputStream()));

dis.readByte();

DataInputStream fis = new DataInputStream(

new BufferedInputStream(new FileInputStream(filePath)));

DataOutputStream ps = new DataOutputStream(s.getOutputStream());

ps.writeUTF(fi.getName());

ps.flush();

ps.writeLong((long) fi.length());

ps.flush();

int bufferSize = 8192;

byte[] buf = new byte[bufferSize];

while (true) {

int read = 0;

if (fis != null) {

read = fis.read(buf);

}

if (read == -1) {

break;

}

ps.write(buf, 0, read);

}

ps.flush();

fis.close();

s.close();

System.out.println("file transfered completly !");

}

} catch (IOException e) {

e.printStackTrace();

}

}

public static void main(String[] args) {

new ServerTest().start();

}

}

package com.server;

import https://www.360docs.net/doc/3e4806026.html,.*;

import java.io.*;

public class ClientSocket {

private String ip;

private int port;

private Socket socket = null;

DataOutputStream out = null;

DataInputStream getMessageStream = null;

public ClientSocket(String ip,int port){

this.ip = ip;

this.port = port;

}

//setup socket connection

public void CteateConnection() throws Exception{

try{

socket = new Socket(ip,port);

}catch(Exception e){

e.printStackTrace();

if(socket !=null)

socket.close();

throw e;

}finally{}

}

public void sendMessage(String sendMessage) throws Exception{ try{

out = new DataOutputStream(socket.getOutputStream());

if(sendMessage.equals("Windows")){

out.writeByte(0x1);

out.flush();

return;

}

if(sendMessage.equals("Unix")){

out.writeByte(0x2);

out.flush();

return;

}

if(sendMessage.equals("Linux")){

out.writeByte(0x3);

out.flush();

}else{

out.writeUTF(sendMessage);

out.flush();

}

}catch(Exception e){

e.printStackTrace();

if(out != null)

out.close();

throw e;

}finally{

}

}

public DataInputStream getMessageStream() throws Exception{

try{

getMessageStream = new DataInputStream(new BufferedInputStream(socket.getInputStream()));

return getMessageStream;

}catch(Exception e){

e.printStackTrace();

if(getMessageStream != null)

getMessageStream.close();

throw e;

}finally{

}

}

public void shutDownConnection(){

try{

if (out!=null)

out.close();

if(getMessageStream!=null)

getMessageStream.close();

if(socket!=null)

socket.close();

}catch(Exception e){}

}

}

Client端：

package com.client;

import java.io.*;

import com.server.ClientSocket;

public class ClientTest {

private ClientSocket cs = null;

private String ip = "localhost";

private int port = 8821;

private String sendMessage = "Windows";

public ClientTest() {

try {

if (createConnection()) {

sendMessage();

getMessage();

}

} catch (Exception e) {

e.printStackTrace();

}

}

private boolean createConnection(){

cs = new ClientSocket(ip,port);

try{

cs.CteateConnection();

System.out.println("connect to the server successfully.");

return true;

}catch(Exception e){

System.out.println("connect to the server failed.");

return false;

}

}

private void sendMessage(){

if(cs==null)

return;

try{

cs.sendMessage(sendMessage);

}catch(Exception e){

System.out.println("send message failed.");

}

}

private void getMessage(){

if(cs==null)

return;

DataInputStream inputStream = null;

try{

inputStream = cs.getMessageStream();

}catch(Exception e){

System.out.println("receive message cache erro.");

return;

}

try{

//the path save the file

String savePath = "D://test/testReceive/";

int bufferSize = 8192;

byte[] buf = new byte[bufferSize];

int passedlen = 0;

long len =0;

savePath +=inputStream.readUTF();

DataOutputStream fileOut = new DataOutputStream(new BufferedOutputStream(new BufferedOutputStream(new FileOutputStream(savePath))));

len = inputStream.readLong();

System.out.println("the length of the file is : "+len);

System.err.println("begin receiving the file......");

while(true){

int read = 0;

if(inputStream!=null){

read = inputStream.read(buf);

}

passedlen +=read;

if(read == -1){

break;

}

System.out.println("have been received "+(passedlen*100/len)+"%");

fileOut.write(buf,0,read);

}

System.out.println("received completly and save as "+savePath);

fileOut.close();

}catch(Exception e){

System.out.println("receive erro :");

e.printStackTrace();

return;

}

}

public static void main(String arg[]){

new ClientTest();

}

}