//! # Imap handling module //! //! uses [async-email/async-imap](https://github.com/async-email/async-imap) //! to implement connect, fetch, delete functionality with standard IMAP servers. use std::collections::BTreeMap; use async_imap::{ error::Result as ImapResult, types::{Capability, Fetch, Flag, Mailbox, Name, NameAttribute}, }; use async_std::prelude::*; use async_std::sync::Receiver; use num_traits::FromPrimitive; use crate::config::*; use crate::constants::*; use crate::context::Context; use crate::dc_receive_imf::{ dc_receive_imf, from_field_to_contact_id, is_msgrmsg_rfc724_mid_in_list, }; use crate::events::Event; use crate::headerdef::{HeaderDef, HeaderDefMap}; use crate::job::{self, Action}; use crate::login_param::{CertificateChecks, LoginParam}; use crate::message::{self, update_server_uid, MessageState}; use crate::mimeparser; use crate::oauth2::dc_get_oauth2_access_token; use crate::param::Params; use crate::provider::get_provider_info; use crate::{chat, scheduler::InterruptInfo, stock::StockMessage}; mod client; mod idle; pub mod select_folder; mod session; use client::Client; use message::Message; use session::Session; type Result = std::result::Result; #[derive(Debug, thiserror::Error)] pub enum Error { #[error("IMAP Connect without configured params")] ConnectWithoutConfigure, #[error("IMAP Connection Failed params: {0}")] ConnectionFailed(String), #[error("IMAP No Connection established")] NoConnection, #[error("IMAP Could not get OAUTH token")] OauthError, #[error("IMAP Could not login as {0}")] LoginFailed(String), #[error("IMAP Could not fetch")] FetchFailed(#[from] async_imap::error::Error), #[error("IMAP operation attempted while it is torn down")] InTeardown, #[error("IMAP operation attempted while it is torn down")] SqlError(#[from] crate::sql::Error), #[error("IMAP got error from elsewhere")] WrappedError(#[from] crate::error::Error), #[error("IMAP select folder error")] SelectFolderError(#[from] select_folder::Error), #[error("Mail parse error")] MailParseError(#[from] mailparse::MailParseError), #[error("No mailbox selected, folder: {0}")] NoMailbox(String), #[error("IMAP other error: {0}")] Other(String), } #[derive(Debug, Display, Clone, Copy, PartialEq, Eq)] pub enum ImapActionResult { Failed, RetryLater, AlreadyDone, Success, } /// Prefetch: /// - Message-ID to check if we already have the message. /// - In-Reply-To and References to check if message is a reply to chat message. /// - Chat-Version to check if a message is a chat message /// - Autocrypt-Setup-Message to check if a message is an autocrypt setup message, /// not necessarily sent by Delta Chat. const PREFETCH_FLAGS: &str = "(UID BODY.PEEK[HEADER.FIELDS (\ MESSAGE-ID \ FROM \ IN-REPLY-TO REFERENCES \ CHAT-VERSION \ AUTOCRYPT-SETUP-MESSAGE\ )])"; const DELETE_CHECK_FLAGS: &str = "(UID BODY.PEEK[HEADER.FIELDS (MESSAGE-ID)])"; const JUST_UID: &str = "(UID)"; const BODY_FLAGS: &str = "(FLAGS BODY.PEEK[])"; const SELECT_ALL: &str = "1:*"; #[derive(Debug)] pub struct Imap { idle_interrupt: Receiver, config: ImapConfig, session: Option, connected: bool, interrupt: Option, should_reconnect: bool, login_failed_once: bool, } #[derive(Debug)] struct OAuth2 { user: String, access_token: String, } impl async_imap::Authenticator for OAuth2 { type Response = String; fn process(&self, _data: &[u8]) -> Self::Response { format!( "user={}\x01auth=Bearer {}\x01\x01", self.user, self.access_token ) } } #[derive(Debug, PartialEq)] enum FolderMeaning { Unknown, SentObjects, Other, } #[derive(Debug)] struct ImapConfig { pub addr: String, pub imap_server: String, pub imap_port: u16, pub imap_user: String, pub imap_pw: String, pub strict_tls: bool, pub server_flags: usize, pub selected_folder: Option, pub selected_mailbox: Option, pub selected_folder_needs_expunge: bool, pub can_idle: bool, /// True if the server has MOVE capability as defined in /// https://tools.ietf.org/html/rfc6851 pub can_move: bool, } impl Default for ImapConfig { fn default() -> Self { ImapConfig { addr: "".into(), imap_server: "".into(), imap_port: 0, imap_user: "".into(), imap_pw: "".into(), strict_tls: false, server_flags: 0, selected_folder: None, selected_mailbox: None, selected_folder_needs_expunge: false, can_idle: false, can_move: false, } } } impl Imap { pub fn new(idle_interrupt: Receiver) -> Self { Imap { idle_interrupt, config: Default::default(), session: Default::default(), connected: Default::default(), interrupt: Default::default(), should_reconnect: Default::default(), login_failed_once: Default::default(), } } pub fn is_connected(&self) -> bool { self.connected } pub fn should_reconnect(&self) -> bool { self.should_reconnect } pub fn trigger_reconnect(&mut self) { self.should_reconnect = true; } async fn setup_handle_if_needed(&mut self, context: &Context) -> Result<()> { if self.config.imap_server.is_empty() { return Err(Error::InTeardown); } if self.should_reconnect() { self.unsetup_handle(context).await; self.should_reconnect = false; } else if self.is_connected() { return Ok(()); } let server_flags = self.config.server_flags as i32; let connection_res: ImapResult = if (server_flags & (DC_LP_IMAP_SOCKET_STARTTLS | DC_LP_IMAP_SOCKET_PLAIN)) != 0 { let config = &mut self.config; let imap_server: &str = config.imap_server.as_ref(); let imap_port = config.imap_port; match Client::connect_insecure((imap_server, imap_port)).await { Ok(client) => { if (server_flags & DC_LP_IMAP_SOCKET_STARTTLS) != 0 { client.secure(imap_server, config.strict_tls).await } else { Ok(client) } } Err(err) => Err(err), } } else { let config = &self.config; let imap_server: &str = config.imap_server.as_ref(); let imap_port = config.imap_port; Client::connect_secure((imap_server, imap_port), imap_server, config.strict_tls) .await }; let login_res = match connection_res { Ok(client) => { let config = &self.config; let imap_user: &str = config.imap_user.as_ref(); let imap_pw: &str = config.imap_pw.as_ref(); if (server_flags & DC_LP_AUTH_OAUTH2) != 0 { let addr: &str = config.addr.as_ref(); if let Some(token) = dc_get_oauth2_access_token(context, addr, imap_pw, true).await { let auth = OAuth2 { user: imap_user.into(), access_token: token, }; client.authenticate("XOAUTH2", &auth).await } else { return Err(Error::OauthError); } } else { client.login(imap_user, imap_pw).await } } Err(err) => { let message = { let config = &self.config; let imap_server: &str = config.imap_server.as_ref(); let imap_port = config.imap_port; context .stock_string_repl_str2( StockMessage::ServerResponse, format!("IMAP {}:{}", imap_server, imap_port), err.to_string(), ) .await }; // IMAP connection failures are reported to users emit_event!(context, Event::ErrorNetwork(message)); return Err(Error::ConnectionFailed(err.to_string())); } }; self.should_reconnect = false; match login_res { Ok(session) => { // needs to be set here to ensure it is set on reconnects. self.connected = true; self.session = Some(session); self.login_failed_once = false; Ok(()) } Err((err, _)) => { let imap_user = self.config.imap_user.to_owned(); let message = context .stock_string_repl_str(StockMessage::CannotLogin, &imap_user) .await; warn!(context, "{} ({})", message, err); emit_event!(context, Event::ErrorNetwork(message.clone())); let lock = context.wrong_pw_warning_mutex.lock().await; if self.login_failed_once && context.get_config_bool(Config::NotifyAboutWrongPw).await { if let Err(e) = context.set_config(Config::NotifyAboutWrongPw, None).await { warn!(context, "{}", e); } drop(lock); let mut msg = Message::new(Viewtype::Text); msg.text = Some(message); if let Err(e) = chat::add_device_msg_with_importance(context, None, Some(&mut msg), true) .await { warn!(context, "{}", e); } } else { self.login_failed_once = true; } self.trigger_reconnect(); Err(Error::LoginFailed(format!("cannot login as {}", imap_user))) } } } async fn unsetup_handle(&mut self, context: &Context) { // Close folder if messages should be expunged if let Err(err) = self.close_folder(context).await { warn!(context, "failed to close folder: {:?}", err); } // Logout from the server if let Some(mut session) = self.session.take() { if let Err(err) = session.logout().await { warn!(context, "failed to logout: {:?}", err); } } self.connected = false; self.config.selected_folder = None; self.config.selected_mailbox = None; } async fn free_connect_params(&mut self) { let mut cfg = &mut self.config; cfg.addr = "".into(); cfg.imap_server = "".into(); cfg.imap_user = "".into(); cfg.imap_pw = "".into(); cfg.imap_port = 0; cfg.can_idle = false; cfg.can_move = false; } /// Connects to imap account using already-configured parameters. pub async fn connect_configured(&mut self, context: &Context) -> Result<()> { if self.is_connected() && !self.should_reconnect() { return Ok(()); } if !context.is_configured().await { return Err(Error::ConnectWithoutConfigure); } let param = LoginParam::from_database(context, "configured_").await; // the trailing underscore is correct if self.connect(context, ¶m).await { self.ensure_configured_folders(context, true).await } else { Err(Error::ConnectionFailed(format!("{}", param))) } } /// Tries connecting to imap account using the specific login parameters. pub async fn connect(&mut self, context: &Context, lp: &LoginParam) -> bool { if lp.mail_server.is_empty() || lp.mail_user.is_empty() || lp.mail_pw.is_empty() { return false; } { let addr = &lp.addr; let imap_server = &lp.mail_server; let imap_port = lp.mail_port as u16; let imap_user = &lp.mail_user; let imap_pw = &lp.mail_pw; let server_flags = lp.server_flags as usize; let mut config = &mut self.config; config.addr = addr.to_string(); config.imap_server = imap_server.to_string(); config.imap_port = imap_port; config.imap_user = imap_user.to_string(); config.imap_pw = imap_pw.to_string(); let provider = get_provider_info(&lp.addr); config.strict_tls = match lp.imap_certificate_checks { CertificateChecks::Automatic => { provider.map_or(false, |provider| provider.strict_tls) } CertificateChecks::Strict => true, CertificateChecks::AcceptInvalidCertificates | CertificateChecks::AcceptInvalidCertificates2 => false, }; config.server_flags = server_flags; } if let Err(err) = self.setup_handle_if_needed(context).await { warn!(context, "failed to setup imap handle: {}", err); self.free_connect_params().await; return false; } let teardown = match &mut self.session { Some(ref mut session) => match session.capabilities().await { Ok(caps) => { if !context.sql.is_open().await { warn!(context, "IMAP-LOGIN as {} ok but ABORTING", lp.mail_user,); true } else { let can_idle = caps.has_str("IDLE"); let can_move = caps.has_str("MOVE"); let caps_list = caps.iter().fold(String::new(), |s, c| { if let Capability::Atom(x) = c { s + &format!(" {}", x) } else { s + &format!(" {:?}", c) } }); self.config.can_idle = can_idle; self.config.can_move = can_move; self.connected = true; emit_event!( context, Event::ImapConnected(format!( "IMAP-LOGIN as {}, capabilities: {}", lp.mail_user, caps_list, )) ); false } } Err(err) => { info!(context, "CAPABILITY command error: {}", err); true } }, None => true, }; if teardown { self.disconnect(context).await; false } else { true } } pub async fn disconnect(&mut self, context: &Context) { self.unsetup_handle(context).await; self.free_connect_params().await; } pub async fn fetch(&mut self, context: &Context, watch_folder: &str) -> Result<()> { if !context.sql.is_open().await { // probably shutdown return Err(Error::InTeardown); } self.setup_handle_if_needed(context).await?; while self.fetch_new_messages(context, &watch_folder).await? { // We fetch until no more new messages are there. } Ok(()) } async fn get_config_last_seen_uid>( &self, context: &Context, folder: S, ) -> (u32, u32) { let key = format!("imap.mailbox.{}", folder.as_ref()); if let Some(entry) = context.sql.get_raw_config(context, &key).await { // the entry has the format `imap.mailbox.=:` let mut parts = entry.split(':'); ( parts .next() .unwrap_or_default() .parse() .unwrap_or_else(|_| 0), parts .next() .unwrap_or_default() .parse() .unwrap_or_else(|_| 0), ) } else { (0, 0) } } /// return Result with (uid_validity, last_seen_uid) tuple. pub(crate) async fn select_with_uidvalidity( &mut self, context: &Context, folder: &str, ) -> Result<(u32, u32)> { self.select_folder(context, Some(folder)).await?; // compare last seen UIDVALIDITY against the current one let (uid_validity, last_seen_uid) = self.get_config_last_seen_uid(context, &folder).await; let config = &mut self.config; let mailbox = config .selected_mailbox .as_ref() .ok_or_else(|| Error::NoMailbox(folder.to_string()))?; let new_uid_validity = match mailbox.uid_validity { Some(v) => v, None => { let s = format!("No UIDVALIDITY for folder {:?}", folder); return Err(Error::Other(s)); } }; if new_uid_validity == uid_validity { return Ok((uid_validity, last_seen_uid)); } if mailbox.exists == 0 { info!(context, "Folder \"{}\" is empty.", folder); // set lastseenuid=0 for empty folders. // id we do not do this here, we'll miss the first message // as we will get in here again and fetch from lastseenuid+1 then self.set_config_last_seen_uid(context, &folder, new_uid_validity, 0) .await; return Ok((new_uid_validity, 0)); } // uid_validity has changed or is being set the first time. // find the last seen uid within the new uid_validity scope. let new_last_seen_uid = match mailbox.uid_next { Some(uid_next) => { uid_next - 1 // XXX could uid_next be 0? } None => { warn!( context, "IMAP folder has no uid_next, fall back to fetching" ); if let Some(ref mut session) = &mut self.session { // note that we use fetch by sequence number // and thus we only need to get exactly the // last-index message. let set = format!("{}", mailbox.exists); match session.fetch(set, JUST_UID).await { Ok(mut list) => { if let Some(Ok(msg)) = list.next().await { msg.uid.unwrap_or_default() } else { return Err(Error::Other("failed to fetch".into())); } } Err(err) => { return Err(Error::FetchFailed(err)); } } } else { return Err(Error::NoConnection); } } }; self.set_config_last_seen_uid(context, &folder, new_uid_validity, new_last_seen_uid) .await; info!( context, "uid/validity change: new {}/{} current {}/{}", new_last_seen_uid, new_uid_validity, uid_validity, last_seen_uid ); Ok((new_uid_validity, new_last_seen_uid)) } async fn fetch_new_messages>( &mut self, context: &Context, folder: S, ) -> Result { let show_emails = ShowEmails::from_i32(context.get_config_int(Config::ShowEmails).await) .unwrap_or_default(); let (uid_validity, last_seen_uid) = self .select_with_uidvalidity(context, folder.as_ref()) .await?; let msgs = self.fetch_after(context, last_seen_uid).await?; let read_cnt = msgs.len(); let folder: &str = folder.as_ref(); let mut read_errors = 0; let mut uids = Vec::with_capacity(msgs.len()); let mut new_last_seen_uid = None; for (current_uid, msg) in msgs.into_iter() { let (headers, msg_id) = match get_fetch_headers(&msg) { Ok(headers) => { let msg_id = prefetch_get_message_id(&headers).unwrap_or_default(); (headers, msg_id) } Err(err) => { warn!(context, "{}", err); read_errors += 1; continue; } }; if message_needs_processing( context, current_uid, &headers, &msg_id, folder, show_emails, ) .await { // Trigger download and processing for this message. uids.push(current_uid); } else if read_errors == 0 { // No errors so far, but this was skipped, so mark as last_seen_uid new_last_seen_uid = Some(current_uid); } } // check passed, go fetch the emails let (new_last_seen_uid_processed, error_cnt) = self.fetch_many_msgs(context, &folder, &uids).await; read_errors += error_cnt; // determine which last_seen_uid to use to update to let new_last_seen_uid_processed = new_last_seen_uid_processed.unwrap_or_default(); let new_last_seen_uid = new_last_seen_uid.unwrap_or_default(); let last_one = new_last_seen_uid.max(new_last_seen_uid_processed); if last_one > last_seen_uid { self.set_config_last_seen_uid(context, &folder, uid_validity, last_one) .await; } if read_errors == 0 { info!(context, "{} mails read from \"{}\".", read_cnt, folder,); } else { warn!( context, "{} mails read from \"{}\" with {} errors.", read_cnt, folder, read_errors ); } Ok(read_cnt > 0) } /// Fetch all uids larger than the passed in. Returns a sorted list of fetch results. async fn fetch_after( &mut self, context: &Context, uid: u32, ) -> Result> { if self.session.is_none() { return Err(Error::NoConnection); } let session = self.session.as_mut().unwrap(); // fetch messages with larger UID than the last one seen // `(UID FETCH lastseenuid+1:*)`, see RFC 4549 let set = format!("{}:*", uid + 1); let mut list = session .uid_fetch(set, PREFETCH_FLAGS) .await .map_err(Error::FetchFailed)?; let mut msgs = BTreeMap::new(); while let Some(fetch) = list.next().await { let msg = fetch.map_err(|err| Error::Other(err.to_string()))?; if let Some(msg_uid) = msg.uid { msgs.insert(msg_uid, msg); } } drop(list); // If the mailbox is not empty, results always include // at least one UID, even if last_seen_uid+1 is past // the last UID in the mailbox. It happens because // uid+1:* is interpreted the same way as *:uid+1. // See https://tools.ietf.org/html/rfc3501#page-61 for // standard reference. Therefore, sometimes we receive // already seen messages and have to filter them out. let new_msgs = msgs.split_off(&(uid + 1)); for current_uid in msgs.keys() { info!( context, "fetch_new_messages: ignoring uid {}, last seen was {}", current_uid, uid ); } Ok(new_msgs) } async fn set_config_last_seen_uid>( &self, context: &Context, folder: S, uidvalidity: u32, lastseenuid: u32, ) { let key = format!("imap.mailbox.{}", folder.as_ref()); let val = format!("{}:{}", uidvalidity, lastseenuid); context .sql .set_raw_config(context, &key, Some(&val)) .await .ok(); } /// Fetches a list of messages by server UID. /// The passed in list of uids must be sorted. /// /// Returns the last uid fetch successfully and an error count. async fn fetch_many_msgs>( &mut self, context: &Context, folder: S, server_uids: &[u32], ) -> (Option, usize) { let set = match server_uids { [] => return (None, 0), [server_uid] => server_uid.to_string(), [first_uid, .., last_uid] => { // XXX: it is assumed that UIDs are sorted and // contiguous. If UIDs are not contiguous, more // messages than needed will be downloaded. debug_assert!(first_uid < last_uid, "uids must be sorted"); format!("{}:{}", first_uid, last_uid) } }; if !self.is_connected() { warn!(context, "Not connected"); return (None, server_uids.len()); } if self.session.is_none() { // we could not get a valid imap session, this should be retried self.trigger_reconnect(); warn!(context, "Could not get IMAP session"); return (None, server_uids.len()); } let session = self.session.as_mut().unwrap(); let mut msgs = match session.uid_fetch(&set, BODY_FLAGS).await { Ok(msgs) => msgs, Err(err) => { // TODO: maybe differentiate between IO and input/parsing problems // so we don't reconnect if we have a (rare) input/output parsing problem? self.should_reconnect = true; warn!( context, "Error on fetching messages #{} from folder \"{}\"; error={}.", &set, folder.as_ref(), err ); return (None, server_uids.len()); } }; let folder = folder.as_ref().to_string(); let mut read_errors = 0; let mut last_uid = None; let mut count = 0; while let Some(Ok(msg)) = msgs.next().await { let server_uid = msg.uid.unwrap_or_default(); if !server_uids.contains(&server_uid) { // skip if there are some in between we are not interested in continue; } count += 1; let is_deleted = msg.flags().any(|flag| flag == Flag::Deleted); if is_deleted || msg.body().is_none() { // No need to process these. continue; } // XXX put flags into a set and pass them to dc_receive_imf let context = context.clone(); let folder = folder.clone(); // safe, as we checked above that there is a body. let body = msg.body().unwrap(); let is_seen = msg.flags().any(|flag| flag == Flag::Seen); match dc_receive_imf(&context, &body, &folder, server_uid, is_seen).await { Ok(_) => last_uid = Some(server_uid), Err(err) => { warn!(context, "dc_receive_imf error: {}", err); read_errors += 1; } }; } if count != server_uids.len() { warn!( context, "failed to fetch all uids: got {}, requested {}", count, server_uids.len() ); } (last_uid, read_errors) } pub async fn can_move(&self) -> bool { self.config.can_move } pub async fn mv( &mut self, context: &Context, folder: &str, uid: u32, dest_folder: &str, ) -> ImapActionResult { if folder == dest_folder { info!( context, "Skip moving message; message {}/{} is already in {}...", folder, uid, dest_folder, ); return ImapActionResult::AlreadyDone; } if let Some(imapresult) = self .prepare_imap_operation_on_msg(context, folder, uid) .await { return imapresult; } // we are connected, and the folder is selected let set = format!("{}", uid); let display_folder_id = format!("{}/{}", folder, uid); if self.can_move().await { if let Some(ref mut session) = &mut self.session { match session.uid_mv(&set, &dest_folder).await { Ok(_) => { emit_event!( context, Event::ImapMessageMoved(format!( "IMAP Message {} moved to {}", display_folder_id, dest_folder )) ); return ImapActionResult::Success; } Err(err) => { warn!( context, "Cannot move message, fallback to COPY/DELETE {}/{} to {}: {}", folder, uid, dest_folder, err ); } } } else { unreachable!(); }; } else { info!( context, "Server does not support MOVE, fallback to COPY/DELETE {}/{} to {}", folder, uid, dest_folder ); } if let Some(ref mut session) = &mut self.session { if let Err(err) = session.uid_copy(&set, &dest_folder).await { warn!(context, "Could not copy message: {}", err); return ImapActionResult::Failed; } } else { unreachable!(); } if !self.add_flag_finalized(context, uid, "\\Deleted").await { warn!(context, "Cannot mark {} as \"Deleted\" after copy.", uid); emit_event!( context, Event::ImapMessageMoved(format!( "IMAP Message {} copied to {} (delete FAILED)", display_folder_id, dest_folder )) ); ImapActionResult::Failed } else { self.config.selected_folder_needs_expunge = true; emit_event!( context, Event::ImapMessageMoved(format!( "IMAP Message {} copied to {} (delete successfull)", display_folder_id, dest_folder )) ); ImapActionResult::Success } } async fn add_flag_finalized(&mut self, context: &Context, server_uid: u32, flag: &str) -> bool { // return true if we successfully set the flag or we otherwise // think add_flag should not be retried: Disconnection during setting // the flag, or other imap-errors, returns true as well. // // returning false means that the operation can be retried. if server_uid == 0 { return true; // might be moved but we don't want to have a stuck job } let s = server_uid.to_string(); self.add_flag_finalized_with_set(context, &s, flag).await } async fn add_flag_finalized_with_set( &mut self, context: &Context, uid_set: &str, flag: &str, ) -> bool { if self.should_reconnect() { return false; } if let Some(ref mut session) = &mut self.session { let query = format!("+FLAGS ({})", flag); match session.uid_store(uid_set, &query).await { Ok(_) => {} Err(err) => { warn!( context, "IMAP failed to store: ({}, {}) {:?}", uid_set, query, err ); } } true // we tried once, that's probably enough for setting flag } else { unreachable!(); } } pub async fn prepare_imap_operation_on_msg( &mut self, context: &Context, folder: &str, uid: u32, ) -> Option { if uid == 0 { return Some(ImapActionResult::RetryLater); } if !self.is_connected() { // currently jobs are only performed on the INBOX thread // TODO: make INBOX/SENT/MVBOX perform the jobs on their // respective folders to avoid select_folder network traffic // and the involved error states if let Err(err) = self.connect_configured(context).await { warn!(context, "prepare_imap_op failed: {}", err); return Some(ImapActionResult::RetryLater); } } match self.select_folder(context, Some(&folder)).await { Ok(()) => None, Err(select_folder::Error::ConnectionLost) => { warn!(context, "Lost imap connection"); Some(ImapActionResult::RetryLater) } Err(select_folder::Error::NoSession) => { warn!(context, "no imap session"); Some(ImapActionResult::Failed) } Err(select_folder::Error::BadFolderName(folder_name)) => { warn!(context, "invalid folder name: {:?}", folder_name); Some(ImapActionResult::Failed) } Err(err) => { warn!(context, "failed to select folder: {:?}: {:?}", folder, err); Some(ImapActionResult::RetryLater) } } } pub async fn set_seen( &mut self, context: &Context, folder: &str, uid: u32, ) -> ImapActionResult { if let Some(imapresult) = self .prepare_imap_operation_on_msg(context, folder, uid) .await { return imapresult; } // we are connected, and the folder is selected info!(context, "Marking message {}/{} as seen...", folder, uid,); if self.add_flag_finalized(context, uid, "\\Seen").await { ImapActionResult::Success } else { warn!( context, "Cannot mark message {} in folder {} as seen, ignoring.", uid, folder ); ImapActionResult::Failed } } pub async fn delete_msg( &mut self, context: &Context, message_id: &str, folder: &str, uid: u32, ) -> ImapActionResult { if let Some(imapresult) = self .prepare_imap_operation_on_msg(context, folder, uid) .await { return imapresult; } // we are connected, and the folder is selected let set = format!("{}", uid); let display_imap_id = format!("{}/{}", folder, uid); // double-check that we are deleting the correct message-id // this comes at the expense of another imap query if let Some(ref mut session) = &mut self.session { match session.uid_fetch(set, DELETE_CHECK_FLAGS).await { Ok(mut msgs) => { let fetch = if let Some(Ok(fetch)) = msgs.next().await { fetch } else { warn!( context, "Cannot delete on IMAP, {}: imap entry gone '{}'", display_imap_id, message_id, ); return ImapActionResult::AlreadyDone; }; let remote_message_id = get_fetch_headers(&fetch) .and_then(|headers| prefetch_get_message_id(&headers)) .unwrap_or_default(); if remote_message_id != message_id { warn!( context, "Cannot delete on IMAP, {}: remote message-id '{}' != '{}'", display_imap_id, remote_message_id, message_id, ); return ImapActionResult::Failed; } } Err(err) => { warn!( context, "Cannot delete on IMAP, {}: {}", display_imap_id, err, ); return ImapActionResult::RetryLater; } } } // mark the message for deletion if !self.add_flag_finalized(context, uid, "\\Deleted").await { warn!( context, "Cannot mark message {} as \"Deleted\".", display_imap_id ); ImapActionResult::RetryLater } else { emit_event!( context, Event::ImapMessageDeleted(format!( "IMAP Message {} marked as deleted [{}]", display_imap_id, message_id )) ); self.config.selected_folder_needs_expunge = true; ImapActionResult::Success } } pub async fn ensure_configured_folders( &mut self, context: &Context, create_mvbox: bool, ) -> Result<()> { let folders_configured = context .sql .get_raw_config_int(context, "folders_configured") .await; if folders_configured.unwrap_or_default() >= DC_FOLDERS_CONFIGURED_VERSION { return Ok(()); } self.configure_folders(context, create_mvbox).await } pub async fn configure_folders(&mut self, context: &Context, create_mvbox: bool) -> Result<()> { if !self.is_connected() { return Err(Error::NoConnection); } if let Some(ref mut session) = &mut self.session { let mut folders = match session.list(Some(""), Some("*")).await { Ok(f) => f, Err(err) => { return Err(Error::Other(format!("list_folders failed {:?}", err))); } }; let mut delimiter = ".".to_string(); let mut delimiter_is_default = true; let mut sentbox_folder = None; let mut mvbox_folder = None; let mut fallback_folder = get_fallback_folder(&delimiter); while let Some(folder) = folders.next().await { let folder = folder.map_err(|err| Error::Other(err.to_string()))?; info!(context, "Scanning folder: {:?}", folder); // Update the delimiter iff there is a different one, but only once. if let Some(d) = folder.delimiter() { if delimiter_is_default && !d.is_empty() && delimiter != d { delimiter = d.to_string(); fallback_folder = get_fallback_folder(&delimiter); delimiter_is_default = false; } } if folder.name() == "DeltaChat" { // Always takes precendent mvbox_folder = Some(folder.name().to_string()); } else if folder.name() == fallback_folder { // only set iff none has been already set if mvbox_folder.is_none() { mvbox_folder = Some(folder.name().to_string()); } } else if let FolderMeaning::SentObjects = get_folder_meaning(&folder) { // Always takes precedent sentbox_folder = Some(folder.name().to_string()); } else if let FolderMeaning::SentObjects = get_folder_meaning_by_name(&folder) { // only set iff none has been already set if sentbox_folder.is_none() { sentbox_folder = Some(folder.name().to_string()); } } } drop(folders); info!(context, "Using \"{}\" as folder-delimiter.", delimiter); info!(context, "sentbox folder is {:?}", sentbox_folder); if mvbox_folder.is_none() && create_mvbox { info!(context, "Creating MVBOX-folder \"DeltaChat\"...",); match session.create("DeltaChat").await { Ok(_) => { mvbox_folder = Some("DeltaChat".into()); info!(context, "MVBOX-folder created.",); } Err(err) => { warn!( context, "Cannot create MVBOX-folder, trying to create INBOX subfolder. ({})", err ); match session.create(&fallback_folder).await { Ok(_) => { mvbox_folder = Some(fallback_folder); info!( context, "MVBOX-folder created as INBOX subfolder. ({})", err ); } Err(err) => { warn!(context, "Cannot create MVBOX-folder. ({})", err); } } } } // SUBSCRIBE is needed to make the folder visible to the LSUB command // that may be used by other MUAs to list folders. // for the LIST command, the folder is always visible. if let Some(ref mvbox) = mvbox_folder { if let Err(err) = session.subscribe(mvbox).await { warn!(context, "could not subscribe to {:?}: {:?}", mvbox, err); } } } context .set_config(Config::ConfiguredInboxFolder, Some("INBOX")) .await?; if let Some(ref mvbox_folder) = mvbox_folder { context .set_config(Config::ConfiguredMvboxFolder, Some(mvbox_folder)) .await?; } if let Some(ref sentbox_folder) = sentbox_folder { context .set_config(Config::ConfiguredSentboxFolder, Some(sentbox_folder)) .await?; } context .sql .set_raw_config_int(context, "folders_configured", DC_FOLDERS_CONFIGURED_VERSION) .await?; } info!(context, "FINISHED configuring IMAP-folders."); Ok(()) } pub async fn empty_folder(&mut self, context: &Context, folder: &str) { info!(context, "emptying folder {}", folder); // we want to report all error to the user // (no retry should be attempted) if folder.is_empty() { error!(context, "cannot perform empty, folder not set"); return; } if let Err(err) = self.setup_handle_if_needed(context).await { error!(context, "could not setup imap connection: {:?}", err); return; } if let Err(err) = self.select_folder(context, Some(&folder)).await { error!( context, "Could not select {} for expunging: {:?}", folder, err ); return; } if !self .add_flag_finalized_with_set(context, SELECT_ALL, "\\Deleted") .await { error!(context, "Cannot mark messages for deletion {}", folder); return; } // we now trigger expunge to actually delete messages self.config.selected_folder_needs_expunge = true; match self.select_folder::(context, None).await { Ok(()) => { emit_event!(context, Event::ImapFolderEmptied(folder.to_string())); } Err(err) => { error!(context, "expunge failed {}: {:?}", folder, err); } } if let Err(err) = context .sql .execute( "UPDATE msgs SET server_folder='',server_uid=0 WHERE server_folder=?", paramsv![folder], ) .await { warn!( context, "Failed to reset server_uid and server_folder for deleted messages: {}", err ); } } } /// Try to get the folder meaning by the name of the folder only used if the server does not support XLIST. // TODO: lots languages missing - maybe there is a list somewhere on other MUAs? // however, if we fail to find out the sent-folder, // only watching this folder is not working. at least, this is no show stopper. // CAVE: if possible, take care not to add a name here that is "sent" in one language // but sth. different in others - a hard job. fn get_folder_meaning_by_name(folder_name: &Name) -> FolderMeaning { let sent_names = vec!["sent", "sentmail", "sent objects", "gesendet"]; let lower = folder_name.name().to_lowercase(); if sent_names.into_iter().any(|s| s == lower) { FolderMeaning::SentObjects } else { FolderMeaning::Unknown } } fn get_folder_meaning(folder_name: &Name) -> FolderMeaning { let special_names = vec!["\\Spam", "\\Trash", "\\Drafts", "\\Junk"]; for attr in folder_name.attributes() { if let NameAttribute::Custom(ref label) = attr { if special_names.iter().any(|s| *s == label) { return FolderMeaning::Other; } else if label == "\\Sent" { return FolderMeaning::SentObjects; } } } FolderMeaning::Unknown } async fn precheck_imf( context: &Context, rfc724_mid: &str, server_folder: &str, server_uid: u32, ) -> Result { if let Some((old_server_folder, old_server_uid, msg_id)) = message::rfc724_mid_exists(context, &rfc724_mid).await? { if old_server_folder.is_empty() && old_server_uid == 0 { info!( context, "[move] detected bcc-self {} as {}/{}", rfc724_mid, server_folder, server_uid ); let delete_server_after = context.get_config_delete_server_after().await; if delete_server_after != Some(0) { if msg_id .needs_move(context, server_folder) .await .unwrap_or_default() { // If the bcc-self message is not moved, directly // add MarkSeen job, otherwise MarkSeen job is // added after the Move Job completed. job::add( context, job::Job::new(Action::MoveMsg, msg_id.to_u32(), Params::new(), 0), ) .await; } else { job::add( context, job::Job::new(Action::MarkseenMsgOnImap, msg_id.to_u32(), Params::new(), 0), ) .await; } } } else if old_server_folder != server_folder { info!( context, "[move] detected message {} moved by other device from {}/{} to {}/{}", rfc724_mid, old_server_folder, old_server_uid, server_folder, server_uid ); } else if old_server_uid == 0 { info!( context, "[move] detected message {} moved by us from {}/{} to {}/{}", rfc724_mid, old_server_folder, old_server_uid, server_folder, server_uid ); } else if old_server_uid != server_uid { warn!( context, "UID for message {} in folder {} changed from {} to {}", rfc724_mid, server_folder, old_server_uid, server_uid ); } if old_server_folder != server_folder || old_server_uid != server_uid { update_server_uid(context, rfc724_mid, server_folder, server_uid).await; if let Ok(MessageState::InSeen) = msg_id.get_state(context).await { job::add( context, job::Job::new(Action::MarkseenMsgOnImap, msg_id.to_u32(), Params::new(), 0), ) .await; }; context .interrupt_inbox(InterruptInfo::new(false, Some(msg_id))) .await; info!(context, "Updating server_uid and interrupting") } Ok(true) } else { Ok(false) } } fn get_fetch_headers(prefetch_msg: &Fetch) -> Result> { let header_bytes = match prefetch_msg.header() { Some(header_bytes) => header_bytes, None => return Ok(Vec::new()), }; let (headers, _) = mailparse::parse_headers(header_bytes)?; Ok(headers) } fn prefetch_get_message_id(headers: &[mailparse::MailHeader]) -> Result { if let Some(message_id) = headers.get_header_value(HeaderDef::MessageId) { Ok(crate::mimeparser::parse_message_id(&message_id)?) } else { Err(Error::Other("prefetch: No message ID found".to_string())) } } async fn prefetch_is_reply_to_chat_message( context: &Context, headers: &[mailparse::MailHeader<'_>], ) -> bool { if let Some(value) = headers.get_header_value(HeaderDef::InReplyTo) { if is_msgrmsg_rfc724_mid_in_list(context, &value).await { return true; } } if let Some(value) = headers.get_header_value(HeaderDef::References) { if is_msgrmsg_rfc724_mid_in_list(context, &value).await { return true; } } false } pub(crate) async fn prefetch_should_download( context: &Context, headers: &[mailparse::MailHeader<'_>], show_emails: ShowEmails, ) -> Result { let is_chat_message = headers.get_header_value(HeaderDef::ChatVersion).is_some(); let is_reply_to_chat_message = prefetch_is_reply_to_chat_message(context, &headers).await; let maybe_ndn = if let Some(from) = headers.get_header_value(HeaderDef::From_) { let from = from.to_ascii_lowercase(); from.contains("mailer-daemon") || from.contains("mail-daemon") } else { false }; // Autocrypt Setup Message should be shown even if it is from non-chat client. let is_autocrypt_setup_message = headers .get_header_value(HeaderDef::AutocryptSetupMessage) .is_some(); let (_contact_id, blocked_contact, origin) = from_field_to_contact_id(context, &mimeparser::get_from(headers)).await?; let accepted_contact = origin.is_known(); let show = is_autocrypt_setup_message || maybe_ndn || match show_emails { ShowEmails::Off => is_chat_message || is_reply_to_chat_message, ShowEmails::AcceptedContacts => { is_chat_message || is_reply_to_chat_message || accepted_contact } ShowEmails::All => true, }; let show = show && !blocked_contact; Ok(show) } async fn message_needs_processing( context: &Context, current_uid: u32, headers: &[mailparse::MailHeader<'_>], msg_id: &str, folder: &str, show_emails: ShowEmails, ) -> bool { let skip = match precheck_imf(context, &msg_id, folder, current_uid).await { Ok(skip) => skip, Err(err) => { warn!(context, "precheck_imf error: {}", err); true } }; if skip { // we know the message-id already or don't want the message otherwise. info!( context, "Skipping message {} from \"{}\" by precheck.", msg_id, folder, ); return false; } // we do not know the message-id // or the message-id is missing (in this case, we create one in the further process) // or some other error happened let show = match prefetch_should_download(context, &headers, show_emails).await { Ok(show) => show, Err(err) => { warn!(context, "prefetch_should_download error: {}", err); true } }; if !show { info!( context, "Ignoring new message {} from \"{}\".", msg_id, folder, ); return false; } true } fn get_fallback_folder(delimiter: &str) -> String { format!("INBOX{}DeltaChat", delimiter) }