import rsa
# 生成 2048 位密鑰
(public_key, private_key) = rsa.newkeys(2048)
# 保存私鑰（供授權(quán)中心使用）
with open("private.pem", "wb") as priv_file:
    priv_file.write(private_key.save_pkcs1())
# 保存公鑰（供客戶端使用）
with open("public.pem", "wb") as pub_file:
    pub_file.write(public_key.save_pkcs1())
print("RSA 密鑰對(duì)已生成：private.pem（私鑰） 和 public.pem（公鑰）")

import rsa
import json
import base64
import os
from datetime import datetime
# 讀取私鑰
with open("private.pem", "rb") as priv_file:
    private_key = rsa.PrivateKey.load_pkcs1(priv_file.read())
# 獲取目標(biāo)設(shè)備的 CPU ID
# def get_machine_id():
    # try:
        # if os.name == 'nt':
            # machine_id = os.popen("wmic cpu get ProcessorId").read().split()[1]
            # print('授權(quán)設(shè)備的唯一標(biāo)識(shí):'+machine_id)
        # else:
            # return os.popen("cat /etc/machine-id").read().strip()
    # except:
        # return "UNKNOWN"
# 生成授權(quán)文件
def generate_license(user, expire_date, machine_id):
    license_data = {
        "user": user,
        "expire": expire_date,
        "machine_id": machine_id
    }
    print(license_data)
    # 簽名授權(quán)數(shù)據(jù)
    data_str = f"{user}|{expire_date}|{machine_id}".encode()
    signature = rsa.sign(data_str, private_key, 'SHA-256')
    # 添加簽名并保存 JSON
    license_data["signature"] = base64.b64encode(signature).decode()
    with open("license.json", "w") as license_file:
        json.dump(license_data, license_file, indent=4)
    print("? 授權(quán)文件 license.json 生成成功！")
# 示例：為某個(gè)客戶生成授權(quán)
generate_license(user="client1", expire_date="2025-12-31", machine_id="機(jī)器碼")

{
    "user": "client1",
    "expire": "2025-12-31",
    "machine_id": "BFE***ED",
    "signature": "R62******8/w=="
}

import os
import time
import threading
import json
import tkinter as tk
from tkinter import messagebox, ttk
from datetime import datetime
import boto3
import configparser
from elasticsearch import Elasticsearch
from urllib3.exceptions import InsecureRequestWarning
import warnings
import base64
import logging  # 添加 logging 模塊
import rsa
import hashlib
from botocore.exceptions import ClientError, EndpointConnectionError
warnings.filterwarnings("ignore", category=InsecureRequestWarning)
# ==================== 配置日志 ====================
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s',
    filename=f'logs/file_search_{datetime.now().strftime("%Y%m%d_%H%M%S")}.log',
    filemode='w'
)
logger = logging.getLogger(__name__)
# ==================== 授權(quán)管理 ====================
LICENSE_FILE = "license.json"
PUBLIC_KEY_FILE = "public.pem"
def get_machine_id():
    """ 獲取當(dāng)前設(shè)備的唯一標(biāo)識(shí)（CPU ID）"""
    try:
        if os.name == 'nt':
            machine_id = os.popen("wmic cpu get ProcessorId").read().split()[1]
            print('獲取當(dāng)前設(shè)備的唯一標(biāo)識(shí):'+machine_id)
            return machine_id
        else:
            return os.popen("cat /etc/machine-id").read().strip()
    except:
        return "UNKNOWN"
def load_rsa_public_key():
    """ 加載 RSA 公鑰 """
    try:
        with open(PUBLIC_KEY_FILE, "rb") as f:
            return rsa.PublicKey.load_pkcs1(f.read())
    except Exception as e:
        logger.error(f"無(wú)法加載公鑰: {e}")
        messagebox.showerror("授權(quán)錯(cuò)誤", "授權(quán)系統(tǒng)異常，無(wú)法加載公鑰")
        exit(1)
def verify_license():
    """ 驗(yàn)證授權(quán)碼 """
    if not os.path.exists(LICENSE_FILE):
        messagebox.showerror("授權(quán)錯(cuò)誤", "未找到授權(quán)文件，請(qǐng)聯(lián)系管理員")
        exit(1)
    try:
        with open(LICENSE_FILE, "r") as f:
            license_data = json.load(f)
        user = license_data.get("user")
        expire = license_data.get("expire")
        machine_id = license_data.get("machine_id")
        signature = base64.b64decode(license_data.get("signature"))
        if not all([user, expire, machine_id, signature]):
            messagebox.showerror("授權(quán)錯(cuò)誤", "授權(quán)文件損壞")
            exit(1)
        # 校驗(yàn)過(guò)期時(shí)間
        if datetime.strptime(expire, "%Y-%m-%d") < datetime.now():
            messagebox.showerror("授權(quán)錯(cuò)誤", "授權(quán)已過(guò)期，請(qǐng)聯(lián)系管理員續(xù)費(fèi)")
            exit(1)
        # 校驗(yàn)機(jī)器碼
        if machine_id != get_machine_id():
            print(machine_id)
            print(get_machine_id())
            messagebox.showerror("授權(quán)錯(cuò)誤", "設(shè)備不匹配，授權(quán)無(wú)效")
            exit(1)
        # 驗(yàn)證簽名
        public_key = load_rsa_public_key()
        data_str = f"{user}|{expire}|{machine_id}".encode()
        try:
            rsa.verify(data_str, signature, public_key)
        except rsa.VerificationError:
            messagebox.showerror("授權(quán)錯(cuò)誤", "授權(quán)碼無(wú)效或被篡改")
            exit(1)
        logger.info("授權(quán)驗(yàn)證成功")
    except Exception as e:
        logger.error(f"授權(quán)驗(yàn)證失敗: {e}")
        messagebox.showerror("授權(quán)錯(cuò)誤", "授權(quán)文件無(wú)效")
        exit(1)