mirror of
https://github.com/internetarchive/brozzler.git
synced 2025-02-24 00:29:53 -05:00
291 lines
12 KiB
Python
291 lines
12 KiB
Python
#!/usr/bin/env python
|
|
# vim: set sw=4 et:
|
|
|
|
import logging
|
|
import json
|
|
import urllib.request
|
|
import itertools
|
|
import websocket
|
|
import time
|
|
import threading
|
|
import subprocess
|
|
import signal
|
|
import tempfile
|
|
import os
|
|
import socket
|
|
from umbra.behaviors import Behavior
|
|
|
|
class BrowserPool:
|
|
logger = logging.getLogger(__module__ + "." + __qualname__)
|
|
|
|
BASE_PORT = 9200
|
|
|
|
def __init__(self, size=3, chrome_exe='chromium-browser'):
|
|
self._available = set()
|
|
self._in_use = set()
|
|
|
|
for i in range(0, size):
|
|
browser = Browser(BrowserPool.BASE_PORT + i, chrome_exe)
|
|
self._available.add(browser)
|
|
|
|
self._lock = threading.Lock()
|
|
|
|
self.logger.info("browser ports: {}".format([browser.chrome_port for browser in self._available]))
|
|
|
|
def acquire(self):
|
|
"""Returns browser from pool if available, raises KeyError otherwise."""
|
|
with self._lock:
|
|
browser = self._available.pop()
|
|
self._in_use.add(browser)
|
|
return browser
|
|
|
|
def release(self, browser):
|
|
with self._lock:
|
|
self._available.add(browser)
|
|
self._in_use.remove(browser)
|
|
|
|
def shutdown_now(self):
|
|
for browser in self._in_use:
|
|
browser.abort_browse_page()
|
|
|
|
class BrowsingException(Exception):
|
|
pass
|
|
|
|
class Browser:
|
|
"""Runs chrome/chromium to synchronously browse one page at a time using
|
|
worker.browse_page(). Currently the implementation starts up a new instance
|
|
of chrome for each page browsed, always on the same debug port. (In the
|
|
future, it may keep the browser running indefinitely.)"""
|
|
|
|
logger = logging.getLogger(__module__ + "." + __qualname__)
|
|
|
|
HARD_TIMEOUT_SECONDS = 20 * 60
|
|
|
|
def __init__(self, chrome_port=9222, chrome_exe='chromium-browser'):
|
|
self.command_id = itertools.count(1)
|
|
self.chrome_port = chrome_port
|
|
self.chrome_exe = chrome_exe
|
|
self._behavior = None
|
|
self._websock = None
|
|
self._abort_browse_page = False
|
|
self._chrome_instance = None
|
|
|
|
def __repr__(self):
|
|
return "{}.{}:{}".format(Browser.__module__, Browser.__qualname__, self.chrome_port)
|
|
|
|
def __enter__(self):
|
|
self.start()
|
|
return self
|
|
|
|
def __exit__(self, *args):
|
|
self.stop()
|
|
|
|
def start(self):
|
|
# these can raise exceptions
|
|
self._work_dir = tempfile.TemporaryDirectory()
|
|
self._chrome_instance = Chrome(self.chrome_port, self.chrome_exe,
|
|
self._work_dir.name, os.sep.join([self._work_dir.name, "chrome-user-data"]))
|
|
self._websocket_url = self._chrome_instance.start()
|
|
|
|
def stop(self):
|
|
self._chrome_instance.stop()
|
|
self._work_dir.cleanup()
|
|
|
|
def abort_browse_page(self):
|
|
self._abort_browse_page = True
|
|
|
|
def browse_page(self, url, on_request=None):
|
|
"""Synchronously browses a page and runs behaviors.
|
|
|
|
Raises BrowsingException if browsing the page fails in a non-critical
|
|
way.
|
|
"""
|
|
self.url = url
|
|
self.on_request = on_request
|
|
|
|
self._websock = websocket.WebSocketApp(self._websocket_url,
|
|
on_open=self._visit_page, on_message=self._handle_message)
|
|
|
|
import random
|
|
threadName = "WebsockThread{}-{}".format(self.chrome_port,
|
|
''.join((random.choice('abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789') for _ in range(6))))
|
|
websock_thread = threading.Thread(target=self._websock.run_forever, name=threadName, kwargs={'ping_timeout':0.5})
|
|
websock_thread.start()
|
|
self._start = time.time()
|
|
aborted = False
|
|
|
|
try:
|
|
while True:
|
|
time.sleep(0.5)
|
|
if not self._websock or not self._websock.sock or not self._websock.sock.connected:
|
|
raise BrowsingException("websocket closed, did chrome die? {}".format(self._websocket_url))
|
|
elif time.time() - self._start > Browser.HARD_TIMEOUT_SECONDS:
|
|
self.logger.info("finished browsing page, reached hard timeout of {} seconds url={}".format(Browser.HARD_TIMEOUT_SECONDS, self.url))
|
|
return
|
|
elif self._behavior != None and self._behavior.is_finished():
|
|
self.logger.info("finished browsing page according to behavior url={}".format(self.url))
|
|
return
|
|
elif self._abort_browse_page:
|
|
raise BrowsingException("browsing page aborted")
|
|
finally:
|
|
if self._websock and self._websock.sock and self._websock.sock.connected:
|
|
try:
|
|
self._websock.close()
|
|
except BaseException as e:
|
|
self.logger.error("exception closing websocket {} - {}".format(self._websock, e))
|
|
|
|
websock_thread.join()
|
|
self._behavior = None
|
|
|
|
def send_to_chrome(self, **kwargs):
|
|
msg_id = next(self.command_id)
|
|
kwargs['id'] = msg_id
|
|
msg = json.dumps(kwargs)
|
|
self.logger.debug('sending message to {}: {}'.format(self._websock, msg))
|
|
self._websock.send(msg)
|
|
return msg_id
|
|
|
|
def _visit_page(self, websock):
|
|
self.send_to_chrome(method="Network.enable")
|
|
self.send_to_chrome(method="Page.enable")
|
|
self.send_to_chrome(method="Console.enable")
|
|
self.send_to_chrome(method="Debugger.enable")
|
|
self.send_to_chrome(method="Runtime.enable")
|
|
|
|
# disable google analytics, see _handle_message() where breakpoint is caught "Debugger.paused"
|
|
self.send_to_chrome(method="Debugger.setBreakpointByUrl", params={"lineNumber": 1, "urlRegex":"https?://www.google-analytics.com/analytics.js"})
|
|
|
|
# navigate to the page!
|
|
self.send_to_chrome(method="Page.navigate", params={"url": self.url})
|
|
|
|
def _handle_message(self, websock, message):
|
|
# self.logger.debug("message from {} - {}".format(websock.url, message[:95]))
|
|
# self.logger.debug("message from {} - {}".format(websock.url, message))
|
|
message = json.loads(message)
|
|
if "method" in message and message["method"] == "Network.requestWillBeSent":
|
|
if self._behavior:
|
|
self._behavior.notify_of_activity()
|
|
if message["params"]["request"]["url"].lower().startswith("data:"):
|
|
self.logger.debug("ignoring data url {}".format(message["params"]["request"]["url"][:80]))
|
|
elif self.on_request:
|
|
self.on_request(message)
|
|
elif "method" in message and message["method"] == "Page.loadEventFired":
|
|
if self._behavior is None:
|
|
self.logger.info("Page.loadEventFired, starting behaviors url={} message={}".format(self.url, message))
|
|
self._behavior = Behavior(self.url, self)
|
|
self._behavior.start()
|
|
else:
|
|
self.logger.warn("Page.loadEventFired again, perhaps original url had a meta refresh, or behaviors accidentally navigated to another page? starting behaviors again url={} message={}".format(self.url, message))
|
|
self._behavior = Behavior(self.url, self)
|
|
self._behavior.start()
|
|
elif "method" in message and message["method"] == "Console.messageAdded":
|
|
self.logger.debug("{} console.{} {}".format(websock.url,
|
|
message["params"]["message"]["level"],
|
|
message["params"]["message"]["text"]))
|
|
elif "method" in message and message["method"] == "Debugger.paused":
|
|
# We hit the breakpoint set in visit_page. Get rid of google
|
|
# analytics script!
|
|
|
|
self.logger.debug("debugger paused! message={}".format(message))
|
|
scriptId = message['params']['callFrames'][0]['location']['scriptId']
|
|
|
|
# replace script
|
|
self.send_to_chrome(method="Debugger.setScriptSource", params={"scriptId": scriptId, "scriptSource":"console.log('google analytics is no more!');"})
|
|
|
|
# resume execution
|
|
self.send_to_chrome(method="Debugger.resume")
|
|
elif "result" in message:
|
|
if self._behavior and self._behavior.is_waiting_on_result(message['id']):
|
|
self._behavior.notify_of_result(message)
|
|
# elif "method" in message and message["method"] in ("Network.dataReceived", "Network.responseReceived", "Network.loadingFinished"):
|
|
# pass
|
|
# elif "method" in message:
|
|
# self.logger.debug("{} {}".format(message["method"], message))
|
|
# else:
|
|
# self.logger.debug("[no-method] {}".format(message))
|
|
|
|
|
|
class Chrome:
|
|
logger = logging.getLogger(__module__ + "." + __qualname__)
|
|
|
|
def __init__(self, port, executable, user_home_dir, user_data_dir):
|
|
self.port = port
|
|
self.executable = executable
|
|
self.user_home_dir = user_home_dir
|
|
self.user_data_dir = user_data_dir
|
|
|
|
# returns websocket url to chrome window with about:blank loaded
|
|
def __enter__(self):
|
|
return self.start()
|
|
|
|
def __exit__(self, *args):
|
|
self.stop()
|
|
|
|
# returns websocket url to chrome window with about:blank loaded
|
|
def start(self):
|
|
timeout_sec = 60
|
|
new_env = os.environ.copy()
|
|
new_env["HOME"] = self.user_home_dir
|
|
chrome_args = [self.executable,
|
|
"--use-mock-keychain", # mac thing
|
|
"--user-data-dir={}".format(self.user_data_dir),
|
|
"--remote-debugging-port={}".format(self.port),
|
|
"--disable-web-sockets", "--disable-cache",
|
|
"--window-size=1100,900", "--no-default-browser-check",
|
|
"--disable-first-run-ui", "--no-first-run",
|
|
"--homepage=about:blank", "--disable-direct-npapi-requests",
|
|
"about:blank"]
|
|
self.logger.info("running {}".format(chrome_args))
|
|
self.chrome_process = subprocess.Popen(chrome_args, env=new_env, start_new_session=True)
|
|
self.logger.info("chrome running, pid {}".format(self.chrome_process.pid))
|
|
self._start = time.time() # member variable just so that kill -QUIT reports it
|
|
|
|
json_url = "http://localhost:%s/json" % self.port
|
|
|
|
while True:
|
|
try:
|
|
raw_json = urllib.request.urlopen(json_url).read()
|
|
all_debug_info = json.loads(raw_json.decode('utf-8'))
|
|
debug_info = [x for x in all_debug_info if x['url'] == 'about:blank']
|
|
|
|
if debug_info and 'webSocketDebuggerUrl' in debug_info[0]:
|
|
self.logger.debug("{} returned {}".format(json_url, raw_json))
|
|
url = debug_info[0]['webSocketDebuggerUrl']
|
|
self.logger.info('got chrome window websocket debug url {} from {}'.format(url, json_url))
|
|
return url
|
|
except:
|
|
pass
|
|
finally:
|
|
if time.time() - self._start > timeout_sec:
|
|
raise Exception("failed to retrieve {} after {} seconds".format(json_url, time.time() - self._start))
|
|
else:
|
|
time.sleep(0.5)
|
|
|
|
def stop(self):
|
|
timeout_sec = 60
|
|
self.logger.info("terminating chrome pid {}".format(self.chrome_process.pid))
|
|
|
|
self.chrome_process.terminate()
|
|
first_sigterm = last_sigterm = time.time()
|
|
|
|
while time.time() - first_sigterm < timeout_sec:
|
|
time.sleep(0.5)
|
|
|
|
status = self.chrome_process.poll()
|
|
if status is not None:
|
|
if status == 0:
|
|
self.logger.info("chrome pid {} exited normally".format(self.chrome_process.pid, status))
|
|
else:
|
|
self.logger.warn("chrome pid {} exited with nonzero status {}".format(self.chrome_process.pid, status))
|
|
return
|
|
|
|
# sometimes a hung chrome process will terminate on repeated sigterms
|
|
if time.time() - last_sigterm > 10:
|
|
self.chrome_process.terminate()
|
|
last_sigterm = time.time()
|
|
|
|
self.logger.warn("chrome pid {} still alive {} seconds after sending SIGTERM, sending SIGKILL".format(self.chrome_process.pid, timeout_sec))
|
|
self.chrome_process.kill()
|
|
status = self.chrome_process.wait()
|
|
self.logger.warn("chrome pid {} reaped (status={}) after killing with SIGKILL".format(self.chrome_process.pid, status))
|