fix: prevent parser crash on invalid colon prefix in expressions

ArgonLang · Jul 26, 2024 · e9e66b5 · e9e66b5
1 parent 9c1dde4
commit e9e66b5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/argon/lang/parser2/parser2.cpp b/argon/lang/parser2/parser2.cpp
@@ -975,9 +975,11 @@ Node *Parser::ParseFuncNameParam(Context *context, bool parse_pexpr) {
 
     bool identifier = false;
 
-    if (parse_pexpr)
+    if (parse_pexpr) {
         id = this->ParseExpression(context, PeekPrecedence(TokenType::COMMA));
-    else if (this->CheckIDExt()) {
+        if (!id)
+            throw ParserException(TKCUR_LOC, kStandardError[0]);
+    } else if (this->CheckIDExt()) {
         id = Parser::ParseIdentifierSimple(&this->tkcur_);
         identifier = true;
 
@@ -2637,7 +2639,7 @@ Node *Parser::ParsePrefix(Context *context) {
     this->Eat(true);
 
     auto *right = this->ParseExpression(context, PeekPrecedence(TokenType::ASTERISK));
-    if(right == nullptr)
+    if (right == nullptr)
         throw ParserException(this->tkcur_.loc, kStandardError[0]);
 
     auto *unary = NewNode<Unary>(type_ast_prefix_, false, NodeType::PREFIX);