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") || osName.contains("OS X")))

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 {

fc2.transferFrom(fc1, 0L, 0);

throw new RuntimeException("NonReadableChannelException expected");

} catch (NonReadableChannelException expected) {

} finally {

fc1.close();

fc2.close();

}

}

/**

private static void initTestFile(File blah, long size) throws Exception {

if (blah.exists())

blah.delete();

FileOutputStream fos = new FileOutputStream(blah);

BufferedWriter awriter

= new BufferedWriter(new OutputStreamWriter(fos, "8859_1"));

for(int i=0; i<size; i++) {

awriter.write("e");

}

awriter.flush();

awriter.close();

}

/**

static class EchoServer implements Runnable {

private SocketChannel sc;

public EchoServer(SocketChannel sc) {

this.sc = sc;

}

public void run() {

ByteBuffer bb = ByteBuffer.allocateDirect(1024);

try {

for (;;) {

int n = sc.read(bb);

if (n < 0)

break;

bb.flip();

while (bb.remaining() > 0) {

sc.write(bb);

}

bb.clear();

}

} catch (IOException x) {

x.printStackTrace();

} finally {

try {

sc.close();

} catch (IOException ignore) { }

}

}

}

}