需求

最近在后臺項目代碼中一段自定義的AES加解密的程序在平時的測試工作中應(yīng)用頻繁。因為寫腳本經(jīng)常會需要使用，而經(jīng)過各種嘗試，比如jpype等，都不盡如人意。最后轉(zhuǎn)換思路找到了一種好方法：

1. 將java項目打成jar包；

2. 設(shè)定jar包的Main-class屬性；

3. 在python中通過subprocess執(zhí)行jar包。

修改main函數(shù)打jar包

在intellij中打jar包：

1. File--> Project Structure... --> Artifacts --> + --> JAR --> From modules with dependencies --> OK --> 勾選Include in project build --> 記錄上方Output directory路徑 --> OK；

2. Build --> Build Project --> 之后就可以在Output directory路徑看到打的jar包了；

3. 用WinRAR打開jar包，修改其中META-INF文件夾下的MANIFEST.MF文件，增加一行Main-class入口，例如："Main-class: com.package.name.Classname"，注意冒號后面空一格，文件最后空一行。

注意：執(zhí)行jar包就是執(zhí)行jar包的Main-class中的main函數(shù)。所以如果要在python中調(diào)用java方法并傳參的話，要在打jar包前修改Main-class中的main方法，Main-class也就是目標(biāo)java類。

在Main-class的main函數(shù)中可以設(shè)置傳參：

String data= args[0];
String key = args[1];
// 調(diào)用解密程序并打印解密結(jié)果

這里data是密文，key是密鑰。

編寫python腳本

然后對應(yīng)寫python腳本，這里設(shè)定python文件名為aes.py，假定我們用來解密的jar包名為aes.jar：

import subprocess
import chardet
import sys

class AES(object):
  def __init__(self, data, key):
    self.data = data
    self.key = key

  def decrypt(self):    
    command = "java -jar aes.jar"
    arg0 = self.data
    arg1 = self.key
    cmd = [command,arg0,arg1]
    new_cmd = " ".join(cmd)
    stdout,stderr = subprocess.Popen(new_cmd,stdout=subprocess.PIPE,stderr=subprocess.PIPE).communicate()
    encoding = chardet.detect(stdout)["encoding"]
    result = stdout.decode(encoding)
    return result

if __name__ == '__main__':
  data = sys.argv[1]
  key = sys.argv[2]
  AES = AES(data,key)
  print(AES.decrypt())

這里為了與jar包的兩個傳參相對應(yīng)也設(shè)置python腳本需要傳入兩個參數(shù)，第一個是待解密的數(shù)據(jù)data，第二個是解密鑰匙key。

這里我們創(chuàng)建一個文件夾aes，將aes.py和aes.jar放到該文件夾下，二者同處于aes根目錄下。

然后在當(dāng)前目錄下啟動cmd，執(zhí)行python腳本，傳入data和key兩個參數(shù)給python腳本：

python aes.py "待解密的字符串" "密鑰"

在python腳本會將這兩個參數(shù)再傳遞給jar包并調(diào)用jar包Main-class的main函數(shù)中調(diào)用解密過程并打印解密結(jié)果。

這樣就可以在python中調(diào)用java的解密程序，從而打印出解密后的字符串了。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。