rust/src/lib/generic_os.rs

63 lines
1.6 KiB
Rust
Raw Normal View History

2011-07-17 19:11:40 -05:00
import str::sbuf;
2011-07-17 19:11:40 -05:00
#[cfg(target_os = "linux")]
#[cfg(target_os = "macos")]
fn getenv(n: &istr) -> option::t<istr> {
let n = istr::to_estr(n);
2011-07-27 07:19:39 -05:00
let s = os::libc::getenv(str::buf(n));
ret if s as int == 0 {
option::none::<istr>
} else {
let s = unsafe::reinterpret_cast(s);
option::some::<istr>(istr::str_from_cstr(s))
};
2011-07-17 19:11:40 -05:00
}
#[cfg(target_os = "linux")]
#[cfg(target_os = "macos")]
fn setenv(n: &istr, v: &istr) {
let n = istr::to_estr(n);
let v = istr::to_estr(v);
2011-07-27 07:19:39 -05:00
let nbuf = str::buf(n);
let vbuf = str::buf(v);
2011-07-17 19:11:40 -05:00
os::libc::setenv(nbuf, vbuf, 1);
}
#[cfg(target_os = "win32")]
fn getenv(n: &istr) -> option::t<istr> {
let n = istr::to_estr(n);
2011-07-27 07:19:39 -05:00
let nbuf = str::buf(n);
let nsize = 256u;
while true {
let vstr = str::alloc(nsize - 1u);
let vbuf = str::buf(vstr);
let res = os::kernel32::GetEnvironmentVariableA(nbuf, vbuf, nsize);
if res == 0u {
2011-07-17 19:11:40 -05:00
ret option::none;
} else if res < nsize {
let vbuf = unsafe::reinterpret_cast(vbuf);
ret option::some(istr::str_from_cstr(vbuf));
2011-07-27 07:19:39 -05:00
} else { nsize = res; }
2011-07-17 19:11:40 -05:00
}
fail;
}
#[cfg(target_os = "win32")]
fn setenv(n: &istr, v: &istr) {
let n = istr::to_estr(n);
let v = istr::to_estr(v);
2011-07-27 07:19:39 -05:00
let nbuf = str::buf(n);
let vbuf = str::buf(v);
2011-07-17 19:11:40 -05:00
os::kernel32::SetEnvironmentVariableA(nbuf, vbuf);
}
// Local Variables:
// mode: rust;
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C .. 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End: