rust/src/lib/generic_os.rs
Marijn Haverbeke fc6b7c8b38 Reformat for new mode syntax, step 1
Long lines were fixed in a very crude way, as I'll be following up
with another reformat in a bit.
2011-09-12 12:04:14 +02:00

77 lines
2.2 KiB
Rust

import str::sbuf;
#[cfg(target_os = "linux")]
#[cfg(target_os = "macos")]
fn getenv(n: str) -> option::t<str> {
let s = str::as_buf(n, {|buf| os::libc::getenv(buf) });
ret if unsafe::reinterpret_cast(s) == 0 {
option::none::<str>
} else {
let s = unsafe::reinterpret_cast(s);
option::some::<str>(str::str_from_cstr(s))
};
}
#[cfg(target_os = "linux")]
#[cfg(target_os = "macos")]
fn setenv(n: str, v: str) {
// FIXME (868)
let _: () =
str::as_buf(n,
// FIXME (868)
{|nbuf|
let _: () =
str::as_buf(v,
{|vbuf|
os::libc::setenv(nbuf, vbuf, 1);
});
});
}
#[cfg(target_os = "win32")]
fn getenv(n: str) -> option::t<str> {
let nsize = 256u;
while true {
let v: [u8] = [];
vec::reserve(v, nsize);
let res =
str::as_buf(n,
{|nbuf|
let vbuf = vec::to_ptr(v);
os::kernel32::GetEnvironmentVariableA(nbuf, vbuf,
nsize)
});
if res == 0u {
ret option::none;
} else if res < nsize {
vec::unsafe::set_len(v, res);
ret option::some(str::unsafe_from_bytes(v));
} else { nsize = res; }
}
fail;
}
#[cfg(target_os = "win32")]
fn setenv(n: str, v: str) {
// FIXME (868)
let _: () =
str::as_buf(n,
{|nbuf|
let _: () =
str::as_buf(v,
{|vbuf|
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: