UserMailer.php

Go to the documentation of this file.

<?php
class UserMailer {
    private static $mErrorString;

    protected static function sendWithPear( $mailer, $dest, $headers, $body ) {
        $mailResult = $mailer->send( $dest, $headers, $body );

        # Based on the result return an error string,
        if ( PEAR::isError( $mailResult ) ) {
            wfDebug( "PEAR::Mail failed: " . $mailResult->getMessage() . "\n" );
            return Status::newFatal( 'pear-mail-error', $mailResult->getMessage() );
        } else {
            return Status::newGood();
        }
    }

    static function arrayToHeaderString( $headers, $endl = "\n" ) {
        $strings = array();
        foreach ( $headers as $name => $value ) {
            // Prevent header injection by stripping newlines from value
            $value = self::sanitizeHeaderValue( $value );
            $strings[] = "$name: $value";
        }
        return implode( $endl, $strings );
    }

    static function makeMsgId() {
        global $wgSMTP, $wgServer;

        $msgid = uniqid( wfWikiID() . ".", true ); /* true required for cygwin */
        if ( is_array( $wgSMTP ) && isset( $wgSMTP['IDHost'] ) && $wgSMTP['IDHost'] ) {
            $domain = $wgSMTP['IDHost'];
        } else {
            $url = wfParseUrl( $wgServer );
            $domain = $url['host'];
        }
        return "<$msgid@$domain>";
    }

    public static function send( $to, $from, $subject, $body, $replyto = null,
        $contentType = 'text/plain; charset=UTF-8'
    ) {
        global $wgSMTP, $wgEnotifMaxRecips, $wgAdditionalMailParams, $wgAllowHTMLEmail;
        $mime = null;
        if ( !is_array( $to ) ) {
            $to = array( $to );
        }

        // mail body must have some content
        $minBodyLen = 10;
        // arbitrary but longer than Array or Object to detect casting error

        // body must either be a string or an array with text and body
        if (
            !(
                !is_array( $body ) &&
                strlen( $body ) >= $minBodyLen
            )
            &&
            !(
                is_array( $body ) &&
                isset( $body['text'] ) &&
                isset( $body['html'] ) &&
                strlen( $body['text'] ) >= $minBodyLen &&
                strlen( $body['html'] ) >= $minBodyLen
            )
        ) {
            // if it is neither we have a problem
            return Status::newFatal( 'user-mail-no-body' );
        }

        if ( !$wgAllowHTMLEmail && is_array( $body ) ) {
            // HTML not wanted.  Dump it.
            $body = $body['text'];
        }

        wfDebug( __METHOD__ . ': sending mail to ' . implode( ', ', $to ) . "\n" );

        # Make sure we have at least one address
        $has_address = false;
        foreach ( $to as $u ) {
            if ( $u->address ) {
                $has_address = true;
                break;
            }
        }
        if ( !$has_address ) {
            return Status::newFatal( 'user-mail-no-addy' );
        }

        # Forge email headers
        # -------------------
        #
        # WARNING
        #
        # DO NOT add To: or Subject: headers at this step. They need to be
        # handled differently depending upon the mailer we are going to use.
        #
        # To:
        #  PHP mail() first argument is the mail receiver. The argument is
        #  used as a recipient destination and as a To header.
        #
        #  PEAR mailer has a recipient argument which is only used to
        #  send the mail. If no To header is given, PEAR will set it to
        #  to 'undisclosed-recipients:'.
        #
        #  NOTE: To: is for presentation, the actual recipient is specified
        #  by the mailer using the Rcpt-To: header.
        #
        # Subject:
        #  PHP mail() second argument to pass the subject, passing a Subject
        #  as an additional header will result in a duplicate header.
        #
        #  PEAR mailer should be passed a Subject header.
        #
        # -- hashar 20120218

        $headers['From'] = $from->toString();
        $returnPath = $from->address;
        $extraParams = $wgAdditionalMailParams;

        // Hook to generate custom VERP address for 'Return-Path'
        wfRunHooks( 'UserMailerChangeReturnPath', array( $to, &$returnPath ) );
        # Add the envelope sender address using the -f command line option when PHP mail() is used.
        # Will default to the $from->address when the UserMailerChangeReturnPath hook fails and the
        # generated VERP address when the hook runs effectively.
        $extraParams .= ' -f ' . $returnPath;

        $headers['Return-Path'] = $returnPath;

        if ( $replyto ) {
            $headers['Reply-To'] = $replyto->toString();
        }

        $headers['Date'] = MWTimestamp::getLocalInstance()->format( 'r' );
        $headers['Message-ID'] = self::makeMsgId();
        $headers['X-Mailer'] = 'MediaWiki mailer';

        # Line endings need to be different on Unix and Windows due to
        # the bug described at http://trac.wordpress.org/ticket/2603
        if ( wfIsWindows() ) {
            $endl = "\r\n";
        } else {
            $endl = "\n";
        }

        if ( is_array( $body ) ) {
            // we are sending a multipart message
            wfDebug( "Assembling multipart mime email\n" );
            if ( !stream_resolve_include_path( 'Mail/mime.php' ) ) {
                wfDebug( "PEAR Mail_Mime package is not installed. Falling back to text email.\n" );
                // remove the html body for text email fall back
                $body = $body['text'];
            } else {
                require_once 'Mail/mime.php';
                if ( wfIsWindows() ) {
                    $body['text'] = str_replace( "\n", "\r\n", $body['text'] );
                    $body['html'] = str_replace( "\n", "\r\n", $body['html'] );
                }
                $mime = new Mail_mime( array(
                    'eol' => $endl,
                    'text_charset' => 'UTF-8',
                    'html_charset' => 'UTF-8'
                ) );
                $mime->setTXTBody( $body['text'] );
                $mime->setHTMLBody( $body['html'] );
                $body = $mime->get(); // must call get() before headers()
                $headers = $mime->headers( $headers );
            }
        }
        if ( $mime === null ) {
            // sending text only, either deliberately or as a fallback
            if ( wfIsWindows() ) {
                $body = str_replace( "\n", "\r\n", $body );
            }
            $headers['MIME-Version'] = '1.0';
            $headers['Content-type'] = ( is_null( $contentType ) ?
                'text/plain; charset=UTF-8' : $contentType );
            $headers['Content-transfer-encoding'] = '8bit';
        }

        $ret = wfRunHooks( 'AlternateUserMailer', array( $headers, $to, $from, $subject, $body ) );
        if ( $ret === false ) {
            // the hook implementation will return false to skip regular mail sending
            return Status::newGood();
        } elseif ( $ret !== true ) {
            // the hook implementation will return a string to pass an error message
            return Status::newFatal( 'php-mail-error', $ret );
        }

        if ( is_array( $wgSMTP ) ) {
            #
            # PEAR MAILER
            #

            if ( !stream_resolve_include_path( 'Mail.php' ) ) {
                throw new MWException( 'PEAR mail package is not installed' );
            }
            require_once 'Mail.php';

            wfSuppressWarnings();

            // Create the mail object using the Mail::factory method
            $mail_object =& Mail::factory( 'smtp', $wgSMTP );
            if ( PEAR::isError( $mail_object ) ) {
                wfDebug( "PEAR::Mail factory failed: " . $mail_object->getMessage() . "\n" );
                wfRestoreWarnings();
                return Status::newFatal( 'pear-mail-error', $mail_object->getMessage() );
            }

            wfDebug( "Sending mail via PEAR::Mail\n" );

            $headers['Subject'] = self::quotedPrintable( $subject );

            # When sending only to one recipient, shows it its email using To:
            if ( count( $to ) == 1 ) {
                $headers['To'] = $to[0]->toString();
            }

            # Split jobs since SMTP servers tends to limit the maximum
            # number of possible recipients.
            $chunks = array_chunk( $to, $wgEnotifMaxRecips );
            foreach ( $chunks as $chunk ) {
                $status = self::sendWithPear( $mail_object, $chunk, $headers, $body );
                # FIXME : some chunks might be sent while others are not!
                if ( !$status->isOK() ) {
                    wfRestoreWarnings();
                    return $status;
                }
            }
            wfRestoreWarnings();
            return Status::newGood();
        } else {
            #
            # PHP mail()
            #
            if ( count( $to ) > 1 ) {
                $headers['To'] = 'undisclosed-recipients:;';
            }
            $headers = self::arrayToHeaderString( $headers, $endl );

            wfDebug( "Sending mail via internal mail() function\n" );

            self::$mErrorString = '';
            $html_errors = ini_get( 'html_errors' );
            ini_set( 'html_errors', '0' );
            set_error_handler( 'UserMailer::errorHandler' );

            try {
                $safeMode = wfIniGetBool( 'safe_mode' );

                foreach ( $to as $recip ) {
                    if ( $safeMode ) {
                        $sent = mail( $recip, self::quotedPrintable( $subject ), $body, $headers );
                    } else {
                        $sent = mail(
                            $recip,
                            self::quotedPrintable( $subject ),
                            $body,
                            $headers,
                            $extraParams
                        );
                    }
                }
            } catch ( Exception $e ) {
                restore_error_handler();
                throw $e;
            }

            restore_error_handler();
            ini_set( 'html_errors', $html_errors );

            if ( self::$mErrorString ) {
                wfDebug( "Error sending mail: " . self::$mErrorString . "\n" );
                return Status::newFatal( 'php-mail-error', self::$mErrorString );
            } elseif ( !$sent ) {
                // mail function only tells if there's an error
                wfDebug( "Unknown error sending mail\n" );
                return Status::newFatal( 'php-mail-error-unknown' );
            } else {
                return Status::newGood();
            }
        }
    }

