/*

 * 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

 * @bug 4434723 4482726 4559072 4638365 4795550 5081340 5103988 6253145

 *   6984545

 * @summary Test FileChannel.transferFrom and transferTo

 * @library ..

 */

import java.io.*;

import java.net.*;

import java.nio.*;

import java.nio.channels.*;

import java.nio.channels.spi.SelectorProvider;

import java.nio.file.StandardOpenOption;

import java.nio.file.FileAlreadyExistsException;

import java.util.Random;

public class Transfer {

    private static Random generator = new Random();

    private static int[] testSizes = {

        0, 10, 1023, 1024, 1025, 2047, 2048, 2049 };

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

        testFileChannel();

        for (int i=0; i<testSizes.length; i++)

            testReadableByteChannel(testSizes[i]);

        xferTest02(); // for bug 4482726

        xferTest03(); // for bug 4559072

        xferTest04(); // for bug 4638365

        xferTest05(); // for bug 4638365

        xferTest06(); // for bug 5081340

        xferTest07(); // for bug 5103988

        xferTest08(); // for bug 6253145

        xferTest09(); // for bug 6984545

    }

    private static void testFileChannel() throws Exception {

        File source = File.createTempFile("source", null);

        source.deleteOnExit();

        File sink = File.createTempFile("sink", null);

        sink.deleteOnExit();

        FileOutputStream fos = new FileOutputStream(source);

        FileChannel sourceChannel = fos.getChannel();

        sourceChannel.write(ByteBuffer.wrap(

            "Use the source, Luke!".getBytes()));

        sourceChannel.close();

        FileInputStream fis = new FileInputStream(source);

        sourceChannel = fis.getChannel();

        RandomAccessFile raf = new RandomAccessFile(sink, "rw");

        FileChannel sinkChannel = raf.getChannel();

        long oldSinkPosition = sinkChannel.position();

        long oldSourcePosition = sourceChannel.position();

        long bytesWritten = sinkChannel.transferFrom(sourceChannel, 0, 10);

        if (bytesWritten != 10)

            throw new RuntimeException("Transfer failed");

        if (sourceChannel.position() == oldSourcePosition)

            throw new RuntimeException("Source position didn't change");

        if (sinkChannel.position() != oldSinkPosition)

            throw new RuntimeException("Sink position changed");

        if (sinkChannel.size() != 10)

            throw new RuntimeException("Unexpected sink size");

        bytesWritten = sinkChannel.transferFrom(sourceChannel, 1000, 10);

        if (bytesWritten > 0)

            throw new RuntimeException("Wrote past file size");

        sourceChannel.close();

        sinkChannel.close();

        source.delete();

        sink.delete();

    }

    private static void testReadableByteChannel(int size) throws Exception {

        SelectorProvider sp = SelectorProvider.provider();

        Pipe p = sp.openPipe();

        Pipe.SinkChannel sink = p.sink();

        Pipe.SourceChannel source = p.source();

        sink.configureBlocking(false);

        ByteBuffer outgoingdata = ByteBuffer.allocateDirect(size + 10);

        byte[] someBytes = new byte[size + 10];

        generator.nextBytes(someBytes);

        outgoingdata.put(someBytes);

        outgoingdata.flip();

        int totalWritten = 0;

        while (totalWritten < size + 10) {

            int written = sink.write(outgoingdata);

            if (written < 0)

                throw new Exception("Write failed");

            totalWritten += written;

        }

        File f = File.createTempFile("blah"+size, null);

        f.deleteOnExit();

        RandomAccessFile raf = new RandomAccessFile(f, "rw");

        FileChannel fc = raf.getChannel();

        long oldPosition = fc.position();

        long bytesWritten = fc.transferFrom(source, 0, size);

        fc.force(true);

        if (bytesWritten != size)

            throw new RuntimeException("Transfer failed");

        if (fc.position() != oldPosition)

            throw new RuntimeException("Position changed");

        if (fc.size() != size)

            throw new RuntimeException("Unexpected sink size "+ fc.size());

        fc.close();

        sink.close();

        source.close();

        f.delete();

    }

    public static void xferTest02() throws Exception {

        byte[] srcData = new byte[5000];

        for (int i=0; i<5000; i++)

            srcData[i] = (byte)generator.nextInt();

        // get filechannel for the source file.

        File source = File.createTempFile("source", null);

        source.deleteOnExit();

        RandomAccessFile raf1 = new RandomAccessFile(source, "rw");

        FileChannel fc1 = raf1.getChannel();

        // write out data to the file channel

        long bytesWritten = 0;

        while (bytesWritten < 5000) {

            bytesWritten = fc1.write(ByteBuffer.wrap(srcData));

        }

        // get filechannel for the dst file.

        File dest = File.createTempFile("dest", null);

        dest.deleteOnExit();

        RandomAccessFile raf2 = new RandomAccessFile(dest, "rw");

        FileChannel fc2 = raf2.getChannel();

        int bytesToWrite = 3000;

        int startPosition = 1000;

        bytesWritten = fc1.transferTo(startPosition, bytesToWrite, fc2);

        fc1.close();

        fc2.close();

        raf1.close();

        raf2.close();

        source.delete();

        dest.delete();

    }

    public static void xferTest03() throws Exception {

        byte[] srcData = new byte[] {1,2,3,4} ;

        // get filechannel for the source file.

        File source = File.createTempFile("source", null);

        source.deleteOnExit();

        RandomAccessFile raf1 = new RandomAccessFile(source, "rw");

        FileChannel fc1 = raf1.getChannel();

        fc1.truncate(0);

        // write out data to the file channel

        int bytesWritten = 0;

        while (bytesWritten < 4) {

            bytesWritten = fc1.write(ByteBuffer.wrap(srcData));

        }

        // get filechannel for the dst file.

        File dest = File.createTempFile("dest", null);

        dest.deleteOnExit();

        RandomAccessFile raf2 = new RandomAccessFile(dest, "rw");

        FileChannel fc2 = raf2.getChannel();

        fc2.truncate(0);

        fc1.transferTo(0, srcData.length + 1, fc2);

        if (fc2.size() > 4)

            throw new Exception("xferTest03 failed");

        fc1.close();

        fc2.close();

        raf1.close();

        raf2.close();

        source.delete();

        dest.delete();

    }

    // Test transferTo with large file

    public static void xferTest04() throws Exception {

        // Windows and Linux can't handle the really large file sizes for a

        // truncate or a positional write required by the test for 4563125

        String osName = System.getProperty("os.name");

        if (!osName.startsWith("SunOS"))

            return;

        File source = File.createTempFile("blah", null);

        source.deleteOnExit();

        long testSize = ((long)Integer.MAX_VALUE) * 2;

        initTestFile(source, 10);

        RandomAccessFile raf = new RandomAccessFile(source, "rw");

        FileChannel fc = raf.getChannel();

        fc.write(ByteBuffer.wrap("Use the source!".getBytes()), testSize - 40);

        fc.close();

        raf.close();

        File sink = File.createTempFile("sink", null);

        sink.deleteOnExit();

        FileInputStream fis = new FileInputStream(source);

        FileChannel sourceChannel = fis.getChannel();

        raf = new RandomAccessFile(sink, "rw");

        FileChannel sinkChannel = raf.getChannel();

        long bytesWritten = sourceChannel.transferTo(testSize -40, 10,

                                                     sinkChannel);

        if (bytesWritten != 10) {

            throw new RuntimeException("Transfer test 4 failed " +

                                       bytesWritten);

        }

        sourceChannel.close();

        sinkChannel.close();

        source.delete();

        sink.delete();

    }

    // Test transferFrom with large file

    public static void xferTest05() throws Exception {

        // Create a source file & large sink file for the test

        File source = File.createTempFile("blech", null);

        source.deleteOnExit();

        initTestFile(source, 100);

        // Create the sink file as a sparse file if possible

        File sink = null;

        FileChannel fc = null;

        while (fc == null) {

            sink = File.createTempFile("sink", null);

            // re-create as a sparse file

            sink.delete();

            try {

                fc = FileChannel.open(sink.toPath(),

                                      StandardOpenOption.CREATE_NEW,

                                      StandardOpenOption.WRITE,

                                      StandardOpenOption.SPARSE);

            } catch (FileAlreadyExistsException ignore) {

                // someone else got it

            }

        }

        sink.deleteOnExit();

        long testSize = ((long)Integer.MAX_VALUE) * 2;

        try {

            fc.write(ByteBuffer.wrap("Use the source!".getBytes()),

                     testSize - 40);

        } catch (IOException e) {

            // Can't set up the test, abort it

            System.err.println("xferTest05 was aborted.");

            return;

        } finally {

            fc.close();

        }

        // Get new channels for the source and sink and attempt transfer

        FileChannel sourceChannel = new FileInputStream(source).getChannel();

        try {

            FileChannel sinkChannel = new RandomAccessFile(sink, "rw").getChannel();

            try {

                long bytesWritten = sinkChannel.transferFrom(sourceChannel,

                                                             testSize - 40, 10);

                if (bytesWritten != 10) {

                    throw new RuntimeException("Transfer test 5 failed " +

                                               bytesWritten);

                }

            } finally {

                sinkChannel.close();

            }

        } finally {

            sourceChannel.close();

        }

        source.delete();

        sink.delete();

    }

    static void checkFileData(File file, String expected) throws Exception {

        FileInputStream fis = new FileInputStream(file);

        Reader r = new BufferedReader(new InputStreamReader(fis, "ASCII"));

        StringBuilder sb = new StringBuilder();

        int c;

        while ((c = r.read()) != -1)

            sb.append((char)c);

        String contents = sb.toString();

        if (! contents.equals(expected))

            throw new Exception("expected: " + expected

                                + ", got: " + contents);

        r.close();

    }

    // Test transferFrom asking for more bytes than remain in source

    public static void xferTest06() throws Exception {

        String data = "Use the source, Luke!";

        File source = File.createTempFile("source", null);

        source.deleteOnExit();

        File sink = File.createTempFile("sink", null);

        sink.deleteOnExit();

        FileOutputStream fos = new FileOutputStream(source);

        fos.write(data.getBytes("ASCII"));

        fos.close();

        FileChannel sourceChannel =

            new RandomAccessFile(source, "rw").getChannel();

        sourceChannel.position(7);

        long remaining = sourceChannel.size() - sourceChannel.position();

        FileChannel sinkChannel =

            new RandomAccessFile(sink, "rw").getChannel();

        long n = sinkChannel.transferFrom(sourceChannel, 0L,

                                          sourceChannel.size()); // overflow

        if (n != remaining)

            throw new Exception("n == " + n + ", remaining == " + remaining);

        sinkChannel.close();

        sourceChannel.close();

        checkFileData(source, data);

        checkFileData(sink, data.substring(7,data.length()));

        source.delete();

    }

    // Test transferTo to non-blocking socket channel

    public static void xferTest07() throws Exception {

        File source = File.createTempFile("source", null);

        source.deleteOnExit();

        FileChannel sourceChannel = new RandomAccessFile(source, "rw")

            .getChannel();

        sourceChannel.position(32000L)

            .write(ByteBuffer.wrap("The End".getBytes()));

        // The sink is a non-blocking socket channel

        ServerSocketChannel ssc = ServerSocketChannel.open();

        ssc.socket().bind(new InetSocketAddress(0));

        InetSocketAddress sa = new InetSocketAddress(

            InetAddress.getLocalHost(), ssc.socket().getLocalPort());

        SocketChannel sink = SocketChannel.open(sa);

        sink.configureBlocking(false);

        SocketChannel other = ssc.accept();

        long size = sourceChannel.size();

        // keep sending until congested

        long n;

        do {

            n = sourceChannel.transferTo(0, size, sink);

        } while (n > 0);

        sourceChannel.close();

        sink.close();

        other.close();

        ssc.close();

        source.delete();

    }

    // Test transferTo with file positions larger than 2 and 4GB

    public static void xferTest08() throws Exception {

        // Creating a sparse 6GB file on Windows takes too long

        String osName = System.getProperty("os.name");

        if (osName.startsWith("Windows"))

            return;

        final long G = 1024L * 1024L * 1024L;

        // Create 6GB file

        File file = File.createTempFile("source", null);

        file.deleteOnExit();

        RandomAccessFile raf = new RandomAccessFile(file, "rw");

        FileChannel fc = raf.getChannel();

        try {

            fc.write(ByteBuffer.wrap("0123456789012345".getBytes("UTF-8")), 6*G);

        } catch (IOException x) {

            System.err.println("Unable to create test file:" + x);

            fc.close();

            return;

        }

        // Setup looback connection and echo server

        ServerSocketChannel ssc = ServerSocketChannel.open();

        ssc.socket().bind(new InetSocketAddress(0));

        InetAddress lh = InetAddress.getLocalHost();

        InetSocketAddress isa = new InetSocketAddress(lh, ssc.socket().getLocalPort());

        SocketChannel source = SocketChannel.open(isa);

        SocketChannel sink = ssc.accept();

        Thread thr = new Thread(new EchoServer(sink));

        thr.start();

        // Test data is array of positions and counts

        long testdata[][] = {

            { 2*G-1,    1 },

            { 2*G-1,    10 },       // across 2GB boundary

            { 2*G,      1 },

            { 2*G,      10 },

            { 2*G+1,    1 },

            { 4*G-1,    1 },

            { 4*G-1,    10 },       // across 4GB boundary

            { 4*G,      1 },

            { 4*G,      10 },

            { 4*G+1,    1 },

            { 5*G-1,    1 },

            { 5*G-1,    10 },

            { 5*G,      1 },

            { 5*G,      10 },

            { 5*G+1,    1 },

            { 6*G,      1 },

        };

        ByteBuffer sendbuf = ByteBuffer.allocateDirect(100);

        ByteBuffer readbuf = ByteBuffer.allocateDirect(100);

        try {

            byte value = 0;

            for (int i=0; i<testdata.length; i++) {

                long position = testdata[(int)i][0];

                long count = testdata[(int)i][1];

                // generate bytes

                for (long j=0; j<count; j++) {

                    sendbuf.put(++value);

                }

                sendbuf.flip();

                // write to file and transfer to echo server

                fc.write(sendbuf, position);

                fc.transferTo(position, count, source);

                // read from echo server

                long nread = 0;

                while (nread < count) {

                    int n = source.read(readbuf);

                    if (n < 0)

                        throw new RuntimeException("Premature EOF!");

                    nread += n;

                }

                // check reply from echo server

                readbuf.flip();

                sendbuf.flip();

                if (!readbuf.equals(sendbuf))

                    throw new RuntimeException("Echo'ed bytes do not match!");

                readbuf.clear();

                sendbuf.clear();

            }

        } finally {

            source.close();

            ssc.close();

            fc.close();

            file.delete();

        }

    }

    // Test that transferFrom with FileChannel source that is not readable

    // throws NonReadableChannelException

    static void xferTest09() throws Exception {

        File source = File.createTempFile("source", null);

        source.deleteOnExit();

        File target = File.createTempFile("target", null);

        target.deleteOnExit();

        FileChannel fc1 = new FileOutputStream(source).getChannel();

        FileChannel fc2 = new RandomAccessFile(target, "rw").getChannel();

        try {