实现简易用户登录以及SQL注入现象
准备数据
1、设计表
2、加入相关信息
代码
package jdbc;
import java.sql.*;
import java.util.ResourceBundle;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
public class simpLogin {
public static void main(String[] args) {
//初始化界面
Map<String,String> userLoginInfo = initUI();
//验证用户名和密码
boolean loginSuccess = login(userLoginInfo);
System.out.println(loginSuccess ? "登录成功" : "登录失败");
}
/**
* @param userLoginInfo 用户登录信息
* @return false表示失败，true表示成功
*/
private static boolean login(Map<String, String> userLoginInfo) {
boolean loginSuccess = false;
String loginName = userLoginInfo.get("loginName");
String loginPwd = userLoginInfo.get("loginPwd");
//JDBC代码
ResourceBundle bundle = ResourceBundle.getBundle("jdbc.info");
String driver = bundle.getString("driver");
String url = bundle.getString("url");
String user = bundle.getString("user");
String password = bundle.getString("password");
Connection conn = null;
Statement stmt = null;
ResultSet rs = null;
try{
Class.forName(driver);
conn = DriverManager.getConnection(url, user, password);
stmt = conn.createStatement();
String sql = "select * from user where loginName='" +loginName +"' and loginPwd='" +loginPwd+ "'";
rs = stmt.executeQuery(sql);
while (rs.next()){
loginSuccess = true;
}
}catch (SQLException | ClassNotFoundException e){
e.printStackTrace();
}finally {
if (rs != null) {
try {
rs.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if (stmt != null) {
try {
stmt.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if (conn != null) {
try {
conn.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
return loginSuccess;
}
/**
* 初始化用户界面
* @return 用户输入的用户名和密码等登录信息
*/
private static Map<String, String> initUI() {
Scanner in = new Scanner(System.in);
System.out.println("用户名：");
String loginName = in.nextLine();
System.out.println("密码：");
String loginPwd = in.nextLine();
Map<String, String> userLoginInfo = new HashMap<>();
userLoginInfo.put("loginName", loginName);
userLoginInfo.put("loginPwd", loginPwd);
return userLoginInfo;
}
}
SQL注入现象
示例
用户名输入：aaa
密码输入：aaa' or '1'='1
还是会显示登录成功，这就产生了SQL注入现象。所以当前程序存在安全隐患。
原因
1、Debug：
select * from user where
loginName='aaa' and
loginPwd='aaa' or '1'='1'
可以看出SQL语句完全被修改了。
2、在数据库查询：
3、分析
String sql = "select * from user where loginName='" +loginName +"' and loginPwd='" +loginPwd+ "'";
rs = stmt.executeQuery(sql);
以上代码先完成了SQL语句的拼接，再发送SQL语句给DBMS进行SQL编译。（先拼接，再编译）
在拼接的时候将用户提供的“非法信息”编译进去，导致了原SQL语句的含义被扭曲了。
4、根本原因
用户输入的信息中含有SQL语句的关键字，并且这些关键字参与SQL语句的编译过程，导致SQL语句的原意被扭曲，进而达到SQL注入。
解决SQL注入问题
1、只要用户提供的信息不参与SQL语句的编译过程，问题就解决了。即使用户提供的信息中含有SQL语句的关键字，但是没有参与编译，不起作用。
2、要想用户信息不参与SQL语句的编译，那么必须使用java.sql.PreparedStatement，该接口继承了java.sql.Statement
3、PreparedStatement是属于预编译的数据库操作对象。原理是：预先对SQL语句的框架进行编译，然后再给SQL语句传"值"。
4、帮助文档
public interface PreparedStatement extends Statement
表示预编译的SQL语句的对象。
SQL语句已预编译并存储在PreparedStatement对象中。 然后可以使用该对象多次有效地执行此语句。
注意：setter方法（ setShort ， setString用于设置IN参数值必须指定与所定义的SQL类型的输入参数的兼容的类型，等等）。 例如，如果IN参数具有SQL类型INTEGER ，则应使用方法setInt 。
如果需要任意参数类型转换，方法setObject应与目标SQL类型一起使用。
在设定的参数的以下示例中， con表示一个活动连接：
PreparedStatement pstmt = con.prepareStatement("UPDATE EMPLOYEES
SET SALARY = ? WHERE ID = ?");
pstmt.setBigDecimal(1, 153833.00)
pstmt.setInt(2, 110592)
解决SQL注入代码示例
package jdbc;
import java.sql.*;
import java.util.HashMap;
import java.util.Map;
import java.util.ResourceBundle;
import java.util.Scanner;
/**
* 解决SQL注入问题
*/
public class simpLoginPlus {
public static void main(String[] args) {
//初始化界面
Map<String,String> userLoginInfo = initUI();
//验证用户名和密码
boolean loginSuccess = login(userLoginInfo);
System.out.println(loginSuccess ? "登录成功" : "登录失败");
}
/**
* @param userLoginInfo 用户登录信息
* @return false表示失败，true表示成功
*/
private static boolean login(Map<String, String> userLoginInfo) {
boolean loginSuccess = false;
String loginName = userLoginInfo.get("loginName");
String loginPwd = userLoginInfo.get("loginPwd");
//JDBC代码
ResourceBundle bundle = ResourceBundle.getBundle("jdbc.info");
String driver = bundle.getString("driver");
String url = bundle.getString("url");
String user = bundle.getString("user");
String password = bundle.getString("password");
Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;
try{
Class.forName(driver);
conn = DriverManager.getConnection(url, user, password);
/*这相当于SQL语句的架子。
其中一个问号(?)表示一个占位符，
一个?将来接收一个"值"。
注意：占位符不能使用单引号括起来。*/
String sql = "select * from user where loginName= ? and loginPwd= ?";
/*程序执行到此处，会发送SQL语句架子给DBMS，
然后DBMS进行SQL语句的预先编译。*/
pstmt = conn.prepareStatement(sql);
/*给占位符?传值，
* 第一个?下标是1，
* 第二个?下标是2*/
pstmt.setString(1,loginName);
pstmt.setString(2,loginPwd);
//执行SQL
rs = pstmt.executeQuery();
while (rs.next()){
loginSuccess = true;
}
}catch (SQLException | ClassNotFoundException e){
e.printStackTrace();
}finally {
if (rs != null) {
try {
rs.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if (pstmt != null) {
try {
pstmt.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if (conn != null) {
try {
conn.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
return loginSuccess;
}
/**
* 初始化用户界面
* @return 用户输入的用户名和密码等登录信息
*/
private static Map<String, String> initUI() {
Scanner in = new Scanner(System.in);
System.out.println("用户名：");
String loginName = in.nextLine();
System.out.println("密码：");
String loginPwd = in.nextLine();
Map<String, String> userLoginInfo = new HashMap<>();
userLoginInfo.put("loginName", loginName);
userLoginInfo.put("loginPwd", loginPwd);
return userLoginInfo;
}
}
再输入asd' or '1'='1已经不起作用了。
解决SQL注入的关键是：用户提供的信息中即使含有SQL语句的关键字，但是这些关键字并没有参与编译，不起作用。
