use libc::uintptr_t; use std::ffi::OsString; use std::path::{Path, PathBuf}; use std::sync::{Arc, Condvar, Mutex, RwLock}; use crate::chat::*; use crate::constants::*; use crate::contact::*; use crate::dc_tools::*; use crate::error::*; use crate::imap::*; use crate::job::*; use crate::job_thread::JobThread; use crate::key::*; use crate::login_param::LoginParam; use crate::lot::Lot; use crate::message::*; use crate::param::Params; use crate::smtp::*; use crate::sql::Sql; /// Callback function type for [Context] /// /// # Parameters /// /// * `context` - The context object as returned by [Context::new]. /// * `event` - One of the [Event] items. /// * `data1` - Depends on the event parameter, see [Event]. /// * `data2` - Depends on the event parameter, see [Event]. /// /// # Returns /// /// This callback must return 0 unless stated otherwise in the event /// description at [Event]. pub type ContextCallback = dyn Fn(&Context, Event, uintptr_t, uintptr_t) -> uintptr_t + Send + Sync; #[derive(DebugStub)] pub struct Context { pub(crate) dbfile: PathBuf, pub(crate) blobdir: PathBuf, pub sql: Sql, pub inbox: Arc>, pub perform_inbox_jobs_needed: Arc>, pub probe_imap_network: Arc>, pub sentbox_thread: Arc>, pub mvbox_thread: Arc>, pub smtp: Arc>, pub smtp_state: Arc<(Mutex, Condvar)>, pub oauth2_critical: Arc>, #[debug_stub = "Callback"] pub cb: Box, pub os_name: Option, pub cmdline_sel_chat_id: Arc>, pub bob: Arc>, pub last_smeared_timestamp: Arc>, pub running_state: Arc>, /// Mutex to avoid generating the key for the user more than once. pub generating_key_mutex: Mutex<()>, } #[derive(Debug, PartialEq, Eq)] pub struct RunningState { pub ongoing_running: bool, pub shall_stop_ongoing: bool, } impl Context { pub fn new(cb: Box, os_name: String, dbfile: PathBuf) -> Result { let mut blob_fname = OsString::new(); blob_fname.push(dbfile.file_name().unwrap_or_default()); blob_fname.push("-blobs"); let blobdir = dbfile.with_file_name(blob_fname); if !blobdir.exists() { std::fs::create_dir_all(&blobdir)?; } Context::with_blobdir(cb, os_name, dbfile, blobdir) } pub fn with_blobdir( cb: Box, os_name: String, dbfile: PathBuf, blobdir: PathBuf, ) -> Result { ensure!( blobdir.is_dir(), "Blobdir does not exist: {}", blobdir.display() ); let ctx = Context { blobdir, dbfile, inbox: Arc::new(RwLock::new(Imap::new())), cb, os_name: Some(os_name), running_state: Arc::new(RwLock::new(Default::default())), sql: Sql::new(), smtp: Arc::new(Mutex::new(Smtp::new())), smtp_state: Arc::new((Mutex::new(Default::default()), Condvar::new())), oauth2_critical: Arc::new(Mutex::new(())), bob: Arc::new(RwLock::new(Default::default())), last_smeared_timestamp: Arc::new(RwLock::new(0)), cmdline_sel_chat_id: Arc::new(RwLock::new(0)), sentbox_thread: Arc::new(RwLock::new(JobThread::new( "SENTBOX", "configured_sentbox_folder", Imap::new(), ))), mvbox_thread: Arc::new(RwLock::new(JobThread::new( "MVBOX", "configured_mvbox_folder", Imap::new(), ))), probe_imap_network: Arc::new(RwLock::new(false)), perform_inbox_jobs_needed: Arc::new(RwLock::new(false)), generating_key_mutex: Mutex::new(()), }; ensure!( ctx.sql.open(&ctx, &ctx.dbfile, 0), "Failed opening sqlite database" ); Ok(ctx) } pub fn get_dbfile(&self) -> &Path { self.dbfile.as_path() } pub fn get_blobdir(&self) -> &Path { self.blobdir.as_path() } pub fn call_cb(&self, event: Event, data1: uintptr_t, data2: uintptr_t) -> uintptr_t { (*self.cb)(self, event, data1, data2) } } impl Drop for Context { fn drop(&mut self) { info!(self, "disconnecting INBOX-watch",); self.inbox.read().unwrap().disconnect(self); info!(self, "disconnecting sentbox-thread",); self.sentbox_thread.read().unwrap().imap.disconnect(self); info!(self, "disconnecting mvbox-thread",); self.mvbox_thread.read().unwrap().imap.disconnect(self); info!(self, "disconnecting SMTP"); self.smtp.clone().lock().unwrap().disconnect(); self.sql.close(self); } } impl Default for RunningState { fn default() -> Self { RunningState { ongoing_running: false, shall_stop_ongoing: true, } } } #[derive(Debug, Default)] pub struct BobStatus { pub expects: i32, pub status: i32, pub qr_scan: Option, } #[derive(Default, Debug)] pub struct SmtpState { pub idle: bool, pub suspended: bool, pub doing_jobs: bool, pub perform_jobs_needed: i32, pub probe_network: bool, } /* ****************************************************************************** * INI-handling, Information ******************************************************************************/ pub unsafe fn dc_get_info(context: &Context) -> *mut libc::c_char { let unset = "0"; let l = LoginParam::from_database(context, ""); let l2 = LoginParam::from_database(context, "configured_"); let displayname = context.sql.get_config(context, "displayname"); let chats = get_chat_cnt(context) as usize; let real_msgs = dc_get_real_msg_cnt(context) as usize; let deaddrop_msgs = dc_get_deaddrop_msg_cnt(context) as usize; let contacts = Contact::get_real_cnt(context) as usize; let is_configured = context .sql .get_config_int(context, "configured") .unwrap_or_default(); let dbversion = context .sql .get_config_int(context, "dbversion") .unwrap_or_default(); let e2ee_enabled = context .sql .get_config_int(context, "e2ee_enabled") .unwrap_or_else(|| 1); let mdns_enabled = context .sql .get_config_int(context, "mdns_enabled") .unwrap_or_else(|| 1); let prv_key_cnt: Option = context.sql.query_get_value( context, "SELECT COUNT(*) FROM keypairs;", rusqlite::NO_PARAMS, ); let pub_key_cnt: Option = context.sql.query_get_value( context, "SELECT COUNT(*) FROM acpeerstates;", rusqlite::NO_PARAMS, ); let fingerprint_str = if let Some(key) = Key::from_self_public(context, &l2.addr, &context.sql) { key.fingerprint() } else { "".into() }; let inbox_watch = context .sql .get_config_int(context, "inbox_watch") .unwrap_or_else(|| 1); let sentbox_watch = context .sql .get_config_int(context, "sentbox_watch") .unwrap_or_else(|| 1); let mvbox_watch = context .sql .get_config_int(context, "mvbox_watch") .unwrap_or_else(|| 1); let mvbox_move = context .sql .get_config_int(context, "mvbox_move") .unwrap_or_else(|| 1); let folders_configured = context .sql .get_config_int(context, "folders_configured") .unwrap_or_default(); let configured_sentbox_folder = context .sql .get_config(context, "configured_sentbox_folder") .unwrap_or_else(|| "".to_string()); let configured_mvbox_folder = context .sql .get_config(context, "configured_mvbox_folder") .unwrap_or_else(|| "".to_string()); let res = format!( "deltachat_core_version=v{}\n\ sqlite_version={}\n\ sqlite_thread_safe={}\n\ arch={}\n\ number_of_chats={}\n\ number_of_chat_messages={}\n\ messages_in_contact_requests={}\n\ number_of_contacts={}\n\ database_dir={}\n\ database_version={}\n\ blobdir={}\n\ display_name={}\n\ is_configured={}\n\ entered_account_settings={}\n\ used_account_settings={}\n\ inbox_watch={}\n\ sentbox_watch={}\n\ mvbox_watch={}\n\ mvbox_move={}\n\ folders_configured={}\n\ configured_sentbox_folder={}\n\ configured_mvbox_folder={}\n\ mdns_enabled={}\n\ e2ee_enabled={}\n\ private_key_count={}\n\ public_key_count={}\n\ fingerprint={}\n\ level=awesome\n", &*DC_VERSION_STR, rusqlite::version(), rusqlite::ffi::sqlite3_threadsafe(), // arch (::std::mem::size_of::<*mut libc::c_void>()).wrapping_mul(8), chats, real_msgs, deaddrop_msgs, contacts, context.get_dbfile().to_str().unwrap(), dbversion, context.get_blobdir().to_str().unwrap(), displayname.unwrap_or_else(|| unset.into()), is_configured, l, l2, inbox_watch, sentbox_watch, mvbox_watch, mvbox_move, folders_configured, configured_sentbox_folder, configured_mvbox_folder, mdns_enabled, e2ee_enabled, prv_key_cnt.unwrap_or_default(), pub_key_cnt.unwrap_or_default(), fingerprint_str, ); res.strdup() } pub unsafe fn dc_get_version_str() -> *mut libc::c_char { (&*DC_VERSION_STR).strdup() } pub fn dc_get_fresh_msgs(context: &Context) -> Vec { let show_deaddrop = 0; context .sql .query_map( "SELECT m.id FROM msgs m LEFT JOIN contacts ct \ ON m.from_id=ct.id LEFT JOIN chats c ON m.chat_id=c.id WHERE m.state=? \ AND m.hidden=0 \ AND m.chat_id>? \ AND ct.blocked=0 \ AND (c.blocked=0 OR c.blocked=?) ORDER BY m.timestamp DESC,m.id DESC;", &[10, 9, if 0 != show_deaddrop { 2 } else { 0 }], |row| row.get(0), |rows| { let mut ret = Vec::new(); for row in rows { let id: u32 = row?; ret.push(id); } Ok(ret) }, ) .unwrap() } #[allow(non_snake_case)] pub fn dc_search_msgs(context: &Context, chat_id: u32, query: *const libc::c_char) -> Vec { if query.is_null() { return Vec::new(); } let real_query = as_str(query).trim(); if real_query.is_empty() { return Vec::new(); } let strLikeInText = format!("%{}%", real_query); let strLikeBeg = format!("{}%", real_query); let query = if 0 != chat_id { "SELECT m.id, m.timestamp FROM msgs m LEFT JOIN contacts ct ON m.from_id=ct.id WHERE m.chat_id=? \ AND m.hidden=0 \ AND ct.blocked=0 AND (txt LIKE ? OR ct.name LIKE ?) ORDER BY m.timestamp,m.id;" } else { "SELECT m.id, m.timestamp FROM msgs m LEFT JOIN contacts ct ON m.from_id=ct.id \ LEFT JOIN chats c ON m.chat_id=c.id WHERE m.chat_id>9 AND m.hidden=0 \ AND (c.blocked=0 OR c.blocked=?) \ AND ct.blocked=0 AND (m.txt LIKE ? OR ct.name LIKE ?) ORDER BY m.timestamp DESC,m.id DESC;" }; context .sql .query_map( query, params![chat_id as libc::c_int, &strLikeInText, &strLikeBeg], |row| row.get::<_, i32>(0), |rows| { let mut ret = Vec::new(); for id in rows { ret.push(id? as u32); } Ok(ret) }, ) .unwrap_or_default() } pub fn dc_is_inbox(_context: &Context, folder_name: impl AsRef) -> bool { folder_name.as_ref() == "INBOX" } pub fn dc_is_sentbox(context: &Context, folder_name: impl AsRef) -> bool { let sentbox_name = context.sql.get_config(context, "configured_sentbox_folder"); if let Some(name) = sentbox_name { name == folder_name.as_ref() } else { false } } pub fn dc_is_mvbox(context: &Context, folder_name: impl AsRef) -> bool { let mvbox_name = context.sql.get_config(context, "configured_mvbox_folder"); if let Some(name) = mvbox_name { name == folder_name.as_ref() } else { false } } pub fn do_heuristics_moves(context: &Context, folder: &str, msg_id: u32) { if context .sql .get_config_int(context, "mvbox_move") .unwrap_or_else(|| 1) == 0 { return; } if !dc_is_inbox(context, folder) && !dc_is_sentbox(context, folder) { return; } if let Ok(msg) = dc_msg_new_load(context, msg_id) { if dc_msg_is_setupmessage(&msg) { // do not move setup messages; // there may be a non-delta device that wants to handle it return; } if dc_is_mvbox(context, folder) { dc_update_msg_move_state(context, msg.rfc724_mid, MoveState::Stay); } // 1 = dc message, 2 = reply to dc message if 0 != msg.is_dc_message { job_add( context, Action::MoveMsg, msg.id as libc::c_int, Params::new(), 0, ); dc_update_msg_move_state(context, msg.rfc724_mid, MoveState::Moving); } } } #[cfg(test)] mod tests { use super::*; use crate::test_utils::*; #[test] fn test_wrong_db() { let tmp = tempfile::tempdir().unwrap(); let dbfile = tmp.path().join("db.sqlite"); std::fs::write(&dbfile, b"123").unwrap(); let res = Context::new(Box::new(|_, _, _, _| 0), "FakeOs".into(), dbfile); assert!(res.is_err()); } #[test] fn test_blobdir_exists() { let tmp = tempfile::tempdir().unwrap(); let dbfile = tmp.path().join("db.sqlite"); Context::new(Box::new(|_, _, _, _| 0), "FakeOS".into(), dbfile).unwrap(); let blobdir = tmp.path().join("db.sqlite-blobs"); assert!(blobdir.is_dir()); } #[test] fn test_wrong_blogdir() { let tmp = tempfile::tempdir().unwrap(); let dbfile = tmp.path().join("db.sqlite"); let blobdir = tmp.path().join("db.sqlite-blobs"); std::fs::write(&blobdir, b"123").unwrap(); let res = Context::new(Box::new(|_, _, _, _| 0), "FakeOS".into(), dbfile); assert!(res.is_err()); } #[test] fn test_sqlite_parent_not_exists() { let tmp = tempfile::tempdir().unwrap(); let subdir = tmp.path().join("subdir"); let dbfile = subdir.join("db.sqlite"); let dbfile2 = dbfile.clone(); Context::new(Box::new(|_, _, _, _| 0), "FakeOS".into(), dbfile).unwrap(); assert!(subdir.is_dir()); assert!(dbfile2.is_file()); } #[test] fn test_with_blobdir_not_exists() { let tmp = tempfile::tempdir().unwrap(); let dbfile = tmp.path().join("db.sqlite"); let blobdir = tmp.path().join("blobs"); let res = Context::with_blobdir(Box::new(|_, _, _, _| 0), "FakeOS".into(), dbfile, blobdir); assert!(res.is_err()); } #[test] fn no_crashes_on_context_deref() { let t = dummy_context(); std::mem::drop(t.ctx); } }