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) 


503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 503

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) 


525
526
527
528
529
530
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 525

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 



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
360
361
362
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 334

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 



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

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
175
176
177
 
# 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
    if stmt.expression
      _collect_heap_release_names(stmt.expression, result)
    end
    _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 



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
307
308
309
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 280

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 



452
453
454
455
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 452

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 



484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 484

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 



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

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_in_expr(stmt.expression, result) if stmt.expression
    _collect_released_names(stmt.body, result) if stmt.body.is_a?(Array)
  end
end

#_expr_contains_transfer?(expr, name) ⇒ Boolean

Returns:

  • (Boolean) 


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

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) 


624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 624

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 ────────────────────────────────────────────



446
447
448
449
450
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 446

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

#build_owning_type_set ⇒ Object 



740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 740

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 ─────────────────────────────



313
314
315
316
317
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 313

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 



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
606
607
608
609
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 581

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 ──────────────────────────────────



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
566
567
568
569
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 534

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 



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

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) 


652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 652

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 



715
716
717
718
719
720
721
722
723
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 715

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 



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

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 



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

def extract_line(stmt)
  stmt.line
end

#find_decl_node_id(graph, name) ⇒ Object 



571
572
573
574
575
576
577
578
579
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 571

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 



675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 675

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) 


196
197
198
199
200
201
202
203
204
205
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 196

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 



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

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) 


248
249
250
251
252
253
254
255
256
257
258
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 248

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) 


260
261
262
263
264
265
266
267
268
269
270
271
272
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 260

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) 


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

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) 


412
413
414
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 412

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

#owning_base_name(qname) ⇒ Object 



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

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

#owning_local_bindings(stmts) ⇒ Object 



274
275
276
277
278
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 274

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

#owning_type_by_name?(type_name) ⇒ Boolean

Returns:

  • (Boolean) 


734
735
736
737
738
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 734

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 



401
402
403
404
405
406
407
408
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 401

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) 


706
707
708
709
710
711
712
713
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 706

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) 


611
612
613
614
615
616
617
618
619
620
621
622
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 611

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

  # DeferStmt: inline form `defer x.release()` or block form `defer: …`
  if stmt.is_a?(AST::DeferStmt)
    return true if stmt.expression && _expr_has_release?(stmt.expression, name)
    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) 


697
698
699
700
701
702
703
704
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 697

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 ────────────────────────────────────────────────



366
367
368
369
370
371
372
373
374
375
376
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 366

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) 


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

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) 


416
417
418
419
420
421
422
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 416

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) 


424
425
426
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 424

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) 


647
648
649
650
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 647

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

#type_base_name(type) ⇒ Object 



758
759
760
761
762
 
# File 'lib/milk_tea/tooling/linter/release_rules.rb', line 758

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