Add ExtensionFunctionDispatcher::Delegate::GetAssociatedTabContents

chrome/browser/extensions/extension_infobar_delegate.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/extensions/extension_infobar_delegate.h"
 
 #include "chrome/browser/browser.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/common/notification_details.h"
 #include "chrome/common/notification_source.h"
 #include "chrome/common/notification_type.h"
 
 ExtensionInfoBarDelegate::ExtensionInfoBarDelegate(Browser* browser,
                                                    TabContents* tab_contents,
                                                    Extension* extension,
                                                    const GURL& url)
     : InfoBarDelegate(tab_contents),
       observer_(NULL),
       extension_(extension),
       tab_contents_(tab_contents) {
   ExtensionProcessManager* manager =
       browser->profile()->GetExtensionProcessManager();
   extension_host_.reset(manager->CreateInfobar(url, browser));
+  extension_host_->set_associated_tab_contents(tab_contents);
 
   registrar_.Add(this, NotificationType::EXTENSION_HOST_VIEW_SHOULD_CLOSE,
                  Source<Profile>(browser->profile()));
   registrar_.Add(this, NotificationType::EXTENSION_UNLOADED,
                  Source<Profile>(browser->profile()));
 }
 
 ExtensionInfoBarDelegate::~ExtensionInfoBarDelegate() {
   if (observer_)
     observer_->OnDelegateDeleted();
@@ skipping 41 matching lines @@
       if (extension_ == extension)
         tab_contents_->RemoveInfoBar(this);
       break;
     }
     default: {
       NOTREACHED() << "Unknown message";
       break;
     }
   }
 }