Escape colons in BagOStuff key segments
For the sake of safety and correctness, the following BagOStuff::makeKey() invocations should return distinct keys: $cache->makeKey( 'ab:', 'cd' ); $cache->makeKey( 'ab', ':cd' ); That is not currently the case, because while we use ':' as a key path separator, we don't escape ':' in the input supplied to makeKey(). So, make BagOStuff::makeKeyInternal() URL-encode colons. To prevent this from messing up the max. key length calculations, reproduce this logic in MemcachedBagOStuff::makeKeyInternal(), in lieu of having the method call its parent. Change-Id: I83ea7e7336a1c9e64aa42284c2517089a736efe5
This commit is contained in:
parent
cdb5432728
commit
0c9fb12265
4 changed files with 17 additions and 7 deletions
|
|
@ -627,7 +627,11 @@ abstract class BagOStuff implements LoggerAwareInterface {
|
|||
* @return string
|
||||
*/
|
||||
public function makeKeyInternal( $keyspace, $args ) {
|
||||
$key = $keyspace . ':' . implode( ':', $args );
|
||||
$key = $keyspace;
|
||||
foreach ( $args as $arg ) {
|
||||
$arg = str_replace( ':', '%3A', $arg );
|
||||
$key = $key . ':' . $arg;
|
||||
}
|
||||
return strtr( $key, ' ', '_' );
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -118,9 +118,10 @@ class MemcachedBagOStuff extends BagOStuff {
|
|||
// the separator character needed for each argument.
|
||||
$charsLeft = 255 - strlen( $keyspace ) - count( $args );
|
||||
|
||||
$that = $this;
|
||||
$args = array_map(
|
||||
function ( $arg ) use ( $that, &$charsLeft ) {
|
||||
function ( $arg ) use ( &$charsLeft ) {
|
||||
$arg = strtr( $arg, ' ', '_' );
|
||||
|
||||
// Make sure %, #, and non-ASCII chars are escaped
|
||||
$arg = preg_replace_callback(
|
||||
'/[^\x21-\x22\x24\x26-\x7e]+/',
|
||||
|
|
@ -142,10 +143,10 @@ class MemcachedBagOStuff extends BagOStuff {
|
|||
);
|
||||
|
||||
if ( $charsLeft < 0 ) {
|
||||
$args = array( '##' . md5( implode( ':', $args ) ) );
|
||||
return $keyspace . ':##' . md5( implode( ':', $args ) );
|
||||
}
|
||||
|
||||
return parent::makeKeyInternal( $keyspace, $args );
|
||||
return $keyspace . ':' . implode( ':', $args );
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
|
|
@ -50,6 +50,11 @@ class BagOStuffTest extends MediaWikiTestCase {
|
|||
$globalKey,
|
||||
'Local key and global key with same parameters should not be equal'
|
||||
);
|
||||
|
||||
$this->assertNotEquals(
|
||||
$cache->makeKeyInternal( 'prefix', array( 'a', 'bc:', 'de' ) ),
|
||||
$cache->makeKeyInternal( 'prefix', array( 'a', 'bc', ':de' ) )
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
|||
|
|
@ -32,7 +32,7 @@ class MemcachedBagOStuffTest extends MediaWikiTestCase {
|
|||
|
||||
$this->assertEquals(
|
||||
$this->cache->makeKey( 'but spaces', 'hashes#', "and\nnewlines", 'are_not' ),
|
||||
'test:but%20spaces:hashes%23:and%0Anewlines:are_not'
|
||||
'test:but_spaces:hashes%23:and%0Anewlines:are_not'
|
||||
);
|
||||
|
||||
$this->assertEquals(
|
||||
|
|
@ -43,7 +43,7 @@ class MemcachedBagOStuffTest extends MediaWikiTestCase {
|
|||
|
||||
$this->assertEquals(
|
||||
$this->cache->makeKey( 'this', 'key', 'contains', '𝕥𝕠𝕠 𝕞𝕒𝕟𝕪 𝕞𝕦𝕝𝕥𝕚𝕓𝕪𝕥𝕖 𝕔𝕙𝕒𝕣𝕒𝕔𝕥𝕖𝕣𝕤' ),
|
||||
'test:this:key:contains:#60190c8f5a63ba5438b124f5c10b91d0'
|
||||
'test:this:key:contains:#c118f92685a635cb843039de50014c9c'
|
||||
);
|
||||
|
||||
$this->assertEquals(
|
||||
|
|
|
|||
Loading…
Reference in a new issue