Apply content-security-policy directive to chrome://workers page.

chrome/browser/ui/webui/workers_ui.cc

 // Copyright (c) 2011 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/ui/webui/workers_ui.h"
 
 #include "base/json/json_writer.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
@@ skipping 16 matching lines @@
 static const char kOpenDevToolsCommand[]  = "openDevTools";
 
 static const char kWorkerProcessHostIdField[]  = "workerProcessHostId";
 static const char kWorkerRouteIdField[]  = "workerRouteId";
 static const char kUrlField[]  = "url";
 static const char kNameField[]  = "name";
 static const char kPidField[]  = "pid";
 
 namespace {
 
-class WorkersUIHTMLSource : public ChromeURLDataManager::DataSource {
+class WorkersUIHTMLSource : public ChromeWebUIDataSource {
  public:
   WorkersUIHTMLSource();
 
   virtual void StartDataRequest(const std::string& path,
                                 bool is_incognito,
                                 int request_id);
-
-  virtual std::string GetMimeType(const std::string&) const;
-
  private:
   ~WorkersUIHTMLSource() {}
-
   void SendSharedWorkersData(int request_id);
-
   DISALLOW_COPY_AND_ASSIGN(WorkersUIHTMLSource);
 };
 
 WorkersUIHTMLSource::WorkersUIHTMLSource()
-    : DataSource(chrome::kChromeUIWorkersHost, NULL) {
+    : ChromeWebUIDataSource(chrome::kChromeUIWorkersHost, NULL) {
+  add_resource_path("workers.js", IDR_WORKERS_INDEX_JS);
+  set_default_resource(IDR_WORKERS_INDEX_HTML);
 }
 
 void WorkersUIHTMLSource::StartDataRequest(const std::string& path,
                                            bool is_incognito,
                                            int request_id) {
   if (path == kWorkersDataFile) {
     SendSharedWorkersData(request_id);
   } else {
-    int idr = IDR_WORKERS_INDEX_HTML;
-    scoped_refptr<RefCountedStaticMemory> response(
-        ResourceBundle::GetSharedInstance().LoadDataResourceBytes(idr));
-    SendResponse(request_id, response);
+    ChromeWebUIDataSource::StartDataRequest(path, is_incognito, request_id);
   }
 }
 
-std::string WorkersUIHTMLSource::GetMimeType(const std::string& path) const {
-  if (EndsWith(path, ".css", false))
-    return "text/css";
-  if (EndsWith(path, ".js", false))
-    return "application/javascript";
-  if (EndsWith(path, ".json", false))
-    return "plain/text";
-  return "text/html";
-}
-
 void WorkersUIHTMLSource::SendSharedWorkersData(int request_id) {
     ListValue workers_list;
     BrowserChildProcessHost::Iterator iter(ChildProcessInfo::WORKER_PROCESS);
     for (; !iter.Done(); ++iter) {
       WorkerProcessHost* worker = static_cast<WorkerProcessHost*>(*iter);
       const WorkerProcessHost::Instances& instances = worker->instances();
       for (WorkerProcessHost::Instances::const_iterator i = instances.begin();
            i != instances.end(); ++i) {
         if (!i->shared())
           continue;
@@ skipping 66 matching lines @@
   WorkersDOMHandler* handler = new WorkersDOMHandler();
   AddMessageHandler(handler);
   handler->Attach(this);
 
   WorkersUIHTMLSource* html_source = new WorkersUIHTMLSource();
 
   // Set up the chrome://workers/ source.
   Profile* profile = Profile::FromBrowserContext(contents->browser_context());
   profile->GetChromeURLDataManager()->AddDataSource(html_source);
 }