Java代碼審計詳解

一、SQL注入漏洞

SQL注入漏洞是Java代碼審計中最常見的漏洞之一。這種漏洞是由於程序在動態生成SQL語句時未對用戶輸入進行過濾，從而導致攻擊者可以注入任意SQL代碼。以下是一個SQL注入漏洞的代碼示例：

String sql = "SELECT * FROM users WHERE username='" + username + "' AND password='" + password + "'";
PreparedStatement statement = connection.prepareStatement(sql);
ResultSet result = statement.executeQuery();

在上述代碼中，我們可以看到，程序是將參數直接拼接到SQL語句中，攻擊者可以利用這一點來注入惡意代碼。為了避免這種漏洞，我們應該使用參數化查詢，如下所示：

String sql = "SELECT * FROM users WHERE username=? AND password=?";
PreparedStatement statement = connection.prepareStatement(sql);
statement.setString(1, username);
statement.setString(2, password);
ResultSet result = statement.executeQuery();

注意，上述代碼中使用了佔位符，將實際參數傳遞給prepared statement，從而避免了SQL注入攻擊。

二、XSS漏洞

XSS漏洞是另一種常見的Java代碼漏洞。這種漏洞通常發生在Web應用程序中，攻擊者通過注入可執行腳本來竊取用戶信息或偽造用戶輸入。以下是一個XSS漏洞的代碼示例：

<input type="text" name="search" value="<% out.println(request.getParameter("search")); %>" />

在上述代碼中，我們可以看到，程序直接將用戶輸入輸出到HTML頁面上，攻擊者可以在用戶輸入中注入惡意腳本。為了避免這種漏洞，我們應該使用HTML實體化來轉義用戶輸入，如下所示：

<input type="text" name="search" value="<% out.println(escapeHTML(request.getParameter("search"))); %>" />

private static final String[][] HTML_ENTITIES = {
    {"\"", """},
    {"&", "&"},
    {"", ">"}
};

private static String escapeHTML(String s) {
    for (String[] pair : HTML_ENTITIES) {
        s = s.replaceAll(pair[0], pair[1]);
    }
    return s;
}

上述代碼中使用了escapeHTML方法來將HTML特殊字符轉義為實體。

三、敏感信息泄露

Java代碼中還可能存在敏感信息泄露的漏洞。這種漏洞通常發生在未正確配置系統或應用程序中。以下是一個敏感信息泄露漏洞的代碼示例：

String password = "password123";
OutputStream os = new FileOutputStream("config.txt");
os.write(password.getBytes());
os.flush();
os.close();

在上述代碼中，程序直接將敏感信息寫入到文件中，導致任何人都可以讀取文件並獲取密碼。為了避免這種漏洞，我們應該使用安全的存儲方式，如使用加密算法對敏感信息進行加密。

String password = "password123";
byte[] encrypted = encrypt(password);
OutputStream os = new FileOutputStream("config.txt");
os.write(encrypted);
os.flush();
os.close();

private static byte[] encrypt(String data) {
    String key = "mysecretkey";
    Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
    SecretKeySpec keySpec = new SecretKeySpec(key.getBytes(), "AES");
    IvParameterSpec ivSpec = new IvParameterSpec(key.getBytes());
    cipher.init(Cipher.ENCRYPT_MODE, keySpec, ivSpec);
    return cipher.doFinal(data.getBytes());
}

上述代碼中使用了AES加密算法對密碼進行加密，使用密鑰mysecretkey。

四、HTTP響應-splitting漏洞

HTTP響應-splitting漏洞是一種攻擊，攻擊者可以利用此漏洞將HTTP響應拆分為多個部分，從而繞過跨站點腳本(XSS)和其他安全控制。以下是一個HTTP響應-splitting漏洞的代碼示例：

String input = request.getParameter("input");
response.getWriter().write("Input received: " + input);

在上述代碼中，程序直接將用戶輸入輸出到HTTP響應中，攻擊者可以在用戶輸入中注入回車符，從而將HTTP響應拆分成多個部分，繞過安全控制。為了避免這種漏洞，我們應該將用戶輸入轉義為安全字符，如下所示：

String input = request.getParameter("input");
response.getWriter().write("Input received: " + input.replaceAll("\\r|\\n", ""));

上述代碼中使用了replaceAll方法將用戶輸入中的回車符替換為安全字符，避免了HTTP響應-splitting漏洞。

五、文件包含漏洞

Java代碼中還可能存在文件包含漏洞。這種漏洞通常發生在程序未正確檢查用戶輸入，從而導致攻擊者可以在程序中包含惡意文件。以下是一個文件包含漏洞的代碼示例：

String page = request.getParameter("page");
File file = new File(page);
Scanner scanner = new Scanner(file);
while (scanner.hasNext()) {
    response.getWriter().write(scanner.nextLine());
}

在上述代碼中，程序直接從用戶輸入中獲取文件名，並包含該文件。攻擊者可以注入任意文件名，並包含惡意文件。為了避免這種漏洞，我們應該檢查文件名並限制文件訪問，如下所示：

String page = request.getParameter("page");
if (!page.endsWith(".html")) {
    throw new SecurityException("Invalid page requested");
}
File file = new File("/path/to/files/" + page);
if (!file.exists()) {
    throw new FileNotFoundException();
}
BufferedReader reader = new BufferedReader(new FileReader(file));
String line;
while ((line = reader.readLine()) != null) {
    response.getWriter().write(line);
}
reader.close();

上述代碼中使用了endsWith方法和文件路徑檢查來限制文件訪問，並使用BufferedReader和FileReader從文件中讀取內容。