mt-lang 0.2.19 → 0.2.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '076938689f867b9d3d7f60eab7f4654c03a45ea69b429f1af17c86a3afef51a5'
-  data.tar.gz: 6f01bf9d070e1307666949acf47d99170d6b28c0048c868a6310663f2d55c919
+  metadata.gz: 98c4dc25b7d1fb3ab4931eed5fd30c29e30448a3b2c3fe85d6b58d823fab7baa
+  data.tar.gz: a025a925117c16e69f85667f4411a5d09b92ac0ea65e6ec061111578db39001a
 SHA512:
-  metadata.gz: 8141047ac65832b64b8eab1d565b5c3eb3bf8841b5541624605179a1e4bc0479f4dab16a92b3b7055e2eb844e37781aac7acab2aa812b348e53d9a151a16fe97
-  data.tar.gz: e5ead28d45b33a6c7fdb5c46f84cdd9a887e6bca593b841687a1d1124ba904f02ee09bdb1ef8dab87f2b8dcfcd6480190e94cea57e8b3b4853148d06d199f6e6
+  metadata.gz: e18ef0444a502981b66eed892a94ccc92918c2d1254669d912e27a8e43b552d360cbc9d4eaf386935b141700ee992eedb365093c00bd67e35e3d878e20f17fbb
+  data.tar.gz: 1e642ded750895e1cabc818bff33c7a88beffc463975d46dc98a2a45cb7eb07d53ee57580fee90fd31b3d8af011e4477a41779044c535513a444457bcd12a902

data/lib/milk_tea/base.rb

@@ -3,7 +3,7 @@
 require "pathname"
 
 module MilkTea
-  VERSION = "0.2.19"
+  VERSION = "0.2.20"
 
   def self.root
     @root ||= Pathname.new(File.expand_path("../..", __dir__))

data/lib/milk_tea/core/semantic_analyzer/expressions.rb

@@ -450,6 +450,21 @@ module MilkTea
           raise_sema_error("safe array indexing requires an addressable array value; bind it to a local first")
         end
 
+        if array_type?(receiver_type) && expression.index.is_a?(AST::IntegerLiteral)
+          length = array_length(receiver_type)
+          index_val = expression.index.value
+          if index_val >= length || index_val < 0
+            raise_sema_error("array index #{index_val} is out of bounds for array[T, #{length}]", expression)
+          end
+        end
+
+        if array_type?(receiver_type) && expression.index.is_a?(AST::UnaryOp) && expression.index.operator == "-" &&
+           expression.index.operand.is_a?(AST::IntegerLiteral)
+          length = array_length(receiver_type)
+          index_val = -expression.index.operand.value
+          raise_sema_error("array index #{index_val} is out of bounds for array[T, #{length}]", expression)
+        end
+
         infer_index_result_type(receiver_type, index_type)
       end
 

data/lib/milk_tea/core/semantic_analyzer/statements.rb

@@ -185,6 +185,7 @@ module MilkTea
 
             validate_consuming_foreign_expression!(statement.value, scopes:, root_allowed: false) if statement.value
             value_type = statement.value ? infer_expression(statement.value, scopes:, expected_type: return_type) : @ctx.types.fetch("void")
+            check_stack_pointer_escape_in_return(statement.value, scopes:) if statement.value
             ensure_assignable!(
               value_type,
               return_type,
@@ -1322,6 +1323,64 @@ module MilkTea
         end
       end
 
+      # Walks the returned expression tree to detect ptr_of/ref_of/const_ptr_of
+      # applied to a stack-local variable whose address must not escape the
+      # function. Only flags when the pointer value itself (not a computed
+      # result through a function call) reaches the return.
+      def check_stack_pointer_escape_in_return(expression, scopes:)
+        return unless expression
+
+        if expression.is_a?(AST::Call)
+          check_stack_escape_call(expression, scopes:)
+        elsif expression.is_a?(AST::BinaryOp)
+          check_stack_pointer_escape_in_return(expression.left, scopes:)
+          check_stack_pointer_escape_in_return(expression.right, scopes:)
+        elsif expression.is_a?(AST::UnaryOp)
+          check_stack_pointer_escape_in_return(expression.operand, scopes:)
+        end
+      end
+
+      def check_stack_escape_call(expression, scopes:)
+        callee = expression.callee
+        return unless callee.is_a?(AST::Identifier)
+        return unless %w[ptr_of ref_of const_ptr_of].include?(callee.name)
+        return if expression.arguments.empty?
+
+        arg = expression.arguments.first
+        source = arg.respond_to?(:value) ? arg.value : arg
+        check_stack_local_source(source, scopes:, builtin: callee.name)
+      end
+
+      def check_stack_local_source(expression, scopes:, builtin:)
+        case expression
+        when AST::Identifier
+          binding = lookup_local_binding(expression.name, scopes)
+          return unless binding
+          return unless %i[let var param].include?(binding.kind)
+          return if stack_safe_pointer_source?(binding)
+          raise_sema_error(
+            "#{builtin}(#{expression.name}) cannot be returned: pointer to function-local storage",
+            expression,
+          )
+        when AST::MemberAccess
+          check_stack_local_source(expression.receiver, scopes:, builtin:)
+        when AST::IndexAccess
+          check_stack_local_source(expression.receiver, scopes:, builtin:)
+        end
+      end
+
+      def lookup_local_binding(name, scopes)
+        scopes.reverse_each do |scope|
+          return scope[name] if scope.key?(name)
+        end
+        nil
+      end
+
+      def stack_safe_pointer_source?(binding)
+        return false unless binding.type
+        ref_type?(binding.type) || own_type?(binding.type)
+      end
+
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mt-lang
 version: !ruby/object:Gem::Version
-  version: 0.2.19
+  version: 0.2.20
 platform: ruby
 authors:
 - Long (Teefan) Tran
@@ -586,7 +586,7 @@ metadata:
   homepage_uri: https://teefan.github.io/mt-lang/
   source_code_uri: https://github.com/teefan/mt-lang
 post_install_message: |
-  Milk Tea 0.2.19 installed!
+  Milk Tea 0.2.20 installed!
 
   System requirements:
     - A C compiler (gcc or clang) must be available on PATH