odin-lang.Odin/core/os_linux.odin

foreign_system_library (
	dl   "dl";
	libc "c";
)
import "strings.odin";

type (
	Handle   i32;
	FileTime u64;
	Errno    i32;
)

const (
	O_RDONLY   = 0x00000;
	O_WRONLY   = 0x00001;
	O_RDWR     = 0x00002;
	O_CREAT    = 0x00040;
	O_EXCL     = 0x00080;
	O_NOCTTY   = 0x00100;
	O_TRUNC    = 0x00200;
	O_NONBLOCK = 0x00800;
	O_APPEND   = 0x00400;
	O_SYNC     = 0x01000;
	O_ASYNC    = 0x02000;
	O_CLOEXEC  = 0x80000;
	SEEK_SET   = 0;
	SEEK_CUR   = 1;
	SEEK_END   = 2;
	SEEK_DATA  = 3;
	SEEK_HOLE  = 4;
	SEEK_MAX   = SEEK_HOLE;

	// NOTE(zangent): These are OS specific!
	// Do not mix these up!
	RTLD_LAZY         = 0x001;
	RTLD_NOW          = 0x002;
	RTLD_BINDING_MASK = 0x3;
	RTLD_GLOBAL       = 0x100;
)

// "Argv" arguments converted to Odin strings
let args = _alloc_command_line_arguments();

type _FileTime struct #ordered {
	seconds:     i64,
	nanoseconds: i32,
	reserved:    i32,
}

// Translated from
//  https://android.googlesource.com/platform/prebuilts/gcc/linux-x86/host/x86_64-linux-glibc2.7-4.6/+/jb-dev/sysroot/usr/include/bits/stat.h
// Validity is not guaranteed.

type Stat struct #ordered {
	device_id:     u64, // ID of device containing file
	serial:        u64, // File serial number
	nlink:         u32, // Number of hard links
	mode:          u32, // Mode of the file
	uid:           u32, // User ID of the file's owner
	gid:           u32, // Group ID of the file's group
	_padding:      i32, // 32 bits of padding
	rdev:          u64, // Device ID, if device
	size:          i64, // Size of the file, in bytes
	block_size:    i64, // Optimal bllocksize for I/O
	blocks:        i64, // Number of 512-byte blocks allocated

	last_access:   _FileTime, // Time of last access
	modified:      _FileTime, // Time of last modification
	status_change: _FileTime, // Time of last status change

	_reserve1,
	_reserve2,
	_reserve3:     i64,
	serial_numbe:  u64, // File serial number...? Maybe.
	_reserve4:     i64,
};

// File type
const (
	S_IFMT   = 0170000; // Type of file mask
	S_IFIFO  = 0010000; // Named pipe (fifo)
	S_IFCHR  = 0020000; // Character special
	S_IFDIR  = 0040000; // Directory
	S_IFBLK  = 0060000; // Block special
	S_IFREG  = 0100000; // Regular
	S_IFLNK  = 0120000; // Symbolic link
	S_IFSOCK = 0140000; // Socket

	// File mode
	// Read, write, execute/search by owner
	S_IRWXU = 0000700; // RWX mask for owner
	S_IRUSR = 0000400; // R for owner
	S_IWUSR = 0000200; // W for owner
	S_IXUSR = 0000100; // X for owner

	// Read, write, execute/search by group
	S_IRWXG = 0000070; // RWX mask for group
	S_IRGRP = 0000040; // R for group
	S_IWGRP = 0000020; // W for group
	S_IXGRP = 0000010; // X for group

	// Read, write, execute/search by others
	S_IRWXO = 0000007; // RWX mask for other
	S_IROTH = 0000004; // R for other
	S_IWOTH = 0000002; // W for other
	S_IXOTH = 0000001; // X for other

	S_ISUID = 0004000; // Set user id on execution
	S_ISGID = 0002000; // Set group id on execution
	S_ISVTX = 0001000; // Directory restrcted delete
)

proc S_ISLNK (m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFLNK; }
proc S_ISREG (m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFREG; }
proc S_ISDIR (m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFDIR; }
proc S_ISCHR (m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFCHR; }
proc S_ISBLK (m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFBLK; }
proc S_ISFIFO(m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFIFO; }
proc S_ISSOCK(m: u32) -> bool #inline  {return (m & S_IFMT) == S_IFSOCK;}

const (
	R_OK = 4; // Test for read permission
	W_OK = 2; // Test for write permission
	X_OK = 1; // Test for execute permission
	F_OK = 0; // Test for file existance
)

