Improved SVG cleanup code

Aleksander Machniak

2016-01-16 ed1d212ae2daea5e4bd043417610177093e99f19

 program/steps/settings/save_identity.inc

@@ -1,12 +1,15 @@
<?php

/*
/**
 +-----------------------------------------------------------------------+
 | program/steps/settings/save_identity.inc                              |
 |                                                                       |
 | This file is part of the RoundCube Webmail client                     |
 | Copyright (C) 2005, RoundCube Dev. - Switzerland                      |
 | Licensed under the GNU GPL                                            |
 | This file is part of the Roundcube Webmail client                     |
 | Copyright (C) 2005-2013, The Roundcube Dev Team                       |
 |                                                                       |
 | Licensed under the GNU General Public License version 3 or            |
 | any later version with exceptions for skins & plugins.                |
 | See the README file for a full license statement.                     |
 |                                                                       |
 | PURPOSE:                                                              |
 |   Save an identity record or to add a new one                         |
@@ -14,122 +17,250 @@
 +-----------------------------------------------------------------------+
 | Author: Thomas Bruederli <roundcube@gmail.com>                        |
 +-----------------------------------------------------------------------+

 $Id$

*/

$a_save_cols = array('name', 'email', 'organization', 'reply-to', 'bcc', 'default');
define('IDENTITIES_LEVEL', intval($RCMAIL->config->get('identities_level', 0)));

$a_save_cols    = array('name', 'email', 'organization', 'reply-to', 'bcc', 'standard', 'signature', 'html_signature');
$a_boolean_cols = array('standard', 'html_signature');
$updated = $default_id = false;

// check input
if (empty($_POST['_name']) || empty($_POST['_email']))
  {
  show_message('formincomplete', 'warning');
  rcmail_overwrite_action('edit-identitiy');
  return;
  }
if (empty($_POST['_email']) && (IDENTITIES_LEVEL == 0 || IDENTITIES_LEVEL == 2)) {
    $OUTPUT->show_message('noemailwarning', 'warning');
    $RCMAIL->overwrite_action('edit-identity');
    return;
}

$save_data = array();
foreach ($a_save_cols as $col) {
    $fname = '_'.$col;
    if (isset($_POST[$fname])) {
        $save_data[$col] = rcube_utils::get_input_value($fname, rcube_utils::INPUT_POST, true);
    }
}

// set "off" values for checkboxes that were not checked, and therefore
// not included in the POST body.
foreach ($a_boolean_cols as $col) {
    $fname = '_' . $col;
    if (!isset($_POST[$fname])) {
        $save_data[$col] = 0;
    }
}

// make the identity a "default" if only one identity is allowed
if (IDENTITIES_LEVEL > 1) {
    $save_data['standard'] = 1;
}

// unset email address if user has no rights to change it
if (IDENTITIES_LEVEL == 1 || IDENTITIES_LEVEL == 3) {
    unset($save_data['email']);
}
// unset all fields except signature
else if (IDENTITIES_LEVEL == 4) {
    foreach ($save_data as $idx => $value) {
        if ($idx != 'signature' && $idx != 'html_signature') {
            unset($save_data[$idx]);
        }
    }
}

// Validate e-mail addresses
$email_checks = array(rcube_utils::idn_to_ascii($save_data['email']));
foreach (array('reply-to', 'bcc') as $item) {
    foreach (rcube_mime::decode_address_list($save_data[$item], null, false) as $rcpt) {
        $email_checks[] = rcube_utils::idn_to_ascii($rcpt['mailto']);
    }
}

foreach ($email_checks as $email) {
    if ($email && !rcube_utils::check_email($email)) {
        // show error message
        $OUTPUT->show_message('emailformaterror', 'error', array('email' => rcube_utils::idn_to_utf8($email)), false);
        $RCMAIL->overwrite_action('edit-identity');
        return;
    }
}

if (!empty($save_data['signature']) && !empty($save_data['html_signature'])) {
    // replace uploaded images with data URIs
    $save_data['signature'] = rcmail_attach_images($save_data['signature']);

    // XSS protection in HTML signature (#1489251)
    $save_data['signature'] = rcmail_wash_html($save_data['signature']);

    // clear POST data of signature, we want to use safe content
    // when the form is displayed again
    unset($_POST['_signature']);
}

// update an existing contact
if ($_POST['_iid'])
  {
  $a_write_sql = array();
if ($_POST['_iid']) {
    $iid = rcube_utils::get_input_value('_iid', rcube_utils::INPUT_POST);

  foreach ($a_save_cols as $col)
    {
    $fname = '_'.$col;
    if (!isset($_POST[$fname]))
      continue;
    if (in_array(IDENTITIES_LEVEL, array(1,3,4))) {
        // merge with old identity data, fixes #1488834
        $identity  = $RCMAIL->user->get_identity($iid);
        $save_data = array_merge($identity, $save_data);

    $a_write_sql[] = sprintf("%s=%s", $DB->quoteIdentifier($col), $DB->quote(strip_tags($_POST[$fname])));
        unset($save_data['changed'], $save_data['del'], $save_data['user_id'], $save_data['identity_id']);
    }

  if (sizeof($a_write_sql))
    {
    $DB->query("UPDATE ".get_table_name('identities')."
                SET ".join(', ', $a_write_sql)."
                WHERE  identity_id=?
                AND    user_id=?
                AND    del<>'1'",
                $_POST['_iid'],
                $_SESSION['user_id']);
                       
    $updated = $DB->affected_rows();
    }
       
  if ($updated)
    {
    show_message('successfullysaved', 'confirmation');
    $plugin = $RCMAIL->plugins->exec_hook('identity_update', array('id' => $iid, 'record' => $save_data));
    $save_data = $plugin['record'];

    // mark all other identities as 'not-default'
    $DB->query("UPDATE ".get_table_name('identities')."
                SET ".$DB->quoteIdentifier('default')."='0'
                WHERE  user_id=?
                AND    identity_id<>?
                AND    del<>'1'",
                $_SESSION['user_id'],
                $_POST['_iid']);
    
    if ($_POST['_framed'])
      {
      // update the changed col in list
      // ...      
      }
    if ($save_data['email']) {
        $save_data['email'] = rcube_utils::idn_to_ascii($save_data['email']);
    }
  else
    {
    // show error message
    show_message('errorsaving', 'error');
    rcmail_overwrite_action('edit-identitiy');
    }
  }

// insert a new contact
else
  {
  $a_insert_cols = $a_insert_values = array();
    if (!$plugin['abort'])
        $updated = $RCMAIL->user->update_identity($iid, $save_data);
    else
        $updated = $plugin['result'];

  foreach ($a_save_cols as $col)
    {
    $fname = '_'.$col;
    if (!isset($_POST[$fname]))
      continue;
    
    $a_insert_cols[] = $DB->quoteIdentifier($col);
    $a_insert_values[] = $DB->quote(strip_tags($_POST[$fname]));
    }
    
  if (sizeof($a_insert_cols))
    {
    $DB->query("INSERT INTO ".get_table_name('identities')."
                (user_id, ".join(', ', $a_insert_cols).")
                VALUES (?, ".join(', ', $a_insert_values).")",
                $_SESSION['user_id']);
                       
    $insert_id = $DB->insert_id();
    }
    
  if ($insert_id)
    {
    $_GET['_iid'] = $insert_id;
    if ($updated) {
        $OUTPUT->show_message('successfullysaved', 'confirmation');

    if ($_POST['_framed'])
      {
      // add contact row or jump to the page where it should appear
      // ....
      }
    }
  else
    {
    // show error message
    show_message('errorsaving', 'error');
    rcmail_overwrite_action('edit-identitiy');
    }
  }
        if (!empty($save_data['standard'])) {
            $default_id = $iid;
        }

        if ($_POST['_framed']) {
            // update the changed col in list
            $name = $save_data['name'] . ' <' . rcube_utils::idn_to_utf8($save_data['email']) .'>';
            $OUTPUT->command('parent.update_identity_row', $iid, rcube::Q(trim($name)));
        }
    }
    else {
        // show error message
        $OUTPUT->show_message($plugin['message'] ?: 'errorsaving', 'error', null, false);
        $RCMAIL->overwrite_action('edit-identity');
        return;
    }
}
// insert a new identity record
else if (IDENTITIES_LEVEL < 2) {
    if (IDENTITIES_LEVEL == 1) {
        $save_data['email'] = $RCMAIL->get_user_email();
    }

    $plugin = $RCMAIL->plugins->exec_hook('identity_create', array('record' => $save_data));
    $save_data = $plugin['record'];

    if ($save_data['email']) {
        $save_data['email'] = rcube_utils::idn_to_ascii($save_data['email']);
    }

    if (!$plugin['abort'])
        $insert_id = $save_data['email'] ? $RCMAIL->user->insert_identity($save_data) : null;
    else
        $insert_id = $plugin['result'];

    if ($insert_id) {
        $RCMAIL->plugins->exec_hook('identity_create_after', array('id' => $insert_id, 'record' => $save_data));

        $OUTPUT->show_message('successfullysaved', 'confirmation', null, false);

        $_GET['_iid'] = $insert_id;

        if (!empty($save_data['standard'])) {
            $default_id = $insert_id;
        }

        if ($_POST['_framed']) {
            // add a new row to the list
            $name = $save_data['name'] . ' <' . rcube_utils::idn_to_utf8($save_data['email']) .'>';
            $OUTPUT->command('parent.update_identity_row', $insert_id, rcube::Q(trim($name)), true);
        }
    }
    else {
        // show error message
        $OUTPUT->show_message($plugin['message'] ?: 'errorsaving', 'error', null, false);
        $RCMAIL->overwrite_action('edit-identity');
        return;
    }
}
else {
    $OUTPUT->show_message('opnotpermitted', 'error');
}

// mark all other identities as 'not-default'
if ($default_id) {
    $RCMAIL->user->set_default($default_id);
}

// go to next step
rcmail_overwrite_action($_POST['_framed'] ? 'edit-identitiy' : 'identities');
if (!empty($_REQUEST['_framed'])) {
    $RCMAIL->overwrite_action('edit-identity');
}
else {
    $RCMAIL->overwrite_action('identities');
}

?>

/**
 * Attach uploaded images into signature as data URIs
 */
function rcmail_attach_images($html)
{
    global $RCMAIL;

    $offset = 0;
    $regexp = '/\s(poster|src)\s*=\s*[\'"]*\S+upload-display\S+file=rcmfile(\w+)[\s\'"]*/';

    while (preg_match($regexp, $html, $matches, 0, $offset)) {
        $file_id  = $matches[2];
        $data_uri = ' ';

        if ($file_id && ($file = $_SESSION['identity']['files'][$file_id])) {
            $file = $RCMAIL->plugins->exec_hook('attachment_get', $file);

            $data_uri .= 'src="data:' . $file['mimetype'] . ';base64,';
            $data_uri .= base64_encode($file['data'] ?: file_get_contents($file['path']));
            $data_uri .= '" ';
        }

        $html    = str_replace($matches[0], $data_uri, $html);
        $offset += strlen($data_uri) - strlen($matches[0]) + 1;
    }

    return $html;
}

/**
 * Sanity checks/cleanups on HTML body of signature
 */
function rcmail_wash_html($html)
{
    // Add header with charset spec., washtml cannot work without that
    $html = '<html><head>'
        . '<meta http-equiv="Content-Type" content="text/html; charset='.RCUBE_CHARSET.'" />'
        . '</head><body>' . $html . '</body></html>';

    // clean HTML with washhtml by Frederic Motte
    $wash_opts = array(
        'show_washed'   => false,
        'allow_remote'  => 1,
        'charset'       => RCUBE_CHARSET,
        'html_elements' => array('body', 'link'),
        'html_attribs'  => array('rel', 'type'),
    );

    // initialize HTML washer
    $washer = new rcube_washtml($wash_opts);

    //$washer->add_callback('form', 'rcmail_washtml_callback');
    //$washer->add_callback('style', 'rcmail_washtml_callback');

    // Remove non-UTF8 characters (#1487813)
    $html = rcube_charset::clean($html);

    $html = $washer->wash($html);

    // remove unwanted comments and tags (produced by washtml)
    $html = preg_replace(array('/<!--[^>]+-->/', '/<\/?body>/'), '', $html);

    return $html;
}