feat: add fullpage and maxheight parameters for enhanced screenshot options

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

CLAUDE.md

@@ -0,0 +1,76 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Project
+
+http2pic - PHP website renderer that takes screenshots of URLs and returns them as images. Live at https://http2pic.haschek.at/
+
+## Architecture
+
+**Entry point:** `web/index.php` - PSR-4 router with two paths:
+- `/api` - Chrome/Chromium screenshot via Selenium WebDriver (php-webdriver). Params: `url`, `viewport` (WIDTHxHEIGHT), `js` (true/false), `fullpage` (true/false), `maxheight` (1–30000, default 15000). Connects to `localhost:4444` ChromeDriver, sets window size, takes screenshot as PNG. In fullpage mode, resizes window to full `scrollHeight` (capped at `maxheight`) instead of hiding overflow.
+- default - renders `src/templates/index.html.php` (landing page).
+
+**Legacy class:** `src/http2pic.class.php` - Old `wkhtmltoimage`-based renderer (deprecated, not used in production). Supports PNG/JPG, viewport, resize, URL reachability check, file caching.
+
+**Helpers:** `src/helpers.php` - `renderTemplate()`, `addToLog()`, `getUserIP()`.
+
+**Config:** `src/config.inc.php` - set at build time by `start.sh` from `URL` env var.
+
+## Running
+
+**Docker (production):**
+```
+docker compose up -d
+```
+Services: Caddy (:80), PHP-FPM, ChromeDriver (:4444). Volumes: `./cache` and `./logs`. Configured via `URL` env var.
+
+**Dev container:** `.devcontainer/` - same stack, run `./devcontainer/start.sh`. Forward port 8080.
+
+**Quick test:**
+```
+php -S localhost:8080 -t web/
+# Then visit http://localhost:8080/api?url=<target>
+```
+
+## Key files
+
+| File | Purpose |
+|------|---------|
+| `web/index.php` | API + template router |
+| `src/http2pic.class.php` | Legacy wkhtmltoimage renderer |
+| `src/helpers.php` | Template render, logging, IP helper |
+| `src/config.inc.php` | Runtime config (URL, API_KEY, BLOCK_PRIVATE_IPS) |
+| `docker/Caddyfile` | Reverse proxy, PHP-FPM, file server |
+| `docker/start.sh` | Boots PHP-FPM, ChromeDriver, writes config |
+| `docker-compose.yml` | Production compose |
+
+## API Key
+
+Set `API_KEY` env var in docker-compose to require authentication on all `/api` requests.
+Leave unset (default) for open access.
+
+```bash
+# via header (preferred — not logged in access logs)
+curl -H "X-API-Key: your-secret-key" "http://host/api?url=https://example.com"
+
+# via query param
+curl "http://host/api?key=your-secret-key&url=https://example.com"
+```
+
+## SSRF Protection
+
+Set `BLOCK_PRIVATE_IPS=true` to reject requests to LAN, loopback, and cloud metadata IPs.
+Recommended when hosting publicly. Default is off (allows local/LAN addresses).
+
+Note: DNS rebinding attacks can bypass this protection (attacker-controlled DNS can return a public IP
+during validation and a private IP when Chrome actually connects). Full protection requires a network-level
+egress firewall.
+
+## Caveats
+
+- `web/index.php` has a `var_dump($cmd)` debug statement left in `http2pic.class.php:181` - remove before shipping.
+- Legacy `http2pic.class.php` has a variable scoping bug: line 109 uses `$url` instead of `$this->params['url']`.
+- Cache dir permissions must be `777` (set by `start.sh`).
+- ChromeDriver must be running on `localhost:4444` for the API to work.

README.md

@@ -28,18 +28,41 @@ Whenever you come to this page you can just [download](https://github.com/chrisi
 
 ## Usage
 
-After you extracted the contents of this repo to your webserver and can access the page and it will tell you how to use the API.
-
-But it's as simple as:
-
 ```
-https://your-url-and.path/api.php?[OPTIONS]&url=[WEBSITE_URL]
+https://your-host/api?url=[WEBSITE_URL]&[OPTIONS]
 ```
 
-The requested page will render as image (not provide a link). So you can use the path to your api.php file like so:
+### Parameters
+
+| Parameter | Default | Description |
+|-----------|---------|-------------|
+| `url` | — | Target URL to screenshot (required) |
+| `viewport` | `1024x768` | Viewport size as `WIDTHxHEIGHT` (max 3840x2160) |
+| `js` | `true` | Enable JavaScript (`true`/`false`) |
+| `fullpage` | `false` | Capture full page height instead of viewport only |
+| `maxheight` | `15000` | Max pixel height for full-page captures (1–30000) |
+| `key` | — | API key (if `API_KEY` env var is set) |
+
+### Examples
+
+```bash
+# Standard viewport screenshot
+curl "https://your-host/api?url=https://example.com" -o screenshot.png
+
+# Full-page screenshot (great for LLM analysis)
+curl "https://your-host/api?url=https://example.com&fullpage=true" -o full.png
+
+# Full-page with custom width and height cap
+curl "https://your-host/api?url=https://example.com&fullpage=true&viewport=1280x768&maxheight=20000" -o full.png
+
+# With API key
+curl -H "X-API-Key: your-secret" "https://your-host/api?url=https://example.com&fullpage=true" -o full.png
+```
+
+Use as an `<img>` src:
 
 ```html
-<img src="https://your-url-and.path/api.php?url=http://xkcd.com" title="screenshot of xkcd.com" />
+<img src="https://your-host/api?url=https://example.com" title="screenshot" />
 ```
 
 ### Example php script to proxy an image to the local server

docker-compose-dev.yml

@@ -1,4 +1,3 @@
-version: '3.3'
 services:
   http2pic:
     build:
@@ -13,6 +12,8 @@ services:
       - ./logs:/srv/logs
 
     environment:
-      - URL=http://localhostxxx:8080
+      - URL=http://localhost:8080
+      # - API_KEY=your-secret-key      # if set, all /api requests must provide it
+      # - BLOCK_PRIVATE_IPS=true       # block LAN/loopback/metadata IPs (recommended for public hosting)
     ports:
       - 8080:80

docker-compose.yml

@@ -7,5 +7,7 @@ services:
       - ./logs:/srv/logs
     environment:
       - URL=http://localhost:8080
+      # - API_KEY=your-secret-key      # if set, all /api requests must provide it
+      # - BLOCK_PRIVATE_IPS=true       # block LAN/loopback/metadata IPs (recommended for public hosting)
     ports:
       - 8080:80

docker/dokploy-compose.yml

@@ -8,6 +8,7 @@ services:
       - 80
     environment:
       - URL=https://http2pic.haschek.at
+      - BLOCK_PRIVATE_IPS=true
     networks:
       - dokploy-network
     labels:

docker/start.sh

@@ -14,9 +14,19 @@ chmod 777 /srv/logs
 
 echo ' [+] Building config'
 _buildConfig() {
+    local block_private api_key
+    case "${BLOCK_PRIVATE_IPS:-false}" in
+        true|1|yes) block_private=true ;;
+        *)          block_private=false ;;
+    esac
+    api_key="${API_KEY:-}"
+    api_key="${api_key//\\/}"
+    api_key="${api_key//\'/}"
     echo "<?php"
     echo "date_default_timezone_set('Europe/Vienna');"
     echo "define('URL','${URL:-http://localhost:8080}');"
