Module: MilkTea::Linter::LinterReleaseRules

Included in:
MilkTea::Linter
Defined in:
lib/milk_tea/tooling/linter/release_rules.rb

Instance Method Summary collapse

Instance Method Details

#_any_stmt?(stmts, &pred) ⇒ Boolean

Returns:

  • (Boolean) 


499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 499

def _any_stmt?(stmts, &pred)
  return false if stmts.nil?

  stmts.any? do |stmt|
    if pred.call(stmt)
      true
    else
      case stmt
      when AST::IfStmt
        stmt.branches.any? { |b| _any_stmt?(b.body, &pred) } ||
          _any_stmt?(stmt.else_body, &pred)
      when AST::WhileStmt then _any_stmt?(stmt.body, &pred)
      when AST::ForStmt then _any_stmt?(stmt.body, &pred)
      when AST::MatchStmt then stmt.arms.any? { |arm| _any_stmt?(arm.body, &pred) }
      when AST::UnsafeStmt then _any_stmt?(stmt.body, &pred)
      when AST::DeferStmt then stmt.body.is_a?(Array) && _any_stmt?(stmt.body, &pred)
      else false
      end
    end
  end
end

#_any_stmt_or_expr?(stmts, &pred) ⇒ Boolean

Returns:

  • (Boolean) 


521
522
523
524
525
526
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 521

def _any_stmt_or_expr?(stmts, &pred)
  _any_stmt?(stmts) do |stmt|
    expr = extract_expr(stmt)
    expr && pred.call(expr)
  end
end

#_check_double_release_in_seq(stmts, result) ⇒ Object 



331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 331

def _check_double_release_in_seq(stmts, result)
  return if stmts.nil?

  scope_releases = Hash.new { |hash, key| hash[key] = [] }

  stmts.each do |stmt|
    expr = extract_expr(stmt)
    if expr && release_call_on_binding?(expr)
      name = expr.callee.receiver.name
      scope_releases[name] << extract_line(stmt)
    elsif expr && heap_release_call_on_name?(expr)
      name = heap_release_target_name(expr)
      scope_releases[name] << extract_line(stmt) if name
    end

    scope_releases.delete(stmt.target.name) if reassigns_owning_binding?(stmt)

    case stmt
    when AST::UnsafeStmt
      _check_double_release_in_seq(stmt.body, result)
    when AST::DeferStmt
      _check_double_release_in_seq(stmt.body, result) if stmt.body.is_a?(Array)
    end
  end

  scope_releases.each do |name, lines|
    result[name] = lines if lines.length > 1
  end
end

#_collect_heap_release_names(expr, result) ⇒ Object 



176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 176

def _collect_heap_release_names(expr, result)
  return if expr.nil?
  if heap_release_call_on_name?(expr)
    result << heap_release_target_name(expr)
  end
  case expr
  when AST::Call
    expr.arguments.each { |a| _collect_heap_release_names(a.value, result) }
    _collect_heap_release_names(expr.callee, result) if expr.callee.is_a?(AST::Call)
  when AST::BinaryOp
    _collect_heap_release_names(expr.left, result)
    _collect_heap_release_names(expr.right, result)
  when AST::UnaryOp
    _collect_heap_release_names(expr.operand, result)
  end
end

#_collect_own_released_names(stmts, result) ⇒ Object 



145
146
147
148
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 145

def _collect_own_released_names(stmts, result)
  return if stmts.nil?
  stmts.each { |s| _collect_own_released_names_in_stmt(s, result) }
end

#_collect_own_released_names_in_stmt(stmt, result) ⇒ Object 



150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 150

def _collect_own_released_names_in_stmt(stmt, result)
  case stmt
  when AST::ExpressionStmt
    _collect_heap_release_names(stmt.expression, result)
  when AST::ReturnStmt
    _collect_heap_release_names(stmt.value, result) if stmt.value
  when AST::LocalDecl
    _collect_heap_release_names(stmt.value, result) if stmt.value
  when AST::Assignment
    _collect_heap_release_names(stmt.value, result)
  when AST::IfStmt
    stmt.branches.each { |b| _collect_own_released_names(b.body, result) }
    _collect_own_released_names(stmt.else_body, result) if stmt.else_body
  when AST::WhileStmt
    _collect_own_released_names(stmt.body, result)
  when AST::ForStmt
    _collect_own_released_names(stmt.body, result)
  when AST::MatchStmt
    stmt.arms.each { |arm| _collect_own_released_names(arm.body, result) }
  when AST::UnsafeStmt
    _collect_own_released_names(stmt.body, result)
  when AST::DeferStmt
    _collect_own_released_names(stmt.body, result) if stmt.body.is_a?(Array)
  end
