OOPIFs: Transitioning MHTML generation from view-oriented to frame-oriented.

content/renderer/web_frame_utils.cc

 // 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.
 
 #include "content/renderer/web_frame_utils.h"
 
 #include "content/renderer/render_frame_impl.h"
 #include "content/renderer/render_frame_proxy.h"
 #include "ipc/ipc_message.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
+#include "third_party/WebKit/public/web/WebLocalFrame.h"
+#include "third_party/WebKit/public/web/WebRemoteFrame.h"
 
 namespace content {
 
 int GetRoutingIdForFrameOrProxy(blink::WebFrame* web_frame) {
   if (!web_frame)
     return MSG_ROUTING_NONE;
   if (web_frame->isWebRemoteFrame())
     return RenderFrameProxy::FromWebFrame(web_frame)->routing_id();
   return RenderFrameImpl::FromWebFrame(web_frame)->GetRoutingID();
 }
 
+blink::WebFrame* GetWebFrameFromRoutingIdForFrameOrProxy(int routing_id) {
+  auto* render_frame = RenderFrameImpl::FromRoutingID(routing_id);
+  if (render_frame)
+    return render_frame->GetWebFrame();
+
+  auto* render_frame_proxy = RenderFrameProxy::FromRoutingID(routing_id);
+  if (render_frame_proxy)
+    return render_frame_proxy->web_frame();
+
+  return nullptr;
+}
+
 }  // namespace content