Set a reasonable max count on open files and don't panic if set fails

This commit is contained in:
Connor Imes 2015-10-13 09:04:40 -05:00
parent efdaf5e7e5
commit fd710eebc1

View file

@ -102,15 +102,37 @@ use dom::bindings::codegen::RegisterBindings;
#[allow(unsafe_code)] #[allow(unsafe_code)]
fn perform_platform_specific_initialization() { fn perform_platform_specific_initialization() {
use std::mem; use std::mem;
const RLIMIT_NOFILE: libc::c_int = 7; // 4096 is default max on many linux systems
const MAX_FILE_LIMIT: libc::rlim_t = 4096;
// Bump up our number of file descriptors to save us from impending doom caused by an onslaught // Bump up our number of file descriptors to save us from impending doom caused by an onslaught
// of iframes. // of iframes.
unsafe { unsafe {
let mut rlim = mem::uninitialized(); let mut rlim: libc::rlimit = mem::uninitialized();
assert!(libc::getrlimit(RLIMIT_NOFILE, &mut rlim) == 0); match libc::getrlimit(libc::RLIMIT_NOFILE, &mut rlim) {
rlim.rlim_cur = rlim.rlim_max; 0 => {
assert!(libc::setrlimit(RLIMIT_NOFILE, &mut rlim) == 0); if rlim.rlim_cur >= MAX_FILE_LIMIT {
// we have more than enough
return;
}
rlim.rlim_cur = match rlim.rlim_max {
libc::RLIM_INFINITY => MAX_FILE_LIMIT,
_ => {
if rlim.rlim_max < MAX_FILE_LIMIT {
rlim.rlim_max
} else {
MAX_FILE_LIMIT
}
}
};
match libc::setrlimit(libc::RLIMIT_NOFILE, &mut rlim) {
0 => (),
_ => warn!("Failed to set file count limit"),
};
},
_ => warn!("Failed to get file count limit"),
};
} }
} }