drivers/gpu/msd-intel-gen/tests/mock/mock_address_space.cc

// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "mock_address_space.h"

bool MockAddressSpace::AllocLocked(size_t size, uint8_t align_pow2, uint64_t* addr_out)
{
    DASSERT(magma::is_page_aligned(size));
    uint64_t addr = magma::round_up(next_addr_, 1ul << align_pow2);
    allocations_[addr] = Allocation{size, true, true};
    *addr_out = addr;
    next_addr_ = addr + size;
    return true;
}

bool MockAddressSpace::FreeLocked(uint64_t addr)
{
    auto iter = allocations_.find(addr);
    if (iter == allocations_.end())
        return false;
    iter->second.allocated = false;
    return true;
}

bool MockAddressSpace::ClearLocked(uint64_t addr, uint64_t page_count)
{
    auto iter = allocations_.find(addr);
    if (iter == allocations_.end())
        return false;
    iter->second.clear = true;
    return true;
}

bool MockAddressSpace::InsertLocked(uint64_t addr,
                                    magma::PlatformBusMapper::BusMapping* bus_mapping)
{
    auto iter = allocations_.find(addr);
    if (iter == allocations_.end())
        return false;
    iter->second.clear = false;
    return true;
}