15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <wincrypt.h> 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h" 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/scoped_temp_dir.h" 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 11868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/self_cleaning_temp_dir.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns a string of 8 characters consisting of the letter 'R' followed by 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// seven random hex digits. 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)std::wstring GetRandomFilename() { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint8 data[4]; 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HCRYPTPROV crypt_ctx = NULL; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get four bytes of randomness. Use CAPI rather than the CRT since I've 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // seen the latter trivially repeat. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_NE(FALSE, CryptAcquireContext(&crypt_ctx, NULL, NULL, PROV_RSA_FULL, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CRYPT_VERIFYCONTEXT)); 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_NE(FALSE, CryptGenRandom(crypt_ctx, arraysize(data), &data[0])); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_NE(FALSE, CryptReleaseContext(crypt_ctx, 0)); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Hexify the value. 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string result(base::HexEncode(&data[0], arraysize(data))); 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(8, result.size()); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Replace the first digit with the letter 'R' (for "random", get it?). 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result[0] = 'R'; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return base::ASCIIToWide(result); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace installer { 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SelfCleaningTempDirTest : public testing::Test { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test the implementation of GetTopDirToCreate when given the root of a 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// volume. 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SelfCleaningTempDirTest, TopLevel) { 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath base_dir; 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SelfCleaningTempDir::GetTopDirToCreate(base::FilePath(L"C:\\"), &base_dir); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(base_dir.empty()); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test the implementation of GetTopDirToCreate when given a non-existant dir 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// under the root of a volume. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SelfCleaningTempDirTest, TopLevelPlusOne) { 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath base_dir; 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath parent_dir(L"C:\\"); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parent_dir = parent_dir.Append(GetRandomFilename()); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SelfCleaningTempDir::GetTopDirToCreate(parent_dir, &base_dir); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(parent_dir, base_dir); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test that all intermediate dirs are cleaned up if they're empty when 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Delete() is called. 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SelfCleaningTempDirTest, RemoveUnusedOnDelete) { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make a directory in which we'll work. 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir work_dir; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.CreateUniqueTempDir()); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make up some path under the temp dir. 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath parent_temp_dir(work_dir.path().Append(L"One").Append(L"Two")); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SelfCleaningTempDir temp_dir; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir.Initialize(parent_temp_dir, L"Three")); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(parent_temp_dir.Append(L"Three"), temp_dir.path()); 777dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(temp_dir.path())); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir.Delete()); 797dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.Append(L"Three"))); 807dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir)); 817dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName())); 827dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.Delete()); 847dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test that two clients can work in the same area. 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SelfCleaningTempDirTest, TwoClients) { 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make a directory in which we'll work. 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir work_dir; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.CreateUniqueTempDir()); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make up some path under the temp dir. 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath parent_temp_dir(work_dir.path().Append(L"One").Append(L"Two")); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SelfCleaningTempDir temp_dir1; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SelfCleaningTempDir temp_dir2; 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // First client is created. 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir1.Initialize(parent_temp_dir, L"Three")); 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Second client is created in the same space. 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir2.Initialize(parent_temp_dir, L"Three")); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Both clients are where they are expected. 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(parent_temp_dir.Append(L"Three"), temp_dir1.path()); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(parent_temp_dir.Append(L"Three"), temp_dir2.path()); 1047dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(temp_dir1.path())); 1057dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(temp_dir2.path())); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Second client goes away. 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir2.Delete()); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The first is now useless. 1097dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(temp_dir1.path())); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // But the intermediate dirs are still present 1117dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(parent_temp_dir)); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Now the first goes away. 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir1.Delete()); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // And cleans up after itself. 1157dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.Append(L"Three"))); 1167dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir)); 1177dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName())); 1187dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.Delete()); 1207dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test that all intermediate dirs are cleaned up if they're empty when the 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// destructor is called. 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SelfCleaningTempDirTest, RemoveUnusedOnDestroy) { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make a directory in which we'll work. 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir work_dir; 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.CreateUniqueTempDir()); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make up some path under the temp dir. 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath parent_temp_dir(work_dir.path().Append(L"One").Append(L"Two")); 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SelfCleaningTempDir temp_dir; 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir.Initialize(parent_temp_dir, L"Three")); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(parent_temp_dir.Append(L"Three"), temp_dir.path()); 1367dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(temp_dir.path())); 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1387dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.Append(L"Three"))); 1397dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir)); 1407dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName())); 1417dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.Delete()); 1437dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test that intermediate dirs are left behind if they're not empty when the 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// destructor is called. 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SelfCleaningTempDirTest, LeaveUsedOnDestroy) { 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const char kHiHon[] = "hi, hon"; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make a directory in which we'll work. 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir work_dir; 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.CreateUniqueTempDir()); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make up some path under the temp dir. 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath parent_temp_dir(work_dir.path().Append(L"One").Append(L"Two")); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SelfCleaningTempDir temp_dir; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(temp_dir.Initialize(parent_temp_dir, L"Three")); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(parent_temp_dir.Append(L"Three"), temp_dir.path()); 1617dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(temp_dir.path())); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Drop a file somewhere. 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(arraysize(kHiHon) - 1, 164a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::WriteFile(parent_temp_dir.Append(GetRandomFilename()), 165a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) kHiHon, arraysize(kHiHon) - 1)); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1677dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.Append(L"Three"))); 1687dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::DirectoryExists(parent_temp_dir)); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(work_dir.Delete()); 1707dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::DirectoryExists(parent_temp_dir.DirName().DirName())); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace installer 174