rust/src/lib/generic_os.rs

import str::sbuf;


#[cfg(target_os = "linux")]
#[cfg(target_os = "macos")]
fn getenv(n: &istr) -> option::t<istr> {
    let s = str::as_buf(n, { |buf|
        os::libc::getenv(buf)
    });
    ret if unsafe::reinterpret_cast(s) == 0 {
        option::none::<istr>
    } else {
        let s = unsafe::reinterpret_cast(s);
        option::some::<istr>(str::str_from_cstr(s))
    };
}

#[cfg(target_os = "linux")]
#[cfg(target_os = "macos")]
fn setenv(n: &istr, v: &istr) {
    // FIXME (868)
    let _: () = str::as_buf(n, { |nbuf|
        // FIXME (868)
        let _: () = str::as_buf(v, { |vbuf|
            os::libc::setenv(nbuf, vbuf, 1);
        });
    });
}

#[cfg(target_os = "win32")]
fn getenv(n: &istr) -> option::t<istr> {
    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: &istr, v: &istr) {
    // 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: