diff --git a/includes/menu.inc b/includes/menu.inc index f23eb0d..d301c6a 100644 --- a/includes/menu.inc +++ b/includes/menu.inc @@ -2633,8 +2633,8 @@ function menu_rebuild() { $transaction = db_transaction(); try { - list($menu, $masks) = menu_router_build(); - _menu_router_save($menu, $masks); + list($menu, $masks, $path_roots) = menu_router_build(); + _menu_router_save($menu, $masks, $path_roots); _menu_navigation_links_rebuild($menu); // Clear the menu, page and block caches. menu_cache_clear_all(); @@ -2669,10 +2669,10 @@ function menu_router_build() { } // Alter the menu as defined in modules, keys are like user/%user. drupal_alter('menu', $callbacks); - list($menu, $masks) = _menu_router_build($callbacks); + list($menu, $masks, $path_roots) = _menu_router_build($callbacks); _menu_router_cache($menu); - return array($menu, $masks); + return array($menu, $masks, $path_roots); } /** @@ -3454,6 +3454,7 @@ function _menu_router_build($callbacks) { // matching. Calculate fitness, and fill some default values. $menu = array(); $masks = array(); + $path_roots = array(); foreach ($callbacks as $path => $item) { $load_functions = array(); $to_arg_functions = array(); @@ -3461,6 +3462,7 @@ function _menu_router_build($callbacks) { $move = FALSE; $parts = explode('/', $path, MENU_MAX_PARTS); + $path_roots[$parts[0]] = $parts[0]; $number_parts = count($parts); // We store the highest index of parts here to save some work in the fit // calculation loop. @@ -3670,13 +3672,13 @@ function _menu_router_build($callbacks) { $masks = array_keys($masks); rsort($masks); - return array($menu, $masks); + return array($menu, $masks, $path_roots); } /** * Helper function to save data from menu_router_build() to the router table. */ -function _menu_router_save($menu, $masks) { +function _menu_router_save($menu, $masks, $path_roots) { // Delete the existing router since we have some data to replace it. db_truncate('menu_router')->execute(); @@ -3749,6 +3751,7 @@ function _menu_router_save($menu, $masks) { $insert->execute(); // Store the masks. variable_set('menu_masks', $masks); + variable_set('menu_path_roots', $path_roots); return $menu; } diff --git a/includes/path.inc b/includes/path.inc index 630b34c..8386594 100644 --- a/includes/path.inc +++ b/includes/path.inc @@ -55,21 +55,13 @@ function drupal_lookup_path($action, $path = '', $path_language = NULL) { $cache = array( 'map' => array(), 'no_source' => array(), - 'whitelist' => NULL, + 'whitelist' => new PathAliasWhitelist('path_alias_whitelist', 'cache'), 'system_paths' => array(), 'no_aliases' => array(), 'first_call' => TRUE, ); } - // Retrieve the path alias whitelist. - if (!isset($cache['whitelist'])) { - $cache['whitelist'] = variable_get('path_alias_whitelist', NULL); - if (!isset($cache['whitelist'])) { - $cache['whitelist'] = drupal_path_alias_whitelist_rebuild(); - } - } - // If no language is explicitly specified we default to the current URL // language. If we used a language different from the one conveyed by the // requested URL, we might end up being unable to check if there is a path @@ -80,7 +72,7 @@ function drupal_lookup_path($action, $path = '', $path_language = NULL) { $cache = array(); $cache['whitelist'] = drupal_path_alias_whitelist_rebuild(); } - elseif ($cache['whitelist'] && $path != '') { + elseif ($path != '') { if ($action == 'alias') { // During the first call to drupal_lookup_path() per language, load the // expected system paths for the page from cache. @@ -355,6 +347,76 @@ function current_path() { return $_GET['q']; } +/* + * Extends DrupalCacheArray to build the path alias whitelist over time. + */ +class PathAliasWhitelist extends DrupalCacheArray { + + public function __construct($cid, $bin) { + parent::__construct($cid, $bin); + + // On a cold start $this->storage will be empty and the whitelist will + // need to be rebuilt from scratch. Prepare an initial version of the + // whitelist by processing the menu router to create an array where the + // first part of each router item is the key, with the value initialized + // to NULL. During the request, each path requested that matches one of + // these keys will be looked up and the array value set to either TRUE or + // FALSE. This ensures that paths which do not exist in the router are not + // looked up, and that paths that do exist in the router are only looked + // up once. Since url() can be called during maintenance mode, avoid + // trying to build the menu router in this state. + if (empty($this->storage) && !defined('MAINTENANCE_MODE')) { + foreach (variable_get('menu_path_roots', array()) as $root) { + $this->storage[$root] = NULL; + $this->persist($root); + } + } + } + + public function offsetGet($offset) { + // url() may be called with paths that are not represented by menu router + // items such as paths that will be rewritten by hook_url_outbound_alter(). + // Therefore internally TRUE is used to indicate whitelisted paths. FALSE is + // used to indicate paths that have already been checked but are not + // whitelisted, and NULL indicates paths that have not been checked yet. + if (isset($this->storage[$offset])) { + if ($this->storage[$offset]) { + return TRUE; + } + } + elseif (array_key_exists($offset, $this->storage)) { + return $this->resolveCacheMiss($offset); + } + } + + function resolveCacheMiss($root) { + $query = db_select('url_alias', 'u'); + $query->addExpression(1); + $query->condition('u.source', db_like($root) . '%', 'LIKE') ->range(0, 1); + $exists = (bool) $query->execute()->fetchField(); + $this->storage[$root] = $exists; + $this->persist($root); + if ($exists) { + return TRUE; + } + } + + public function set($cid, $data, $bin, $lock = TRUE) { + $lock_name = $cid . ':' . $bin; + if (!$lock || lock_acquire($lock_name)) { + if ($cached = cache($bin)->get($cid)) { + // Use array merge instead of union so that filled in values in $data + // overwrite empty values in the current cache. + $data = array_merge($cached->data, $data); + } + cache($bin)->set($cid, $data); + if ($lock) { + lock_release($lock_name); + } + } + } +} + /** * Rebuild the path alias white list. * @@ -362,27 +424,19 @@ function current_path() { * An optional system path for which an alias is being inserted. * * @return - * An array containing a white list of path aliases. + * An instance of the PathAliasWhitelist class. */ function drupal_path_alias_whitelist_rebuild($source = NULL) { // When paths are inserted, only rebuild the whitelist if the system path // has a top level component which is not already in the whitelist. if (!empty($source)) { - $whitelist = variable_get('path_alias_whitelist', NULL); + $whitelist = new PathAliasWhitelist('path_alias_whitelist', 'cache'); if (isset($whitelist[strtok($source, '/')])) { return $whitelist; } } - // For each alias in the database, get the top level component of the system - // path it corresponds to. This is the portion of the path before the first - // '/', if present, otherwise the whole path itself. - $whitelist = array(); - $result = db_query("SELECT DISTINCT SUBSTRING_INDEX(source, '/', 1) AS path FROM {url_alias}"); - foreach ($result as $row) { - $whitelist[$row->path] = TRUE; - } - variable_set('path_alias_whitelist', $whitelist); - return $whitelist; + cache_clear_all('path_alias_whitelist', 'cache'); + return new PathAliasWhitelist('path_alias_whitelist', 'cache'); } /**