    static function errorHandler( $code, $string ) {
        self::$mErrorString = preg_replace( '/^mail\(\)(\s*\[.*?\])?: /', '', $string );
    }

    public static function sanitizeHeaderValue( $val ) {
        return strtr( $val, array( "\r" => '', "\n" => '' ) );
    }

    public static function rfc822Phrase( $phrase ) {
        // Remove line breaks
        $phrase = self::sanitizeHeaderValue( $phrase );
        // Remove quotes
        $phrase = str_replace( '"', '', $phrase );
        return '"' . $phrase . '"';
    }

    public static function quotedPrintable( $string, $charset = '' ) {
        # Probably incomplete; see RFC 2045
        if ( empty( $charset ) ) {
            $charset = 'UTF-8';
        }
        $charset = strtoupper( $charset );
        $charset = str_replace( 'ISO-8859', 'ISO8859', $charset ); // ?

        $illegal = '\x00-\x08\x0b\x0c\x0e-\x1f\x7f-\xff=';
        $replace = $illegal . '\t ?_';
        if ( !preg_match( "/[$illegal]/", $string ) ) {
            return $string;
        }
        $out = "=?$charset?Q?";
        $out .= preg_replace_callback( "/([$replace])/",
            array( __CLASS__, 'quotedPrintableCallback' ), $string );
        $out .= '?=';
        return $out;
    }

    protected static function quotedPrintableCallback( $matches ) {
        return sprintf( "=%02X", ord( $matches[1] ) );
    }
}