134 lines
4.2 KiB
C++
134 lines
4.2 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2; -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef mozilla_devtools_HeapSnapshot__
|
|
#define mozilla_devtools_HeapSnapshot__
|
|
|
|
#include "js/HashTable.h"
|
|
#include "mozilla/ErrorResult.h"
|
|
#include "mozilla/devtools/DeserializedNode.h"
|
|
#include "mozilla/dom/BindingDeclarations.h"
|
|
#include "mozilla/HashFunctions.h"
|
|
#include "mozilla/Maybe.h"
|
|
#include "mozilla/RefPtr.h"
|
|
#include "mozilla/UniquePtr.h"
|
|
|
|
#include "CoreDump.pb.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "nsCRTGlue.h"
|
|
#include "nsCycleCollectionParticipant.h"
|
|
#include "nsISupports.h"
|
|
#include "nsWrapperCache.h"
|
|
#include "nsXPCOM.h"
|
|
|
|
namespace mozilla {
|
|
namespace devtools {
|
|
|
|
struct NSFreePolicy {
|
|
void operator()(void* ptr) {
|
|
NS_Free(ptr);
|
|
}
|
|
};
|
|
|
|
using UniqueString = UniquePtr<char16_t[], NSFreePolicy>;
|
|
|
|
struct UniqueStringHashPolicy {
|
|
struct Lookup {
|
|
const char16_t* str;
|
|
size_t length;
|
|
|
|
Lookup(const char16_t* str, size_t length)
|
|
: str(str)
|
|
, length(length)
|
|
{ }
|
|
};
|
|
|
|
static js::HashNumber hash(const Lookup& lookup) {
|
|
MOZ_ASSERT(lookup.str);
|
|
return HashString(lookup.str, lookup.length);
|
|
}
|
|
|
|
static bool match(const UniqueString& existing, const Lookup& lookup) {
|
|
MOZ_ASSERT(lookup.str);
|
|
return NS_strncmp(existing.get(), lookup.str, lookup.length) == 0;
|
|
}
|
|
};
|
|
|
|
class HeapSnapshot final : public nsISupports
|
|
, public nsWrapperCache
|
|
{
|
|
friend struct DeserializedNode;
|
|
|
|
explicit HeapSnapshot(JSContext* cx, nsISupports* aParent)
|
|
: timestamp(Nothing())
|
|
, rootId(0)
|
|
, nodes(cx)
|
|
, strings(cx)
|
|
, mParent(aParent)
|
|
{
|
|
MOZ_ASSERT(aParent);
|
|
};
|
|
|
|
// Initialize this HeapSnapshot from the given buffer that contains a
|
|
// serialized core dump. Do NOT take ownership of the buffer, only borrow it
|
|
// for the duration of the call. Return false on failure.
|
|
bool init(const uint8_t* buffer, uint32_t size);
|
|
|
|
// Save the given `protobuf::Node` message in this `HeapSnapshot` as a
|
|
// `DeserializedNode`.
|
|
bool saveNode(const protobuf::Node& node);
|
|
|
|
// If present, a timestamp in the same units that `PR_Now` gives.
|
|
Maybe<uint64_t> timestamp;
|
|
|
|
// The id of the root node for this deserialized heap graph.
|
|
NodeId rootId;
|
|
|
|
// The set of nodes in this deserialized heap graph, keyed by id.
|
|
using NodeMap = js::HashMap<NodeId, UniquePtr<DeserializedNode>>;
|
|
NodeMap nodes;
|
|
|
|
// Core dump files have many duplicate strings: type names are repeated for
|
|
// each node, and although in theory edge names are highly customizable for
|
|
// specific edges, in practice they are also highly duplicated. Rather than
|
|
// make each Deserialized{Node,Edge} malloc their own copy of their edge and
|
|
// type names, we de-duplicate the strings here and Deserialized{Node,Edge}
|
|
// get borrowed pointers into this set.
|
|
using UniqueStringSet = js::HashSet<UniqueString, UniqueStringHashPolicy>;
|
|
UniqueStringSet strings;
|
|
|
|
protected:
|
|
nsCOMPtr<nsISupports> mParent;
|
|
|
|
virtual ~HeapSnapshot() { }
|
|
|
|
public:
|
|
// Create a `HeapSnapshot` from the given buffer that contains a serialized
|
|
// core dump. Do NOT take ownership of the buffer, only borrow it for the
|
|
// duration of the call.
|
|
static already_AddRefed<HeapSnapshot> Create(JSContext* cx,
|
|
dom::GlobalObject& global,
|
|
const uint8_t* buffer,
|
|
uint32_t size,
|
|
ErrorResult& rv);
|
|
|
|
NS_DECL_CYCLE_COLLECTING_ISUPPORTS
|
|
NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS(HeapSnapshot)
|
|
MOZ_DECLARE_REFCOUNTED_TYPENAME(HeapSnapshot)
|
|
|
|
nsISupports* GetParentObject() const { return mParent; }
|
|
|
|
virtual JSObject* WrapObject(JSContext* aCx,
|
|
JS::Handle<JSObject*> aGivenProto) override;
|
|
|
|
const char16_t* borrowUniqueString(const char16_t* duplicateString,
|
|
size_t length);
|
|
};
|
|
|
|
} // namespace devtools
|
|
} // namespace mozilla
|
|
|
|
#endif // mozilla_devtools_HeapSnapshot__
|