end

#_collect_own_typed_locals(stmts, result) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 74

def _collect_own_typed_locals(stmts, result)
  return if stmts.nil?

  stmts.each do |stmt|
    case stmt
    when AST::LocalDecl
      type_name = resolve_local_own_pointer_type(stmt)
      if type_name
        result[stmt.name] = {
          line: stmt.line,
          column: stmt.column,
          type_name: type_name,
        }
      end
    when AST::IfStmt
      stmt.branches.each { |b| _collect_own_typed_locals(b.body, result) }
      _collect_own_typed_locals(stmt.else_body, result) if stmt.else_body
    when AST::WhileStmt
      _collect_own_typed_locals(stmt.body, result)
    when AST::ForStmt
      _collect_own_typed_locals(stmt.body, result)
    when AST::MatchStmt
      stmt.arms.each { |arm| _collect_own_typed_locals(arm.body, result) }
    when AST::UnsafeStmt
      _collect_own_typed_locals(stmt.body, result)
    when AST::DeferStmt
      _collect_own_typed_locals(stmt.body, result) if stmt.body.is_a?(Array)
    end
  end
end

#_collect_owning_locals(stmts, result) ⇒ Object 



277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 277

def _collect_owning_locals(stmts, result)
  return if stmts.nil?

  stmts.each do |stmt|
    case stmt
    when AST::LocalDecl
      type_name = resolve_local_owning_type(stmt)
      if type_name
        result[stmt.name] = {
          line: stmt.line,
          column: stmt.column,
          type_name: type_name,
        }
      end
    when AST::IfStmt
      stmt.branches.each { |b| _collect_owning_locals(b.body, result) }
      _collect_owning_locals(stmt.else_body, result) if stmt.else_body
    when AST::WhileStmt
      _collect_owning_locals(stmt.body, result)
    when AST::ForStmt
      _collect_owning_locals(stmt.body, result)
    when AST::MatchStmt
      stmt.arms.each { |arm| _collect_owning_locals(arm.body, result) }
    when AST::UnsafeStmt
      _collect_owning_locals(stmt.body, result)
    when AST::DeferStmt
      _collect_owning_locals(stmt.body, result) if stmt.body.is_a?(Array)
    end
  end
end

#_collect_released_names(stmts, result) ⇒ Object 



449
450
451
452
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 449

def _collect_released_names(stmts, result)
  return if stmts.nil?
  stmts.each { |s| _collect_released_names_in_stmt(s, result) }
end

#_collect_released_names_in_expr(expr, result) ⇒ Object 



480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 480

def _collect_released_names_in_expr(expr, result)
  return if expr.nil?

  result << expr.callee.receiver.name if release_call_on_binding?(expr)

  case expr
  when AST::Call
    expr.arguments.each { |a| _collect_released_names_in_expr(a.value, result) }
    _collect_released_names_in_expr(expr.callee, result)
  when AST::MemberAccess
    _collect_released_names_in_expr(expr.receiver, result)
  when AST::BinaryOp
    _collect_released_names_in_expr(expr.left, result)
    _collect_released_names_in_expr(expr.right, result)
  when AST::UnaryOp
    _collect_released_names_in_expr(expr.operand, result)
  end
end

#_collect_released_names_in_stmt(stmt, result) ⇒ Object 



454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 454

