libcruft-util/io_win32.hpp

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * Copyright 2014-2016 Danny Robson <danny@nerdcruft.net>
 */

#ifndef __UTIL_IO_WIN32_HPP
#define __UTIL_IO_WIN32_HPP

#include "io.hpp"
#include "win32/handle.hpp"
#include "view.hpp"

#include <cstdint>
#include <experimental/filesystem>
#include <windows.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>


///////////////////////////////////////////////////////////////////////////////
// compatibility definitions
enum : int {
    PROT_NONE   = 0,
    PROT_READ   = 1 << 0,
    PROT_EXEC   = 1 << 1,
    PROT_WRITE  = 1 << 2
};


//-----------------------------------------------------------------------------
enum : int {
    MAP_SHARED,
    MAP_PRIVATE,

    MAP_ANONYMOUS
};


///////////////////////////////////////////////////////////////////////////////
// implementation definitions
namespace util {
    namespace detail::win32 {
        class mapped_file {
        public:
            mapped_file (::util::win32::handle &&,
                         int fflags = O_RDONLY,
                         int mflags = PROT_READ);
            mapped_file (const std::experimental::filesystem::path &path,
                         int fflags = O_RDONLY,
                         int mflags = PROT_READ);
            mapped_file (const util::posix::fd&,
                         int fflag = O_RDONLY,
                         int mflags = PROT_READ);

            mapped_file (const mapped_file&) = delete;
            mapped_file& operator= (const mapped_file&) = delete;

            size_t size  (void) const;
            bool   empty (void) const;

            const uint8_t* data (void) const &;
            uint8_t*       data (void) &;

            uint8_t* begin (void) &;
            uint8_t* end   (void) &;

            const uint8_t* begin (void) const &;
            const uint8_t* end   (void) const &;

            const uint8_t* cbegin (void) const &;
            const uint8_t* cend   (void) const &;

            template <typename T>
            util::view<std::add_const_t<T>*>
            as_view () const &
            {
                return {
                    reinterpret_cast<const T*> (cbegin ()),
                    reinterpret_cast<const T*> (align (cend (), alignof (T)))
                };
            }

            template <typename T>
            util::view<T*>
            as_view () &
            {
                return {
                    reinterpret_cast<T *> (begin ()),
                    reinterpret_cast<T *> (align (end (), alignof(T)))
                };
            }

        private:
            ::util::win32::handle m_file;
            ::util::win32::handle m_mapping;

            std::unique_ptr<uint8_t,BOOL(*)(LPCVOID)> m_data;
            uint64_t m_size;
        };
    }

    typedef detail::win32::mapped_file mapped_file;
}

#endif