Skip to content

Commit

Permalink
donaldh first hack on sockets for JVM.
Browse files Browse the repository at this point in the history
  • Loading branch information
@donaldh
donaldh committed Aug 30, 2013
1 parent cba09d3 commit 83ea6f2
Show file tree
Hide file tree
Showing 3 changed files with 205 additions and 167 deletions.
178 changes: 11 additions & 167 deletions src/vm/jvm/runtime/org/perl6/nqp/io/FileHandle.java
View file
Expand Up @@ -2,70 +2,55 @@

import java.io.File;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.channels.FileChannel;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CharsetEncoder;
import java.nio.file.Path;
import java.nio.file.StandardOpenOption;
import java.util.ArrayList;

import org.perl6.nqp.runtime.ExceptionHandling;
import org.perl6.nqp.runtime.ThreadContext;

public class FileHandle implements IIOClosable, IIOSeekable, IIOEncodable, IIOSyncReadable, IIOSyncWritable {
private FileChannel chan;
private CharsetEncoder enc;
private CharsetDecoder dec;
private boolean eof = false;
private ByteBuffer readBuffer;
public class FileHandle extends SyncHandle implements IIOSeekable {

FileChannel fc;

public FileHandle(ThreadContext tc, String filename, String mode) {
try {
Path p = new File(filename).toPath();
if (mode.equals("r")) {
chan = FileChannel.open(p, StandardOpenOption.READ);
fc = FileChannel.open(p, StandardOpenOption.READ);
}
else if (mode.equals("w")) {
chan = FileChannel.open(p, StandardOpenOption.WRITE,
fc = FileChannel.open(p, StandardOpenOption.WRITE,
StandardOpenOption.CREATE,
StandardOpenOption.TRUNCATE_EXISTING);
}
else if (mode.equals("wa")) {
chan = FileChannel.open(p, StandardOpenOption.WRITE,
fc = FileChannel.open(p, StandardOpenOption.WRITE,
StandardOpenOption.CREATE,
StandardOpenOption.APPEND);
}
else {
ExceptionHandling.dieInternal(tc, "Unhandled file open mode '" + mode + "'");
}
chan = fc;
setEncoding(tc, Charset.forName("UTF-8"));
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public void close(ThreadContext tc) {
try {
chan.close();
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public void seek(ThreadContext tc, long offset, long whence) {
try {
switch ((int)whence) {
case 0:
chan.position(offset);
fc.position(offset);
break;
case 1:
chan.position(chan.position() + offset);
fc.position(fc.position() + offset);
break;
case 2:
chan.position(chan.size());
fc.position(fc.size());
break;
default:
throw ExceptionHandling.dieInternal(tc, "Invalid seek mode");
Expand All @@ -77,151 +62,10 @@ public void seek(ThreadContext tc, long offset, long whence) {

public long tell(ThreadContext tc) {
try {
long position = chan.position();
long position = fc.position();
return readBuffer != null ? position - readBuffer.remaining() : position;
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public void setEncoding(ThreadContext tc, Charset cs) {
enc = cs.newEncoder();
dec = cs.newDecoder();
}

public synchronized String slurp(ThreadContext tc) {
try {
// Read in file.
ArrayList<ByteBuffer> buffers = new ArrayList<ByteBuffer>();
ByteBuffer curBuffer = ByteBuffer.allocate(32768);
int total = 0;
int read;
if (readBuffer != null) {
total = readBuffer.limit() - readBuffer.position();
buffers.add(ByteBuffer.wrap(readBuffer.array(), readBuffer.position(), total));
readBuffer = null;
}
while ((read = chan.read(curBuffer)) != -1) {
curBuffer.flip();
buffers.add(curBuffer);
curBuffer = ByteBuffer.allocate(32768);
total += read;
}
eof = true;

return decodeBuffers(buffers, total);
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public synchronized String readline(ThreadContext tc) {
try {
boolean foundLine = false;
ArrayList<ByteBuffer> lineChunks = new ArrayList<ByteBuffer>();
int total = 0;

while (!foundLine) {
/* Ensure we have a buffer available. */
if (readBuffer == null) {
readBuffer = ByteBuffer.allocate(32768);
if (chan.read(readBuffer) == -1) {
/* End of file, so what we have is fine. */
eof = true;
foundLine = true;
readBuffer.flip();
break;
}
readBuffer.flip();
}

/* Look for a line end. */
int start = readBuffer.position();
int end = start;
while (!foundLine && end < readBuffer.limit()) {
if (readBuffer.get(end) == '\n')
foundLine = true;
end++;
}

/* Copy what we found into the results. */
byte[] lineBytes = new byte[end - start];
readBuffer.get(lineBytes);
lineChunks.add(ByteBuffer.wrap(lineBytes));
total += lineBytes.length;

/* If we didn't find a line, will cross chunk boundary. */
if (!foundLine)
readBuffer = null;
}

if (lineChunks.size() == 1)
return dec.decode(lineChunks.get(0)).toString();
else
return decodeBuffers(lineChunks, total);
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

private String decodeBuffers(ArrayList<ByteBuffer> buffers, int total) throws IOException {
// Copy to a single buffer and decode (could be smarter, but need
// to be wary as UTF-8 chars may span a buffer boundary).
ByteBuffer allBytes = ByteBuffer.allocate(total);
for (ByteBuffer bb : buffers) {
int amount = total < bb.limit() ? total : bb.limit();
allBytes.put(bb.array(), 0, amount);
total -= amount;
}
allBytes.rewind();
return dec.decode(allBytes).toString();
}

public boolean eof(ThreadContext tc) {
return eof;
}

public byte[] read(ThreadContext tc, int bytes) {
try {
ByteBuffer buffer = ByteBuffer.allocate(bytes);
chan.read(buffer);
buffer.flip();
byte[] res = new byte[buffer.limit()];
buffer.get(res);
return res;
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public void write(ThreadContext tc, byte[] array) {
ByteBuffer buffer = ByteBuffer.wrap(array);
write(tc, buffer);
}

protected void write(ThreadContext tc, ByteBuffer buffer) {
try {
int toWrite = buffer.limit();
int written = 0;
while (written < toWrite) {
written += chan.write(buffer);
}
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public void print(ThreadContext tc, String s) {
try {
ByteBuffer buffer = enc.encode(CharBuffer.wrap(s));
write(tc, buffer);
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}

public void say(ThreadContext tc, String s) {
print(tc, s);
print(tc, System.lineSeparator());
}
}
22 changes: 22 additions & 0 deletions src/vm/jvm/runtime/org/perl6/nqp/io/SocketHandle.java
View file
@@ -0,0 +1,22 @@
package org.perl6.nqp.io;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;

import org.perl6.nqp.runtime.ExceptionHandling;
import org.perl6.nqp.runtime.ThreadContext;

public class SocketHandle extends SyncHandle {

public SocketHandle(ThreadContext tc, String host, int port) {
try {
InetSocketAddress addr = InetSocketAddress.createUnresolved(host, port);
chan = SocketChannel.open(addr);
setEncoding(tc, Charset.forName("UTF-8"));
} catch (IOException e) {
throw ExceptionHandling.dieInternal(tc, e);
}
}
}

0 comments on commit 83ea6f2

Please sign in to comment.