chatmail-core/src/peerstate.rs

use std::collections::HashSet;
use std::ffi::CString;
use std::fmt;

use num_traits::FromPrimitive;

use crate::aheader::*;
use crate::constants::*;
use crate::context::Context;
use crate::dc_chat::*;
use crate::dc_sqlite3::*;
use crate::dc_tools::{to_cstring, to_string};
use crate::key::*;
use crate::types::*;

/// Peerstate represents the state of an Autocrypt peer.
pub struct Peerstate<'a> {
    pub context: &'a Context,
    pub addr: Option<String>,
    pub last_seen: u64,
    pub last_seen_autocrypt: u64,
    pub prefer_encrypt: EncryptPreference,
    pub public_key: Option<Key>,
    pub public_key_fingerprint: Option<String>,
    pub gossip_key: Option<Key>,
    pub gossip_timestamp: u64,
    pub gossip_key_fingerprint: Option<String>,
    verified_key: VerifiedKey,
    pub verified_key_fingerprint: Option<String>,
    pub to_save: Option<ToSave>,
    pub degrade_event: Option<DegradeEvent>,
}

impl<'a> PartialEq for Peerstate<'a> {
    fn eq(&self, other: &Peerstate) -> bool {
        self.addr == other.addr
            && self.last_seen == other.last_seen
            && self.last_seen_autocrypt == other.last_seen_autocrypt
            && self.prefer_encrypt == other.prefer_encrypt
            && self.public_key == other.public_key
            && self.public_key_fingerprint == other.public_key_fingerprint
            && self.gossip_key == other.gossip_key
            && self.gossip_timestamp == other.gossip_timestamp
            && self.gossip_key_fingerprint == other.gossip_key_fingerprint
            && self.verified_key == other.verified_key
            && self.verified_key_fingerprint == other.verified_key_fingerprint
            && self.to_save == other.to_save
            && self.degrade_event == other.degrade_event
    }
}

impl<'a> Eq for Peerstate<'a> {}

impl<'a> fmt::Debug for Peerstate<'a> {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.debug_struct("Peerstate")
            .field("addr", &self.addr)
            .field("last_seen", &self.last_seen)
            .field("last_seen_autocrypt", &self.last_seen_autocrypt)
            .field("prefer_encrypt", &self.prefer_encrypt)
            .field("public_key", &self.public_key)
            .field("public_key_fingerprint", &self.public_key_fingerprint)
            .field("gossip_key", &self.gossip_key)
            .field("gossip_timestamp", &self.gossip_timestamp)
            .field("gossip_key_fingerprint", &self.gossip_key_fingerprint)
            .field("verified_key", &self.verified_key)
            .field("verified_key_fingerprint", &self.verified_key_fingerprint)
            .field("to_save", &self.to_save)
            .field("degrade_event", &self.degrade_event)
            .finish()
    }
}

#[derive(Debug, Copy, Clone, PartialEq, Eq, FromPrimitive, ToPrimitive)]
#[repr(u8)]
pub enum ToSave {
    Timestamps = 0x01,
    All = 0x02,
}

#[derive(Debug, Copy, Clone, PartialEq, Eq, FromPrimitive, ToPrimitive)]
#[repr(u8)]
pub enum DegradeEvent {
    /// Recoverable by an incoming encrypted mail.
    EncryptionPaused = 0x01,
    /// Recoverable by a new verify.
    FingerprintChanged = 0x02,
}

#[derive(Debug, Copy, Clone, PartialEq, Eq)]
pub enum VerifiedKey {
    Gossip,
    Public,
    None,
}

impl Default for VerifiedKey {
    fn default() -> Self {
        VerifiedKey::None
    }
}

impl VerifiedKey {
    pub fn is_none(&self) -> bool {
        match self {
            VerifiedKey::None => true,
            _ => false,
        }
    }

    pub fn is_some(&self) -> bool {
        !self.is_none()
    }
}

