jdk/test/sun/net/www/http/ChunkedInputStream/ChunkedEncodingTest.java

/*
 * Copyright 2004-2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/**
 * @test
 * @bug 4333920
 * @library ../../../../../sun/net/www/httptest/
 * @build HttpCallback HttpServer ClosedChannelList HttpTransaction
 * @run main ChunkedEncodingTest
 * @summary ChunkedEncodingTest unit test
 */

import java.io.*;
import java.net.*;
import java.security.*;

public class ChunkedEncodingTest implements HttpCallback {
    private static String FNPrefix;
    private String[] respBody = new String[52];
    private byte[][] bufs = new byte[52][8*1024];
    private static MessageDigest md5;
    private static byte[] file1Mac, file2Mac;
    public void request (HttpTransaction req) {
        try {
            FileInputStream fis = new FileInputStream(FNPrefix+"test.txt");
            DigestInputStream dis = null;
            md5.reset();
            dis = new DigestInputStream(fis, md5);
            for (int i = 0; i < 52; i++) {
                int n = dis.read(bufs[i]);
                respBody[i] = new String(bufs[i], 0, n);
            }
            file1Mac = dis.getMessageDigest().digest();
            dis.close();
            req.setResponseEntityBody(respBody);
            req.sendResponse(200, "OK");
            req.orderlyClose();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    static void read (InputStream is) throws IOException {
        int c;
        System.out.println ("reading");

        DigestInputStream dis = null;
        md5.reset();
        dis = new DigestInputStream(is, md5);
        while ((c=dis.read()) != -1);
        file2Mac = dis.getMessageDigest().digest();
        dis.close();
        System.out.println ("finished reading");
    }

    static void client (String u) throws Exception {
        URL url = new URL (u);
        System.out.println ("client opening connection to: " + u);
        URLConnection urlc = url.openConnection ();
        InputStream is = urlc.getInputStream ();
        read (is);
        is.close();
    }

    static HttpServer server;

    public static void test () throws Exception {
        try {

            FNPrefix = System.getProperty("test.src", ".")+"/";
            md5 = MessageDigest.getInstance("MD5");
            server = new HttpServer (new ChunkedEncodingTest(), 1, 10, 0);
            System.out.println ("Server: listening on port: " + server.getLocalPort());
            client ("http://localhost:"+server.getLocalPort()+"/d1/foo.html");
        } catch (Exception e) {
            if (server != null) {
                server.terminate();
            }
            throw e;
        }
        if (!MessageDigest.isEqual(file1Mac, file2Mac)) {
            except ("The file sent by server is different from the original file");
        }

        server.terminate();
    }

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

    public static void except (String s) {
        server.terminate();
        throw new RuntimeException (s);
    }
}