ctr_Repair/branches/work/RW_Aging/sources/ConsoleBackup/SavedataChecker.cpp

/*---------------------------------------------------------------------------*
 Project:  Horizon
 File:     NandSavedataChecker.cpp

 Copyright 2009 Nintendo.  All rights reserved.

 These coded instructions, statements, and computer programs contain
 proprietary information of Nintendo of America Inc. and/or Nintendo
 Company Ltd., and are protected by Federal copyright law.  They may
 not be disclosed to third parties or copied or duplicated in any form,
 in whole or in part, without the prior written consent of Nintendo.

 $Rev$
 *---------------------------------------------------------------------------*/

#include <nn/fs/fs_ApiSysSaveData.h>
#include <nn/fs/fs_ApiSharedExtSaveData.h>

#include "SavedataChecker.h"
#include "CommonLogger.h"
#include "FileTransfer.h"

namespace ConsoleBackup
{

namespace
{

}

SavedataCheckerBase::SavedataCheckerBase(void* buf, size_t size) :
        m_Buf(buf), m_Bufsize(size), m_CalculatedFileSize(0), m_TotalReadSize(0)
{

}

SavedataCheckerBase::~SavedataCheckerBase()
{
    NN_LOG("m_TotalReadSize = %lld\n", m_TotalReadSize);
}

nn::Result SavedataCheckerBase::CleanUpFilesRecursively(bool* modified, std::wstring currentDirectory)
{
    nn::fs::Directory dir;
    nn::fs::DirectoryEntry entry;
    nn::Result result;

    NN_LOG("%s\n", common::GetCharStr(currentDirectory.c_str()));
    result = dir.TryInitialize(currentDirectory.c_str());
    if(result.IsFailure())
    {
        COMMON_LOGGER_WARN("Delete Directory %s\n", common::GetCharStr(currentDirectory.c_str()));
        result = nn::fs::TryDeleteDirectory(currentDirectory.c_str());
        *modified = true;
        return result;
    }

    for (;;)
    {
        s32 numRead;
        result = dir.TryRead(&numRead, &entry, 1);
        if(result.IsFailure())
        {
            dir.Finalize();
            COMMON_LOGGER_WARN("Delete Directory %s\n", common::GetCharStr(currentDirectory.c_str()));
            result = nn::fs::TryDeleteDirectoryRecursively(currentDirectory.c_str());
            *modified = true;
            continue;
        }


        if(numRead == 0)
        {
            break;
        }

        if (std::wcscmp(entry.entryName, L".") == 0 || std::wcscmp(entry.entryName, L"..") == 0)
        {
            continue;
        }


        // ディレクトリの場合
        if (entry.attributes.isDirectory)
        {
            return CleanUpFilesRecursively(modified, currentDirectory + std::wstring(entry.entryName) + std::wstring(L"/"));
        }
        // ファイルの場合
        else
        {
            nn::fs::FileInputStream file;
            std::wstring filePath = (currentDirectory + std::wstring(entry.entryName)).c_str();
            const wchar_t* path = filePath.c_str();

            result = file.TryInitialize(path);
            if(result.IsFailure())
            {
                nn::dbg::PrintResult(result);
                COMMON_LOGGER_WARN("Cannot Initialize %s, delete.\n", common::GetCharStr(entry.entryName));
                result = nn::fs::TryDeleteFile(path);
                *modified = true;
                continue;
            }

            for (;;)
            {
                s32 readSize;
                result = file.TryRead(&readSize, m_Buf, m_Bufsize);
                if(result.IsFailure())
                {
                    nn::dbg::PrintResult(result);
                    COMMON_LOGGER_WARN("Cannot read %s, delete.\n", common::GetCharStr(entry.entryName));
                    m_TotalReadSize += file.GetSize();
                    file.Finalize();
                    result = nn::fs::TryDeleteFile(path);
                    COMMON_LOGGER_RESULT_IF_FAILED(result);
                    *modified = true;
                    break;
                }
                else
                {
                    m_TotalReadSize += readSize;
                }

                if(readSize == 0)
                {
                    break;
                }
            }
        }
    }

    return nn::ResultSuccess();
}

nn::Result SavedataCheckerBase::GetFileSize(std::wstring currentDirectory)
{
    nn::fs::Directory dir;
    nn::fs::DirectoryEntry entry;
    nn::Result result;

    NN_LOG("%s\n", common::GetCharStr(currentDirectory.c_str()));
    result = dir.TryInitialize(currentDirectory.c_str());
    if(result.IsFailure())
    {
        return result;
    }

    for (;;)
    {
        s32 numRead;
        result = dir.TryRead(&numRead, &entry, 1);
        if(result.IsFailure())
        {
            continue;
        }


        if(numRead == 0)
        {
            break;
        }

        if (std::wcscmp(entry.entryName, L".") == 0 || std::wcscmp(entry.entryName, L"..") == 0)
        {
            continue;
        }


        // ディレクトリの場合
        if (entry.attributes.isDirectory)
        {
            return GetFileSize(currentDirectory + std::wstring(entry.entryName) + std::wstring(L"/"));
        }
        // ファイルの場合
        else
        {
            nn::fs::FileInputStream file;
            std::wstring filePath = (currentDirectory + std::wstring(entry.entryName)).c_str();
            const wchar_t* path = filePath.c_str();

            result = file.TryInitialize(path);
            if(result.IsFailure())
            {
                continue;
            }

            m_CalculatedFileSize += file.GetSize();
        }
    }

    return nn::ResultSuccess();
}

s64 SavedataCheckerBase::GetCalculatedSize()
{
    return m_CalculatedFileSize;
}

s64 SavedataCheckerBase::GetTotalReadSize()
{
    return m_TotalReadSize;
}


NandSavedataChecker::NandSavedataChecker()
{
    // TODO 自動生成されたコンストラクター・スタブ

}

NandSavedataChecker::NandSavedataChecker(void* buf, size_t size) : m_Buf(buf), m_Bufsize(size)
{
    m_pSharedExtSaveChecker = new SharedExtSavedataChecker(buf, size);
    m_pSysSaveChecker = new SystemSavedataChecker(buf, size);
}

NandSavedataChecker::~NandSavedataChecker()
{
    delete m_pSharedExtSaveChecker;
    delete m_pSysSaveChecker;
}

nn::Result NandSavedataChecker::CleanUp()
{
    nn::Result result;

    m_pSharedExtSaveChecker->CalculateFileSize();
    m_pSysSaveChecker->CalculateFileSize();

    result = m_pSharedExtSaveChecker->CleanUp();
    COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

    result = m_pSysSaveChecker->CleanUp();
    COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

    return nn::ResultSuccess();
}

s64 NandSavedataChecker::GetProgress()
{
    if (m_pSharedExtSaveChecker->GetCalculatedSize() == 0 || m_pSysSaveChecker->GetCalculatedSize() == 0)
    {
        return 0;
    }
    else
    {
        return (m_pSharedExtSaveChecker->GetTotalReadSize() + m_pSysSaveChecker->GetTotalReadSize()) * 100
                / (m_pSharedExtSaveChecker->GetCalculatedSize() + m_pSysSaveChecker->GetCalculatedSize());
    }
}

SystemSavedataChecker::SystemSavedataChecker()
{

}

SystemSavedataChecker::SystemSavedataChecker(void* buf, size_t size) : SavedataCheckerBase(buf, size)
{
}


SystemSavedataChecker::~SystemSavedataChecker()
{
}



nn::Result SystemSavedataChecker::CleanUp()
{
    nn::Result result;

    bool modified = false;
    std::wstring currentDirectory;
    for (s32 i = 0; i < SYSTEM_SAVE_DATA_NUM; i++)
    {
        result = nn::fs::MountSystemSaveData(SYSTEM_SAVEDATA_ARCHIVE_NAME, SYSTEM_SAVEDATA_COUPLE_LIST[i].id );
        if (result.IsFailure())
        {
            if(result <= nn::fs::ResultVerificationFailed())
            {
                NN_LOG("Mount Error: %ls\n", SYSTEM_SAVEDATA_COUPLE_LIST[i].name.c_str());
                // 削除する
                COMMON_LOGGER_WARN("Delete Savedata %s\n", common::GetCharStr(SYSTEM_SAVEDATA_COUPLE_LIST[i].name.c_str()));
                result = nn::fs::DeleteSystemSaveData(SYSTEM_SAVEDATA_COUPLE_LIST[i].id);
                COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);
            }
        }
        else
        {
            NN_LOG("Mount %ls\n", SYSTEM_SAVEDATA_COUPLE_LIST[i].name.c_str());
            // ファイルを個別にチェックする
            result = CleanUpFilesRecursively(&modified, L"ssave:/");
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

            if(modified)
            {
                result = nn::fs::CommitSystemSaveData(SYSTEM_SAVEDATA_ARCHIVE_NAME);
            }

            result = nn::fs::Unmount(SYSTEM_SAVEDATA_ARCHIVE_NAME);
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);
        }
    }

    return nn::ResultSuccess();
}