impl<'a> Peerstate<'a> {
    pub fn new(context: &'a Context) -> Self {
        Peerstate {
            context,
            addr: None,
            last_seen: 0,
            last_seen_autocrypt: 0,
            prefer_encrypt: Default::default(),
            public_key: None,
            public_key_fingerprint: None,
            gossip_key: None,
            gossip_key_fingerprint: None,
            gossip_timestamp: 0,
            verified_key: Default::default(),
            verified_key_fingerprint: None,
            to_save: None,
            degrade_event: None,
        }
    }

    pub fn verified_key(&self) -> Option<&Key> {
        match self.verified_key {
            VerifiedKey::Public => self.public_key.as_ref(),
            VerifiedKey::Gossip => self.gossip_key.as_ref(),
            VerifiedKey::None => None,
        }
    }

    pub fn from_header(context: &'a Context, header: &Aheader, message_time: u64) -> Self {
        let mut res = Self::new(context);

        res.addr = Some(header.addr.clone());
        res.last_seen = message_time;
        res.last_seen_autocrypt = message_time;
        res.to_save = Some(ToSave::All);
        res.prefer_encrypt = header.prefer_encrypt;
        res.public_key = Some(header.public_key.clone());
        res.recalc_fingerprint();

        res
    }

    pub fn from_gossip(context: &'a Context, gossip_header: &Aheader, message_time: u64) -> Self {
        let mut res = Self::new(context);

        res.addr = Some(gossip_header.addr.clone());
        res.gossip_timestamp = message_time;
        res.to_save = Some(ToSave::All);
        res.gossip_key = Some(gossip_header.public_key.clone());
        res.recalc_fingerprint();

        res
    }

    pub fn from_addr(context: &'a Context, sql: &dc_sqlite3_t, addr: &str) -> Option<Self> {
        let mut res = None;

        let stmt = unsafe {
            dc_sqlite3_prepare(
                context,
                sql,
                b"SELECT addr, last_seen, last_seen_autocrypt, prefer_encrypted, public_key, gossip_timestamp, gossip_key, public_key_fingerprint, gossip_key_fingerprint, verified_key, verified_key_fingerprint FROM acpeerstates  WHERE addr=? COLLATE NOCASE;\x00"
                    as *const u8 as *const libc::c_char)
        };
        let addr_c = CString::new(addr.as_bytes()).unwrap();
        unsafe { sqlite3_bind_text(stmt, 1, addr_c.as_ptr(), -1, None) };
        if unsafe { sqlite3_step(stmt) } == 100 {
            res = Some(Self::from_stmt(context, stmt));
        }

        unsafe { sqlite3_finalize(stmt) };
        res
    }

    pub fn from_fingerprint(
        context: &'a Context,
        sql: &dc_sqlite3_t,
        fingerprint: &str,
    ) -> Option<Self> {
        let mut res = None;

        let stmt = unsafe {
            dc_sqlite3_prepare(
                context,
                sql,
                b"SELECT addr, last_seen, last_seen_autocrypt, prefer_encrypted, public_key, gossip_timestamp, gossip_key, public_key_fingerprint, gossip_key_fingerprint, verified_key, verified_key_fingerprint FROM acpeerstates  WHERE public_key_fingerprint=? COLLATE NOCASE     OR gossip_key_fingerprint=? COLLATE NOCASE  ORDER BY public_key_fingerprint=? DESC;\x00"
                    as *const u8 as *const libc::c_char)
        };

        let fp_c = CString::new(fingerprint.as_bytes()).unwrap();
        unsafe {
            sqlite3_bind_text(stmt, 1, fp_c.as_ptr(), -1, None);
            sqlite3_bind_text(stmt, 2, fp_c.as_ptr(), -1, None);
            sqlite3_bind_text(stmt, 3, fp_c.as_ptr(), -1, None);
        }
        if unsafe { sqlite3_step(stmt) == 100 } {
            res = Some(Self::from_stmt(context, stmt));
        }

        unsafe { sqlite3_finalize(stmt) };

        res
    }

