ISAR/rusqlite
3
0
Fork 0
mirror of https://github.com/isar/rusqlite.git synced 2024-11-29 21:52:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix i32 overflow in Connection::busy_timeout

Browse Source
This commit is contained in:
arthurprs 2019-12-28 20:18:10 +01:00
parent ff2a5a5588
commit cae86b2bfe
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/busy.rs
View File

@ -1,4 +1,5 @@
///! Busy handler (when the database is locked) ///! Busy handler (when the database is locked)
use std::convert::TryInto;
use std::mem; use std::mem;
use std::os::raw::{c_int, c_void}; use std::os::raw::{c_int, c_void};
use std::panic::catch_unwind; use std::panic::catch_unwind;
@ -21,12 +22,13 @@ impl Connection {
/// (using `busy_handler`) prior to calling this routine, that other /// (using `busy_handler`) prior to calling this routine, that other
/// busy handler is cleared. /// busy handler is cleared.
pub fn busy_timeout(&self, timeout: Duration) -> Result<()> { pub fn busy_timeout(&self, timeout: Duration) -> Result<()> {
let ms = timeout let ms: i32 = timeout
.as_secs() .as_secs()
.checked_mul(1000) .checked_mul(1000)
.and_then(|t| t.checked_add(timeout.subsec_millis().into())) .and_then(|t| t.checked_add(timeout.subsec_millis().into()))
.and_then(|t| t.try_into().ok())
.expect("too big"); .expect("too big");
self.db.borrow_mut().busy_timeout(ms as i32) self.db.borrow_mut().busy_timeout(ms)
} }
/// Register a callback to handle `SQLITE_BUSY` errors. /// Register a callback to handle `SQLITE_BUSY` errors.