50 lines
1.0 KiB
Java
50 lines
1.0 KiB
Java
package com.ryanmichela.sshd;
|
|
|
|
import org.apache.sshd.common.SshException;
|
|
|
|
import java.io.IOException;
|
|
import java.io.OutputStream;
|
|
|
|
/**
|
|
* Copyright 2013 Ryan Michela
|
|
*/
|
|
public class FlushyOutputStream extends OutputStream {
|
|
|
|
private OutputStream base;
|
|
private boolean isClosed = false;
|
|
|
|
public FlushyOutputStream(OutputStream base) {
|
|
this.base = base;
|
|
}
|
|
|
|
@Override
|
|
public void write(int b) throws IOException {
|
|
if (isClosed) return;
|
|
base.write(b);
|
|
base.flush();
|
|
}
|
|
|
|
@Override
|
|
public void write(byte[] b) throws IOException {
|
|
if (isClosed) return;
|
|
base.write(b);
|
|
base.flush();
|
|
}
|
|
|
|
@Override
|
|
public void write(byte[] b, int off, int len) throws IOException {
|
|
if (isClosed) return;
|
|
try {
|
|
base.write(b, off, len);
|
|
base.flush();
|
|
} catch (SshException e) {
|
|
if (!e.getMessage().contains("channel already closed")) throw e;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void close() {
|
|
isClosed = true;
|
|
}
|
|
}
|