<?php
# The home-server project produces a multi-purpose setup using Ansible.
# Copyright © 2018 Y. Gablin, under the GPL-3.0-or-later license.
# Full licensing information in the LICENSE file, or gnu.org/licences/gpl-3.0.txt if the file is missing.

/*
  PHP script to handle file uploads and downloads for Prosody's mod_http_upload_external

  Tested with Apache 2.2+ and PHP 5.3+

  ** Why this script?

  This script only allows uploads that have been authorized by mod_http_upload_external. It
  attempts to make the upload/download as safe as possible, considering that there are *many*
  security concerns involved with allowing arbitrary file upload/download on a web server.

  With that said, I do not consider myself a PHP developer, and at the time of writing, this
  code has had no external review. Use it at your own risk. I make no claims that this code
  is secure.

  ** How to use?

  Drop this file somewhere it will be served by your web server. Edit the config options below.

  In Prosody set:

    http_upload_external_base_url = "https://your.example.com/path/to/share.php/"
    http_upload_external_secret = "this is your secret string"

  ** License

  (C) 2016 Matthew Wild <mwild1@gmail.com>

  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.

*/

/*\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\*/
/*         CONFIGURATION OPTIONS                   */
/*\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\*/

/* Change this to a directory that is writable by your web server, but is outside your web root */
$CONFIG_STORE_DIR = '/var/lib/prosody/http_upload';

/* This must be the same as 'http_upload_external_secret' that you set in Prosody's config file */
$CONFIG_SECRET = '{{xmpp_upload_secret}}';

/* Debug your server setup by replacing `false` with `true` on this line */
$DEBUG = false;

/* For people who need options to tweak that they don't understand... here you are */
$CONFIG_CHUNK_SIZE = 4096;

/*\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\*/
/*         END OF CONFIGURATION                    */
/*\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\/\*/

/* Do not edit below this line unless you know what you are doing (spoiler: nobody does) */

if ($DEBUG) {
	error_log('$_SERVER[DOCUMENT_URI]       ='.$_SERVER['DOCUMENT_URI']);
	error_log('$_SERVER[SCRIPT_NAME]        ='.$_SERVER['SCRIPT_NAME']);
	error_log('$_SERVER[REQUEST_METHOD]     ='.$_SERVER['REQUEST_METHOD']);
	error_log('$_SERVER[HTTP_CONTENT_LENGTH]='.$_SERVER['HTTP_CONTENT_LENGTH']);
	error_log('$_GET[v]                     ='.$_GET['v']);
}

// $_SERVER['PATH_INFO'] would serve us better here, but it _might_ be incorrectly set...
$upload_file_name = substr($_SERVER['DOCUMENT_URI'], strlen($_SERVER['SCRIPT_NAME'])+1);
$store_file_name = $CONFIG_STORE_DIR . '/store-' . hash('sha256', $upload_file_name);

$request_method = $_SERVER['REQUEST_METHOD'];

if(array_key_exists('v', $_GET) === TRUE && $request_method === 'PUT') {
	$upload_file_size = array_key_exists('CONTENT_LENGTH', $_SERVER)
		?	$_SERVER['CONTENT_LENGTH']
		:	(array_key_exists('HTTP_CONTENT_LENGTH', $_SERVER)
			?	$_SERVER['HTTP_CONTENT_LENGTH']
			:	getallheaders()['Content-Length']);
	$upload_token = $_GET['v'];

	$calculated_token = hash_hmac('sha256', "$upload_file_name $upload_file_size", $CONFIG_SECRET);
	if($upload_token !== $calculated_token) {
		if ($DEBUG) error_log('EXIT 403 (Content-Length probably wrong)');
		header('HTTP/1.0 403 Forbidden');
		exit;
	}

	/* Open a file for writing */
	$store_file = fopen($store_file_name, 'x');

	if($store_file === FALSE) {
		if ($DEBUG) error_log('EXIT 409 (permissions probably wrong on the storage directory)');
		header('HTTP/1.0 409 Conflict');
		exit;
	}

	/* PUT data comes in on the stdin stream */
	$incoming_data = fopen('php://input', 'r');

	/* Read the data a chunk at a time and write to the file */
	while ($data = fread($incoming_data, $CONFIG_CHUNK_SIZE)) {
  		fwrite($store_file, $data);
	}

	/* Close the streams */
	fclose($incoming_data);
	fclose($store_file);
} else if($request_method === 'GET' || $request_method === 'HEAD') {
	// Send file (using X-Sendfile would be nice here...)
	if(file_exists($store_file_name)) {
		if ($DEBUG) error_log('EXIT 200 (all seems to be fine)');
		header('Content-Disposition: attachment');
		header('Content-Type: '.mime_content_type($store_file_name));
		header('Content-Length: '.filesize($store_file_name));
		if($request_method !== 'HEAD') {
			readfile($store_file_name);
		}
	} else {
		if ($DEBUG) error_log('EXIT 404 (file probably absent or not readable by your web server)');
		header('HTTP/1.0 404 Not Found');
	}
} else {
	if ($DEBUG) error_log('EXIT 400 (REQUEST_METHOD not set, or PUT without a token)');
	header('HTTP/1.0 400 Bad Request');
}

exit;