git.saurik.com Git - apple/javascriptcore.git/blame

Commit	Line	Data
81345200 A	1	/*
	2	* Copyright (C) 2013 Apple Inc. All rights reserved.
	3	*
	4	* Redistribution and use in source and binary forms, with or without
	5	* modification, are permitted provided that the following conditions
	6	* are met:
	7	* 1. Redistributions of source code must retain the above copyright
	8	* notice, this list of conditions and the following disclaimer.
	9	* 2. Redistributions in binary form must reproduce the above copyright
	10	* notice, this list of conditions and the following disclaimer in the
	11	* documentation and/or other materials provided with the distribution.
	12	*
	13	* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
	14	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
	15	* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	16	* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
	17	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	18	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	19	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	20	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	21	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	22	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
	23	* THE POSSIBILITY OF SUCH DAMAGE.
	24	*/
	25
	26	#ifndef WeakMapData_h
	27	#define WeakMapData_h
	28
	29	#include "JSCell.h"
	30	#include "Structure.h"
	31	#include <wtf/HashFunctions.h>
	32	#include <wtf/HashMap.h>
	33	#include <wtf/MathExtras.h>
	34
	35	namespace JSC {
	36
ed1e77d3	37	class WeakMapData final : public JSCell {
81345200 A	38	public:
81345200 A	39	typedef JSCell Base;
ed1e77d3	40	static const unsigned StructureFlags = Base::StructureFlags \| StructureIsImmortal;
81345200 A	41
	42	static WeakMapData* create(VM& vm)
	43	{
	44	WeakMapData* weakMapData = new (NotNull, allocateCell<WeakMapData>(vm.heap)) WeakMapData(vm);
	45	weakMapData->finishCreation(vm);
	46	return weakMapData;
	47	}
	48
	49	static Structure* createStructure(VM& vm, JSGlobalObject* globalObject, JSValue prototype)
	50	{
ed1e77d3	51	return Structure::create(vm, globalObject, prototype, TypeInfo(CellType, StructureFlags), info());
81345200 A	52	}
	53
	54	static const bool needsDestruction = true;
81345200 A	55
	56	void set(VM&, JSObject*, JSValue);
	57	JSValue get(JSObject*);
	58	bool remove(JSObject*);
	59	bool contains(JSObject*);
	60	void clear();
	61
	62	DECLARE_INFO;
	63
ed1e77d3 A	64	typedef HashMap<JSObject*, WriteBarrier<Unknown>> MapType;
	65	MapType::const_iterator begin() const { return m_map.begin(); }
	66	MapType::const_iterator end() const { return m_map.end(); }
	67
	68	int size() const { return m_map.size(); }
81345200 A	69
	70	private:
	71	WeakMapData(VM&);
	72	static void destroy(JSCell*);
	73	static void visitChildren(JSCell*, SlotVisitor&);
	74	void finishCreation(VM&);
	75
	76	class DeadKeyCleaner : public UnconditionalFinalizer, public WeakReferenceHarvester {
	77	public:
	78	DeadKeyCleaner(WeakMapData* target)
	79	: m_target(target)
	80	{
	81	}
	82	private:
	83	virtual void visitWeakReferences(SlotVisitor&) override;
	84	virtual void finalizeUnconditionally() override;
ed1e77d3	85	unsigned m_liveKeyCount;
81345200 A	86	WeakMapData* m_target;
	87	};
	88	DeadKeyCleaner m_deadKeyCleaner;
81345200 A	89	MapType m_map;
	90	};
	91
	92	}
	93
	94	#endif /* !defined(WeakMapData_h) */