content/browser/download/file_metadata_unittest_linux.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <errno.h>
   6 #include <sys/types.h>
   7 #include <sys/xattr.h>
   8
   9 #include <algorithm>
  10 #include <sstream>
  11 #include <string>
  12
  13 #include "base/files/file_path.h"
  14 #include "base/files/file_util.h"
  15 #include "base/files/scoped_temp_dir.h"
  16 #include "base/logging.h"
  17 #include "base/strings/string_split.h"
  18 #include "content/browser/download/file_metadata_linux.h"
  19 #include "testing/gtest/include/gtest/gtest.h"
  20 #include "url/gurl.h"
  21
  22 namespace content {
  23 namespace {
  24
  25 using std::istringstream;
  26 using std::string;
  27 using std::vector;
  28
  29 class FileMetadataLinuxTest : public testing::Test {
  30  public:
  31   FileMetadataLinuxTest()
  32       : source_url_("http://www.source.com"),
  33         referrer_url_("http://www.referrer.com"),
  34         is_xattr_supported_(false) {}
  35
  36   const base::FilePath& test_file() const {
  37     return test_file_;
  38   }
  39
  40   const GURL& source_url() const {
  41     return source_url_;
  42   }
  43
  44   const GURL& referrer_url() const {
  45     return referrer_url_;
  46   }
  47
  48   bool is_xattr_supported() const {
  49     return is_xattr_supported_;
  50   }
  51
  52  protected:
  53   void SetUp() override {
  54     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
  55     ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &test_file_));
  56     int result = setxattr(test_file_.value().c_str(),
  57                           "user.test", "test", 4, 0);
  58     is_xattr_supported_ = (!result) || (errno != ENOTSUP);
  59     if (!is_xattr_supported_) {
  60       VLOG(0) << "Test will be skipped because extended attributes are not "
  61                 << "supported on this OS/file system.";
  62     }
  63   }
  64
  65   void CheckExtendedAttributeValue(const string attr_name,
  66                                    const string expected_value) const {
  67     ssize_t len = getxattr(test_file().value().c_str(), attr_name.c_str(),
  68                            NULL, 0);
  69     if (len <= static_cast<ssize_t>(0)) {
  70       FAIL() << "Attribute '" << attr_name << "' does not exist";
  71     }
  72     char* buffer = new char[len];
  73     len = getxattr(test_file().value().c_str(), attr_name.c_str(), buffer, len);
  74     EXPECT_EQ(expected_value.size(), static_cast<size_t>(len));
  75     string real_value(buffer, len);
  76     delete[] buffer;
  77     EXPECT_EQ(expected_value, real_value);
  78   }
  79
  80   void GetExtendedAttributeNames(vector<string>* attr_names) const {
  81     ssize_t len = listxattr(test_file().value().c_str(), NULL, 0);
  82     if (len <= static_cast<ssize_t>(0)) return;
  83     char* buffer = new char[len];
  84     len = listxattr(test_file().value().c_str(), buffer, len);
  85     attr_names->clear();
  86     base::SplitString(string(buffer, len), '\0', attr_names);
  87     delete[] buffer;
  88   }
  89
  90   void VerifyAttributesAreSetCorrectly() const {
  91     vector<string> attr_names;
  92     GetExtendedAttributeNames(&attr_names);
  93
  94     // Check if the attributes are set on the file
  95     vector<string>::const_iterator pos = find(attr_names.begin(),
  96         attr_names.end(), kSourceURLAttrName);
  97     EXPECT_NE(pos, attr_names.end());
  98     pos = find(attr_names.begin(), attr_names.end(), kReferrerURLAttrName);
  99     EXPECT_NE(pos, attr_names.end());
 100
 101     // Check if the attribute values are set correctly
 102     CheckExtendedAttributeValue(kSourceURLAttrName, source_url().spec());
 103     CheckExtendedAttributeValue(kReferrerURLAttrName, referrer_url().spec());
 104   }
 105
 106  private:
 107   base::ScopedTempDir temp_dir_;
 108   base::FilePath test_file_;
 109   GURL source_url_;
 110   GURL referrer_url_;
 111   bool is_xattr_supported_;
 112 };
 113
 114 TEST_F(FileMetadataLinuxTest, CheckMetadataSetCorrectly) {
 115   if (!is_xattr_supported()) return;
 116   AddOriginMetadataToFile(test_file(), source_url(), referrer_url());
 117   VerifyAttributesAreSetCorrectly();
 118 }
 119
 120 TEST_F(FileMetadataLinuxTest, SetMetadataMultipleTimes) {
 121   if (!is_xattr_supported()) return;
 122   GURL dummy_url("http://www.dummy.com");
 123   AddOriginMetadataToFile(test_file(), dummy_url, dummy_url);
 124   AddOriginMetadataToFile(test_file(), source_url(), referrer_url());
 125   VerifyAttributesAreSetCorrectly();
 126 }
 127
 128 TEST_F(FileMetadataLinuxTest, InvalidSourceURLTest) {
 129   if (!is_xattr_supported()) return;
 130   GURL invalid_url;
 131   vector<string> attr_names;
 132   AddOriginMetadataToFile(test_file(), invalid_url, referrer_url());
 133   GetExtendedAttributeNames(&attr_names);
 134   EXPECT_EQ(attr_names.end(), find(attr_names.begin(), attr_names.end(),
 135       kSourceURLAttrName));
 136   CheckExtendedAttributeValue(kReferrerURLAttrName, referrer_url().spec());
 137 }
 138
 139 TEST_F(FileMetadataLinuxTest, InvalidReferrerURLTest) {
 140   if (!is_xattr_supported()) return;
 141   GURL invalid_url;
 142   vector<string> attr_names;
 143   AddOriginMetadataToFile(test_file(), source_url(), invalid_url);
 144   GetExtendedAttributeNames(&attr_names);
 145   EXPECT_EQ(attr_names.end(), find(attr_names.begin(), attr_names.end(),
 146       kReferrerURLAttrName));
 147   CheckExtendedAttributeValue(kSourceURLAttrName, source_url().spec());
 148 }
 149
 150 TEST_F(FileMetadataLinuxTest, InvalidURLsTest) {
 151   if (!is_xattr_supported()) return;
 152   GURL invalid_url;
 153   vector<string> attr_names;
 154   AddOriginMetadataToFile(test_file(), invalid_url, invalid_url);
 155   GetExtendedAttributeNames(&attr_names);
 156   EXPECT_EQ(attr_names.end(), find(attr_names.begin(), attr_names.end(),
 157       kSourceURLAttrName));
 158   EXPECT_EQ(attr_names.end(), find(attr_names.begin(), attr_names.end(),
 159       kReferrerURLAttrName));
 160 }
 161
 162 }  // namespace
 163 }  // namespace content