    fn from_stmt(context: &'a Context, stmt: *mut sqlite3_stmt) -> Self {
        let mut res = Self::new(context);

        res.addr = Some(to_string(unsafe {
            sqlite3_column_text(stmt, 0) as *const _
        }));
        res.last_seen = unsafe { sqlite3_column_int64(stmt, 1) } as u64;
        res.last_seen_autocrypt = unsafe { sqlite3_column_int64(stmt, 2) } as u64;
        res.prefer_encrypt =
            EncryptPreference::from_i32(unsafe { sqlite3_column_int(stmt, 3) }).unwrap_or_default();
        res.gossip_timestamp = unsafe { sqlite3_column_int(stmt, 5) } as u64;
        let pkf = to_string(unsafe { sqlite3_column_text(stmt, 7) as *const _ });
        res.public_key_fingerprint = if pkf.is_empty() { None } else { Some(pkf) };
        let gkf = to_string(unsafe { sqlite3_column_text(stmt, 8) as *const _ });
        res.gossip_key_fingerprint = if gkf.is_empty() { None } else { Some(gkf) };
        let vkf = to_string(unsafe { sqlite3_column_text(stmt, 10) as *const _ });
        res.verified_key_fingerprint = if vkf.is_empty() { None } else { Some(vkf) };

        if unsafe { sqlite3_column_type(stmt, 4) } != 5 {
            res.public_key = Key::from_stmt(stmt, 4, KeyType::Public);
        }
        if unsafe { sqlite3_column_type(stmt, 6) } != 5 {
            res.gossip_key = Key::from_stmt(stmt, 6, KeyType::Public);
        }
        if unsafe { sqlite3_column_type(stmt, 9) } != 5 {
            let vk = Key::from_stmt(stmt, 9, KeyType::Public);
            res.verified_key = if vk == res.gossip_key {
                VerifiedKey::Gossip
            } else if vk == res.public_key {
                VerifiedKey::Public
            } else {
                VerifiedKey::None
            };
        }

        res
    }

    pub fn recalc_fingerprint(&mut self) {
        if let Some(ref public_key) = self.public_key {
            let old_public_fingerprint = self.public_key_fingerprint.take();
            self.public_key_fingerprint = Some(public_key.fingerprint());

            if old_public_fingerprint.is_none()
                || self.public_key_fingerprint.is_none()
                || old_public_fingerprint != self.public_key_fingerprint
            {
                self.to_save = Some(ToSave::All);
                if old_public_fingerprint.is_some() {
                    self.degrade_event = Some(DegradeEvent::FingerprintChanged);
                }
            }
        }

        if let Some(ref gossip_key) = self.gossip_key {
            let old_gossip_fingerprint = self.gossip_key_fingerprint.take();
            self.gossip_key_fingerprint = Some(gossip_key.fingerprint());

            if old_gossip_fingerprint.is_none()
                || self.gossip_key_fingerprint.is_none()
                || old_gossip_fingerprint != self.gossip_key_fingerprint
            {
                self.to_save = Some(ToSave::All);
                if old_gossip_fingerprint.is_some() {
                    self.degrade_event = Some(DegradeEvent::FingerprintChanged);
                }
            }
        }
    }

    pub fn degrade_encryption(&mut self, message_time: u64) {
        if self.prefer_encrypt == EncryptPreference::Mutual {
            self.degrade_event = Some(DegradeEvent::EncryptionPaused);
        }

        self.prefer_encrypt = EncryptPreference::Reset;
        self.last_seen = message_time;
        self.to_save = Some(ToSave::All);
    }

