libcruft-util/io_win32.cpp

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2014 Danny Robson <danny@nerdcruft.net>
 */

#include "io_win32.hpp"

#include "debug.hpp"
#include "except.hpp"
#include "win32/error.hpp"

#include <windows.h>

using util::detail::win32::mapped_file;


///////////////////////////////////////////////////////////////////////////////
static constexpr
DWORD
fflags_to_generic (int flags) {
    switch (flags) {
    case O_RDONLY:  return GENERIC_READ;
    case O_WRONLY:  return GENERIC_WRITE;
    case O_RDWR:    return GENERIC_READ | GENERIC_WRITE;
    }

    unreachable ();
}


//-----------------------------------------------------------------------------
static constexpr
DWORD
fflags_to_access (int fflags)
{
    switch (fflags) {
    case O_RDONLY:  return FILE_MAP_READ;
    case O_WRONLY:  return FILE_MAP_WRITE;
    case O_RDWR:    return FILE_MAP_WRITE;
    }

    unreachable ();
}


//-----------------------------------------------------------------------------
static constexpr
DWORD
mflags_to_protect (int mflags) {
    DWORD res = 0;

    if (mflags & PROT_READ)  res |= PAGE_READONLY;
    if (mflags & PROT_WRITE) res |= PAGE_READWRITE;
    if (mflags & PROT_EXEC)  res |= PAGE_EXECUTE;

    return res;
}


///////////////////////////////////////////////////////////////////////////////
mapped_file::mapped_file (::util::win32::handle &&src,
                          int fflags,
                          int mflags):
    m_file (std::forward<::util::win32::handle> (src)),
    m_data (nullptr, UnmapViewOfFile)
{
    // I would rather perform checks on filesize after mapping, but mapping
    // requires a check for empty files before we perform the mapping to
    // detect errors it throws in that specific situation.
    DWORD hi_size, lo_size = GetFileSize (m_file, &hi_size);
    m_size = (uint64_t)hi_size << 32 | lo_size;

    m_mapping.reset (
        CreateFileMapping (
            m_file,
            nullptr,
            mflags_to_protect (mflags),
            0, 0,
            nullptr
        )
    );

    // Apparently Windows lacks the ability to map zero length files; fucking
    // hell. Try not to collapse, but instead bail with a null mapping and
    // pray the user doesn't do something stupid with the result.
    if (!m_mapping) {
        auto err = win32::error::last_code ();
        if (err == ERROR_FILE_INVALID && m_size == 0)
            return;

        win32::error::throw_code (err);
    }

    auto view = MapViewOfFile (
        m_mapping,
        fflags_to_access (fflags),
        0, 0,
        0
    );

    if (!view)
        win32::error::throw_code ();

    m_data.reset (
        static_cast<unsigned char*> (view)
    );
}


//-----------------------------------------------------------------------------
mapped_file::mapped_file (const std::experimental::filesystem::path &path,
                          int fflags,
                          int mflags):
    mapped_file (
        ::util::win32::handle (
            ::CreateFile (
                path.string ().c_str (),
                fflags_to_generic (fflags),
                fflags == O_RDONLY ? FILE_SHARE_READ : 0,
                nullptr,
                OPEN_EXISTING,
                FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN,
                nullptr
            )
        ),
        fflags,
        mflags
    )
{ ; }


//-----------------------------------------------------------------------------
mapped_file::mapped_file (const util::posix::fd &src,
                          int fflags,
                          int mflags):
    mapped_file (util::win32::handle (reinterpret_cast<HANDLE> (_get_osfhandle (src))),
                 fflags,
                 mflags)
{ };


///////////////////////////////////////////////////////////////////////////////
size_t
mapped_file::size (void) const
{
    return m_size;
}


//-----------------------------------------------------------------------------
bool
mapped_file::empty (void) const
{
    return size () == 0;
}


//-----------------------------------------------------------------------------
const uint8_t*
mapped_file::data (void) const &
{
    return m_data.get ();
}


//-----------------------------------------------------------------------------
uint8_t*
mapped_file::data (void) &
{
    return m_data.get ();
}


//-----------------------------------------------------------------------------
uint8_t*
mapped_file::begin (void) &
{
    return data ();
}


//-----------------------------------------------------------------------------
uint8_t*
mapped_file::end (void) &
{
    return data () + size ();
}


//-----------------------------------------------------------------------------
const uint8_t*
mapped_file::begin (void) const &
{
    return cbegin ();
}


//-----------------------------------------------------------------------------
const uint8_t*
mapped_file::end (void) const &
{
    return cend ();
}


//-----------------------------------------------------------------------------
const uint8_t*
mapped_file::cbegin (void) const &
{
    return data ();
}


//-----------------------------------------------------------------------------
const uint8_t*
mapped_file::cend (void) const &
{
    return data () + size ();
}