106 lines
3.5 KiB
C++
106 lines
3.5 KiB
C++
#include <iostream>
|
||
#include <string>
|
||
#include <regex>
|
||
|
||
// Powered by ChatGPT.
|
||
|
||
enum WalletType {
|
||
WALLET_UNKNOWN = 0,
|
||
WALLET_BTC_P2PKH,
|
||
WALLET_BTC_P2SH,
|
||
WALLET_BTC_BECH32,
|
||
WALLET_ETH_ERC20, // ETH、ERC20（含 USDT-ERC20）
|
||
WALLET_USDT_OMNI, // USDT Omni，BTC 网络，格式同 BTC
|
||
WALLET_USDT_TRC20, // USDT TRC20
|
||
WALLET_TRON,
|
||
WALLET_SOLANA,
|
||
WALLET_XRP,
|
||
WALLET_POLKADOT,
|
||
WALLET_CARDANO_SHELLEY,
|
||
WALLET_CARDANO_BYRON,
|
||
WALLET_DOGE // Dogecoin
|
||
};
|
||
|
||
enum AddressType {
|
||
ADDR_BTC = 0,
|
||
ADDR_ERC20,
|
||
ADDR_OMNI,
|
||
ADDR_TRC20,
|
||
ADDR_SOL,
|
||
ADDR_XRP,
|
||
ADDR_ADA,
|
||
ADDR_DOGE,
|
||
ADDR_DOT,
|
||
ADDR_TRON,
|
||
MAX_WALLET_NUM,
|
||
};
|
||
|
||
inline WalletType detectWalletType(const std::string& address_raw) {
|
||
std::string address = address_raw;
|
||
address.erase(0, address.find_first_not_of(" \t\n\r"));
|
||
address.erase(address.find_last_not_of(" \t\n\r") + 1);
|
||
|
||
// 1. ETH/ERC20（0x 开头）
|
||
static const std::regex eth_regex("^0x[a-fA-F0-9]{40}$");
|
||
if (std::regex_match(address, eth_regex)) return WALLET_ETH_ERC20;
|
||
|
||
// 2. TRC20（T 开头）
|
||
static const std::regex trc20_regex("^T[1-9A-HJ-NP-Za-km-z]{33}$");
|
||
if (std::regex_match(address, trc20_regex)) return WALLET_USDT_TRC20;
|
||
|
||
// 3. BTC Bech32（bc1 开头）
|
||
static const std::regex btc_bech32_regex("^bc1[0-9a-z]{6,}$");
|
||
if (std::regex_match(address, btc_bech32_regex)) return WALLET_BTC_BECH32;
|
||
|
||
// 4. BTC P2PKH（1 开头）
|
||
static const std::regex btc_p2pkh_regex("^1[1-9A-HJ-NP-Za-km-z]{25,34}$");
|
||
if (std::regex_match(address, btc_p2pkh_regex)) return WALLET_BTC_P2PKH;
|
||
|
||
// 5. BTC P2SH（3 开头）
|
||
static const std::regex btc_p2sh_regex("^3[1-9A-HJ-NP-Za-km-z]{25,34}$");
|
||
if (std::regex_match(address, btc_p2sh_regex)) return WALLET_BTC_P2SH;
|
||
|
||
// 6. XRP（r 开头，Base58）
|
||
static const std::regex xrp_regex("^r[1-9A-HJ-NP-Za-km-z]{24,34}$");
|
||
if (std::regex_match(address, xrp_regex)) return WALLET_XRP;
|
||
|
||
// 7. Dogecoin（D 开头，Base58）
|
||
static const std::regex doge_regex("^D[5-9A-HJ-NP-Ua-km-z]{33}$");
|
||
if (std::regex_match(address, doge_regex)) return WALLET_DOGE;
|
||
|
||
// 8. Cardano Shelley（addr1 开头）
|
||
static const std::regex ada_shelley_regex("^addr1[0-9a-z]{20,}$");
|
||
if (std::regex_match(address, ada_shelley_regex)) return WALLET_CARDANO_SHELLEY;
|
||
|
||
// 9. Cardano Byron（DdzFF 开头）
|
||
if (address.find("DdzFF") == 0) return WALLET_CARDANO_BYRON;
|
||
|
||
// 10. Polkadot（长度 47–48，Base58）
|
||
static const std::regex dot_regex("^[1-9A-HJ-NP-Za-km-z]{47,48}$");
|
||
if (std::regex_match(address, dot_regex)) return WALLET_POLKADOT;
|
||
|
||
// 11. Solana（32–44，无前缀，Base58）→ 容易误判，必须放最后
|
||
static const std::regex solana_regex("^[1-9A-HJ-NP-Za-km-z]{32,44}$");
|
||
if (std::regex_match(address, solana_regex)) return WALLET_SOLANA;
|
||
|
||
return WALLET_UNKNOWN;
|
||
}
|
||
|
||
inline std::string walletTypeToString(WalletType type) {
|
||
switch (type) {
|
||
case WALLET_BTC_P2PKH: return "Bitcoin P2PKH (includes USDT-OMNI)";
|
||
case WALLET_BTC_P2SH: return "Bitcoin P2SH (includes USDT-OMNI)";
|
||
case WALLET_BTC_BECH32: return "Bitcoin Bech32";
|
||
case WALLET_ETH_ERC20: return "Ethereum / ERC20 (includes USDT-ERC20)";
|
||
case WALLET_USDT_TRC20: return "USDT TRC20";
|
||
case WALLET_TRON: return "TRON (same as USDT-TRC20)";
|
||
case WALLET_SOLANA: return "Solana";
|
||
case WALLET_XRP: return "XRP";
|
||
case WALLET_POLKADOT: return "Polkadot";
|
||
case WALLET_CARDANO_SHELLEY: return "Cardano Shelley";
|
||
case WALLET_CARDANO_BYRON: return "Cardano Byron";
|
||
case WALLET_DOGE: return "Dogecoin";
|
||
default: return "Unknown or Unsupported";
|
||
}
|
||
}
|