diskfreespace.hxx

/*
** Copyright 2007-2024, RTE (https://www.rte-france.com)
** See AUTHORS.txt
** SPDX-License-Identifier: MPL-2.0
** This file is part of Antares-Simulator,
** Adequacy and Performance assessment for interconnected energy networks.
**
** Antares_Simulator is free software: you can redistribute it and/or modify
** it under the terms of the Mozilla Public Licence 2.0 as published by
** the Mozilla Foundation, either version 2 of the License, or
** (at your option) any later version.
**
** Antares_Simulator is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
** Mozilla Public Licence 2.0 for more details.
**
** You should have received a copy of the Mozilla Public Licence 2.0
** along with Antares_Simulator. If not, see <https://opensource.org/license/mpl-2-0/>.
*/
 
#include <yuni/yuni.h>
#include <yuni/core/system/memory.h>
#include <yuni/core/system/windows.hdr.h>
#include <yuni/core/string/wstring.h>
#include <antares/logs/logs.h>
 
using namespace Yuni;
 
template<class StringT>
static inline StringT& BytesToStringW(StringT& out, uint64_t size)
{
    if (0 == size)
        return out << L"0 byte";
 
    // bytes
    if (size < 1024)
        return out << size << L" bytes";
    // KiB
    if (size < 1024 * 1024)
        return out << (size / 1024) << L" KiB";
    // MiB
    if (size < 1024 * 1024 * 1024)
        return out << (size / (1024 * 1024)) << L" MiB";
 
    // GiB
    double s = size / (1024. * 1024 * 1024);
    return out << Math::Round(s, 1) << L" GiB";
}
 
static inline uint64_t DiskFreeSpace(const AnyString& folder)
{
    if (folder.empty())
        return (uint64_t)-1; // error
 
#ifdef YUNI_OS_WINDOWS
    unsigned __int64 i64FreeBytesToCaller;
    // unsigned __int64 i64TotalBytes;
    // unsigned __int64 i64FreeBytes;
 
    {
        Yuni::WString wstr(folder);
        if (wstr.empty())
            return (uint64_t)-1; // error
 
        if (GetDiskFreeSpaceExW(wstr.c_str(),
                                (PULARGE_INTEGER)&i64FreeBytesToCaller,
                                nullptr, //(PULARGE_INTEGER)&i64TotalBytes,
                                nullptr /*(PULARGE_INTEGER)&i64FreeBytes*/)
            != 0)
        {
            // +1 to not be strictly equal to 0
            // It won't change anything anyway
            return i64FreeBytesToCaller;
        }
    }
    {
        // the previous call may abort due to the UNC path
        // retrying with the standard path
        Yuni::WString wstr(folder);
        if (wstr.empty())
            return (uint64_t)-1; // error
 
        if (GetDiskFreeSpaceExW(wstr.c_str(),
                                (PULARGE_INTEGER)&i64FreeBytesToCaller,
                                nullptr, //(PULARGE_INTEGER)&i64TotalBytes,
                                nullptr /*(PULARGE_INTEGER)&i64FreeBytes*/)
            != 0)
        {
            // +1 to not be strictly equal to 0
            // It won't change anything anyway
            return i64FreeBytesToCaller;
        }
        else
        {
            // Antares::logs.error() << "last error : " << GetLastError() << " : " << folder;
        }
    }
#endif
 
    (void)folder;
    return (uint64_t)-1; // obviously an error
}