mirror of
https://github.com/Instadapp/Swap-Aggregator-Subgraph.git
synced 2024-07-29 21:57:12 +00:00
105 lines
2.9 KiB
JavaScript
105 lines
2.9 KiB
JavaScript
"use strict";
|
|
|
|
exports.Alias = void 0;
|
|
|
|
var _constants = require("../constants");
|
|
|
|
var _errors = require("../errors");
|
|
|
|
var _toJSON = require("../toJSON");
|
|
|
|
var _Collection = require("./Collection");
|
|
|
|
var _Node = require("./Node");
|
|
|
|
var _Pair = require("./Pair");
|
|
|
|
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.get(node.source);
|
|
return anchor.count * anchor.aliasCount;
|
|
} else if (node instanceof _Collection.Collection) {
|
|
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.Pair) {
|
|
const kc = getAliasCount(node.key, anchors);
|
|
const vc = getAliasCount(node.value, anchors);
|
|
return Math.max(kc, vc);
|
|
}
|
|
|
|
return 1;
|
|
};
|
|
|
|
class Alias extends _Node.Node {
|
|
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.toJSON)(this.source, arg, ctx);
|
|
const {
|
|
anchors,
|
|
maxAliasCount
|
|
} = ctx;
|
|
const anchor = anchors.get(this.source);
|
|
/* istanbul ignore if */
|
|
|
|
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.Alias = Alias;
|
|
|
|
_defineProperty(Alias, "default", true); |