wechat_protocol/

client.rs

#[path = "group_mutation.rs"]
mod group_mutation;
pub use group_mutation::*;
#[path = "group_detail.rs"]
mod group_detail;
pub use group_detail::*;
#[path = "group_inventory.rs"]
mod group_inventory;
pub use group_inventory::*;
#[path = "logout.rs"]
mod logout;
#[path = "revoke.rs"]
mod revoke;
#[path = "status_notify.rs"]
mod status_notify;
pub use crate::wire::{
    AddMessage, AvatarBytes, ContactLookup, ContactRecord, Heartbeat, InitContactPage, RoomMember,
    RoomRoster, SyncBatch, SyncCommand, TextSendResult, VoiceChunk,
};
use crate::{
    business::{self, AuthenticatedPacket, Prelogin},
    crypto::Ecdh,
    mmtls::{self, Session},
    protobuf as p, wire,
};
use anyhow::{bail, ensure, Result};
pub use logout::*;
use openssl::{nid::Nid, pkey::PKey, rsa::Rsa};
pub use revoke::*;
use serde::{Deserialize, Serialize};
pub use status_notify::*;
use std::{fmt, time::Duration};
use zeroize::Zeroize;

#[derive(Clone, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct DeviceProfile {
    pub device_id: Vec<u8>,
    pub device_type: String,
    pub client_version: i32,
}
impl fmt::Debug for DeviceProfile {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("DeviceProfile")
            .field("device_id", &"[redacted]")
            .field("device_type", &self.device_type)
            .field("client_version", &self.client_version)
            .finish()
    }
}
impl DeviceProfile {
    pub fn generate(device_type: String) -> Result<Self> {
        let bytes = crate::crypto::random::<16>()?;
        let id = format!("l{}", hex::encode(bytes)).into_bytes()[..16].to_vec();
        let p = Self {
            device_id: id,
            device_type,
            client_version: -227271398,
        };
        p.validate()?;
        Ok(p)
    }
    pub fn validate(&self) -> Result<()> {
        ensure!(
            self.device_id.len() == 16
                && !self.device_type.is_empty()
                && self.device_type.len() <= 1024
                && !self.device_type.contains('\0'),
            "Invalid native device profile"
        );
        ensure!(
            self.client_version == -227271398,
            "Unsupported native client version"
        );
        Ok(())
    }
}
/// Raw accepted auth sections, retained only inside the encrypted account state.
/// The update mask describes native application gates; presence alone does not
/// mean an optional section changed active account credentials.
#[derive(Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct AuthSnapshot {
    pub update_mask: u32,
    pub response: Vec<u8>,
    pub auth: Vec<u8>,
    pub account: Vec<u8>,
    pub network: Vec<u8>,
    pub ax_auth: Vec<u8>,
}
impl fmt::Debug for AuthSnapshot {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("AuthSnapshot")
            .field("sections", &"[redacted]")
            .finish()
    }
}
impl Drop for AuthSnapshot {
    fn drop(&mut self) {
        self.response.zeroize();
        self.auth.zeroize();
        self.account.zeroize();
        self.network.zeroize();
        self.ax_auth.zeroize();
    }
}
/// Accepted credentials. Serialize only into the service's encrypted state vault.
#[derive(Clone, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct AccountState {
    pub schema: u32,
    pub device: DeviceProfile,
    pub uin: u32,
    pub own_username: String,
    pub cookie: Vec<u8>,
    pub session_key: Vec<u8>,
    pub request_key: Vec<u8>,
    pub response_key: Vec<u8>,
    pub auto_auth_key: Vec<u8>,
    pub key_info: Vec<u8>,
    /// Native AuthSect field 9 SKBuffer. This is the official profile password,
    /// distinct from transport keys, and is never part of an HTTP response.
    #[serde(default)]
    pub database_password: Vec<u8>,
    #[serde(default)]
    pub auth_snapshot: AuthSnapshot,
    pub long_host: String,
    pub short_host: String,
}
impl fmt::Debug for AccountState {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("AccountState")
            .field("schema", &self.schema)
            .field("credentials", &"[redacted]")
            .finish()
    }
}
impl Drop for AccountState {
    fn drop(&mut self) {
        self.cookie.zeroize();
        self.session_key.zeroize();
        self.request_key.zeroize();
        self.response_key.zeroize();
        self.auto_auth_key.zeroize();
        self.key_info.zeroize();
        self.database_password.zeroize();
    }
}
impl AccountState {
    pub fn validate(&self) -> Result<()> {
        ensure!(
            self.schema == 1
                && self.uin != 0
                && !self.own_username.is_empty()
                && self.own_username.len() <= 256
                && !self.own_username.contains('\0'),
            "Invalid native account identity"
        );
        self.device.validate()?;
        mmtls::validate_host(&self.long_host)?;
        mmtls::validate_host(&self.short_host)?;
        ensure!(
            self.cookie.len() <= 15
                && self.session_key.len() == 16
                && ((self.request_key.is_empty() && self.response_key.is_empty())
                    || (self.request_key.len() == 24 && self.response_key.len() == 24))
                && self.auto_auth_key.len() <= 1024 * 1024
                && self.key_info.len() <= 1024 * 1024
                && (self.database_password.is_empty() || self.database_password.len() == 32)
                && [
                    &self.auth_snapshot.response,
                    &self.auth_snapshot.auth,
                    &self.auth_snapshot.account,
                    &self.auth_snapshot.network,
                    &self.auth_snapshot.ax_auth
                ]
                .iter()
                .all(|v| v.len() <= 1024 * 1024),
            "Invalid native account keys"
        );
        Ok(())
    }
}
impl AccountState {
    /// Import an operator-selected private baseline checkpoint. This only converts
    /// explicit standalone credentials; it never searches or reads official files.
    /// Return diagnostics contain field names, never supplied credential values.
    pub fn from_baseline_json(input: &[u8]) -> Result<Self> {
        use base64::{engine::general_purpose::STANDARD, Engine};
        ensure!(
            input.len() <= 8 * 1024 * 1024,
            "Baseline checkpoint too large"
        );
        let value: serde_json::Value = serde_json::from_slice(input)
            .map_err(|_| anyhow::anyhow!("Invalid baseline checkpoint JSON"))?;
        let object = value
            .as_object()
            .ok_or_else(|| anyhow::anyhow!("Invalid baseline checkpoint object"))?;
        let string = |name: &str| -> Result<String> {
            object
                .get(name)
                .and_then(|v| v.as_str())
                .map(str::to_string)
                .ok_or_else(|| anyhow::anyhow!("Missing baseline field {name}"))
        };
        let decode = |name: &str, required: bool| -> Result<Vec<u8>> {
            let encoded = match object.get(name) {
                Some(v) => v
                    .as_str()
                    .ok_or_else(|| anyhow::anyhow!("Invalid baseline field {name}"))?,
                None if !required => "",
                _ => bail!("Missing baseline field {name}"),
            };
            let bytes = STANDARD
                .decode(encoded)
                .map_err(|_| anyhow::anyhow!("Invalid baseline encoding {name}"))?;
            ensure!(
                STANDARD.encode(&bytes) == encoded,
                "Noncanonical baseline encoding {name}"
            );
            Ok(bytes)
        };
        ensure!(
            object.get("schema").and_then(|v| v.as_u64()) == Some(1),
            "Unsupported baseline schema"
        );
        let uin = match object.get("uin") {
            Some(serde_json::Value::String(s)) => s
                .parse::<u32>()
                .map_err(|_| anyhow::anyhow!("Invalid baseline UIN"))?,
            Some(n) => u32::try_from(
                n.as_u64()
                    .ok_or_else(|| anyhow::anyhow!("Invalid baseline UIN"))?,
            )?,
            None => bail!("Missing baseline UIN"),
        };
        let account = decode("accountSect", false)?;
        let section_username = if account.is_empty() {
            None
        } else {
            let f = p::decode(&account)?;
            p::optional_data(&f, 1)?
                .map(|v| std::str::from_utf8(v).map(str::to_string))
                .transpose()
                .map_err(|_| anyhow::anyhow!("Invalid baseline account username"))?
        };
        let own_username = object
            .get("ownUsername")
            .or_else(|| object.get("username"))
            .map(|v| {
                v.as_str()
                    .map(str::to_string)
                    .ok_or_else(|| anyhow::anyhow!("Invalid baseline username"))
            })
            .transpose()?
            .or(section_username.clone())
            .ok_or_else(|| anyhow::anyhow!("Missing baseline account identity"))?;
        ensure!(
            section_username.is_none_or(|s| s == own_username),
            "Baseline account identity mismatch"
        );
        let state =
            Self {
                schema: 1,
                device: DeviceProfile {
                    device_id: decode("deviceId", true)?,
                    device_type: string("deviceType")?,
                    client_version: i32::try_from(
                        object
                            .get("clientVersion")
                            .and_then(|v| v.as_i64())
                            .ok_or_else(|| anyhow::anyhow!("Invalid baseline client version"))?,
                    )?,
                },
                uin,
                own_username,
                cookie: decode("cookie", false)?,
                session_key: decode("sessionKey", true)?,
                request_key: decode("field23", false)?,
                response_key: decode("field24", false)?,
                auto_auth_key: decode("autoAuthKey", false)?,
                key_info: decode("keyInfo", false)?,
                database_password: decode("databasePassword", false)?,
                auth_snapshot: AuthSnapshot {
                    update_mask: match object.get("authSections") {
                        Some(serde_json::Value::String(value)) => value.parse()?,
                        Some(value) => u32::try_from(value.as_u64().ok_or_else(|| {
                            anyhow::anyhow!("Invalid baseline auth section mask")
                        })?)?,
                        None => 0,
                    },
                    response: decode("authResponse", false)?,
                    auth: decode("authSect", false)?,
                    account,
                    network: decode("networkSect", false)?,
                    ax_auth: decode("axAuthSect", false)?,
                },
                long_host: object
                    .get("longHost")
                    .and_then(|v| v.as_str())
                    .unwrap_or("long.weixin.qq.com")
                    .to_string(),
                short_host: object
                    .get("shortHost")
                    .and_then(|v| v.as_str())
                    .unwrap_or("short.weixin.qq.com")
                    .to_string(),
            };
        state.validate()?;
        Ok(state)
    }
}

