fune/dom/html/HTMLTableElement.h
Kris Maglione 4275cd1039 Bug 1406278: Part 1 - Pass subject principal to SetAttribute and friends. r=bz
In order to tailor certain security checks to the caller that is attempting to
load a particular piece of content, we need to be able to attach an
appropriate triggering principal to the corresponding requests. Since most
HTML content is loaded based on attribute values, that means capturing the
subject principal of the caller who sets those attributes, which means making
it available to AfterSetAttr hooks.

MozReview-Commit-ID: BMDL2Uepg0X

--HG--
extra : rebase_source : 25e438c243700a9368c393e40e3a6002d968d6c8
2017-10-09 14:33:38 -07:00

249 lines
7.5 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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_dom_HTMLTableElement_h
#define mozilla_dom_HTMLTableElement_h
#include "mozilla/Attributes.h"
#include "nsGenericHTMLElement.h"
#include "mozilla/dom/HTMLTableCaptionElement.h"
#include "mozilla/dom/HTMLTableSectionElement.h"
namespace mozilla {
namespace dom {
class TableRowsCollection;
class HTMLTableElement final : public nsGenericHTMLElement
{
public:
explicit HTMLTableElement(already_AddRefed<mozilla::dom::NodeInfo>& aNodeInfo);
NS_IMPL_FROMCONTENT_HTML_WITH_TAG(HTMLTableElement, table)
// nsISupports
NS_DECL_ISUPPORTS_INHERITED
HTMLTableCaptionElement* GetCaption() const
{
return static_cast<HTMLTableCaptionElement*>(GetChild(nsGkAtoms::caption));
}
void SetCaption(HTMLTableCaptionElement* aCaption, ErrorResult& aError)
{
DeleteCaption();
if (aCaption) {
nsCOMPtr<nsINode> firstChild = nsINode::GetFirstChild();
nsINode::InsertBefore(*aCaption, firstChild, aError);
}
}
void DeleteTFoot();
already_AddRefed<nsGenericHTMLElement> CreateCaption();
void DeleteCaption();
HTMLTableSectionElement* GetTHead() const
{
return static_cast<HTMLTableSectionElement*>(GetChild(nsGkAtoms::thead));
}
void SetTHead(HTMLTableSectionElement* aTHead, ErrorResult& aError)
{
if (aTHead && !aTHead->IsHTMLElement(nsGkAtoms::thead)) {
aError.Throw(NS_ERROR_DOM_HIERARCHY_REQUEST_ERR);
return;
}
DeleteTHead();
if (aTHead) {
nsCOMPtr<nsIContent> refNode = nullptr;
for (refNode = nsINode::GetFirstChild();
refNode;
refNode = refNode->GetNextSibling()) {
if (refNode->IsHTMLElement() &&
!refNode->IsHTMLElement(nsGkAtoms::caption) &&
!refNode->IsHTMLElement(nsGkAtoms::colgroup)) {
break;
}
}
nsINode::InsertBefore(*aTHead, refNode, aError);
}
}
already_AddRefed<nsGenericHTMLElement> CreateTHead();
void DeleteTHead();
HTMLTableSectionElement* GetTFoot() const
{
return static_cast<HTMLTableSectionElement*>(GetChild(nsGkAtoms::tfoot));
}
void SetTFoot(HTMLTableSectionElement* aTFoot, ErrorResult& aError)
{
if (aTFoot && !aTFoot->IsHTMLElement(nsGkAtoms::tfoot)) {
aError.Throw(NS_ERROR_DOM_HIERARCHY_REQUEST_ERR);
return;
}
DeleteTFoot();
if (aTFoot) {
nsINode::AppendChild(*aTFoot, aError);
}
}
already_AddRefed<nsGenericHTMLElement> CreateTFoot();
nsIHTMLCollection* TBodies();
already_AddRefed<nsGenericHTMLElement> CreateTBody();
nsIHTMLCollection* Rows();
already_AddRefed<nsGenericHTMLElement> InsertRow(int32_t aIndex,
ErrorResult& aError);
void DeleteRow(int32_t aIndex, ErrorResult& aError);
void GetAlign(DOMString& aAlign)
{
GetHTMLAttr(nsGkAtoms::align, aAlign);
}
void SetAlign(const nsAString& aAlign, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::align, aAlign, aError);
}
void GetBorder(DOMString& aBorder)
{
GetHTMLAttr(nsGkAtoms::border, aBorder);
}
void SetBorder(const nsAString& aBorder, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::border, aBorder, aError);
}
void GetFrame(DOMString& aFrame)
{
GetHTMLAttr(nsGkAtoms::frame, aFrame);
}
void SetFrame(const nsAString& aFrame, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::frame, aFrame, aError);
}
void GetRules(DOMString& aRules)
{
GetHTMLAttr(nsGkAtoms::rules, aRules);
}
void SetRules(const nsAString& aRules, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::rules, aRules, aError);
}
void GetSummary(nsString& aSummary)
{
GetHTMLAttr(nsGkAtoms::summary, aSummary);
}
void GetSummary(DOMString& aSummary)
{
GetHTMLAttr(nsGkAtoms::summary, aSummary);
}
void SetSummary(const nsAString& aSummary, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::summary, aSummary, aError);
}
void GetWidth(DOMString& aWidth)
{
GetHTMLAttr(nsGkAtoms::width, aWidth);
}
void SetWidth(const nsAString& aWidth, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::width, aWidth, aError);
}
void GetBgColor(DOMString& aBgColor)
{
GetHTMLAttr(nsGkAtoms::bgcolor, aBgColor);
}
void SetBgColor(const nsAString& aBgColor, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::bgcolor, aBgColor, aError);
}
void GetCellPadding(DOMString& aCellPadding)
{
GetHTMLAttr(nsGkAtoms::cellpadding, aCellPadding);
}
void SetCellPadding(const nsAString& aCellPadding, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::cellpadding, aCellPadding, aError);
}
void GetCellSpacing(DOMString& aCellSpacing)
{
GetHTMLAttr(nsGkAtoms::cellspacing, aCellSpacing);
}
void SetCellSpacing(const nsAString& aCellSpacing, ErrorResult& aError)
{
SetHTMLAttr(nsGkAtoms::cellspacing, aCellSpacing, aError);
}
virtual bool ParseAttribute(int32_t aNamespaceID,
nsAtom* aAttribute,
const nsAString& aValue,
nsAttrValue& aResult) override;
virtual nsMapRuleToAttributesFunc GetAttributeMappingFunction() const override;
NS_IMETHOD_(bool) IsAttributeMapped(const nsAtom* aAttribute) const override;
virtual nsresult Clone(mozilla::dom::NodeInfo *aNodeInfo, nsINode **aResult,
bool aPreallocateChildren) const override;
virtual nsresult BindToTree(nsIDocument* aDocument, nsIContent* aParent,
nsIContent* aBindingParent,
bool aCompileEventHandlers) override;
virtual void UnbindFromTree(bool aDeep = true,
bool aNullParent = true) override;
/**
* Called when an attribute is about to be changed
*/
virtual nsresult BeforeSetAttr(int32_t aNameSpaceID, nsAtom* aName,
const nsAttrValueOrString* aValue,
bool aNotify) override;
/**
* Called when an attribute has just been changed
*/
virtual nsresult AfterSetAttr(int32_t aNameSpaceID, nsAtom* aName,
const nsAttrValue* aValue,
const nsAttrValue* aOldValue,
nsIPrincipal* aSubjectPrincipal,
bool aNotify) override;
NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(HTMLTableElement,
nsGenericHTMLElement)
nsMappedAttributes* GetAttributesMappedForCell();
protected:
virtual ~HTMLTableElement();
virtual JSObject* WrapNode(JSContext *aCx, JS::Handle<JSObject*> aGivenProto) override;
nsIContent* GetChild(nsAtom *aTag) const
{
for (nsIContent* cur = nsINode::GetFirstChild(); cur;
cur = cur->GetNextSibling()) {
if (cur->IsHTMLElement(aTag)) {
return cur;
}
}
return nullptr;
}
RefPtr<nsContentList> mTBodies;
RefPtr<TableRowsCollection> mRows;
nsMappedAttributes *mTableInheritedAttributes;
void BuildInheritedAttributes();
void ReleaseInheritedAttributes();
private:
static void MapAttributesIntoRule(const nsMappedAttributes* aAttributes,
GenericSpecifiedValues* aGenericData);
};
} // namespace dom
} // namespace mozilla
#endif /* mozilla_dom_HTMLTableElement_h */