added parameters id and mailbox to formadBody / magicHTML for better/ easier
[squirrelmail.git] / functions / mime.php
index 178828892bd489c77440bd43a2363594d11625b5..ce2be9b894cef4b9b36c9f8bad8ff4bf27d625e6 100644 (file)
 require_once('../functions/imap.php');
 require_once('../functions/attachment_common.php');
 
-/** Setting up the objects that have the structure for the message **/
-class msg_header {
-    /** msg_header contains generic variables for values that **/
-    /** could be in a header.                                 **/
-
-    var $type0 = '', $type1 = '', $boundary = '', $charset = '',
-        $encoding = '', $size = 0, $to = array(), $from = '', $date = '',
-        $cc = array(), $bcc = array(), $reply_to = '', $subject = '',
-        $id = 0, $mailbox = '', $description = '', $filename = '',
-        $entity_id = 0, $message_id = 0, $name = '', $priority = 3, $type = '';
-}
-
-class message {
-    /** message is the object that contains messages.  It is a recursive
-      object in that through the $entities variable, it can contain
-      more objects of type message.  See documentation in mime.txt for
-      a better description of how this works.
-    **/
-    var $header = '', $entities = array();
-
-    function addEntity ($msg) {
-        $this->entities[] = $msg;
-    }
-}
-
 /* --------------------------------------------------------------------------------- */
 /* MIME DECODING                                                                     */
 /* --------------------------------------------------------------------------------- */