def _collect_released_names_in_stmt(stmt, result)
  case stmt
  when AST::ExpressionStmt
    _collect_released_names_in_expr(stmt.expression, result)
  when AST::ReturnStmt
    _collect_released_names_in_expr(stmt.value, result) if stmt.value
  when AST::LocalDecl
    _collect_released_names_in_expr(stmt.value, result) if stmt.value
  when AST::Assignment
    _collect_released_names_in_expr(stmt.value, result)
  when AST::IfStmt
    stmt.branches.each { |b| _collect_released_names(b.body, result) }
    _collect_released_names(stmt.else_body, result) if stmt.else_body
  when AST::WhileStmt
    _collect_released_names(stmt.body, result)
  when AST::ForStmt
    _collect_released_names(stmt.body, result)
  when AST::MatchStmt
    stmt.arms.each { |arm| _collect_released_names(arm.body, result) }
  when AST::UnsafeStmt
    _collect_released_names(stmt.body, result)
  when AST::DeferStmt
    _collect_released_names(stmt.body, result) if stmt.body.is_a?(Array)
  end
end

#_expr_contains_transfer?(expr, name) ⇒ Boolean

Returns:

  • (Boolean) 


425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 425

def _expr_contains_transfer?(expr, name)
  return false unless expr.is_a?(AST::Call)

  expr.arguments.any? do |arg|
    next false unless arg.is_a?(AST::Argument) && arg.name

    if arg.value.is_a?(AST::Identifier) && arg.value.name == name
      true
    elsif arg.value.is_a?(AST::Call)
      _expr_contains_transfer?(arg.value, name)
    else
      false
    end
  end
end

#_expr_has_release?(expr, name) ⇒ Boolean

Returns:

  • (Boolean) 


619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 619

def _expr_has_release?(expr, name)
  return false unless expr

  return true if release_call_on_binding?(expr) && expr.callee.receiver.name == name

  case expr
  when AST::Call
    expr.arguments.any? { |a| _expr_has_release?(a.value, name) } ||
      _expr_has_release?(expr.callee, name)
  when AST::MemberAccess then _expr_has_release?(expr.receiver, name)
  when AST::BinaryOp
    _expr_has_release?(expr.left, name) || _expr_has_release?(expr.right, name)
  when AST::UnaryOp then _expr_has_release?(expr.operand, name)
  else false
  end
end

#all_own_released_names(stmts) ⇒ Object

── own release detection ──────────────────────────────────────



139
140
141
142
143
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 139

def all_own_released_names(stmts)
  result = Set.new
  _collect_own_released_names(stmts, result)
  result
end

#all_released_names(stmts) ⇒ Object

── Shared AST walkers ────────────────────────────────────────────



443
444
445
446
447
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 443

def all_released_names(stmts)
  result = Set.new
  _collect_released_names(stmts, result)
  result
end

#build_owning_type_set ⇒ Object 



735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 735

def build_owning_type_set
  set = Set.new
  return set unless @sema_facts

  all_methods = [@sema_facts.methods]
  (@imported_modules || {}).each_value do |mod_binding|
    all_methods << mod_binding.methods if mod_binding.respond_to?(:methods)
  end

  all_methods.compact.each do |methods_hash|
    methods_hash.each do |type, methods|
      next unless methods.is_a?(Hash) && methods.key?("release")
      set << type_base_name(type)
    end
  end
  set
end

#check_own_release_leaks(stmts) ⇒ Object

Check only own typed leaks (separate from struct-type leaks)



45
46
47
48
49
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 45

def check_own_release_leaks(stmts)
  own_locals = own_typed_local_bindings(stmts)
  own_released = all_own_released_names(stmts)
  own_locals.reject { |name, _info| own_released.include?(name) || own_transferred?(stmts, name) }
end

#check_owning_release_double(stmts) ⇒ Object

── Simple sequential double-release ─────────────────────────────



310
311
312
313
314
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 310

def check_owning_release_double(stmts)
  result = {}
  _check_double_release_in_seq(stmts, result)
  result
end

#check_owning_release_leaks(stmts) ⇒ Object

── Simple leak detection ──────────────────────────────────────────



38
39
40
41
42
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 38

def check_owning_release_leaks(stmts)
  locals = owning_local_bindings(stmts)
  released = all_released_names(stmts)
  locals.reject { |name, _info| released.include?(name) || ownership_transferred?(stmts, name) }
end

#classify_cfg_nodes(graph, reachable_ids, name) ⇒ Object 



577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 577

