pdfjs-dist/lib/test/unit/evaluator_spec.js

/**
 * @licstart The following is the entire license notice for the
 * JavaScript code in this page
 *
 * Copyright 2022 Mozilla Foundation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * @licend The above is the entire license notice for the
 * JavaScript code in this page
 */
"use strict";

var _test_utils = require("./test_utils.js");

var _primitives = require("../../core/primitives.js");

var _util = require("../../shared/util.js");

var _stream = require("../../core/stream.js");

var _operator_list = require("../../core/operator_list.js");

var _evaluator = require("../../core/evaluator.js");

var _worker = require("../../core/worker.js");

describe("evaluator", function () {
  function HandlerMock() {
    this.inputs = [];
  }

  HandlerMock.prototype = {
    send(name, data) {
      this.inputs.push({
        name,
        data
      });
    }

  };

  function ResourcesMock() {}

  ResourcesMock.prototype = {
    get(name) {
      return this[name];
    }

  };

  async function runOperatorListCheck(evaluator, stream, resources) {
    const operatorList = new _operator_list.OperatorList();
    const task = new _worker.WorkerTask("OperatorListCheck");
    await evaluator.getOperatorList({
      stream,
      task,
      resources,
      operatorList
    });
    return operatorList;
  }

  let partialEvaluator;
  beforeAll(function () {
    partialEvaluator = new _evaluator.PartialEvaluator({
      xref: new _test_utils.XRefMock(),
      handler: new HandlerMock(),
      pageIndex: 0,
      idFactory: (0, _test_utils.createIdFactory)(0)
    });
  });
  afterAll(function () {
    partialEvaluator = null;
  });
  describe("splitCombinedOperations", function () {
    it("should reject unknown operations", async function () {
      const stream = new _stream.StringStream("fTT");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(1);
      expect(result.fnArray[0]).toEqual(_util.OPS.fill);
      expect(result.argsArray[0]).toEqual(null);
    });
    it("should handle one operation", async function () {
      const stream = new _stream.StringStream("Q");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(1);
      expect(result.fnArray[0]).toEqual(_util.OPS.restore);
    });
    it("should handle two glued operations", async function () {
      const imgDict = new _primitives.Dict();
      imgDict.set("Subtype", _primitives.Name.get("Image"));
      imgDict.set("Width", 1);
      imgDict.set("Height", 1);
      const imgStream = new _stream.Stream([0]);
      imgStream.dict = imgDict;
      const xObject = new _primitives.Dict();
      xObject.set("Res1", imgStream);
      const resources = new ResourcesMock();
      resources.XObject = xObject;
      const stream = new _stream.StringStream("/Res1 DoQ");
      const result = await runOperatorListCheck(partialEvaluator, stream, resources);
      expect(result.fnArray.length).toEqual(3);
      expect(result.fnArray[0]).toEqual(_util.OPS.dependency);
      expect(result.fnArray[1]).toEqual(_util.OPS.paintImageXObject);
      expect(result.fnArray[2]).toEqual(_util.OPS.restore);
      expect(result.argsArray.length).toEqual(3);
      expect(result.argsArray[0]).toEqual(["img_p0_1"]);
      expect(result.argsArray[1]).toEqual(["img_p0_1", 1, 1]);
      expect(result.argsArray[2]).toEqual(null);
    });
    it("should handle three glued operations", async function () {
      const stream = new _stream.StringStream("fff");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(3);
      expect(result.fnArray[0]).toEqual(_util.OPS.fill);
      expect(result.fnArray[1]).toEqual(_util.OPS.fill);
      expect(result.fnArray[2]).toEqual(_util.OPS.fill);
    });
    it("should handle three glued operations #2", async function () {
      const resources = new ResourcesMock();
      resources.Res1 = {};
      const stream = new _stream.StringStream("B*Bf*");
      const result = await runOperatorListCheck(partialEvaluator, stream, resources);
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(3);
      expect(result.fnArray[0]).toEqual(_util.OPS.eoFillStroke);
      expect(result.fnArray[1]).toEqual(_util.OPS.fillStroke);
      expect(result.fnArray[2]).toEqual(_util.OPS.eoFill);
    });
    it("should handle glued operations and operands", async function () {
      const stream = new _stream.StringStream("f5 Ts");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(2);
      expect(result.fnArray[0]).toEqual(_util.OPS.fill);
      expect(result.fnArray[1]).toEqual(_util.OPS.setTextRise);
      expect(result.argsArray.length).toEqual(2);
      expect(result.argsArray[1].length).toEqual(1);
      expect(result.argsArray[1][0]).toEqual(5);
    });
    it("should handle glued operations and literals", async function () {
      const stream = new _stream.StringStream("trueifalserinulln");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(3);
      expect(result.fnArray[0]).toEqual(_util.OPS.setFlatness);
      expect(result.fnArray[1]).toEqual(_util.OPS.setRenderingIntent);
      expect(result.fnArray[2]).toEqual(_util.OPS.endPath);
      expect(result.argsArray.length).toEqual(3);
      expect(result.argsArray[0].length).toEqual(1);
      expect(result.argsArray[0][0]).toEqual(true);
      expect(result.argsArray[1].length).toEqual(1);
      expect(result.argsArray[1][0]).toEqual(false);
      expect(result.argsArray[2]).toEqual(null);
    });
  });
  describe("validateNumberOfArgs", function () {
    it("should execute if correct number of arguments", async function () {
      const stream = new _stream.StringStream("5 1 d0");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(result.argsArray[0][0]).toEqual(5);
      expect(result.argsArray[0][1]).toEqual(1);
      expect(result.fnArray[0]).toEqual(_util.OPS.setCharWidth);
    });
    it("should execute if too many arguments", async function () {
      const stream = new _stream.StringStream("5 1 4 d0");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(result.argsArray[0][0]).toEqual(1);
      expect(result.argsArray[0][1]).toEqual(4);
      expect(result.fnArray[0]).toEqual(_util.OPS.setCharWidth);
    });
    it("should execute if nested commands", async function () {
      const gState = new _primitives.Dict();
      gState.set("LW", 2);
      gState.set("CA", 0.5);
      const extGState = new _primitives.Dict();
      extGState.set("GS2", gState);
      const resources = new ResourcesMock();
      resources.ExtGState = extGState;
      const stream = new _stream.StringStream("/F2 /GS2 gs 5.711 Tf");
      const result = await runOperatorListCheck(partialEvaluator, stream, resources);
      expect(result.fnArray.length).toEqual(3);
      expect(result.fnArray[0]).toEqual(_util.OPS.setGState);
      expect(result.fnArray[1]).toEqual(_util.OPS.dependency);
      expect(result.fnArray[2]).toEqual(_util.OPS.setFont);
      expect(result.argsArray.length).toEqual(3);
      expect(result.argsArray[0]).toEqual([[["LW", 2], ["CA", 0.5]]]);
      expect(result.argsArray[1]).toEqual(["g_font_error"]);
      expect(result.argsArray[2]).toEqual(["g_font_error", 5.711]);
    });
    it("should skip if too few arguments", async function () {
      const stream = new _stream.StringStream("5 d0");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(result.argsArray).toEqual([]);
      expect(result.fnArray).toEqual([]);
    });
    it("should error if (many) path operators have too few arguments " + "(bug 1443140)", async function () {
      const NUM_INVALID_OPS = 25;
      const invalidMoveText = "10 Td\n".repeat(NUM_INVALID_OPS);
      const moveTextStream = new _stream.StringStream(invalidMoveText);
      const result = await runOperatorListCheck(partialEvaluator, moveTextStream, new ResourcesMock());
      expect(result.argsArray).toEqual([]);
      expect(result.fnArray).toEqual([]);
      const invalidLineTo = "20 l\n".repeat(NUM_INVALID_OPS);
      const lineToStream = new _stream.StringStream(invalidLineTo);

      try {
        await runOperatorListCheck(partialEvaluator, lineToStream, new ResourcesMock());
        expect(false).toEqual(true);
      } catch (reason) {
        expect(reason instanceof _util.FormatError).toEqual(true);
        expect(reason.message).toEqual("Invalid command l: expected 2 args, but received 1 args.");
      }
    });
    it("should close opened saves", async function () {
      const stream = new _stream.StringStream("qq");
      const result = await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
      expect(!!result.fnArray && !!result.argsArray).toEqual(true);
      expect(result.fnArray.length).toEqual(4);
      expect(result.fnArray[0]).toEqual(_util.OPS.save);
      expect(result.fnArray[1]).toEqual(_util.OPS.save);
      expect(result.fnArray[2]).toEqual(_util.OPS.restore);
      expect(result.fnArray[3]).toEqual(_util.OPS.restore);
    });
    it("should error on paintXObject if name is missing", async function () {
      const stream = new _stream.StringStream("/ Do");

      try {
        await runOperatorListCheck(partialEvaluator, stream, new ResourcesMock());
        expect(false).toEqual(true);
      } catch (reason) {
        expect(reason instanceof _util.FormatError).toEqual(true);
        expect(reason.message).toEqual("XObject should be a stream");
      }
    });
    it("should skip paintXObject if subtype is PS", async function () {
      const xobjStreamDict = new _primitives.Dict();
      xobjStreamDict.set("Subtype", _primitives.Name.get("PS"));
      const xobjStream = new _stream.Stream([], 0, 0, xobjStreamDict);
      const xobjs = new _primitives.Dict();
      xobjs.set("Res1", xobjStream);
      const resources = new _primitives.Dict();
      resources.set("XObject", xobjs);
      const stream = new _stream.StringStream("/Res1 Do");
      const result = await runOperatorListCheck(partialEvaluator, stream, resources);
      expect(result.argsArray).toEqual([]);
      expect(result.fnArray).toEqual([]);
    });
  });
  describe("thread control", function () {
    it("should abort operator list parsing", async function () {
      const stream = new _stream.StringStream("qqQQ");
      const resources = new ResourcesMock();
      const result = new _operator_list.OperatorList();
      const task = new _worker.WorkerTask("OperatorListAbort");
      task.terminate();

      try {
        await partialEvaluator.getOperatorList({
          stream,
          task,
          resources,
          operatorList: result
        });
        expect(false).toEqual(true);
      } catch (_) {
        expect(!!result.fnArray && !!result.argsArray).toEqual(true);
        expect(result.fnArray.length).toEqual(0);
      }
    });
    it("should abort text content parsing", async function () {
      const resources = new ResourcesMock();
      const stream = new _stream.StringStream("qqQQ");
      const task = new _worker.WorkerTask("TextContentAbort");
      task.terminate();

      try {
        await partialEvaluator.getTextContent({
          stream,
          task,
          resources
        });
        expect(false).toEqual(true);
      } catch (_) {
        expect(true).toEqual(true);
      }
    });
  });
  describe("operator list", function () {
    class StreamSinkMock {
      enqueue() {}

    }

    it("should get correct total length after flushing", function () {
      const operatorList = new _operator_list.OperatorList(null, new StreamSinkMock());
      operatorList.addOp(_util.OPS.save, null);
      operatorList.addOp(_util.OPS.restore, null);
      expect(operatorList.totalLength).toEqual(2);
      expect(operatorList.length).toEqual(2);
      operatorList.flush();
      expect(operatorList.totalLength).toEqual(2);
      expect(operatorList.length).toEqual(0);
    });
  });
});