@@ -48,38 +23,50 @@ class message {
  * It will return this object for use with all relevant header information and
  * fully parsed into the standard "message" object format.
  */
-function mime_structure ($imap_stream, $header) {
-
-    $ssid = sqimap_session_id();
-    $lsid = strlen( $ssid );
-    $id = $header->id;
-    fputs ($imap_stream, "$ssid FETCH $id BODYSTRUCTURE\r\n");
-    //
-    // This should use sqimap_read_data instead of reading it itself
-    //
-    $read = fgets ($imap_stream, 9216);
-    $bodystructure = '';
-    while ( substr($read, 0, $lsid) <> $ssid &&
-         !feof( $imap_stream ) ) {
-        $bodystructure .= $read;
-        $read = fgets ($imap_stream, 9216);
-    }
-    $read = $bodystructure;
 
-    // isolate the body structure and remove beginning and end parenthesis
-    $read = trim(substr ($read, strpos(strtolower($read), 'bodystructure') + 13));
+function mime_structure ($bodystructure, $flags=array()) {
 
+    // isolate the body structure and remove beginning and end parenthesis
+    $read = trim(substr ($bodystructure, strpos(strtolower($bodystructure), 'bodystructure') + 13));
+    $msg = &new message();
     $read = trim(substr ($read, 0, -1));
-    $end = mime_match_parenthesis(0, $read);
-    while ($end == strlen($read)-1) {
-        $read = trim(substr ($read, 0, -1));
-        $read = trim(substr ($read, 1));
-        $end = mime_match_parenthesis(0, $read);
+    $msg = $msg->parseStructure($read,0);
+    $msg->setEnt('0');
+    if (count($flags)) {
+       foreach ($flags as $flag) {
+          $char = strtoupper($flag{1});
+         switch ($char) {
+            case 'S':
+              if (strtolower($flag) == '\\seen') {
+                 $msg->is_seen = true;
+              }
+              break;
+            case 'A':
+              if (strtolower($flag) == '\\answered') {
+                 $msg->is_answered = true;
+              }
+              break;
+            case 'D':
+              if (strtolower($flag) == '\\deleted') {
+                 $msg->is_deleted = true;
+              }
+              break;
+            case 'F':
+              if (strtolower($flag) == '\\flagged') {
+                 $msg->is_flagged = true;
+              }
+              break;
+            case 'M':
+              if (strtolower($flag) == '\$mdnsent') {
+                 $msg->is_mdn = true;
+              }
+              break;
+            default:
+              break;
+          }
+       }
     }
-
-    $msg = mime_parse_structure ($read, 0);
-    $msg->header = $header;
-
+//    listEntities($msg);
     return( $msg );
 }
 
@@ -91,358 +78,9 @@ function mime_structure ($imap_stream, $header) {
  * then it parses it and adds the necessary header information (by calling out
  * to mime_get_elements()
  */
-function mime_parse_structure ($structure, $ent_id) {
-  global $mailbox;
-  $properties = array();
-  $msg = new message();
-  if ($structure{0} == '(') {
-     $old_ent_id = $ent_id;
-     $ent_id = mime_new_element_level($ent_id);
-     $start = $end = -1;
-     do {
-        $start = $end+1;
-        $end = mime_match_parenthesis ($start, $structure);
-
-       /* check if we are dealing with a new entity-level */
-       $i = strrpos($ent_id,'.');
-       if ($i>0) {
-           $ent = substr($ent_id, $i+1);
-       } else {
-           $ent = '';
-       }
-        /* add "forgotten"  parent entities (alternative and relative) */
-       if ($ent == '0') {
-           /* new entity levels have information about the type (type1) and 
-           *  the properties. This information is situated at the end of the 
-           *  structure string like for example (example between the brackets) 
-           *  [ "RELATED" ("BOUNDARY" "myboundary" "TYPE" "plain/html") ]
-           */
-           
-           /* get the involved properties for parsing to mime_get_properties */
-           $startprop = strrpos($structure,'(');
-           $properties_str = substr($structure,$startprop);
-           $endprop = mime_match_parenthesis ($startprop, $structure);
-           $propstr = substr($structure, $startprop + 1, ($endprop - $startprop)-1);
-           /* cut off the used properties */
-           if ($startprop) { 
-               $structure_end = substr($structure, $endprop+2);
-               $structure = trim(substr($structure,0,$startprop));
-           }
-           
-           /* get type1 */
-           $pos = strrpos($structure,' ');
-           $type1 = strtolower(substr($structure, $pos+2, (count($structure)-2)));
-
-           /* cut off  type1 */
-           if ($pos && $startprop) {
-               $structure = trim(substr($structure, 0, $pos));
-           }
-
-           /* process the found information */
-            $properties = mime_get_props($properties, $properties_str);
-           if (count($properties)>0) {
-               $msg->header->entity_id = $old_ent_id;
-               $msg->header->type0 = 'multipart';
-               $msg->header->type1 = $type1;
-               for ($i=0; $i < count($properties); $i++) {
-                   $msg->header->{$properties[$i]['name']} = $properties[$i]['value'];
-               }
-           }
-           $structure = $structure . ' ' . $structure_end;
-       } 
-       $element = substr($structure, $start+1, ($end - $start)-1);
-       $ent_id = mime_increment_id ($ent_id);
-       $newmsg = mime_parse_structure ($element, $ent_id);
-       /* set mailbox in case of message/rfc822 entities */
-       if (isset($newmsg->header->type0) && isset($newmsg->header->type1)) {
-           if ($newmsg->header->type0 == 'message' && $newmsg->header->type1 == 'rfc822') {
-               $newmsg->header->mailbox=$mailbox;
-           }
-       }
-       $msg->addEntity ($newmsg);
-
-     } while ($structure{$end+1} == '(');
-  } else {
-     // parse the elements
-    $msg = mime_get_element ($structure, $msg, $ent_id);
-  }
-  return $msg;
-}
-
-
-/* Increments the element ID.  An element id can look like any of
- * the following:  1, 1.2, 4.3.2.4.1, etc.  This function increments
- * the last number of the element id, changing 1.2 to 1.3.
- */
-function mime_increment_id ($id) {
-
-    if (strpos($id, '.')) {
-        $first = substr($id, 0, strrpos($id, '.'));
-        $last = substr($id, strrpos($id, '.')+1);
-        $last++;
-        $new = $first . '.' .$last;
-    } else {
-        $new = $id + 1;
-    }
-
-    return $new;
-}
-
-/*
- * See comment for mime_increment_id().
- * This adds another level on to the entity_id changing 1.3 to 1.3.0
- * NOTE:  1.3.0 is not a valid element ID.  It MUST be incremented
- *        before it can be used.  I left it this way so as not to have
- *        to make a special case if it is the first entity_id.  It
- *        always increments it, and that works fine.
- */
-function mime_new_element_level ($id) {
-
-    if (!$id) {
-        $id = 0;
-    } else {
-        $id = $id . '.0';
-    }
-
-    return( $id );
-}
-
-function mime_get_element (&$structure, $msg, $ent_id) {
-
-  $elem_num = 1;
-  $msg->header = new msg_header();
-  $msg->header->entity_id = $ent_id;
-  $properties = array();
-  while (strlen($structure) > 0) {
-     $structure = trim($structure);
-     $char = $structure{0};
-
-     if (strtolower(substr($structure, 0, 3)) == 'nil') {
-        $text = '';
-        $structure = substr($structure, 3);
-     } else if ($char == '"') {
-        // loop through until we find the matching quote, and return that as a string
-        $pos = 1;
-        $text = '';
-        while ( ($char = $structure{$pos} ) <> '"' && $pos < strlen($structure)) {
-           $text .= $char;
-           $pos++;
-        }
-        $structure = substr($structure, strlen($text) + 2);
-     } else if ($char == '{') {
-         /**
-          * loop through until we find the matching quote, 
-          * and return that as a string
-          */
-         $pos = 1;
-         $len = '';
-         while (($char = $structure{$pos}) != '}' 
-                && $pos < strlen($structure)) {
-             $len .= $char;
-             $pos++;
-         }
-         $structure = substr($structure, strlen($len) + 4);
-         $text = substr($structure, 0, $len);
-         $structure = substr($structure, $len + 1);
-     } else if ($char == '(') {
-        // comment me
-        $end = mime_match_parenthesis (0, $structure);
-        $sub = substr($structure, 1, $end-1);
-        $properties = mime_get_props($properties, $sub);
-        $structure = substr($structure, strlen($sub) + 2);
-     } else {
-        // loop through until we find a space or an end parenthesis
-        $pos = 0;
-        $char = $structure{$pos};
-        $text = '';
-        while ($char != ' ' && $char != ')' && $pos < strlen($structure)) {
-           $text .= $char;
-           $pos++;
-           $char = $structure{$pos};
-        }
-        $structure = substr($structure, strlen($text));
-     }
-
-     // This is where all the text parts get put into the header
-     switch ($elem_num) {
-        case 1:
-           $msg->header->type0 = strtolower($text);
-           break;
-        case 2:
-           $msg->header->type1 = strtolower($text);
-           break;
-        case 4: // Id
-           // Invisimail enclose images with <>
-           $msg->header->id = str_replace( '<', '', str_replace( '>', '', $text ) );
-           break;
-        case 5:
-           $msg->header->description = $text;
-           break;
-        case 6:
-           $msg->header->encoding = strtolower($text);
-           break;
-        case 7:
-           $msg->header->size = $text;
-           break;
-        default:
-           if ($msg->header->type0 == 'text' && $elem_num == 8) {
-              // This is a plain text message, so lets get the number of lines
-              // that it contains.
-              $msg->header->num_lines = $text;
-
-           } else if ($msg->header->type0 == 'message' && $msg->header->type1 == 'rfc822' && $elem_num == 8) {
-              // This is an encapsulated message, so lets start all over again and
-              // parse this message adding it on to the existing one.
-              $structure = trim($structure);
-              if ( $structure{0} == '(' ) {
-                 $e = mime_match_parenthesis (0, $structure);
-                 $structure = substr($structure, 0, $e);
-                 $structure = substr($structure, 1);
-                 $m = mime_parse_structure($structure, $msg->header->entity_id);
-
-                 // the following conditional is there to correct a bug that wasn't
-                 // incrementing the entity IDs correctly because of the special case
-                 // that message/rfc822 is.  This fixes it fine.
-                 if (substr($structure, 1, 1) != '(')
-                    $m->header->entity_id = mime_increment_id(mime_new_element_level($ent_id));
-
-                 // Now we'll go through and reformat the results.
-                 if ($m->entities) {
-                    for ($i=0; $i < count($m->entities); $i++) {
-                       $msg->addEntity($m->entities[$i]);
-                    }
-                 } else {
-                    $msg->addEntity($m);
-                 }
-                 $structure = "";
-              }
-           }
-           break;
-     }
-     $elem_num++;
-     $text = "";
-  }
-  // loop through the additional properties and put those in the various headers
-  for ($i=0; $i < count($properties); $i++) {
-     $msg->header->{$properties[$i]['name']} = $properties[$i]['value'];
-  }
-
-  return $msg;
-}
-
-/*
- * I did most of the MIME stuff yesterday (June 20, 2000), but I couldn't
- * figure out how to do this part, so I decided to go to bed.  I woke up
- * in the morning and had a flash of insight.  I went to the white-board
- * and scribbled it out, then spent a bit programming it, and this is the
- * result.  Nothing complicated, but I think my brain was fried yesterday.
- * Funny how that happens some times.
- *
- * This gets properties in a nested parenthesisized list.  For example,
- * this would get passed something like:  ("attachment" ("filename" "luke.tar.gz"))
- * This returns an array called $props with all paired up properties.
- * It ignores the "attachment" for now, maybe that should change later
- * down the road.  In this case, what is returned is:
- *    $props[0]["name"] = "filename";
- *    $props[0]["value"] = "luke.tar.gz";
- */
-function mime_get_props ($props, $structure) {
-
-  while (strlen($structure) > 0) {
-     $structure = trim($structure);
-     $char = $structure{0};
-     if ($char == '"') {
-        $pos = 1;
-        $tmp = '';
-        while ( ( $char = $structure{$pos} ) != '"' &&
-                $pos < strlen($structure)) {
-           $tmp .= $char;
-           $pos++;
-        }
-        $structure = trim(substr($structure, strlen($tmp) + 2));
-        $char = $structure{0};
-
-        if ($char == '"') {
-           $pos = 1;
-           $value = '';
-           while ( ( $char = $structure{$pos} ) != '"' &&
-                   $pos < strlen($structure) ) {
-              $value .= $char;
-              $pos++;
-           }
-           $structure = trim(substr($structure, strlen($value) + 2));
-           $k = count($props);
-           $props[$k]['name'] = strtolower($tmp);
-           $props[$k]['value'] = $value;
-          if ($structure != '') {
-               mime_get_props($props, $structure);
-          } else {
-            return $props;
-          }            
-        } else if ($char == '(') {
-           $end = mime_match_parenthesis (0, $structure);
-           $sub = substr($structure, 1, $end-1);
-          if (! isset($props))
-              $props = array();
-              $props = mime_get_props($props, $sub);
-              $structure = substr($structure, strlen($sub) + 2);
-          return $props;      
-        }
-     } else if ($char == '(') {
-        $end = mime_match_parenthesis (0, $structure);
-        $sub = substr($structure, 1, $end-1);
-        $props = mime_get_props($props, $sub);
-        $structure = substr($structure, strlen($sub) + 2);
-        return $props;
-     } else {
-        return $props;
-     }
-  }
-}
-
-/*
- *  Matches parenthesis.  It will return the position of the matching
- *  parenthesis in $structure.  For instance, if $structure was:
- *     ("text" "plain" ("val1name", "1") nil ... )
- *     x                                         x
- *  then this would return 42 to match up those two.
- */
-function mime_match_parenthesis ($pos, $structure) {
-
-    $j = strlen( $structure );
-    
-    // ignore all extra characters
-    // If inside of a string, skip string -- Boundary IDs and other
-    // things can have ) in them.
-    if ( $structure{$pos} != '(' ) {
-        return( $j );
-    }
-    
-    while ( $pos < $j ) {
-        $pos++;
-        if ($structure{$pos} == ')') {
-            return $pos;
-        } elseif ($structure{$pos} == '"') {
-            $pos++;
-            while ( $structure{$pos} != '"' &&
-                    $pos < $j ) {
-               if (substr($structure, $pos, 2) == '\\"') {
-                  $pos++;
-               } elseif (substr($structure, $pos, 2) == '\\\\') {
-                  $pos++;
-               }
-               $pos++;
-            }
-        } elseif ( $structure{$pos} == '(' ) {
-            $pos = mime_match_parenthesis ($pos, $structure);
-        }
-    }
-    echo _("Error decoding mime structure.  Report this as a bug!") . '<br>';
-    return( $pos );
-}
-
-function mime_fetch_body($imap_stream, $id, $ent_id ) {
 
+function mime_fetch_body($imap_stream, $id, $ent_id) {
+    global $uid_support; 
     /*
      * do a bit of error correction.  If we couldn't find the entity id, just guess
      * that it is the first one.  That is usually the case anyway.
@@ -450,13 +88,13 @@ function mime_fetch_body($imap_stream, $id, $ent_id ) {
     if (!$ent_id) {
         $ent_id = 1;
     }
-
     $cmd = "FETCH $id BODY[$ent_id]";
-    $data = sqimap_run_command ($imap_stream, $cmd, true, $response, $message);
-
+    
+    $data = sqimap_run_command ($imap_stream, $cmd, true, $response, $message, $uid_support);
     do {
         $topline = trim(array_shift( $data ));
     } while( $topline && $topline[0] == '*' && !preg_match( '/\* [0-9]+ FETCH.*/i', $topline )) ;
+
     $wholemessage = implode('', $data);
     if (ereg('\\{([^\\}]*)\\}', $topline, $regs)) {
 
@@ -466,33 +104,7 @@ function mime_fetch_body($imap_stream, $id, $ent_id ) {
             in order to parse html messages. Let's get them here.
         */
         if ( $ret{0} == '<' ) {
-            $data = sqimap_run_command ($imap_stream, "FETCH $id BODY[$ent_id.MIME]", true, $response, $message);
-            /* BASE within HTML documents is illegal (see w3 spec)
-*            $base = '';
-*            $k = 10;
-*            foreach( $data as $d ) {
-*                if ( substr( $d, 0, 13 ) == 'Content-Base:' ) {
-*                    $j = strlen( $d );
-*                    $i = 13;
-*                    $base = '';
-*                    while ( $i < $j &&
-*                           ( !isNoSep( $d{$i} ) || $d{$i} == '"' )  )
-*                        $i++;
-*                    while ( $i < $j ) {
-*                        if ( isNoSep( $d{$i} ) )
-*                            $base .= $d{$i};
-*                        $i++;
-*                    }
-*                    $k = 0;
-*                } elseif ( $k == 1 && !isnosep( $d{0} ) ) {
-*                    $base .= substr( $d, 1 );
-*                }
-*                $k++;
-*            }
-*            if ( $base <> '' ) {
-*                $ret = "<base href=\"$base\">" . $ret;
-*            }
-*           */
+            $data = sqimap_run_command ($imap_stream, "FETCH $id BODY[$ent_id.MIME]", true, $response, $message, $uid_support);
         }
     } else if (ereg('"([^"]*)"', $topline, $regs)) {
         $ret = $regs[1];
@@ -521,7 +133,7 @@ function mime_fetch_body($imap_stream, $id, $ent_id ) {
                '<td><b>' . _("FETCH line:") . "</td><td>$topline</td></tr>" .
                "</table><BR></tt></font><hr>";
 
-        $data = sqimap_run_command ($imap_stream, "FETCH $passed_id BODY[]", true, $response, $message);
+        $data = sqimap_run_command ($imap_stream, "FETCH $passed_id BODY[]", true, $response, $message, $uid_support);
         array_shift($data);
         $wholemessage = implode('', $data);
 
@@ -531,12 +143,13 @@ function mime_fetch_body($imap_stream, $id, $ent_id ) {
 }
 
 function mime_print_body_lines ($imap_stream, $id, $ent_id, $encoding) {
+    global $uid_support;
     // do a bit of error correction.  If we couldn't find the entity id, just guess
     // that it is the first one.  That is usually the case anyway.
     if (!$ent_id) {
         $ent_id = 1;
     }
-    $sid = sqimap_session_id();
+    $sid = sqimap_session_id($uid_support);
     // Don't kill the connection if the browser is over a dialup
     // and it would take over 30 seconds to download it.
 
@@ -544,14 +157,37 @@ function mime_print_body_lines ($imap_stream, $id, $ent_id, $encoding) {
     if (!ini_get("safe_mode")) {
         set_time_limit(0);
     }
+    if ($uid_support) {
+       $sid_s = substr($sid,0,strpos($sid, ' '));
+    } else {
+       $sid_s = $sid;
+    }
 
     fputs ($imap_stream, "$sid FETCH $id BODY[$ent_id]\r\n");
     $cnt = 0;
     $continue = true;
-    $read = fgets ($imap_stream,4096);
+    $read = fgets ($imap_stream,8192);
+
+
+//    if (preg_match('/.*\{(\d+)\}.*/',$read,$regs)) {
+/*
+       $size = $regs[1];
+       $size_div = (int) ($size / 4096);
+       $size_mod = $size % 4096;
+       if (!$size_mod) $size_div++;
+       $read = '';
+       for ($i=0;$i<$size_div;$i++) { 
+           $read .= fread ($imap_stream,4096);
+       }
+       if ($size_mod > 0) {
+          $read .= fread ($imap_stream, $size_mod);
+       }
+       echo decodeBody($read, $encoding);
+    }
+*/
     // This could be bad -- if the section has sqimap_session_id() . ' OK'
     // or similar, it will kill the download.
-    while (!ereg("^".$sid." (OK|BAD|NO)(.*)$", $read, $regs)) {
+    while (!ereg("^".$sid_s." (OK|BAD|NO)(.*)$", $read, $regs)) {
       if (trim($read) == ')==') {
           $read1 = $read;
           $read = fgets ($imap_stream,4096);
@@ -566,88 +202,84 @@ function mime_print_body_lines ($imap_stream, $id, $ent_id, $encoding) {
       }
       $read = fgets ($imap_stream,4096);
       $cnt++;
+//      break;
     }
 }
 
 /* -[ END MIME DECODING ]----------------------------------------------------------- */
 
 
-
-/* This is the first function called.  It decides if this is a multipart
-   message or if it should be handled as a single entity
- */
-function decodeMime ($imap_stream, &$header) {
-    global $username, $key, $imapServerAddress, $imapPort;
-    return mime_structure ($imap_stream, $header);
-}
-
 // This is here for debugging purposese.  It will print out a list
 // of all the entity IDs that are in the $message object.
 
 function listEntities ($message) {
-if ($message) {
- if ($message->header->entity_id)
- echo "<tt>" . $message->header->entity_id . ' : ' . $message->header->type0 . '/' . $message->header->type1 . '<br>';
- for ($i = 0; $message->entities[$i]; $i++) {
-    $msg = listEntities($message->entities[$i], $ent_id);
-    if ($msg)
-       return $msg;
- }
-}
+  if ($message) {
+    echo "<tt>" . $message->entity_id . ' : ' . $message->type0 . '/' . $message->type1 . ' parent = '. $message->parent->entity_id. '<br>';
+    for ($i = 0;isset($message->entities[$i]); $i++) {
+       echo "$i : ";
+       $msg = listEntities($message->entities[$i]);
+    
+       if ($msg) {
+          echo "return: ";
+          return $msg;
+       }
+    }
+  } 
 }
 
 
 /* returns a $message object for a particular entity id */
 function getEntity ($message, $ent_id) {
-    if ($message) {
-        if ($message->header->entity_id == $ent_id && strlen($ent_id) == strlen($message->header->entity_id))
-       {
-            return $message;
-        } else {
-            for ($i = 0; isset($message->entities[$i]); $i++) {
-                $msg = getEntity ($message->entities[$i], $ent_id);
-                if ($msg) {
-                    return $msg;
-               }
-            }
-        }
-    }
+    return $message->getEntity($ent_id);
 }
 
 /*
  * figures out what entity to display and returns the $message object
  * for that entity.
  */
-function findDisplayEntity ($msg, $textOnly = 1)   {
+function findDisplayEntity ($msg, $textOnly = true, $entity = array() )   {
     global $show_html_default;
     
-    $entity = 0;
-    
+    $found = false;    
     if ($msg) {
-        if ( $msg->header->type0 == 'multipart' &&
-             ( $msg->header->type1 == 'alternative' ||
-               $msg->header->type1 == 'mixed' ||              
-               $msg->header->type1 == 'related' ) &&
-             $show_html_default && ! $textOnly ) {
-            $entity = findDisplayEntityHTML($msg);
-        }
-
-        // Show text/plain or text/html -- the first one we find.
-        if ( $entity == 0 &&
-             $msg->header->type0 == 'text' &&
-             ( $msg->header->type1 == 'plain' ||
-               $msg->header->type1 == 'html' ) &&
-             isset($msg->header->entity_id) ) {
-            $entity = $msg->header->entity_id;
-        }
-    
-        $i = 0;
-        while ($entity == 0 && isset($msg->entities[$i]) ) {
-            $entity = findDisplayEntity($msg->entities[$i], $textOnly);
-            $i++;
-        }
+        $type = $msg->type0.'/'.$msg->type1;
+        if ( $type == 'multipart/alternative') {
+           $msg = findAlternativeEntity($msg, $textOnly);
+           if (count($msg->entities) == 0) {
+               $entity[] = $msg->entity_id;
+           } else {
+               $found = true;
+                $entity =findDisplayEntity($msg,$textOnly, $entity);
+           }
+       } else  if ( $type == 'multipart/related') {
+            $msgs = findRelatedEntity($msg);
+           for ($i = 0; $i < count($msgs); $i++) {
+               $msg = $msgs[$i];
+               if (count($msg->entities) == 0) {
+                   $entity[] = $msg->entity_id;
+               } else {
+                   $found = true;
+                    $entity =findDisplayEntity($msg,$textOnly, $entity);
+               }
+           }
+       } else if ( count($entity) == 0 &&
+             $msg->type0 == 'text' &&
+             ( $msg->type1 == 'plain' ||
+               $msg->type1 == 'html' ) &&
+             isset($msg->entity_id) ) {
+            if (count($msg->entities) == 0) {
+               $entity[] = $msg->entity_id;
+            }
+        } 
+       $i = 0;
+       while ( isset($msg->entities[$i]) && count($entity) == 0 && !$found )  {
+           $entity = findDisplayEntity($msg->entities[$i], $textOnly, $entity);
+           $i++;
+       }
+    }
+    if ( !isset($entity[0]) ) {
+        $entity[]="";
     }
-    
     return( $entity );
 }
 
@@ -665,6 +297,7 @@ function findDisplayEntityHTML ($message) {
             isset($message->header->entity_id)) {
            return 0;
        }
+       
         $entity = findDisplayEntityHTML($message->entities[$i]);
         if ($entity != 0) {
             return $entity;
@@ -674,6 +307,46 @@ function findDisplayEntityHTML ($message) {
     return 0;
 }
 
+function findAlternativeEntity ($message, $textOnly) {
+    global $show_html_default;
+    /* if we are dealing with alternative parts then we choose the best 
+     * viewable message supported by SM.
+     */
+    if ($show_html_default && !$textOnly) {     
+       $alt_order = array ('text/plain','text/html');
+    } else {
+       $alt_order = array ('text/plain');
+    }
+    $best_view = 0;
+    $ent_id = 0;
+    $k = 0; 
+    for ($i = 0; $i < count($message->entities); $i ++) {
+        $type = $message->entities[$i]->header->type0.'/'.$message->entities[$i]->header->type1;
+       if ($type == 'multipart/related') {
+          $type = $message->entities[$i]->header->type;
+       }
+       for ($j = $k; $j < count($alt_order); $j++) {
+           if ($alt_order[$j] == $type && $j > $best_view) {
+               $best_view = $j;
+               $ent_id = $i;
+               $k = $j;
+           }
+       }
+    }
+    return $message->entities[$ent_id];
+}
+
+function findRelatedEntity ($message) {
+    $msgs = array(); 
+    for ($i = 0; $i < count($message->entities); $i ++) {
+        $type = $message->entities[$i]->header->type0.'/'.$message->entities[$i]->header->type1;
+        if ($message->header->type == $type) {
+           $msgs[] = $message->entities[$i];
+       }
+    }
+    return $msgs;
+}    
+
 /*
  * translateText
  * Extracted from strings.php 23/03/2002
@@ -729,61 +402,30 @@ function translateText(&$body, $wrap_at, $charset) {
     $body = '<pre>' . implode("\n", $body_ary) . '</pre>';
 }
 
-/* debugfunction for looping through entities and displaying correct entities */
-function listMyEntities ($message) {
-
-if ($message) {
-    if ($message->header->entity_id) {
-       echo "<tt>" . $message->header->entity_id . ' : ' . $message->header->type0 . '/' . $message->header->type1 . '<br>';
-    } 
-    if (!($message->header->type0 == 'message' &&  $message->header->type1 == 'rfc822')) {
-       if (isset($message->header->boundary) ) {
-           $ent_id = $message->header->entity_id;
-           $var = $message->header->boundary;
-           if ($var !='')
-           echo "<b>$ent_id boundary = $var</b><br>";
-       } 
-       if (isset($message->header->type) ) {
-           $var = $message->header->type;
-           if ($var !='')
-           echo "<b>$ent_id type = $var</b><br>";
-       } 
-       for ($i = 0; $message->entities[$i]; $i++) {
-           $msg = listMyEntities($message->entities[$i]);
-       }
-
-       if ($msg )  return $msg;
-    }
-}
-
-}
-
-
 
 /* This returns a parsed string called $body. That string can then
 be displayed as the actual message in the HTML. It contains
 everything needed, including HTML Tags, Attachments at the
 bottom, etc.
 */
-function formatBody($imap_stream, $message, $color, $wrap_at) {
+function formatBody($imap_stream, $message, $color, $wrap_at, $ent_num, $id, $mailbox='INBOX') {
     // this if statement checks for the entity to show as the
     // primary message. To add more of them, just put them in the
     // order that is their priority.
-    global $startMessage, $username, $key, $imapServerAddress, $imapPort, $body,
+    global $startMessage, $username, $key, $imapServerAddress, $imapPort,
            $show_html_default, $has_unsafe_images, $view_unsafe_images, $sort;
 
     $has_unsafe_images = 0;
 
-    $id = $message->header->id;
-
-    $urlmailbox = urlencode($message->header->mailbox);
-    // Get the right entity and redefine message to be this entity
-    // Pass the 0 to mean that we want the 'best' viewable one
-    $ent_num = findDisplayEntity ($message, 0);
+    if ($message->type0 == 'message' && $message->type1 == 'rfc822') {
+        $message = $message->entities[0];
+    }
+    $urlmailbox = urlencode($message->mailbox);
     $body_message = getEntity($message, $ent_num);
     if (($body_message->header->type0 == 'text') ||
         ($body_message->header->type0 == 'rfc822')) {
-        $body = mime_fetch_body ($imap_stream, $id, $ent_num);
+       $body = mime_fetch_body ($imap_stream, $id, $ent_num);
+       
         $body = decodeBody($body, $body_message->header->encoding);
         $hookResults = do_hook("message_body", $body);
         $body = $hookResults[1];
@@ -794,12 +436,11 @@ function formatBody($imap_stream, $message, $color, $wrap_at) {
                 $body = strip_tags( $body );
                 translateText($body, $wrap_at, $body_message->header->charset);
             } else {
-                $body = MagicHTML( $body, $id );
+                $body = magicHTML( $body, $id, $message, $mailbox );
             }
         } else {
             translateText($body, $wrap_at, $body_message->header->charset);
         }
-
         $body .= "<CENTER><SMALL><A HREF=\"../src/download.php?absolute_dl=true&amp;passed_id=$id&amp;passed_ent_id=$ent_num&amp;mailbox=$urlmailbox&amp;showHeaders=1\">". _("Download this as a file") ."</A></SMALL></CENTER><BR>";
         if ($has_unsafe_images) {
             if ($view_unsafe_images) {
@@ -811,10 +452,16 @@ function formatBody($imap_stream, $message, $color, $wrap_at) {
 
         /** Display the ATTACHMENTS: message if there's more than one part **/
         if (isset($message->entities[1])) {
-            $body .= formatAttachments ($message, $ent_num, $message->header->mailbox, $id);
+           /* Header-type alternative means we choose the best one to display 
+              so don't show the alternatives as attachment. Header-type related
+              means that the attachments are already part of the related message.
+           */   
+           if ($message->header->type1 !='related' && $message->header->type1 !='alternative') {
+               $body .= formatAttachments ($message, $ent_num, $mailbox, $id);
+           }
         }
     } else {
-        $body = formatAttachments ($message, -1, $message->header->mailbox, $id);
+        $body = formatAttachments ($message, -1, $message->mailbox, $id);
     }
     return ($body);
 }
@@ -850,23 +497,14 @@ function formatAttachments($message, $ent_id, $mailbox, $id) {
        }
        if ($type0 =='message' && $type1 == 'rfc822') {
         
-            $filename = decodeHeader($message->header->filename);
-            if (trim($filename) == '') {
-                if (trim($name) == '') {
-                    $display_filename = 'untitled-[' . $message->header->entity_id . ']' ;
-                } else {
-                    $display_filename = $name;
-                    $filename = $name;
-                }
-            } else {
-                $display_filename = $filename;
-            }
-
+            $filename = decodeHeader($message->header->subject);
+            $display_filename = $filename;
+            
             $urlMailbox = urlencode($mailbox);
-            $ent = urlencode($message->header->entity_id);
+            $ent = urlencode($message->entity_id);
 
             $DefaultLink =
-                "../src/download.php?startMessage=$startMessage&amp;passed_id=$id&amp;mailbox=$urlMailbox&amp;passed_ent_id=$ent";
+                "../src/read_body.php?startMessage=$startMessage&amp;passed_id=$id&amp;mailbox=$urlMailbox&amp;passed_ent_id=$ent";
             if ($where && $what) {
                 $DefaultLink .= '&amp;where=' . urlencode($where) . '&amp;what=' . urlencode($what);
             }
@@ -895,12 +533,16 @@ function formatAttachments($message, $ent_id, $mailbox, $id) {
                         '</b>&nbsp;&nbsp;</small></TD>' .
                         "<TD><SMALL>[ $type0/$type1 ]&nbsp;</SMALL></TD>" .
                         '<TD><SMALL>';
-            if ($message->header->description) {
-                $body .= '<b>' . htmlspecialchars(_($message->header->description)) . '</b>';
-            }
+           $from_o = $message->header->from;
+           if (isset($from_o)) {
+               $from_name = $from_o->getAddress(false);
+           } else {
+               $from_name = _("Unknown sender");
+           }
+           $from_name = decodeHeader(htmlspecialchars($from_name));
+            $body .= '<b>' . $from_name . '</b>';
             $body .= '</SMALL></TD><TD><SMALL>&nbsp;';
 
-
             $SkipSpaces = 1;
             foreach ($Links as $Val) {
                 if ($SkipSpaces) {
@@ -923,15 +565,15 @@ function formatAttachments($message, $ent_id, $mailbox, $id) {
             $type1 = strtolower($message->header->type1);
             $name = decodeHeader($message->header->name);
 
-            if ($message->header->entity_id != $ent_id) {
+            if ($message->entity_id != $ent_id) {
             $filename = decodeHeader($message->header->filename);
             if (trim($filename) == '') {
                 if (trim($name) == '') {
                     if ( trim( $message->header->id ) == '' )
-                        $display_filename = 'untitled-[' . $message->header->entity_id . ']' ;
+                        $display_filename = 'untitled-[' . $message->entity_id . ']' ;
                     else
                         $display_filename = 'cid: ' . $message->header->id;
-                    // $display_filename = 'untitled-[' . $message->header->entity_id . ']' ;
+                    // $display_filename = 'untitled-[' . $message->entity_id . ']' ;
                 } else {
                     $display_filename = $name;
                     $filename = $name;
@@ -941,7 +583,7 @@ function formatAttachments($message, $ent_id, $mailbox, $id) {
             }
 
             $urlMailbox = urlencode($mailbox);
-            $ent = urlencode($message->header->entity_id);
+            $ent = urlencode($message->entity_id);
 
             $DefaultLink =
                 "../src/download.php?startMessage=$startMessage&amp;passed_id=$id&amp;mailbox=$urlMailbox&amp;passed_ent_id=$ent";
@@ -1015,7 +657,6 @@ function decodeBody($body, $encoding) {
       $encoding == 'quoted_printable') {
      $body = quoted_printable_decode($body);
 
-
      while (ereg("=\n", $body))
         $body = ereg_replace ("=\n", "", $body);
 
@@ -1111,15 +752,12 @@ function encodeHeader ($string) {
 function find_ent_id( $id, $message ) {
     $ret = '';
     for ($i=0; $ret == '' && $i < count($message->entities); $i++) {
-       if (( $message->entities[$i]->header->type1 == 'alternative') ||         
-           ( $message->entities[$i]->header->type1 == 'related') ||     
-           ( $message->entities[$i]->header->type1 == 'mixed')) {       
+       if ( $message->entities[$i]->header->type0 == 'multipart')  {    
            $ret = find_ent_id( $id, $message->entities[$i] );
         } else {
             if ( strcasecmp( $message->entities[$i]->header->id, $id ) == 0 )
-                $ret = $message->entities[$i]->header->entity_id;
+                $ret = $message->entities[$i]->entity_id;
         }
-
     }
     return( $ret );
 }
@@ -1290,10 +928,12 @@ function sq_getnxtag($body, $offset){
          * A comment or an SGML declaration.
          */
         if (substr($body, $pos+1, 2) == "--"){
-            $gt = strpos($body, "-->", $pos)+2;
+            $gt = strpos($body, "-->", $pos);
             if ($gt === false){
                 $gt = strlen($body);
-            }
+            } else {
+               $gt += 2;
+           }
             return Array(false, false, false, $lt, $gt);
         } else {
             $gt = sq_findnxstr($body, $pos, ">");
@@ -1591,7 +1231,8 @@ function sq_fixatts($tagname,
                     $bad_attvals,
                     $add_attr_to_tag,
                     $message,
-                    $id
+                    $id,
+                   $mailbox
                     ){
     $me = "sq_fixatts";
     while (list($attname, $attvalue) = each($attary)){
@@ -1642,7 +1283,7 @@ function sq_fixatts($tagname,
          * Turn cid: urls into http-friendly ones.
          */
         if (preg_match("/^[\'\"]\s*cid:/si", $attvalue)){
-            $attary{$attname} = sq_cid2http($message, $id, $attvalue);
+            $attary{$attname} = sq_cid2http($message, $id, $attvalue, $mailbox);
         }
     }
     /**
@@ -1702,10 +1343,14 @@ function sq_fixstyle($message, $id, $content){
     }
 
     /**
-     * Fix stupid expression: declarations which lead to vulnerabilities
+     * Fix stupid css declarations which lead to vulnerabilities
      * in IE.
      */
-    $content = preg_replace("/expression\s*:/si", "idiocy:", $content);
+    $match   = Array('/expression/si',
+                    '/behaviou*r/si',
+                    '/binding/si');
+    $replace = Array('idiocy', 'idiocy', 'idiocy');
+    $content = preg_replace($match, $replace, $content);
     return $content;
 }
 
@@ -1718,7 +1363,7 @@ function sq_fixstyle($message, $id, $content){
  * @param  $cidurl   the cid: url.
  * @return           a string with a http-friendly url
  */
-function sq_cid2http($message, $id, $cidurl){
+function sq_cid2http($message, $id, $cidurl, $mailbox){
     /**
      * Get rid of quotes.
      */
@@ -1726,7 +1371,7 @@ function sq_cid2http($message, $id, $cidurl){
     $cidurl = str_replace($quotchar, "", $cidurl);
     $cidurl = substr(trim($cidurl), 4);
     $httpurl = $quotchar . "../src/download.php?absolute_dl=true&amp;" .
-        "passed_id=$id&amp;mailbox=" . urlencode($message->header->mailbox) .
+        "passed_id=$id&amp;mailbox=" . urlencode($mailbox) .
         "&amp;passed_ent_id=" . find_ent_id($cidurl, $message) . $quotchar;
     return $httpurl;
 }
@@ -1795,7 +1440,8 @@ function sq_sanitize($body,
                      $bad_attvals,
                      $add_attr_to_tag,
                      $message,
-                     $id
+                     $id,
+                    $mailbox
                      ){
     $me = "sq_sanitize";
     /**
@@ -1814,6 +1460,11 @@ function sq_sanitize($body,
     $open_tags = Array();
     $trusted = "<!-- begin sanitized html -->\n";
     $skip_content = false;
+    /**
+     * Take care of netscape's stupid javascript entities like
+     * &{alert('boo')};
+     */
+    $body = preg_replace("/&(\{.*?\};)/si", "&amp;\\1", $body);
 
     while (($curtag=sq_getnxtag($body, $curpos)) != FALSE){
         list($tagname, $attary, $tagtype, $lt, $gt) = $curtag;
@@ -1899,7 +1550,8 @@ function sq_sanitize($body,
                                                      $bad_attvals,
                                                      $add_attr_to_tag,
                                                      $message,
-                                                     $id
+                                                     $id,
+                                                    $mailbox
                                                      );
                             }
                             /**
@@ -1942,9 +1594,9 @@ function sq_sanitize($body,
  * @param  $id    the id of the message
  * @return        a string with html safe to display in the browser.
  */
-function magicHTML($body, $id){
+function magicHTML($body, $id, $message, $mailbox = 'INBOX'){
     global $attachment_common_show_images, $view_unsafe_images,
-        $has_unsafe_images, $message;
+        $has_unsafe_images;
     /**
      * Don't display attached images in HTML mode.
      */
@@ -1978,7 +1630,9 @@ function magicHTML($body, $id){
                          "/.*/" =>
                          Array(
                                "/target/si",
-                               "/^on.*/si"
+                               "/^on.*/si",
+                              "/^dynsrc/si",
+                              "/^data.*/si"
                                )
                          );
 
@@ -1986,28 +1640,55 @@ function magicHTML($body, $id){
     $bad_attvals = Array(
         "/.*/" =>
             Array(
-                "/^src|background|href|action/i" =>
+                "/^src|background/i" =>
                     Array(
                           Array(
                                 "|^([\'\"])\s*\.\./.*([\'\"])|si",
-                                "/^([\'\"])\s*\S+script\s*:.*([\'\"])/si"
+                                "/^([\'\"])\s*\S+script\s*:.*([\'\"])/si",
+                               "/^([\'\"])\s*mocha\s*:*.*([\'\"])/si",
+                               "/^([\'\"])\s*about\s*:.*([\'\"])/si"
                                 ),
                           Array(
                                 "\\1$secremoveimg\\2",
-                                "\\1$secremoveimg\\2"
+                                "\\1$secremoveimg\\2",
+                               "\\1$secremoveimg\\2",
+                               "\\1$secremoveimg\\2"
+                                )
+                        ),
+                "/^href|action/i" =>
+                    Array(
+                          Array(
+                                "|^([\'\"])\s*\.\./.*([\'\"])|si",
+                                "/^([\'\"])\s*\S+script\s*:.*([\'\"])/si",
+                               "/^([\'\"])\s*mocha\s*:*.*([\'\"])/si",
+                               "/^([\'\"])\s*about\s*:.*([\'\"])/si"
+                                ),
+                          Array(
+                                "\\1#\\2",
+                                "\\1#\\2",
+                               "\\1#\\2",
+                               "\\1#\\2"
                                 )
                         ),
                 "/^style/si" =>
                     Array(
                           Array(
-                                "/expression\s*:/si",
+                                "/expression/si",
+                               "/binding/si",
+                               "/behaviou*r/si",
                                 "|url\(([\'\"])\s*\.\./.*([\'\"])\)|si",
-                                "/url\(([\'\"])\s*\S+script:.*([\'\"])\)/si"
+                                "/url\(([\'\"])\s*\S+script\s*:.*([\'\"])\)/si",
+                               "/url\(([\'\"])\s*mocha\s*:.*([\'\"])\)/si",
+                               "/url\(([\'\"])\s*about\s*:.*([\'\"])\)/si"
                                ),
                           Array(
-                                "idiocy:",
-                                "url(\\1$secremoveimg\\2)",
-                                "url(\\1$secremoveimg\\2)"
+                                "idiocy",
+                               "idiocy",
+                               "idiocy",
+                                "url(\\1#\\2)",
+                                "url(\\1#\\2)",
+                               "url(\\1#\\2)",
+                               "url(\\1#\\2)"
                                )
                           )
                 )
@@ -2017,30 +1698,14 @@ function magicHTML($body, $id){
          * Remove any references to http/https if view_unsafe_images set
          * to false.
          */
-        $addendum = Array(
-          "/.*/" =>
-            Array(
-                "/^src|background/i" =>
-                    Array(
-                          Array(
-                                "/^([\'\"])\s*https*:.*([\'\"])/si"
-                                ),
-                          Array(
-                                "\\1$secremoveimg\\2"
-                                )
-                        ),
-                "/^style/si" =>
-                    Array(
-                          Array(
-                                "/url\(([\'\"])\s*https*:.*([\'\"])\)/si"
-                               ),
-                          Array(
-                                "url(\\1$secremoveimg\\2)"
-                               )
-                          )
-                )
-          );
-        $bad_attvals = array_merge($bad_attvals, $addendum);
+         array_push($bad_attvals{'/.*/'}{'/^src|background/i'}[0],
+                    '/^([\'\"])\s*https*:.*([\'\"])/si');
+         array_push($bad_attvals{'/.*/'}{'/^src|background/i'}[1],
+                    "\\1$secremoveimg\\2");
+         array_push($bad_attvals{'/.*/'}{'/^style/si'}[0],
+                    '/url\(([\'\"])\s*https*:.*([\'\"])\)/si');
+         array_push($bad_attvals{'/.*/'}{'/^style/si'}[1],
+                    "url(\\1$secremoveimg\\2)");
     }
 
     $add_attr_to_tag = Array(
@@ -2055,11 +1720,13 @@ function magicHTML($body, $id){
                            $bad_attvals,
                            $add_attr_to_tag,
                            $message,
-                           $id
+                           $id,
+                          $mailbox
                            );
     if (preg_match("|$secremoveimg|si", $trusted)){
         $has_unsafe_images = true;
     }
     return $trusted;
 }
+
 ?>
\ No newline at end of file