<?php

/**
 * Filesystem Attachments
 *
 * This is a core plugin which provides basic, filesystem based
 * attachment temporary file handling.  This includes storing
 * attachments of messages currently being composed, writing attachments
 * to disk when drafts with attachments are re-opened and writing
 * attachments to disk for inline display in current html compositions.
 * It also handles uploaded files for other uses, so not only attachments.
 *
 * Developers may wish to extend this class when creating attachment
 * handler plugins:
 *   class myCustom_attachments extends filesystem_attachments { ... }
 *
 * Note for developers: It is plugin's responsibility to care about security.
 * So, e.g. if the plugin is asked about some file path it should check
 * if it's really the storage path of the plugin and not e.g. /etc/passwd.
 * It is done by setting 'status' flag on every plugin hook it uses.
 * Roundcube core will trust the returned path if status=true.
 *
 * @license GNU GPLv3+
 * @author Ziba Scott <ziba@umich.edu>
 * @author Thomas Bruederli <roundcube@gmail.com>
 */
class filesystem_attachments extends rcube_plugin
{
    public $task = '?(?!login).*';
    public $initialized = false;

    #[Override]
    public function init()
    {
        // Find filesystem_attachments-based plugins, we can use only one
        foreach ($this->api->loaded_plugins() as $plugin_name) {
            $plugin = $this->api->get_plugin($plugin_name);
            if (($plugin instanceof self) && $plugin->initialized) {
                rcube::raise_error([
                    'message' => "Can use only one plugin for attachments/file uploads! Using '{$plugin_name}', ignoring others.",
                ], true, false);
                return;
            }
        }

        $this->initialized = true;

        // Save a newly uploaded attachment
        $this->add_hook('attachment_upload', [$this, 'upload']);

        // Save an attachment from a non-upload source (draft or forward)
        $this->add_hook('attachment_save', [$this, 'save']);

        // Remove an attachment from storage
        $this->add_hook('attachment_delete', [$this, 'remove']);

        // When composing an html message, image attachments may be shown
        $this->add_hook('attachment_display', [$this, 'display']);

        // Get the attachment from storage and place it on disk to be sent
        $this->add_hook('attachment_get', [$this, 'get']);

        // Delete all temp files associated with this user
        $this->add_hook('attachments_cleanup', [$this, 'cleanup']);
        $this->add_hook('session_destroy', [$this, 'cleanup']);
    }

    /**
     * Save a newly uploaded attachment
     */
    public function upload($args)
    {
        $args['status'] = false;
        $group = $args['group'];

        // use common temp dir for file uploads
        $tmpfname = rcube_utils::temp_filename('attmnt');

        if (!empty($args['path']) && move_uploaded_file($args['path'], $tmpfname) && file_exists($tmpfname)) {
            $args['id'] = $this->file_id();
            $args['path'] = $tmpfname;
            $args['status'] = true;
            @chmod($tmpfname, 0600);  // set correct permissions (#1488996)
        }

        return $args;
    }

    /**
     * Save an attachment from a non-upload source (draft or forward)
     */
    public function save($args)
    {
        $group = $args['group'];
        $args['status'] = false;

        if (empty($args['path'])) {
            $tmp_path = rcube_utils::temp_filename('attmnt');

            if ($fp = fopen($tmp_path, 'w')) {
                fwrite($fp, $args['data']);
                fclose($fp);
                $args['path'] = $tmp_path;
            } else {
                return $args;
            }
        }

        $args['id'] = $this->file_id();
        $args['status'] = true;

        return $args;
    }

    /**
     * Remove an attachment from storage
     * This is triggered by the remove attachment button on the compose screen
     */
    public function remove($args)
    {
        $args['status'] = $this->verify_path($args['path']) && @unlink($args['path']);
        return $args;
    }

    /**
     * When composing an html message, image attachments may be shown
     * For this plugin, the file is already in place, just check for
     * the existence of the proper metadata
     */
    public function display($args)
    {
        $args['status'] = $this->verify_path($args['path']) && file_exists($args['path']);
        return $args;
    }

    /**
     * This attachment plugin doesn't require any steps to put the file
     * on disk for use. This stub function is kept here to make this
     * class handy as a parent class for other plugins which may need it.
     */
    public function get($args)
    {
        if (!$this->verify_path($args['path'])) {
            $args['path'] = null;
        }

        return $args;
    }

    /**
     * Delete all temp files associated with this user session
     */
    public function cleanup($args)
    {
        $rcube = rcube::get_instance();
        $group = $args['group'] ?? null;

        // @phpstan-ignore-next-line
        foreach ($rcube->list_uploaded_files($group) as $file) {
            if ($file['path'] && $this->verify_path($file['path']) && file_exists($file['path'])) {
                unlink($file['path']);
            }
        }

        return $args;
    }

    protected static function file_id()
    {
        $rcube = rcube::get_instance();
        [$usec, $sec] = explode(' ', microtime());
        $id = preg_replace('/[^0-9]/', '', $rcube->user->ID . $sec . $usec);

        // make sure the ID is really unique (#1489546)
        // @phpstan-ignore-next-line
        while ($rcube->get_uploaded_file($id)) {
            // increment last four characters
            $x = intval(substr($id, -4)) + 1;
            $id = substr($id, 0, -4) . sprintf('%04d', $x > 9999 ? $x - 9999 : $x);
        }

        return $id;
    }

    /**
     * For security we'll always verify the file path stored in session,
     * as session entries can be faked in various ways e.g. #6026.
     * We allow only files in Roundcube temp dir
     */
    protected static function verify_path($path)
    {
        if (empty($path)) {
            return false;
        }

        $rcmail = rcube::get_instance();
        $temp_dir = $rcmail->config->get('temp_dir');
        $file_path = pathinfo($path, \PATHINFO_DIRNAME);

        if ($temp_dir !== $file_path) {
            // When the configured directory is not writable, or out of open_basedir path
            // tempnam() fallbacks to system temp without a warning.
            // We allow that, but we'll let to know the user about the misconfiguration.
            if ($file_path == sys_get_temp_dir()) {
                rcube::raise_error([
                    'message' => "Detected 'temp_dir' change. "
                        . "Access to '{$temp_dir}' restricted by filesystem permissions or open_basedir",
                ], true, false);

                return true;
            }

            rcube::raise_error([
                'message' => sprintf("%s can't read %s (not in temp_dir)",
                    $rcmail->get_user_name(), substr($path, 0, 512)),
            ], true, false);

            return false;
        }

        return true;
    }
}