nn::Result SystemSavedataChecker::CalculateFileSize()
{
    nn::Result result;

    for (s32 i = 0; i < SYSTEM_SAVE_DATA_NUM; i++)
    {
        result = nn::fs::MountSystemSaveData(SYSTEM_SAVEDATA_ARCHIVE_NAME, SYSTEM_SAVEDATA_COUPLE_LIST[i].id );
        if (result.IsSuccess())
        {
            result = GetFileSize(L"ssave:/");
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

            result = nn::fs::Unmount(SYSTEM_SAVEDATA_ARCHIVE_NAME);
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);
        }
    }

    NN_LOG("CalculatedFileSize = %lld\n", m_CalculatedFileSize);
    return result;
}

SharedExtSavedataChecker::SharedExtSavedataChecker()
{
}

SharedExtSavedataChecker::SharedExtSavedataChecker(void* buf, size_t size) : SavedataCheckerBase(buf, size)
{
}



SharedExtSavedataChecker::~SharedExtSavedataChecker()
{

}

nn::Result SharedExtSavedataChecker::CleanUp()
{
    nn::Result result;

    const size_t ARRAY_SIZE = 256;
    s32 numId;
    bit32 IdArray[ARRAY_SIZE];

    result = nn::fs::EnumerateSharedExtSaveData(&numId, IdArray, ARRAY_SIZE);
    COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

    NN_LOG_DEBUG("ExtData num = %d\n", numId);
    bool modified = false;
    for (s32 i = 0; i < numId; i++ )
    {
        if (nn::fs::MountSharedExtSaveData(SHARED_EXT_SAVEDATA_ARCHIVE_NAME, IdArray[i]).IsSuccess())
        {
            NN_LOG("Mount %x\n", IdArray[i]);

            // アーカイブ内のファイル・ディレクトリをチェックする
            result = CleanUpFilesRecursively(&modified, L"shext:/");
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

            result = nn::fs::Unmount(SHARED_EXT_SAVEDATA_ARCHIVE_NAME);
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);
        }
        else
        {
            // アーカイブごと削除する
            if(result <= nn::fs::ResultVerificationFailed())
            {
                NN_LOG("Mount Error: %x\n", IdArray[i]);

                // 削除する
                COMMON_LOGGER_WARN("Delete Shared-Ext-Savedata %x\n", IdArray[i]);
                result = nn::fs::DeleteSharedExtSaveData(IdArray[i]);
                COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);
            }
        }
    }

    return nn::ResultSuccess();
}

nn::Result SharedExtSavedataChecker::CalculateFileSize()
{
    nn::Result result;

    const size_t ARRAY_SIZE = 256;
    s32 numId;
    bit32 IdArray[ARRAY_SIZE];

    result = nn::fs::EnumerateSharedExtSaveData(&numId, IdArray, ARRAY_SIZE);
    COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

    NN_LOG_DEBUG("ExtData num = %d\n", numId);
    for (s32 i = 0; i < numId; i++ )
    {
        result = nn::fs::MountSharedExtSaveData(SHARED_EXT_SAVEDATA_ARCHIVE_NAME, IdArray[i]);
        if (result.IsSuccess())
        {
            result = GetFileSize(L"shext:/");
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);

            result = nn::fs::Unmount(SHARED_EXT_SAVEDATA_ARCHIVE_NAME);
            COMMON_LOGGER_RETURN_RESULT_IF_FAILED(result);
        }
    }

    NN_LOG("CalculatedFileSize = %lld\n", m_CalculatedFileSize);
    return nn::ResultSuccess();
}

} /* namespace ConsoleBackup */