Mercurial > hg > rc1
comparison plugins/enigma/lib/enigma_mime_message.php @ 0:1e000243b222
vanilla 1.3.3 distro, I hope
| author | Charlie Root |
|---|---|
| date | Thu, 04 Jan 2018 15:50:29 -0500 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:1e000243b222 |
|---|---|
| 1 <?php | |
| 2 | |
| 3 /** | |
| 4 +-------------------------------------------------------------------------+ | |
| 5 | Mail_mime wrapper for the Enigma Plugin | | |
| 6 | | | |
| 7 | Copyright (C) 2010-2015 The Roundcube Dev Team | | |
| 8 | | | |
| 9 | Licensed under the GNU General Public License version 3 or | | |
| 10 | any later version with exceptions for skins & plugins. | | |
| 11 | See the README file for a full license statement. | | |
| 12 | | | |
| 13 +-------------------------------------------------------------------------+ | |
| 14 | Author: Aleksander Machniak <alec@alec.pl> | | |
| 15 +-------------------------------------------------------------------------+ | |
| 16 */ | |
| 17 | |
| 18 class enigma_mime_message extends Mail_mime | |
| 19 { | |
| 20 const PGP_SIGNED = 1; | |
| 21 const PGP_ENCRYPTED = 2; | |
| 22 | |
| 23 protected $type; | |
| 24 protected $message; | |
| 25 protected $body; | |
| 26 protected $signature; | |
| 27 protected $encrypted; | |
| 28 protected $micalg; | |
| 29 | |
| 30 | |
| 31 /** | |
| 32 * Object constructor | |
| 33 * | |
| 34 * @param Mail_mime Original message | |
| 35 * @param int Output message type | |
| 36 */ | |
| 37 function __construct($message, $type) | |
| 38 { | |
| 39 $this->message = $message; | |
| 40 $this->type = $type; | |
| 41 | |
| 42 // clone parameters | |
| 43 foreach (array_keys($this->build_params) as $param) { | |
| 44 $this->build_params[$param] = $message->getParam($param); | |
| 45 } | |
| 46 | |
| 47 // clone headers | |
| 48 $this->headers = $message->headers(); | |
| 49 | |
| 50 // \r\n is must-have here | |
| 51 $this->body = $message->get() . "\r\n"; | |
| 52 } | |
| 53 | |
| 54 /** | |
| 55 * Check if the message is multipart (requires PGP/MIME) | |
| 56 * | |
| 57 * @return bool True if it is multipart, otherwise False | |
| 58 */ | |
| 59 public function isMultipart() | |
| 60 { | |
| 61 return $this->message instanceof enigma_mime_message | |
| 62 || $this->message->isMultipart() || $this->message->getHTMLBody(); | |
| 63 } | |
| 64 | |
| 65 /** | |
| 66 * Get e-mail address of message sender | |
| 67 * | |
| 68 * @return string Sender address | |
| 69 */ | |
| 70 public function getFromAddress() | |
| 71 { | |
| 72 // get sender address | |
| 73 $headers = $this->message->headers(); | |
| 74 $from = rcube_mime::decode_address_list($headers['From'], 1, false, null, true); | |
| 75 $from = $from[1]; | |
| 76 | |
| 77 return $from; | |
| 78 } | |
| 79 | |
| 80 /** | |
| 81 * Get recipients' e-mail addresses | |
| 82 * | |
| 83 * @return array Recipients' addresses | |
| 84 */ | |
| 85 public function getRecipients() | |
| 86 { | |
| 87 // get sender address | |
| 88 $headers = $this->message->headers(); | |
| 89 $to = rcube_mime::decode_address_list($headers['To'], null, false, null, true); | |
| 90 $cc = rcube_mime::decode_address_list($headers['Cc'], null, false, null, true); | |
| 91 $bcc = rcube_mime::decode_address_list($headers['Bcc'], null, false, null, true); | |
| 92 | |
| 93 $recipients = array_unique(array_merge($to, $cc, $bcc)); | |
| 94 $recipients = array_diff($recipients, array('undisclosed-recipients:')); | |
| 95 | |
| 96 return $recipients; | |
| 97 } | |
| 98 | |
| 99 /** | |
| 100 * Get original message body, to be encrypted/signed | |
| 101 * | |
| 102 * @return string Message body | |
| 103 */ | |
| 104 public function getOrigBody() | |
| 105 { | |
| 106 $_headers = $this->message->headers(); | |
| 107 $headers = array(); | |
| 108 | |
| 109 if ($_headers['Content-Transfer-Encoding'] | |
| 110 && stripos($_headers['Content-Type'], 'multipart') === false | |
| 111 ) { | |
| 112 $headers[] = 'Content-Transfer-Encoding: ' . $_headers['Content-Transfer-Encoding']; | |
| 113 } | |
| 114 $headers[] = 'Content-Type: ' . $_headers['Content-Type']; | |
| 115 | |
| 116 return implode("\r\n", $headers) . "\r\n\r\n" . $this->body; | |
| 117 } | |
| 118 | |
| 119 /** | |
| 120 * Register signature attachment | |
| 121 * | |
| 122 * @param string Signature body | |
| 123 * @param string Hash algorithm name | |
| 124 */ | |
| 125 public function addPGPSignature($body, $algorithm = null) | |
| 126 { | |
| 127 $this->signature = $body; | |
| 128 $this->micalg = $algorithm; | |
| 129 | |
| 130 // Reset Content-Type to be overwritten with valid boundary | |
| 131 unset($this->headers['Content-Type']); | |
| 132 unset($this->headers['Content-Transfer-Encoding']); | |
| 133 } | |
| 134 | |
| 135 /** | |
| 136 * Register encrypted body | |
| 137 * | |
| 138 * @param string Encrypted body | |
| 139 */ | |
| 140 public function setPGPEncryptedBody($body) | |
| 141 { | |
| 142 $this->encrypted = $body; | |
| 143 | |
| 144 // Reset Content-Type to be overwritten with valid boundary | |
| 145 unset($this->headers['Content-Type']); | |
| 146 unset($this->headers['Content-Transfer-Encoding']); | |
| 147 } | |
| 148 | |
| 149 /** | |
| 150 * Builds the multipart message. | |
| 151 * | |
| 152 * @param array $params Build parameters that change the way the email | |
| 153 * is built. Should be associative. See $_build_params. | |
| 154 * @param resource $filename Output file where to save the message instead of | |
| 155 * returning it | |
| 156 * @param boolean $skip_head True if you want to return/save only the message | |
| 157 * without headers | |
| 158 * | |
| 159 * @return mixed The MIME message content string, null or PEAR error object | |
| 160 */ | |
| 161 public function get($params = null, $filename = null, $skip_head = false) | |
| 162 { | |
| 163 if (!empty($params)) { | |
| 164 foreach ($params as $key => $value) { | |
| 165 $this->build_params[$key] = $value; | |
| 166 } | |
| 167 } | |
| 168 | |
| 169 $this->checkParams(); | |
| 170 | |
| 171 if ($this->type == self::PGP_SIGNED) { | |
| 172 $params = array( | |
| 173 'preamble' => "This is an OpenPGP/MIME signed message (RFC 4880 and 3156)", | |
| 174 'content_type' => "multipart/signed; protocol=\"application/pgp-signature\"", | |
| 175 'eol' => $this->build_params['eol'], | |
| 176 ); | |
| 177 | |
| 178 if ($this->micalg) { | |
| 179 $params['content_type'] .= "; micalg=pgp-" . $this->micalg; | |
| 180 } | |
| 181 | |
| 182 $message = new Mail_mimePart('', $params); | |
| 183 | |
| 184 if (!empty($this->body)) { | |
| 185 $headers = $this->message->headers(); | |
| 186 $params = array('content_type' => $headers['Content-Type']); | |
| 187 | |
| 188 if ($headers['Content-Transfer-Encoding'] | |
| 189 && stripos($headers['Content-Type'], 'multipart') === false | |
| 190 ) { | |
| 191 $params['encoding'] = $headers['Content-Transfer-Encoding']; | |
| 192 } | |
| 193 | |
| 194 $message->addSubpart($this->body, $params); | |
| 195 } | |
| 196 | |
| 197 if (!empty($this->signature)) { | |
| 198 $message->addSubpart($this->signature, array( | |
| 199 'filename' => 'signature.asc', | |
| 200 'content_type' => 'application/pgp-signature', | |
| 201 'disposition' => 'attachment', | |
| 202 'description' => 'OpenPGP digital signature', | |
| 203 )); | |
| 204 } | |
| 205 } | |
| 206 else if ($this->type == self::PGP_ENCRYPTED) { | |
| 207 $params = array( | |
| 208 'preamble' => "This is an OpenPGP/MIME encrypted message (RFC 4880 and 3156)", | |
| 209 'content_type' => "multipart/encrypted; protocol=\"application/pgp-encrypted\"", | |
| 210 'eol' => $this->build_params['eol'], | |
| 211 ); | |
| 212 | |
| 213 $message = new Mail_mimePart('', $params); | |
| 214 | |
| 215 $message->addSubpart('Version: 1', array( | |
| 216 'content_type' => 'application/pgp-encrypted', | |
| 217 'description' => 'PGP/MIME version identification', | |
| 218 )); | |
| 219 | |
| 220 $message->addSubpart($this->encrypted, array( | |
| 221 'content_type' => 'application/octet-stream', | |
| 222 'description' => 'PGP/MIME encrypted message', | |
| 223 'disposition' => 'inline', | |
| 224 'filename' => 'encrypted.asc', | |
| 225 )); | |
| 226 } | |
| 227 | |
| 228 // Use saved boundary | |
| 229 if (!empty($this->build_params['boundary'])) { | |
| 230 $boundary = $this->build_params['boundary']; | |
| 231 } | |
| 232 else { | |
| 233 $boundary = null; | |
| 234 } | |
| 235 | |
| 236 // Write output to file | |
| 237 if ($filename) { | |
| 238 // Append mimePart message headers and body into file | |
| 239 $headers = $message->encodeToFile($filename, $boundary, $skip_head); | |
| 240 | |
| 241 if ($this->isError($headers)) { | |
| 242 return $headers; | |
| 243 } | |
| 244 | |
| 245 $this->headers = array_merge($this->headers, $headers); | |
| 246 | |
| 247 return; | |
| 248 } | |
| 249 else { | |
| 250 $output = $message->encode($boundary, $skip_head); | |
| 251 | |
| 252 if ($this->isError($output)) { | |
| 253 return $output; | |
| 254 } | |
| 255 | |
| 256 $this->headers = array_merge($this->headers, $output['headers']); | |
| 257 | |
| 258 return $output['body']; | |
| 259 } | |
| 260 } | |
| 261 | |
| 262 /** | |
| 263 * Get Content-Type and Content-Transfer-Encoding headers of the message | |
| 264 * | |
| 265 * @return array Headers array | |
| 266 */ | |
| 267 protected function contentHeaders() | |
| 268 { | |
| 269 $this->checkParams(); | |
| 270 | |
| 271 $eol = $this->build_params['eol'] ?: "\r\n"; | |
| 272 | |
| 273 // multipart message: and boundary | |
| 274 if (!empty($this->build_params['boundary'])) { | |
| 275 $boundary = $this->build_params['boundary']; | |
| 276 } | |
| 277 else if (!empty($this->headers['Content-Type']) | |
| 278 && preg_match('/boundary="([^"]+)"/', $this->headers['Content-Type'], $m) | |
| 279 ) { | |
| 280 $boundary = $m[1]; | |
| 281 } | |
| 282 else { | |
| 283 $boundary = '=_' . md5(rand() . microtime()); | |
| 284 } | |
| 285 | |
| 286 $this->build_params['boundary'] = $boundary; | |
| 287 | |
| 288 if ($this->type == self::PGP_SIGNED) { | |
| 289 $headers['Content-Type'] = "multipart/signed;$eol" | |
| 290 ." protocol=\"application/pgp-signature\";$eol" | |
| 291 ." boundary=\"$boundary\""; | |
| 292 | |
| 293 if ($this->micalg) { | |
| 294 $headers['Content-Type'] .= ";{$eol} micalg=pgp-" . $this->micalg; | |
| 295 } | |
| 296 } | |
| 297 else if ($this->type == self::PGP_ENCRYPTED) { | |
| 298 $headers['Content-Type'] = "multipart/encrypted;$eol" | |
| 299 ." protocol=\"application/pgp-encrypted\";$eol" | |
| 300 ." boundary=\"$boundary\""; | |
| 301 } | |
| 302 | |
| 303 return $headers; | |
| 304 } | |
| 305 } |
