1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
|
/* @test @(#)file2.1 1.1 03/08/09
* @summary Unit test for java.net.ResponseCacheHandler
* @bug 4837267
* @author Yingxian Wang
*/
import java.net.*;
import java.util.*;
import java.io.*;
import java.nio.*;
import java.nio.channels.ReadableByteChannel;
import sun.net.www.ParseUtil;
/**
* Request should get serviced by the cache handler. Response get
* saved through the cache handler.
*/
public class ResponseCacheHandlerTest implements Runnable {
ServerSocket ss;
static URL url1;
static URL url2;
static boolean flag = false;
/*
* Our "http" server to return a 404 */
public void run() {
try {
Socket s = ss.accept();
// check request contains "Cookie"
InputStream is = s.getInputStream ();
BufferedReader r = new BufferedReader(new InputStreamReader(is));
boolean flag = false;
String x;
while ((x=r.readLine()) != null) {
if (x.length() ==0) {
break;
}
}
PrintStream out = new PrintStream(
new BufferedOutputStream(
s.getOutputStream() ));
/* send file2.1 */
File file2 = new File("file2.1");
out.print("HTTP/1.1 200 OK\r\n");
out.print("Content-Type: text/html; charset=iso-8859-1\r\n");
out.print("Content-Length: "+file2.length()+"\r\n");
out.print("Connection: close\r\n");
out.print("\r\n");
FileInputStream fis = new FileInputStream(file2);
byte[] buf = new byte[(int)file2.length()];
int len;
while ((len = fis.read(buf)) != -1) {
out.print(new String(buf));
}
out.flush();
s.close();
ss.close();
} catch (Exception e) {
e.printStackTrace();
}
}
ResponseCacheHandlerTest() throws Exception {
/* start the server */
ss = new ServerSocket(0);
(new Thread(this)).start();
/* establish http connection to server */
url1 = new URL("http://localhost/file1.cache");
HttpURLConnection http = (HttpURLConnection)url1.openConnection();
InputStream is = null;
System.out.println("headers1:"+http.getHeaderFields());
try {
is = http.getInputStream();
} catch (IOException ioex) {
throw new RuntimeException(ioex.getMessage());
}
BufferedReader r = new BufferedReader(new InputStreamReader(is));
String x;
File fileout = new File("file1");
PrintStream out = new PrintStream(
new BufferedOutputStream(
new FileOutputStream(fileout)));
while ((x=r.readLine()) != null) {
out.print(x+"\n");
}
out.flush();
out.close();
// if !(file1.out == file1.cache)
// throw new RuntimeException("testing ResponseCacheHandler.get() failed");
http.disconnect();
// testing ResponseCacheHandler.put()
url2 = new URL("http://localhost:" +
Integer.toString(ss.getLocalPort())+"/file2.1");
http = (HttpURLConnection)url2.openConnection();
System.out.println("headers2:"+http.getHeaderFields());
try {
is = http.getInputStream();
} catch (IOException ioex) {
throw new RuntimeException(ioex.getMessage());
}
DataInputStream dis = new DataInputStream(is);
fileout = new File("file2.2");
byte[] buf = new byte[(int)new File("file2.1").length()];
DataOutputStream dos = new DataOutputStream(new FileOutputStream(fileout));
int len;
while ((len=dis.read(buf)) != -1) {
dos.write(buf, 0, len);
}
dos.flush();
dos.close();
// if !(file2.1 == file2.2 == file2.cache)
// throw new RuntimeException("testing ResponseCacheHandler.put() failed");
}
public static void main(String args[]) throws Exception {
ResponseCacheHandler.setDefault(new MyResponseCacheHandler());
new ResponseCacheHandlerTest();
}
static class MyResponseCacheHandler extends ResponseCacheHandler {
public Response
get(URI uri, Map requestHeaders)
throws IOException {
if (uri.equals(ParseUtil.toURI(url1))) {
return new MyResponse("file1.cache");
} if (uri.equals(ParseUtil.toURI(url2)) && !flag) {
flag = true;
return null;
} else {
return new MyResponse("file2.cache");
}
}
public boolean put(URI uri, Response response) throws IOException {
// save cache to file2.cache
// 1. serialize headers into file2.cache
// 2. write data to file2.cache
File file = new File("file2.cache");
FileOutputStream fos = new FileOutputStream(file);
ObjectOutputStream oos = new ObjectOutputStream(fos);
oos.writeObject(response.getHeaders());
ReadableByteChannel rbc = response.getBody();
int len = (int)new File("file2.1").length();
ByteBuffer buf = ByteBuffer.allocate(len);
byte[] b = new byte[len];
while ((len = rbc.read(buf)) != -1) {
buf.flip();
buf.get(b, 0, len);
buf.compact();
oos.write(b, 0, len);
}
oos.close();
return true;
}
}
static class MyResponse extends Response {
FileInputStream fis;
Map headers;
public MyResponse(String filename) {
try {
fis = new FileInputStream(new File(filename));
headers = (Map)new ObjectInputStream(fis).readObject();
} catch (Exception ex) {
throw new RuntimeException(ex.getMessage());
}
}
public ReadableByteChannel getBody() throws IOException {
return fis.getChannel();
}
public Map getHeaders() throws IOException {
return headers;
}
}
}
|
