soutade/libgourou
3
6
Fork 4
Code Issues 1 Pull Requests Packages Releases 4 Activity

Remove use of tempnam function and fix bug (bad check of rename return)

Browse Source
This commit is contained in:
Grégory Soutadé 2022-04-23 17:41:54 +02:00
parent f568b5d3a8
commit 4f9b2de5a5
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
utils/adept_remove.cpp
View File

@ -114,16 +114,16 @@ public:
// Use temp file for PDF
if (type == gourou::DRMProcessor::ITEM_TYPE::PDF)
{
char* tempFile = tempnam("/tmp", NULL);
processor.removeDRM(inputFile, tempFile, type, encryptionKey, encryptionKeySize);
std::string tempFile = filename + ".tmp";
/* Be sure there is not already a temp file */
unlink(tempFile.c_str());
processor.removeDRM(filename, tempFile, type, encryptionKey, encryptionKeySize);
/* Original file must be removed before doing a copy... */
unlink(inputFile);
if (!rename(tempFile, filename.c_str()))
unlink(filename.c_str());
if (rename(tempFile.c_str(), filename.c_str()))
{
free(tempFile);
EXCEPTION(gourou::DRM_FILE_ERROR, "Unable to copy " << tempFile << " into " << filename);
}
free(tempFile);
}
else
processor.removeDRM(inputFile, filename, type, encryptionKey, encryptionKeySize);