mirror of
https://github.com/hierynomus/sshj.git
synced 2025-12-08 00:00:54 +03:00
Reformatted
This commit is contained in:
@@ -1,12 +1,12 @@
|
|||||||
/**
|
/**
|
||||||
* Copyright 2009 sshj contributors
|
* Copyright 2009 sshj contributors
|
||||||
*
|
* <p/>
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
* you may not use this file except in compliance with the License.
|
* you may not use this file except in compliance with the License.
|
||||||
* You may obtain a copy of the License at
|
* You may obtain a copy of the License at
|
||||||
*
|
* <p/>
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
*
|
* <p/>
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
@@ -34,37 +34,31 @@ public class RemoteFile
|
|||||||
super(requester, path, handle);
|
super(requester, path, handle);
|
||||||
}
|
}
|
||||||
|
|
||||||
public FileAttributes fetchAttributes()
|
public FileAttributes fetchAttributes() throws IOException {
|
||||||
throws IOException {
|
|
||||||
return requester.request(newRequest(PacketType.FSTAT))
|
return requester.request(newRequest(PacketType.FSTAT))
|
||||||
.retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS)
|
.retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS)
|
||||||
.ensurePacketTypeIs(PacketType.ATTRS)
|
.ensurePacketTypeIs(PacketType.ATTRS)
|
||||||
.readFileAttributes();
|
.readFileAttributes();
|
||||||
}
|
}
|
||||||
|
|
||||||
public long length()
|
public long length() throws IOException {
|
||||||
throws IOException {
|
|
||||||
return fetchAttributes().getSize();
|
return fetchAttributes().getSize();
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setLength(long len)
|
public void setLength(long len) throws IOException {
|
||||||
throws IOException {
|
|
||||||
setAttributes(new FileAttributes.Builder().withSize(len).build());
|
setAttributes(new FileAttributes.Builder().withSize(len).build());
|
||||||
}
|
}
|
||||||
|
|
||||||
public int read(long fileOffset, byte[] to, int offset, int len)
|
public int read(long fileOffset, byte[] to, int offset, int len) throws IOException {
|
||||||
throws IOException {
|
|
||||||
final Response res = asyncRead(fileOffset, len).retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS);
|
final Response res = asyncRead(fileOffset, len).retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS);
|
||||||
return checkReadResponse(res, to, offset);
|
return checkReadResponse(res, to, offset);
|
||||||
}
|
}
|
||||||
|
|
||||||
protected Promise<Response, SFTPException> asyncRead(long fileOffset, int len)
|
protected Promise<Response, SFTPException> asyncRead(long fileOffset, int len) throws IOException {
|
||||||
throws IOException {
|
|
||||||
return requester.request(newRequest(PacketType.READ).putUInt64(fileOffset).putUInt32(len));
|
return requester.request(newRequest(PacketType.READ).putUInt64(fileOffset).putUInt32(len));
|
||||||
}
|
}
|
||||||
|
|
||||||
protected int checkReadResponse(Response res, byte[] to, int offset)
|
protected int checkReadResponse(Response res, byte[] to, int offset) throws Buffer.BufferException, SFTPException {
|
||||||
throws Buffer.BufferException, SFTPException {
|
|
||||||
switch (res.getType()) {
|
switch (res.getType()) {
|
||||||
case DATA:
|
case DATA:
|
||||||
int recvLen = res.readUInt32AsInt();
|
int recvLen = res.readUInt32AsInt();
|
||||||
@@ -80,8 +74,7 @@ public class RemoteFile
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void write(long fileOffset, byte[] data, int off, int len)
|
public void write(long fileOffset, byte[] data, int off, int len) throws IOException {
|
||||||
throws IOException {
|
|
||||||
checkWriteResponse(asyncWrite(fileOffset, data, off, len));
|
checkWriteResponse(asyncWrite(fileOffset, data, off, len));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -95,13 +88,11 @@ public class RemoteFile
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void checkWriteResponse(Promise<Response, SFTPException> responsePromise)
|
private void checkWriteResponse(Promise<Response, SFTPException> responsePromise) throws SFTPException {
|
||||||
throws SFTPException {
|
|
||||||
responsePromise.retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS).ensureStatusPacketIsOK();
|
responsePromise.retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS).ensureStatusPacketIsOK();
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setAttributes(FileAttributes attrs)
|
public void setAttributes(FileAttributes attrs) throws IOException {
|
||||||
throws IOException {
|
|
||||||
requester.request(newRequest(PacketType.FSETSTAT).putFileAttributes(attrs))
|
requester.request(newRequest(PacketType.FSETSTAT).putFileAttributes(attrs))
|
||||||
.retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS).ensureStatusPacketIsOK();
|
.retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS).ensureStatusPacketIsOK();
|
||||||
}
|
}
|
||||||
@@ -141,15 +132,13 @@ public class RemoteFile
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void write(int w)
|
public void write(int w) throws IOException {
|
||||||
throws IOException {
|
|
||||||
b[0] = (byte) w;
|
b[0] = (byte) w;
|
||||||
write(b, 0, 1);
|
write(b, 0, 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void write(byte[] buf, int off, int len)
|
public void write(byte[] buf, int off, int len) throws IOException {
|
||||||
throws IOException {
|
|
||||||
if (unconfirmedWrites.size() > maxUnconfirmedWrites) {
|
if (unconfirmedWrites.size() > maxUnconfirmedWrites) {
|
||||||
checkWriteResponse(unconfirmedWrites.remove());
|
checkWriteResponse(unconfirmedWrites.remove());
|
||||||
}
|
}
|
||||||
@@ -158,23 +147,20 @@ public class RemoteFile
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void flush()
|
public void flush() throws IOException {
|
||||||
throws IOException {
|
|
||||||
while (!unconfirmedWrites.isEmpty()) {
|
while (!unconfirmedWrites.isEmpty()) {
|
||||||
checkWriteResponse(unconfirmedWrites.remove());
|
checkWriteResponse(unconfirmedWrites.remove());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void close()
|
public void close() throws IOException {
|
||||||
throws IOException {
|
|
||||||
flush();
|
flush();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public class RemoteFileInputStream
|
public class RemoteFileInputStream extends InputStream {
|
||||||
extends InputStream {
|
|
||||||
|
|
||||||
private final byte[] b = new byte[1];
|
private final byte[] b = new byte[1];
|
||||||
|
|
||||||
@@ -202,32 +188,30 @@ public class RemoteFile
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void reset()
|
public void reset() throws IOException {
|
||||||
throws IOException {
|
|
||||||
fileOffset = markPos;
|
fileOffset = markPos;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public long skip(long n)
|
public long skip(long n) throws IOException {
|
||||||
throws IOException {
|
|
||||||
return (this.fileOffset = Math.min(fileOffset + n, length()));
|
return (this.fileOffset = Math.min(fileOffset + n, length()));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public int read()
|
public int read() throws IOException {
|
||||||
throws IOException {
|
|
||||||
return read(b, 0, 1) == -1 ? -1 : b[0] & 0xff;
|
return read(b, 0, 1) == -1 ? -1 : b[0] & 0xff;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public int read(byte[] into, int off, int len)
|
public int read(byte[] into, int off, int len) throws IOException {
|
||||||
throws IOException {
|
|
||||||
int read = RemoteFile.this.read(fileOffset, into, off, len);
|
int read = RemoteFile.this.read(fileOffset, into, off, len);
|
||||||
if (read != -1) {
|
if (read != -1) {
|
||||||
fileOffset += read;
|
fileOffset += read;
|
||||||
if (markPos != 0 && read > readLimit) // Invalidate mark position
|
if (markPos != 0 && read > readLimit) {
|
||||||
|
// Invalidate mark position
|
||||||
markPos = 0;
|
markPos = 0;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return read;
|
return read;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -260,33 +244,33 @@ public class RemoteFile
|
|||||||
this.requestOffset = this.responseOffset = fileOffset;
|
this.requestOffset = this.responseOffset = fileOffset;
|
||||||
}
|
}
|
||||||
|
|
||||||
private ByteArrayInputStream pending = new ByteArrayInputStream( new byte[0] );
|
private ByteArrayInputStream pending = new ByteArrayInputStream(new byte[0]);
|
||||||
|
|
||||||
private boolean retrieveUnconfirmedRead(boolean blocking) throws IOException {
|
private boolean retrieveUnconfirmedRead(boolean blocking) throws IOException {
|
||||||
|
if (unconfirmedReads.size() <= 0) {
|
||||||
if (unconfirmedReads.size() <= 0)
|
|
||||||
return false;
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
if (!blocking && !unconfirmedReads.peek().isDelivered())
|
if (!blocking && !unconfirmedReads.peek().isDelivered()) {
|
||||||
return false;
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
unconfirmedReadOffsets.remove();
|
unconfirmedReadOffsets.remove();
|
||||||
final Response res = unconfirmedReads.remove().retrieve( requester.getTimeoutMs(), TimeUnit.MILLISECONDS );
|
final Response res = unconfirmedReads.remove().retrieve(requester.getTimeoutMs(), TimeUnit.MILLISECONDS);
|
||||||
switch (res.getType()) {
|
switch (res.getType()) {
|
||||||
case DATA:
|
case DATA:
|
||||||
int recvLen = res.readUInt32AsInt();
|
int recvLen = res.readUInt32AsInt();
|
||||||
|
|
||||||
responseOffset += recvLen;
|
responseOffset += recvLen;
|
||||||
pending = new ByteArrayInputStream( res.array(), res.rpos(), recvLen );
|
pending = new ByteArrayInputStream(res.array(), res.rpos(), recvLen);
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case STATUS:
|
case STATUS:
|
||||||
res.ensureStatusIs( Response.StatusCode.EOF );
|
res.ensureStatusIs(Response.StatusCode.EOF);
|
||||||
eof = true;
|
eof = true;
|
||||||
break;
|
break;
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new SFTPException( "Unexpected packet: " + res.getType() );
|
throw new SFTPException("Unexpected packet: " + res.getType());
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@@ -307,9 +291,9 @@ public class RemoteFile
|
|||||||
|
|
||||||
while (unconfirmedReads.size() <= maxUnconfirmedReads) {
|
while (unconfirmedReads.size() <= maxUnconfirmedReads) {
|
||||||
// Send read requests as long as there is no EOF and we have not reached the maximum parallelism
|
// Send read requests as long as there is no EOF and we have not reached the maximum parallelism
|
||||||
int reqLen = Math.max( 1024, len ); // don't be shy!
|
int reqLen = Math.max(1024, len); // don't be shy!
|
||||||
unconfirmedReads.add( RemoteFile.this.asyncRead( requestOffset, reqLen ) );
|
unconfirmedReads.add(RemoteFile.this.asyncRead(requestOffset, reqLen));
|
||||||
unconfirmedReadOffsets.add( requestOffset );
|
unconfirmedReadOffsets.add(requestOffset);
|
||||||
requestOffset += reqLen;
|
requestOffset += reqLen;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -324,36 +308,38 @@ public class RemoteFile
|
|||||||
assert (nextOffset - responseOffset) <= Integer.MAX_VALUE;
|
assert (nextOffset - responseOffset) <= Integer.MAX_VALUE;
|
||||||
|
|
||||||
byte[] buf = new byte[(int) (nextOffset - responseOffset)];
|
byte[] buf = new byte[(int) (nextOffset - responseOffset)];
|
||||||
int recvLen = RemoteFile.this.read( responseOffset, buf, 0, buf.length );
|
int recvLen = RemoteFile.this.read(responseOffset, buf, 0, buf.length);
|
||||||
|
|
||||||
if (recvLen < 0) {
|
if (recvLen < 0) {
|
||||||
eof = true;
|
eof = true;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (0 == recvLen) // avoid infinite loops
|
if (0 == recvLen) {
|
||||||
throw new SFTPException( "Unexpected response size (0), bailing out" );
|
// avoid infinite loops
|
||||||
|
throw new SFTPException("Unexpected response size (0), bailing out");
|
||||||
|
}
|
||||||
|
|
||||||
responseOffset += recvLen;
|
responseOffset += recvLen;
|
||||||
pending = new ByteArrayInputStream( buf, 0, recvLen );
|
pending = new ByteArrayInputStream(buf, 0, recvLen);
|
||||||
}
|
} else if (!retrieveUnconfirmedRead(true /*blocking*/)) {
|
||||||
else
|
|
||||||
if (!retrieveUnconfirmedRead( true /*blocking*/ )) {
|
|
||||||
|
|
||||||
// this may happen if we change prefetch strategy
|
// this may happen if we change prefetch strategy
|
||||||
// currently, we should never get here...
|
// currently, we should never get here...
|
||||||
|
|
||||||
throw new IllegalStateException( "Could not retrieve data for pending read request" );
|
throw new IllegalStateException("Could not retrieve data for pending read request");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return pending.read( into, off, len );
|
return pending.read(into, off, len);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public int available() throws IOException {
|
public int available() throws IOException {
|
||||||
while (!eof && (pending.available() <= 0) && retrieveUnconfirmedRead( false /*blocking*/ ))
|
boolean lastRead = true;
|
||||||
/*loop*/;
|
while (!eof && (pending.available() <= 0) && lastRead) {
|
||||||
|
lastRead = retrieveUnconfirmedRead(false /*blocking*/);
|
||||||
|
}
|
||||||
return pending.available();
|
return pending.available();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user