public String getFileBinaryBase64(String fileName) throws APIException {
if ((new File(fileName)).exists()) {
FileInputStream stream = null;
try {
stream = new FileInputStream(new File(fileName));
FileChannel fc = stream.getChannel();
MappedByteBuffer bb = fc.map(FileChannel.MapMode.READ_ONLY, 0, fc.size());
/* Instead of using default, pass in a decoder. */
byte[] b = new byte[bb.remaining()];
bb.get(b);
return Base64.encodeBytes(b);
} catch (Exception e) {
throw new APIException(fileName + " could not have its files extracte!");
} finally {
try {
stream.close();
} catch (Exception e) {
throw new APIException(fileName + " could not be closed!");
}
}
} else {
throw new APIException(fileName + " doesn't exist!");
}
}
public boolean setFileBinaryBase64(String fileName, String base64) throws APIException {
FileOutputStream stream = null;
try {
File f = new File(fileName);
f.createNewFile();
stream = new FileOutputStream(f);
stream.write(Base64.decode(base64));
try {
stream.close();
} catch (IOException e) {
throw new APIException(fileName + " could not be closed!");
}
} catch (Exception e) {
throw new APIException(fileName + " could not have its files extracte!");
} finally {
try {
stream.flush();
stream.close();
} catch (Exception e) {
throw new APIException(fileName + " could not be closed!");
}
}
return true;
}
private String generateFinalKey(String in) {
String seckey = in.trim();
String acc = seckey + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
MessageDigest sh1;
try {
sh1 = MessageDigest.getInstance("SHA1");
} catch (NoSuchAlgorithmException e) {
throw new RuntimeException(e);
}
return Base64.encodeBytes(sh1.digest(acc.getBytes()));
}
@Override
public ClientHandshakeBuilder postProcessHandshakeRequestAsClient(
ClientHandshakeBuilder request) {
request.put("Upgrade", "websocket");
request.put("Connection", "Upgrade"); // to respond to a Connection keep alives
request.put("Sec-WebSocket-Version", "8");
byte[] random = new byte[16];
reuseableRandom.nextBytes(random);
request.put("Sec-WebSocket-Key", Base64.encodeBytes(random));
return request;
}
// FIXME - normalize further
public Response serveFile(
String uri,
Map<String, String> header,
File homeDir,
boolean allowDirectoryListing,
boolean isCustomFile) {
String username;
String password;
//
if (webgui.requiresSecurity()) { // && !authorized) {
try {
if (header.containsKey("authorization")) {
String up = header.get("authorization");
int pos = up.indexOf("Basic ");
if (pos != -1) {
up = up.substring(pos + 6);
}
// FIXME - depends on commons !!!!
String usernameAndPassword = new String(Base64.decode(up)); // SHWEET
// CURRENTLY
// USING
// WEBSOCKETS
// VERSION
// !!!
// :P
username = usernameAndPassword.substring(0, usernameAndPassword.lastIndexOf(":"));
password = usernameAndPassword.substring(usernameAndPassword.lastIndexOf(":") + 1);
String token = BasicSecurity.authenticate(username, password);
if (token != null) {
// authorized = true;
} else {
throw new Exception(String.format("no token for user %s", username));
}
log.info(usernameAndPassword);
} else {
throw new Exception("no authorization in header");
}
} catch (Exception e) {
Logging.logException(e);
Response r = new Response(e.getMessage());
r.setStatus(Response.Status.FORBIDDEN);
r.addHeader("WWW-Authenticate", "Basic realm=\"MyRobotLab\"");
return r;
}
}
// Remove URL arguments
uri = uri.trim().replace(File.separatorChar, '/');
if (uri.indexOf('?') >= 0) {
uri = uri.substring(0, uri.indexOf('?'));
}
// Prohibit getting out of current directory
if (uri.startsWith("..") || uri.endsWith("..") || uri.indexOf("../") >= 0) {
return new Response(
Response.Status.FORBIDDEN,
WSServer.MIME_PLAINTEXT,
"FORBIDDEN: Won't serve ../ for security reasons.");
}
File f;
if (!isCustomFile) {
// ---- begin resource ----------------
// TODO ? hard coded /resource - are they going to pull up
// WebGUI.class how useful is that ??
String resoucePath = String.format("/resource%s", uri);
InputStream fis = null;
if (resoucePath.endsWith("/")) {
// ends with a slash .. might be a directory - try index.html
String documentIndex = String.format("%sindex.html", resoucePath);
fis = FileIO.class.getResourceAsStream(documentIndex);
if (fis == null) { // couldn't find document index try directory
// listing, if allowed
fis = FileIO.class.getResourceAsStream(resoucePath);
} else {
uri = documentIndex;
}
} else {
fis = FileIO.class.getResourceAsStream(resoucePath);
}
if (fis == null)
return new Response(
Response.Status.NOT_FOUND, WSServer.MIME_PLAINTEXT, "Error 404, file not found.");
try {
// Get MIME type from file name extension, if possible
String mime = null;
int dot = uri.lastIndexOf('.');
if (dot >= 0) {
mime = (String) NanoHTTPD.theMimeTypes.get(uri.substring(dot + 1).toLowerCase());
}
if (mime == null) {
// mime = NanoHTTPD.MIME_DEFAULT_BINARY;
mime = WSServer.MIME_PLAINTEXT;
}
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
int nRead;
byte[] data = new byte[16384];
while ((nRead = fis.read(data, 0, data.length)) != -1) {
buffer.write(data, 0, nRead);
}
fis.close();
buffer.flush();
byte[] content = null;
// if ("/resource/WebGUI/myrobotlab.html".equals(uri)) {
if (uri.endsWith(".mrl")) {
content = filter(new String(buffer.toByteArray()), header);
mime = NanoHTTPD.MIME_HTML;
} else {
content = buffer.toByteArray();
}
Response r = new Response(Response.Status.OK, mime, new ByteArrayInputStream(content));
r.addHeader("Content-length", "" + content.length);
return r;
} catch (IOException ioe) {
return new Response(
Response.Status.FORBIDDEN, WSServer.MIME_PLAINTEXT, "FORBIDDEN: Reading file failed.");
}
// ---- end resource -------------------
} else {
// ---- begin file -----------------------
f = new File(homeDir, uri);
if (!f.exists())
return new Response(
Response.Status.NOT_FOUND, WSServer.MIME_PLAINTEXT, "Error 404, file not found.");
// List the directory, if necessary
if (f.isDirectory()) {
// Browsers get confused without '/' after the
// directory, send a redirect.
if (!uri.endsWith("/")) {
uri += "/";
Response r =
new Response(
Response.Status.REDIRECT,
NanoHTTPD.MIME_HTML,
"<html><body>Redirected: <a href=\"" + uri + "\">" + uri + "</a></body></html>");
r.addHeader("Location", uri);
return r;
}
// First try index.html and index.htm
if (new File(f, "index.html").exists()) f = new File(homeDir, uri + "/index.html");
else if (new File(f, "index.htm").exists()) f = new File(homeDir, uri + "/index.htm");
// No index file, list the directory
else if (allowDirectoryListing) {
String[] files = f.list();
String msg = "<html><body><h1>Directory " + uri + "</h1><br/>";
if (uri.length() > 1) {
String u = uri.substring(0, uri.length() - 1);
int slash = u.lastIndexOf('/');
if (slash >= 0 && slash < u.length())
msg += "<b><a href=\"" + uri.substring(0, slash + 1) + "\">..</a></b><br/>";
}
for (int i = 0; i < files.length; ++i) {
File curFile = new File(f, files[i]);
boolean dir = curFile.isDirectory();
if (dir) {
msg += "<b>";
files[i] += "/";
}
msg += "<a href=\"" + NanoHTTPD.encodeUri(uri + files[i]) + "\">" + files[i] + "</a>";
// Show file size
if (curFile.isFile()) {
long len = curFile.length();
msg += " <font size=2>(";
if (len < 1024) msg += curFile.length() + " bytes";
else if (len < 1024 * 1024)
msg += curFile.length() / 1024 + "." + (curFile.length() % 1024 / 10 % 100) + " KB";
else
msg +=
curFile.length() / (1024 * 1024)
+ "."
+ curFile.length() % (1024 * 1024) / 10 % 100
+ " MB";
msg += ")</font>";
}
msg += "<br/>";
if (dir) msg += "</b>";
}
return new Response(Response.Status.OK, NanoHTTPD.MIME_HTML, msg);
} else {
return new Response(
Response.Status.FORBIDDEN,
WSServer.MIME_PLAINTEXT,
"FORBIDDEN: No directory listing.");
}
}
// ----- end file ------------------------
}
// }
try {
// Get MIME type from file name extension, if possible
String mime = null;
// int dot = f.getCanonicalPath().lastIndexOf('.');
int dot = uri.lastIndexOf('.');
if (dot >= 0) {
mime = (String) NanoHTTPD.theMimeTypes.get(uri.substring(dot + 1).toLowerCase());
}
if (mime == null) {
mime = WSServer.MIME_PLAINTEXT;
}
// Support (simple) skipping:
long startFrom = 0;
String range = header.get("Range");
if (range != null) {
if (range.startsWith("bytes=")) {
range = range.substring("bytes=".length());
int minus = range.indexOf('-');
if (minus > 0) range = range.substring(0, minus);
try {
startFrom = Long.parseLong(range);
} catch (NumberFormatException nfe) {
}
}
}
InputStream fis;
if (!isCustomFile) {
fis = FileIO.class.getResourceAsStream(uri);
if (fis == null) {
return new Response(
Response.Status.NOT_FOUND, WSServer.MIME_PLAINTEXT, "Error 404, file not found.");
}
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
int nRead;
byte[] data = new byte[16384];
while ((nRead = fis.read(data, 0, data.length)) != -1) {
buffer.write(data, 0, nRead);
}
fis.close();
buffer.flush();
byte[] content = null;
// FIXME - this is not normalized - because the code around it
// is
// not normalized
if (uri.endsWith(".mrl")) {
content = filter(new String(buffer.toByteArray()), header);
mime = NanoHTTPD.MIME_HTML;
} else {
content = buffer.toByteArray();
}
Response r = new Response(Response.Status.OK, mime, new ByteArrayInputStream(content));
r.addHeader("Content-length", "" + content.length);
return r;
} else {
fis = new FileInputStream(f);
// fis.skip(startFrom); // TODO support skip in the future
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
int nRead;
byte[] data = new byte[16384];
while ((nRead = fis.read(data, 0, data.length)) != -1) {
buffer.write(data, 0, nRead);
}
fis.close();
buffer.flush();
byte[] content = null;
// FIXME - this is not normalized - because the code around it
// is
// not normalized
if (uri.endsWith(".mrl")) {
content = filter(new String(buffer.toByteArray()), header);
mime = NanoHTTPD.MIME_HTML;
} else {
content = buffer.toByteArray();
}
Response r = new Response(Response.Status.OK, mime, new ByteArrayInputStream(content));
r.addHeader("Content-length", "" + content.length);
// r.addHeader("Content-length", "" + (f.length() - startFrom));
// r.addHeader("Content-range", "" + startFrom + "-" +
// (f.length() - 1) + "/" + f.length());
return r;
}
} catch (IOException ioe) {
return new Response(
Response.Status.FORBIDDEN, WSServer.MIME_PLAINTEXT, "FORBIDDEN: Reading file failed.");
}
}