| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109 |
- "use strict";
-
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.default = void 0;
-
- var _constants = require("../constants");
-
- var _errors = require("../errors");
-
- var _toJSON = _interopRequireDefault(require("../toJSON"));
-
- var _Collection = _interopRequireDefault(require("./Collection"));
-
- var _Node = _interopRequireDefault(require("./Node"));
-
- var _Pair = _interopRequireDefault(require("./Pair"));
-
- function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
- function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
-
- const getAliasCount = (node, anchors) => {
- if (node instanceof Alias) {
- const anchor = anchors.find(a => a.node === node.source);
- return anchor.count * anchor.aliasCount;
- } else if (node instanceof _Collection.default) {
- let count = 0;
-
- for (const item of node.items) {
- const c = getAliasCount(item, anchors);
- if (c > count) count = c;
- }
-
- return count;
- } else if (node instanceof _Pair.default) {
- const kc = getAliasCount(node.key, anchors);
- const vc = getAliasCount(node.value, anchors);
- return Math.max(kc, vc);
- }
-
- return 1;
- };
-
- class Alias extends _Node.default {
- static stringify({
- range,
- source
- }, {
- anchors,
- doc,
- implicitKey,
- inStringifyKey
- }) {
- let anchor = Object.keys(anchors).find(a => anchors[a] === source);
- if (!anchor && inStringifyKey) anchor = doc.anchors.getName(source) || doc.anchors.newName();
- if (anchor) return `*${anchor}${implicitKey ? ' ' : ''}`;
- const msg = doc.anchors.getName(source) ? 'Alias node must be after source node' : 'Source node not found for alias node';
- throw new Error(`${msg} [${range}]`);
- }
-
- constructor(source) {
- super();
- this.source = source;
- this.type = _constants.Type.ALIAS;
- }
-
- set tag(t) {
- throw new Error('Alias nodes cannot have tags');
- }
-
- toJSON(arg, ctx) {
- if (!ctx) return (0, _toJSON.default)(this.source, arg, ctx);
- const {
- anchors,
- maxAliasCount
- } = ctx;
- const anchor = anchors.find(a => a.node === this.source);
-
- if (!anchor || anchor.res === undefined) {
- const msg = 'This should not happen: Alias anchor was not resolved?';
- if (this.cstNode) throw new _errors.YAMLReferenceError(this.cstNode, msg);else throw new ReferenceError(msg);
- }
-
- if (maxAliasCount >= 0) {
- anchor.count += 1;
- if (anchor.aliasCount === 0) anchor.aliasCount = getAliasCount(this.source, anchors);
-
- if (anchor.count * anchor.aliasCount > maxAliasCount) {
- const msg = 'Excessive alias count indicates a resource exhaustion attack';
- if (this.cstNode) throw new _errors.YAMLReferenceError(this.cstNode, msg);else throw new ReferenceError(msg);
- }
- }
-
- return anchor.res;
- } // Only called when stringifying an alias mapping key while constructing
- // Object output.
-
-
- toString(ctx) {
- return Alias.stringify(this, ctx);
- }
-
- }
-
- exports.default = Alias;
-
- _defineProperty(Alias, "default", true);
|