    pub fn apply_header(&mut self, header: &Aheader, message_time: u64) {
        if self.addr.is_none()
            || self.addr.as_ref().unwrap().to_lowercase() != header.addr.to_lowercase()
        {
            return;
        }

        if message_time > self.last_seen_autocrypt {
            self.last_seen = message_time;
            self.last_seen_autocrypt = message_time;
            self.to_save = Some(ToSave::Timestamps);
            if (header.prefer_encrypt == EncryptPreference::Mutual
                || header.prefer_encrypt == EncryptPreference::NoPreference)
                && header.prefer_encrypt != self.prefer_encrypt
            {
                if self.prefer_encrypt == EncryptPreference::Mutual
                    && header.prefer_encrypt != EncryptPreference::Mutual
                {
                    self.degrade_event = Some(DegradeEvent::EncryptionPaused);
                }
                self.prefer_encrypt = header.prefer_encrypt;
                self.to_save = Some(ToSave::All)
            }

            if self.public_key.as_ref() != Some(&header.public_key) {
                self.public_key = Some(header.public_key.clone());
                self.recalc_fingerprint();
                self.to_save = Some(ToSave::All);
            }
        }
    }

    pub fn apply_gossip(&mut self, gossip_header: &Aheader, message_time: u64) {
        if self.addr.is_none()
            || self.addr.as_ref().unwrap().to_lowercase() != gossip_header.addr.to_lowercase()
        {
            return;
        }

        if message_time > self.gossip_timestamp {
            self.gossip_timestamp = message_time;
            self.to_save = Some(ToSave::Timestamps);
            if self.gossip_key.as_ref() != Some(&gossip_header.public_key) {
                self.gossip_key = Some(gossip_header.public_key.clone());
                self.recalc_fingerprint();
                self.to_save = Some(ToSave::All)
            }
        };
    }

    pub fn render_gossip_header(&self, min_verified: usize) -> Option<String> {
        if let Some(ref addr) = self.addr {
            if let Some(key) = self.peek_key(min_verified) {
                // TODO: avoid cloning
                let header = Aheader::new(
                    addr.to_string(),
                    key.clone(),
                    EncryptPreference::NoPreference,
                );
                return Some(header.to_string());
            }
        }

        None
    }

    pub fn peek_key(&self, min_verified: usize) -> Option<&Key> {
        if self.public_key.is_none() && self.gossip_key.is_none() && self.verified_key.is_none() {
            return None;
        }

        if 0 != min_verified {
            return self.verified_key();
        }
        if self.public_key.is_some() {
            return self.public_key.as_ref();
        }

        self.gossip_key.as_ref()
    }

    pub fn set_verified(&mut self, which_key: usize, fingerprint: &str, verified: usize) -> bool {
        let mut success = false;
        if !(which_key != 0 && which_key != 1 || verified != 2) {
            if which_key == 1
                && self.public_key_fingerprint.is_some()
                && self.public_key_fingerprint.as_ref().unwrap() == fingerprint
            {
                self.to_save = Some(ToSave::All);
                self.verified_key = VerifiedKey::Public;
                self.verified_key_fingerprint = self.public_key_fingerprint.clone();
                success = true;
            }
            if which_key == 0
                && self.gossip_key_fingerprint.is_some()
                && self.gossip_key_fingerprint.as_ref().unwrap() == fingerprint
            {
                self.to_save = Some(ToSave::All);
                self.verified_key = VerifiedKey::Gossip;
                self.verified_key_fingerprint = self.gossip_key_fingerprint.clone();
                success = true;
            }
        }

        success
    }

