/*

 * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/* @test

 * @summary Unit test for java.net.ResponseCache

 * @bug 4837267

 * @author Yingxian Wang

 */

import java.net.*;

import java.util.*;

import java.io.*;

import sun.net.www.ParseUtil;

import javax.net.ssl.*;

/**

 * Request should get serviced by the cache handler. Response get

 * saved through the cache handler.

 */

public class ResponseCacheTest implements Runnable {

    ServerSocket ss;

    static URL url1;

    static URL url2;

    static String FNPrefix, OutFNPrefix;

    static List<Closeable> streams = new ArrayList<>();

    static List<File> files = new ArrayList<>();

    /*

     * Our "http" server to return a 404 */

    public void run() {

        Socket s = null;

        FileInputStream fis = null;

        try {

            s = ss.accept();

            InputStream is = s.getInputStream ();

            BufferedReader r = new BufferedReader(new InputStreamReader(is));

            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(FNPrefix+"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");

            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();

        } finally {

            try { ss.close(); } catch (IOException unused) {}

            try { s.close(); } catch (IOException unused) {}

            try { fis.close(); } catch (IOException unused) {}

        }

    }

static class NameVerifier implements HostnameVerifier {

        public boolean verify(String hostname, SSLSession session) {

            return true;

        }

    }

    ResponseCacheTest() 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("request headers: "+http.getRequestProperties());

        System.out.println("responsecode is :"+http.getResponseCode());

        Map<String,List<String>> 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(OutFNPrefix+"file1");

        PrintStream out = new PrintStream(

                                 new BufferedOutputStream(

                                    new FileOutputStream(fileout)));

        while ((x=r.readLine()) != null) {

            out.print(x+"\n");

        }

        out.flush();

        out.close();

        http.disconnect();

        // testing ResponseCacheHandler.put()

        url2 = new URL("http://localhost:" +

                       Integer.toString(ss.getLocalPort())+"/file2.1");

        http = (HttpURLConnection)url2.openConnection();

        System.out.println("responsecode2 is :"+http.getResponseCode());

        Map<String,List<String>> headers2 = http.getHeaderFields();

        try {

            is = http.getInputStream();

        } catch (IOException ioex) {

            throw new RuntimeException(ioex.getMessage());

        }

        r = new BufferedReader(new InputStreamReader(is));

        fileout = new File(OutFNPrefix+"file2.2");

        out = new PrintStream(

                                 new BufferedOutputStream(

                                    new FileOutputStream(fileout)));

        while ((x=r.readLine()) != null) {

            out.print(x+"\n");

        }

        out.flush();

        out.close();

        // assert (headers1 == headers2 && file1 == file2.2)

        File file1 = new File(OutFNPrefix+"file1");

        File file2 = new File(OutFNPrefix+"file2.2");

        files.add(file1);

        files.add(file2);

        System.out.println("headers1"+headers1+"\nheaders2="+headers2);

        if (!headers1.equals(headers2) || file1.length() != file2.length()) {

            throw new RuntimeException("test failed");

        }

    }

    public static void main(String args[]) throws Exception {

        try {

            ResponseCache.setDefault(new MyResponseCache());

            FNPrefix = System.getProperty("test.src", ".")+"/";

            OutFNPrefix = System.getProperty("test.scratch", ".")+"/";

            new ResponseCacheTest();

        } finally{

            ResponseCache.setDefault(null);

            for (Closeable c: streams) {

                try { c.close(); } catch (IOException unused) {}

            }

            for (File f: files) {

                f.delete();

            }

        }

    }

    static class MyResponseCache extends ResponseCache {

        public CacheResponse

        get(URI uri, String rqstMethod, Map<String,List<String>> rqstHeaders)

            throws IOException {

            if (uri.equals(ParseUtil.toURI(url1))) {

                return new MyCacheResponse(FNPrefix+"file1.cache");

            }

            return null;

        }

        public CacheRequest put(URI uri, URLConnection conn)  throws IOException {

            // save cache to file2.cache

            // 1. serialize headers into file2.cache

            // 2. write data to file2.cache

            return new MyCacheRequest(OutFNPrefix+"file2.cache", conn.getHeaderFields());

        }

    }

    static class MyCacheResponse extends CacheResponse {

        FileInputStream fis;

        Map<String,List<String>> headers;

        public MyCacheResponse(String filename) {

            try {

                fis = new FileInputStream(new File(filename));

                streams.add(fis);

                ObjectInputStream ois = new ObjectInputStream(fis);

                headers = (Map<String,List<String>>)ois.readObject();

            } catch (Exception ex) {

                // throw new RuntimeException(ex.getMessage());

            }

        }

        public InputStream getBody() throws IOException {

            return fis;

        }

        public Map<String,List<String>> getHeaders() throws IOException {

            return headers;

        }

    }

    static class MyCacheRequest extends CacheRequest {

        FileOutputStream fos;

        public MyCacheRequest(String filename, Map<String,List<String>> rspHeaders) {

            try {

                File file = new File(filename);

                fos = new FileOutputStream(file);

                streams.add(fos);

                files.add(file);

                ObjectOutputStream oos = new ObjectOutputStream(fos);

                oos.writeObject(rspHeaders);

            } catch (Exception ex) {

                throw new RuntimeException(ex.getMessage());

            }

        }

        public OutputStream getBody() throws IOException {

            return fos;

        }

        public void abort() {

            // no op

        }

    }

}