def classify_cfg_nodes(graph, reachable_ids, name)
  releases = Set.new
  consumers = Set.new
  reassigns = Set.new

  graph.each_node do |node|
    next unless reachable_ids.include?(node.id)
    next unless node.statement

    is_defer = node.statement.is_a?(AST::DeferStmt)

    if release_call_in_stmt?(node.statement, name)
      consumers << node.id
      releases << node.id unless is_defer
    end

    if return_for_name?(node.statement, name) || struct_transfer_in_stmt?(node.statement, name)
      consumers << node.id
    end

    if node.kind == :assignment
      node.writes_info.each do |write|
        reassigns << node.id if write[:origin] == :assignment && write[:name] == name
      end
    end
  end

  [releases, consumers, reassigns]
end

#emit_owning_release_cfg_warnings(cfg_analysis, body) ⇒ Object

── CFG-based precision analysis ──────────────────────────────────



530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 530

def emit_owning_release_cfg_warnings(cfg_analysis, body)
  graph = cfg_analysis.graph
  reachable_ids = cfg_analysis.reachability.reachable_ids
          exit_kinds = %i[exit return break_exit continue_exit].to_set.freeze
  owning_locals = owning_local_bindings(body)
  return if owning_locals.empty?

  owning_locals.each do |name, info|
    decl_id = find_decl_node_id(graph, name)
    next unless decl_id && reachable_ids.include?(decl_id)

    releases, consumers, reassigns = classify_cfg_nodes(graph, reachable_ids, name)

    unless every_path_to_exit_has_consumer?(graph, decl_id, consumers, exit_kinds, reachable_ids)
      emit_owning_release_leak_warnings({ name => info })
    end

    releases.each do |rel_id|
      reachable_release = find_reachable_release(graph, rel_id, releases, reassigns, reachable_ids)
      next unless reachable_release

      node = graph.nodes[reachable_release]
      @warnings << Warning.new(
        path: @path,
        line: node&.line,
        column: nil,
        length: nil,
        code: "owning-release-double",
        message: "owning binding '#{name}' may be released more than once",
        severity: :warning,
        symbol_name: name,
      )
      break
    end
  end
end

#emit_owning_release_double_warnings(double_names_lines) ⇒ Object 



316
317
318
319
320
321
322
323
324
325
326
327
328
329
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 316

def emit_owning_release_double_warnings(double_names_lines)
  double_names_lines.each do |name, lines|
    @warnings << Warning.new(
      path: @path,
      line: lines.last,
      column: nil,
      length: nil,
      code: "owning-release-double",
      message: "owning binding '#{name}' may be released more than once",
      severity: :warning,
      symbol_name: name,
    )
  end
end

#emit_owning_release_leak_warnings(leak_names) ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
64
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 51

def emit_owning_release_leak_warnings(leak_names)
  leak_names.each do |name, info|
    @warnings << Warning.new(
      path: @path,
      line: info[:line],
      column: info[:column],
      length: name.length,
      code: "owning-release-leak",
      message: "owning binding '#{name}' of type '#{info[:type_name]}' is never released",
      severity: :warning,
      symbol_name: name,
    )
  end
end

#emit_owning_release_warnings(function_or_body) ⇒ Object

Entry point — called from visit_function and proc expr in full_tier mode.



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 7

def emit_owning_release_warnings(function_or_body)
  body = function_or_body.is_a?(Array) ? function_or_body : function_or_body.body
  return unless body && !body.empty?

  cfg_available = false
  if @sema_facts
    cfg_available = begin
      analysis = profile_phase("rule.owning_release.cfg") { statement_flow_analysis(body) }
      analysis || false
    rescue StandardError
      false
    end
  end

  if cfg_available
    emit_owning_release_cfg_warnings(cfg_available, body)
  else
    # No CFG — emit struct-type leaks via simple pattern-matching
    leak_names = check_owning_release_leaks(body)
    emit_owning_release_leak_warnings(leak_names)
  end
  # Always run own[T] checks (AST-based, works with or without CFG)
  own_leaks = check_own_release_leaks(body)
  emit_owning_release_leak_warnings(own_leaks)

  double_names_lines = check_owning_release_double(body)
  emit_owning_release_double_warnings(double_names_lines)
end

#every_path_to_exit_has_consumer?(graph, start_id, consumers, exit_kinds, reachable_ids) ⇒ Boolean

