Merge pull request #252 from link2xt/dc_get_config-safe

Make dc_get_config_t safe
This commit is contained in:
Friedel Ziegelmayer
2019-07-30 00:25:29 +02:00
committed by GitHub
3 changed files with 13 additions and 32 deletions

View File

@@ -249,16 +249,8 @@ unsafe fn cb_set_config(context: &Context, key: *const libc::c_char, value: *con
* *
* @private @memberof Context * @private @memberof Context
*/ */
unsafe fn cb_get_config( fn cb_get_config(context: &Context, key: &str) -> Option<String> {
context: &Context, context.sql.get_config(context, key)
key: *const libc::c_char,
def: *const libc::c_char,
) -> *mut libc::c_char {
let res = context
.sql
.get_config(context, as_str(key))
.unwrap_or_else(|| to_string(def));
to_cstring(res)
} }
pub unsafe fn dc_context_unref(context: &mut Context) { pub unsafe fn dc_context_unref(context: &mut Context) {

View File

@@ -5,7 +5,7 @@ use std::time::{Duration, SystemTime};
use crate::constants::*; use crate::constants::*;
use crate::context::Context; use crate::context::Context;
use crate::dc_loginparam::*; use crate::dc_loginparam::*;
use crate::dc_tools::{as_str, to_cstring}; use crate::dc_tools::to_cstring;
use crate::oauth2::dc_get_oauth2_access_token; use crate::oauth2::dc_get_oauth2_access_token;
use crate::types::*; use crate::types::*;
use crate::x::free; use crate::x::free;
@@ -705,26 +705,16 @@ impl Imap {
fn get_config_last_seen_uid<S: AsRef<str>>(&self, context: &Context, folder: S) -> (u32, u32) { fn get_config_last_seen_uid<S: AsRef<str>>(&self, context: &Context, folder: S) -> (u32, u32) {
let key = format!("imap.mailbox.{}", folder.as_ref()); let key = format!("imap.mailbox.{}", folder.as_ref());
let val1 = unsafe { if let Some(entry) = (self.get_config)(context, &key) {
let key_c = to_cstring(key);
let val = (self.get_config)(context, key_c, 0 as *const libc::c_char);
free(key_c as *mut _);
val
};
if val1.is_null() {
return (0, 0);
}
let entry = as_str(val1);
if entry.is_empty() {
return (0, 0);
}
// the entry has the format `imap.mailbox.<folder>=<uidvalidity>:<lastseenuid>` // the entry has the format `imap.mailbox.<folder>=<uidvalidity>:<lastseenuid>`
let mut parts = entry.split(':'); let mut parts = entry.split(':');
( (
parts.next().unwrap().parse().unwrap_or_else(|_| 0), parts.next().unwrap().parse().unwrap_or_else(|_| 0),
parts.next().unwrap().parse().unwrap_or_else(|_| 0), parts.next().unwrap().parse().unwrap_or_else(|_| 0),
) )
} else {
(0, 0)
}
} }
fn fetch_from_single_folder<S: AsRef<str>>(&self, context: &Context, folder: S) -> usize { fn fetch_from_single_folder<S: AsRef<str>>(&self, context: &Context, folder: S) -> usize {

View File

@@ -36,8 +36,7 @@ pub type dc_precheck_imf_t =
unsafe fn(_: &Context, _: *const libc::c_char, _: &str, _: u32) -> libc::c_int; unsafe fn(_: &Context, _: *const libc::c_char, _: &str, _: u32) -> libc::c_int;
pub type dc_set_config_t = pub type dc_set_config_t =
unsafe fn(_: &Context, _: *const libc::c_char, _: *const libc::c_char) -> (); unsafe fn(_: &Context, _: *const libc::c_char, _: *const libc::c_char) -> ();
pub type dc_get_config_t = pub type dc_get_config_t = fn(_: &Context, _: &str) -> Option<String>;
unsafe fn(_: &Context, _: *const libc::c_char, _: *const libc::c_char) -> *mut libc::c_char;
pub type sqlite_int64 = i64; pub type sqlite_int64 = i64;
pub type sqlite3_int64 = sqlite_int64; pub type sqlite3_int64 = sqlite_int64;