/**
* Gets a new browser session
*
* @param browserString
* @param startURL
* @param extensionJs per-session user extension Javascript
* @param configuration Remote Control configuration. Cannot be null.
* @param useCached if a cached session should be used if one is available
* @param ensureClean if a clean session (e.g. no previous cookies) is required.
* @return the BrowserSessionInfo for the new browser session.
* @throws RemoteCommandException
*/
protected BrowserSessionInfo getNewBrowserSession(
String browserString,
String startURL,
String extensionJs,
Capabilities browserConfigurations,
boolean useCached,
boolean ensureClean,
RemoteControlConfiguration configuration)
throws RemoteCommandException {
BrowserSessionInfo sessionInfo = null;
browserString = validateBrowserString(browserString, configuration);
if (configuration.getProxyInjectionModeArg()) {
InjectionHelper.setBrowserSideLogEnabled(configuration.isBrowserSideLogEnabled());
InjectionHelper.init();
}
if (useCached) {
log.info("grabbing available session...");
sessionInfo = grabAvailableSession(browserString, startURL);
}
// couldn't find one in the cache, or not reusing sessions.
if (null == sessionInfo) {
log.info("creating new remote session");
sessionInfo =
createNewRemoteSession(
browserString,
startURL,
extensionJs,
browserConfigurations,
ensureClean,
configuration);
}
assert null != sessionInfo;
if (false /* ensureClean */) {
// need to add this to the launcher API.
// sessionInfo.launcher.hideCurrentSessionData();
}
return sessionInfo;
}
public static long injectJavaScript(
HttpRequest request, HttpResponse response, InputStream in, OutputStream out, String debugURL)
throws IOException {
if (!contentTransformations.containsKey("__SELENIUM_JS__")) {
init();
}
int len = 102400;
byte[] buf = new byte[len];
len = readStream(in, buf, len);
if (len == -1) {
return -1;
}
int lengthOfBOM = getBOMLength(buf);
String data = new String(buf, lengthOfBOM, len);
boolean isKnownToBeHtml =
HtmlIdentifier.shouldBeInjected(request.getPath(), response.getContentType(), data);
String url = response.getHttpRequest().getRequestURL().toString();
if (debugURL.equals(url)) {
log.info("debug URL seen");
}
if (!isKnownToBeHtml) {
out.write(buf, 0, len);
}
// else if (lengthOfBOM>0) {
// out.write(buf, 0, lengthOfBOM);
// }
String sessionId = SeleniumDriverResourceHandler.getLastSessionId();
long bytesCopied = len;
log.fine(url + " (InjectionHelper looking)");
if (!isKnownToBeHtml) {
bytesCopied += ModifiedIO.copy(in, out);
} else {
log.fine("injecting...");
response.removeField("Content-Length"); // added js will make it wrong, lead to page getting
// truncated
ByteArrayOutputStream baos = new ByteArrayOutputStream();
if (INJECT_SCRIPT_TAGS) {
writeScriptTags(baos);
}
InputStream jsIn = new ClassPathResource(InjectionHelper.injectionHtml).getInputStream();
contentTransformations.put("@SESSION_ID@", sessionId);
writeDataWithUserTransformations("", jsIn, baos);
jsIn.close();
baos.write(setSomeJsVars(sessionId));
for (String filename : userJsInjectionFiles) {
jsIn = new FileInputStream(filename);
IO.copy(jsIn, baos);
}
int headIndex;
if (tryToInjectInHead) {
headIndex = data.toLowerCase().indexOf("<head>");
} else {
headIndex = -1;
}
if (headIndex != -1) {
data = data.substring(0, headIndex + 6) + baos.toString() + data.substring(headIndex + 6);
} else {
data = baos.toString() + data;
}
bytesCopied += writeDataWithUserTransformations(data, in, out);
}
return bytesCopied;
}
