mirror of
https://github.com/vlang/v.git
synced 2025-09-13 14:32:26 +03:00
470 lines
14 KiB
V
470 lines
14 KiB
V
// Copyright (c) 2019-2024 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, mut pref_ pref.Preferences, ast_files []&ast.File) {
|
|
mut all_fns, all_consts, all_globals := all_fn_const_and_global(ast_files)
|
|
util.timing_start(@METHOD)
|
|
defer {
|
|
util.timing_measure(@METHOD)
|
|
}
|
|
// Functions that must be generated and can't be skipped
|
|
mut all_fn_root_names := if pref_.backend == .native {
|
|
// Note: this is temporary, until the native backend supports more features!
|
|
['main.main']
|
|
} else {
|
|
byteptr_idx_str := '${ast.byteptr_type_idx}'
|
|
charptr_idx_str := '${ast.charptr_type_idx}'
|
|
u8_idx_str := '${ast.u8_type_idx}'
|
|
string_idx_str := '${ast.string_type_idx}'
|
|
array_idx_str := '${ast.array_type_idx}'
|
|
map_idx_str := '${ast.map_type_idx}'
|
|
ref_array_idx_str := '${int(ast.array_type.ref())}'
|
|
[
|
|
'main.main',
|
|
'__new_array',
|
|
'str_intp',
|
|
'format_sb',
|
|
'__new_array_with_default',
|
|
'__new_array_with_multi_default',
|
|
'__new_array_with_array_default',
|
|
'init_global_allocator', // needed for linux_bare and wasm_bare
|
|
'v_realloc', // needed for _STR
|
|
'malloc',
|
|
'malloc_noscan',
|
|
'vcalloc',
|
|
'vcalloc_noscan',
|
|
'new_array_from_c_array',
|
|
'v_fixed_index',
|
|
'memdup',
|
|
'memdup_uncollectable',
|
|
'vstrlen',
|
|
'__as_cast',
|
|
'tos',
|
|
'tos2',
|
|
'tos3',
|
|
'isnil',
|
|
'_option_ok',
|
|
'_result_ok',
|
|
'error',
|
|
// 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
|
|
byteptr_idx_str + '.vstring',
|
|
byteptr_idx_str + '.vstring_with_len',
|
|
byteptr_idx_str + '.vstring_literal',
|
|
charptr_idx_str + '.vstring',
|
|
charptr_idx_str + '.vstring_with_len',
|
|
charptr_idx_str + '.vstring_literal',
|
|
// byte. methods
|
|
u8_idx_str + '.str_escaped',
|
|
// string. methods
|
|
string_idx_str + '.add',
|
|
string_idx_str + '.trim_space',
|
|
string_idx_str + '.repeat',
|
|
string_idx_str + '.replace',
|
|
string_idx_str + '.clone',
|
|
string_idx_str + '.clone_static',
|
|
string_idx_str + '.trim',
|
|
string_idx_str + '.substr',
|
|
string_idx_str + '.substr_ni',
|
|
string_idx_str + '.substr_with_check',
|
|
string_idx_str + '.at',
|
|
string_idx_str + '.at_with_check',
|
|
string_idx_str + '.index_kmp',
|
|
// string. ==, !=, etc...
|
|
string_idx_str + '.eq',
|
|
string_idx_str + '.ne',
|
|
string_idx_str + '.lt',
|
|
string_idx_str + '.gt',
|
|
string_idx_str + '.le',
|
|
string_idx_str + '.ge',
|
|
'fast_string_eq',
|
|
// other array methods
|
|
array_idx_str + '.get',
|
|
array_idx_str + '.set',
|
|
array_idx_str + '.get_unsafe',
|
|
array_idx_str + '.set_unsafe',
|
|
array_idx_str + '.get_with_check', // used for `x := a[i] or {}`
|
|
array_idx_str + '.clone_static_to_depth',
|
|
array_idx_str + '.clone_to_depth',
|
|
array_idx_str + '.first',
|
|
array_idx_str + '.last',
|
|
array_idx_str + '.pointers', // TODO: handle generic methods calling array primitives more precisely in pool_test.v
|
|
array_idx_str + '.reverse',
|
|
array_idx_str + '.repeat_to_depth',
|
|
array_idx_str + '.slice',
|
|
array_idx_str + '.slice_ni',
|
|
// map methods
|
|
map_idx_str + '.get',
|
|
map_idx_str + '.set',
|
|
// reference array methods
|
|
ref_array_idx_str + '.last',
|
|
ref_array_idx_str + '.pop',
|
|
ref_array_idx_str + '.push',
|
|
ref_array_idx_str + '.insert_many',
|
|
ref_array_idx_str + '.prepend_many',
|
|
ref_array_idx_str + '.reverse',
|
|
ref_array_idx_str + '.set',
|
|
ref_array_idx_str + '.set_unsafe',
|
|
// TODO: process the _vinit const initializations automatically too
|
|
'json.decode_string',
|
|
'json.decode_int',
|
|
'json.decode_bool',
|
|
'json.decode_u64',
|
|
'json.encode_int',
|
|
'json.encode_string',
|
|
'json.encode_bool',
|
|
'json.encode_u64',
|
|
'json.json_print',
|
|
'json.json_parse',
|
|
'main.nasserts',
|
|
'main.vtest_init',
|
|
'main.vtest_new_metainfo',
|
|
'main.vtest_new_filemetainfo',
|
|
'os.getwd',
|
|
'v.embed_file.find_index_entry_by_path',
|
|
]
|
|
}
|
|
|
|
if pref_.is_bare {
|
|
all_fn_root_names << [
|
|
'strlen',
|
|
'memcmp',
|
|
'memcpy',
|
|
'realloc',
|
|
'vsnprintf',
|
|
'vsprintf',
|
|
]
|
|
}
|
|
|
|
is_noscan_whitelisted := pref_.gc_mode in [.boehm_full_opt, .boehm_incr_opt]
|
|
|
|
for k, mut mfn in all_fns {
|
|
$if trace_skip_unused_all_fns ? {
|
|
println('k: ${k} | mfn: ${mfn.name}')
|
|
}
|
|
// _noscan functions/methods are selected when the `-gc boehm` is on:
|
|
if is_noscan_whitelisted && mfn.name.ends_with('_noscan') {
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
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 pref_.is_prof {
|
|
if k.starts_with('time.vpc_now') || k.starts_with('v.profile.') {
|
|
// needed for -profile
|
|
all_fn_root_names << k
|
|
continue
|
|
}
|
|
}
|
|
|
|
if k.ends_with('before_request') {
|
|
// TODO: add a more specific check for the .before_request() method in vweb apps
|
|
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
|
|
}
|
|
if mfn.receiver.typ != ast.void_type && mfn.generic_names.len > 0 {
|
|
// generic methods may be used in cgen after specialisation :-|
|
|
// TODO: move generic method specialisation from cgen to before markused
|
|
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_option_not_set'
|
|
all_fn_root_names << 'panic_result_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_sym('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
|
|
}
|
|
if isym.info !is ast.Interface {
|
|
// Do not remove this check, isym.info could be &IError.
|
|
continue
|
|
}
|
|
interface_info := isym.info as ast.Interface
|
|
if interface_info.methods.len == 0 {
|
|
continue
|
|
}
|
|
for itype in interface_info.types {
|
|
ptype := itype.set_nr_muls(1)
|
|
ntype := itype.set_nr_muls(0)
|
|
interface_types := [ptype, ntype]
|
|
for method in interface_info.methods {
|
|
for typ in interface_types {
|
|
interface_implementation_method_name := '${int(typ)}.${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(mut table, mut all_fn_root_names, 'vweb.Result', 'vweb.filter', 'vweb.Context')
|
|
handle_vweb(mut table, mut all_fn_root_names, 'x.vweb.Result', 'x.vweb.filter', 'x.vweb.Context')
|
|
|
|
// handle ORM drivers:
|
|
orm_connection_implementations := table.iface_types['orm.Connection'] or { []ast.Type{} }
|
|
if orm_connection_implementations.len > 0 {
|
|
for k, _ in all_fns {
|
|
if k.starts_with('orm.') {
|
|
all_fn_root_names << k
|
|
}
|
|
}
|
|
for orm_type in orm_connection_implementations {
|
|
all_fn_root_names << '${int(orm_type)}.select'
|
|
all_fn_root_names << '${int(orm_type)}.insert'
|
|
all_fn_root_names << '${int(orm_type)}.update'
|
|
all_fn_root_names << '${int(orm_type)}.delete'
|
|
all_fn_root_names << '${int(orm_type)}.create'
|
|
all_fn_root_names << '${int(orm_type)}.drop'
|
|
all_fn_root_names << '${int(orm_type)}.last_id'
|
|
}
|
|
}
|
|
|
|
mut walker := Walker{
|
|
table: table
|
|
files: ast_files
|
|
all_fns: all_fns
|
|
all_consts: all_consts
|
|
all_globals: all_globals
|
|
pref: pref_
|
|
}
|
|
// println( all_fns.keys() )
|
|
walker.mark_markused_fns() // tagged with `@[markused]`
|
|
walker.mark_markused_consts() // tagged with `@[markused]`
|
|
walker.mark_markused_globals() // tagged with `@[markused]`
|
|
walker.mark_exported_fns()
|
|
walker.mark_root_fns(all_fn_root_names)
|
|
|
|
if walker.n_asserts > 0 {
|
|
unsafe { 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)
|
|
}
|
|
if pref_.gc_mode in [.boehm_full_opt, .boehm_incr_opt] {
|
|
if k in ['new_map_noscan_key', 'new_map_noscan_value', 'new_map_noscan_key_value',
|
|
'new_map_init_noscan_key', 'new_map_init_noscan_value',
|
|
'new_map_init_noscan_key_value'] {
|
|
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}')
|
|
}
|
|
}
|
|
|
|
for kcon, con in all_consts {
|
|
if pref_.is_shared && con.is_pub {
|
|
walker.mark_const_as_used(kcon)
|
|
}
|
|
if !pref_.is_shared && con.is_pub && con.name.starts_with('main.') {
|
|
walker.mark_const_as_used(kcon)
|
|
}
|
|
}
|
|
|
|
table.used_fns = walker.used_fns.move()
|
|
table.used_consts = walker.used_consts.move()
|
|
table.used_globals = walker.used_globals.move()
|
|
|
|
$if trace_skip_unused ? {
|
|
eprintln('>> t.used_fns: ${table.used_fns.keys()}')
|
|
eprintln('>> t.used_consts: ${table.used_consts.keys()}')
|
|
eprintln('>> t.used_globals: ${table.used_globals.keys()}')
|
|
eprintln('>> walker.table.used_maps: ${walker.table.used_maps}')
|
|
}
|
|
}
|
|
|
|
fn all_fn_const_and_global(ast_files []&ast.File) (map[string]ast.FnDecl, map[string]ast.ConstField, map[string]ast.GlobalField) {
|
|
util.timing_start(@METHOD)
|
|
defer {
|
|
util.timing_measure(@METHOD)
|
|
}
|
|
mut all_fns := map[string]ast.FnDecl{}
|
|
mut all_consts := map[string]ast.ConstField{}
|
|
mut all_globals := map[string]ast.GlobalField{}
|
|
for i in 0 .. ast_files.len {
|
|
file := ast_files[i]
|
|
for node in file.stmts {
|
|
match node {
|
|
ast.FnDecl {
|
|
fkey := node.fkey()
|
|
all_fns[fkey] = node
|
|
}
|
|
ast.ConstDecl {
|
|
for cfield in node.fields {
|
|
ckey := cfield.name
|
|
all_consts[ckey] = cfield
|
|
}
|
|
}
|
|
ast.GlobalDecl {
|
|
for gfield in node.fields {
|
|
gkey := gfield.name
|
|
all_globals[gkey] = gfield
|
|
}
|
|
}
|
|
else {}
|
|
}
|
|
}
|
|
}
|
|
return all_fns, all_consts, all_globals
|
|
}
|
|
|
|
fn handle_vweb(mut table ast.Table, mut all_fn_root_names []string, result_name string, filter_name string,
|
|
context_name string) {
|
|
// handle vweb magic router methods:
|
|
result_type_idx := table.find_type_idx(result_name)
|
|
if result_type_idx != 0 {
|
|
all_fn_root_names << filter_name
|
|
typ_vweb_context := ast.Type(table.find_type_idx(context_name)).set_nr_muls(1)
|
|
all_fn_root_names << '${int(typ_vweb_context)}.html'
|
|
for vgt in table.used_vweb_types {
|
|
sym_app := table.sym(vgt)
|
|
for m in sym_app.methods {
|
|
mut skip := true
|
|
if m.name == 'before_request' {
|
|
// TODO: handle expansion of method calls in generic functions in a more universal way
|
|
skip = false
|
|
}
|
|
if m.return_type == result_type_idx {
|
|
skip = false
|
|
}
|
|
if skip {
|
|
continue
|
|
}
|
|
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 << '${int(pvgt)}.${m.name}'
|
|
}
|
|
}
|
|
}
|
|
}
|