Handle pointer arithmetics with function operands

fuhsnn · Dec 28, 2024 · 2a9c631 · 2a9c631
1 parent ada543f
commit 2a9c631
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/parse.c b/parse.c
@@ -3383,6 +3383,11 @@ static Node *new_add(Node *lhs, Node *rhs, Token *tok) {
   if (is_numeric(lhs->ty) && is_numeric(rhs->ty))
     return new_binary(ND_ADD, lhs, rhs, tok);
 
+  if (lhs->ty->kind == TY_FUNC)
+    lhs = new_cast(lhs, pointer_to(lhs->ty));
+  if (rhs->ty->kind == TY_FUNC)
+    rhs = new_cast(rhs, pointer_to(rhs->ty));
+
   Node **ofs = is_integer(lhs->ty) ? &lhs : is_integer(rhs->ty) ? &rhs : NULL;
   Node *ptr = lhs->ty->base ? lhs : rhs->ty->base ? rhs : NULL;
 
@@ -3407,6 +3412,11 @@ static Node *new_sub(Node *lhs, Node *rhs, Token *tok) {
   if (is_numeric(lhs->ty) && is_numeric(rhs->ty))
     return new_binary(ND_SUB, lhs, rhs, tok);
 
+  if (lhs->ty->kind == TY_FUNC)
+    lhs = new_cast(lhs, pointer_to(lhs->ty));
+  if (rhs->ty->kind == TY_FUNC)
+    rhs = new_cast(rhs, pointer_to(rhs->ty));
+
   // ptr - num
   if (lhs->ty->base && is_integer(rhs->ty)) {
     if (lhs->ty->base->kind == TY_VLA)