代理服务器只能给客户端发送第一组消息

方兆国

最近作一个代理服务器程序，使用来转发视频的，可是他只能给我把第一条请求转发过来，剩下的数据就发布回来了，然后视频就无法播放了

1. package com.liujiaqi.proxy;
   
2. 
   
3. import java.io.BufferedReader;
   
4. import java.io.File;
   
5. import java.io.FileNotFoundException;
   
6. import java.io.FileReader;
   
7. import java.io.IOException;
   
8. import java.io.PrintWriter;
   
9. 
   
10. public class proxy {
    
11.         private static File file = new File("proxy.ini");       //配置文件
    
12.         private static String host = "http://localhost/";  //Web服务器地址
    
13.         private static int port = 9090;
    
14. 
    
15.         public static void createFile() {                          //使用默认参数创建配置文件的函数
    
16.                 try {
    
17.                         file.createNewFile();
    
18.                         PrintWriter writer = new PrintWriter(file);
    
19.                         writer.println("<host>" + host);
    
20.                         writer.println("<port>" + port);
    
21.                         writer.flush();
    
22.                         writer.close();
    
23.                 } catch (IOException e) {
    
24.                         e.printStackTrace();
    
25.                 }
    
26.         }
    
27. 
    
28.         public static void readFile() {                            //读取配置文件中的参数
    
29.                 try {
    
30.                         String temp = null;
    
31.                         BufferedReader reader = new BufferedReader(new FileReader(file));
    
32.                         while ((temp = reader.readLine()) != null) {
    
33.                                 switch (temp.substring(temp.indexOf("<") + 1,
    
34.                                                 temp.lastIndexOf(">"))) {
    
35.                                 case "host":
    
36.                                         host = temp.substring(temp.lastIndexOf(">") + 1);
    
37.                                         break;
    
38.                                 case "port":
    
39.                                         port = Integer.parseInt(temp.substring(temp
    
40.                                                         .lastIndexOf(">") + 1));
    
41.                                         break;
    
42.                                 default:
    
43.                                         break;
    
44.                                 }
    
45.                         }
    
46.                         reader.close();
    
47.                 } catch (FileNotFoundException e) {
    
48.                         e.printStackTrace();
    
49.                 } catch (IOException e) {
    
50.                         e.printStackTrace();
    
51.                 }
    
52.         }
    
53. 
    
54.         public static void main(String[] args) {
    
55.                 if (file.exists()) {                            //如果配置文件存在，则从配置文件中载入启动参数
    
56.                         proxy.readFile();
    
57.                 } else {                                        //如果配置文件不存在，则创建配置文件，然后使用默认的参数来启动
    
58.                         proxy.createFile();
    
59.                 }
    
60.                 ProxyServer server = new ProxyServer(host, port);
    
61.                 server.start();
    
62.         }
    
63. }
    

复制代码

方兆国

这是用来监听代理端口的类

1. package com.fangzhaoguo.proxy;
   
2. 
   
3. import java.io.IOException;
   
4. import java.net.ServerSocket;
   
5. import java.net.Socket;
   
6. import java.util.concurrent.ExecutorService;
   
7. import java.util.concurrent.Executors;
   
8. 
   
9. public class ProxyServer {
   
10.         private static ServerSocket serverSocket;
    
11.         private boolean isAlive = true;
    
12.         private String host = null;
    
13.         private int port = 0;
    
14. 
    
15.         public ProxyServer(final String host, final int port) {
    
16.                 try {
    
17.                         this.host = host;
    
18.                         this.port = port;
    
19.                         serverSocket = new ServerSocket(this.port);
    
20.                 } catch (IOException e) {
    
21.                         e.printStackTrace();
    
22.                 }
    
23.         }
    
24. 
    
25.         public void start() {
    
26.                 ExecutorService service = Executors.newFixedThreadPool(20);   //创建线程池
    
27.                 while (isAlive) {
    
28.                         try {
    
29.                                 Socket socket = serverSocket.accept();
    
30.                                 service.execute(new ProxyConnect(host, socket));    //使用线程池中的线程处理任务
    
31.                         } catch (IOException e) {
    
32.                                 e.printStackTrace();
    
33.                         }
    
34.                 }
    
35.                 try {
    
36.                         serverSocket.close();
    
37.                 } catch (IOException e) {
    
38.                         e.printStackTrace();
    
39.                 }
    
40.         }
    
41. 
    
42.         public void close() {
    
43.                 isAlive = false;
    
44.         }
    
45. }
    

