// Copyright (c) 2006-2008 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/basictypes.h" #include "googleurl/src/gurl.h" #include "net/base/mime_sniffer.h" #include "testing/gtest/include/gtest/gtest.h" namespace { class MimeSnifferTest : public testing::Test { }; } struct SnifferTest { const char* content; size_t content_len; std::string url; std::string type_hint; const char* mime_type; }; static void TestArray(SnifferTest* tests, size_t count) { std::string mime_type; for (size_t i = 0; i < count; ++i) { net::SniffMimeType(tests[i].content, tests[i].content_len, GURL(tests[i].url), tests[i].type_hint, &mime_type); EXPECT_EQ(tests[i].mime_type, mime_type); } } // TODO(evanm): convert other tests to use SniffMimeType instead of TestArray, // so the error messages produced by test failures are more useful. static std::string SniffMimeType(const std::string& content, const std::string& url, const std::string& mime_type_hint) { std::string mime_type; net::SniffMimeType(content.data(), content.size(), GURL(url), mime_type_hint, &mime_type); return mime_type; } TEST(MimeSnifferTest, BoundaryConditionsTest) { std::string mime_type; std::string type_hint; char buf[] = { 'd', '\x1f', '\xFF' }; GURL url; net::SniffMimeType(buf, 0, url, type_hint, &mime_type); EXPECT_EQ("text/plain", mime_type); net::SniffMimeType(buf, 1, url, type_hint, &mime_type); EXPECT_EQ("text/plain", mime_type); net::SniffMimeType(buf, 2, url, type_hint, &mime_type); EXPECT_EQ("application/octet-stream", mime_type); } TEST(MimeSnifferTest, BasicSniffingTest) { SnifferTest tests[] = { { "", sizeof("")-1, "http://www.example.com/foo.gif", "application/octet-stream", "application/octet-stream" }, { "GIF89a\x1F\x83\x94", sizeof("GIF89a\xAF\x83\x94")-1, "http://www.example.com/foo", "text/plain", "image/gif" }, { "Gif87a\x1F\x83\x94", sizeof("Gif87a\xAF\x83\x94")-1, "http://www.example.com/foo?param=tt.gif", "", "application/octet-stream" }, { "%!PS-Adobe-3.0", sizeof("%!PS-Adobe-3.0")-1, "http://www.example.com/foo", "text/plain", "text/plain" }, { "\x89" "PNG\x0D\x0A\x1A\x0A", sizeof("\x89" "PNG\x0D\x0A\x1A\x0A")-1, "http://www.example.com/foo", "application/octet-stream", "image/png" }, { "\xFF\xD8\xFF\x23\x49\xAF", sizeof("\xFF\xD8\xFF\x23\x49\xAF")-1, "http://www.example.com/foo", "", "image/jpeg" }, }; TestArray(tests, arraysize(tests)); } TEST(MimeSnifferTest, MozillaCompatibleTest) { SnifferTest tests[] = { { " \n \n \n \n \n \n", sizeof("\n")-1, "http://www.example.com/foo", "", "text/xml" }, { "\n", sizeof("\n")-1, "http://www.example.com/foo", "application/octet-stream", "application/octet-stream" }, }; TestArray(tests, arraysize(tests)); } TEST(MimeSnifferTest, DontAllowPrivilegeEscalationTest) { SnifferTest tests[] = { { "GIF87a\n\n" "" "\n", sizeof("GIF87a\n\n" "" "\n")-1, "http://www.example.com/foo", "", "image/gif" }, { "GIF87a\n\n" "" "\n", sizeof("GIF87a\n\n" "" "\n")-1, "http://www.example.com/foo?q=ttt.html", "", "image/gif" }, { "GIF87a\n\n" "" "\n", sizeof("GIF87a\n\n" "" "\n")-1, "http://www.example.com/foo#ttt.html", "", "image/gif" }, { "a\n\n" "" "\n", sizeof("a\n\n" "" "\n")-1, "http://www.example.com/foo", "", "text/plain" }, { "a\n\n" "" "\n", sizeof("a\n\n" "" "\n")-1, "http://www.example.com/foo?q=ttt.html", "", "text/plain" }, { "a\n\n" "" "\n", sizeof("a\n\n" "" "\n")-1, "http://www.example.com/foo#ttt.html", "", "text/plain" }, { "a\n\n" "" "\n", sizeof("a\n\n" "" "\n")-1, "http://www.example.com/foo.html", "", "text/plain" }, }; TestArray(tests, arraysize(tests)); } TEST(MimeSnifferTest, UnicodeTest) { SnifferTest tests[] = { { "\xEF\xBB\xBF" "Hi there", sizeof("\xEF\xBB\xBF" "Hi there")-1, "http://www.example.com/foo", "", "text/plain" }, { "\xEF\xBB\xBF\xED\x7A\xAD\x7A\x0D\x79", sizeof("\xEF\xBB\xBF\xED\x7A\xAD\x7A\x0D\x79")-1, "http://www.example.com/foo", "", "text/plain" }, { "\xFE\xFF\xD0\xA5\xD0\xBE\xD0\xBB\xD1\x83\xD0\xB9", sizeof("\xFE\xFF\xD0\xA5\xD0\xBE\xD0\xBB\xD1\x83\xD0\xB9")-1, "http://www.example.com/foo", "", "text/plain" }, { "\xFE\xFF\x00\x41\x00\x20\xD8\x00\xDC\x00\xD8\x00\xDC\x01", sizeof("\xFE\xFF\x00\x41\x00\x20\xD8\x00\xDC\x00\xD8\x00\xDC\x01")-1, "http://www.example.com/foo", "", "text/plain" }, }; TestArray(tests, arraysize(tests)); } TEST(MimeSnifferTest, FlashTest) { SnifferTest tests[] = { { "CWSdd\x00\xB3", sizeof("CWSdd\x00\xB3")-1, "http://www.example.com/foo", "", "application/octet-stream" }, { "FLVjdkl*(#)0sdj\x00", sizeof("FLVjdkl*(#)0sdj\x00")-1, "http://www.example.com/foo?q=ttt.swf", "", "application/octet-stream" }, { "FWS3$9\r\b\x00", sizeof("FWS3$9\r\b\x00")-1, "http://www.example.com/foo#ttt.swf", "", "application/octet-stream" }, { "FLVjdkl*(#)0sdj", sizeof("FLVjdkl*(#)0sdj")-1, "http://www.example.com/foo.swf", "", "text/plain" }, { "FLVjdkl*(#)0s\x01dj", sizeof("FLVjdkl*(#)0s\x01dj")-1, "http://www.example.com/foo/bar.swf", "", "application/octet-stream" }, { "FWS3$9\r\b\x1A", sizeof("FWS3$9\r\b\x1A")-1, "http://www.example.com/foo.swf?clickTAG=http://www.adnetwork.com/bar", "", "application/octet-stream" }, { "FWS3$9\r\x1C\b", sizeof("FWS3$9\r\x1C\b")-1, "http://www.example.com/foo.swf?clickTAG=http://www.adnetwork.com/bar", "text/plain", "application/octet-stream" }, }; TestArray(tests, arraysize(tests)); } TEST(MimeSnifferTest, XMLTest) { // An easy feed to identify. EXPECT_EQ("application/atom+xml", SniffMimeType("\r\n" "" "" ""; // CNN's RSS EXPECT_EQ("application/rss+xml", SniffMimeType(kCNNRSS, "", "text/xml")); EXPECT_EQ("text/plain", SniffMimeType(kCNNRSS, "", "text/plain")); // Don't sniff random XML as something different. EXPECT_EQ("text/xml", SniffMimeType("", "", "text/xml")); EXPECT_EQ("application/xhtml+xml", SniffMimeType("", "", "application/xml")); // Following our behavior with HTML, don't call other mime types XHTML. EXPECT_EQ("text/plain", SniffMimeType("", "", "text/plain")); EXPECT_EQ("application/rss+xml", SniffMimeType("", "", "application/rss+xml")); // Don't sniff other HTML-looking bits as HTML. EXPECT_EQ("text/xml", SniffMimeType("", "", "text/xml")); EXPECT_EQ("text/xml", SniffMimeType("", "", "text/xml")); } // Test content which is >= 512 bytes, and includes no open angle bracket. // http://code.google.com/p/chromium/issues/detail?id=3521 TEST(MimeSnifferTest, XMLTestLargeNoAngledBracket) { // Make a large input, with 600 bytes of "x". std::string content; content.resize(600); std::fill(content.begin(), content.end(), 'x'); // content.size() >= kMaxBytesToSniff (512) so the sniff is unambiguous. std::string mime_type; EXPECT_TRUE(net::SniffMimeType(content.data(), content.size(), GURL(), "text/xml", &mime_type)); }