public CmdResult getFile(String url, boolean calcHash) throws HoneycombTestException {
Log.DEBUG("getFile: " + url);
CmdResult cr = new CmdResult();
byte[] readBuf = new byte[BUFSIZE];
try {
MessageDigest sha = null; // for hash
if (calcHash) sha = MessageDigest.getInstance(HC_HASH);
cr.filename = url;
long t1 = System.currentTimeMillis();
HTTPResponse rsp = conn.Get(url);
if (rsp.getStatusCode() >= 300) throw makeTestException("getFile", url, rsp);
cr.filesize = 0;
InputStream is = rsp.getInputStream();
while (true) {
int ct = is.read(readBuf);
if (ct == -1) break;
cr.filesize += ct;
if (calcHash) sha.update(readBuf, 0, ct);
}
cr.time = System.currentTimeMillis() - t1;
cr.pass = true;
if (calcHash) cr.datasha1 = HCUtil.convertHashBytesToString(sha.digest());
} catch (HoneycombTestException e) {
throw e;
} catch (Exception e) {
e.printStackTrace();
cr.pass = false;
cr.addException(e);
}
return cr;
}
/** The real "store file" method. */
private CmdResult putFile(
String url,
// Only one of these two should be given:
int nBytes,
ReadableByteChannel contents,
boolean calcHash,
NVPair[] extraHeaders)
throws HoneycombTestException, IOException, ModuleException {
NVPair[] headers = null;
MessageDigest sha = null;
if (calcHash)
try {
sha = MessageDigest.getInstance(HC_HASH);
} catch (NoSuchAlgorithmException e) {
throw new RuntimeException(e);
}
CmdResult cr = new CmdResult();
cr.filename = url;
byte[] byte_buffer = new byte[BUFSIZE];
ByteBuffer buff = ByteBuffer.wrap(byte_buffer);
int toWrite, written = 0;
long startTime = System.currentTimeMillis();
if (contents != null)
// If we're not writing from our buffer, don't set Content-type
headers = extraHeaders;
else if (extraHeaders == null) headers = putHeaders;
else {
// Need to combine the two arrays
NVPair[] newHeaders = new NVPair[putHeaders.length + extraHeaders.length];
int i = 0;
for (int j = 0; j < putHeaders.length; j++) newHeaders[i++] = putHeaders[j];
for (int j = 0; j < extraHeaders.length; j++) newHeaders[i++] = extraHeaders[j];
headers = newHeaders;
}
HttpOutputStream out = new HttpOutputStream();
HTTPResponse response = conn.Put(url, out, headers);
// Write into the HTTP stream
for (; ; ) {
if (contents != null) {
// Read from the "contents" channel and write it out
if ((toWrite = contents.read(buff)) < 0) break;
byte[] data = buff.array();
if (sha != null) sha.update(data);
out.write(data, 0, toWrite);
buff.clear();
} else {
// Write from the buffer writeBuf
if (written >= nBytes) break;
toWrite = nBytes - written;
if (toWrite > BUFSIZE) {
out.write(writeBuf);
written += BUFSIZE;
if (sha != null) sha.update(writeBuf);
} else {
out.write(writeBuf, 0, toWrite);
if (sha != null) sha.update(writeBuf, 0, toWrite);
break;
}
}
}
out.close();
if (response.getStatusCode() >= 300) throw makeTestException("putFile", url, response);
cr.mdoid = response.getHeader("ETag");
cr.time = System.currentTimeMillis() - startTime;
cr.pass = true;
if (sha != null) cr.datasha1 = HCUtil.convertHashBytesToString(sha.digest());
return cr;
}
