Changeset 32164

Timestamp:

04/20/2015 05:41:37 AM (10 years ago)

Author:

pento

Message:

Clean up some edge cases in sanitize_sql_orderby().

Props vortfu, dd32.

Location:

trunk

Files:

• src/wp-includes/formatting.php (modified) (1 diff)
• tests/phpunit/tests/formatting/SanitizeOrderby.php (modified) (1 diff)

trunk/src/wp-includes/formatting.php

@@ -1363 +1363 @@
 
 /**
- * Ensures a string is a valid SQL order by clause.
- *
- * Accepts one or more columns, with or without ASC/DESC, and also accepts
- * RAND().
+ * Ensures a string is a valid SQL 'order by' clause.
+ *
+ * Accepts one or more columns, with or without a sort order (ASC / DESC).
+ * e.g. 'column_1', 'column_1, column_2', 'column_1 ASC, column_2 DESC' etc.
+ *
+ * Also accepts 'RAND()'.
  *
  * @since 2.5.1
  *
- * @param string $orderby Order by string to be checked.
- * @return false|string Returns the order by clause if it is a match, false otherwise.
- */
-function sanitize_sql_orderby( $orderby ){
-    preg_match('/^\s*([a-z0-9_]+(\s+(ASC|DESC))?(\s*,\s*|\s*$))+|^\s*RAND\(\s*\)\s*$/i', $orderby, $obmatches);
-    if ( !$obmatches )
-        return false;
-    return $orderby;
+ * @param string $orderby Order by clause to be validated.
+ * @return string|bool Returns $orderby if valid, false otherwise.
+ */
+function sanitize_sql_orderby( $orderby ) {
+    if ( preg_match( '/^\s*(([a-z0-9_]+|`[a-z0-9_]+`)(\s+(ASC|DESC))?\s*(,\s*(?=[a-z0-9_`])|$))+$/i', $orderby ) || preg_match( '/^\s*RAND\(\s*\)\s*$/i', $orderby ) ) {
+        return $orderby;
+    }
+    return false;
 }
 

trunk/tests/phpunit/tests/formatting/SanitizeOrderby.php

@@ -1 +1 @@
 <?php
 
-/* // @todo These tests need to be rewritten for sanitize_sql_orderby
+/**
+ * @group sanitize_sql_orderby
+ */
 class Tests_Formatting_SanitizeOrderby extends WP_UnitTestCase {
-    function test_empty() {
-        $cols = array('a' => 'a');
-        $this->assertEquals( '', sanitize_sql_orderby('', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('  ', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby("\t", $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby(null, $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby(0, $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('+', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('-', $cols) );
+
+    /**
+     * @covers ::sanitize_sql_orderby
+     * @dataProvider valid_orderbys
+     */
+    function test_valid( $orderby ) {
+        $this->assertEquals( $orderby, sanitize_sql_orderby( $orderby ) );
+    }
+    function valid_orderbys() {
+        return array(
+            array( '1' ),
+            array( '1 ASC' ),
+            array( '1 ASC, 2' ),
+            array( '1 ASC, 2 DESC' ),
+            array( '1 ASC, 2 DESC, 3' ),
+            array( '       1      DESC' ),
+            array( 'field ASC' ),
+            array( 'field1 ASC, field2' ),
+            array( 'field_1 ASC, field_2 DESC' ),
+            array( 'field1, field2 ASC' ),
+            array( '`field1`' ),
+            array( '`field1` ASC' ),
+            array( '`field` ASC, `field2`' ),
+            array( 'RAND()' ),
+            array( '   RAND(  )   ' ),
+        );
     }
 
-    function test_unknown_column() {
-        $cols = array('name' => 'post_name', 'date' => 'post_date');
-        $this->assertEquals( '', sanitize_sql_orderby('unknown_column', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('+unknown_column', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('-unknown_column', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('-unknown1,+unknown2,unknown3', $cols) );
-        $this->assertEquals( 'post_name ASC', sanitize_sql_orderby('name,unknown_column', $cols) );
-        $this->assertEquals( '', sanitize_sql_orderby('!@#$%^&*()_=~`\'",./', $cols) );
+    /**
+     * @covers ::sanitize_sql_orderby
+     * @dataProvider invalid_orderbys
+     */
+    function test_invalid( $orderby ) {
+        $this->assertFalse( sanitize_sql_orderby( $orderby ) );
     }
-
-    function test_valid() {
-        $cols = array('name' => 'post_name', 'date' => 'post_date', 'random' => 'rand()');
-        $this->assertEquals( 'post_name ASC', sanitize_sql_orderby('name', $cols) );
-        $this->assertEquals( 'post_name ASC', sanitize_sql_orderby('+name', $cols) );
-        $this->assertEquals( 'post_name DESC', sanitize_sql_orderby('-name', $cols) );
-        $this->assertEquals( 'post_date ASC, post_name ASC', sanitize_sql_orderby('date,name', $cols) );
-        $this->assertEquals( 'post_date ASC, post_name ASC', sanitize_sql_orderby(' date , name ', $cols) );
-        $this->assertEquals( 'post_name DESC, post_date ASC', sanitize_sql_orderby('-name,date', $cols) );
-        $this->assertEquals( 'post_name ASC, post_date ASC', sanitize_sql_orderby('name ,+ date', $cols) );
-        $this->assertEquals( 'rand() ASC', sanitize_sql_orderby('random', $cols) );
+    function invalid_orderbys() {
+        return array(
+            array( '' ),
+            array( '1 2' ),
+            array( '1, 2 3' ),
+            array( '1 DESC, ' ),
+            array( 'field-1' ),
+            array( 'field DESC,' ),
+            array( 'field1 field2' ),
+            array( 'field RAND()' ),
+            array( 'RAND() ASC' ),
+            array( '`field1` ASC, `field2' ),
+            array( 'field, !@#$%^' ),
+        );
     }
 }
-*/