portable-network-archive 0.37.0

use crate::utils::{EmbedExt, TestResources, diff::assert_dirs_equal, setup};
use clap::Parser;
use portable_network_archive::cli;

#[test]
fn aes_ctr_archive() {
    setup();
    TestResources::extract_in("raw/", "zstd_aes_ctr/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_aes_ctr/zstd_aes_ctr.pna",
        "--overwrite",
        "zstd_aes_ctr/in/",
        "--password",
        "password",
        "--aes",
        "ctr",
    ])
    .unwrap()
    .execute()
    .unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_aes_ctr/zstd_aes_ctr.pna",
        "--overwrite",
        "--out-dir",
        "zstd_aes_ctr/out/",
        "--password",
        "password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute()
    .unwrap();

    assert_dirs_equal("zstd_aes_ctr/in/", "zstd_aes_ctr/out/");
}

#[test]
fn aes_cbc_archive() {
    setup();
    TestResources::extract_in("raw/", "zstd_aes_cbc/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_aes_cbc/zstd_aes_cbc.pna",
        "--overwrite",
        "zstd_aes_cbc/in/",
        "--password",
        "password",
        "--aes",
        "cbc",
    ])
    .unwrap()
    .execute()
    .unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_aes_cbc/zstd_aes_cbc.pna",
        "--overwrite",
        "--out-dir",
        "zstd_aes_cbc/out/",
        "--password",
        "password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute()
    .unwrap();

    assert_dirs_equal("zstd_aes_cbc/in/", "zstd_aes_cbc/out/");
}

#[test]
fn camellia_ctr_archive() {
    setup();
    TestResources::extract_in("raw/", "zstd_camellia_ctr/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_camellia_ctr/zstd_camellia_ctr.pna",
        "--overwrite",
        "zstd_camellia_ctr/in/",
        "--password",
        "password",
        "--camellia",
        "ctr",
    ])
    .unwrap()
    .execute()
    .unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_camellia_ctr/zstd_camellia_ctr.pna",
        "--overwrite",
        "--out-dir",
        "zstd_camellia_ctr/out/",
        "--password",
        "password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute()
    .unwrap();

    assert_dirs_equal("zstd_camellia_ctr/in/", "zstd_camellia_ctr/out/");
}

#[test]
fn camellia_cbc_archive() {
    setup();
    TestResources::extract_in("raw/", "zstd_camellia_cbc/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_camellia_cbc/zstd_camellia_cbc.pna",
        "--overwrite",
        "zstd_camellia_cbc/in/",
        "--password",
        "password",
        "--camellia",
        "cbc",
    ])
    .unwrap()
    .execute()
    .unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_camellia_cbc/zstd_camellia_cbc.pna",
        "--overwrite",
        "--out-dir",
        "zstd_camellia_cbc/out/",
        "--password",
        "password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute()
    .unwrap();

    assert_dirs_equal("zstd_camellia_cbc/in/", "zstd_camellia_cbc/out/");
}

#[test]
fn aes_gcm_archive() {
    setup();
    TestResources::extract_in("raw/", "zstd_aes_gcm/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_aes_gcm/zstd_aes_gcm.pna",
        "--overwrite",
        "zstd_aes_gcm/in/",
        "--password",
        "password",
        "--aes",
        "gcm",
    ])
    .unwrap()
    .execute()
    .unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_aes_gcm/zstd_aes_gcm.pna",
        "--overwrite",
        "--out-dir",
        "zstd_aes_gcm/out/",
        "--password",
        "password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute()
    .unwrap();

    assert_dirs_equal("zstd_aes_gcm/in/", "zstd_aes_gcm/out/");
}

#[test]
fn camellia_gcm_archive() {
    setup();
    TestResources::extract_in("raw/", "zstd_camellia_gcm/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_camellia_gcm/zstd_camellia_gcm.pna",
        "--overwrite",
        "zstd_camellia_gcm/in/",
        "--password",
        "password",
        "--camellia",
        "gcm",
    ])
    .unwrap()
    .execute()
    .unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_camellia_gcm/zstd_camellia_gcm.pna",
        "--overwrite",
        "--out-dir",
        "zstd_camellia_gcm/out/",
        "--password",
        "password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute()
    .unwrap();

    assert_dirs_equal("zstd_camellia_gcm/in/", "zstd_camellia_gcm/out/");
}

/// Precondition: Archive is encrypted with AES-GCM (AEAD) and a password.
/// Action: Extract with an incorrect password.
/// Expectation: Extraction fails and reports a key mismatch rather than a
/// corrupted datastream (a distinction CBC/CTR modes cannot make).
#[test]
fn aes_gcm_archive_wrong_password_is_key_mismatch() {
    setup();
    TestResources::extract_in("raw/", "zstd_aes_gcm_wrong_password/in/").unwrap();
    cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "c",
        "-f",
        "zstd_aes_gcm_wrong_password/zstd_aes_gcm_wrong_password.pna",
        "--overwrite",
        "zstd_aes_gcm_wrong_password/in/",
        "--password",
        "password",
        "--aes",
        "gcm",
    ])
    .unwrap()
    .execute()
    .unwrap();
    let result = cli::Cli::try_parse_from([
        "pna",
        "--quiet",
        "x",
        "-f",
        "zstd_aes_gcm_wrong_password/zstd_aes_gcm_wrong_password.pna",
        "--overwrite",
        "--out-dir",
        "zstd_aes_gcm_wrong_password/out/",
        "--password",
        "wrong_password",
        "--strip-components",
        "2",
    ])
    .unwrap()
    .execute();

    let err = result.unwrap_err();
    assert!(
        format!("{err:?}").contains("key mismatch"),
        "error should be classified as a key mismatch: {err:?}"
    );
}