“演示使用PreparedStatement解决SQL注入问题”的版本间的差异
跳到导航
跳到搜索
Jihongchang(讨论 | 贡献) (建立内容为“https://www.bilibili.com/video/BV1eJ411c7rf?p=23”的新页面) |
Jihongchang(讨论 | 贡献) |
||
(未显示同一用户的2个中间版本) | |||
第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 注入问题 | ||
+ | * | ||
+ | * 除了解决 Statement 的拼串、sql问题之外,PreparedStatement 还有哪些好处呢? | ||
+ | * 1.PreparedStatement 操作 Blob 的数据,而Statement做不到。 | ||
+ | * 2.PreparedStatement 可以实现更高效地批量操作 | ||
+ | */ | ||
+ | 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>https://github.com/jihch/jdbc/blob/main/src/main/java/io/github/jihch/preparedstatement/crud/PreparedStatementTest.java |
2022年12月21日 (三) 05:54的最新版本
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 注入问题
*
* 除了解决 Statement 的拼串、sql问题之外,PreparedStatement 还有哪些好处呢?
* 1.PreparedStatement 操作 Blob 的数据,而Statement做不到。
* 2.PreparedStatement 可以实现更高效地批量操作
*/
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;
}
}