Index: database/updates.inc
===================================================================
RCS file: /cvs/drupal/drupal/database/updates.inc,v
retrieving revision 1.212
diff -u -F^f -r1.212 updates.inc
--- database/updates.inc	7 Mar 2006 19:08:46 -0000	1.212
+++ database/updates.inc	8 Mar 2006 00:25:17 -0000
@@ -1684,6 +1684,8 @@ function system_update_177() {
     'pass_subject',
     'pass_body',
   );
+
+  $ret = array();
   foreach ($message_ids as $message_id) {
     if ($admin_setting = variable_get('user_mail_' . $message_id, FALSE)) {
       $admin_setting = '<pre>'. $admin_setting .'</pre>'; //else it renders horrible
@@ -1698,3 +1700,70 @@ function system_update_177() {
   return $ret;
 }
 
+function _update_178_url_fix($text) {
+  //key is attribute to replace.
+  $urlpatterns['href'] = "/<a[^>]+href=\"([^\"]+)/i";
+  $urlpatterns['src']  = "/<img[^>]+src=\"([^\"]+)/i";
+
+  foreach ($urlpatterns as $type => $pattern) {
+    preg_match_all($pattern, $text, $matches);
+    foreach($matches[1] as $url) {
+      if ($url != '' && !strstr($url, 'mailto:') && !strstr($url, '://') && !strstr($url, '../') && !strstr($url, './') && $url[0] != '/' && $url[0] != '#') {
+        $text = preg_replace('|'. $type .'\s*=\s*"'. $url .'\s*"|', $type. '="'.base_path(). $url  .'"', $text);
+      }
+    }
+  }
+  return $text;
+}
+
+/**
+ * Update relative paths in node and comment content if clean_urls are enabled.
+ */
+function system_update_178() {
+
+  if(variable_get('clean_url', 0) == 1) {
+    // Multi-part update
+    if (!isset($_SESSION['system_update_178_comment'])) {
+      $_SESSION['system_update_178_comment'] = 0;
+      $_SESSION['system_update_178_node'] = 0;
+      $_SESSION['system_update_178_comment_max'] = db_result(db_query('SELECT MAX(cid) FROM {comments}'));
+      $_SESSION['system_update_178_node_max'] = db_result(db_query('SELECT MAX(nid) FROM {node_revisions}'));
+    }
+
+    $limit = 20;
+    $result = db_query_range("SELECT cid, comment FROM {comments} WHERE cid > %d", $_SESSION['system_update_178_comment'], 0, $limit);
+    while ($comment = db_fetch_object($result)) {
+      $_SESSION['system_update_178_comment'] = $comment->cid;
+      $comment->comment = _update_178_url_fix($comment->comment);
+      db_query("UPDATE {comments} SET comment = '%s' WHERE cid = %d", $comment->comment, $comment->cid);
+    }
+
+    $result = db_query_range("SELECT nid, vid, teaser, body FROM {node_revisions} WHERE nid > %d", $_SESSION['system_update_178_node'], 0, $limit);
+    while ($node = db_fetch_object($result)) {
+      // Catch the infinite loop for node revisions. We ignore updating
+      // content past the 20th revision.
+      if ($_SESSION['system_update_178_node'] == $node->nid) {
+        $_SESSION['system_update_178_node']++;
+      }
+      else {
+        $_SESSION['system_update_178_node'] = $node->nid;
+      }
+      $node->teaser = _update_178_url_fix($node->teaser);
+      $node->body = _update_178_url_fix($node->body);
+      db_query('UPDATE {node_revisions} SET body = "%s", teaser = "%s" WHERE nid = %d AND vid = %d', $node->body, $node->teaser, $node->nid, $node->vid);
+    }
+
+    if ($_SESSION['system_update_178_comment'] == $_SESSION['system_update_178_comment_max'] &&
+        $_SESSION['system_update_178_node'] == $_SESSION['system_update_178_node_max']) {
+      unset($_SESSION['system_update_178_comment']);
+      unset($_SESSION['system_update_178_comment_max']);
+      unset($_SESSION['system_update_178_node']);
+      unset($_SESSION['system_update_178_node_max']);
+      return array();
+    }
+    return array('#finished' => $_SESSION['system_update_178_comment'] / $_SESSION['system_update_178_comment_max']
+                                && $_SESSION['system_update_178_node'] / $_SESSION['system_update_178_node_max']);
+  }
+
+  return array();
+}