PowerToys/src/modules/FileLocksmith/FileLocksmithLib/IPC.cpp

#include "pch.h"

#include "IPC.h"
#include "Constants.h"

#include <common/SettingsAPI/settings_helpers.h>

constexpr DWORD DefaultPipeBufferSize = 8192;
constexpr DWORD DefaultPipeTimeoutMillis = 200;

namespace ipc
{
    Writer::Writer()
    {
        start();
    }

    Writer::~Writer()
    {
        finish();
    }

    HRESULT Writer::start()
    {
        std::wstring path = PTSettingsHelper::get_module_save_folder_location(constants::nonlocalizable::PowerToyName);
        path += L"\\";
        path += constants::nonlocalizable::LastRunPath;

        try
        {
            m_stream = std::ofstream(path, std::ios::binary);
            if (!m_stream.is_open())
            {
                return E_FAIL;
            }

            return S_OK;
        }
        catch (...)
        {
            return E_FAIL;
        }
    }

    HRESULT Writer::add_path(LPCWSTR path)
    {
        int length = lstrlenW(path);
        if (!m_stream.write(reinterpret_cast<const char*>(path), length * sizeof(WCHAR)))
        {
            return E_FAIL;
        }

        WCHAR line_break = L'\n';
        if (!m_stream.write(reinterpret_cast<const char*>(&line_break), sizeof(WCHAR)))
        {
            return E_FAIL;
        }

        return S_OK;
    }

    void Writer::finish()
    {
        add_path(L"");
        m_stream.close();
    }
}