foreign libc {
	proc _unix_open  (path: ^u8, mode: int) -> Handle                  #link_name "open";
	proc _unix_close (fd: Handle) -> i32                               #link_name "close";
	proc _unix_read  (fd: Handle, buf: rawptr, size: int) -> int       #link_name "read";
	proc _unix_write (fd: Handle, buf: rawptr, size: int) -> int       #link_name "write";
	proc _unix_seek  (fd: Handle, offset: i64, whence: i32) -> i64     #link_name "lseek64";
	proc _unix_gettid() -> u64                                         #link_name "gettid";
	proc _unix_stat  (path: ^u8, stat: ^Stat) -> i32                   #link_name "stat";
	proc _unix_access(path: ^u8, mask: int) -> i32                     #link_name "access";

	proc _unix_malloc (size: int) -> rawptr                            #link_name "malloc";
	proc _unix_free   (ptr: rawptr)                                    #link_name "free";
	proc _unix_realloc(ptr: rawptr, size: int) -> rawptr               #link_name "realloc";
	proc _unix_getenv (^u8) -> ^u8                                     #link_name "getenv";

	proc _unix_exit(status: int)                                       #link_name "exit";
}
foreign dl {
	proc _unix_dlopen (filename: ^u8, flags: int) -> rawptr            #link_name "dlopen";
	proc _unix_dlsym  (handle: rawptr, symbol: ^u8) ->  (proc() #cc_c) #link_name "dlsym";
	proc _unix_dlclose(handle: rawptr) -> int                          #link_name "dlclose";
	proc _unix_dlerror() -> ^u8                                        #link_name "dlerror";
}

// TODO(zangent): Change this to just `open` when Bill fixes overloading.
proc open_simple(path: string, mode: int) -> (Handle, Errno) {

	var cstr = strings.new_c_string(path);
	var handle = _unix_open(cstr, mode);
	free(cstr);
	if(handle == -1) {
		return 0, 1;
	}
	return handle, 0;
}
// NOTE(zangent): This is here for compatability reasons. Should this be here?
proc open(path: string, mode: int, perm: u32) -> (Handle, Errno) {
	return open_simple(path, mode);
}

proc close(fd: Handle) {
	_unix_close(fd);
}

proc read(fd: Handle, data: []u8) -> (int, Errno) {
	var sz = _unix_read(fd, &data[0], len(data));
	return sz, 0;
}

proc write(fd: Handle, data: []u8) -> (int, Errno) {
	var sz = _unix_write(fd, &data[0], len(data));
	return sz, 0;
}

proc seek(fd: Handle, offset: i64, whence: int) -> (i64, Errno) {
	var res = _unix_seek(fd, offset, i32(whence));
	return res, 0;
}

proc file_size(fd: Handle) -> (i64, Errno) {
	var prev, _ = seek(fd, 0, SEEK_CUR);
	var size, err = seek(fd, 0, SEEK_END);
	seek(fd, prev, SEEK_SET);
	return size, err;
}


// NOTE(bill): Uses startup to initialize it
var (
	stdin:  Handle = 0;
	stdout: Handle = 1;
	stderr: Handle = 2;
)
/* TODO(zangent): Implement these!
proc last_write_time(fd: Handle) -> FileTime {}
proc last_write_time_by_name(name: string) -> FileTime {}
*/

proc stat(path: string) -> (Stat, int) #inline {
	var s: Stat;
	var cstr = strings.new_c_string(path);
	defer free(cstr);
	var ret_int = _unix_stat(cstr, &s);
	return s, int(ret_int);
}

proc access(path: string, mask: int) -> bool #inline {
	var cstr = strings.new_c_string(path);
	defer free(cstr);
	return _unix_access(cstr, mask) == 0;
}

proc heap_alloc(size: int) -> rawptr {
	assert(size > 0);
	return _unix_malloc(size);
}

proc heap_resize(ptr: rawptr, new_size: int) -> rawptr {
	return _unix_realloc(ptr, new_size);
}

proc heap_free(ptr: rawptr) {
	_unix_free(ptr);
}

proc getenv(name: string) -> (string, bool) {
	var path_str = strings.new_c_string(name);
	var cstr: ^u8 = _unix_getenv(path_str);
	free(path_str);
	if(cstr == nil) {
		return "", false;
	}
	return strings.to_odin_string(cstr), true;
}

proc exit(code: int) {
	_unix_exit(code);
}

proc current_thread_id() -> int {
	// return int(_unix_gettid());
	return 0;
}

proc dlopen(filename: string, flags: int) -> rawptr #inline {
	var cstr = strings.new_c_string(filename);
	var handle = _unix_dlopen(cstr, flags);
	free(cstr);
	return handle;
}
proc dlsym(handle: rawptr, symbol: string) -> (proc() #cc_c) #inline {
	assert(handle != nil);
	var cstr = strings.new_c_string(symbol);
	var proc_handle = _unix_dlsym(handle, cstr);
	free(cstr);
	return proc_handle;
}
proc dlclose(handle: rawptr) -> bool #inline {
	assert(handle != nil);
	return _unix_dlclose(handle) == 0;
}
proc dlerror() -> string {
	return strings.to_odin_string(_unix_dlerror());
}


proc _alloc_command_line_arguments() -> []string {
	// TODO(bill):
	return nil;
}