复制代码

方兆国

这是存放视频数据内容的表

1. create table if not exists Data
   
2. (
   
3.         id int not null auto_increment,
   
4.         name varchar(100) not null,
   
5.         data blob not null,
   
6.         size int not null,
   
7.         log timestamp not null default now(),
   
8.         primary key (id)
   
9. );
   

复制代码

id是自动增长的序号，以免数据其他字段的重复导致数据被误刷新
name是视频片断的名称
data是视频片断的数据内容
size是数据片断的大小
log是数据写入日期

方兆国

这是存放视频名称的表

1. create table if not exists Information
   
2. (
   
3.         id int not null auto_increment,
   
4.         url char(100) not null,
   
5.         log timestamp not null default now(),
   
6.         primary key (id)
   
7. );

复制代码

id是自动增长的序号
url是视频的名称，旧版本中使用的是url地址，因此保留了这个名称
log是视频最后一次被浏览的日期

方兆国

这是用来关联视频名称和视频片断的表

1. create table if not exists DataIndex
   
2. (
   
3.         file_id int not null,
   
4.         block_id int not null,
   
5.         primary key (file_id,block_id),
   
6.         foreign key(file_id) references Information(id),
   
7.         foreign key(block_id) references Data(id)
   
8. );

复制代码

file_id是视频名称的id,外键，关联至视频名称表Information的id字段
block_id是视频片断的id,外键，关联至视频数据内容表 Data的id字段

方兆国

这是视频名称表Information上的存储函数，用来插入新行并且返回id

1. CREATE function fun_Information(address char(100))
   
2. RETURNS int
   
3. BEGIN
   
4.         DECLARE file_id int;
   
5.         select count(*) into file_id from Information where url=address;
   
6.         if file_id < 1  then
   
7.                 insert into Information(url) values(address);
   
8.         end if;
   
9.         select id into file_id from Information where url=address;
   
10.         return file_id;
    
11. END

复制代码

方兆国

这是视频数据内容表Data上的存储函数，用来插入新行并且返回id

1. CREATE FUNCTION fun_Data(nValue char(100),dValue blob,sValue int,fValue int)
   
2. RETURNS int
   
3. BEGIN
   
4.         DECLARE bValue int;
   
5.         select count(*) into @Value from Data where name=nValue;
   
6.         if @Value<1  then
   
7.                 insert into Data(name,data,size) values(nValue,dValue,sValue);
   
8.         end if;
   
9.         select id into bValue from Data where name=nValue;
   
10.         select count(*) into @Value from DataIndex inner join Data on Data.id=DataIndex.block_id where Data.name=nValue and DataIndex.file_id=fValue;
    
11.         if @Value<1  then
    
12.                 insert DataIndex(file_id,block_id) values(fValue,bValue);
    
13.         end if;       
    
14.         return bValue;
    
15. END

复制代码

方兆国

接下来就是最关键的内容，也就是客户端和代理服务器之间通信的函数了，问题就出在这儿，代理服务器只能接受并且正常返回第一条请求，剩下的就都出问题了

方兆国

ProxyConnect 使用了 Runnable
run()如下所示，我自己对这一块儿也很糊涂，关键就是http请求和响应的问题

