v/vlib/builtin/builtin.v

221 lines
4.5 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-06-22 20:20:28 +02:00
module builtin
2019-12-11 19:42:22 +01:00
__global g_m2_buf byteptr
2019-12-11 16:41:25 +01:00
__global g_m2_ptr byteptr
pub fn exit(code int) {
2019-06-23 10:15:30 +02:00
C.exit(code)
2019-06-22 20:20:28 +02:00
}
// isnil returns true if an object is nil (only for C objects).
pub fn isnil(v voidptr) bool {
return v == 0
}
/*
2019-12-19 21:52:45 +01:00
fn on_panic(f fn(int)int) {
2019-06-22 20:20:28 +02:00
// TODO
}
*/
2019-06-22 20:20:28 +02:00
2019-12-19 21:52:45 +01:00
pub fn print_backtrace() {
// at the time of backtrace_symbols_fd call, the C stack would look something like this:
// 1 frame for print_backtrace_skipping_top_frames
// 1 frame for print_backtrace itself
// ... print the rest of the backtrace frames ...
// => top 2 frames should be skipped, since they will not be informative to the developer
print_backtrace_skipping_top_frames(2)
2019-06-22 20:20:28 +02:00
}
2019-07-30 15:08:14 +02:00
// replaces panic when -debug arg is passed
2019-12-19 21:52:45 +01:00
fn panic_debug(line_no int, file, mod, fn_name, s string) {
// NB: the order here is important for a stabler test output
// module is less likely to change than function, etc...
// During edits, the line number will change most frequently,
// so it is last
eprintln('================ V panic ================')
eprintln(' module: $mod')
eprintln(' function: ${fn_name}()')
eprintln(' message: $s')
eprintln(' file: $file')
eprintln(' line: ' + line_no.str())
eprintln('=========================================')
print_backtrace_skipping_top_frames(1)
2019-07-30 15:08:14 +02:00
C.exit(1)
}
2019-06-22 20:20:28 +02:00
pub fn panic(s string) {
eprintln('V panic: $s')
2019-06-22 20:20:28 +02:00
print_backtrace()
C.exit(1)
}
pub fn eprintln(s string) {
// eprintln is used in panics, so it should not fail at all
if s.str == 0 {
eprintln('eprintln(NIL)')
2019-10-04 14:48:09 +02:00
}
2019-11-28 09:46:52 +01:00
$if !windows {
C.fflush(C.stdout)
C.fflush(C.stderr)
C.fprintf(C.stderr, '%.*s\n', s.len, s.str)
C.fflush(C.stderr)
2019-11-25 11:54:56 +01:00
return
}
2019-06-22 20:20:28 +02:00
// TODO issues with stderr and cross compiling for Linux
println(s)
2019-06-22 20:20:28 +02:00
}
2019-12-28 11:02:06 +01:00
pub fn eprint(s string) {
if s.str == 0 {
eprintln('eprint(NIL)')
2019-12-28 11:02:06 +01:00
}
$if !windows {
C.fflush(C.stdout)
C.fflush(C.stderr)
C.fprintf(C.stderr, '%.*s', s.len, s.str)
C.fflush(C.stderr)
2019-12-28 11:02:06 +01:00
return
}
print(s)
}
2019-06-22 20:20:28 +02:00
pub fn print(s string) {
2019-07-24 12:16:45 +02:00
$if windows {
output_handle := C.GetStdHandle(C.STD_OUTPUT_HANDLE)
mut bytes_written := 0
if is_atty(1) > 0 {
wide_str := s.to_wide()
wide_len := C.wcslen(wide_str)
C.WriteConsole(output_handle, wide_str, wide_len, &bytes_written, 0)
2020-05-22 11:18:18 +02:00
unsafe {
free(wide_str)
}
} else {
C.WriteFile(output_handle, s.str, s.len, &bytes_written, 0)
}
2019-07-24 12:16:45 +02:00
} $else {
C.printf('%.*s', s.len, s.str)
}
2019-06-22 20:20:28 +02:00
}
2020-05-22 11:18:18 +02:00
const (
new_line_character = '\n'
)
pub fn println(s string) {
$if windows {
print(s)
print(new_line_character)
} $else {
// TODO: a syscall sys_write on linux works, except for the v repl.
// Probably it is a stdio buffering issue. Needs more testing...
// $if linux {
// $if !android {
// snl := s + '\n'
// C.syscall(/* sys_write */ 1, /* stdout_value */ 1, snl.str, s.len+1)
// return
// }
// }
C.printf('%.*s\n', s.len, s.str)
}
}
2019-12-19 21:52:45 +01:00
__global total_m i64=0
__global nr_mallocs int=0
2019-12-11 16:41:25 +01:00
fn looo(){} // TODO remove, [ pratt
2019-10-20 19:29:24 +02:00
[unsafe_fn]
2019-06-22 20:20:28 +02:00
pub fn malloc(n int) byteptr {
2019-12-16 19:29:32 +01:00
if n <= 0 {
2019-12-16 20:22:04 +01:00
panic('malloc(<=0)')
2019-06-22 20:20:28 +02:00
}
2019-12-14 00:46:55 +01:00
$if prealloc {
2019-12-11 16:41:25 +01:00
res := g_m2_ptr
g_m2_ptr += n
nr_mallocs++
return res
} $else {
ptr := C.malloc(n)
if ptr == 0 {
panic('malloc($n) failed')
}
return ptr
2019-12-14 00:46:55 +01:00
}
2019-12-19 21:52:45 +01:00
/*
TODO
2019-06-22 20:20:28 +02:00
#ifdef VPLAY
if n > 10000 {
panic('allocating more than 10 KB is not allowed in the playground')
}
#endif
#ifdef DEBUG_ALLOC
total_m += n
println('\n\n\nmalloc($n) total=$total_m')
2019-06-22 20:20:28 +02:00
print_backtrace()
#endif
*/
}
[unsafe_fn]
pub fn v_realloc(b byteptr, n int) byteptr {
ptr := C.realloc(b, n)
if ptr == 0 {
panic('realloc($n) failed')
}
2019-12-19 21:52:45 +01:00
return ptr
2019-06-22 20:20:28 +02:00
}
pub fn v_calloc(n int) byteptr {
return C.calloc(n, 1)
}
2019-06-22 20:20:28 +02:00
pub fn vcalloc(n int) byteptr {
if n < 0 {
2019-12-16 20:22:04 +01:00
panic('calloc(<=0)')
} else if n == 0 {
return byteptr(0)
} else {
return C.calloc(n, 1)
2019-06-22 20:20:28 +02:00
}
}
2019-10-20 19:29:24 +02:00
[unsafe_fn]
pub fn free(ptr voidptr) {
C.free(ptr)
}
2019-10-31 11:08:01 +01:00
pub fn memdup(src voidptr, sz int) voidptr {
if sz == 0 {
return vcalloc(1)
}
2019-06-22 20:20:28 +02:00
mem := malloc(sz)
return C.memcpy(mem, src, sz)
}
fn v_ptr_free(ptr voidptr) {
C.free(ptr)
}
2019-06-22 20:20:28 +02:00
2019-11-06 21:04:40 +01:00
pub fn is_atty(fd int) int {
$if windows {
2019-11-16 00:30:50 +01:00
mut mode := u32(0)
osfh := voidptr(C._get_osfhandle(fd))
C.GetConsoleMode(osfh, voidptr(&mode))
return int(mode)
} $else {
2019-11-06 21:04:40 +01:00
return C.isatty(fd)
}
}
2020-04-25 08:36:53 +02:00
fn __as_cast(obj voidptr, obj_type, expected_type int) voidptr {
if obj_type != expected_type {
panic('as cast: cannot cast $obj_type to $expected_type')
}
return obj
}