1 files changed, 10 insertions, 0 deletions
diff --git a/program/include/rcube_message.php b/program/include/rcube_message.php
index 4d3cb6ca9..d810ef3fa 100644
--- a/program/include/rcube_message.php
+++ b/program/include/rcube_message.php
@@ -86,6 +86,16 @@ class rcube_message
     return $raw ? $value : $this->imap->decode_header($value);
   }
   
+  /**
+   * Set is_safe var and session data
+   *
+   * @param bool enable/disable
+   */
+  public function set_safe($safe = true)
+  {
+    $this->is_safe = $safe;
+    $_SESSION['safe_messages'][$this->uid] = $this->is_safe;
+  }
   
   /**
    * Compose a valid URL for getting a message part