JDBC模拟用户登录并防止SQL注入

import java.sql.*;
import java.util.HashMap;
import java.util.Map;
import java.util.ResourceBundle;
import java.util.Scanner;

/**
 * 解决sql注入问题
 *      只要用户的信息不参与编译即可
 *      使用PreparedStatement预编译
 */
public class JDBCTest07 {
    public static void main(String[] args) {
        //初始化用户界面
        Map<String,String> loginPage = initLoginPage();

        //登录
        boolean loginStatus = login(loginPage);

        System.out.println(loginStatus == true ? "sign in successfully":"sign in failed");

    }

    /**
     * 登录
     * @param loginPage
     * @return
     */
    private static boolean login(Map<String,String> loginPage) {

        boolean isSuccess = false;

        String name = loginPage.get("username");
        String pw = loginPage.get("password");

        Connection conn = null;
        //第一个改动
        PreparedStatement ps = null;
        ResultSet rs = null;

        ResourceBundle bundle = ResourceBundle.getBundle("JDBC");

        String driver = bundle.getString("driver");
        String url = bundle.getString("url");
        String user = bundle.getString("user");
        String pword = bundle.getString("password");

        try {

            Class.forName(driver);

            conn = DriverManager.getConnection(url,user,pword);

            //第二个改动
            String sql = "select loginname,password from t_user where loginname = ? and password = ?";//?为占位符
            //第三个改动 在此处将sql语句的框架传给DBMS进行sql语句的预编译
            ps = conn.prepareStatement(sql);

            //传值第一个?下标为1 类推
            ps.setString(1,name);
            ps.setString(2,pw);
            //执行
            rs = ps.executeQuery();

            if (rs.next()){
                isSuccess = true;
            }

        } catch (ClassNotFoundException | SQLException e) {
            e.printStackTrace();
        }finally {
            if (rs != null) {
                try {
                    rs.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }

            if (ps != null) {
                try {
                    ps.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }

            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }

        return isSuccess;
    }

    /**
     * 初始化用户界面
     * @return 用户的登录账号和密码信息
     */
    private static Map<String,String> initLoginPage() {

        Scanner s = new Scanner(System.in);

        System.out.println("username:");
        String username = s.nextLine();

        System.out.println("password:");
        String password = s.nextLine();

        Map<String,String> loginPage = new HashMap<>();
        loginPage.put("username",username);
        loginPage.put("password",password);

        return loginPage;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值