/*

 * Copyright 2009 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 4244499 4532049 4700978 4820807 4980042

 * @summary Test ZipInputStream, ZipOutputStream and ZipFile with non-UTF8 encoding

 */

import java.io.*;

import java.nio.charset.*;

import java.util.*;

import java.util.zip.*;

public class ZipCoding {

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

        test("MS932",

             "\u4e00\u4e01", "\uff67\uff68\uff69\uff6a\uff6b\uff6c");

        test("ibm437",

             "\u00e4\u00fc", "German Umlaut \u00fc in comment");

        test("utf-8",

             "\u4e00\u4e01", "\uff67\uff68\uff69\uff6a\uff6b\uff6c");

        test("utf-8",

             "\u00e4\u00fc", "German Umlaut \u00fc in comment");

        test("utf-8",

             "Surrogate\ud801\udc01", "Surrogates \ud800\udc00 in comment");

    }

    static void testZipInputStream(InputStream is, Charset cs,

                                   String name, String comment, byte[] bb)

        throws Exception

    {

        ZipInputStream zis = new ZipInputStream(is, cs);

        ZipEntry e = zis.getNextEntry();

        if (e == null || ! name.equals(e.getName()))

            throw new RuntimeException("ZipIS name doesn't match!");

        byte[] bBuf = new byte[bb.length << 1];

        int n = zis.read(bBuf, 0, bBuf.length);

        if (n != bb.length ||

            !Arrays.equals(bb, Arrays.copyOf(bBuf, n))) {

            throw new RuntimeException("ZipIS content doesn't match!");

        }

        zis.close();

    }

    static void testZipFile(File f, Charset cs,

                            String name, String comment, byte[] bb)

        throws Exception

    {

        ZipFile zf = new ZipFile(f, cs);

        Enumeration<? extends ZipEntry> zes = zf.entries();

        ZipEntry e = (ZipEntry)zes.nextElement();

        if (! name.equals(e.getName()) ||

            ! comment.equals(e.getComment()))

            throw new RuntimeException("ZipFile: name/comment doesn't match!");

        InputStream is = zf.getInputStream(e);

        if (is == null)

            throw new RuntimeException("ZipFile: getIS failed!");

        byte[] bBuf = new byte[bb.length << 1];

        int n = 0;

        int nn =0;

        while ((nn = is.read(bBuf, n, bBuf.length-n)) != -1) {

            n += nn;

        }

        if (n != bb.length ||

            !Arrays.equals(bb, Arrays.copyOf(bBuf, n))) {

            throw new RuntimeException("ZipFile content doesn't match!");

        }

        zf.close();

    }

    static void test(String csn, String name, String comment)

        throws Exception

    {

        byte[] bb = "This is the conent of the zipfile".getBytes("ISO-8859-1");

        Charset cs = Charset.forName(csn);

        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        ZipOutputStream zos = new ZipOutputStream(baos, cs);

        ZipEntry e = new ZipEntry(name);

        e.setComment(comment);

        zos.putNextEntry(e);

        zos.write(bb, 0, bb.length);

        zos.closeEntry();

        zos.close();

        ByteArrayInputStream bis = new ByteArrayInputStream(baos.toByteArray());

        testZipInputStream(bis, cs, name, comment, bb);

        if ("utf-8".equals(csn)) {

            // EFS should be set

            bis.reset();

            testZipInputStream(bis, Charset.forName("MS932"), name, comment, bb);

        }

        File f = new File(new File(System.getProperty("test.dir", ".")),

                          "zfcoding.zip");

        FileOutputStream fos = new FileOutputStream(f);

        baos.writeTo(fos);

        fos.close();

        testZipFile(f, cs, name, comment, bb);

        if ("utf-8".equals(csn)) {

            testZipFile(f, Charset.forName("MS932"), name, comment, bb);

        }

        f.delete();

    }

}