Returns:

  • (Boolean) 


647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 647

def every_path_to_exit_has_consumer?(graph, start_id, consumers, exit_kinds, reachable_ids)
  visited = Set.new
  queue = (graph.nodes[start_id]&.succs || []).dup
  until queue.empty?
    nid = queue.shift
    next if visited.include?(nid)
    visited << nid
    next unless reachable_ids.include?(nid)

    node = graph.nodes[nid]
    next unless node

    if consumers.include?(nid)
      nil
    elsif exit_kinds.include?(node.kind)
      return false
    else
      node.succs.each { |s| queue << s }
    end
  end
  true
end

#extract_expr(stmt) ⇒ Object 



710
711
712
713
714
715
716
717
718
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 710

def extract_expr(stmt)
  case stmt
  when AST::ExpressionStmt then stmt.expression
  when AST::ReturnStmt then stmt.value
  when AST::LocalDecl then stmt.value
  when AST::Assignment then stmt.value
  else nil
  end
end

#extract_initializer_type(expr) ⇒ Object 



375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 375

def extract_initializer_type(expr)
  return nil unless expr.is_a?(AST::Call)
  return nil unless expr.arguments.empty?

  callee = expr.callee
  method_name = case callee
                when AST::MemberAccess then callee.member
                when AST::Specialization
                  inner = callee.callee
                  inner.is_a?(AST::MemberAccess) ? inner.member : nil
                else nil
                end
  return nil unless %w[create with_capacity empty from_str].include?(method_name)

  receiver = case callee
             when AST::MemberAccess then callee.receiver
             when AST::Specialization then callee
             else nil
             end

  owning_type_from_receiver(receiver)
end

#extract_line(stmt) ⇒ Object 



720
721
722
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 720

def extract_line(stmt)
  stmt.line
end

#find_decl_node_id(graph, name) ⇒ Object 



567
568
569
570
571
572
573
574
575
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 567

def find_decl_node_id(graph, name)
  graph.each_node do |node|
    next unless node.kind == :local_decl
    node.writes_info.each do |write|
      return node.id if write[:origin] == :declaration && write[:name] == name
    end
  end
  nil
end

#find_reachable_release(graph, start_id, release_ids, reassign_ids, reachable_ids) ⇒ Object 



670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 670

def find_reachable_release(graph, start_id, release_ids, reassign_ids, reachable_ids)
  visited = Set.new
  queue = (graph.nodes[start_id]&.succs || []).dup
  until queue.empty?
    nid = queue.shift
    next if visited.include?(nid)
    visited << nid
    next unless reachable_ids.include?(nid)

    return nid if release_ids.include?(nid) && nid != start_id
    next if reassign_ids.include?(nid)

    node = graph.nodes[nid]
    next unless node

    queue.concat(node.succs)
  end
  nil
end

#heap_release_call_on_name?(expr) ⇒ Boolean

Returns:

  • (Boolean) 


193
194
195
196
197
198
199
200
201
202
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 193

def heap_release_call_on_name?(expr)
  return false unless expr.is_a?(AST::Call)
  return false unless expr.callee.is_a?(AST::MemberAccess)
  return false unless %w[release release_and_null].include?(expr.callee.member)
  return false unless expr.arguments.length >= 1
  arg = expr.arguments.first
  return false unless arg.is_a?(AST::Argument)
  target_name = heap_release_target_name(expr)
  target_name && !target_name.empty?
end

#heap_release_target_name(expr) ⇒ Object 



204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 204

def heap_release_target_name(expr)
  return nil unless expr.is_a?(AST::Call)
  return nil unless expr.arguments.length >= 1
  # For multi-arg release (e.g. tracking.release(tracker, p)), find
  # the argument that names the released value by preferring the last
  # direct-identifier argument.
  released = nil
  expr.arguments.each do |arg|
    next unless arg.is_a?(AST::Argument)
    value = arg.value
    value = value.expression if value.is_a?(AST::UnsafeStmt)
    if value.is_a?(AST::Identifier)
      released = value.name
    elsif value.is_a?(AST::Call) && value.callee.is_a?(AST::Identifier) && value.callee.name == "ref_of"
      ref_arg = value.arguments.first
      if ref_arg.is_a?(AST::Argument) && ref_arg.value.is_a?(AST::Identifier)
        released = ref_arg.value.name
      end
    end
  end
  released