+    echo "define('API_KEY','${api_key}');"
+    echo "define('BLOCK_PRIVATE_IPS',${block_private});"
     echo ""
 }
 

docs/superpowers/plans/2026-04-20-security-hardening.md

@@ -0,0 +1,564 @@
+# Security Hardening & Bug Fixes Implementation Plan
+
+> **For agentic workers:** REQUIRED SUB-SKILL: Use superpowers:subagent-driven-development (recommended) or superpowers:executing-plans to implement this plan task-by-task. Steps use checkbox (`- [ ]`) syntax for tracking.
+
+**Goal:** Harden http2pic with optional API key auth, opt-in SSRF protection, viewport cap, and fix two correctness bugs (60ms timeout, viewport set after page load).
+
+**Architecture:** All PHP changes stay in `web/index.php` and `src/helpers.php`. Config vars flow from `docker-compose.yml` env → `docker/start.sh` → `src/config.inc.php` defines. No new dependencies.
+
+**Tech Stack:** PHP 8.4, php-webdriver, Caddy, Docker
+
+---
+
+## File Map
+
+| File | What changes |
+|------|-------------|
+| `tests/test_helpers.php` | New — unit tests for `isPrivateIP()` and `getUserIP()` |
+| `src/helpers.php` | Add `isPrivateIP()`, fix `getUserIP()`, fix `addToLog()` log injection |
+| `docker/start.sh` | Write `API_KEY` and `BLOCK_PRIVATE_IPS` defines into config |
+| `docker-compose.yml` | Add commented env var examples |
+| `web/index.php` | API key check, SSRF check, viewport cap, viewport-before-get, timeout fix, generic errors |
+
+---
+
+## Task 1: Write failing tests for helpers
+
+**Files:**
+- Create: `tests/test_helpers.php`
+
+- [ ] **Step 1: Create test file**
+
+```php
+<?php
+define('DS', DIRECTORY_SEPARATOR);
+define('ROOT', dirname(__FILE__) . '/..');
+require_once ROOT . '/src/helpers.php';
+
+$passed = 0;
+$failed = 0;
+
+function check(bool $condition, string $label): void {
+    global $passed, $failed;
+    if ($condition) { echo "PASS: $label\n"; $passed++; }
+    else            { echo "FAIL: $label\n"; $failed++; }
+}
+
+// --- isPrivateIP ---
+check(isPrivateIP('127.0.0.1'),        'loopback 127.0.0.1');
+check(isPrivateIP('127.255.255.255'),   'loopback 127.255.255.255');
+check(isPrivateIP('10.0.0.1'),          '10.x private');
+check(isPrivateIP('10.255.255.255'),    '10.255.x private');
+check(isPrivateIP('172.16.0.1'),        '172.16.x private');
+check(isPrivateIP('172.31.255.255'),    '172.31.x private');
+check(!isPrivateIP('172.15.255.255'),   '172.15.x is public');
+check(!isPrivateIP('172.32.0.0'),       '172.32.x is public');
+check(isPrivateIP('192.168.0.1'),       '192.168.x private');
+check(isPrivateIP('169.254.169.254'),   'AWS metadata IP');
+check(!isPrivateIP('8.8.8.8'),          'Google DNS is public');
+check(!isPrivateIP('93.184.216.34'),    'example.com IP is public');
+check(isPrivateIP('not-an-ip'),         'unparseable IP blocked');
+
+// --- getUserIP ---
+unset($_SERVER['HTTP_X_FORWARDED_FOR']);
+$_SERVER['REMOTE_ADDR'] = '5.6.7.8';
+check(getUserIP() === '5.6.7.8', 'getUserIP falls back to REMOTE_ADDR');
+
+$_SERVER['HTTP_X_FORWARDED_FOR'] = '1.2.3.4, 10.0.0.1, 172.16.0.5';
+check(getUserIP() === '1.2.3.4', 'getUserIP takes first X-Forwarded-For IP');
+
+$_SERVER['HTTP_X_FORWARDED_FOR'] = '  9.9.9.9  ';
+check(getUserIP() === '9.9.9.9', 'getUserIP trims whitespace');
+
+echo "\n$passed passed, $failed failed\n";
+exit($failed > 0 ? 1 : 0);
+```
+
+- [ ] **Step 2: Run tests — expect failures**
+
+```bash
+php tests/test_helpers.php
+```
+
+Expected: several `FAIL:` lines (functions don't exist yet or have wrong logic). PHP fatal error on `isPrivateIP` is fine.
+
+---
+
+## Task 2: Implement helpers changes
+
+**Files:**
+- Modify: `src/helpers.php`
+
+- [ ] **Step 1: Replace `getUserIP()` and fix `addToLog()`, add `isPrivateIP()`**
+
+Replace the entire contents of `src/helpers.php` with:
+
+```php
+<?php
+
+function renderTemplate($templatename, $variables = [], $basepath = ROOT . '/src')
+{
+    ob_start();
+    if (is_array($variables))
+        extract($variables);
+    if (file_exists($basepath . DS . 'templates' . DS . $templatename . '.php'))
+        include($basepath . DS . 'templates' . DS . $templatename . '.php');
+    else if (file_exists($basepath . DS . 'templates' . DS . $templatename))
+        include($basepath . DS . 'templates' . DS . $templatename);
+    $rendered = ob_get_contents();
+    ob_end_clean();
+    return $rendered;
+}
+
+function addToLog(string $data): void
+{
+    $data = str_replace(["\n", "\r", "\t"], ' ', $data);
+    $fp = fopen(ROOT . DS . 'logs' . DS . 'app.log', 'a');
+    fwrite($fp, date("d.m.y H:i") . "\t" . $data . "\n");
+    fclose($fp);
+}
+
+function getUserIP(): string
+{
+    if (!empty($_SERVER['HTTP_X_FORWARDED_FOR']))
+        return trim(explode(',', $_SERVER['HTTP_X_FORWARDED_FOR'])[0]);
+    return $_SERVER['REMOTE_ADDR'];
+}
+
+function isPrivateIP(string $ip): bool
+{
+    if (filter_var($ip, FILTER_VALIDATE_IP) === false) return true;
+
+    $long = ip2long($ip);
+    if ($long === false) return true;
+
+    foreach ([
+        [ip2long('127.0.0.0'),   0xFF000000], // 127.0.0.0/8  loopback
+        [ip2long('10.0.0.0'),    0xFF000000], // 10.0.0.0/8   RFC1918
+        [ip2long('172.16.0.0'),  0xFFF00000], // 172.16.0.0/12 RFC1918
+        [ip2long('192.168.0.0'), 0xFFFF0000], // 192.168.0.0/16 RFC1918
+        [ip2long('169.254.0.0'), 0xFFFF0000], // 169.254.0.0/16 link-local/metadata
+    ] as [$base, $mask]) {
+        if (($long & $mask) === ($base & $mask)) return true;
+    }
+    return false;
+}
+```
+
+- [ ] **Step 2: Run tests — expect all pass**
+
+```bash
+php tests/test_helpers.php
+```
+
+Expected output ends with: `N passed, 0 failed`
+
+- [ ] **Step 3: Commit**
+
+```bash
+git add src/helpers.php tests/test_helpers.php
+git commit -m "feat: add isPrivateIP helper, fix getUserIP and addToLog"
+```
+
+---
+
+## Task 3: Update config generation
+
+**Files:**
+- Modify: `docker/start.sh`
+- Modify: `docker-compose.yml`
+
+- [ ] **Step 1: Update `_buildConfig` in `docker/start.sh`**
+
+Replace the `_buildConfig` function:
+
+```bash
+_buildConfig() {
+    echo "<?php"
+    echo "date_default_timezone_set('Europe/Vienna');"
+    echo "define('URL','${URL:-http://localhost:8080}');"
+    echo "define('API_KEY','${API_KEY:-}');"
+    echo "define('BLOCK_PRIVATE_IPS',${BLOCK_PRIVATE_IPS:-false});"
+    echo ""
+}
+```
+
+- [ ] **Step 2: Add commented env var examples to `docker-compose.yml`**
+
+Replace the `environment:` block:
+
+```yaml
+    environment:
+      - URL=http://localhost:8080
+      # - API_KEY=your-secret-key      # if set, all /api requests must provide it
+      # - BLOCK_PRIVATE_IPS=true       # block LAN/loopback/metadata IPs (recommended for public hosting)
+```
+
+- [ ] **Step 3: Verify config generation manually**
+
+```bash
+URL=http://localhost:8080 API_KEY=test BLOCK_PRIVATE_IPS=true bash -c '
+_buildConfig() {
+    echo "<?php"
+    echo "date_default_timezone_set('"'"'Europe/Vienna'"'"');"
+    echo "define('"'"'URL'"'"','"'"'${URL:-http://localhost:8080}'"'"');"
+    echo "define('"'"'API_KEY'"'"','"'"'${API_KEY:-}'"'"');"
+    echo "define('"'"'BLOCK_PRIVATE_IPS'"'"',${BLOCK_PRIVATE_IPS:-false});"
+    echo ""
+}
+_buildConfig'
+```
+
+Expected output:
+```
+<?php
+date_default_timezone_set('Europe/Vienna');
+define('URL','http://localhost:8080');
+define('API_KEY','test');
+define('BLOCK_PRIVATE_IPS',true);
+```
+
+- [ ] **Step 4: Commit**
+
+```bash
+git add docker/start.sh docker-compose.yml
+git commit -m "feat: add API_KEY and BLOCK_PRIVATE_IPS config vars"
+```
+
+---
+
+## Task 4: Fix index.php bugs + viewport cap + generic errors
+
+**Files:**
+- Modify: `web/index.php`
+
+This task rewrites the `case 'api':` block. Read the current state of `web/index.php` first, then replace the entire `case 'api':` block (from `case 'api':` through `break;`) with the following.
+
+- [ ] **Step 1: Replace `case 'api':` block**
+
+```php
+    case 'api':
+        $target = substr($_SERVER['REQUEST_URI'], 5);
+        if (!$target || !filter_var($target, FILTER_VALIDATE_URL))
+            $target = $_REQUEST['url'];
+        if (!filter_var($target, FILTER_VALIDATE_URL)) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Invalid URL';
+            exit;
+        }
+        $ip = getUserIP();
+
+        $viewport = $_REQUEST['viewport'] ?: '1024x768';
+        if (!preg_match('/^\d+x\d+$/', $viewport)) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Invalid viewport format. Use WIDTHxHEIGHT (e.g., 1024x768)';
+            exit;
+        }
+        $vpParts = array_map('intval', explode('x', $viewport));
+        if ($vpParts[0] > 3840 || $vpParts[1] > 2160) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Viewport exceeds maximum (3840x2160)';
+            exit;
+        }
+
+        $js = $_REQUEST['js'] == 'false' ? false : true;
+
+        $serverUrl = 'http://localhost:4444';
+        $options = new \Facebook\WebDriver\Chrome\ChromeOptions();
+        $options->addArguments(['--headless', '--disable-gpu', '--no-sandbox', '--disable-dev-shm-usage']);
+
+        $capabilities = DesiredCapabilities::chrome();
+        $capabilities->setCapability(\Facebook\WebDriver\Chrome\ChromeOptions::CAPABILITY, $options);
+
+        if (!$js)
+            $capabilities->setCapability('javascriptEnabled', false);
+
+        $driver = null;
+        $error = null;
+        try {
+            $driver = RemoteWebDriver::create($serverUrl, $capabilities, 30000, 60000);
+            $driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension($vpParts[0], $vpParts[1]));
+            $driver->get($target);
+            $driver->executeScript('document.body.style.overflow = "hidden";');
+
+            addToLog("$ip\tRequested $target with viewport $viewport and js " . ($js ? 'enabled' : 'disabled'));
+
+            $screenshot = $driver->takeScreenshot();
+            header('Content-Type: image/png');
+            header('Content-Length: ' . strlen($screenshot));
+            echo $screenshot;
+        } catch (Exception $e) {
+            $error = $e->getMessage();
+            addToLog("$ip\tRequested $target but resulted in error:\t" . $error);
+        } finally {
+            if ($driver instanceof RemoteWebDriver) {
+                try { $driver->quit(); } catch (Exception $q) {}
+            }
+        }
+        if ($error !== null) {
+            header('HTTP/1.0 500 Internal Server Error');
+            echo 'Screenshot failed';
+        }
+
+        break;
+```
+
+- [ ] **Step 2: Test viewport cap with curl**
+
+```bash
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=https://example.com&viewport=99999x99999"
+```
+
+Expected: `400`
+
+```bash
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=https://example.com&viewport=1024x768"
+```
+
+Expected: `200`
+
+- [ ] **Step 3: Commit**
+
+```bash
+git add web/index.php
+git commit -m "fix: viewport before page load, 60ms→60s timeout, viewport cap, generic errors"
+```
+
+---
+
+## Task 5: Add API key authentication
+
+**Files:**
+- Modify: `web/index.php`
+
+- [ ] **Step 1: Add API key check at top of `case 'api':` block**
+
+Insert immediately after `case 'api':` (before the `$target` line):
+
+```php
+        if (defined('API_KEY') && API_KEY !== '') {
+            $provided = $_SERVER['HTTP_X_API_KEY'] ?? $_REQUEST['key'] ?? '';
+            if ($provided !== API_KEY) {
+                header('HTTP/1.0 401 Unauthorized');
+                echo 'Invalid or missing API key';
+                exit;
+            }
+        }
+```
+
+- [ ] **Step 2: Test without key (no API_KEY configured)**
+
+```bash
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=https://example.com"
+```
+
+Expected: `200` (API_KEY not set in local config → open access)
+
+- [ ] **Step 3: Test with key via header**
+
+Temporarily set `API_KEY` in `src/config.inc.php` to `'testkey'`, then:
+
+```bash
+# missing key → 401
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=https://example.com"
+# header → 200
+curl -s -o /dev/null -w "%{http_code}" -H "X-API-Key: testkey" "http://localhost:8080/api?url=https://example.com"
+# query param → 200
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?key=testkey&url=https://example.com"
+# wrong key → 401
+curl -s -o /dev/null -w "%{http_code}" -H "X-API-Key: wrongkey" "http://localhost:8080/api?url=https://example.com"
+```
+
+Expected: `401`, `200`, `200`, `401`
+
+Revert the manual edit to `src/config.inc.php` after testing (it gets regenerated by `start.sh` in Docker).
+
+- [ ] **Step 4: Document API key usage in CLAUDE.md**
+
+In `CLAUDE.md`, add a new section under the existing content:
+
+```markdown
+## API Key
+
+Set `API_KEY` env var in docker-compose to require authentication on all `/api` requests.
+Leave unset (default) for open access.
+
+```bash
+# via header (preferred — not logged in access logs)
+curl -H "X-API-Key: your-secret-key" "http://host/api?url=https://example.com"
+
+# via query param
+curl "http://host/api?key=your-secret-key&url=https://example.com"
+```
+
+## SSRF Protection
+
+Set `BLOCK_PRIVATE_IPS=true` to reject requests to LAN, loopback, and cloud metadata IPs.
+Recommended when hosting publicly. Default is off (allows local/LAN addresses).
+```
+
+- [ ] **Step 5: Commit**
+
+```bash
+git add web/index.php CLAUDE.md
+git commit -m "feat: optional API key auth via X-API-Key header or ?key= param"
+```
+
+---
+
+## Task 6: Add SSRF protection
+
+**Files:**
+- Modify: `web/index.php`
+
+- [ ] **Step 1: Add SSRF check after viewport validation**
+
+Insert after the `$js = ...` line (before `$serverUrl = ...`):
+
+```php
+        if (defined('BLOCK_PRIVATE_IPS') && BLOCK_PRIVATE_IPS) {
+            $host = parse_url($target, PHP_URL_HOST);
+            if (filter_var($host, FILTER_VALIDATE_IP)) {
+                $resolvedIp = $host;
+            } else {
+                $resolvedIp = gethostbyname($host);
+                if ($resolvedIp === $host) {
+                    header('HTTP/1.0 403 Forbidden');
+                    echo 'URL not allowed';
+                    exit;
+                }
+            }
+            if (isPrivateIP($resolvedIp)) {
+                header('HTTP/1.0 403 Forbidden');
+                echo 'URL not allowed';
+                exit;
+            }
+        }
+```
+
+- [ ] **Step 2: Test SSRF protection**
+
+Temporarily set `BLOCK_PRIVATE_IPS` to `true` in `src/config.inc.php`, then:
+
+```bash
+# private IP → 403
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=http://192.168.1.1/"
+# loopback → 403
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=http://127.0.0.1/"
+# metadata → 403
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=http://169.254.169.254/"
+# public URL → 200
+curl -s -o /dev/null -w "%{http_code}" "http://localhost:8080/api?url=https://example.com"
+```
+
+Expected: `403`, `403`, `403`, `200`
+
+Revert the manual edit to `src/config.inc.php` after testing.
+
+- [ ] **Step 3: Commit**
+
+```bash
+git add web/index.php
+git commit -m "feat: opt-in SSRF protection via BLOCK_PRIVATE_IPS env var"
+```
+
+---
+
+## Final state of `web/index.php` `case 'api':` block
+
+For reference, the complete block after all tasks:
+
+```php
+    case 'api':
+        if (defined('API_KEY') && API_KEY !== '') {
+            $provided = $_SERVER['HTTP_X_API_KEY'] ?? $_REQUEST['key'] ?? '';
+            if ($provided !== API_KEY) {
+                header('HTTP/1.0 401 Unauthorized');
+                echo 'Invalid or missing API key';
+                exit;
+            }
+        }
+
+        $target = substr($_SERVER['REQUEST_URI'], 5);
+        if (!$target || !filter_var($target, FILTER_VALIDATE_URL))
+            $target = $_REQUEST['url'];
+        if (!filter_var($target, FILTER_VALIDATE_URL)) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Invalid URL';
+            exit;
+        }
+        $ip = getUserIP();
+
+        $viewport = $_REQUEST['viewport'] ?: '1024x768';
+        if (!preg_match('/^\d+x\d+$/', $viewport)) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Invalid viewport format. Use WIDTHxHEIGHT (e.g., 1024x768)';
+            exit;
+        }
+        $vpParts = array_map('intval', explode('x', $viewport));
+        if ($vpParts[0] > 3840 || $vpParts[1] > 2160) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Viewport exceeds maximum (3840x2160)';
+            exit;
+        }
+
+        $js = $_REQUEST['js'] == 'false' ? false : true;
+
+        if (defined('BLOCK_PRIVATE_IPS') && BLOCK_PRIVATE_IPS) {
+            $host = parse_url($target, PHP_URL_HOST);
+            if (filter_var($host, FILTER_VALIDATE_IP)) {
+                $resolvedIp = $host;
+            } else {
+                $resolvedIp = gethostbyname($host);
+                if ($resolvedIp === $host) {
+                    header('HTTP/1.0 403 Forbidden');
+                    echo 'URL not allowed';
+                    exit;
+                }
+            }
+            if (isPrivateIP($resolvedIp)) {
+                header('HTTP/1.0 403 Forbidden');
+                echo 'URL not allowed';
+                exit;
+            }
+        }
+
+        $serverUrl = 'http://localhost:4444';
+        $options = new \Facebook\WebDriver\Chrome\ChromeOptions();
+        $options->addArguments(['--headless', '--disable-gpu', '--no-sandbox', '--disable-dev-shm-usage']);
+
+        $capabilities = DesiredCapabilities::chrome();
+        $capabilities->setCapability(\Facebook\WebDriver\Chrome\ChromeOptions::CAPABILITY, $options);
+
+        if (!$js)
+            $capabilities->setCapability('javascriptEnabled', false);
+
+        $driver = null;
+        $error = null;
+        try {
+            $driver = RemoteWebDriver::create($serverUrl, $capabilities, 30000, 60000);
+            $driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension($vpParts[0], $vpParts[1]));
+            $driver->get($target);
+            $driver->executeScript('document.body.style.overflow = "hidden";');
+
+            addToLog("$ip\tRequested $target with viewport $viewport and js " . ($js ? 'enabled' : 'disabled'));
+
+            $screenshot = $driver->takeScreenshot();
+            header('Content-Type: image/png');
+            header('Content-Length: ' . strlen($screenshot));
+            echo $screenshot;
+        } catch (Exception $e) {
+            $error = $e->getMessage();
+            addToLog("$ip\tRequested $target but resulted in error:\t" . $error);
+        } finally {
+            if ($driver instanceof RemoteWebDriver) {
+                try { $driver->quit(); } catch (Exception $q) {}
+            }
+        }
+        if ($error !== null) {
+            header('HTTP/1.0 500 Internal Server Error');
+            echo 'Screenshot failed';
+        }
+
+        break;
+```

docs/superpowers/specs/2026-04-20-security-hardening-design.md

@@ -0,0 +1,169 @@
+# http2pic Security Hardening & Bug Fixes
+
+**Date:** 2026-04-20
+**Status:** Approved
+
+## Summary
+
+Harden http2pic against abuse and fix two correctness bugs. All changes target `web/index.php`, `src/helpers.php`, `docker/start.sh`, and `docker-compose.yml`. No new dependencies.
+
+---
+
+## 1. Config & Environment
+
+Two new optional env vars. Both default to off so existing deployments are unaffected.
+
+| Var | Default | Effect |
+|-----|---------|--------|
+| `API_KEY` | `""` | If non-empty, all `/api` requests must supply it |
+| `BLOCK_PRIVATE_IPS` | `false` | If `true`, rejects URLs that resolve to private/loopback/metadata IPs |
+
+`docker/start.sh` writes both into `src/config.inc.php` alongside the existing `URL` define:
+
+```php
+define('API_KEY', '${API_KEY:-}');
+define('BLOCK_PRIVATE_IPS', ${BLOCK_PRIVATE_IPS:-false});
+```
+
+`docker-compose.yml` gets commented-out examples:
+
+```yaml
+environment:
+  - URL=http://localhost:8080
+  # - API_KEY=your-secret-key       # if set, all /api requests must provide it
+  # - BLOCK_PRIVATE_IPS=true        # block LAN/loopback/metadata IPs (safe default for public hosting)
+```
+
+---
+
+## 2. API Key Authentication
+
+Checked at the top of `case 'api':`, before any Chrome work.
+
+**Logic:** if `API_KEY` is defined and non-empty, require the key. Accept it from either:
+- HTTP header: `X-API-Key: <key>`
+- Query param: `?key=<key>`
+
+Header takes priority. Returns `401 Unauthorized` if missing or wrong.
+
+```php
+if (defined('API_KEY') && API_KEY !== '') {
+    $provided = $_SERVER['HTTP_X_API_KEY'] ?? $_REQUEST['key'] ?? '';
+    if ($provided !== API_KEY) {
+        header('HTTP/1.0 401 Unauthorized');
+        echo 'Invalid or missing API key';
+        exit;
+    }
+}
+```
+
+**Usage examples (to add to README):**
+
+```bash
+# via header (preferred — not logged in access logs)
+curl -H "X-API-Key: secret" "http://host/api?url=https://example.com"
+
+# via query param (simpler for browser use)
+curl "http://host/api?key=secret&url=https://example.com"
+```
+
+---
+
+## 3. SSRF Protection (opt-in)
+
+Active only when `BLOCK_PRIVATE_IPS=true`. Runs after URL format validation, before spawning Chrome.
+
+**Flow:**
+1. Extract hostname from target URL via `parse_url()`
+2. Resolve to IP via `gethostbyname()`
+3. If resolution fails (returns same string) or IP is private → `403 Forbidden`
+
+**Blocked ranges:** `127.0.0.0/8`, `10.0.0.0/8`, `172.16.0.0/12`, `192.168.0.0/16`, `169.254.0.0/16` (AWS/cloud metadata).
+
+Implementation uses `ip2long()` + bitmask checks — no regex.
+
+A new `isPrivateIP(string $ip): bool` helper lives in `src/helpers.php`.
+
+Returns `403 Forbidden` with body `URL not allowed` (no detail about why, to avoid enumeration).
+
+---
+
+## 4. Bug Fixes
+
+### 4a. RemoteWebDriver request timeout (60ms → 60s)
+
+```php
+// before
+RemoteWebDriver::create($serverUrl, $capabilities, 30000, 60);
+// after
+RemoteWebDriver::create($serverUrl, $capabilities, 30000, 60000);
+```
+
+4th param is milliseconds. `60` = 60ms, effectively zero. Fix to `60000`.
+
+### 4b. Viewport set before page load
+
+Move `window()->setSize()` to before `driver->get()` so the page loads at the correct viewport from the start. This ensures responsive layouts and media queries fire at the intended dimensions.
+
+---
+
+## 5. Security Hardening (always applied)
+
+### 5a. Viewport dimension cap
+
+After format validation, reject dimensions exceeding 3840×2160:
+
+```php
+$parts = explode('x', $viewport);
+if ((int)$parts[0] > 3840 || (int)$parts[1] > 2160) {
+    header('HTTP/1.0 400 Bad Request');
+    echo 'Viewport exceeds maximum (3840x2160)';
+    exit;
+}
+```
+
+Prevents memory exhaustion via `99999x99999` requests.
+
+### 5b. Generic error responses
+
+Catch block logs full exception message but returns only:
+
+```
+Screenshot failed
+```
+
+Stops ChromeDriver internals (internal hostnames, paths, stack traces) leaking to clients.
+
+### 5c. Log injection fix
+
+`addToLog()` strips `\n`, `\r`, `\t` from data before writing:
+
+```php
+$data = str_replace(["\n", "\r", "\t"], ' ', $data);
+```
+
+Prevents crafted URLs/IPs from injecting fake log entries.
+
+### 5d. getUserIP() fix
+
+Drop `HTTP_CLIENT_IP` (client-controlled, untrustworthy). Take only the first IP from `X-Forwarded-For` (Caddy appends subsequent hops; first entry is the real client):
+
+```php
+function getUserIP() {
+    if (!empty($_SERVER['HTTP_X_FORWARDED_FOR']))
+        return trim(explode(',', $_SERVER['HTTP_X_FORWARDED_FOR'])[0]);
+    return $_SERVER['REMOTE_ADDR'];
+}
+```
+
+---
+
+## Files Changed
+
+| File | Changes |
+|------|---------|
+| `web/index.php` | API key check, SSRF check, viewport cap, viewport-before-get, timeout fix, generic errors |
+| `src/helpers.php` | `isPrivateIP()` helper, log injection fix, `getUserIP()` fix |
+| `docker/start.sh` | Write `API_KEY` and `BLOCK_PRIVATE_IPS` into config |
+| `docker-compose.yml` | Commented env var examples |
+| `src/config.inc.php` | (generated) gets two new defines |

src/helpers.php

@@ -11,25 +11,40 @@ function renderTemplate($templatename,$variables=[],$basepath=ROOT.'/src')
         include($basepath . DS . 'templates' . DS . $templatename);
     $rendered = ob_get_contents();
     ob_end_clean();
-
     return $rendered;
 }
 
