/**
* Send a plugin JAR file to the client
*
* @param data the plugin name
*/
private void sendPlugin(ObjectConnection oc, String data) {
DataInputStream dis = null;
try {
dis = new DataInputStream(new FileInputStream(APPLICATIONS_DIRECTORY + data + ".jar"));
byte[] buf = new byte[dis.available()];
dis.readFully(buf);
oc.write(buf);
} catch (Exception e) {
Logging.getLogger().warning("Unable to send the file: " + data + ".jar");
} finally {
if (dis != null) {
try {
dis.close();
} catch (IOException ioe) {
}
}
}
}
@Override
public void run() {
try {
// Create data input and output streams
DataInputStream inputFromClient = new DataInputStream(socket.getInputStream());
DataOutputStream outputToClient = new DataOutputStream(socket.getOutputStream());
// Set up a byte buffer to capture the file from the client
byte[] buffer = new byte[BUFFER_SIZE];
OutputStream outputStream = null;
String fileName = "";
boolean createFile = true;
int bytesReceived = 0;
long totalBytesReceived = 0;
long fileSize = 0;
// Continuously serve the client
while (true) {
bytesReceived = inputFromClient.read(buffer);
if (bytesReceived
> 0) { // Get the file transmission header from the initial client packet
String transmitHeader = new String(buffer, 0, bytesReceived);
// transmitHeader = transmitHeader.substring(0, bytesReceived).trim().;
String[] header = transmitHeader.split(HEADER_DEL);
fileSize = Long.parseLong(header[0]);
fileName = header[1];
// Send receipt acknowledgment back to the client. Just send back the number of bytes
// received.
outputToClient.writeInt(bytesReceived);
// Reinitialize buffer values
buffer = new byte[BUFFER_SIZE];
bytesReceived = 0;
}
// Wait for client to send bytes
while ((bytesReceived = inputFromClient.read(buffer)) != -1) {
if (inputFromClient.available() > 0) {
if (createFile) { // Get a unique name for the file to be received
fileName = textFolder.getText() + fileName; // getUniqueFileName();
outputStream = createFile(fileName);
createFile = false;
textArea.append("Receiving file from client.\n");
}
// Write bytes to file
outputStream.write(buffer, 0, bytesReceived);
} else { // We get here if no more data is available, but some bytes were already
// received
// If bytes were received and the file wasn't already created,
// it means that the file was smaller than our buffer size.
// Create the file...
if (bytesReceived > 0
&& createFile) { // Get a unique name for the file to be received
fileName = textFolder.getText() + fileName; // getUniqueFileName();
outputStream = createFile(fileName);
createFile = false;
textArea.append("Receiving file from client.\n");
}
if (outputStream != null) {
if (bytesReceived > 0) { // Write remaining bytes to file, if any
outputStream.write(buffer, 0, bytesReceived);
}
outputStream.flush();
outputStream.close();
textArea.append("Received file successfully. Saved as " + fileName + "\n");
// Return success to client.
outputToClient.writeInt(0);
}
// Reset creation flag
createFile = true;
break;
}
// Reinitialize buffer values
buffer = new byte[BUFFER_SIZE];
bytesReceived = 0;
}
}
} catch (IOException e) {
System.err.println(e);
}
}