Changeset 2416

Timestamp:

03/07/2005 08:56:47 AM (20 years ago)

Author:

matt

Message:

Cleaned up enclosures, factored out header code, fixed validation -
​http://mosquito.wordpress.org/view.php?id=963

Location:

trunk/wp-includes

Files:

• feed-functions.php (modified) (1 diff)
• functions.php (modified) (2 diffs)

trunk/wp-includes/feed-functions.php

@@ -147 +147 @@
                     foreach($val as $enc) {
                         $enclosure = split( "\n", $enc );
-                        print "<enclosure url='".trim( $enclosure[ 0 ] )."' length='".trim( $enclosure[ 1 ] )."' type='".trim( $enclosure[ 2 ] )."'/>\n";
+                        print "<enclosure url='".trim( htmlspecialchars($enclosure[ 0 ]) )."' length='".trim( $enclosure[ 1 ] )."' type='".trim( $enclosure[ 2 ] )."'/>\n";
                     }
                 }

trunk/wp-includes/functions.php

@@ -749 +749 @@
     endforeach;
 
-    foreach ($post_links as $url){
-                if( $url != '' && in_array($url, $pung) == false ) {
-                    set_time_limit( 60 ); 
-                    $file = str_replace( "http://", "", $url );
-                    $host = substr( $file, 0, strpos( $file, "/" ) );
-                    $file = substr( $file, strpos( $file, "/" ) );
-                    $headers = "HEAD $file HTTP/1.1\r\nHOST: $host\r\n\r\n";
-                    $port    = 80;
-                    $timeout = 3;
-                    $fp = @fsockopen($host, $port, $err_num, $err_msg, $timeout);
-                    if( $fp ) {
-                        fputs($fp, $headers );
-                        $response = '';
-                        while ( !feof($fp) && strpos( $response, "\r\n\r\n" ) == false )
-                            $response .= fgets($fp, 2048);
-                        fclose( $fp );
-                    } else {
-                        $response = '';
-                    }
-                    if( $response != '' ) {
-                        $len = substr( $response, strpos( $response, "Content-Length:" ) + 16 );
-                        $len = substr( $len, 0, strpos( $len, "\n" ) );
-                        $type = substr( $response, strpos( $response, "Content-Type:" ) + 14 );
-                        $type = substr( $type, 0, strpos( $type, "\n" ) + 1 );
-                        $allowed_types = array( 'video', 'audio' );
-                        if( in_array( substr( $type, 0, strpos( $type, "/" ) ), $allowed_types ) ) {
-                            $meta_value = "$url\n$len\n$type\n";
-                            $query = "INSERT INTO `".$wpdb->postmeta."` ( `meta_id` , `post_id` , `meta_key` , `meta_value` )
-                                VALUES ( NULL, '$post_ID', 'enclosure' , '".$meta_value."')";
-                            $wpdb->query( $query );
-                        }
-                    }
-                }
-        }
+    foreach ($post_links as $url) :
+        if ( $url != '' && in_array($url, $pung) == false ) {
+            if ( $headers = wp_get_http_headers( $url) ) {
+                $len  = $headers['content-length'];
+                $type = $headers['content-type'];
+                $allowed_types = array( 'video', 'audio' );
+                if( in_array( substr( $type, 0, strpos( $type, "/" ) ), $allowed_types ) ) {
+                    $meta_value = "$url\n$len\n$type\n";
+                    $wpdb->query( "INSERT INTO `".$wpdb->postmeta."` ( `post_id` , `meta_key` , `meta_value` )
+                    VALUES ( '$post_ID', 'enclosure' , '".$meta_value."')" );
+                }
+            }
+        }
+    endforeach;
+}
+
+function wp_get_http_headers( $url ) {
+    set_time_limit( 60 ); 
+    $parts = parse_url( $url );
+    $file  = $parts['path'] . $parts['query'];
+    $host  = $parts['host'];
+    if ( !isset( $parts['port'] ) )
+        $parts['port'] = 80;
+
+    $head = "HEAD $file HTTP/1.1\r\nHOST: $host\r\n\r\n";
+
+    $fp = @fsockopen($host, $parts['port'], $err_num, $err_msg, 3);
+    if ( !$fp )
+        return false;
+
+    $response = '';
+    fputs( $fp, $head );
+    while ( !feof( $fp ) && strpos( $response, "\r\n\r\n" ) == false )
+        $response .= fgets( $fp, 2048 );
+    fclose( $fp );
+    preg_match_all('/(.*?): (.*)\r/', $response, $matches);
+    $count = count($matches[1]);
+    for ( $i = 0; $i < $count; $i++) {
+        $key = strtolower($matches[1][$i]);
+        $headers["$key"] = $matches[2][$i];
+    }
+
+    preg_match('/.*([0-9]{3}).*/', $response, $return);
+    $headers['response'] = $return[1]; // HTTP response code eg 204, 200, 404
+    return $headers;
 }
 
@@ -864 +875 @@
 
     }
+}
 
     if (isset($wp_filter[$tag]))