Fixing mime type so that MHTML files can be opened directly from Google Drive.

chrome/browser/chromeos/gdata/gdata_protocol_handler.cc

 // Copyright (c) 2012 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/chromeos/gdata/gdata_protocol_handler.h"
 
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
@@ skipping 41 matching lines @@
 const int kHTTPInternalError = 500;
 
 const char kHTTPOkText[] = "OK";
 const char kHTTPNotAllowedText[] = "Not Allowed";
 const char kHTTPNotFoundText[] = "Not Found";
 const char kHTTPInternalErrorText[] = "Internal Error";
 
 // Initial size of download buffer, same as kBufferSize used for URLFetcherCore.
 const int kInitialDownloadBufferSizeInBytes = 4096;
 
+struct MimeTypeReplacement {
+  const char* original_type;
+  const char* new_type;
+};
+
+const MimeTypeReplacement kMimeTypeReplacements[] = {
+    {"message/rfc822","multipart/related"} // Fixes MHTML
+};
+
+std::string FixupMimeType(const std::string& type) {
+  for (size_t i = 0; i < arraysize(kMimeTypeReplacements); i++) {
+    if (type == kMimeTypeReplacements[i].original_type)
+      return kMimeTypeReplacements[i].new_type;
+  }
+  return type;
+}
+
 // Empty callback for net::FileStream::Close().
 void EmptyCompletionCallback(int) {
 }
 
 // Helper function that reads file size.
 void GetFileSizeOnBlockingPool(const FilePath& file_path,
                                int64* file_size) {
   if (!file_util::GetFileSize(file_path, file_size))
     *file_size = 0;
 }
@@ skipping 246 matching lines @@
         base::Bind(&CancelGDataDownloadOnUIThread,
                    gdata_file_path_));
   }
 
   net::URLRequestJob::Kill();
   weak_ptr_factory_->InvalidateWeakPtrs();
 }
 
 bool GDataURLRequestJob::GetMimeType(std::string* mime_type) const {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  mime_type->assign(mime_type_);
+  mime_type->assign(FixupMimeType(mime_type_));
   return !mime_type->empty();
 }
 
 void GDataURLRequestJob::GetResponseInfo(net::HttpResponseInfo* info) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   if (response_info_.get())
     *info = *response_info_;
 }
 
 int GDataURLRequestJob::GetResponseCode() const {
@@ skipping 564 matching lines @@
 GDataProtocolHandler::~GDataProtocolHandler() {
 }
 
 net::URLRequestJob* GDataProtocolHandler::MaybeCreateJob(
     net::URLRequest* request) const {
   DVLOG(1) << "Handling url: " << request->url().spec();
   return new GDataURLRequestJob(request);
 }
 
 }  // namespace gdata