1. public void run() {
   
2.                 int file_id = -1;
   
3.                 int[] block_id = null;
   
4.                 String filename = this.read();
   
5.                 String sql = "{?=call fun_Information(?)}";
   
6.                 file_id = database.Rrocedure(sql, filename);
   
7.                 sql = "update Information set log=now() where id=" + file_id;
   
8.                 database.Write(sql);
   
9. 
   
10.                 while (isAlive) {
    
11.                         sql = "select Data.id,Data.size from Data inner join DataIndex on Data.id=DataIndex.block_id inner join Information on Information.id=DataIndex.file_id where Information.id="
    
12.                                         + file_id + " and Data.name='" + filename + "';";
    
13.                         rs = database.Read(sql);
    
14.                         try {
    
15.                                 if (rs.next()) {
    
16.                                         block_id = new int[rs.getRow()];
    
17.                                         rs.beforeFirst();
    
18.                                         int i = 0, length = 0;
    
19.                                         while (rs.next()) {
    
20.                                                 block_id[i] = rs.getInt(1);
    
21.                                                 length += rs.getInt(2);
    
22.                                                 i++;
    
23.                                         }
    
24.                                         rs.close();
    
25.                                         database.ResultSetClose();
    
26.                                         this.SendHead(length);
    
27.                                         for (i = 0; i < block_id.length; i++) {
    
28.                                                 this.WorkDB(block_id[i]);
    
29.                                                 try {
    
30.                                                         out.flush();
    
31.                                                 } catch (IOException e) {
    
32.                                                         e.printStackTrace();
    
33.                                                 }
    
34.                                         }
    
35.                                 } else {
    
36.                                         rs.close();
    
37.                                         database.ResultSetClose();
    
38.                                         this.WorkNet(file_id, filename);
    
39.                                 }
    
40.                         } catch (SQLException e) {
    
41.                                 e.printStackTrace();
    
42.                         }
    
43.                         filename = this.read();
    
44.                 }
    
45. 
    
46.                 this.close();
    
47.         }

复制代码

方兆国

这是从web服务器读取数据的函数，并且对数据进行处理

1. private String read() {
   
2.                 String buffer = null;
   
3.                 System.out.println("I am the " + Thread.currentThread().getId() + "\t"
   
4.                                 + Thread.currentThread().getName());      //用来显示该线程是第几个线程
   
5.                 try {
   
6.                         do {
   
7.                                 buffer = in.readLine();
   
8.                                 if (buffer.startsWith("Connection: ")) {                                         //如果是以Connection开头，则记录时Keep-Alive还是Close
   
9.                                         connection = buffer.substring(buffer.indexOf(" ") + 1);
   
10.                                         System.out.println(connection);
    
11.                                         if (connection.equals("close")) {
    
12.                                                 isAlive = false;
    
13.                                         }
    
14.                                 }
    
15.                                 if (buffer.startsWith("null")) {
    
16.                                         isAlive = false;
    
17.                                 }
    
18.                         } while (!buffer.startsWith("GET") && isAlive);
    
19.                         buffer = buffer.substring(5, buffer.lastIndexOf(" "));
    
20.                 } catch (IOException e) {
    
21.                         e.printStackTrace();
    
22.                 }
    
23.                 return buffer;
    
24.         }

复制代码

System.out.println("I am the " + Thread.currentThread().getId() + "\t"+ Thread.currentThread().getName());     
用来显示该线程是第几个线程，之前对浏览器的请求有疑惑，因此加入了调试代码


if (buffer.startsWith("Connection: ")) {                                       
                                        connection = buffer.substring(buffer.indexOf(" ") + 1);
                                        System.out.println(connection);
                                        if (connection.equals("close")) {
                                                isAlive = false;
                                        }
如果是以Connection开头，则记录时Keep-Alive还是Close
但是我还是对这个Keep-Alive和Close的区别不是很理解，Close好像是本次会话结束之后就关闭的意思
而且使用buffer.substring(buffer.indexOf("HTTP"));检索的时候会看到有一会儿是HTTP 1.0 有些时候是HTTP 1.1
更诡异的是User-Agent字段，有些时候是GPAC/0.5.0-rev4065，有些时候是 Lavf52.33.0，用的是同一个播放器