-function addToLog($data)
+function addToLog(string $data): void
 {
+    $data = str_replace(["\n", "\r", "\t"], ' ', $data);
     $fp = fopen(ROOT . DS . 'logs' . DS . 'app.log', 'a');
     fwrite($fp, date("d.m.y H:i") . "\t" . $data . "\n");
     fclose($fp);
 }
 
-function getUserIP()
+function getUserIP(): string
 {
-    if (!empty($_SERVER['HTTP_CLIENT_IP'])) {
+    if (!empty($_SERVER['HTTP_X_FORWARDED_FOR']))
-        $ip = $_SERVER['HTTP_CLIENT_IP'];
+        return trim(explode(',', $_SERVER['HTTP_X_FORWARDED_FOR'])[0]);
-    } elseif (!empty($_SERVER['HTTP_X_FORWARDED_FOR'])) {
+    return $_SERVER['REMOTE_ADDR'];
-        $ip = $_SERVER['HTTP_X_FORWARDED_FOR'];
-    } else {
-        $ip = $_SERVER['REMOTE_ADDR'];
 }
-    return $ip;
+
+function isPrivateIP(string $ip): bool
+{
+    if (filter_var($ip, FILTER_VALIDATE_IP) === false) return true;
+
+    $long = ip2long($ip);
+    if ($long === false) return true;
+
+    foreach ([
+        [ip2long('0.0.0.0'),     0xFF000000], // 0.0.0.0/8    unspecified (routes to loopback on Linux)
+        [ip2long('127.0.0.0'),   0xFF000000], // 127.0.0.0/8  loopback
+        [ip2long('10.0.0.0'),    0xFF000000], // 10.0.0.0/8   RFC1918
+        [ip2long('172.16.0.0'),  0xFFF00000], // 172.16.0.0/12 RFC1918
+        [ip2long('192.168.0.0'), 0xFFFF0000], // 192.168.0.0/16 RFC1918
+        [ip2long('169.254.0.0'), 0xFFFF0000], // 169.254.0.0/16 link-local/metadata
+    ] as [$base, $mask]) {
+        if (($long & $mask) === ($base & $mask)) return true;
+    }
+    return false;
 }

src/http2pic.class.php

@@ -104,7 +104,7 @@ class http2pic
 	
 		//validate URL and check if exists
 		if ($this->isBase64($this->params['url']))
-			$this->params['url'] = base64_decode($url);
+			$this->params['url'] = base64_decode($this->params['url']);
 		else
 			$this->params['url'] = rawurldecode($_GET['url']);
 		
@@ -171,18 +171,24 @@ class http2pic
 			$cmd.=' -f png';
 		
 		//add url to cmd
-		$cmd.=' \''.addslashes($this->params['url']).'\'';
+		$cmd.=' '.escapeshellarg($this->params['url']);
 
 		//add storage path to cmd
 		$cmd.=' '.escapeshellarg($this->params['file']);
 
 		$cmd.=' --wait-for-network-idle';
 
-		var_dump($cmd);
+		$output = [];
-			
+		$rc = 0;
-		$cmd = escapeshellcmd($cmd);
+		exec($cmd . ' 2>&1', $output, $rc);
-		shell_exec($cmd);
 		$this->params['cmd'] = $cmd;
+		if ($rc !== 0) {
+			$this->params['render_error'] = implode("\n", $output);
+			header('Content-Type: image/jpeg');
+			$result = imagecreatefromjpeg($this->params['onfail']);
+			imagejpeg($result, NULL, 100);
+			return $cmd;
+		}
 
 		$this->postRender();
 
@@ -273,6 +279,8 @@ class http2pic
 	{
 		$ch = curl_init($url);
 		curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
+		curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 10);
+		curl_setopt($ch, CURLOPT_TIMEOUT, 15);
 		if(curl_exec($ch) != false){
 			//We were able to connect to a webserver, what did it return?
 			$code = curl_getinfo($ch, CURLINFO_HTTP_CODE);

src/templates/index.html.php

@@ -102,6 +102,18 @@
                                 <td>Allows you to enable/disable JavaScript in the rendered Website. Default value: yes</td>
                                 <td>js=false</td>
                             </tr>
+                            <tr>
+                                <td>fullpage</td>
+                                <td>true|false</td>
+                                <td>Capture the full page height instead of just the viewport. Useful for long pages and LLM analysis. Default: false</td>
+                                <td>fullpage=true</td>
+                            </tr>
+                            <tr>
+                                <td>maxheight</td>
+                                <td>1&ndash;30000</td>
+                                <td>Maximum pixel height when using fullpage=true. Default: 15000</td>
+                                <td>maxheight=20000</td>
+                            </tr>
                         </tbody>
                     </table>
                 </section>

tests/test_helpers.php

@@ -0,0 +1,50 @@
+<?php
+define('DS', DIRECTORY_SEPARATOR);
+define('ROOT', dirname(__FILE__) . '/..');
+require_once ROOT . '/src/helpers.php';
+
+$passed = 0;
+$failed = 0;
+
+function check(bool $condition, string $label): void {
+    global $passed, $failed;
+    if ($condition) { echo "PASS: $label\n"; $passed++; }
+    else            { echo "FAIL: $label\n"; $failed++; }
+}
+
+// --- isPrivateIP ---
+check(isPrivateIP('127.0.0.1'),        'loopback 127.0.0.1');
+check(isPrivateIP('127.255.255.255'),   'loopback 127.255.255.255');
+check(isPrivateIP('10.0.0.1'),          '10.x private');
+check(isPrivateIP('10.255.255.255'),    '10.255.x private');
+check(isPrivateIP('172.16.0.1'),        '172.16.x private');
+check(isPrivateIP('172.31.255.255'),    '172.31.x private');
+check(!isPrivateIP('172.15.255.255'),   '172.15.x is public');
+check(!isPrivateIP('172.32.0.0'),       '172.32.x is public');
+check(isPrivateIP('192.168.0.1'),       '192.168.x private');
+check(isPrivateIP('169.254.169.254'),   'AWS metadata IP');
+check(!isPrivateIP('8.8.8.8'),          'Google DNS is public');
+check(!isPrivateIP('93.184.216.34'),    'example.com IP is public');
+check(isPrivateIP('not-an-ip'),         'unparseable IP blocked');
+check(isPrivateIP('0.0.0.0'),           '0.0.0.0 blocked (routes to loopback)');
+
+// --- getUserIP ---
+unset($_SERVER['HTTP_CLIENT_IP']);
+unset($_SERVER['HTTP_X_FORWARDED_FOR']);
+$_SERVER['REMOTE_ADDR'] = '5.6.7.8';
+check(getUserIP() === '5.6.7.8', 'getUserIP falls back to REMOTE_ADDR');
+
+$_SERVER['HTTP_X_FORWARDED_FOR'] = '1.2.3.4, 10.0.0.1, 172.16.0.5';
+check(getUserIP() === '1.2.3.4', 'getUserIP takes first X-Forwarded-For IP');
+
+$_SERVER['HTTP_X_FORWARDED_FOR'] = '  9.9.9.9  ';
+check(getUserIP() === '9.9.9.9', 'getUserIP trims whitespace');
+
+$_SERVER['HTTP_CLIENT_IP'] = '6.6.6.6';
+$_SERVER['HTTP_X_FORWARDED_FOR'] = '1.1.1.1';
+check(getUserIP() === '1.1.1.1', 'HTTP_CLIENT_IP ignored, X-Forwarded-For used');
+unset($_SERVER['HTTP_CLIENT_IP']);
+unset($_SERVER['HTTP_X_FORWARDED_FOR']);
+
+echo "\n$passed passed, $failed failed\n";
+exit($failed > 0 ? 1 : 0);

web/index.php

@@ -6,6 +6,10 @@ use Facebook\WebDriver\Remote\DesiredCapabilities;
 define('DS', DIRECTORY_SEPARATOR);
 define('ROOT', dirname(__FILE__) . DS . '..');
 
+// increase PHP timeout - rendering can take 60s+
+set_time_limit(120);
+ignore_user_abort(true);
+
 require_once(ROOT . DS . 'src' . DS . 'config.inc.php');
 require_once(ROOT . DS . 'src' . DS . 'helpers.php');
 require_once(ROOT . DS . 'src' . DS . 'http2pic.class.php');
@@ -20,6 +24,14 @@ if (php_sapi_name() == 'cli-server' && file_exists(ROOT . DS . 'web' . DS . impl
 
 switch ($url[0]) {
     case 'api':
+        if (defined('API_KEY') && API_KEY !== '') {
+            $provided = $_SERVER['HTTP_X_API_KEY'] ?? $_REQUEST['key'] ?? '';
+            if (!hash_equals(API_KEY, $provided)) {
+                header('HTTP/1.0 401 Unauthorized');
+                echo 'Invalid or missing API key';
+                exit;
+            }
+        }
         $target = substr($_SERVER['REQUEST_URI'], 5);
         if (!$target || !filter_var($target, FILTER_VALIDATE_URL))
             $target = $_REQUEST['url'];
@@ -28,6 +40,12 @@ switch ($url[0]) {
             echo 'Invalid URL';
             exit;
         }
+        $scheme = strtolower(parse_url($target, PHP_URL_SCHEME) ?? '');
+        if (!in_array($scheme, ['http', 'https'], true)) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Invalid URL';
+            exit;
+        }
         $ip = getUserIP();
 
         $viewport = $_REQUEST['viewport'] ?: '1024x768';
@@ -36,8 +54,46 @@ switch ($url[0]) {
             echo 'Invalid viewport format. Use WIDTHxHEIGHT (e.g., 1024x768)';
             exit;
         }
+        $vpParts = array_map('intval', explode('x', $viewport));
+        if ($vpParts[0] < 1 || $vpParts[1] < 1 || $vpParts[0] > 3840 || $vpParts[1] > 2160) {
+            header('HTTP/1.0 400 Bad Request');
+            echo 'Viewport dimensions must be between 1x1 and 3840x2160';
+            exit;
+        }
+
         $js = $_REQUEST['js'] == 'false' ? false : true;
 
+        $fullpage = isset($_REQUEST['fullpage']) && $_REQUEST['fullpage'] === 'true';
+        $maxheight = 15000;
+        if (isset($_REQUEST['maxheight'])) {
+            $mh = intval($_REQUEST['maxheight']);
+            if ($mh < 1 || $mh > 30000) {
+                header('HTTP/1.0 400 Bad Request');
+                echo 'maxheight must be between 1 and 30000';
+                exit;
+            }
+            $maxheight = $mh;
+        }
+
+        if (defined('BLOCK_PRIVATE_IPS') && BLOCK_PRIVATE_IPS) {
+            $host = parse_url($target, PHP_URL_HOST);
+            if (filter_var($host, FILTER_VALIDATE_IP)) {
+                $resolvedIp = $host;
+            } else {
+                $resolvedIp = gethostbyname($host);
+                if ($resolvedIp === $host) {
+                    header('HTTP/1.0 403 Forbidden');
+                    echo 'URL not allowed';
+                    exit;
+                }
+            }
+            if (isPrivateIP($resolvedIp)) {
+                header('HTTP/1.0 403 Forbidden');
+                echo 'URL not allowed';
+                exit;
+            }
+        }
+
         $serverUrl = 'http://localhost:4444';
         $options = new \Facebook\WebDriver\Chrome\ChromeOptions();
         $options->addArguments(['--headless', '--disable-gpu', '--no-sandbox', '--disable-dev-shm-usage']);
@@ -45,40 +101,45 @@ switch ($url[0]) {
         $capabilities = DesiredCapabilities::chrome();
         $capabilities->setCapability(\Facebook\WebDriver\Chrome\ChromeOptions::CAPABILITY, $options);
 
-        //disable javascript if $js is false
         if (!$js)
             $capabilities->setCapability('javascriptEnabled', false);
 
-
+        $driver = null;
+        $error = null;
         try {
-            $driver = RemoteWebDriver::create($serverUrl, $capabilities);
+            $driver = RemoteWebDriver::create($serverUrl, $capabilities, 30000, 60000);
+            $driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension($vpParts[0], $vpParts[1]));
             $driver->get($target);
-            //hide scroll bars
-            $driver->executeScript('document.body.style.overflow = "hidden";');
 
-            //set screenshot size to 1920x1080
+            if ($fullpage) {
-            //$driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension(1024, 768));
+                $fullH = (int)$driver->executeScript('return Math.max(document.body.scrollHeight, document.documentElement.scrollHeight)');
-            //if $viewport is set, set window size
+                $cappedH = min($fullH, $maxheight);
-            if ($viewport) {
+                if ($cappedH < $fullH) {
-                $viewport = explode('x', $viewport);
+                    addToLog($ip . ' Full-page height capped at ' . $maxheight . 'px (actual: ' . $fullH . 'px) for ' . $target);
-                $driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension($viewport[0], $viewport[1]));
+                }
+                $driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension($vpParts[0], $cappedH));
             } else {
-                $viewport = [1024, 768];
+                $driver->executeScript('document.body.style.overflow = "hidden";');
-                $driver->manage()->window()->setSize(new \Facebook\WebDriver\WebDriverDimension($viewport[0], $viewport[1]));
-            }
-            $viewportLabel = is_array($viewport) ? implode('x', $viewport) : (string) $viewport;
-            addToLog("$ip\tRequested $target with viewport " . $viewportLabel . " and js " . ($js ? 'enabled' : 'disabled'));
-        } catch (Exception $e) {
-            header('HTTP/1.0 500 Internal Server Error');
-            addToLog("$ip\tRequested $target but resulted in error:\t" . $e->getMessage());
-            echo 'Error: ' . $e->getMessage();
-            exit;
             }
 
-        // take screenshot and save to file
+            addToLog($ip . ' Requested ' . $target . ' viewport=' . $viewport . ' js=' . ($js ? 'enabled' : 'disabled') . ($fullpage ? ' fullpage=true' : ''));
-        //header for png
+
+            $screenshot = $driver->takeScreenshot();
             header('Content-Type: image/png');
-        echo $driver->takeScreenshot();
+            header('Content-Length: ' . strlen($screenshot));
+            echo $screenshot;
+        } catch (Exception $e) {
+            $error = $e->getMessage();
+            addToLog($ip . ' Error requesting ' . $target . ': ' . $error);
+        } finally {
+            if ($driver instanceof RemoteWebDriver) {
+                try { $driver->quit(); } catch (Exception $q) {}
+            }
+        }
+        if ($error !== null) {
+            header('HTTP/1.0 500 Internal Server Error');
+            echo 'Screenshot failed';
+        }
 
         break;
     default: