1137
|
1 package luan.webserver;
|
|
2
|
|
3 import java.io.InputStream;
|
|
4 import java.io.OutputStream;
|
|
5 import java.io.IOException;
|
|
6 import java.net.Socket;
|
|
7 import org.slf4j.Logger;
|
|
8 import org.slf4j.LoggerFactory;
|
|
9 import luan.lib.parser.ParseException;
|
|
10
|
|
11
|
|
12 final class Connection {
|
|
13 private static final Logger logger = LoggerFactory.getLogger(Connection.class);
|
|
14
|
|
15 private final Server server;
|
|
16 private final Socket socket;
|
|
17
|
|
18 Connection(Server server,Socket socket) {
|
|
19 this.server = server;
|
|
20 this.socket = socket;
|
|
21 handle();
|
|
22 }
|
|
23
|
|
24 private void handle() {
|
|
25 try {
|
|
26 InputStream in = socket.getInputStream();
|
|
27 byte[] a = new byte[8192];
|
|
28 int endOfHeader;
|
|
29 int size = 0;
|
|
30 int left = a.length;
|
|
31 outer: while(true) {
|
|
32 int n = in.read(a,size,left);
|
|
33 if( n == -1 )
|
|
34 throw new IOException("unexpected end of input at "+size);
|
|
35 size += n;
|
|
36 for( int i=0; i<=size-4; i++ ) {
|
|
37 if( a[i]=='\r' && a[i+1]=='\n' && a[i+2]=='\r' && a[i+3]=='\n' ) {
|
|
38 endOfHeader = i + 4;
|
|
39 break outer;
|
|
40 }
|
|
41 }
|
|
42 left -= n;
|
|
43 if( left == 0 ) {
|
|
44 byte[] a2 = new byte[2*a.length];
|
|
45 System.arraycopy(a,0,a2,0,size);
|
|
46 a = a2;
|
|
47 left = a.length - size;
|
|
48 }
|
|
49 }
|
|
50 String rawRequest = new String(a,0,endOfHeader);
|
|
51 //System.out.println(rawRequest);
|
|
52 Request request = RequestHeadParser.parse(rawRequest);
|
|
53 //System.out.println(request.headers);
|
|
54
|
|
55 Response response = server.handler.handle(request);
|
|
56 response.headers.put("Connection","close");
|
|
57 response.headers.put("Content-Length",Long.toString(response.body.length));
|
|
58 byte[] header = response.toHeaderString().getBytes();
|
|
59
|
|
60 OutputStream out = socket.getOutputStream();
|
|
61 out.write(header);
|
|
62 copyAll(response.body.content,out);
|
|
63 out.close();
|
|
64 socket.close();
|
|
65 } catch(IOException e) {
|
|
66 logger.info("",e);
|
|
67 } catch(ParseException e) {
|
|
68 logger.info("",e);
|
|
69 }
|
|
70 }
|
|
71
|
|
72 private static void copyAll(InputStream in,OutputStream out)
|
|
73 throws IOException
|
|
74 {
|
|
75 byte[] a = new byte[8192];
|
|
76 int n;
|
|
77 while( (n=in.read(a)) != -1 ) {
|
|
78 out.write(a,0,n);
|
|
79 }
|
|
80 }
|
|
81
|
|
82 }
|