trojan 0.10.0

A Rust implementation of the Trojan protocol
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

//! Unified trojan CLI.
//!
//! This binary provides a unified interface to all trojan components:
//! - `trojan server` - Run the trojan server
//! - `trojan client` - Run the trojan client (SOCKS5 proxy)
//! - `trojan entry` - Run the relay chain entry node (A)
//! - `trojan relay` - Run the relay chain relay node (B)
//! - `trojan auth` - Manage authentication users (SQL backend)
//! - `trojan cert` - Generate self-signed certificates (requires `cert` feature)
//! - `trojan upgrade` - Self-upgrade from GitHub releases (requires `upgrade` feature)
//!
//! Each subcommand can also be run as a standalone binary.

use std::process::ExitCode;

use clap::{Parser, Subcommand};

#[cfg(feature = "upgrade")]
mod upgrade;

/// Trojan unified CLI.
#[derive(Parser)]
#[command(
    name = "trojan",
    version,
    about = "A Rust implementation of the Trojan protocol",
    propagate_version = true
)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Run the trojan server.
    #[command(name = "server", alias = "serve")]
    Server(Box<trojan_server::ServerArgs>),

    /// Run the trojan client (SOCKS5 proxy).
    #[command(name = "client")]
    Client(trojan_client::ClientArgs),

    /// Run the relay chain entry node (A).
    #[command(name = "entry")]
    Entry(trojan_relay::cli::EntryArgs),

    /// Run the relay chain relay node (B).
    #[command(name = "relay")]
    Relay(trojan_relay::cli::RelayArgs),

    /// Manage authentication users (SQL backend).
    #[command(name = "auth")]
    Auth(trojan_auth::AuthArgs),

    /// Run the panel agent (managed mode).
    #[cfg(feature = "agent")]
    #[command(name = "agent")]
    Agent(trojan_agent::AgentArgs),

    /// Generate and manage TLS certificates.
    #[cfg(feature = "cert")]
    #[command(name = "cert")]
    Cert(trojan_cert::CertArgs),

    /// Upgrade to latest version from GitHub releases.
    #[cfg(feature = "upgrade")]
    #[command(name = "upgrade", alias = "update")]
    Upgrade(upgrade::UpgradeArgs),
}

#[tokio::main]
async fn main() -> ExitCode {
    rustls::crypto::aws_lc_rs::default_provider()
        .install_default()
        .expect("failed to install default CryptoProvider");

    let cli = Cli::parse();

    let result: Result<(), String> = match cli.command {
        Commands::Server(args) => trojan_server::cli::run(*args)
            .await
            .map_err(|e| e.to_string()),
        Commands::Client(args) => trojan_client::cli::run(args)
            .await
            .map_err(|e| e.to_string()),
        Commands::Entry(args) => trojan_relay::cli::run_entry(args)
            .await
            .map_err(|e| e.to_string()),
        Commands::Relay(args) => trojan_relay::cli::run_relay(args)
            .await
            .map_err(|e| e.to_string()),
        Commands::Auth(args) => trojan_auth::cli::run(args).await.map_err(|e| e.to_string()),
        #[cfg(feature = "agent")]
        Commands::Agent(args) => trojan_agent::cli::run(args)
            .await
            .map_err(|e| e.to_string()),
        #[cfg(feature = "cert")]
        Commands::Cert(args) => trojan_cert::run(args).map_err(|e| e.to_string()),
        #[cfg(feature = "upgrade")]
        Commands::Upgrade(args) => upgrade::run(args).await.map_err(|e| e.to_string()),
    };

    match result {
        Ok(()) => ExitCode::SUCCESS,
        Err(e) => {
            eprintln!("Error: {}", e);
            ExitCode::FAILURE
        }
    }
}