#[derive(Clone, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct LoginAttempt {
    pub schema: u32,
    pub device: DeviceProfile,
    pub computer_name: String,
    pub qr_png: Vec<u8>,
    pub width: u32,
    pub height: u32,
    pub created_at: u32,
    pub expires_at: u32,
    pub identifier: Vec<u8>,
    pub notify_key: Vec<u8>,
    pub cookie: Vec<u8>,
    pub key_version: u16,
    pub long_host: String,
    pub short_host: String,
    pub status: LoginStatus,
    #[serde(default)]
    key_info: Vec<u8>,
    credential3: Option<Vec<u8>>,
    credential4: Option<Vec<u8>>,
    #[serde(default)]
    lan_upload_attempted: bool,
    // Socket/token ownership never survives serialization or daemon restart.
    #[serde(skip)]
    lan_challenge: Option<std::sync::Arc<crate::lan_login::Challenge>>,
}
impl fmt::Debug for LoginAttempt {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("LoginAttempt")
            .field("status", &self.status)
            .field("created_at", &self.created_at)
            .field("expires_at", &self.expires_at)
            .field("credentials", &"[redacted]")
            .finish()
    }
}
impl Drop for LoginAttempt {
    fn drop(&mut self) {
        self.stop_lan_login();
        self.identifier.zeroize();
        self.notify_key.zeroize();
        self.cookie.zeroize();
        self.key_info.zeroize();
        self.credential3.zeroize();
        self.credential4.zeroize();
    }
}
impl LoginAttempt {
    pub fn lan_login_state(&self) -> Option<&'static str> {
        self.lan_challenge
            .as_ref()
            .map(|challenge| challenge.status())
    }
    fn stop_lan_login(&mut self) {
        if let Some(challenge) = self.lan_challenge.take() {
            challenge.stop();
        }
    }
    pub fn qr_svg(&self) -> Result<String> {
        use base64::Engine;
        ensure!(
            !self.qr_png.is_empty(),
            "This saved-device attempt has no QR"
        );
        let encoded = base64::engine::general_purpose::STANDARD.encode(&self.qr_png);
        Ok(format!("<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"530\" height=\"530\" viewBox=\"0 0 265 265\"><rect width=\"265\" height=\"265\" fill=\"white\"/><image x=\"40\" y=\"40\" width=\"185\" height=\"185\" style=\"image-rendering:pixelated\" href=\"data:image/png;base64,{encoded}\"/></svg>"))
    }
    fn validate(&self) -> Result<()> {
        self.validate_at(wire::unix_time()?)
    }
    fn locally_expired(&self, now: u32) -> bool {
        // A scanned QR is an approval challenge. Its image expiry must not end
        // that challenge; the operation and this hard bound still limit it.
        let hard_deadline = self.created_at.saturating_add(900);
        let deadline = if self.qr_png.is_empty()
            || matches!(self.status, LoginStatus::Scanned | LoginStatus::Approved)
        {
            hard_deadline
        } else {
            self.expires_at.min(hard_deadline)
        };
        now >= deadline
    }
    fn validate_at(&self, now: u32) -> Result<()> {
        ensure!(
            self.schema == 1 && !self.locally_expired(now),
            "Login attempt expired"
        );
        self.device.validate()?;
        ensure!(
            !self.identifier.is_empty()
                && self.identifier.len() <= 4096
                && !self.notify_key.is_empty()
                && self.notify_key.len() <= 1024
                && self.cookie.len() <= 15
                && self.key_info.len() <= 1024 * 1024,
            "Invalid login attempt"
        );
        mmtls::validate_host(&self.long_host)?;
        mmtls::validate_host(&self.short_host)?;
        ensure!(
            [10014, 50005].contains(&self.key_version),
            "Unsupported login key version"
        );
        Ok(())
    }
    fn manual_auth_request(&self, point: &[u8]) -> Result<Vec<u8>> {
        wire::manual_auth(
            &self.device,
            &self.computer_name,
            self.credential3
                .as_deref()
                .ok_or_else(|| anyhow::anyhow!("Missing approval"))?,
            self.credential4
                .as_deref()
                .ok_or_else(|| anyhow::anyhow!("Missing approval"))?,
            point,
            &self.key_info,
        )
    }
}
#[derive(Clone, Copy, Debug, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum LoginStatus {
    Pending,
    Scanned,
    Approved,
    Expired,
    Cancelled,
    Rejected,
    Completed,
}
#[derive(Clone, Debug)]
pub struct NativeOptions {
    pub long_host: String,
    pub short_host: String,
    pub timeout: Duration,
    pub receive_budget: usize,
}
impl Default for NativeOptions {
    fn default() -> Self {
        Self {
            long_host: "long.weixin.qq.com".to_string(),
            short_host: "short.weixin.qq.com".to_string(),
            timeout: Duration::from_secs(30),
            receive_budget: crate::MAX_MESSAGE_BYTES,
        }
    }
}
/// Blocking client for supported WeChat server operations.
///
/// Construct with [`Self::new`], obtain an [`AccountState`] through approved login,
/// then use methods such as [`Self::send_text`] or [`Self::sync`]. The state is
/// passed mutably because authenticated responses can update credentials/routing.
/// This client does not own a daemon, profile, send queue or recovery journal.
pub struct NativeClient {
    options: NativeOptions,
}
/// Polling an issued login challenge is repeatable. This does not authorize
/// retrying QR creation, credential exchange, sends or rejected authentication.
pub fn is_retryable_login_poll_error(error: &anyhow::Error) -> bool {
    if crate::login_lan_error_reason(error).is_some() {
        return false;
    }
    error.chain().any(|cause| {
        if let Some(io) = cause.downcast_ref::<std::io::Error>() {
            return matches!(
                io.kind(),
                std::io::ErrorKind::TimedOut
                    | std::io::ErrorKind::WouldBlock
                    | std::io::ErrorKind::ConnectionReset
                    | std::io::ErrorKind::ConnectionAborted
                    | std::io::ErrorKind::ConnectionRefused
                    | std::io::ErrorKind::UnexpectedEof
                    | std::io::ErrorKind::BrokenPipe
                    | std::io::ErrorKind::NotConnected
            );
        }
        if let Some(http) = cause.downcast_ref::<reqwest::Error>() {
            return http.is_timeout() || http.is_connect();
        }
        let message = cause.to_string();
        matches!(
            message.as_str(),
            "MMTLS deadline exceeded" | "MMTLS connection closed"
        ) || message
            .strip_prefix("Native shortlink HTTP status ")
            .and_then(|s| s.parse::<u16>().ok())
            .is_some_and(|status| matches!(status, 408 | 429 | 500 | 502 | 503 | 504))
    })
}
impl NativeClient {
    /// Validate transport options. Construction itself makes no network request.
    pub fn new(options: NativeOptions) -> Result<Self> {
        mmtls::validate_host(&options.long_host)?;
        mmtls::validate_host(&options.short_host)?;
        ensure!(
            !options.timeout.is_zero() && options.timeout <= Duration::from_secs(120),
            "Invalid native timeout"
        );
        ensure!(
            (1024..=crate::MAX_MESSAGE_BYTES).contains(&options.receive_budget),
            "Invalid receive budget"
        );
        Ok(Self { options })
    }
    fn session(&self, host: &str) -> Result<Session> {
        Session::connect(host, self.options.timeout, self.options.receive_budget)
    }
    pub fn push_connection(&self, state: &AccountState) -> Result<crate::push::Connection> {
        crate::push::Connection::open(state, self.options.timeout, self.options.receive_budget)
    }
    #[allow(clippy::too_many_arguments)]
    fn prelogin_rpc(
        &self,
        long_host: &str,
        short_host: &str,
        cgi: u16,
        path: &str,
        plain: &[u8],
        cookie: &[u8],
        auth_key: &[u8],
        fixed_version: Option<u16>,
    ) -> Result<(Vec<u8>, Vec<u8>, u16)> {
        let session = self.session(long_host)?;
        let profile = if let Some(version) = fixed_version {
            business::profile_for_region(if version == 50005 { 2 } else { 1 })?
        } else {
            business::profile_for_region(session.certificate_region)?
        };
        let prepared = Prelogin::prepare(plain, &profile, auth_key)?;
        let packet = business::prelogin_packet(&prepared, cgi, profile.version, cookie)?;
        let reply = mmtls::short_exchange(
            &session.short,
            short_host,
            path,
            &packet,
            self.options.timeout,
        )?;
        let envelope = business::split(&reply)?;
        ensure!(
            envelope.return_code == 0,
            "Native envelope returned {}",
            envelope.return_code
        );
        ensure!(
            envelope.cgi == cgi as u64 && envelope.encryption == 12,
            "Unexpected prelogin response policy"
        );
        let plain = prepared.decrypt_reply(&envelope.payload)?;
        Ok((plain, envelope.cookie, profile.version))
    }
    /// Request one QR challenge for the supplied device. Display
    /// [`LoginAttempt::qr_png`] privately, poll for phone approval, then call
    /// [`Self::complete_login`]. This does not grant authorization by itself.
    pub fn start_qr_login(
        &self,
        device: DeviceProfile,
        computer_name: &str,
    ) -> Result<LoginAttempt> {
        self.start_profile_qr_login(device, computer_name, &[])
    }
    /// QR approval for an existing installation retains its selected key-info.
    /// This neither bypasses phone approval nor imports active authorization.
    pub fn start_profile_qr_login(
        &self,
        device: DeviceProfile,
        computer_name: &str,
        key_info: &[u8],
    ) -> Result<LoginAttempt> {
        device.validate()?;
        ensure!(key_info.len() <= 1024 * 1024, "Invalid saved key-info");
        let plain = wire::qr_request(&device, computer_name)?;
        let (reply, cookie, key_version) = self.prelogin_rpc(
            &self.options.long_host,
            &self.options.short_host,
            502,
            "/cgi-bin/micromsg-bin/getloginqrcode",
            &plain,
            &[],
            &[],
            None,
        )?;
        let qr = wire::decode_qr(&reply)?;
        let now = wire::unix_time()?;
        Ok(LoginAttempt {
            schema: 1,
            device,
            computer_name: computer_name.to_string(),
            qr_png: qr.image,
            width: qr.width,
            height: qr.height,
            created_at: now,
            expires_at: now.saturating_add(240),
            identifier: qr.identifier,
            notify_key: qr.notify_key,
            cookie,
            key_version,
            long_host: self.options.long_host.clone(),
            short_host: self.options.short_host.clone(),
            status: LoginStatus::Pending,
            key_info: key_info.to_vec(),
            credential3: None,
            credential4: None,
            lan_upload_attempted: false,
            lan_challenge: None,
        })
    }
    /// Initiates phone approval for an existing remembered device without a QR.
    /// The caller chooses this fallback explicitly; no automatic auth retries occur.
    pub fn start_saved_device_login(
        &self,
        state: &AccountState,
        computer_name: &str,
    ) -> Result<LoginAttempt> {
        state.validate()?;
        self.start_saved_login(
            &state.device,
            &state.own_username,
            &state.auto_auth_key,
            &state.cookie,
            &state.key_info,
            &state.long_host,
            &state.short_host,
            computer_name,
        )
    }
    /// Use remembered official-profile trust without inventing active session keys.
    pub fn start_profile_saved_device_login(
        &self,
        device: &DeviceProfile,
        username: &str,
        auto_auth_key: &[u8],
        cookie: &[u8],
        key_info: &[u8],
        computer_name: &str,
    ) -> Result<LoginAttempt> {
        self.start_saved_login(
            device,
            username,
            auto_auth_key,
            cookie,
            key_info,
            &self.options.long_host,
            &self.options.short_host,
            computer_name,
        )
    }
    #[allow(clippy::too_many_arguments)]
    fn start_saved_login(
        &self,
        device: &DeviceProfile,
        username: &str,
        auto_auth_key: &[u8],
        cookie: &[u8],
        key_info: &[u8],
        long_host: &str,
        short_host: &str,
        computer_name: &str,
    ) -> Result<LoginAttempt> {
        device.validate()?;
        ensure!(
            !username.is_empty()
                && username.len() <= 256
                && !username.contains('\0')
                && !auto_auth_key.is_empty()
                && auto_auth_key.len() <= 1024 * 1024
                && cookie.len() <= 15
                && key_info.len() <= 1024 * 1024,
            "Invalid saved-device credentials"
        );
        let rsa = Rsa::generate(1024)?;
        let public = PKey::from_rsa(rsa)?.public_key_to_pem()?;
        let request = wire::push_login(device, username, auto_auth_key, &public)?;
        let (reply, cookie, key_version) = self.prelogin_rpc(
            long_host,
            short_host,
            654,
            "/cgi-bin/micromsg-bin/pushloginurl",
            &request,
            cookie,
            &[],
            None,
        )?;
        let (identifier, notify_key) = wire::decode_push(&reply)?;
        let now = wire::unix_time()?;
        Ok(LoginAttempt {
            schema: 1,
            device: device.clone(),
            computer_name: computer_name.to_string(),
            qr_png: vec![],
            width: 0,
            height: 0,
            created_at: now,
            expires_at: now.saturating_add(240),
            identifier,
            notify_key,
            cookie,
            key_version,
            long_host: long_host.into(),
            short_host: short_host.into(),
            status: LoginStatus::Pending,
            key_info: key_info.to_vec(),
            credential3: None,
            credential4: None,
            lan_upload_attempted: false,
            lan_challenge: None,
        })
    }
    /// Poll an existing challenge. Only [`LoginStatus::Approved`] permits
    /// [`Self::complete_login`]; expired/rejected/cancelled attempts must stop.
    pub fn poll_login(&self, attempt: &mut LoginAttempt) -> Result<LoginStatus> {
        self.poll_login_guarded(
            attempt,
            std::time::Instant::now() + Duration::from_secs(900),
            || false,
        )
    }
    /// The guard bounds the temporary LAN listener even while a blocking poll
    /// or upload is in flight. It must not perform blocking work itself.
    pub fn poll_login_guarded(
        &self,
        attempt: &mut LoginAttempt,
        deadline: std::time::Instant,
        cancelled: impl Fn() -> bool + Send + Sync + 'static,
    ) -> Result<LoginStatus> {
        let now = wire::unix_time()?;
        if attempt.locally_expired(now) {
            attempt.stop_lan_login();
            attempt.status = LoginStatus::Expired;
            return Ok(attempt.status);
        }
        attempt.validate_at(now)?;
        ensure!(
            !matches!(
                attempt.status,
                LoginStatus::Completed | LoginStatus::Cancelled | LoginStatus::Rejected
            ),
            "Login attempt is terminal"
        );
        let request = wire::poll_request(&attempt.device, &attempt.identifier)?;
        let (reply, cookie, _) = self.prelogin_rpc(
            &attempt.long_host,
            &attempt.short_host,
            503,
            "/cgi-bin/micromsg-bin/checkloginqrcode",
            &request,
            &attempt.cookie,
            &[],
            Some(attempt.key_version),
        )?;
        let response = wire::decode_poll(&reply, &attempt.notify_key)?;
        if !cookie.is_empty() {
            attempt.cookie = cookie;
        }
        let remaining = attempt
            .created_at
            .saturating_add(900)
            .saturating_sub(wire::unix_time()?);
        let deadline =
            deadline.min(std::time::Instant::now() + Duration::from_secs(u64::from(remaining)));
        let scope = crate::lan_login::Scope::new(deadline, cancelled);
        apply_login_notification(attempt, response, |attempt| {
            self.start_lan_login(attempt, scope)
        })
    }
    fn start_lan_login(
        &self,
        attempt: &mut LoginAttempt,
        scope: crate::lan_login::Scope,
    ) -> Result<std::sync::Arc<crate::lan_login::Challenge>> {
        use crate::lan_login as lan;
        let challenge = std::sync::Arc::new(lan::Challenge::bind(scope)?);
        let random = zeroize::Zeroizing::new(crate::crypto::random::<16>()?);
        let request = zeroize::Zeroizing::new(challenge.request(
            &attempt.device,
            &attempt.identifier,
            &random,
        )?);
        let (reply, cookie, _) = self
            .prelogin_rpc(
                &attempt.long_host,
                &attempt.short_host,
                lan::CGI,
                lan::PATH,
                &request,
                &attempt.cookie,
                &[],
                Some(attempt.key_version),
            )
            .map_err(|_| lan::failure("upload_failed"))?;
        let reply = zeroize::Zeroizing::new(reply);
        let token =
            lan::response(&reply).map_err(|_| lan::failure("upload_rejected_or_invalid"))?;
        challenge.arm(&token)?;
        if !cookie.is_empty() {
            attempt.cookie = cookie;
        }
        Ok(challenge)
    }
    /// Exchange an approved challenge for secret account credentials once.
    /// Keep the returned state in memory or an encrypted application vault.
    pub fn complete_login(&self, attempt: &mut LoginAttempt) -> Result<AccountState> {
        attempt.stop_lan_login();
        attempt.validate()?;
        ensure!(
            attempt.status == LoginStatus::Approved,
            "Login has not been approved"
        );
        let inner = Ecdh::generate(Nid::SECP224R1)?;
        let request = attempt.manual_auth_request(&inner.public()?)?;
        let mut long_host = attempt.long_host.clone();
        let mut short_host = attempt.short_host.clone();
        let mut cookie = attempt.cookie.clone();
        for redirected in 0..=1 {
            let (reply, next_cookie, _) = self.prelogin_rpc(
                &long_host,
                &short_host,
                252,
                "/cgi-bin/micromsg-bin/secmanualauth",
                &request,
                &cookie,
                &[],
                None,
            )?;
            let fields = p::decode(&reply)?;
            let code = p::base_code(&fields)?;
            if code == -301 && redirected == 0 {
                (long_host, short_host) = redirect(&fields)?;
                cookie = next_cookie;
                continue;
            }
            ensure!(code == 0, "Account authentication returned {code}");
            let mut state = reduce_auth(&reply, None, &inner, &attempt.device, &next_cookie)?;
            state.long_host = long_host;
            state.short_host = short_host;
            state.validate()?;
            attempt.status = LoginStatus::Completed;
            attempt.credential3.zeroize();
            attempt.credential4.zeroize();
            return Ok(state);
        }
        bail!("Native authentication redirect limit reached")
    }
    /// Replace an expired, unscanned QR while preserving installation key-info.
    pub fn refresh_qr_login(&self, attempt: &LoginAttempt) -> Result<LoginAttempt> {
        ensure!(
            attempt.status == LoginStatus::Expired && !attempt.qr_png.is_empty(),
            "Only an expired QR may be refreshed"
        );
        let options = NativeOptions {
            long_host: attempt.long_host.clone(),
            short_host: attempt.short_host.clone(),
            ..self.options.clone()
        };
        Self::new(options)?.start_profile_qr_login(
            attempt.device.clone(),
            &attempt.computer_name,
            &attempt.key_info,
        )
    }
    /// One trusted AutoAuth exchange, retaining identity and replacing selected key
    /// sections atomically in the returned value. Does not silently fall back to QR.
    pub fn renew_session(
        &self,
        previous: &AccountState,
        computer_name: &str,
    ) -> Result<AccountState> {
        previous.validate()?;
        let inner = Ecdh::generate(Nid::SECP224R1)?;
        let request = wire::auto_auth(previous, &inner.public()?, computer_name)?;
        let mut long = previous.long_host.clone();
        let mut short = previous.short_host.clone();
        let mut cookie = previous.cookie.clone();
        for redirected in 0..=1 {
            let (reply, next_cookie, _) = self.prelogin_rpc(
                &long,
                &short,
                763,
                "/cgi-bin/micromsg-bin/secautoauth",
                &request,
                &cookie,
                &previous.auto_auth_key,
                None,
            )?;
            let fields = p::decode(&reply)?;
            let code = p::base_code(&fields)?;
            if code == -301 && redirected == 0 {
                (long, short) = redirect(&fields)?;
                cookie = next_cookie;
                continue;
            }
            ensure!(code == 0, "Automatic authentication returned {code}");
            let mut next = reduce_auth(
                &reply,
                Some(previous),
                &inner,
                &previous.device,
                &next_cookie,
            )?;
            next.long_host = long;
            next.short_host = short;
            next.validate()?;
            return Ok(next);
        }
        bail!("AutoAuth redirect limit reached")
    }
    /// Perform one experimental call host RPC using its fixed CGI/path pair.
    /// The owner must first retain its intent, hold account mutation authority and
    /// fence account/logout generations. This method does not retry uncertain
    /// operations. An active Ax policy is unsupported here; there is no ordinary
    /// packing fallback. Returned CommonResponse/application results still need
    /// their own checks. Accepted cookie updates follow normal native transport.
    pub fn experimental_call_rpc(
        &self,
        state: &mut AccountState,
        route: crate::calls::host_policy::CallHostRoute,
        request: &[u8],
    ) -> Result<Vec<u8>> {
        self.experimental_call_rpc_guarded(state, route, request, || Ok(()))
    }

