From 0b8a8e88a3c33c14e52241190ee6478cb2acd49d Mon Sep 17 00:00:00 2001 From: Lars-Dominik Braun Date: Fri, 17 Nov 2017 19:54:30 +0100 Subject: Initial import --- .gitignore | 3 + LICENSE.txt | 19 +++ README.rst | 40 ++++++ crocoite/__init__.py | 3 + crocoite/cli.py | 320 ++++++++++++++++++++++++++++++++++++++++++++++++ crocoite/data/scroll.js | 10 ++ setup.py | 24 ++++ 7 files changed, 419 insertions(+) create mode 100644 .gitignore create mode 100644 LICENSE.txt create mode 100644 README.rst create mode 100644 crocoite/__init__.py create mode 100644 crocoite/cli.py create mode 100644 crocoite/data/scroll.js create mode 100644 setup.py diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..836e311 --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +__pycache__ +*.sw? +*.egg-info/ diff --git a/LICENSE.txt b/LICENSE.txt new file mode 100644 index 0000000..ad28e5d --- /dev/null +++ b/LICENSE.txt @@ -0,0 +1,19 @@ +Copyright (c) 2017 crocoite contributors + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in +all copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +THE SOFTWARE. diff --git a/README.rst b/README.rst new file mode 100644 index 0000000..7eea272 --- /dev/null +++ b/README.rst @@ -0,0 +1,40 @@ +crocoite +======== + +Archive websites using Google Chrome and its DevTools protocol. +Tested with Google Chrome 62.0.3202.89 for Linux only. + +Dependencies +------------ + +- Python 3 +- pychrome_ +- warcio_ + +.. _pychrome: https://github.com/fate0/pychrome +.. _warcio: https://github.com/webrecorder/warcio + +Usage +----- + +One-shot commandline interface and pywb_ playback:: + + google-chrome-stable --window-size=1920,1080 --remote-debugging-port=9222 & + crocoite-standalone http://example.com/ example.com.warc.gz + rm -rf collections && wb-manager init test && wb-manager add test example.com.warc.gz + wayback & + $BROWSER http://localhost:8080 + +For `headless Google Chrome`_ add the parameters ``--headless --disable-gpu``. + +.. _pywb: https://github.com/ikreymer/pywb +.. _headless Google Chrome: https://developers.google.com/web/updates/2017/04/headless-chrome + +Caveats +------- + +- Original HTTP requests/responses are not available. They are rebuilt from + data available. Character encoding for text documents is changed to UTF-8. +- Some sites request different assets based on screen resolution, some fetch + different scripts based on user agent. + diff --git a/crocoite/__init__.py b/crocoite/__init__.py new file mode 100644 index 0000000..066c83e --- /dev/null +++ b/crocoite/__init__.py @@ -0,0 +1,3 @@ +#!/usr/bin/env python3 + + diff --git a/crocoite/cli.py b/crocoite/cli.py new file mode 100644 index 0000000..24192fe --- /dev/null +++ b/crocoite/cli.py @@ -0,0 +1,320 @@ +# Copyright (c) 2017 crocoite contributors +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +# THE SOFTWARE. + +import os, sys, json +import pychrome +from urllib.parse import urlsplit +from warcio.warcwriter import WARCWriter +from warcio.statusandheaders import StatusAndHeaders +from base64 import b64decode +import logging +from logging.handlers import BufferingHandler +from http.server import BaseHTTPRequestHandler +from io import BytesIO +import argparse +import tempfile + +logger = logging.getLogger(__name__) + +# 10 MB, encoded! (i.e. actual data can be larger due to compression) +maxBodySize = 10*1024*1024 + +def packageData (path): + """ + Locate package data, see setup.py’s data_files + """ + return os.path.join (os.path.dirname (__file__), 'data', path) + +def packageUrl (path): + """ + Create URL for package data stored into WARC + """ + return 'urn:' + __package__ + ':' + path + +class WARCLogHandler (BufferingHandler): + """ + Buffered log handler, flushing to warcio + """ + + contentType = 'text/plain; charset=utf-8' + + def __init__ (self, capacity, warcfile): + BufferingHandler.__init__ (self, capacity) + self.warcfile = warcfile + + def flush (self): + self.acquire () + try: + buf = '' + for record in self.buffer: + buf += self.format (record) + buf += '\n' + # XXX: record type? + record = self.warcfile.create_warc_record ( + packageUrl ('log'), 'metadata', + payload=BytesIO (buf.encode ('utf8')), + warc_headers_dict={'Content-Type': self.contentType}) + self.warcfile.write_record(record) + self.buffer = [] + finally: + self.release () + +def main (): + def getStatusText (response): + text = response.get ('statusText') + if text: + return text + text = BaseHTTPRequestHandler.responses.get (response['status']) + if text: + return text[0] + return 'No status text available' + + def requestWillBeSent (**kwargs): + req = kwargs.get ('request') + reqId = kwargs['requestId'] + url = urlsplit (req['url']) + if url.scheme in ('http', 'https'): + logger.debug ('sending {} {}'.format (reqId, req['url'])) + if reqId in requests: + redirectResp = kwargs.get ('redirectResponse') + if redirectResp: + requests[reqId]['response'] = redirectResp + # XXX: can we retrieve the original response body right now? + itemToWarc (reqId, requests[reqId], ignoreBody=True) + else: + logger.warn ('request {} already exists, overwriting.'.format (reqId)) + requests[reqId] = {} + requests[reqId]['request'] = req + requests[reqId]['initiator'] = kwargs['initiator'] + else: + logger.warn ('request: ignoring scheme {}'.format (url.scheme)) + + def responseReceived (**kwargs): + resp = kwargs['response'] + reqId = kwargs['requestId'] + url = urlsplit (resp['url']) + if url.scheme in ('http', 'https') and reqId in requests: + logger.debug ('response {} {}'.format (reqId, resp['url'])) + requests[reqId]['response'] = resp + else: + logger.warn ('response: ignoring scheme {}'.format (url.scheme)) + + def loadEventFired (**kwargs): + """ + Equivalent to DOM ready JavaScript event + """ + root = tab.DOM.getDocument () + rootId = root['root']['nodeId'] + links = tab.DOM.querySelectorAll (nodeId=rootId, selector='a') + #for i in links['nodeIds']: + # print ('link', tab.DOM.getAttributes (nodeId=i)) + + def loadingFinished (**kwargs): + """ + Item was fully loaded. For some items the request body is not available + when responseReceived is fired, thus move everything here. + """ + reqId = kwargs['requestId'] + # we never recorded this request + if reqId not in requests: + return + item = requests[reqId] + req = item['request'] + resp = item['response'] + url = urlsplit (resp['url']) + if url.scheme in ('http', 'https'): + logger.debug ('finished {} {}'.format (reqId, req['url'])) + itemToWarc (reqId, item, kwargs['encodedDataLength']) + del requests[reqId] + + def itemToWarc (reqId, item, encodedDataLength=0, ignoreBody=False): + req = item['request'] + resp = item['response'] + url = urlsplit (resp['url']) + + # overwrite request headers with those actually sent + newReqHeaders = resp.get ('requestHeaders') + if newReqHeaders: + req['headers'] = newReqHeaders + + postData = req.get ('postData') + if postData: + postData = BytesIO (postData.encode ('utf8')) + path = url.path + if url.query: + path += '?' + url.query + httpHeaders = StatusAndHeaders('{} {} HTTP/1.1'.format (req['method'], path), + req['headers'], protocol='HTTP/1.1', is_http_request=True) + initiator = item['initiator'] + warcHeaders = { + 'X-Chrome-Initiator': json.dumps (initiator), + } + record = writer.create_warc_record(req['url'], 'request', + payload=postData, http_headers=httpHeaders, + warc_headers_dict=warcHeaders) + writer.write_record(record) + concurrentTo = record.rec_headers['WARC-Record-ID'] + + # check body size first, since we’re loading everything into memory + if encodedDataLength < maxBodySize: + try: + if ignoreBody: + rawBody = b'' + base64Encoded = True + else: + body = tab.Network.getResponseBody (requestId=reqId) + rawBody = body['body'] + base64Encoded = body['base64Encoded'] + if base64Encoded: + logger.debug ('body is base64 encoded') + rawBody = b64decode (rawBody) + else: + rawBody = rawBody.encode ('utf8') + + httpHeaders = StatusAndHeaders('{} {}'.format (resp['status'], + getStatusText (resp)), resp['headers'], protocol='HTTP/1.1') + + # Content is saved decompressed and decoded, remove these headers + blacklistedHeaders = {'transfer-encoding', 'content-encoding'} + for h in blacklistedHeaders: + httpHeaders.remove_header (h) + + # chrome sends nothing but utf8 encoded text. Fortunately HTTP + # headers take precedence over the document’s , thus we can + # easily override those. + contentType = resp['mimeType'] + if not base64Encoded: + contentType += '; charset=utf-8' + httpHeaders.replace_header ('content-type', contentType) + + httpHeaders.replace_header ('content-length', '{:d}'.format (len (rawBody))) + + warcHeaders = { + 'WARC-Concurrent-To': concurrentTo, + 'WARC-IP-Address': resp.get ('remoteIPAddress', ''), + 'X-Chrome-Protocol': resp.get ('protocol', ''), + 'X-Chrome-FromDiskCache': str (resp.get ('fromDiskCache')), + 'X-Chrome-ConnectionReused': str (resp.get ('connectionReused')), + } + record = writer.create_warc_record(resp['url'], 'response', + warc_headers_dict=warcHeaders, payload=BytesIO (rawBody), + http_headers=httpHeaders) + writer.write_record(record) + except pychrome.exceptions.CallMethodException: + logger.error ('no data for {} {} {}'.format (resp['url'], + resp['status'], kwargs.get ('requestId'))) + else: + logger.warn ('body for {} is too large, {} bytes'.format (resp['url'], kwargs['encodedDataLength'])) + + def loadingFailed (**kwargs): + reqId = kwargs['requestId'] + logger.debug ('failed {} {}'.format (reqId, kwargs['errorText'], kwargs.get ('blockedReason'))) + if reqId in requests: + del requests[reqId] + + logging.basicConfig (level=logging.DEBUG) + + parser = argparse.ArgumentParser(description='Save website to WARC using Google Chrome.') + parser.add_argument('--browser', default='http://127.0.0.1:9222', help='DevTools URL') + parser.add_argument('--timeout', default=10, type=int, help='Maximum time for archival') + parser.add_argument('--waitstep', default=2, type=int, help='') + parser.add_argument('--onload', default=packageData ('scroll.js'), help='') + parser.add_argument('--log-buffer', default=1000, type=int, dest='logBuffer') + parser.add_argument('--keep-tab', action='store_true', default=False, dest='keepTab', help='Keep tab open') + parser.add_argument('url', help='Website URL') + parser.add_argument('output', help='WARC filename') + + args = parser.parse_args () + + with open (args.onload, 'r') as fd: + onload = fd.read () + + # temporary store for requests + requests = {} + + # create a browser instance + browser = pychrome.Browser(url=args.browser) + + # create a tab + tab = browser.new_tab() + + # setup callbacks + tab.Network.requestWillBeSent = requestWillBeSent + tab.Network.responseReceived = responseReceived + tab.Network.loadingFinished = loadingFinished + tab.Network.loadingFailed = loadingFailed + tab.Page.loadEventFired = loadEventFired + + # start the tab + tab.start() + + fd = open (args.output, 'wb') + writer = WARCWriter (fd, gzip=True) + version = tab.Browser.getVersion () + # assuming these don’t change + layoutMetrics = tab.Page.getLayoutMetrics () + payload = { + 'software': __package__, + 'browser': version['product'], + 'useragent': version['userAgent'], + # XXX: I’m not entirely sure which one we should use here + 'viewport': '{}x{}'.format (layoutMetrics['layoutViewport']['clientWidth'], + layoutMetrics['layoutViewport']['clientHeight']) + } + warcinfo = writer.create_warcinfo_record (filename=None, info=payload) + writer.write_record (warcinfo) + + warcLogger = WARCLogHandler (args.logBuffer, writer) + logger.addHandler (warcLogger) + + # save onload script + record = writer.create_warc_record (packageUrl ('onload'), 'metadata', + payload=BytesIO (onload.encode ('utf8')), + warc_headers_dict={'Content-Type': 'application/javascript; charset=utf-8'}) + writer.write_record (record) + + # enable events + tab.Network.enable() + tab.Page.enable () + tab.Network.clearBrowserCache () + if tab.Network.canClearBrowserCookies ()['result']: + tab.Network.clearBrowserCookies () + # inject our custom javascript to the page before loading + tab.Page.addScriptToEvaluateOnNewDocument (source=onload) + + tab.Page.navigate(url=args.url) + + # smaller waitstep might miss some requests issued very late, larger + # waitstep may waste time + for i in range (0, args.timeout, args.waitstep): + tab.wait (args.waitstep) + if len (requests) == 0: + break + + tab.stop() + if not args.keepTab: + browser.close_tab(tab) + + logger.removeHandler (warcLogger) + warcLogger.flush () + fd.close () + + return True + diff --git a/crocoite/data/scroll.js b/crocoite/data/scroll.js new file mode 100644 index 0000000..cf060fc --- /dev/null +++ b/crocoite/data/scroll.js @@ -0,0 +1,10 @@ +(function(){ +function scroll (event) { + window.scrollBy (0, window.innerHeight/2); + return true; +} +function onload (event) { + window.setInterval (scroll, 200); +} +document.addEventListener("DOMContentLoaded", onload); +}()); diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..939cec9 --- /dev/null +++ b/setup.py @@ -0,0 +1,24 @@ +from distutils.core import setup + +setup( + name='crocoite', + version='0.1.0', + author='Lars-Dominik Braun', + author_email='lars+crocoite@6xq.net', + packages=['crocoite'], + license='LICENSE.txt', + description='Save website to WARC using Google Chrome.', + long_description=open('README.rst').read(), + install_requires=[ + 'pychrome', + 'warcio', + ], + entry_points={ + 'console_scripts': [ + 'crocoite-standalone = crocoite.cli:main', + ], + }, + data_files=[ + ('crocoite/data', ['crocoite/data/onload.js']), + ], +) -- cgit v1.2.3