1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/builtin/builtin.v

213 lines
4.2 KiB
V
Raw Normal View History

2019-06-23 05:21:30 +03:00
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-06-22 21:20:28 +03:00
module builtin
2019-12-11 21:42:22 +03:00
__global g_m2_buf byteptr
2019-12-11 18:41:25 +03:00
__global g_m2_ptr byteptr
fn init() {
$if windows {
2019-11-06 23:04:40 +03:00
if is_atty(0) > 0 {
C._setmode(C._fileno(C.stdin), C._O_U16TEXT)
2019-12-19 23:52:45 +03:00
}
else {
C._setmode(C._fileno(C.stdin), C._O_U8TEXT)
}
C._setmode(C._fileno(C.stdout), C._O_U8TEXT)
C.SetConsoleMode(C.GetStdHandle(C.STD_OUTPUT_HANDLE), C.ENABLE_PROCESSED_OUTPUT | 0x0004) // ENABLE_VIRTUAL_TERMINAL_PROCESSING
2019-12-19 23:52:45 +03:00
C.setbuf(C.stdout, 0)
}
}
pub fn exit(code int) {
2019-06-23 11:15:30 +03:00
C.exit(code)
2019-06-22 21:20:28 +03: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 23:52:45 +03:00
fn on_panic(f fn(int)int) {
2019-06-22 21:20:28 +03:00
// TODO
}
pub fn print_backtrace_skipping_top_frames(skipframes int) {
2019-11-13 11:05:06 +03:00
$if windows {
$if msvc {
2019-12-19 23:52:45 +03:00
if print_backtrace_skipping_top_frames_msvc(skipframes) {
return
}
}
2019-11-13 11:05:06 +03:00
$if mingw {
2019-12-19 23:52:45 +03:00
if print_backtrace_skipping_top_frames_mingw(skipframes) {
return
}
}
} $else {
if print_backtrace_skipping_top_frames_nix(skipframes) {
return
2019-11-13 11:05:06 +03:00
}
2019-06-22 21:20:28 +03:00
}
println('print_backtrace_skipping_top_frames is not implemented on this platform for now...\n')
}
2019-10-25 17:59:41 +03:00
2019-12-19 23:52:45 +03: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 21:20:28 +03:00
}
2019-07-30 16:08:14 +03:00
// replaces panic when -debug arg is passed
2019-12-19 23:52:45 +03:00
fn panic_debug(line_no int, file, mod, fn_name, s string) {
2019-07-30 16:08:14 +03:00
println('================ V panic ================')
println(' module: $mod')
println(' function: ${fn_name}()')
println(' file: $file')
println(' line: ' + line_no.str())
println(' message: $s')
println('=========================================')
print_backtrace_skipping_top_frames(1)
2019-07-30 16:08:14 +03:00
C.exit(1)
}
2019-06-22 21:20:28 +03:00
pub fn panic(s string) {
2019-06-23 11:01:55 +03:00
println('V panic: $s')
2019-06-22 21:20:28 +03:00
print_backtrace()
C.exit(1)
}
pub fn eprintln(s string) {
if isnil(s.str) {
panic('eprintln(NIL)')
2019-10-04 15:48:09 +03:00
}
2019-11-28 11:46:52 +03:00
$if !windows {
C.fflush(stdout)
C.fflush(stderr)
2019-11-25 13:54:56 +03:00
C.fprintf(stderr, '%.*s\n', s.len, s.str)
C.fflush(stderr)
return
}
2019-06-22 21:20:28 +03:00
// TODO issues with stderr and cross compiling for Linux
println(s)
2019-06-22 21:20:28 +03:00
}
2019-12-28 13:02:06 +03:00
pub fn eprint(s string) {
if isnil(s.str) {
panic('eprint(NIL)')
}
$if !windows {
C.fflush(stdout)
C.fflush(stderr)
C.fprintf(stderr, '%.*s', s.len, s.str)
C.fflush(stderr)
return
}
print(s)
}
2019-06-22 21:20:28 +03:00
pub fn print(s string) {
2019-07-24 13:16:45 +03:00
$if windows {
C.wprintf(s.to_wide())
} $else {
C.printf('%.*s', s.len, s.str)
}
2019-06-22 21:20:28 +03:00
}
2019-12-19 23:52:45 +03:00
__global total_m i64=0
__global nr_mallocs int=0
2019-12-11 18:41:25 +03:00
2019-10-20 20:29:24 +03:00
[unsafe_fn]
2019-06-22 21:20:28 +03:00
pub fn malloc(n int) byteptr {
2019-12-16 21:29:32 +03:00
if n <= 0 {
2019-12-16 22:22:04 +03:00
panic('malloc(<=0)')
2019-06-22 21:20:28 +03:00
}
2019-12-14 02:46:55 +03:00
$if prealloc {
2019-12-11 18:41:25 +03: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 02:46:55 +03:00
}
2019-12-19 23:52:45 +03:00
/*
TODO
2019-06-22 21:20:28 +03: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 21:20:28 +03:00
print_backtrace()
#endif
*/
2019-12-19 23:52:45 +03:00
2019-06-22 21:20:28 +03:00
}
pub fn calloc(n int) byteptr {
2019-12-16 22:22:04 +03:00
if n <= 0 {
panic('calloc(<=0)')
2019-06-22 21:20:28 +03:00
}
2019-06-24 09:06:38 +03:00
return C.calloc(n, 1)
2019-06-22 21:20:28 +03:00
}
2019-10-20 20:29:24 +03:00
[unsafe_fn]
pub fn free(ptr voidptr) {
C.free(ptr)
}
2019-10-31 13:08:01 +03:00
pub fn memdup(src voidptr, sz int) voidptr {
2019-06-22 21:20:28 +03:00
mem := malloc(sz)
return C.memcpy(mem, src, sz)
}
fn v_ptr_free(ptr voidptr) {
C.free(ptr)
}
2019-06-22 21:20:28 +03:00
2019-11-06 23:04:40 +03:00
pub fn is_atty(fd int) int {
$if windows {
2019-11-16 02:30:50 +03:00
mut mode := u32(0)
osfh := voidptr(C._get_osfhandle(fd))
C.GetConsoleMode(osfh, voidptr(&mode))
return int(mode)
} $else {
2019-11-06 23:04:40 +03:00
return C.isatty(fd)
}
}
2019-12-19 23:52:45 +03:00
2019-12-26 09:55:34 +03:00
/*
fn C.va_start()
fn C.va_end()
fn C.vsnprintf() int
fn C.vsprintf() int
pub fn str2_(fmt charptr, ...) string {
argptr := C.va_list{}
C.va_start(argptr, fmt)
len := C.vsnprintf(0, 0, fmt, argptr) + 1
C.va_end(argptr)
buf := malloc(len)
C.va_start(argptr, fmt)
C.vsprintf(charptr(buf), fmt, argptr)
C.va_end(argptr)
//#ifdef DEBUG_ALLOC
// puts("_STR:");
// puts(buf);
//#endif
return tos2(buf)
}
*/