// Oooh scary recursion
private static void parse(BufferedOutputStream out, HashSet<File> visited, File file)
throws IOException {
// Added File generic to HashSet for safe conversion - TJB
if (!file.exists() || file.isDirectory()) {
out.write(("[SSI include not found: " + file.getCanonicalPath() + "]").getBytes());
return;
}
if (visited.contains(file)) {
out.write(("[SSI circular inclusion rejected: " + file.getCanonicalPath() + "]").getBytes());
return;
}
visited.add(file);
// Work out the filename extension. If there isn't one, we keep
// it as the empty string ("").
String extension = WebServerConfig.getExtension(file);
if (WebServerConfig.SSI_EXTENSIONS.contains(extension)) {
// process this ssi page line by line
BufferedReader reader = new BufferedReader(new FileReader(file));
String line = null;
while ((line = reader.readLine()) != null) {
int startIndex;
int endIndex;
while ((startIndex = line.indexOf("<!--#include file=\"")) >= 0) {
if ((endIndex = line.indexOf("\" -->", startIndex)) > startIndex) {
out.write(line.substring(0, startIndex).getBytes());
String filename = line.substring(startIndex + 19, endIndex);
parse(out, visited, new File(file.getParentFile(), filename));
line = line.substring(endIndex + 5, line.length());
} else {
out.write(line.substring(0, 19).getBytes());
line = line.substring(19, line.length());
}
}
out.write(line.getBytes());
out.write(WebServerConfig.LINE_SEPARATOR);
}
} else {
// just dish out the bytes
BufferedInputStream reader = new BufferedInputStream(new FileInputStream(file));
byte[] buffer = new byte[4096];
int bytesRead;
while ((bytesRead = reader.read(buffer, 0, 4096)) != -1) {
out.write(buffer, 0, bytesRead);
}
}
visited.remove(file);
}
@Test
@Parameters(method = "files")
public void getExtensionTest(File files, String expected) {
assertEquals(expected, WebServerConfig.getExtension(files));
}