end

#is_heap_alloc_call?(expr) ⇒ Boolean

Returns:

  • (Boolean) 


118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 118

def is_heap_alloc_call?(expr)
  return false unless expr.is_a?(AST::Call)
  callee = expr.callee
  non_null_method_names = %w[must_alloc must_alloc_aligned
                             must_alloc_zeroed must_resize].to_set
  case callee
  when AST::MemberAccess
    callee.receiver.is_a?(AST::Identifier) &&
      non_null_method_names.include?(callee.member)
  when AST::Specialization
    inner = callee.callee
    inner.is_a?(AST::MemberAccess) &&
      inner.receiver.is_a?(AST::Identifier) &&
      non_null_method_names.include?(inner.member)
  else
    false
  end
end

#own_cast_in_unsafe?(expr, name) ⇒ Boolean

Returns:

  • (Boolean) 


245
246
247
248
249
250
251
252
253
254
255
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 245

def own_cast_in_unsafe?(expr, name)
  case expr
  when AST::UnsafeExpr
    case expr.expression
    when AST::PrefixCast
      expr.expression.expression.is_a?(AST::Identifier) && expr.expression.expression.name == name
    end
  else
    false
  end
end

#own_struct_field_transfer?(expr, name) ⇒ Boolean

Returns:

  • (Boolean) 


257
258
259
260
261
262
263
264
265
266
267
268
269
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 257

def own_struct_field_transfer?(expr, name)
  case expr
  when AST::Call
    expr.arguments.any? do |arg|
      next unless arg.is_a?(AST::Argument)

      arg.value.is_a?(AST::Identifier) && arg.value.name == name ||
        own_struct_field_transfer?(arg.value, name)
    end
  else
    false
  end
end

#own_transferred?(stmts, name) ⇒ Boolean

── own ownership transfer ─────────────────────────────────────

Returns:

  • (Boolean) 


229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 229

def own_transferred?(stmts, name)
  _any_stmt?(stmts) do |stmt|
    if stmt.is_a?(AST::ReturnStmt) && stmt.value.is_a?(AST::Identifier) && stmt.value.name == name
      true
    elsif (expr = extract_expr(stmt)) && own_struct_field_transfer?(expr, name)
      true
    elsif stmt.is_a?(AST::Assignment) && stmt.value.is_a?(AST::Identifier) && stmt.value.name == name
      true
    elsif stmt.is_a?(AST::ReturnStmt) && own_cast_in_unsafe?(stmt.value, name)
      true
    else
      false
    end
  end
end

#own_typed_local_bindings(stmts) ⇒ Object

── own typed local detection ──────────────────────────────────



68
69
70
71
72
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 68

def own_typed_local_bindings(stmts)
  result = {}
  _collect_own_typed_locals(stmts, result)
  result
end

#ownership_transferred?(stmts, name) ⇒ Boolean

── Ownership transfer detection ──────────────────────────────────

Returns:

  • (Boolean) 


409
410
411
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 409

def ownership_transferred?(stmts, name)
  returned_name?(stmts, name) || struct_field_transfer?(stmts, name)
end

#owning_base_name(qname) ⇒ Object 



724
725
726
727
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 724

def owning_base_name(qname)
  last_dot = qname.rindex(".")
  last_dot ? qname[(last_dot + 1)..] : qname
end

#owning_local_bindings(stmts) ⇒ Object 



271
272
273
274
275
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 271

def owning_local_bindings(stmts)
  result = {}
  _collect_owning_locals(stmts, result)
  result
end

#owning_type_by_name?(type_name) ⇒ Boolean

Returns:

  • (Boolean) 


729
730
731
732
733
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 729

def owning_type_by_name?(type_name)
  return false unless type_name
  @owning_type_set ||= build_owning_type_set
  @owning_type_set.include?(type_name)
end

#owning_type_from_receiver(receiver) ⇒ Object 



398
399
400
401
402
403
404
405
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 398

def owning_type_from_receiver(receiver)
  case receiver
  when AST::Identifier then owning_base_name(receiver.name)
  when AST::MemberAccess then owning_base_name(receiver.member)
  when AST::Specialization then owning_type_from_receiver(receiver.callee)
  else nil
  end
