Implement mimetype sniffing for extensions.

chrome/common/extensions/extension_unittest.cc

 // Copyright (c) 2009 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 "base/file_path.h"
+#include "base/file_util.h"
 #include "base/string_util.h"
 #include "base/path_service.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/extensions/extension_error_reporter.h"
 #include "chrome/common/json_value_serializer.h"
+#include "net/base/mime_sniffer.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace keys = extension_manifest_keys;
 namespace values = extension_manifest_values;
 namespace errors = extension_manifest_errors;
 
 class ExtensionTest : public testing::Test {
 };
 
 TEST(ExtensionTest, InitFromValueInvalid) {
@@ skipping 399 matching lines @@
     Extension extension;
     std::string error;
     input_value.SetString(keys::kVersion, "1.0");
     input_value.SetString(keys::kName, "Test");
     input_value.SetString(keys::kUpdateURL, invalid[i]);
 
     EXPECT_FALSE(extension.InitFromValue(input_value, false, &error));
     EXPECT_TRUE(MatchPattern(error, errors::kInvalidUpdateURL));
   }
 }
+
+// This test ensures that the mimetype sniffing code stays in sync with the
+// actual crx files that we test other parts of the system with.
+TEST(ExtensionTest, MimeTypeSniffing) {
+  FilePath path;
+  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &path));
+  path = path.AppendASCII("extensions").AppendASCII("good.crx");
+
+  std::string data;
+  ASSERT_TRUE(file_util::ReadFileToString(path, &data));
+
+  std::string result;
+  EXPECT_TRUE(net::SniffMimeType(data.c_str(), data.size(),
+              GURL("http://www.example.com/foo.crx"), "", &result));
+  EXPECT_EQ(std::string(Extension::kMimeType), result);
+
+  data.clear();
+  result.clear();
+  path = path.DirName().AppendASCII("bad_magic.crx");
+  ASSERT_TRUE(file_util::ReadFileToString(path, &data));
+  EXPECT_TRUE(net::SniffMimeType(data.c_str(), data.size(),
+              GURL("http://www.example.com/foo.crx"), "", &result));
+  EXPECT_EQ("application/octet-stream", result);
+}