    /// As [`Self::experimental_call_rpc`], with one final owner check after
    /// handshake/preparation and before handing business bytes to transport.
    /// A guard cannot retract bytes after transport handoff. Handshake traffic
    /// can already have occurred when this guard stops the business request.
    pub fn experimental_call_rpc_guarded(
        &self,
        state: &mut AccountState,
        route: crate::calls::host_policy::CallHostRoute,
        request: &[u8],
        before_submit: impl FnOnce() -> Result<()>,
    ) -> Result<Vec<u8>> {
        self.call_rpc_with_exchange(state, route, request, before_submit, mmtls::short_exchange)
    }

    /// Allow the embedding daemon to stop awaiting a pre-answer LiveSync.
    /// The returned interruption does not assert that the request was unsent.
    #[doc(hidden)]
    pub fn experimental_call_live_sync_guarded(
        &self,
        state: &mut AccountState,
        request: &[u8],
        before_submit: impl FnOnce() -> Result<()>,
        wait: impl for<'a> FnOnce(
            std::pin::Pin<Box<dyn std::future::Future<Output = Result<Vec<u8>>> + Send + 'a>>,
        ) -> Result<Vec<u8>>,
    ) -> Result<Vec<u8>> {
        self.call_rpc_with_exchange(
            state,
            crate::calls::host_policy::CallHostRoute::LiveSync,
            request,
            before_submit,
            |session, host, path, payload, timeout| {
                wait(Box::pin(mmtls::short_exchange_interruptible(
                    session, host, path, payload, timeout,
                )))
            },
        )
    }

    fn call_rpc_with_exchange(
        &self,
        state: &mut AccountState,
        route: crate::calls::host_policy::CallHostRoute,
        request: &[u8],
        before_submit: impl FnOnce() -> Result<()>,
        exchange: impl FnOnce(&mmtls::ShortSession, &str, &str, &[u8], Duration) -> Result<Vec<u8>>,
    ) -> Result<Vec<u8>> {
        use crate::calls::{
            host_policy::{selected_policy, CallHostPolicy},
            host_transport::{CallHostRpcError, CallHostStage},
        };
        mmtls::validate_short_route(route.path())
            .map_err(|error| CallHostRpcError::from_cause(CallHostStage::Route, &error))?;
        let policy = selected_policy(&state.auth_snapshot, route.command())
            .map_err(|_| CallHostRpcError::new(CallHostStage::Route, "policy", None))?;
        if policy != CallHostPolicy::Ordinary {
            return Err(CallHostRpcError::new(CallHostStage::Route, "ax_unsupported", None).into());
        }
        let session = self
            .session(&state.long_host)
            .map_err(|error| CallHostRpcError::from_cause(CallHostStage::Handshake, &error))?;
        let packet = AuthenticatedPacket::prepare(state, route.command(), request)
            .map_err(|error| CallHostRpcError::from_cause(CallHostStage::Prepare, &error))?;
        before_submit()?;
        let reply = exchange(
            &session.short,
            &state.short_host,
            route.path(),
            &packet.packet,
            self.options.timeout,
        )
        .map_err(|error| {
            if error.is::<crate::calls::client::CallLiveSyncInterrupted>() {
                error
            } else {
                CallHostRpcError::from_cause(CallHostStage::Exchange, &error).into()
            }
        })?;
        let (plain, cookie) =
            crate::calls::host_transport::decode_verified_reply(state, route.command(), &reply)?;
        if !cookie.is_empty() {
            state.cookie = cookie;
        }
        Ok(plain)
    }

    pub(crate) fn authenticated_rpc(
        &self,
        state: &mut AccountState,
        cgi: u16,
        path: &str,
        request: &[u8],
    ) -> Result<Vec<u8>> {
        self.authenticated_rpc_guarded(state, cgi, path, request, || Ok(()))
    }
    fn authenticated_rpc_guarded(
        &self,
        state: &mut AccountState,
        cgi: u16,
        path: &str,
        request: &[u8],
        before_submit: impl FnOnce() -> Result<()>,
    ) -> Result<Vec<u8>> {
        let session = self.session(&state.long_host)?;
        let packet = AuthenticatedPacket::prepare(state, cgi, request)?;
        before_submit()?;
        let reply = mmtls::short_exchange(
            &session.short,
            &state.short_host,
            path,
            &packet.packet,
            self.options.timeout,
        )?;
        let (plain, cookie) = packet.decode_verified_transport_reply(&reply)?;
        if !cookie.is_empty() {
            state.cookie = cookie;
        }
        Ok(plain)
    }
    /// Fetch one scene-6 sync page. An empty cursor starts an initial sync.
    /// Returned commands include their original payloads. Persist and handle all
    /// required commands before committing the returned cursor; this call does
    /// not write history or manage a checkpoint for the application.
    pub fn sync(&self, state: &mut AccountState, cursor: &[u8]) -> Result<SyncBatch> {
        self.sync_scene(state, cursor, 6)
    }
    pub fn sync_scene(
        &self,
        state: &mut AccountState,
        cursor: &[u8],
        scene: u32,
    ) -> Result<SyncBatch> {
        ensure!([6, 7].contains(&scene), "Unsupported sync scene");
        let wire_cursor = if cursor.is_empty() { &[0][..] } else { cursor };
        let request = wire::sync_request(wire_cursor, &state.device.device_type, scene)?;
        wire::decode_sync(
            &self.authenticated_rpc(state, 138, "/cgi-bin/micromsg-bin/newsync", &request)?,
            cursor,
        )
    }
    /// Request heartbeat hints for a committed sync cursor. Hints are not message
    /// contents; the application still needs to fetch and handle sync commands.
    pub fn heartbeat(&self, state: &mut AccountState, cursor: &[u8]) -> Result<Heartbeat> {
        let request = wire::heartbeat_request(state, cursor)?;
        wire::decode_heartbeat(&self.authenticated_rpc(
            state,
            518,
            "/cgi-bin/micromsg-bin/heartbeat",
            &request,
        )?)
    }
    /// Submit exactly once. An I/O error after submission is an uncertain outcome;
    /// services must persist the intent and must not automatically retry this call.
    ///
    /// `destination` is an exact WeChat username, not a display name or daemon
    /// account ID. `created_at` is Unix seconds; `client_message_id` is a nonzero
    /// protocol ID derived with [`wire::client_message_id`] from the caller's task
    /// ID. Check acceptance and both returned-identity matches in the receipt.
    /// No official-profile history is written by this method.
    pub fn send_text(
        &self,
        state: &mut AccountState,
        destination: &str,
        content: &str,
        created_at: u32,
        client_message_id: u32,
    ) -> Result<TextSendResult> {
        self.send_text_guarded(
            state,
            destination,
            content,
            created_at,
            client_message_id,
            || Ok(()),
        )
    }
    /// Recheck the embedding daemon's admission immediately before submitting.
    #[doc(hidden)]
    pub fn send_text_guarded(
        &self,
        state: &mut AccountState,
        destination: &str,
        content: &str,
        created_at: u32,
        client_message_id: u32,
        before_submit: impl FnOnce() -> Result<()>,
    ) -> Result<TextSendResult> {
        let request = wire::text_send_request_with_source(
            destination,
            content,
            created_at,
            client_message_id,
            None,
        )?;
        wire::decode_text_send(
            &self.authenticated_rpc_guarded(
                state,
                522,
                "/cgi-bin/micromsg-bin/newsendmsg",
                &request,
                before_submit,
            )?,
            destination,
            client_message_id,
        )
    }
    /// Submit text with optional native message-source XML. For a simple send use
    /// [`Self::send_text`]. This has the same single-submission/uncertainty contract.
    pub fn send_text_with_source(
        &self,
        state: &mut AccountState,
        destination: &str,
        content: &str,
        created_at: u32,
        client_message_id: u32,
        source: Option<&str>,
    ) -> Result<TextSendResult> {
        let request = wire::text_send_request_with_source(
            destination,
            content,
            created_at,
            client_message_id,
            source,
        )?;
        wire::decode_text_send(
            &self.authenticated_rpc(state, 522, "/cgi-bin/micromsg-bin/newsendmsg", &request)?,
            destination,
            client_message_id,
        )
    }
    /// Look up exact usernames on the server. The returned records are not
    /// automatically written into a local profile.
    pub fn lookup_contacts(
        &self,
        state: &mut AccountState,
        usernames: &[String],
    ) -> Result<ContactLookup> {
        let request = wire::contact_request(state, usernames)?;
        wire::decode_contacts(
            &self.authenticated_rpc(state, 182, "/cgi-bin/micromsg-bin/getcontact", &request)?,
            usernames,
        )
    }
    pub fn lookup_room(
        &self,
        state: &mut AccountState,
        room_id: &str,
        version: u32,
    ) -> Result<RoomRoster> {
        let request = wire::room_request(state, room_id, version)?;
        wire::decode_room(
            &self.authenticated_rpc(
                state,
                551,
                "/cgi-bin/micromsg-bin/getchatroommemberdetail",
                &request,
            )?,
            room_id,
        )
    }
    pub fn init_contacts(
        &self,
        state: &mut AccountState,
        contact_sequence: u32,
        room_sequence: u32,
    ) -> Result<InitContactPage> {
        let request =
            wire::init_contacts_request(&state.own_username, contact_sequence, room_sequence)?;
        let page = wire::decode_init_contacts(&self.authenticated_rpc(
            state,
            851,
            "/cgi-bin/micromsg-bin/initcontact",
            &request,
        )?)?;
        ensure!(
            !page.more
                || page.contact_sequence != contact_sequence
                || page.room_sequence != room_sequence,
            "Contact bootstrap made no cursor progress"
        );
        Ok(page)
    }
    pub fn download_avatars(
        &self,
        state: &mut AccountState,
        usernames: &[String],
    ) -> Result<Vec<AvatarBytes>> {
        let request = wire::avatar_request(state, usernames)?;
        wire::decode_avatars(
            &self.authenticated_rpc(
                state,
                123,
                "/cgi-bin/micromsg-bin/batchgetheadimg",
                &request,
            )?,
            usernames,
        )
    }
    pub fn download_voice_chunk(
        &self,
        state: &mut AccountState,
        server_message_id: u64,
        total_length: u32,
        offset: u32,
    ) -> Result<VoiceChunk> {
        let request = wire::voice_request(state, server_message_id, total_length, offset)?;
        wire::decode_voice(
            &self.authenticated_rpc(state, 128, "/cgi-bin/micromsg-bin/downloadvoice", &request)?,
            server_message_id,
            total_length,
            offset,
        )
    }
    pub fn cdn_config(&self, state: &mut AccountState) -> Result<crate::media::CdnConfig> {
        let request = p::bytes(1, &wire::base_request(&state.device, state.uin, 0)?);
        let reply =
            self.authenticated_rpc(state, 379, "/cgi-bin/micromsg-bin/getcdndns", &request)?;
        crate::media::decode_cdn_config(&reply, state)
    }
    pub fn large_cdn_config(
        &self,
        state: &mut AccountState,
        route: crate::media::LargeCdnRoute,
    ) -> Result<crate::media::CdnConfig> {
        let request = p::bytes(1, &wire::base_request(&state.device, state.uin, 0)?);
        let reply =
            self.authenticated_rpc(state, 379, "/cgi-bin/micromsg-bin/getcdndns", &request)?;
        crate::media::decode_large_cdn_config(&reply, state, route)
    }
    /// Submit a URL card once through the app-message carrier. The caller owns
    /// the durable intent and must inspect an uncertain submission outcome.
    pub fn send_share(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        created_at: u32,
        content: &str,
    ) -> Result<crate::media_send::MediaSendResult> {
        let request = crate::media_send::share_send_request(
            state,
            destination,
            client_id,
            created_at,
            content,
        )?;
        crate::media_send::decode_file_send(
            &self.authenticated_rpc(state, 222, "/cgi-bin/micromsg-bin/sendappmsg", &request)?,
            client_id,
        )
    }
    /// Submit a quoted reply once. The caller owns the persisted XML, intent and
    /// uncertain-submission handling, as for other app-message sends.
    pub fn send_quote(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        created_at: u32,
        content: &str,
    ) -> Result<crate::media_send::MediaSendResult> {
        self.send_quote_with_source(state, destination, client_id, created_at, content, None)
    }
    pub fn send_quote_with_source(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        created_at: u32,
        content: &str,
        source: Option<&str>,
    ) -> Result<crate::media_send::MediaSendResult> {
        let request = crate::quote_send::quote_send_request_with_source(
            state,
            destination,
            client_id,
            created_at,
            content,
            source,
        )?;
        crate::media_send::decode_file_send(
            &self.authenticated_rpc(state, 222, "/cgi-bin/micromsg-bin/sendappmsg", &request)?,
            client_id,
        )
    }
    /// Reserve one file-upload message. An uncertain response must not be
    /// retried automatically; retain the returned receipt with the send intent.
    pub fn reserve_file_upload(
        &self,
        state: &mut AccountState,
        destination: &str,
        progress_client_id: &str,
        file_name: &str,
        plaintext_bytes: usize,
        md5: &str,
    ) -> Result<crate::file_upload::FileUploadReservation> {
        let request = crate::file_upload::file_upload_request(
            state,
            destination,
            progress_client_id,
            file_name,
            plaintext_bytes,
            md5,
        )?;
        crate::file_upload::decode_file_upload_reservation(&self.authenticated_rpc(
            state,
            6691,
            "/cgi-bin/micromsg-bin/sendfileuploadmsg",
            &request,
        )?)
    }
    pub fn send_file(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        created_at: u32,
        upload: &crate::media::UploadedMedia,
    ) -> Result<crate::media_send::MediaSendResult> {
        let request = crate::media_send::file_send_request(
            state,
            destination,
            client_id,
            created_at,
            upload,
        )?;
        crate::media_send::decode_file_send(
            &self.authenticated_rpc(state, 222, "/cgi-bin/micromsg-bin/sendappmsg", &request)?,
            client_id,
        )
    }
    #[allow(clippy::too_many_arguments)]
    pub fn send_file_with_reservation(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        created_at: u32,
        upload: &crate::media::UploadedMedia,
        reservation: Option<&crate::file_upload::FileUploadReservation>,
    ) -> Result<crate::media_send::MediaSendResult> {
        let request = crate::media_send::file_send_request_with_reservation(
            state,
            destination,
            client_id,
            created_at,
            upload,
            reservation,
        )?;
        let reply =
            self.authenticated_rpc(state, 222, "/cgi-bin/micromsg-bin/sendappmsg", &request)?;
        if reservation.is_some() {
            crate::media_send::decode_file_send_correlated(
                &reply,
                client_id,
                &state.own_username,
                destination,
                &upload.descriptor.aes_key,
            )
        } else {
            crate::media_send::decode_file_send(&reply, client_id)
        }
    }
    pub fn send_image(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        upload: &crate::media::UploadedImage,
    ) -> Result<crate::media_send::MediaSendResult> {
        let request = crate::media_send::image_send_request(state, destination, client_id, upload)?;
        crate::media_send::decode_image_send(
            &self.authenticated_rpc(state, 110, "/cgi-bin/micromsg-bin/uploadmsgimg", &request)?,
            client_id,
            &state.own_username,
            destination,
        )
    }
    pub fn send_video(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        upload: &crate::media::UploadedVideo,
        play_seconds: u32,
    ) -> Result<crate::media_send::MediaSendResult> {
        let request = crate::media_send::video_send_request(
            state,
            destination,
            client_id,
            upload,
            play_seconds,
        )?;
        crate::media_send::decode_video_send(
            &self.authenticated_rpc(state, 149, "/cgi-bin/micromsg-bin/uploadvideo", &request)?,
            client_id,
        )
    }
    /// Submit one voice note, as consecutive byte ranges when it exceeds one
    /// request. Each range is sent once; an error after the first submission is
    /// an uncertain outcome that the caller must not retry automatically.
    pub fn send_voice(
        &self,
        state: &mut AccountState,
        destination: &str,
        client_id: &str,
        created_at: u32,
        duration_ms: u32,
        encoded: &[u8],
    ) -> Result<crate::media_send::MediaSendResult> {
        let mut upload = crate::media_send::VoiceUpload::new(
            destination,
            client_id,
            created_at,
            duration_ms,
            encoded,
        )?;
        loop {
            let request = upload.request(state)?;
            let reply =
                self.authenticated_rpc(state, 127, "/cgi-bin/micromsg-bin/uploadvoice", &request)?;
            if let Some(result) = upload.accept_reply(&reply)? {
                return Ok(result);
            }
        }
    }
}

