349 lines
8.4 KiB
V
349 lines
8.4 KiB
V
// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
|
|
// Use of this source code is governed by an MIT license that can be found in the LICENSE file.
|
|
module markused
|
|
|
|
import v.ast
|
|
import v.util
|
|
import v.pref
|
|
|
|
// mark_used walks the AST, starting at main() and marks all used fns transitively
|
|
pub fn mark_used(mut table ast.Table, pref &pref.Preferences, ast_files []&ast.File) {
|
|
mut all_fns, all_consts := all_fn_and_const(ast_files)
|
|
util.timing_start(@METHOD)
|
|
defer {
|
|
util.timing_measure(@METHOD)
|
|
}
|
|
mut all_fn_root_names := [
|
|
'main.main',
|
|
'__new_array',
|
|
'str_intp',
|
|
'format_sb',
|
|
'__new_array_with_default',
|
|
'__new_array_with_array_default',
|
|
'v_realloc' /* needed for _STR */,
|
|
'v_malloc' /* needed for _STR */,
|
|
'new_array_from_c_array',
|
|
'v_fixed_index',
|
|
'memdup',
|
|
'vstrlen',
|
|
'__as_cast',
|
|
'tos',
|
|
'tos2',
|
|
'tos3',
|
|
'isnil',
|
|
'opt_ok',
|
|
'error',
|
|
'__print_assert_failure',
|
|
// utf8_str_visible_length is used by c/str.v
|
|
'utf8_str_visible_length',
|
|
'compare_ints',
|
|
'compare_u64s',
|
|
'compare_strings',
|
|
'compare_ints_reverse',
|
|
'compare_u64s_reverse',
|
|
'compare_strings_reverse',
|
|
'builtin_init',
|
|
// byteptr and charptr
|
|
'3.vstring',
|
|
'3.vstring_with_len',
|
|
'4.vstring',
|
|
'4.vstring_with_len',
|
|
// byte. methods
|
|
'9.str_escaped',
|
|
// string. methods
|
|
'18.add',
|
|
'18.trim_space',
|
|
'18.replace',
|
|
'18.clone',
|
|
'18.clone_static',
|
|
'18.trim',
|
|
'18.substr',
|
|
'18.at',
|
|
'18.index_kmp',
|
|
// string. ==, !=, etc...
|
|
'18.eq',
|
|
'18.ne',
|
|
'18.lt',
|
|
'18.gt',
|
|
'18.le',
|
|
'18.ge',
|
|
'fast_string_eq',
|
|
// ustring. ==, !=, etc...
|
|
'19.eq',
|
|
'19.ne',
|
|
'19.lt',
|
|
'19.gt',
|
|
'19.le',
|
|
'19.ge',
|
|
'19.add',
|
|
// other array methods
|
|
'21.get',
|
|
'21.set',
|
|
'21.get_unsafe',
|
|
'21.set_unsafe',
|
|
'21.get_with_check' /* used for `x := a[i] or {}` */,
|
|
'21.clone_static',
|
|
'21.first',
|
|
'21.last',
|
|
'21.pointers' /* TODO: handle generic methods calling array primitives more precisely in pool_test.v */,
|
|
'21.reverse',
|
|
'21.repeat',
|
|
'21.slice',
|
|
'21.slice2',
|
|
'59.get',
|
|
'59.set',
|
|
'65557.last',
|
|
'65557.pop',
|
|
'65557.push',
|
|
'65557.insert_many',
|
|
'65557.prepend_many',
|
|
'65557.reverse',
|
|
'65557.set',
|
|
'65557.set_unsafe',
|
|
// TODO: process the _vinit const initializations automatically too
|
|
'json__decode_string',
|
|
'os.getwd',
|
|
'os.init_os_args',
|
|
'os.init_os_args_wide',
|
|
]
|
|
|
|
if pref.is_bare {
|
|
all_fn_root_names << [
|
|
'strlen',
|
|
'memcmp',
|
|
'memcpy',
|
|
'realloc',
|
|
'vsnprintf',
|
|
'vsprintf',
|
|
]
|
|
}
|
|
|
|
if pref.gc_mode in [.boehm_full_opt, .boehm_incr_opt] {
|
|
all_fn_root_names << [
|
|
'__new_array_noscan',
|
|
'__new_array_with_default_noscan',
|
|
'__new_array_with_array_default_noscan',
|
|
'new_array_from_c_array_noscan',
|
|
]
|
|
}
|
|
|
|
for k, mut mfn in all_fns {
|
|
mut method_receiver_typename := ''
|
|
if mfn.is_method {
|
|
method_receiver_typename = table.type_to_str(mfn.receiver.typ)
|
|
}
|
|
if method_receiver_typename == '&wyrand.WyRandRNG' {
|
|
// WyRandRNG is the default rand pseudo random generator
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if method_receiver_typename == '&strings.Builder' {
|
|
// implicit string builders are generated in auto_eq_methods.v
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
// auto generated string interpolation functions, may
|
|
// call .str or .auto_str methods for user types:
|
|
if k.ends_with('.str') || k.ends_with('.auto_str') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if k.ends_with('.init') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if k.ends_with('.free') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
|
|
// sync:
|
|
if k == 'sync.new_channel_st' {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if k == 'sync.channel_select' {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if method_receiver_typename == '&sync.Channel' {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if k.ends_with('.lock') || k.ends_with('.unlock') || k.ends_with('.rlock')
|
|
|| k.ends_with('.runlock') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
// testing framework:
|
|
if pref.is_test {
|
|
if k.starts_with('test_') || k.contains('.test_') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if k.starts_with('testsuite_') || k.contains('.testsuite_') {
|
|
// eprintln('>>> test suite: $k')
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
}
|
|
// public/exported functions can not be skipped,
|
|
// especially when producing a shared library:
|
|
if mfn.is_pub && pref.is_shared {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if mfn.name in ['+', '-', '*', '%', '/', '<', '=='] {
|
|
// TODO: mark the used operators in the checker
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
if pref.prealloc && k.starts_with('prealloc_') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
}
|
|
|
|
// handle assertions and testing framework callbacks:
|
|
if pref.is_debug {
|
|
all_fn_root_names << 'panic_debug'
|
|
}
|
|
all_fn_root_names << 'panic_optional_not_set'
|
|
if pref.is_test {
|
|
all_fn_root_names << 'main.cb_assertion_ok'
|
|
all_fn_root_names << 'main.cb_assertion_failed'
|
|
if benched_tests_sym := table.find_type('main.BenchedTests') {
|
|
bts_type := benched_tests_sym.methods[0].params[0].typ
|
|
all_fn_root_names << '${bts_type}.testing_step_start'
|
|
all_fn_root_names << '${bts_type}.testing_step_end'
|
|
all_fn_root_names << '${bts_type}.end_testing'
|
|
all_fn_root_names << 'main.start_testing'
|
|
}
|
|
}
|
|
|
|
// handle interface implementation methods:
|
|
for isym in table.type_symbols {
|
|
if isym.kind != .interface_ {
|
|
continue
|
|
}
|
|
interface_info := isym.info as ast.Interface
|
|
if interface_info.methods.len == 0 {
|
|
continue
|
|
}
|
|
for itype in interface_info.types {
|
|
pitype := itype.set_nr_muls(1)
|
|
for method in interface_info.methods {
|
|
interface_implementation_method_name := '${pitype}.$method.name'
|
|
$if trace_skip_unused_interface_methods ? {
|
|
eprintln('>> isym.name: $isym.name | interface_implementation_method_name: $interface_implementation_method_name')
|
|
}
|
|
all_fn_root_names << interface_implementation_method_name
|
|
}
|
|
}
|
|
}
|
|
|
|
// handle vweb magic router methods:
|
|
typ_vweb_result := table.find_type_idx('vweb.Result')
|
|
if typ_vweb_result != 0 {
|
|
for vgt in table.used_vweb_types {
|
|
sym_app := table.get_type_symbol(vgt)
|
|
for m in sym_app.methods {
|
|
if m.return_type == typ_vweb_result {
|
|
pvgt := vgt.set_nr_muls(1)
|
|
// eprintln('vgt: $vgt | pvgt: $pvgt | sym_app.name: $sym_app.name | m.name: $m.name')
|
|
all_fn_root_names << '${pvgt}.$m.name'
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
|
|
mut walker := Walker{
|
|
table: table
|
|
files: ast_files
|
|
all_fns: all_fns
|
|
all_consts: all_consts
|
|
}
|
|
// println( all_fns.keys() )
|
|
walker.mark_exported_fns()
|
|
walker.mark_root_fns(all_fn_root_names)
|
|
|
|
if walker.n_asserts > 0 {
|
|
walker.fn_decl(mut all_fns['__print_assert_failure'])
|
|
}
|
|
if table.used_maps > 0 {
|
|
for k, mut mfn in all_fns {
|
|
mut method_receiver_typename := ''
|
|
if mfn.is_method {
|
|
method_receiver_typename = table.type_to_str(mfn.receiver.typ)
|
|
}
|
|
if k in ['new_map', 'new_map_init', 'map_hash_string']
|
|
|| method_receiver_typename == '&map' || method_receiver_typename == '&DenseArray'
|
|
|| k.starts_with('map_') {
|
|
walker.fn_decl(mut mfn)
|
|
}
|
|
}
|
|
} else {
|
|
for map_fn_name in ['new_map', 'new_map_init', 'map_hash_string', 'new_dense_array'] {
|
|
walker.used_fns.delete(map_fn_name)
|
|
}
|
|
for k, mut mfn in all_fns {
|
|
if !mfn.is_method {
|
|
continue
|
|
}
|
|
method_receiver_typename := table.type_to_str(mfn.receiver.typ)
|
|
if method_receiver_typename in ['&map', '&mapnode', '&SortedMap', '&DenseArray'] {
|
|
walker.used_fns.delete(k)
|
|
}
|
|
}
|
|
}
|
|
|
|
$if trace_skip_unused_fn_names ? {
|
|
for key, _ in walker.used_fns {
|
|
println('> used fn key: $key')
|
|
}
|
|
}
|
|
|
|
table.used_fns = walker.used_fns.move()
|
|
table.used_consts = walker.used_consts.move()
|
|
|
|
$if trace_skip_unused ? {
|
|
eprintln('>> t.used_fns: $table.used_fns.keys()')
|
|
eprintln('>> t.used_consts: $table.used_consts.keys()')
|
|
eprintln('>> walker.table.used_maps: $walker.table.used_maps')
|
|
}
|
|
}
|
|
|
|
fn all_fn_and_const(ast_files []&ast.File) (map[string]ast.FnDecl, map[string]ast.ConstField) {
|
|
util.timing_start(@METHOD)
|
|
defer {
|
|
util.timing_measure(@METHOD)
|
|
}
|
|
mut all_fns := map[string]ast.FnDecl{}
|
|
mut all_consts := map[string]ast.ConstField{}
|
|
for i in 0 .. ast_files.len {
|
|
file := ast_files[i]
|
|
for node in file.stmts {
|
|
match node {
|
|
ast.FnDecl {
|
|
fkey := if node.is_method {
|
|
'${int(node.receiver.typ)}.$node.name'
|
|
} else {
|
|
node.name
|
|
}
|
|
all_fns[fkey] = node
|
|
}
|
|
ast.ConstDecl {
|
|
for cfield in node.fields {
|
|
ckey := cfield.name
|
|
all_consts[ckey] = cfield
|
|
}
|
|
}
|
|
else {}
|
|
}
|
|
}
|
|
}
|
|
return all_fns, all_consts
|
|
}
|