ConstraintsOptionsExtensions.cs 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. using System.Threading;
  2. using Jint.Constraints;
  3. // ReSharper disable once CheckNamespace
  4. namespace Jint;
  5. public static class ConstraintsOptionsExtensions
  6. {
  7. /// <summary>
  8. /// Limits the allowed statement count that can be run as part of the program.
  9. /// </summary>
  10. public static Options MaxStatements(this Options options, int maxStatements = 0)
  11. {
  12. options.WithoutConstraint(x => x is MaxStatementsConstraint);
  13. if (maxStatements > 0 && maxStatements < int.MaxValue)
  14. {
  15. options.Constraint(new MaxStatementsConstraint(maxStatements));
  16. }
  17. return options;
  18. }
  19. public static Options LimitMemory(this Options options, long memoryLimit)
  20. {
  21. options.WithoutConstraint(x => x is MemoryLimitConstraint);
  22. if (memoryLimit > 0 && memoryLimit < int.MaxValue)
  23. {
  24. options.Constraint(new MemoryLimitConstraint(memoryLimit));
  25. }
  26. return options;
  27. }
  28. public static Options TimeoutInterval(this Options options, TimeSpan timeoutInterval)
  29. {
  30. if (timeoutInterval > TimeSpan.Zero && timeoutInterval < TimeSpan.MaxValue)
  31. {
  32. options.Constraint(new TimeConstraint(timeoutInterval));
  33. }
  34. return options;
  35. }
  36. public static Options CancellationToken(this Options options, CancellationToken cancellationToken)
  37. {
  38. options.WithoutConstraint(x => x is CancellationConstraint);
  39. if (cancellationToken != default)
  40. {
  41. options.Constraint(new CancellationConstraint(cancellationToken));
  42. }
  43. return options;
  44. }
  45. }