summaryrefslogtreecommitdiff
path: root/lib/gitlab/untrusted_regexp.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/gitlab/untrusted_regexp.rb')
-rw-r--r--lib/gitlab/untrusted_regexp.rb23
1 files changed, 23 insertions, 0 deletions
diff --git a/lib/gitlab/untrusted_regexp.rb b/lib/gitlab/untrusted_regexp.rb
index ce1cf737663..70d1a7c6535 100644
--- a/lib/gitlab/untrusted_regexp.rb
+++ b/lib/gitlab/untrusted_regexp.rb
@@ -9,6 +9,8 @@ module Gitlab
# there is a strict limit on total execution time. See the RE2 documentation
# at https://github.com/google/re2/wiki/Syntax for more details.
class UntrustedRegexp
+ require_dependency 're2'
+
delegate :===, :source, to: :regexp
def initialize(pattern, multiline: false)
@@ -52,6 +54,27 @@ module Gitlab
Regexp.new(pattern)
end
+ def self.valid?(pattern)
+ self.fabricate(pattern)
+ rescue RegexpError
+ false
+ end
+
+ def self.fabricate(pattern)
+ matches = pattern.match(%r{^/(?<regexp>.+)/(?<flags>[ismU]*)$})
+
+ if matches
+ expression = matches[:regexp]
+ flags = matches[:flags]
+
+ expression.prepend("(?#{flags})") if flags.present?
+
+ self.new(expression, multiline: false)
+ else
+ self.new(pattern, multiline: false)
+ end
+ end
+
private
attr_reader :regexp