first commit

admin

2 天以前 6e4a6789e948bc6a8dc9fc1aa3eebc2980be3072

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
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import java.util.Scanner;
import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import java.util.Base64;
 
@SpringBootApplication
public class RuoYiApplication {
    // 加密密钥，实际使用时请更换为更复杂的密钥
    private static final String SECRET_KEY = "YourSecretKey123";
    // 加密后的验证码，可以使用 generateEncryptedCode() 方法生成
    private static final String ENCRYPTED_CODE = "YOUR_ENCRYPTED_CODE_HERE";
 
    public static void main(String[] args) {
        if (!validateStartup()) {
            System.out.println("验证失败，程序退出！");
            System.exit(1);
        }
        SpringApplication.run(RuoYiApplication.class, args);
    }
 
    private static boolean validateStartup() {
        try {
            Scanner scanner = new Scanner(System.in);
            System.out.print("请输入启动密码: ");
            String input = scanner.nextLine();
            
            // 加密输入的密码
            String encryptedInput = encrypt(input);
            
            // 比较加密后的结果
            return ENCRYPTED_CODE.equals(encryptedInput);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }
 
    // AES加密方法
    private static String encrypt(String value) throws Exception {
        SecretKeySpec key = new SecretKeySpec(SECRET_KEY.getBytes("UTF-8"), "AES");
        Cipher cipher = Cipher.getInstance("AES");
        cipher.init(Cipher.ENCRYPT_MODE, key);
        byte[] encrypted = cipher.doFinal(value.getBytes());
        return Base64.getEncoder().encodeToString(encrypted);
    }
 
    // 用于生成加密后的验证码
    public static void generateEncryptedCode(String password) throws Exception {
        System.out.println("Encrypted code: " + encrypt(password));
    }
}