Fix

CERT-Polska · Oct 1, 2024 · 3541d91 · 3541d91
1 parent 27f1e5b
commit 3541d91
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/libursa/Query.cpp b/libursa/Query.cpp
@@ -18,7 +18,14 @@ const std::vector<Query> &Query::as_queries() const {
         type != QueryType::MIN_OF) {
         throw std::runtime_error("This query doesn\'t contain subqueries.");
     }
+    return queries;
+}
 
+std::vector<Query> &Query::as_queries() {
+    if (type != QueryType::AND && type != QueryType::OR &&
+        type != QueryType::MIN_OF) {
+        throw std::runtime_error("This query doesn\'t contain subqueries.");
+    }
     return queries;
 }
 

diff --git a/libursa/Query.h b/libursa/Query.h
@@ -51,6 +51,7 @@ class Query {
     Query &operator=(Query &&) = default;
 
     const std::vector<Query> &as_queries() const;
+    std::vector<Query> &as_queries();
     const QString &as_value() const;
     uint32_t as_count() const;
     std::string as_string_repr() const;