/*
 * Copyright (C) 2021 Huawei Device Co., Ltd.
 * 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.
 */

#ifndef AVSHAREDMEMORYBASE_H
#define AVSHAREDMEMORYBASE_H

#include <string>
#include "nocopyable.h"
#include "avsharedmemory.h"

namespace OHOS {
namespace Media {
class __attribute__((visibility("default"))) AVSharedMemoryBase
    : public AVSharedMemory, public NoCopyable {
public:
    /**
     * @brief Construct a new AVSharedMemoryBase object. This function should only be used in the
     * local process.
     *
     * @param size the memory's size, bytes.
     * @param flags the memory's accessible flags, refer to {@AVSharedMemory::Flags}.
     * @param name the debug string
     */
    static std::shared_ptr<AVSharedMemory> CreateFromLocal(
        int32_t size, uint32_t flags, const std::string &name);

    /**
     * @brief Construct a new AVSharedMemoryBase object. This function should only be used in the
     * remote process.
     *
     * @param fd the memory's fd
     * @param size the memory's size, bytes.
     * @param flags the memory's accessible flags, refer to {@AVSharedMemory::Flags}.
     * @param name the debug string
     */
    static std::shared_ptr<AVSharedMemory> CreateFromRemote(
        int32_t fd, int32_t size, uint32_t flags, const std::string &name);

    ~AVSharedMemoryBase();

    /**
     * @brief Construct a new AVSharedMemoryBase object. This function should only be used in the
     * local process.
     *
     * @param size the memory's size, bytes.
     * @param flags the memory's accessible flags, refer to {@AVSharedMemory::Flags}.
     * @param name the debug string
     */
    AVSharedMemoryBase(int32_t size, uint32_t flags, const std::string &name);

    /**
     * @brief Intialize the memory. Call this interface firstly before the other interface.
     * @return MSERR_OK if success, otherwise the errcode.
     */
    int32_t Init();

    /**
     * @brief Get the memory's fd, which only valid when the underlying memory
     * chunk is allocated through the ashmem.
     * @return the memory's fd if the memory is allocated through the ashmem, otherwise -1.
     */
    int32_t GetFd() const
    {
        return fd_;
    }

    std::string GetName() const
    {
        return name_;
    }

    /**
     * @brief Get the memory's virtual address
     * @return the memory's virtual address if the memory is valid, otherwise nullptr.
     */
    virtual uint8_t *GetBase() const override
    {
        return base_;
    }

    /**
     * @brief Get the memory's size
     * @return the memory's size if the memory is valid, otherwise -1.
     */
    virtual int32_t GetSize() const override
    {
        return (base_ != nullptr) ? size_ : -1;
    }

    /**
     * @brief Get the memory's flags set by the creator, refer to {@Flags}
     * @return the memory's flags if the memory is valid, otherwise 0.
     */
    virtual uint32_t GetFlags() const final
    {
        return (base_ != nullptr) ? flags_ : 0;
    }

protected:
    AVSharedMemoryBase(int32_t fd, int32_t size, uint32_t flags, const std::string &name);

private:
    int32_t MapMemory(bool isRemote);
    void Close() noexcept;

    uint8_t *base_;
    int32_t size_;
    uint32_t flags_;
    std::string name_;
    int32_t fd_;
};
} // namespace Media
} // namespace OHOS

#endif