AllocationCommand.h

/*
 * Copyright (c) 2008-2014, Pedigree Developers
 *
 * Please see the CONTRIB file in the root of the source tree for a full
 * list of contributors.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#ifndef ALLOCATIONCOMMAND_H
#define ALLOCATIONCOMMAND_H

#include "pedigree/kernel/debugger/DebuggerCommand.h"
#include "pedigree/kernel/debugger/DebuggerIO.h"
#include "pedigree/kernel/debugger/Scrollable.h"
#include "pedigree/kernel/processor/state_forward.h"
#include "pedigree/kernel/processor/types.h"
#include "pedigree/kernel/utilities/StaticString.h"
#include "pedigree/kernel/utilities/Tree.h"
#include "pedigree/kernel/utilities/Vector.h"

#define NUM_BT_FRAMES 6

class AllocationCommand : public DebuggerCommand, public Scrollable
{
  public:
    AllocationCommand();

    ~AllocationCommand();

    void autocomplete(const HugeStaticString &input, HugeStaticString &output);

    bool execute(
        const HugeStaticString &input, HugeStaticString &output,
        InterruptState &state, DebuggerIO *screen);

    const NormalStaticString getString()
    {
        return NormalStaticString("page-allocations");
    }

    void allocatePage(physical_uintptr_t page);
    void freePage(physical_uintptr_t page);
    void postProcess();

    bool isMallocing()
    {
        return m_bAllocating;
    }
    void setMallocing(bool b)
    {
        m_bAllocating = b;
    }

    void checkpoint();

    //
    // Scrollable interface
    //
    virtual const char *getLine1(
        size_t index, DebuggerIO::Colour &colour, DebuggerIO::Colour &bgColour);
    virtual const char *getLine2(
        size_t index, size_t &colOffset, DebuggerIO::Colour &colour,
        DebuggerIO::Colour &bgColour);
    virtual size_t getLineCount();

  private:
    struct Allocation
    {
        physical_uintptr_t page;
        uintptr_t ra[NUM_BT_FRAMES];
        size_t n;
        size_t pid;
    };
    Vector<Allocation *> m_Allocations;
    Vector<void *> m_Frees;
    size_t m_nLines;
    Tree<size_t, Allocation *> m_Tree;
    Tree<size_t, Allocation *>::Iterator m_It;
    size_t m_nIdx;
    bool m_bAllocating;
};

extern AllocationCommand g_AllocationCommand;

#endif