    pub fn save_to_db(&self, sql: &dc_sqlite3_t, create: bool) -> bool {
        let mut success = false;

        if self.addr.is_none() {
            return success;
        }

        if create {
            let stmt = unsafe {
                dc_sqlite3_prepare(
                    self.context,
                    sql,
                    b"INSERT INTO acpeerstates (addr) VALUES(?);\x00" as *const u8
                        as *const libc::c_char,
                )
            };
            let addr_c = to_cstring(self.addr.as_ref().unwrap());
            unsafe {
                sqlite3_bind_text(stmt, 1, addr_c.as_ptr(), -1, None);
                sqlite3_step(stmt);
                sqlite3_finalize(stmt);
            }
        }

        if self.to_save == Some(ToSave::All) || create {
            let stmt = unsafe {
                dc_sqlite3_prepare(
                    self.context, sql,
		    b"UPDATE acpeerstates \
		      SET last_seen=?, last_seen_autocrypt=?, prefer_encrypted=?, \
		      public_key=?, gossip_timestamp=?, gossip_key=?, public_key_fingerprint=?, gossip_key_fingerprint=?, verified_key=?, verified_key_fingerprint=? \
		      WHERE addr=?;\x00"
                        as *const u8 as *const libc::c_char
                )
            };

            unsafe { sqlite3_bind_int64(stmt, 1, self.last_seen as sqlite3_int64) };
            unsafe { sqlite3_bind_int64(stmt, 2, self.last_seen_autocrypt as sqlite3_int64) };
            unsafe { sqlite3_bind_int64(stmt, 3, self.prefer_encrypt as sqlite3_int64) };

            let pub_bytes = self
                .public_key
                .as_ref()
                .map(|k| k.to_bytes())
                .unwrap_or_default();
            let gossip_bytes = self
                .gossip_key
                .as_ref()
                .map(|k| k.to_bytes())
                .unwrap_or_default();
            let ver_bytes = self
                .verified_key()
                .as_ref()
                .map(|k| k.to_bytes())
                .unwrap_or_default();

            let pkc = self
                .public_key_fingerprint
                .as_ref()
                .map(to_cstring)
                .unwrap_or_default();

            let pkc_ptr = if self.public_key_fingerprint.is_some() {
                pkc.as_ptr()
            } else {
                std::ptr::null()
            };

            let gkc = self
                .gossip_key_fingerprint
                .as_ref()
                .map(to_cstring)
                .unwrap_or_default();

            let gkc_ptr = if self.gossip_key_fingerprint.is_some() {
                gkc.as_ptr()
            } else {
                std::ptr::null_mut()
            };
            let vkc = self
                .verified_key_fingerprint
                .as_ref()
                .map(to_cstring)
                .unwrap_or_default();
            let vkc_ptr = if self.verified_key_fingerprint.is_some() {
                vkc.as_ptr()
            } else {
                std::ptr::null_mut()
            };
            let addr: String = self.addr.clone().unwrap_or_default();
            let addr_c = to_cstring(addr);

            unsafe {
                sqlite3_bind_blob(
                    stmt,
                    4,
                    pub_bytes.as_ptr() as *const _,
                    pub_bytes.len() as libc::c_int,
                    SQLITE_TRANSIENT(),
                )
            };
            unsafe { sqlite3_bind_int64(stmt, 5, self.gossip_timestamp as sqlite3_int64) };
            unsafe {
                sqlite3_bind_blob(
                    stmt,
                    6,
                    gossip_bytes.as_ptr() as *const _,
                    gossip_bytes.len() as libc::c_int,
                    SQLITE_TRANSIENT(),
                )
            };
            unsafe { sqlite3_bind_text(stmt, 7, pkc_ptr as *const _, -1, SQLITE_TRANSIENT()) };
            unsafe { sqlite3_bind_text(stmt, 8, gkc_ptr as *const _, -1, SQLITE_TRANSIENT()) };
            unsafe {
                sqlite3_bind_blob(
                    stmt,
                    9,
                    ver_bytes.as_ptr() as *const _,
                    ver_bytes.len() as libc::c_int,
                    SQLITE_TRANSIENT(),
                )
            };

            unsafe { sqlite3_bind_text(stmt, 10, vkc_ptr as *const _, -1, SQLITE_TRANSIENT()) };
            unsafe { sqlite3_bind_text(stmt, 11, addr_c.as_ptr(), -1, SQLITE_TRANSIENT()) };

            if unsafe { sqlite3_step(stmt) } == 101 {
                success = true;
            }

            unsafe { sqlite3_finalize(stmt) };
        } else if self.to_save == Some(ToSave::Timestamps) {
            let stmt = unsafe {
                dc_sqlite3_prepare(
                    self.context,sql,
                    b"UPDATE acpeerstates SET last_seen=?, last_seen_autocrypt=?, gossip_timestamp=? WHERE addr=?;\x00"
                        as *const u8 as *const libc::c_char)
            };

            let c_addr = self.addr.as_ref().map(to_cstring).unwrap_or_default();
            let addr_ptr = if self.addr.is_some() {
                c_addr.as_ptr()
            } else {
                std::ptr::null()
            };

            unsafe { sqlite3_bind_int64(stmt, 1, self.last_seen as sqlite3_int64) };
            unsafe { sqlite3_bind_int64(stmt, 2, self.last_seen_autocrypt as sqlite3_int64) };
            unsafe { sqlite3_bind_int64(stmt, 3, self.gossip_timestamp as sqlite3_int64) };
            unsafe { sqlite3_bind_text(stmt, 4, addr_ptr, -1, SQLITE_TRANSIENT()) };

            if unsafe { sqlite3_step(stmt) } == 101 {
                success = true;
            }

            unsafe { sqlite3_finalize(stmt) };
        }

        if self.to_save == Some(ToSave::All) || create {
            unsafe { dc_reset_gossiped_timestamp(self.context, 0 as uint32_t) };
        }

        success
    }

