Check page URL origin to determine JavaScript alert titles.

ios/chrome/browser/ui/dialogs/dialog_presenter.mm

 // Copyright 2015 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.
 
 #import "ios/chrome/browser/ui/dialogs/dialog_presenter.h"
 
 #include <deque>
 #include <map>
 
 #import "base/ios/block_types.h"
@@ skipping 73 matching lines @@
 // ShouldShowDialogBlockingOption() for |webState|, optionally adds a button to
 // |alertCoordinator| allowing for the blocking of future dialogs.  In addition
 // to blocking dialogs for the WebState, the added button will call
 // |alertCoordinator|'s |cancelAction|.
 - (void)setUpBlockingOptionForCoordinator:(AlertCoordinator*)alertCoordinator
                                  webState:(web::WebState*)webState;
 
 // The block to use for the JavaScript dialog blocking option for |coordinator|.
 - (ProceduralBlock)blockingActionForCoordinator:(AlertCoordinator*)coordinator;
 
+// Creates a title for the alert based on the URL (|pageURL|), and its
+// relationship to the |mainFrameURL| (typically these are identical except for
+// when posting alerts from an embedded iframe).
++ (NSString*)localizedTitleForJavaScriptAlertFromPage:(const GURL&)pageURL
+                                         mainFrameURL:(const GURL&)mainFrameURL;
+
 @end
 
 @implementation DialogPresenter
 
 @synthesize active = _active;
 @synthesize delegate = _delegate;
 @synthesize viewController = _viewController;
 @synthesize presentedDialogCoordinator = _presentedDialogCoordinator;
 @synthesize blockingConfirmationCoordinator = _blockingConfirmationCoordinator;
 @synthesize presentedDialogWebState = _presentedDialogWebState;
@@ skipping 23 matching lines @@
             self.presentedDialogCoordinator != nil);
   return self.presentedDialogCoordinator != nil;
 }
 
 #pragma mark - Public
 
 - (void)runJavaScriptAlertPanelWithMessage:(NSString*)message
                                 requestURL:(const GURL&)requestURL
                                   webState:(web::WebState*)webState
                          completionHandler:(void (^)(void))completionHandler {
-  NSString* title =
-      [DialogPresenter localizedTitleForJavaScriptAlertFromPage:requestURL];
+  NSString* title = [DialogPresenter
+      localizedTitleForJavaScriptAlertFromPage:requestURL
+                                  mainFrameURL:webState->GetLastCommittedURL()];
   AlertCoordinator* alertCoordinator =
       [[AlertCoordinator alloc] initWithBaseViewController:self.viewController
                                                      title:title
                                                    message:message];
 
   // Handler.
   __weak DialogPresenter* weakSelf = self;
   __weak AlertCoordinator* weakCoordinator = alertCoordinator;
   ProceduralBlock OKHandler = ^{
     if (completionHandler)
@@ skipping 14 matching lines @@
   [self setUpBlockingOptionForCoordinator:alertCoordinator webState:webState];
 
   [self addDialogCoordinator:alertCoordinator forWebState:webState];
 }
 
 - (void)runJavaScriptConfirmPanelWithMessage:(NSString*)message
                                   requestURL:(const GURL&)requestURL
                                     webState:(web::WebState*)webState
                            completionHandler:
                                (void (^)(BOOL isConfirmed))completionHandler {
-  NSString* title =
-      [DialogPresenter localizedTitleForJavaScriptAlertFromPage:requestURL];
+  NSString* title = [DialogPresenter
+      localizedTitleForJavaScriptAlertFromPage:requestURL
+                                  mainFrameURL:webState->GetLastCommittedURL()];
   AlertCoordinator* alertCoordinator =
       [[AlertCoordinator alloc] initWithBaseViewController:self.viewController
                                                      title:title
                                                    message:message];
 
   // Actions.
   ProceduralBlock confirmAction = ^{
     if (completionHandler)
       completionHandler(YES);
   };
@@ skipping 15 matching lines @@
 
   [self addDialogCoordinator:alertCoordinator forWebState:webState];
 }
 
 - (void)runJavaScriptTextInputPanelWithPrompt:(NSString*)message
                                   defaultText:(NSString*)defaultText
                                    requestURL:(const GURL&)requestURL
                                      webState:(web::WebState*)webState
                             completionHandler:
                                 (void (^)(NSString* input))completionHandler {
-  NSString* title =
-      [DialogPresenter localizedTitleForJavaScriptAlertFromPage:requestURL];
+  NSString* title = [DialogPresenter
+      localizedTitleForJavaScriptAlertFromPage:requestURL
+                                  mainFrameURL:webState->GetLastCommittedURL()];
   InputAlertCoordinator* alertCoordinator = [[InputAlertCoordinator alloc]
       initWithBaseViewController:self.viewController
                            title:title
                          message:message];
 
   // Actions.
   __weak InputAlertCoordinator* weakCoordinator = alertCoordinator;
   ProceduralBlock confirmAction = ^{
     if (completionHandler) {
       NSString* textInput = [weakCoordinator textFields].firstObject.text;
@@ skipping 120 matching lines @@
   // Don't try to present if a JavaScript dialog blocking confirmation sheet is
   // displayed.
   if (self.blockingConfirmationCoordinator)
     return;
   // The active TabModel can't be changed while a JavaScript dialog is shown.
   DCHECK(!self.showingDialog);
   if (_active && !_queuedWebStates.empty() && !self.delegate.presenting)
     [self showNextDialog];
 }
 
-+ (NSString*)localizedTitleForJavaScriptAlertFromPage:(const GURL&)pageURL {
++ (NSString*)localizedTitleForJavaScriptAlertFromPage:(const GURL&)pageURL
+                                         mainFrameURL:
+                                             (const GURL&)mainFrameURL {
   NSString* localizedTitle = nil;
   NSString* hostname = base::SysUTF8ToNSString(pageURL.host());
-  if (!hostname.length) {
+
+  bool sameOriginAsMainFrame = pageURL.GetOrigin() == mainFrameURL.GetOrigin();
+
+  if (!sameOriginAsMainFrame) {
     localizedTitle = l10n_util::GetNSString(
         IDS_JAVASCRIPT_MESSAGEBOX_TITLE_NONSTANDARD_URL_IFRAME);
   } else {
     localizedTitle = l10n_util::GetNSStringF(
         IDS_JAVASCRIPT_MESSAGEBOX_TITLE, base::SysNSStringToUTF16(hostname));
   }
   return localizedTitle;
 }
 
 #pragma mark - Private methods.
@@ skipping 141 matching lines @@
                                         style:UIAlertActionStyleDestructive];
     [confirmationCoordinator addItemWithTitle:l10n_util::GetNSString(IDS_CANCEL)
                                        action:cancelHandler
                                         style:UIAlertActionStyleCancel];
     [weakSelf setBlockingConfirmationCoordinator:confirmationCoordinator];
     [[weakSelf blockingConfirmationCoordinator] start];
   } copy];
 }
 
 @end