From a1af15741005290157baa6b1df724a5becee1738 Mon Sep 17 00:00:00 2001
From: Andrew Dolgov <fox@bah.org.ru>
Date: Thu, 22 Apr 2010 10:10:49 +0400
Subject: [PATCH] Fix several problems with image download

Fix fetch_file_contents, so that it returns the contents even if the data isn't
an image. This is needed because the get_favicon_url function tries to download
the webpage using this function, to see if there is a favicon in the page.

The function now takes an optional $type parameter. This parameter control if
the calling function cares about the content-type, or if the function should
just return everything. If the $type parameter is set, the content-type should
contain the string contained in $type, otherwise the function returns false.

The second problem solved with this patch, is that the temporary file that
should contain the image was empty in some cases. I never found out why this
happended, but as curl_exec is capable of returning the fetched data, thus
eliminating the need for the temporary file all together, the function have
been changed to use this way of obtaining the data.

The last problem fixed by this patch is that curl will now follow redirects.

Author: Klaus S. Madsen <ksm@42.dk>
---
 functions.php | 50 +++++++++++++++++++++++++-------------------------
 1 file changed, 25 insertions(+), 25 deletions(-)

diff --git a/functions.php b/functions.php
index 78adfdbcb..7320f0fac 100644
--- a/functions.php
+++ b/functions.php
@@ -353,33 +353,31 @@
 		}
 	}
 
-	function fetch_file_contents($url) {
+	function fetch_file_contents($url, $type) {
 		if (USE_CURL_FOR_ICONS) {
-			$tmpfile = tempnam(TMP_DIRECTORY, "ttrss-tmp");
-
 			$ch = curl_init($url);
-			$fp = fopen($tmpfile, "w");
 
-			if ($fp) {
-				curl_setopt($ch, CURLOPT_FILE, $fp);
-				curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 15);
-				curl_setopt($ch, CURLOPT_TIMEOUT, 45);
-				curl_exec($ch);
+			curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 15);
+			curl_setopt($ch, CURLOPT_TIMEOUT, 45);
+			curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
+			curl_setopt($ch, CURLOPT_MAXREDIRS, 20);
+			curl_setopt($ch, CURLOPT_BINARYTRANSFER, true);
+			curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
 
-				if (strpos(curl_getinfo($ch, CURLINFO_CONTENT_TYPE), "image/") !== false) {
-					curl_close($ch);
-					fclose($fp);					
-					$contents = file_get_contents($tmpfile);
-				} else {
-					curl_close($ch);
-					fclose($fp);					
-				}
+			$contents = curl_exec($ch);
+			if ($contents === false) {
+				curl_close($ch);
+				return false;
 			}
 
-			unlink($tmpfile);
+			$content_type = curl_getinfo($ch, CURLINFO_CONTENT_TYPE);
+			curl_close($ch);
+
+			if ($type && strpos($content_type, "$type") === false) {
+				return false;
+			}
 
 			return $contents;
-
 		} else {
 			return file_get_contents($url);
 		}
@@ -491,14 +489,16 @@
 		$icon_file = ICONS_DIR . "/$feed.ico";
 
 		if ($favicon_url && !file_exists($icon_file)) {
-			$contents = fetch_file_contents($favicon_url);
+			$contents = fetch_file_contents($favicon_url, "image");
 
-			$fp = fopen($icon_file, "w");
+			if ($contents) {
+				$fp = fopen($icon_file, "w");
 
-			if ($fp) {
-				fwrite($fp, $contents);
-				fclose($fp);
-				chmod($icon_file, 0644);
+				if ($fp) {
+					fwrite($fp, $contents);
+					fclose($fp);
+					chmod($icon_file, 0644);
+				}
 			}
 		}