generic-library/poco
1
0
Fork 0
mirror of https://github.com/pocoproject/poco.git synced 2024-12-12 18:20:26 +01:00
Code Issues Projects Releases Wiki Activity

Poco::TemporaryFile: fix possible naming collisions due to random zero increment

Browse Source
This commit is contained in:
Günter Obiltschnig 2023-10-14 11:34:59 +02:00
parent ed43543113
commit 06a03d1ada
4 changed files with 27 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Foundation/include/Poco/TemporaryFile.h
View File

@ -40,6 +40,10 @@ class Foundation_API TemporaryFile: public File
/// The class does, however, delete the temporary
/// file - either in the destructor, or immediately
/// before the application terminates.
///
/// Note: Due to the way the temporary file names are generated,
/// collisions may be possible after 400-500 Million
/// file names generated by a single process.
{
public:
TemporaryFile();

2
Foundation/src/TemporaryFile.cpp
View File

@ -161,7 +161,7 @@ std::string TemporaryFile::tempName(const std::string& tempDir)
randomChars.append(alphabet);
}
}
n = (count += random.next(1000));
n = (count += random.next(1000) + 1);
}
std::ostringstream name;

23
Foundation/testsuite/src/FileTest.cpp
View File

@ -380,6 +380,7 @@ void FileTest::testCopy()
f1.setWriteable().remove();
}
void FileTest::testCopyFailIfDestinationFileExists()
{
std::ofstream ostr("testfile.dat");
@ -414,6 +415,7 @@ void FileTest::testMove()
assertTrue (f1 == f2);
}
void FileTest::testMoveFailIfDestinationFileExists() {
std::ofstream ostr("testfile.dat");
ostr << "Hello, world!" << std::endl;
@ -430,6 +432,7 @@ void FileTest::testMoveFailIfDestinationFileExists() {
f1.setWriteable().remove();
}
void FileTest::testCopyDirectory()
{
Path pd1("testdir");
@ -499,6 +502,7 @@ void FileTest::testCopyDirectory()
fd3.remove(true);
}
void FileTest::testCopyDirectoryFailIfExists()
{
Path pd1("testdir");
@ -538,6 +542,7 @@ void FileTest::testCopyDirectoryFailIfExists()
fd2.remove(true);
}
void FileTest::testRename()
{
std::ofstream ostr("testfile.dat");
@ -555,6 +560,7 @@ void FileTest::testRename()
f2.remove();
}
void FileTest::testRenameFailIfExists() {
std::ofstream ostr("testfile.dat");
ostr << "Hello, world!" << std::endl;
@ -580,8 +586,6 @@ void FileTest::testRenameFailIfExists() {
}
void FileTest::testLongPath()
{
#if defined(_WIN32) && !defined(_WIN32_WCE)
@ -620,6 +624,20 @@ void FileTest::testUnixFileExtension()
assertEqual("", path2.getExtension());
}
void FileTest::testTemporaryFile()
{
const int COUNT = 10000;
std::set<std::string> paths;
for (int i = 0; i < COUNT; i++)
{
Poco::TemporaryFile f;
paths.insert(f.path());
}
assertTrue (paths.size() == COUNT);
}
void FileTest::setUp()
{
File f("testfile.dat");
@ -669,6 +687,7 @@ CppUnit::Test* FileTest::suite()
CppUnit_addTest(pSuite, FileTest, testRootDir);
CppUnit_addTest(pSuite, FileTest, testLongPath);
CppUnit_addTest(pSuite, FileTest, testUnixFileExtension);
CppUnit_addTest(pSuite, FileTest, testTemporaryFile);
return pSuite;
}

1
Foundation/testsuite/src/FileTest.h
View File

@ -43,6 +43,7 @@ public:
void testRootDir();
void testLongPath();
void testUnixFileExtension();
void testTemporaryFile();
void setUp();
void tearDown();