package com.httpget.test;
import java.io.*;
import java.net.*;
public class HttpGetServer {
// 服务器根目录,post.html, upload.html都放在该位置
public static String WEB_ROOT = "c:/root";
// 端口
private int port;
// 用户请求的文件的url
private String requestPath;
public HttpGetServer(String root, int port) {
WEB_ROOT = root;
this.port = port;
requestPath = null;
}
public final static boolean DEBUG = true;//调试用
private static int BUFFER_SIZE = 8096;//缓冲区大小
public void saveToFile(String destUrl, String fileName) throws IOException {
FileOutputStream fos = null;
BufferedInputStream bis = null;
HttpURLConnection httpUrl = null;
URL url = null;
byte[] buf = new byte[BUFFER_SIZE];
int size = 0;
// 建立链接
url = new URL(destUrl);
httpUrl = (HttpURLConnection) url.openConnection();
// 连接指定的资源
httpUrl.connect();
// 获取网络输入流
bis = new BufferedInputStream(httpUrl.getInputStream());
// 建立文件
fos = new FileOutputStream(fileName);
if (this.DEBUG)
System.out.println("正在获取链接[" + destUrl + "]的内容...\n将其保存为文件["
+ fileName + "]");
// 保存文件
while ((size = bis.read(buf)) != -1)
fos.write(buf, 0, size);
fos.close();
bis.close();
httpUrl.disconnect();
}
// 处理GET请求
private void doGet(BufferedReader reader, OutputStream out) throws Exception {
if(this.requestPath.substring(1).indexOf("?")!=-1){
String serverUrl = this.requestPath.substring(this.requestPath.indexOf("?")+1);
System.out.println(serverUrl);
saveToFile(serverUrl,System.getProperty("user.home")+"//Termb.dll");
System.out.println(new File(System.getProperty("user.home")+"//Termb.dll").exists());
}
if (new File(WEB_ROOT + this.requestPath).exists()) {
//从服务器根目录下找到用户请求的文件并发送回浏览器
InputStream fileIn = new FileInputStream(WEB_ROOT + this.requestPath);
byte[] buf = new byte[fileIn.available()];
fileIn.read(buf);
out.write(buf);
out.close();
fileIn.close();
reader.close();
System.out.println("request complete.");
}
}
public void service() throws Exception {
ServerSocket serverSocket = new ServerSocket(this.port, 1, InetAddress
.getByName("localhost"));
System.out.println("server is ok.");
// 开启serverSocket等待用户请求到来,然后根据请求的类别作处理
while (true) {
Socket socket = serverSocket.accept();
System.out.println("new request coming.");
BufferedReader reader = new BufferedReader(new InputStreamReader(
socket.getInputStream(), "UTF-8"));
String line = reader.readLine();
String method = line.substring(0, 4).trim();
OutputStream out = socket.getOutputStream();
this.requestPath = line.split(" ")[1];
System.out.println(method);
this.doGet(reader, out);
socket.close();
System.out.println("socket closed.");
}
}
}