“演示使用PreparedStatement解决SQL注入问题”的版本间的差异

来自姬鸿昌的知识库
跳到导航 跳到搜索
(建立内容为“https://www.bilibili.com/video/BV1eJ411c7rf?p=23”的新页面)
 
第1行: 第1行:
https://www.bilibili.com/video/BV1eJ411c7rf?p=23
+
https://www.bilibili.com/video/BV1eJ411c7rf?p=23<syntaxhighlight lang="java">
 +
import io.github.jihch.statement.crud.User;
 +
import io.github.jihch.util.JDBCUtils;
 +
import org.junit.Test;
 +
 
 +
import java.lang.reflect.Field;
 +
import java.sql.*;
 +
import java.util.HashMap;
 +
import java.util.Map;
 +
import java.util.Scanner;
 +
 
 +
/**
 +
* 演示使用PreparedStatement 替换 Statement 解决SQL 注入问题
 +
*/
 +
public class PreparedStatementTest {
 +
 
 +
    @Test
 +
    public void testLogin() {
 +
 
 +
        Scanner scanner = new Scanner(System.in);
 +
 
 +
        System.out.print("请输入用户名:");
 +
 
 +
        String user = scanner.nextLine();
 +
 
 +
        System.out.print("请输入密码:");
 +
 
 +
        /* 当输入密码的字符串是:  ' or '1' = '1
 +
        * SQL 也会执行并返回数据
 +
        * next() 和 nextLine() 的区别:空格也会被next()方法认为是结束输入的字符,所以这里要用 nextLine()
 +
        */
 +
        String password = scanner.nextLine();
 +
 
 +
        String sql = "SELECT user, password from user_table where user = ? and password = ?";
 +
 
 +
        User user1 = getInstance2(User.class, sql, user, password);
 +
 
 +
        System.out.println(user1);
 +
 
 +
    }
 +
 
 +
    public <T> T getInstance2(Class<T> clazz, String sql, Object... args) {
 +
        T t = null;
 +
 
 +
        Connection conn = null;
 +
        PreparedStatement ps = null;
 +
        ResultSet rs = null;
 +
 
 +
        Map<String, Field> fieldMap = new HashMap<>();
 +
        for (Field declaredField : clazz.getDeclaredFields()) {
 +
            fieldMap.put(declaredField.getName(), declaredField);
 +
        }
 +
 
 +
        try {
 +
            conn = JDBCUtils.getConnection();
 +
            ps = conn.prepareStatement(sql);
 +
 
 +
            for (int i = 0; i < args.length; i++) {
 +
                ps.setObject(i + 1, args[i]);
 +
            }
 +
 
 +
            //执行获取结果集
 +
            rs = ps.executeQuery();
 +
 
 +
            //获取结果集的元数据
 +
            ResultSetMetaData rsmd = rs.getMetaData();
 +
 
 +
            int columnCount = rsmd.getColumnCount();
 +
 
 +
            if (rs.next()) {
 +
 
 +
                t = clazz.newInstance();
 +
 
 +
                for (int i = 0; i < columnCount; i++) {
 +
                    //获取每个列的列值:通过ResultSet
 +
                    Object value = rs.getObject(i + 1);
 +
 
 +
                    //获取每个列的列名:通过ResultSetMetaData
 +
                    //获取列的列名:getColumnName() --不推荐使用
 +
                    //获取列的别名:getColumnLabel()
 +
                    String columnName = rsmd.getColumnName(i + 1);
 +
                    String columnLabel = rsmd.getColumnLabel(i + 1);
 +
                    System.out.printf("columnName:%s, columnLabel:%s\n", columnName, columnLabel);
 +
 
 +
                    //通过反射,将对象指定名 columnName 的属性赋值为指定的值 columnValue
 +
//                    Field field = Order.class.getDeclaredField(columnName);
 +
 
 +
                    if (fieldMap.containsKey(columnLabel)) {
 +
                        Field field = clazz.getDeclaredField(columnLabel);
 +
                        field.setAccessible(true);
 +
                        field.set(t, value);
 +
                    }
 +
                }
 +
                //获取每个列的列名
 +
 
 +
            }//end if
 +
 
 +
        } catch (SQLException e) {
 +
            throw new RuntimeException(e);
 +
 
 +
        } catch (Exception e) {
 +
            throw new RuntimeException(e);
 +
 
 +
        } finally {
 +
            JDBCUtils.closeResource(conn, ps, rs);
 +
 
 +
        }
 +
 
 +
        return t;
 +
 
 +
    }
 +
 
 +
}
 +
</syntaxhighlight>

2022年12月21日 (三) 05:45的版本

https://www.bilibili.com/video/BV1eJ411c7rf?p=23

import io.github.jihch.statement.crud.User;
import io.github.jihch.util.JDBCUtils;
import org.junit.Test;

import java.lang.reflect.Field;
import java.sql.*;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

/**
 * 演示使用PreparedStatement 替换 Statement 解决SQL 注入问题
 */
public class PreparedStatementTest {

    @Test
    public void testLogin() {

        Scanner scanner = new Scanner(System.in);

        System.out.print("请输入用户名:");

        String user = scanner.nextLine();

        System.out.print("请输入密码:");

        /* 当输入密码的字符串是:  ' or '1' = '1
         * SQL 也会执行并返回数据
         * next() 和 nextLine() 的区别:空格也会被next()方法认为是结束输入的字符,所以这里要用 nextLine()
         */
        String password = scanner.nextLine();

        String sql = "SELECT user, password from user_table where user = ? and password = ?";

        User user1 = getInstance2(User.class, sql, user, password);

        System.out.println(user1);

    }

    public <T> T getInstance2(Class<T> clazz, String sql, Object... args) {
        T t = null;

        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;

        Map<String, Field> fieldMap = new HashMap<>();
        for (Field declaredField : clazz.getDeclaredFields()) {
            fieldMap.put(declaredField.getName(), declaredField);
        }

        try {
            conn = JDBCUtils.getConnection();
            ps = conn.prepareStatement(sql);

            for (int i = 0; i < args.length; i++) {
                ps.setObject(i + 1, args[i]);
            }

            //执行获取结果集
            rs = ps.executeQuery();

            //获取结果集的元数据
            ResultSetMetaData rsmd = rs.getMetaData();

            int columnCount = rsmd.getColumnCount();

            if (rs.next()) {

                t = clazz.newInstance();

                for (int i = 0; i < columnCount; i++) {
                    //获取每个列的列值:通过ResultSet
                    Object value = rs.getObject(i + 1);

                    //获取每个列的列名:通过ResultSetMetaData
                    //获取列的列名:getColumnName() --不推荐使用
                    //获取列的别名:getColumnLabel()
                    String columnName = rsmd.getColumnName(i + 1);
                    String columnLabel = rsmd.getColumnLabel(i + 1);
                    System.out.printf("columnName:%s, columnLabel:%s\n", columnName, columnLabel);

                    //通过反射,将对象指定名 columnName 的属性赋值为指定的值 columnValue
//                    Field field = Order.class.getDeclaredField(columnName);

                    if (fieldMap.containsKey(columnLabel)) {
                        Field field = clazz.getDeclaredField(columnLabel);
                        field.setAccessible(true);
                        field.set(t, value);
                    }
                }
                //获取每个列的列名

            }//end if

        } catch (SQLException e) {
            throw new RuntimeException(e);

        } catch (Exception e) {
            throw new RuntimeException(e);

        } finally {
            JDBCUtils.closeResource(conn, ps, rs);

        }

        return t;

    }

}