    pub fn has_verified_key(&self, fingerprints: &HashSet<String>) -> bool {
        if self.verified_key.is_some() && self.verified_key_fingerprint.is_some() {
            let vkc = self.verified_key_fingerprint.as_ref().unwrap();
            if fingerprints.contains(vkc) {
                return true;
            }
        }

        false
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use pretty_assertions::assert_eq;

    use std::ffi::CStr;
    use tempfile::{tempdir, TempDir};

    use crate::context::*;

    #[test]
    fn test_peerstate_save_to_db() {
        let ctx = unsafe { create_test_context() };
        let addr = "hello@mail.com";

        let pub_key = crate::key::Key::from_base64("xsBNBFztUVkBCADYaQl/UOUpRPd32nLRzx8eU0eI+jQEnG+g5anjYA+3oct1rROGl5SygjMULDKdaUy27O3o9Srsti0YjA7uxZnavIqhSopJhFidqY1M1wA9JZa/duucZdNwUGbjGIRsS/4Cjr5+3svscK24hVYub1dvDWXpwUTnj3K6xOEnJdoM+MhCqtSD5+zcJhFc9vyZm9ZTGWUxAhKh0iJTcCD8V6CQ3XZ2z9GruwzZT/FTFovWrz7m3TUI2OdSSHh0eZLRGEoxMCT/vzflAFGAr8ijCaRsEIfqP6FW8uQWnFTqkjxEUCZG6XkeFHB84aj5jqYG/1KCLjL5vEKwfl1tz/WnPhY7ABEBAAHNEDxoZWxsb0BtYWlsLmNvbT7CwIkEEAEIADMCGQEFAlztUVoCGwMECwkIBwYVCAkKCwIDFgIBFiEEgMjHGVbvLXe6ioRROg8oKCvye7gACgkQOg8oKCvye7ijAwf+PTsuawUax9cNPn1bN90H+g9qyHZJMEwKXtUnNaXJxPW3iB7ThhpCiCzsZwP7+l7ArS8tmLeNDw2bENtcf1XCv4wovP2fdXOP3QOUUFX/GdakcTwv7DzC7CO0grB1HtaPhGw/6UX2o2cx2i9xiUf4Givq2MfCbgAW5zloH6WXGPb6yLQYJXxqDIphr4+uZDb+bMAyWHN/DUkAjHrV8nnVki7PMHqzzZpwglalxMX8RGeiGZE39ALJKL/Og87DMFah87/yoxQWGoS7Wqv0XDcCPKoTCPrpk8pOe2KEsq/lz215nefHd4aRpfUX5YCYa8HPvvfPQbGF73uvyQw5w7qjis7ATQRc7VFZAQgAt8ONdnX6KEEQ5Jw6ilJ+LBtY44SP5t0I3eK+goKepgIiKhjGDa+Mntyi4jdhH+HO6kvK5SHMh2sPp4rRO/WKHJwWFySyM1OdyiywhyH0J9R5rBY4vPHsJjf6vSKJdWLWT+ho1fNet2IIC+jVCYli91MAMbRvk6EKVj1nCc+67giOahXEkHt6xxkeCGlOvbw8hxGj1A8+AC1BLms/OR3oc4JMi9O3kq6uG0z9tlUEerac9HVwcjoO1XLe+hJhoT5H+TbnGjPuhuURP3pFiIKHpbRYgUfdSAY0dTObO7t4I5y/drPOrCTnWrBUg2wXAECUhpRKow9/ai2YemLv9KqhhwARAQABwsB2BBgBCAAgBQJc7VFaAhsMFiEEgMjHGVbvLXe6ioRROg8oKCvye7gACgkQOg8oKCvye7jmyggAhs4QzCzIbT2OsAReBxkxtm0AI+g1HZ1KFKof5NDHfgv9C/Qu1I8mKEjlZzA4qFyPmLqntgwJ0RuFy6gLbljZBNCFO7vB478AhYtnWjuKZmA40HUPwcB1hEJ31c42akzfUbioY1TLLepngdsJg7Cm8O+rhI9+1WRA66haJDgFs793SVUDyJh8f9NX50l5zR87/bsV30CFSw0q4OSSy9VI/z+2g5khn1LnuuOrCfFnYIPYtJED1BfkXkosxGlgbzy79VvGmI9d23x4atDK7oBPCzIj+lP8sytJ0u3HOguXi9OgDitKy+Pt1r8gH8frdktMJr5Ts6DW+tIn2vR23KR8aA==", KeyType::Public).unwrap();

        let mut peerstate = Peerstate {
            context: &ctx.ctx,
            addr: Some(addr.into()),
            last_seen: 10,
            last_seen_autocrypt: 11,
            prefer_encrypt: EncryptPreference::Mutual,
            public_key: Some(pub_key.clone()),
            public_key_fingerprint: Some(pub_key.fingerprint()),
            gossip_key: Some(pub_key.clone()),
            gossip_timestamp: 12,
            gossip_key_fingerprint: Some(pub_key.fingerprint()),
            verified_key: VerifiedKey::Gossip,
            verified_key_fingerprint: Some(pub_key.fingerprint()),
            to_save: Some(ToSave::All),
            degrade_event: None,
        };

        assert!(
            peerstate.save_to_db(&ctx.ctx.sql.clone().read().unwrap(), true),
            "failed to save"
        );

        let peerstate_new =
            Peerstate::from_addr(&ctx.ctx, &ctx.ctx.sql.clone().read().unwrap(), addr.into())
                .expect("failed to load peerstate from db");

        // clear to_save, as that is not persissted
        peerstate.to_save = None;
        assert_eq!(peerstate, peerstate_new);
    }

    // TODO: don't copy this from stress.rs
    #[allow(dead_code)]
    struct TestContext {
        ctx: Context,
        dir: TempDir,
    }

    unsafe extern "C" fn cb(
        _context: &Context,
        _event: Event,
        _data1: uintptr_t,
        _data2: uintptr_t,
    ) -> uintptr_t {
        0
    }

    unsafe fn create_test_context() -> TestContext {
        let mut ctx = dc_context_new(cb, std::ptr::null_mut(), std::ptr::null_mut());
        let dir = tempdir().unwrap();
        let dbfile = CString::new(dir.path().join("db.sqlite").to_str().unwrap()).unwrap();
        assert_eq!(
            dc_open(&mut ctx, dbfile.as_ptr(), std::ptr::null()),
            1,
            "Failed to open {}",
            CStr::from_ptr(dbfile.as_ptr() as *const libc::c_char)
                .to_str()
                .unwrap()
        );

        TestContext { ctx: ctx, dir: dir }
    }
}