从Java角度修复SQL注入漏洞

　　　很多情况因为过滤不严导致很多网站存在sql注入，这里以用户登陆为例，简单举例

　　　　　　　首先创建一个测试的数据库

　　　　　　　　比较基础，不写创建过程了

　　　　　　java代码如下:

package cn.basic.jdbc;

import java.awt.image.RescaleOp;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import org.junit.jupiter.api.Test;

public class Test1 {
    /*
     * public static void main(String[] args) throws ClassNotFoundException,
     * SQLException { login("aa","aa"); }
     */
    @Test
    public void testlogin() throws ClassNotFoundException, SQLException {
        login("aa", "aa");
    }

    public static void login(String username, String password) throws SQLException, ClassNotFoundException {
        Class.forName("com.mysql.jdbc.Driver");
        Connection conn = DriverManager.getConnection("jdbc:mysql://127.0.0.1/test", "root", "root");
        Statement st = conn.createStatement();
        String sql = "select * from xxx where username=" + "'" + username + "'" + "and password=" + "'" + password
                + "'";
        ResultSet rs = st.executeQuery(sql);
        if (rs.next()) {
            System.out.println("恭喜" + username + "登陆成功");
            System.out.println(sql);
        } else {
            System.out.println("登录失败");
        }
        if (rs != null) {
            rs.close();
        }

        /*
         * if (st != null) { st.close(); }
         */
        if (conn != null) {
            conn.close();
        }
    }

}

　　　　　　运行

输入正确账号密码可以登陆成功。这里可以被绕过。

很显然这是万能密码。那么如何去修复sql注入呢，这里比较好的方法是采用预编译的开发方式，这是个开发习惯问题。

　　　　修复后的预编译代码如下:

package cn.basic.jdbc;

import java.awt.image.RescaleOp;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import org.junit.jupiter.api.Test;

public class Test1 {
    /*
     * public static void main(String[] args) throws ClassNotFoundException,
     * SQLException { login("aa","aa"); }
     */
    @Test
    public void testlogin() throws ClassNotFoundException, SQLException {
        login("aa", "aa");
    }

    public static void login(String username, String password) throws SQLException, ClassNotFoundException {
        Class.forName("com.mysql.jdbc.Driver");
        Connection conn = DriverManager.getConnection("jdbc:mysql://127.0.0.1/test", "root", "root");
        String sql="select * from xxx where username=? and password=?";
        //使用预编译
        PreparedStatement ps = conn.prepareStatement(sql);
        ps.setString(1, username);
        ps.setString(2, password);
        ResultSet rs = ps.executeQuery();
        if (rs.next()) {
            System.out.println("恭喜" + username + "登陆成功");
            System.out.println(sql);
        } else {
            System.out.println("登录失败");
        }
        if (rs != null) {
            rs.close();
        }
        if(ps!=null) {
            ps.close();
        }

        /*
         * if (st != null) { st.close(); }
         */
        if (conn != null) {
            conn.close();
        }
    }

}