B6660405.java revision 55:b6f7db7d8648 
1/*
2 * Copyright 2008 Sun Microsystems, Inc.  All Rights Reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
20 * CA 95054 USA or visit www.sun.com if you need additional information or
21 * have any questions.
22 */
23
24/*
25 * @test
26 * @bug 6660405
27 * @summary HttpURLConnection returns the wrong InputStream
28 */
29
30import java.net.*;
31import java.util.*;
32import java.io.*;
33import com.sun.net.httpserver.*;
34import java.util.concurrent.Executors;
35import java.util.concurrent.ExecutorService;
36
37public class B6660405
38{
39    com.sun.net.httpserver.HttpServer httpServer;
40    ExecutorService executorService;
41
42    static class MyCacheResponse extends CacheResponse {
43        private byte[] buf = new byte[1024];
44
45        public MyCacheResponse() {
46        }
47
48        @Override
49        public Map<String, List<String>> getHeaders() throws IOException
50        {
51            Map<String, List<String>> h = new HashMap<String, List<String>>();
52            ArrayList<String> l = new ArrayList<String>();
53            l.add("HTTP/1.1 200 OK");
54            h.put(null, l);
55            l = new ArrayList<String>();
56            l.add("1024");
57            h.put("Content-Length", l);
58            return h;
59        }
60
61        @Override
62        public InputStream getBody() throws IOException
63        {
64            return new ByteArrayInputStream(buf);
65        }
66
67    }
68    static class MyResponseCache extends ResponseCache {
69
70        public MyResponseCache() {
71        }
72
73        @Override
74        public CacheResponse get(URI uri, String rqstMethod, Map<String, List<String>> rqstHeaders) throws IOException
75        {
76            if (uri.getPath().equals("/redirect/index.html")) {
77                return new MyCacheResponse();
78            }
79            return null;
80        }
81
82        @Override
83        public CacheRequest put(URI uri, URLConnection conn) throws IOException
84        {
85            return null;
86        }
87
88    }
89
90    public static void main(String[] args)
91    {
92        new B6660405();
93    }
94
95    public B6660405()
96    {
97        try {
98            startHttpServer();
99            doClient();
100        } catch (IOException ioe) {
101            System.err.println(ioe);
102        }
103    }
104
105    void doClient() {
106        ResponseCache.setDefault(new MyResponseCache());
107        try {
108            InetSocketAddress address = httpServer.getAddress();
109
110            // GET Request
111            URL url = new URL("http://localhost:" + address.getPort() + "/test/index.html");
112            HttpURLConnection uc = (HttpURLConnection)url.openConnection();
113            int code = uc.getResponseCode();
114            System.err.println("response code = " + code);
115            int l = uc.getContentLength();
116            System.err.println("content-length = " + l);
117            InputStream in = uc.getInputStream();
118            int i = 0;
119            // Read till end of stream
120            do {
121                i = in.read();
122            } while (i != -1);
123            in.close();
124        } catch (IOException e) {
125            throw new RuntimeException("Got the wrong InputStream after checking headers");
126        } finally {
127            httpServer.stop(1);
128            executorService.shutdown();
129        }
130    }
131
132    /**
133     * Http Server
134     */
135    public void startHttpServer() throws IOException {
136        httpServer = com.sun.net.httpserver.HttpServer.create(new InetSocketAddress(0), 0);
137
138        // create HttpServer context
139        HttpContext ctx = httpServer.createContext("/test/", new MyHandler());
140
141        executorService = Executors.newCachedThreadPool();
142        httpServer.setExecutor(executorService);
143        httpServer.start();
144    }
145
146    class MyHandler implements HttpHandler {
147        public void handle(HttpExchange t) throws IOException {
148            InputStream is = t.getRequestBody();
149            Headers reqHeaders = t.getRequestHeaders();
150            Headers resHeaders = t.getResponseHeaders();
151
152            int i = 0;
153            // Read till end of stream
154            do {
155                i = is.read();
156            } while (i != -1);
157            is.close();
158            resHeaders.add("Location", "http://foo.bar/redirect/index.html");
159            t.sendResponseHeaders(302, -1);
160            t.close();
161        }
162    }
163}
164