end

#reassigns_owning_binding?(stmt) ⇒ Boolean

Returns:

  • (Boolean) 


701
702
703
704
705
706
707
708
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 701

def reassigns_owning_binding?(stmt)
  return false unless stmt.is_a?(AST::Assignment)
  return false unless stmt.target.is_a?(AST::Identifier)
  return false unless stmt.operator == "="

  @scopes.reverse_each { |scope| return true if scope.key?(stmt.target.name) }
  false
end

#release_call_in_stmt?(stmt, name) ⇒ Boolean

Returns:

  • (Boolean) 


607
608
609
610
611
612
613
614
615
616
617
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 607

def release_call_in_stmt?(stmt, name)
  return false unless stmt

  # DeferStmt: block form `defer: …` (inline defers are a single-statement body)
  if stmt.is_a?(AST::DeferStmt)
    return stmt.body.is_a?(Array) && stmt.body.any? { |s| release_call_in_stmt?(s, name) }
  end

  expr = extract_expr(stmt)
  expr && _expr_has_release?(expr, name)
end

#release_call_on_binding?(expr) ⇒ Boolean

── Helpers ───────────────────────────────────────────────────────

Returns:

  • (Boolean) 


692
693
694
695
696
697
698
699
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 692

def release_call_on_binding?(expr)
  return false unless expr.is_a?(AST::Call)
  return false unless expr.callee.is_a?(AST::MemberAccess)
  return false unless expr.callee.member == "release"
  return false unless expr.callee.receiver.is_a?(AST::Identifier)
  return false unless expr.arguments.empty?
  true
end

#resolve_local_own_pointer_type(decl) ⇒ Object 



105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 105

def resolve_local_own_pointer_type(decl)
  if is_heap_alloc_call?(decl.value)
    return "own[<inferred>]"
  end
  type = resolve_expr_type(decl.value) if decl.value
  return nil unless type
  return nil unless type.is_a?(MilkTea::Types::GenericInstance)
  return nil unless type.name == "own"
  type.to_s
rescue StandardError
  nil
end

#resolve_local_owning_type(decl) ⇒ Object

── Type resolution ────────────────────────────────────────────────



363
364
365
366
367
368
369
370
371
372
373
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 363

def resolve_local_owning_type(decl)
  if decl.type && decl.type.is_a?(AST::TypeRef)
    base = owning_base_name(decl.type.name.to_s)
    return base if owning_type_by_name?(base)
  end
  if decl.value
    type_name = extract_initializer_type(decl.value)
    return type_name if type_name && owning_type_by_name?(type_name)
  end
  nil
end

#return_for_name?(stmt, name) ⇒ Boolean

Returns:

  • (Boolean) 


636
637
638
639
640
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 636

def return_for_name?(stmt, name)
  stmt.is_a?(AST::ReturnStmt) &&
    stmt.value.is_a?(AST::Identifier) &&
    stmt.value.name == name
end

#returned_name?(stmts, name) ⇒ Boolean

Returns:

  • (Boolean) 


413
414
415
416
417
418
419
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 413

def returned_name?(stmts, name)
  _any_stmt?(stmts) do |stmt|
    stmt.is_a?(AST::ReturnStmt) &&
      stmt.value.is_a?(AST::Identifier) &&
      stmt.value.name == name
  end
end

#struct_field_transfer?(stmts, name) ⇒ Boolean

Returns:

  • (Boolean) 


421
422
423
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 421

def struct_field_transfer?(stmts, name)
  _any_stmt_or_expr?(stmts) { |expr| _expr_contains_transfer?(expr, name) }
end

#struct_transfer_in_stmt?(stmt, name) ⇒ Boolean

Returns:

  • (Boolean) 


642
643
644
645
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 642

def struct_transfer_in_stmt?(stmt, name)
  expr = extract_expr(stmt)
  expr && _expr_contains_transfer?(expr, name)
end

#type_base_name(type) ⇒ Object 



753
754
755
756
757
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 753

def type_base_name(type)
  return nil unless type
  return type.name.to_s if type.respond_to?(:name) && type.name
  type.to_s
end