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