windows_file_handle.h

Go to the documentation of this file.

// Copyright (c) 2018, Smart Projects Holdings Ltd
// All rights reserved.
// See LICENSE file for license details.

#ifndef _UGCS_VSM_WINDOWS_FILE_HANDLE_H_
#define _UGCS_VSM_WINDOWS_FILE_HANDLE_H_

#include <ugcs/vsm/file_processor.h>
#include <cstring>

#include <windows.h>

/* Some weird code can define this macro in Windows headers. */
#ifdef ERROR
#undef ERROR
#endif

namespace ugcs {
namespace vsm {
namespace internal {

class Windows_file_handle: public File_processor::Stream::Native_handle {
public:

    Windows_file_handle(
            HANDLE handle,
            HANDLE write_handle = INVALID_HANDLE_VALUE);

    Windows_file_handle(
            const std::string &path,
            File_processor::Stream::Mode mode);

    virtual
    ~Windows_file_handle();

    virtual void
    Write() override;

    virtual void
    Read() override;

    virtual File_processor::Stream::Lock_result
    Try_lock() override;

    virtual bool
    Lock() override;

    virtual bool
    Unlock() override;

    virtual bool
    Cancel_write() override;

    virtual bool
    Cancel_read() override;

    void
    Cancel_io(bool write);

    void
    Write_complete_cbk(size_t transfer_size, DWORD error);

    void
    Read_complete_cbk(size_t transfer_size, DWORD error);

    void
    Lock_complete_cbk(DWORD error);

    void
    Io_complete_cbk(OVERLAPPED *io_cb, size_t transfer_size, DWORD error);

    virtual void
    Close() override;

    static Io_result
    Map_error(DWORD error);

protected:
    friend class Overlapped_io_controller;

    HANDLE handle;
    HANDLE write_handle = INVALID_HANDLE_VALUE;
    OVERLAPPED read_cb;
    OVERLAPPED write_cb;
    OVERLAPPED lock_cb;
    HANDLE lock_complete_event = INVALID_HANDLE_VALUE;
    DWORD lock_complete_result = ERROR_ARENA_TRASHED;
    std::shared_ptr<std::vector<uint8_t>> read_buf;
    size_t read_size,
           write_size,
           min_read_size;
    Io_stream::Offset read_offset,
                      write_offset;
    std::mutex write_mutex,
               read_mutex;
};

} /* namespace internal */
} /* namespace vsm */
} /* namespace ugcs */

#endif /* _UGCS_VSM_WINDOWS_FILE_HANDLE_H_ */