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;
    }
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,下面是一个完整的JavaWeb模拟登录案例,包括设计数据库、编写servlet类、service层、DAO层以及JDBC代码。 1. 设计数据库 首先,我们需要设计一个user表,用来存储用户的登录信息。该表包含以下字段: | 字段名 | 类型 | 描述 | | ------ | -------- | ---------- | | id | int | 用户ID | | username | varchar(50) | 用户名 | | password | varchar(50) | 密码 | 可以使用以下SQL语句创建该表: ```sql CREATE TABLE `user` ( `id` int NOT NULL AUTO_INCREMENT, `username` varchar(50) NOT NULL, `password` varchar(50) NOT NULL, PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4; ``` 2. 编写DAO层 在DAO层,我们需要编写一个UserDAO接口和一个UserDAOImpl类,用于实现对用户信息的增删改查操作。 UserDAO.java: ```java public interface UserDAO { User getUserByUsernameAndPassword(String username, String password); } ``` UserDAOImpl.java: ```java public class UserDAOImpl implements UserDAO { private Connection conn; public UserDAOImpl(Connection conn) { this.conn = conn; } @Override public User getUserByUsernameAndPassword(String username, String password) { User user = null; try { Statement stmt = conn.createStatement(); String sql = "SELECT * FROM user WHERE username='" + username + "' AND password='" + password + "'"; ResultSet rs = stmt.executeQuery(sql); if (rs.next()) { user = new User(); user.setId(rs.getInt("id")); user.setUsername(rs.getString("username")); user.setPassword(rs.getString("password")); } } catch (SQLException e) { e.printStackTrace(); } return user; } } ``` 在上面的代码中,我们使用Statement语句实现了查询用户信息的功能,并将查询结果封装成了一个User对象。 需要注意的是,上面的代码中存在SQL注入的安全隐患,因为我们直接将用户输入的用户名密码拼接到SQL语句中。在真实的项目中,我们应该使用PreparedStatement语句,通过参数化查询的方式防止SQL注入。 3. 编写Service层 在Service层,我们需要编写一个UserService接口和一个UserServiceImpl类,用于调用DAO层的方法,并进行业务逻辑的处理。 UserService.java: ```java public interface UserService { User login(String username, String password); } ``` UserServiceImpl.java: ```java public class UserServiceImpl implements UserService { private UserDAO userDao; public UserServiceImpl(UserDAO userDao) { this.userDao = userDao; } @Override public User login(String username, String password) { User user = userDao.getUserByUsernameAndPassword(username, password); if (user != null) { // 登录成功,将用户信息存储到Session中 // ... } return user; } } ``` 在上面的代码中,我们调用了UserDAO的方法来查询用户信息,并进行了登录验证。如果登录成功,我们可以将用户信息存储到Session中,以便在后续的请求中进行身份验证。 4. 编写Servlet类 在Servlet类中,我们需要获取用户输入的用户名密码,调用UserService的方法进行登录验证,然后将结果返回给前端页面。 LoginController.java: ```java @WebServlet("/login") public class LoginController extends HttpServlet { private UserService userService; @Override public void init() throws ServletException { // 在初始化方法中注入UserService的实现类 userDao = new UserDAOImpl(JDBCUtils.getConnection()); userService = new UserServiceImpl(userDao); } @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { String username = request.getParameter("username"); String password = request.getParameter("password"); User user = userService.login(username, password); if (user != null) { // 登录成功,重定向到主页 response.sendRedirect("/index.jsp"); } else { // 登录失败,返回错误信息 request.setAttribute("errorMsg", "用户名密码错误"); request.getRequestDispatcher("/login.jsp").forward(request, response); } } } ``` 在上面的代码中,我们在init()方法中注入了UserService的实现类,并在doPost()方法中获取用户输入的用户名密码,调用UserService的方法进行登录验证,然后根据验证结果进行重定向或错误信息提示。 5. 编写JDBC代码 在JDBC代码中,我们需要获取数据库连接,并实现连接的关闭功能。 JDBCUtils.java: ```java public class JDBCUtils { private static String url = "jdbc:mysql://localhost:3306/test"; private static String user = "root"; private static String password = "123456"; public static Connection getConnection() { Connection conn = null; try { Class.forName("com.mysql.jdbc.Driver"); conn = DriverManager.getConnection(url, user, password); } catch (ClassNotFoundException | SQLException e) { e.printStackTrace(); } return conn; } public static void closeConnection(Connection conn) { if (conn != null) { try { conn.close(); } catch (SQLException e) { e.printStackTrace(); } } } } ``` 在上面的代码中,我们使用了JDBC的标准API来获取数据库连接,并实现了连接的关闭功能。 最后,我们需要在Web.xml中配置Servlet和JDBC的相关信息。 web.xml: ```xml <web-app> <servlet> <servlet-name>LoginController</servlet-name> <servlet-class>com.example.LoginController</servlet-class> </servlet> <servlet-mapping> <servlet-name>LoginController</servlet-name> <url-pattern>/login</url-pattern> </servlet-mapping> </web-app> ``` 以上就是一个完整的JavaWeb模拟登录案例,包括设计数据库、编写DAO层、Service层、Servlet类以及JDBC代码。需要注意的是,在实际开发中,我们应该对用户输入的用户名密码进行安全性验证,以防止SQL注入等攻击。此外,我们还应该使用HTTPS协议来保护用户的登录信息。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值