// Copyright (c) 2010 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/common/plugin_group.h"

#include <string>
#include <vector>

#include "base/scoped_ptr.h"
#include "base/string_util.h"
#include "base/utf_string_conversions.h"
#include "base/values.h"
#include "base/version.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/glue/plugins/webplugininfo.h"

static const PluginGroupDefinition kPluginDef = {
    "MyPlugin", "MyPlugin", "", "", "3.0.44", "http://latest/" };
static const PluginGroupDefinition kPluginDef3 = {
    "MyPlugin 3", "MyPlugin", "0", "4", "3.0.44", "http://latest" };
static const PluginGroupDefinition kPluginDef4 = {
    "MyPlugin 4", "MyPlugin", "4", "5", "4.0.44", "http://latest" };
static const PluginGroupDefinition kPluginDefNotVulnerable = {
    "MyPlugin", "MyPlugin", "", "", "", "http://latest" };

// name, path, version, desc, mime_types, enabled.
static WebPluginInfo kPlugin2043 = {
    ASCIIToUTF16("MyPlugin"), FilePath(), ASCIIToUTF16("2.0.43"),
    ASCIIToUTF16("MyPlugin version 2.0.43"),
    std::vector<WebPluginMimeType>(), true };
static WebPluginInfo kPlugin3043 = {
    ASCIIToUTF16("MyPlugin"), FilePath(), ASCIIToUTF16("3.0.43"),
    ASCIIToUTF16("MyPlugin version 3.0.43"),
    std::vector<WebPluginMimeType>(), true };
static WebPluginInfo kPlugin3044 = {
    ASCIIToUTF16("MyPlugin"), FilePath(), ASCIIToUTF16("3.0.44"),
    ASCIIToUTF16("MyPlugin version 3.0.44"),
    std::vector<WebPluginMimeType>(), true };
static WebPluginInfo kPlugin3045 = {
    ASCIIToUTF16("MyPlugin"), FilePath(), ASCIIToUTF16("3.0.45"),
    ASCIIToUTF16("MyPlugin version 3.0.45"),
    std::vector<WebPluginMimeType>(), true };
static WebPluginInfo kPlugin4043 = {
    ASCIIToUTF16("MyPlugin"), FilePath(), ASCIIToUTF16("4.0.43"),
    ASCIIToUTF16("MyPlugin version 4.0.43"),
    std::vector<WebPluginMimeType>(), true };

class PluginGroupTest : public testing::Test {
};

TEST(PluginGroupTest, PluginGroupMatch) {
  scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition(
      kPluginDef3));
  EXPECT_TRUE(group->Match(kPlugin3045));
  group->AddPlugin(kPlugin3045, 0);
  EXPECT_FALSE(group->IsVulnerable());
}

TEST(PluginGroupTest, PluginGroupMatchCorrectVersion) {
  scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition(
      kPluginDef3));
  EXPECT_TRUE(group->Match(kPlugin2043));
  EXPECT_TRUE(group->Match(kPlugin3043));
  EXPECT_FALSE(group->Match(kPlugin4043));

  group.reset(PluginGroup::FromPluginGroupDefinition(kPluginDef4));
  EXPECT_FALSE(group->Match(kPlugin2043));
  EXPECT_FALSE(group->Match(kPlugin3043));
  EXPECT_TRUE(group->Match(kPlugin4043));
}

TEST(PluginGroupTest, PluginGroupDescription) {
  string16 desc3043(ASCIIToUTF16("MyPlugin version 3.0.43"));
  string16 desc3045(ASCIIToUTF16("MyPlugin version 3.0.45"));
  WebPluginInfo plugin3043(kPlugin3043);
  WebPluginInfo plugin3045(kPlugin3045);

  {
    scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition(
        kPluginDef3));
    EXPECT_TRUE(group->Match(plugin3043));
    group->AddPlugin(plugin3043, 0);
    EXPECT_EQ(desc3043, group->description());
    EXPECT_TRUE(group->IsVulnerable());
    EXPECT_TRUE(group->Match(plugin3045));
    group->AddPlugin(plugin3045, 1);
    EXPECT_EQ(desc3043, group->description());
    EXPECT_TRUE(group->IsVulnerable());
  }

  {
    // Disable the first plugin.
    plugin3043.enabled = false;
    scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition(
        kPluginDef3));
    EXPECT_TRUE(group->Match(plugin3043));
    group->AddPlugin(plugin3043, 0);
    EXPECT_EQ(desc3043, group->description());
    EXPECT_TRUE(group->IsVulnerable());
    EXPECT_TRUE(group->Match(plugin3045));
    group->AddPlugin(plugin3045, 1);
    EXPECT_EQ(desc3045, group->description());
    EXPECT_FALSE(group->IsVulnerable());
  }

  {
    // Disable the second plugin.
    plugin3045.enabled = false;
    scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition(
        kPluginDef3));
    EXPECT_TRUE(group->Match(plugin3043));
    group->AddPlugin(plugin3043, 1);
    EXPECT_EQ(desc3043, group->description());
    EXPECT_TRUE(group->IsVulnerable());
    EXPECT_TRUE(group->Match(plugin3045));
    group->AddPlugin(plugin3045, 0);
    EXPECT_EQ(desc3043, group->description());
    EXPECT_TRUE(group->IsVulnerable());
  }
}

TEST(PluginGroupTest, PluginGroupDefinition) {
  const PluginGroupDefinition* definitions =
      PluginGroup::GetPluginGroupDefinitions();
  for (size_t i = 0; i < PluginGroup::GetPluginGroupDefinitionsSize(); ++i) {
    scoped_ptr<PluginGroup> def_group(
        PluginGroup::FromPluginGroupDefinition(definitions[i]));
    ASSERT_TRUE(def_group.get() != NULL);
    EXPECT_FALSE(def_group->Match(kPlugin2043));
  }
}

TEST(PluginGroupTest, VersionExtraction) {
  // Some real-world plugin versions (spaces, commata, parentheses, 'r', oh my)
  const char* versions[][2] = {
    { "7.6.6 (1671)", "7.6.6.1671" },  // Quicktime
    { "2, 0, 0, 254", "2.0.0.254" },   // DivX
    { "3, 0, 0, 0", "3.0.0.0" },       // Picasa
    { "1, 0, 0, 1", "1.0.0.1" },       // Earth
    { "10,0,45,2", "10.0.45.2" },      // Flash
    { "11.5.7r609", "11.5.7.609"}     // Shockwave
  };

  for (size_t i = 0; i < arraysize(versions); i++) {
    const WebPluginInfo plugin = {
        ASCIIToUTF16("Blah Plugin"), FilePath(), ASCIIToUTF16(versions[i][0]),
        string16(),std::vector<WebPluginMimeType>(), true };
    scoped_ptr<PluginGroup> group(PluginGroup::FromWebPluginInfo(plugin));
    EXPECT_TRUE(group->Match(plugin));
    group->AddPlugin(plugin, 0);
    scoped_ptr<DictionaryValue> data(group->GetDataForUI());
    std::string version;
    data->GetString("version", &version);
    EXPECT_EQ(versions[i][1], version);
  }
}