Home | History | Annotate | Download | only in infobars 
      1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/infobars/infobar_service.h"
      6 
      7 #include "chrome/browser/chrome_notification_types.h"
      8 #include "chrome/browser/infobars/infobar.h"
      9 #include "chrome/browser/infobars/infobar_delegate.h"
     10 #include "chrome/browser/infobars/insecure_content_infobar_delegate.h"
     11 #include "chrome/common/render_messages.h"
     12 #include "content/public/browser/navigation_controller.h"
     13 #include "content/public/browser/notification_service.h"
     14 #include "content/public/browser/web_contents.h"
     15 
     16 
     17 DEFINE_WEB_CONTENTS_USER_DATA_KEY(InfoBarService);
     18 
     19 InfoBar* InfoBarService::AddInfoBar(scoped_ptr<InfoBar> infobar) {
     20   DCHECK(infobar);
     21   if (!infobars_enabled_)
     22     return NULL;
     23 
     24   for (InfoBars::const_iterator i(infobars_.begin()); i != infobars_.end();
     25        ++i) {
     26     if ((*i)->delegate()->EqualsDelegate(infobar->delegate())) {
     27       DCHECK_NE((*i)->delegate(), infobar->delegate());
     28       return NULL;
     29     }
     30   }
     31 
     32   InfoBar* infobar_ptr = infobar.release();
     33   infobars_.push_back(infobar_ptr);
     34   infobar_ptr->SetOwner(this);
     35 
     36   content::NotificationService::current()->Notify(
     37       chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_ADDED,
     38       content::Source<InfoBarService>(this),
     39       content::Details<InfoBar::AddedDetails>(infobar_ptr));
     40   return infobar_ptr;
     41 }
     42 
     43 void InfoBarService::RemoveInfoBar(InfoBar* infobar) {
     44   RemoveInfoBarInternal(infobar, true);
     45 }
     46 
     47 InfoBar* InfoBarService::ReplaceInfoBar(InfoBar* old_infobar,
     48                                         scoped_ptr<InfoBar> new_infobar) {
     49   DCHECK(old_infobar);
     50   if (!infobars_enabled_)
     51     return AddInfoBar(new_infobar.Pass());  // Deletes the infobar.
     52   DCHECK(new_infobar);
     53 
     54   InfoBars::iterator i(std::find(infobars_.begin(), infobars_.end(),
     55                                  old_infobar));
     56   DCHECK(i != infobars_.end());
     57 
     58   InfoBar* new_infobar_ptr = new_infobar.release();
     59   i = infobars_.insert(i, new_infobar_ptr);
     60   new_infobar_ptr->SetOwner(this);
     61   InfoBar::ReplacedDetails replaced_details(old_infobar, new_infobar_ptr);
     62 
     63   // Remove the old infobar before notifying, so that if any observers call back
     64   // to AddInfoBar() or similar, we don't dupe-check against this infobar.
     65   infobars_.erase(++i);
     66 
     67   content::NotificationService::current()->Notify(
     68       chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_REPLACED,
     69       content::Source<InfoBarService>(this),
     70       content::Details<InfoBar::ReplacedDetails>(&replaced_details));
     71 
     72   old_infobar->CloseSoon();
     73   return new_infobar_ptr;
     74 }
     75 
     76 InfoBarService::InfoBarService(content::WebContents* web_contents)
     77     : content::WebContentsObserver(web_contents),
     78       infobars_enabled_(true) {
     79   DCHECK(web_contents);
     80 }
     81 
     82 InfoBarService::~InfoBarService() {
     83   // Destroy all remaining InfoBars.  It's important to not animate here so that
     84   // we guarantee that we'll delete all delegates before we do anything else.
     85   RemoveAllInfoBars(false);
     86 }
     87 
     88 void InfoBarService::RenderProcessGone(base::TerminationStatus status) {
     89   RemoveAllInfoBars(true);
     90 }
     91 
     92 void InfoBarService::NavigationEntryCommitted(
     93     const content::LoadCommittedDetails& load_details) {
     94   // NOTE: It is not safe to change the following code to count upwards or
     95   // use iterators, as the RemoveInfoBar() call synchronously modifies our
     96   // delegate list.
     97   for (size_t i = infobars_.size(); i > 0; --i) {
     98     InfoBar* infobar = infobars_[i - 1];
     99     if (infobar->delegate()->ShouldExpire(load_details))
    100       RemoveInfoBar(infobar);
    101   }
    102 }
    103 
    104 void InfoBarService::WebContentsDestroyed(content::WebContents* web_contents) {
    105   // The WebContents is going away; be aggressively paranoid and delete
    106   // ourselves lest other parts of the system attempt to add infobars or use
    107   // us otherwise during the destruction.
    108   web_contents->RemoveUserData(UserDataKey());
    109   // That was the equivalent of "delete this". This object is now destroyed;
    110   // returning from this function is the only safe thing to do.
    111 }
    112 
    113 bool InfoBarService::OnMessageReceived(const IPC::Message& message) {
    114   bool handled = true;
    115   IPC_BEGIN_MESSAGE_MAP(InfoBarService, message)
    116     IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DidBlockDisplayingInsecureContent,
    117                         OnDidBlockDisplayingInsecureContent)
    118     IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DidBlockRunningInsecureContent,
    119                         OnDidBlockRunningInsecureContent)
    120     IPC_MESSAGE_UNHANDLED(handled = false)
    121   IPC_END_MESSAGE_MAP()
    122   return handled;
    123 }
    124 
    125 void InfoBarService::RemoveInfoBarInternal(InfoBar* infobar, bool animate) {
    126   DCHECK(infobar);
    127   if (!infobars_enabled_) {
    128     DCHECK(infobars_.empty());
    129     return;
    130   }
    131 
    132   InfoBars::iterator i(std::find(infobars_.begin(), infobars_.end(), infobar));
    133   DCHECK(i != infobars_.end());
    134 
    135   // Remove the infobar before notifying, so that if any observers call back to
    136   // AddInfoBar() or similar, we don't dupe-check against this infobar.
    137   infobars_.erase(i);
    138 
    139   // This notification must happen before the call to CloseSoon() below, since
    140   // observers may want to access |infobar| and that call can delete it.
    141   InfoBar::RemovedDetails removed_details(infobar, animate);
    142   content::NotificationService::current()->Notify(
    143       chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_REMOVED,
    144       content::Source<InfoBarService>(this),
    145       content::Details<InfoBar::RemovedDetails>(&removed_details));
    146 
    147   infobar->CloseSoon();
    148 }
    149 
    150 void InfoBarService::RemoveAllInfoBars(bool animate) {
    151   while (!infobars_.empty())
    152     RemoveInfoBarInternal(infobars_.back(), animate);
    153 }
    154 
    155 void InfoBarService::OnDidBlockDisplayingInsecureContent() {
    156   InsecureContentInfoBarDelegate::Create(
    157       this, InsecureContentInfoBarDelegate::DISPLAY);
    158 }
    159 
    160 void InfoBarService::OnDidBlockRunningInsecureContent() {
    161   InsecureContentInfoBarDelegate::Create(this,
    162                                          InsecureContentInfoBarDelegate::RUN);
    163 }
    164