/mordor/streams/temp.cpp

// Copyright (c) 2009 - Mozy, Inc.

#include "temp.h"

#include "mordor/config.h"
#include "mordor/string.h"
#include "mordor/log.h"

namespace Mordor {
static ConfigVar<std::string>::ptr g_tempDir = Config::lookup(
    "mordor.tempdir",
    std::string(""),
    "Temporary directory (blank for system default)");

static Logger::ptr g_log = Log::lookup("mordor:streams");

#ifdef WINDOWS
static bool EnsureFolderExist(const std::wstring & wtempdir)
{
    size_t backslash = 1;
    error_t error;
    std::wstring currentFolder;
    while ((backslash = wtempdir.find_first_of(L'\\', backslash + 1)) != std::wstring::npos) {
        currentFolder = wtempdir.substr(0, backslash);
        CreateDirectoryW(currentFolder.c_str(), NULL);
        error = lastError();
        if (error != ERROR_ALREADY_EXISTS && error != ERROR_SUCCESS) {
            MORDOR_LOG_ERROR(g_log) << "Fail to create folder (" << wtempdir.c_str() << "), LastError:" << error;
            return false;
        }
    }
    CreateDirectoryW(&wtempdir[0], NULL);
    error = lastError();
    if (error == ERROR_ALREADY_EXISTS || error == ERROR_SUCCESS) {
        return true;
    } else {
        MORDOR_LOG_ERROR(g_log) << "Fail to create folder (" << wtempdir.c_str() << "), LastError:" << error;
        return false;
    }
}
#endif

TempStream::TempStream(const std::string &prefix, bool deleteOnClose,
                       IOManager *ioManager, Scheduler *scheduler)
{
    std::string tempdir;
    bool absolutePath =
#ifdef WINDOWS
        (prefix.size() >= 2 && (prefix[1] == ':' || prefix[1] == '\\')) ||
        (!prefix.empty() && prefix[0] == '\\');
#else
        !prefix.empty() && prefix[0] == '/';
#endif
    if (!absolutePath)
        tempdir = g_tempDir->val();
#ifdef WINDOWS
    std::wstring wtempdir = toUtf16(tempdir);
    if (!absolutePath && wtempdir.empty()) {
        wtempdir.resize(MAX_PATH);
        DWORD len = GetTempPathW(MAX_PATH, &wtempdir[0]);
        if (len == 0)
            wtempdir = L".";
        else
            wtempdir.resize(len);
    }
    std::wstring prefixW = toUtf16(prefix);
    size_t backslash = prefixW.rfind(L'\\');
    if (backslash != std::wstring::npos) {
        wtempdir += prefixW.substr(0, backslash);
        prefixW = prefixW.substr(backslash + 1);
    }
    std::wstring tempfile;
    tempfile.resize(MAX_PATH);

    UINT ret = GetTempFileNameW(wtempdir.c_str(),
        prefixW.c_str(),
        0,
        &tempfile[0]);
    error_t error;
    if (ret == 0) {
        //corner case, so call logic here only when GetTempFileNameW fails.
        error = lastError();
        if (GetFileAttributesW(wtempdir.c_str()) == INVALID_FILE_ATTRIBUTES && EnsureFolderExist(wtempdir)) {
            MORDOR_LOG_INFO(g_log) << "try one more time of GetTempFileNameW";
            ret = GetTempFileNameW(wtempdir.c_str(),
                prefixW.c_str(),
                0,
                &tempfile[0]);
            error = lastError();
        }
        if (ret == 0) {
            MORDOR_LOG_ERROR(g_log) << "GetTempFileNameW(" << wtempdir.c_str() << "," << prefixW.c_str() << ") fails with LastError:" << error;
            MORDOR_THROW_EXCEPTION_FROM_LAST_ERROR_API("GetTempFileNameW");
        }
    }
    tempfile.resize(wcslen(tempfile.c_str()));
    init(Mordor::toUtf8(tempfile), FileStream::READWRITE,
        (FileStream::CreateFlags)(FileStream::OPEN |
            (deleteOnClose ? FileStream::DELETE_ON_CLOSE : 0)),
        ioManager, scheduler);
#else
    if (!absolutePath && tempdir.empty()) {
        const char* tmpdirenv = getenv("TMPDIR");
        if (tmpdirenv != NULL) {
            tempdir = tmpdirenv;
        }
    }

    if (!absolutePath && tempdir.empty())
        tempdir = "/tmp/" + prefix + "XXXXXX";
    else if (!absolutePath) {
        if (tempdir[tempdir.length()-1] != '/')
            tempdir += "/";
        tempdir += prefix + "XXXXXX";
    }
    else
        tempdir = prefix + "XXXXXX";
    int fd = mkstemp(&tempdir[0]);
    if (fd < 0)
        MORDOR_THROW_EXCEPTION_FROM_LAST_ERROR_API("mkstemp");
    init(fd, ioManager, scheduler);
    if (deleteOnClose) {
        int rc = unlink(tempdir.c_str());
        if (rc != 0)
            MORDOR_THROW_EXCEPTION_FROM_LAST_ERROR_API("unlink");
    }
    m_path = tempdir;
#endif
}
}