fn redirect(fields: &[p::Field]) -> Result<(String, String)> {
    let network = p::decode(p::data(fields, 5)?)?;
    let hosts = p::decode(p::data(&network, 1)?)?;
    let target = |source: &[u8]| -> Result<String> {
        let mut out = None;
        for f in hosts.iter().filter(|f| f.number == 2) {
            let p::Value::Bytes(raw) = &f.value else {
                bail!("Invalid redirect map");
            };
            let item = p::decode(raw)?;
            if p::data(&item, 1)? == source {
                ensure!(out.is_none(), "Ambiguous redirect");
                let name = std::str::from_utf8(p::data(&item, 2)?)?.to_string();
                mmtls::validate_host(&name)?;
                out = Some(name);
            }
        }
        out.ok_or_else(|| anyhow::anyhow!("Missing redirect mapping"))
    };
    Ok((
        target(b"long.weixin.qq.com")?,
        target(b"short.weixin.qq.com")?,
    ))
}
fn reduce_auth(
    plain: &[u8],
    previous: Option<&AccountState>,
    inner: &Ecdh,
    device: &DeviceProfile,
    cookie: &[u8],
) -> Result<AccountState> {
    let f = p::decode(plain)?;
    ensure!(p::base_code(&f)? == 0, "Rejected auth response");
    let mask = u32::try_from(p::scalar(&f, 2)?)?;
    for (bit, n) in [(1, 3), (2, 4), (4, 5)] {
        if mask & bit != 0 {
            p::data(&f, n)?;
        }
    }
    let mut state = previous.cloned().unwrap_or_else(|| AccountState {
        schema: 1,
        device: device.clone(),
        uin: 0,
        own_username: String::new(),
        cookie: vec![],
        session_key: vec![],
        request_key: vec![],
        response_key: vec![],
        auto_auth_key: vec![],
        key_info: vec![],
        database_password: vec![],
        auth_snapshot: AuthSnapshot::default(),
        long_host: "long.weixin.qq.com".into(),
        short_host: "short.weixin.qq.com".into(),
    });
    ensure!(
        plain.len() <= 1024 * 1024,
        "Auth response snapshot too large"
    );
    state.auth_snapshot.update_mask = mask;
    state.auth_snapshot.response = plain.to_vec();
    for (number, target) in [
        (3, &mut state.auth_snapshot.auth),
        (4, &mut state.auth_snapshot.account),
        (5, &mut state.auth_snapshot.network),
        (6, &mut state.auth_snapshot.ax_auth),
    ] {
        if let Some(section) = p::optional_data(&f, number)? {
            *target = section.to_vec();
        }
    }
    if let Some(account) = p::optional_data(&f, 4)? {
        let a = p::decode(account)?;
        if let Some(name) = p::optional_data(&a, 1)? {
            let name = std::str::from_utf8(name)?;
            ensure!(
                previous.is_none_or(|s| s.own_username == name),
                "Auth account identity changed"
            );
            state.own_username = name.to_string();
        }
    }
    if mask & 1 != 0 {
        let auth = p::data(&f, 3)?;
        let a = p::decode(auth)?;
        let uin = u32::try_from(p::scalar(&a, 1)?)?;
        ensure!(
            uin != 0 && previous.is_none_or(|s| s.uin == uin),
            "Auth UIN changed"
        );
        let (session, request, response) = business::account_keys(inner, auth)?;
        state.uin = uin;
        state.session_key = session;
        state.request_key = request;
        state.response_key = response;
        state.auto_auth_key = p::optional_data(&a, 4)?
            .map(p::unbuffer)
            .transpose()?
            .unwrap_or_default();
        if let Some(password) = p::optional_data(&a, 9)? {
            let password = p::unbuffer(password)?;
            ensure!(
                password.is_empty() || password.len() == 32,
                "Invalid native database password"
            );
            state.database_password.zeroize();
            state.database_password = password;
        }
        state.key_info = p::optional_data(&a, 10)?
            .map(p::unbuffer)
            .transpose()?
            .unwrap_or_default();
    } else {
        ensure!(previous.is_some(), "Initial auth has no key section");
    }
    ensure!(cookie.len() <= 15, "Invalid auth cookie");
    if !cookie.is_empty() {
        state.cookie = cookie.to_vec();
    }
    state.validate()?;
    Ok(state)
}

