From 2aa2b332f6e216ceeabc36ef6b942c40d91bda5a Mon Sep 17 00:00:00 2001 From: alecpl <alec@alec.pl> Date: Wed, 08 Sep 2010 05:40:39 -0400 Subject: [PATCH] - Small performance improvements --- program/include/rcube_imap_generic.php | 149 +++++++++++++++++++++++++++---------------------- 1 files changed, 82 insertions(+), 67 deletions(-) diff --git a/program/include/rcube_imap_generic.php b/program/include/rcube_imap_generic.php index d159b90..0c69c23 100644 --- a/program/include/rcube_imap_generic.php +++ b/program/include/rcube_imap_generic.php @@ -191,7 +191,7 @@ break; } if (!empty($this->prefs['debug_mode'])) { - write_log('imap', 'S: '. chop($buffer)); + write_log('imap', 'S: '. rtrim($buffer)); } $line .= $buffer; } while ($buffer[strlen($buffer)-1] != "\n"); @@ -201,7 +201,7 @@ private function multLine($line, $escape=false) { - $line = chop($line); + $line = rtrim($line); if (preg_match('/\{[0-9]+\}$/', $line)) { $out = ''; @@ -611,14 +611,21 @@ $host = $this->prefs['ssl_mode'] . '://' . $host; } - $this->fp = @fsockopen($host, $this->prefs['port'], $errno, $errstr, 10); + // Connect + if ($this->prefs['timeout'] > 0) + $this->fp = @fsockopen($host, $this->prefs['port'], $errno, $errstr, $this->prefs['timeout']); + else + $this->fp = @fsockopen($host, $this->prefs['port'], $errno, $errstr); + if (!$this->fp) { $this->error = sprintf("Could not connect to %s:%d: %s", $host, $this->prefs['port'], $errstr); $this->errornum = -2; return false; } - stream_set_timeout($this->fp, 10); + if ($this->prefs['timeout'] > 0) + stream_set_timeout($this->fp, $this->prefs['timeout']); + $line = trim(fgets($this->fp, 8192)); if ($this->prefs['debug_mode'] && $line) { @@ -709,6 +716,7 @@ } $this->getNamespace(); $this->logged = true; + return true; } else { return false; @@ -741,31 +749,24 @@ if ($this->putLine("sel1 SELECT \"".$this->escape($mailbox).'"')) { do { - $line = chop($this->readLine(300)); - $a = explode(' ', $line); - if (count($a) == 3) { - $token = strtoupper($a[2]); - if ($token == 'EXISTS') { - $this->exists = (int) $a[1]; - } - else if ($token == 'RECENT') { - $this->recent = (int) $a[1]; - } + $line = rtrim($this->readLine(512)); + + if (preg_match('/^\* ([0-9]+) (EXISTS|RECENT)$/', $line, $m)) { + $token = strtolower($m[2]); + $this->$token = (int) $m[1]; } else if (preg_match('/\[?PERMANENTFLAGS\s+\(([^\)]+)\)\]/U', $line, $match)) { $this->permanentflags = explode(' ', $match[1]); } } while (!$this->startsWith($line, 'sel1', true, true)); - if (strcasecmp($a[1], 'OK') == 0) { + if ($this->parseResult($line) == 0) { $this->selected = $mailbox; return true; } - else { - $this->error = "Couldn't select $mailbox"; - } } + $this->error = "Couldn't select $mailbox"; return false; } @@ -832,7 +833,7 @@ return false; } do { - $line = chop($this->readLine()); + $line = rtrim($this->readLine()); if ($this->startsWith($line, '* SORT')) { $data .= substr($line, 7); } else if (preg_match('/^[0-9 ]+$/', $line)) { @@ -841,7 +842,6 @@ } while (!$this->startsWith($line, 's ', true, true)); $result_code = $this->parseResult($line); - if ($result_code != 0) { $this->error = "Sort: $line"; return false; @@ -917,7 +917,7 @@ $result = array(); do { - $line = chop($this->readLine(200)); + $line = rtrim($this->readLine(200)); $line = $this->multLine($line); if (preg_match('/^\* ([0-9]+) FETCH/', $line, $m)) { @@ -1046,7 +1046,7 @@ $result = -1; if ($this->putLine("fuid FETCH $id (UID)")) { do { - $line = chop($this->readLine(1024)); + $line = rtrim($this->readLine(1024)); if (preg_match("/^\* $id FETCH \(UID (.*)\)/i", $line, $r)) { $result = $r[1]; } @@ -1104,10 +1104,8 @@ if (!$line) break; - $a = explode(' ', $line); - - if (($line[0] == '*') && ($a[2] == 'FETCH')) { - $id = $a[1]; + if (preg_match('/^\* ([0-9]+) FETCH/', $line, $m)) { + $id = $m[1]; $result[$id] = new rcube_mail_header; $result[$id]->id = $id; @@ -1181,7 +1179,7 @@ // to the next valid header line. do { - $line = chop($this->readLine(300), "\r\n"); + $line = rtrim($this->readLine(300), "\r\n"); // The preg_match below works around communigate imap, which outputs " UID <number>)". // Without this, the while statement continues on and gets the "FH0 OK completed" message. @@ -1260,7 +1258,7 @@ } break; case 'in-reply-to': - $result[$id]->in_reply_to = preg_replace('/[\n<>]/', '', $string); + $result[$id]->in_reply_to = str_replace(array("\n", '<', '>'), '', $string); break; case 'references': $result[$id]->references = $string; @@ -1283,8 +1281,6 @@ break; } // end switch () } // end while () - } else { - $a = explode(' ', $line); } // process flags @@ -1293,7 +1289,6 @@ $flags_a = explode(' ', $flags_str); if (is_array($flags_a)) { - // reset($flags_a); foreach($flags_a as $flag) { $flag = strtoupper($flag); if ($flag == 'SEEN') { @@ -1444,7 +1439,7 @@ } do { - $line = $this->readLine(1000); + $line = $this->readLine(); if ($line[0] == '*') { $c++; } @@ -1500,7 +1495,7 @@ { $node = array(); if ($str[$begin] != '(') { - $stop = $begin + strspn($str, "1234567890", $begin, $end - $begin); + $stop = $begin + strspn($str, '1234567890', $begin, $end - $begin); $msg = substr($str, $begin, $stop - $begin); if ($msg == 0) return $node; @@ -1546,30 +1541,40 @@ function thread($folder, $algorithm='REFERENCES', $criteria='', $encoding='US-ASCII') { + $old_sel = $this->selected; + if (!$this->select($folder)) { - return false; + return false; + } + + // return empty result when folder is empty and we're just after SELECT + if ($old_sel != $folder && !$this->exists) { + return array(array(), array(), array()); } $encoding = $encoding ? trim($encoding) : 'US-ASCII'; $algorithm = $algorithm ? trim($algorithm) : 'REFERENCES'; $criteria = $criteria ? 'ALL '.trim($criteria) : 'ALL'; + $data = ''; if (!$this->putLineC("thrd1 THREAD $algorithm $encoding $criteria")) { return false; } do { - $line = trim($this->readLine(10000)); - if (preg_match('/^\* THREAD/', $line)) { - $str = trim(substr($line, 8)); - $depthmap = array(); - $haschildren = array(); - $tree = $this->parseThread($str, 0, strlen($str), null, null, 0, $depthmap, $haschildren); + $line = trim($this->readLine()); + if ($this->startsWith($line, '* THREAD')) { + $data .= substr($line, 9); + } else if (preg_match('/^[0-9() ]+$/', $line)) { + $data .= $line; } } while (!$this->startsWith($line, 'thrd1', true, true)); $result_code = $this->parseResult($line); if ($result_code == 0) { - return array($tree, $depthmap, $haschildren); + $depthmap = array(); + $haschildren = array(); + $tree = $this->parseThread($data, 0, strlen($data), null, null, 0, $depthmap, $haschildren); + return array($tree, $depthmap, $haschildren); } $this->error = "Thread: $line"; @@ -1578,12 +1583,19 @@ function search($folder, $criteria, $return_uid=false) { + $old_sel = $this->selected; + if (!$this->select($folder)) { return false; } + // return empty result when folder is empty and we're just after SELECT + if ($old_sel != $folder && !$this->exists) { + return array(); + } + $data = ''; - $query = 'srch1 ' . ($return_uid ? 'UID ' : '') . 'SEARCH ' . chop($criteria); + $query = 'srch1 ' . ($return_uid ? 'UID ' : '') . 'SEARCH ' . trim($criteria); if (!$this->putLineC($query)) { return false; @@ -1650,8 +1662,11 @@ $command = 'LIST'; } + $ref = $this->escape($ref); + $mailbox = $this->escape($mailbox); + // send command - if (!$this->putLine($key." ".$command." \"". $this->escape($ref) ."\" \"". $this->escape($mailbox) ."\"")) { + if (!$this->putLine($key." ".$command." \"". $ref ."\" \"". $mailbox ."\"")) { $this->error = "Couldn't send $command command"; return false; } @@ -1660,16 +1675,15 @@ do { $line = $this->readLine(500); $line = $this->multLine($line, true); - $a = explode(' ', $line); + $line = trim($line); - if (($line[0] == '*') && ($a[1] == $command)) { - $line = rtrim($line); - // split one line - $a = rcube_explode_quoted_string(' ', $line); - // last string is folder name - $folders[] = preg_replace(array('/^"/', '/"$/'), '', $this->unEscape($a[count($a)-1])); - // second from last is delimiter - $delim = trim($a[count($a)-2], '"'); + if (preg_match('/^\* '.$command.' \(([^\)]*)\) "*([^"]+)"* (.*)$/', $line, $m)) { + // folder name + $folders[] = preg_replace(array('/^"/', '/"$/'), '', $this->unEscape($m[3])); + // attributes +// $attrib = explode(' ', $m[1]); + // delimiter +// $delim = $m[2]; } } while (!$this->startsWith($line, $key, true)); @@ -1708,7 +1722,7 @@ } do { - $line = $this->readLine(1000); + $line = $this->readLine(1024); $line = $this->multLine($line); if (preg_match('/BODY\[([0-9\.]+)\.'.$type.'\]/', $line, $matches)) { @@ -1752,12 +1766,11 @@ $mode = 0; } - $reply_key = '* ' . $id; - $result = false; - // format request - $key = 'ftch0'; - $request = $key . ($is_uid ? ' UID' : '') . " FETCH $id (BODY.PEEK[$part])"; + $reply_key = '* ' . $id; + $key = 'ftch0'; + $request = $key . ($is_uid ? ' UID' : '') . " FETCH $id (BODY.PEEK[$part])"; + // send request if (!$this->putLine($request)) { return false; @@ -1765,11 +1778,12 @@ // receive reply line do { - $line = chop($this->readLine(1000)); + $line = rtrim($this->readLine(1024)); $a = explode(' ', $line); } while (!($end = $this->startsWith($line, $key, true)) && $a[2] != 'FETCH'); - $len = strlen($line); + $len = strlen($line); + $result = false; // handle empty "* X FETCH ()" response if ($line[$len-1] == ')' && $line[$len-2] != '(') { @@ -1865,7 +1879,7 @@ $line = $this->readLine(1024); } while (!$this->startsWith($line, $key, true)); - if ($result) { + if ($result !== false) { if ($file) { fwrite($file, $result); } else if ($print) { @@ -1982,7 +1996,7 @@ return false; } - function appendFromFile($folder, $path, $headers=null, $separator="\n\n") + function appendFromFile($folder, $path, $headers=null) { if (!$folder) { return false; @@ -1998,14 +2012,16 @@ return false; } + $body_separator = "\r\n\r\n"; $len = filesize($path); + if (!$len) { return false; } if ($headers) { $headers = preg_replace('/[\r\n]+$/', '', $headers); - $len += strlen($headers) + strlen($separator); + $len += strlen($headers) + strlen($body_separator); } // send APPEND command @@ -2021,7 +2037,7 @@ // send headers with body separator if ($headers) { - $this->putLine($headers . $separator, false); + $this->putLine($headers . $body_separator, false); } // send file @@ -2089,7 +2105,7 @@ // get line(s) containing quota info if ($this->putLine('QUOT1 GETQUOTAROOT "INBOX"')) { do { - $line = chop($this->readLine(5000)); + $line = rtrim($this->readLine(5000)); if ($this->startsWith($line, '* QUOTA ')) { $quota_lines[] = $line; } @@ -2099,7 +2115,7 @@ // return false if not found, parse if found $min_free = PHP_INT_MAX; foreach ($quota_lines as $key => $quota_line) { - $quota_line = preg_replace('/[()]/', '', $quota_line); + $quota_line = str_replace(array('(', ')'), '', $quota_line); $parts = explode(' ', $quota_line); $storage_part = array_search('STORAGE', $parts); @@ -2212,4 +2228,3 @@ } -?> -- Gitblit v1.9.1