// Improve multishop code

controllers/admin/AdminCategoriesController.php

@@ -106,7 +106,7 @@ class AdminCategoriesControllerCore extends AdminController
 		if (Shop::isFeatureActive() && Shop::getContext() != Shop::CONTEXT_SHOP)
 			unset($this->fields_list['position']);
 		// shop restriction : if category is not available for current shop, we redirect to the list from default category
-		if (!Shop::isCategoryAvailable($this->_category->id))
+		if (!$this->_category->isAssociatedToShop())
 		{
 			$this->redirect_after = self::$currentIndex.'&id_category='.(int)$this->context->shop->getCategory().'&token='.$this->token;
 			$this->redirect();

controllers/front/CategoryController.php

@@ -49,7 +49,7 @@ class CategoryControllerCore extends FrontController
 
 	public function canonicalRedirection($canonicalURL = '')
 	{
-		if (!Shop::isCategoryAvailable(Tools::getValue('id_category')))
+		if (!Validate::isLoadedObject($this->category) || !$this->category->inShop() || !$this->category->isAssociatedToShop())
 		{
 			$this->redirect_after = '404';
 			$this->redirect();
@@ -64,8 +64,6 @@ class CategoryControllerCore extends FrontController
 	 */
 	public function init()
 	{
-		parent::init();
-
 		// Get category ID
 		$id_category = (int)Tools::getValue('id_category');
 		if (!$id_category || !Validate::isUnsignedId($id_category))
@@ -73,14 +71,8 @@ class CategoryControllerCore extends FrontController
 
 		// Instantiate category
 		$this->category = new Category($id_category, $this->context->language->id);
-		if (!Validate::isLoadedObject($this->category) || !$this->category->inShop())
-		{
-			$this->errors[] = Tools::displayError('Category does not exist');
-			header('HTTP/1.1 404 Not Found');
-			header('Status: 404 Not Found');
-		}
-		else
-			$this->canonicalRedirection();
+
+		parent::init();
 
 		if (!$this->category->checkAccess($this->context->customer->id))
 			$this->errors[] = Tools::displayError('You do not have access to this category.');