fn apply_login_notification(
    attempt: &mut LoginAttempt,
    response: wire::PollResponse,
    start_lan: impl FnOnce(&mut LoginAttempt) -> Result<std::sync::Arc<crate::lan_login::Challenge>>,
) -> Result<LoginStatus> {
    if response.lan_challenge && matches!(response.status, Some(0..=2)) {
        // A risk challenge is not approval, including a notification carrying
        // the ordinary approval status without the final credentials yet.
        attempt.credential3.zeroize();
        attempt.credential4.zeroize();
        attempt.credential3 = None;
        attempt.credential4 = None;
        attempt.status = LoginStatus::Scanned;
        if let Some(challenge) = &attempt.lan_challenge {
            if challenge.active() {
                return Ok(attempt.status);
            }
            attempt.stop_lan_login();
            return Err(crate::lan_login::failure("listener_stopped"));
        }
        if attempt.lan_upload_attempted {
            return Err(crate::lan_login::failure("upload_already_attempted"));
        }
        // Record before submission. A lost upload response must not produce an
        // automatic new nonce/token exchange on the next repeatable QR poll.
        attempt.lan_upload_attempted = true;
        attempt.lan_challenge = Some(start_lan(attempt)?);
        return Ok(attempt.status);
    }
    if let Some(status) = response.status {
        attempt.status = match status {
            0 if attempt.status == LoginStatus::Scanned => LoginStatus::Scanned,
            0 => LoginStatus::Pending,
            1 => LoginStatus::Scanned,
            2 => LoginStatus::Approved,
            // Unknown terminal states never count as approval or local expiry.
            _ => LoginStatus::Rejected,
        };
        if attempt.status == LoginStatus::Approved {
            attempt.credential3 = response.credential3;
            attempt.credential4 = response.credential4;
        }
        if matches!(
            attempt.status,
            LoginStatus::Approved | LoginStatus::Rejected
        ) {
            attempt.stop_lan_login();
        }
    }
    if attempt
        .lan_challenge
        .as_ref()
        .is_some_and(|challenge| !challenge.active())
    {
        attempt.stop_lan_login();
        return Err(crate::lan_login::failure("listener_stopped"));
    }
    Ok(attempt.status)
}

#[cfg(test)]
mod tests {
    use super::*;
    fn attempt(status: LoginStatus, qr: bool) -> LoginAttempt {
        LoginAttempt {
            schema: 1,
            device: state().device.clone(),
            computer_name: "synthetic desktop".into(),
            qr_png: if qr { vec![1] } else { Vec::new() },
            width: 0,
            height: 0,
            created_at: 1000,
            expires_at: 1240,
            identifier: vec![1; 16],
            notify_key: vec![2; 16],
            cookie: vec![3; 15],
            key_version: 10014,
            long_host: "long.weixin.qq.com".into(),
            short_host: "short.weixin.qq.com".into(),
            status,
            key_info: b"persisted installation key-info".to_vec(),
            credential3: Some(vec![4; 16]),
            credential4: Some(vec![5; 16]),
            lan_upload_attempted: false,
            lan_challenge: None,
        }
    }
    #[test]
    fn final_approval_request_restores_key_info_for_both_saved_device_and_qr() {
        let inner = Ecdh::generate(Nid::SECP224R1).unwrap();
        for qr in [false, true] {
            let attempt = attempt(LoginStatus::Approved, qr);
            let restored: LoginAttempt =
                serde_json::from_slice(&serde_json::to_vec(&attempt).unwrap()).unwrap();
            assert_eq!(restored.key_info, attempt.key_info);
            let request = p::decode(
                &restored
                    .manual_auth_request(&inner.public().unwrap())
                    .unwrap(),
            )
            .unwrap();
            let aes = p::decode(p::data(&request, 2).unwrap()).unwrap();
            let base = p::decode(p::data(&aes, 1).unwrap()).unwrap();
            // Retained official requests use1 for both approval paths.
            assert_eq!(p::scalar(&base, 6).unwrap(), 1);
            let auth = p::decode(p::data(&aes, 2).unwrap()).unwrap();
            assert_eq!(
                p::unbuffer(p::data(&auth, 5).unwrap()).unwrap(),
                attempt.key_info
            );
            assert!(!format!("{restored:?}").contains("persisted installation"));
        }
    }
    #[test]
    fn scanned_and_saved_device_approval_outlive_image_expiry_but_remain_bounded() {
        let qr = attempt(LoginStatus::Pending, true);
        assert!(qr.validate_at(1239).is_ok());
        assert!(qr.validate_at(1240).is_err());
        for (status, qr) in [
            (LoginStatus::Scanned, true),
            (LoginStatus::Approved, true),
            (LoginStatus::Pending, false),
        ] {
            let attempt = attempt(status, qr);
            assert!(attempt.validate_at(1240).is_ok());
            assert!(attempt.validate_at(1899).is_ok());
            assert!(attempt.validate_at(1900).is_err());
        }
    }
    fn notification(status: u32, lan_challenge: bool) -> wire::PollResponse {
        wire::PollResponse {
            status: Some(status),
            lan_challenge,
            credential3: (status == 2).then(|| vec![4; 16]),
            credential4: (status == 2).then(|| vec![5; 16]),
        }
    }
    fn synthetic_lan() -> std::sync::Arc<crate::lan_login::Challenge> {
        let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap();
        let challenge = crate::lan_login::Challenge::from_listener(
            listener,
            vec![std::net::Ipv4Addr::LOCALHOST],
            crate::lan_login::Scope::new(
                std::time::Instant::now() + Duration::from_secs(5),
                || false,
            ),
        )
        .unwrap();
        challenge.arm(b"synthetic temporary token").unwrap();
        std::sync::Arc::new(challenge)
    }
    #[test]
    fn repeated_risk_notifications_upload_once_and_never_install_approval() {
        let mut attempt = attempt(LoginStatus::Pending, true);
        let challenge = synthetic_lan();
        assert_eq!(
            apply_login_notification(&mut attempt, notification(2, true), |_| Ok(
                challenge.clone()
            ))
            .unwrap(),
            LoginStatus::Scanned
        );
        assert!(attempt.credential3.is_none() && attempt.credential4.is_none());
        assert_eq!(attempt.lan_login_state(), Some("waiting_for_phone"));
        for _ in 0..3 {
            assert_eq!(
                apply_login_notification(&mut attempt, notification(1, true), |_| panic!(
                    "duplicate upload"
                ))
                .unwrap(),
                LoginStatus::Scanned
            );
        }
        // A normal pending poll while the phone works retains the same listener.
        apply_login_notification(&mut attempt, notification(0, false), |_| {
            panic!("ordinary poll started LAN")
        })
        .unwrap();
        assert_eq!(attempt.status, LoginStatus::Scanned);
        assert!(challenge.active());
        let serialized = serde_json::to_string(&attempt).unwrap();
        assert!(!serialized.contains("synthetic temporary token"));
        assert!(!format!("{attempt:?}").contains("synthetic temporary token"));
        let restored: LoginAttempt = serde_json::from_str(&serialized).unwrap();
        assert!(restored.lan_login_state().is_none());
        assert!(restored.lan_upload_attempted); // No restart replay of an uncertain upload.
        assert_eq!(
            apply_login_notification(&mut attempt, notification(2, false), |_| panic!(
                "approval started LAN"
            ))
            .unwrap(),
            LoginStatus::Approved
        );
        assert!(attempt.lan_login_state().is_none());
        assert!(!challenge.active());
        assert!(attempt.credential3.is_some() && attempt.credential4.is_some());
    }
    #[test]
    fn failed_upload_is_not_retried_by_the_repeatable_poll_path() {
        let mut attempt = attempt(LoginStatus::Pending, true);
        let error = apply_login_notification(&mut attempt, notification(1, true), |_| {
            Err(crate::lan_login::failure("upload_failed"))
        })
        .unwrap_err();
        assert!(!is_retryable_login_poll_error(&error));
        assert_eq!(crate::login_lan_error_reason(&error), Some("upload_failed"));
        assert!(attempt.lan_login_state().is_none());
        let error = apply_login_notification(&mut attempt, notification(1, true), |_| {
            panic!("resubmitted upload")
        })
        .unwrap_err();
        assert_eq!(
            crate::login_lan_error_reason(&error),
            Some("upload_already_attempted")
        );
        assert!(attempt.credential3.is_none() && attempt.credential4.is_none());
    }
    #[test]
    fn rejection_or_attempt_drop_closes_the_shared_listener() {
        for reject in [false, true] {
            let mut attempt = attempt(LoginStatus::Pending, false);
            let challenge = synthetic_lan();
            apply_login_notification(&mut attempt, notification(1, true), |_| {
                Ok(challenge.clone())
            })
            .unwrap();
            if reject {
                assert_eq!(
                    apply_login_notification(&mut attempt, notification(3, true), |_| panic!(
                        "rejection started LAN"
                    ))
                    .unwrap(),
                    LoginStatus::Rejected
                );
            }
            drop(attempt);
            assert!(!challenge.active());
        }
    }
    #[test]
    fn ordinary_login_does_not_open_a_lan_listener() {
        let mut attempt = attempt(LoginStatus::Pending, true);
        for status in [0, 1, 2] {
            apply_login_notification(&mut attempt, notification(status, false), |_| {
                panic!("ordinary login opened a socket")
            })
            .unwrap();
            assert!(attempt.lan_login_state().is_none());
            assert!(!attempt.lan_upload_attempted);
        }
    }
    fn state() -> AccountState {
        AccountState {
            schema: 1,
            device: DeviceProfile {
                device_id: vec![1; 16],
                device_type: "UnifiedPCLinux synthetic x86_64".into(),
                client_version: -227271398,
            },
            uin: 123,
            own_username: "synthetic-user".into(),
            cookie: vec![5],
            session_key: vec![6; 16],
            request_key: vec![7; 24],
            response_key: vec![8; 24],
            auto_auth_key: vec![],
            key_info: vec![],
            database_password: vec![],
            auth_snapshot: AuthSnapshot::default(),
            long_host: "long.weixin.qq.com".into(),
            short_host: "short.weixin.qq.com".into(),
        }
    }
    #[test]
    fn challenge_poll_retries_only_transport_availability_failures() {
        use std::io::{Error, ErrorKind};
        for kind in [
            ErrorKind::TimedOut,
            ErrorKind::WouldBlock,
            ErrorKind::ConnectionReset,
            ErrorKind::UnexpectedEof,
        ] {
            assert!(is_retryable_login_poll_error(
                &anyhow::Error::new(Error::from(kind)).context("polling challenge")
            ));
        }
        for message in [
            "MMTLS deadline exceeded",
            "MMTLS connection closed",
            "Native shortlink HTTP status 503",
        ] {
            assert!(is_retryable_login_poll_error(&anyhow::anyhow!("{message}")));
        }
        for message in [
            "Login poll returned -13",
            "Native envelope returned -13",
            "Native shortlink HTTP status 403",
            "Native shortlink HTTP status 5030",
            "Missing authentication tag",
            "Unexpected MMTLS header",
            "Truncated structure",
        ] {
            assert!(!is_retryable_login_poll_error(&anyhow::anyhow!(
                "{message}"
            )));
        }
        assert!(!is_retryable_login_poll_error(
            &Error::from(ErrorKind::PermissionDenied).into()
        ));
    }
    #[test]
    fn secrets_not_debugged() {
        let s = state();
        let text = format!("{s:?}");
        assert!(!text.contains("synthetic-user"));
        assert!(!text.contains("123"));
        assert!(text.contains("redacted"));
    }
    #[test]
    fn account_state_roundtrips_without_number_loss() {
        let s = state();
        let json = serde_json::to_vec(&s).unwrap();
        let restored: AccountState = serde_json::from_slice(&json).unwrap();
        restored.validate().unwrap();
        assert_eq!(restored.uin, s.uin);
    }
    #[test]
    fn database_password_follows_auth_update_gate_and_optional_presence() {
        let mut old = state();
        old.database_password = vec![11; 32];
        let inner = Ecdh::generate(Nid::SECP224R1).unwrap();
        let response = |mask, password: Option<&[u8]>| {
            let mut fields = vec![
                p::uint(1, old.uin as u64),
                p::bytes(3, &p::buffer(&[1; 16])),
                p::uint(19, 0),
            ];
            if let Some(bytes) = password {
                fields.push(p::bytes(9, &p::buffer(bytes)));
            }
            p::message(&[
                p::bytes(1, &p::uint(1, 0)),
                p::uint(2, mask),
                p::bytes(3, &p::message(&fields).unwrap()),
                p::bytes(6, b"synthetic-ax"),
            ])
            .unwrap()
        };
        let gated_off = response(0, Some(&[22; 32]));
        let unchanged = reduce_auth(&gated_off, Some(&old), &inner, &old.device, &[]).unwrap();
        assert_eq!(unchanged.database_password, old.database_password);
        assert_eq!(unchanged.auth_snapshot.update_mask, 0);
        assert_eq!(unchanged.auth_snapshot.response, gated_off);
        assert_eq!(unchanged.auth_snapshot.ax_auth, b"synthetic-ax");
        let accepted = reduce_auth(
            &response(1, Some(&[22; 32])),
            Some(&old),
            &inner,
            &old.device,
            &[],
        )
        .unwrap();
        assert_eq!(accepted.database_password, vec![22; 32]);
        let absent = reduce_auth(
            &response(1, None),
            Some(&accepted),
            &inner,
            &old.device,
            &[],
        )
        .unwrap();
        assert_eq!(absent.database_password, accepted.database_password);
        let empty = reduce_auth(
            &response(1, Some(&[])),
            Some(&absent),
            &inner,
            &old.device,
            &[],
        )
        .unwrap();
        assert!(empty.database_password.is_empty());
        assert!(reduce_auth(
            &response(1, Some(&[22; 31])),
            Some(&old),
            &inner,
            &old.device,
            &[]
        )
        .is_err());
        let mut legacy = serde_json::to_value(&old).unwrap();
        legacy.as_object_mut().unwrap().remove("database_password");
        legacy.as_object_mut().unwrap().remove("auth_snapshot");
        let restored: AccountState = serde_json::from_value(legacy).unwrap();
        restored.validate().unwrap();
        assert!(restored.database_password.is_empty());
    }
    #[test]
    fn refresh_identity_and_clear_tokens() {
        let old = state();
        let inner = Ecdh::generate(Nid::SECP224R1).unwrap();
        let auth = p::message(&[
            p::uint(1, old.uin as u64),
            p::bytes(3, &p::buffer(&[1; 16])),
            p::uint(19, 0),
        ])
        .unwrap();
        let reply = p::message(&[
            p::bytes(1, &p::uint(1, 0)),
            p::uint(2, 1),
            p::bytes(3, &auth),
            p::bytes(4, &p::bytes(1, old.own_username.as_bytes())),
        ])
        .unwrap();
        let next = reduce_auth(&reply, Some(&old), &inner, &old.device, &[]).unwrap();
        assert!(next.auto_auth_key.is_empty());
        assert!(next.request_key.is_empty());
        let bad = p::message(&[
            p::bytes(1, &p::uint(1, 0)),
            p::uint(2, 0),
            p::bytes(4, &p::bytes(1, b"foreign")),
        ])
        .unwrap();
        assert!(reduce_auth(&bad, Some(&old), &inner, &old.device, &[]).is_err());
    }
}