DefaultLockOutProvider.java 6.36 KB
Newer Older
1 2 3 4
/**
 * $Revision$
 * $Date$
 *
5
 * Copyright (C) 2005-2008 Jive Software. All rights reserved.
6
 *
7 8 9 10 11 12 13 14 15 16 17
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
18 19 20
 */
package org.jivesoftware.openfire.lockout;

21 22 23 24 25 26 27
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.Date;

28 29
import org.jivesoftware.database.DbConnectionManager;
import org.jivesoftware.util.StringUtils;
30 31
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
32 33

/**
34
 * The DefaultLockOutProvider works with the ofUserFlag table to maintain a list of disabled/locked out
35 36
 * accounts., and as the name implies, is the default LockOutProvider implementation.
 *
37 38 39 40
 * @author Daniel Henninger
 */
public class DefaultLockOutProvider implements LockOutProvider {

41 42
	private static final Logger Log = LoggerFactory.getLogger(DefaultLockOutProvider.class);

43 44
    private static final String FLAG_ID = "lockout";
    private static final String DELETE_FLAG =
45
            "DELETE FROM ofUserFlag WHERE username=? AND name='"+FLAG_ID+"'";
46
    private static final String ADD_FLAG =
47
            "INSERT INTO ofUserFlag VALUES(?,'"+FLAG_ID+"',?,?)";
48
    private static final String RETRIEVE_FLAG =
49
            "SELECT name,startTime,endTime FROM ofUserFlag WHERE username=? AND name='"+FLAG_ID+"'";
50 51 52 53 54 55 56 57 58

    /**
     * Constructs a new DefaultLockOutProvider
     */
    public DefaultLockOutProvider() {

    }

    /**
59
     * Default provider retrieves disabled status from ofUserFlag table.
60 61
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#getDisabledStatus(String)
     */
62
    @Override
63
    public LockOutFlag getDisabledStatus(String username) {
64 65 66 67 68 69 70 71 72 73
        Connection con = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        LockOutFlag ret = null;
        try {
            con = DbConnectionManager.getConnection();
            pstmt = con.prepareStatement(RETRIEVE_FLAG);
            pstmt.setString(1, username);
            rs = pstmt.executeQuery();
            if (!rs.next()) {
74
                return null;
75 76 77 78 79 80 81 82 83 84 85 86 87
            }
            Date startTime = null;
            if (rs.getString(2) != null) {
                startTime = new Date(Long.parseLong(rs.getString(2).trim()));
            }
            Date endTime = null;
            if (rs.getString(3) != null) {
                endTime = new Date(Long.parseLong(rs.getString(3).trim()));
            }

            ret = new LockOutFlag(username, startTime, endTime);
        }
        catch (Exception e) {
88 89
            Log.error("Error loading lockout information from DB", e);
            return null;
90 91 92 93 94 95 96 97 98
        }
        finally {
            DbConnectionManager.closeConnection(rs, pstmt, con);
        }

        return ret;
    }

    /**
99
     * Default provider deletes existing flag, if it exists, and adds new described flag in ofUserFlag table.
100 101
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#setDisabledStatus(LockOutFlag)
     */
102
    @Override
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
    public void setDisabledStatus(LockOutFlag flag) {
        Connection con = null;
        PreparedStatement pstmt = null;
        try {
            con = DbConnectionManager.getConnection();
            pstmt = con.prepareStatement(DELETE_FLAG);
            pstmt.setString(1, flag.getUsername());
            pstmt.executeUpdate();
        }
        catch (SQLException sqle) {
            // Nothing to do.
        }
        finally {
            DbConnectionManager.closeConnection(pstmt, con);
        }
        try {
            con = DbConnectionManager.getConnection();
            pstmt = con.prepareStatement(ADD_FLAG);
            pstmt.setString(1, flag.getUsername());
            if (flag.getStartTime() != null) {
                pstmt.setString(2, StringUtils.dateToMillis(flag.getStartTime()));
            }
            else {
                pstmt.setNull(2, Types.VARCHAR);
            }
128 129
            if (flag.getEndTime() != null) {
                pstmt.setString(3, StringUtils.dateToMillis(flag.getEndTime()));
130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
            }
            else {
                pstmt.setNull(3, Types.VARCHAR);
            }
            pstmt.executeUpdate();
        }
        catch (SQLException sqle) {
            // Nothing to do.
        }
        finally {
            DbConnectionManager.closeConnection(pstmt, con);
        }
    }

    /**
145
     * Default provider deletes existing flag from ofUserFlag table.
146 147
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#unsetDisabledStatus(String)
     */
148
    @Override
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
    public void unsetDisabledStatus(String username) {
        Connection con = null;
        PreparedStatement pstmt = null;
        try {
            con = DbConnectionManager.getConnection();
            pstmt = con.prepareStatement(DELETE_FLAG);
            pstmt.setString(1, username);
            pstmt.executeUpdate();
        }
        catch (SQLException sqle) {
            // Nothing to do.
        }
        finally {
            DbConnectionManager.closeConnection(pstmt, con);
        }
    }

    /**
     * Default provider allows editing of disabled status.
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#isReadOnly()
     */
170
    @Override
171 172 173 174
    public boolean isReadOnly() {
        return false;
    }

175 176 177 178
    /**
     * Default provider allows delayed start to disabled status.
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#isDelayedStartSupported()
     */
179
    @Override
180 181 182 183 184 185 186 187
    public boolean isDelayedStartSupported() {
        return true;
    }

    /**
     * Default provider allows timeout of disabled status.
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#isTimeoutSupported()
     */
188
    @Override
189 190 191 192
    public boolean isTimeoutSupported() {
        return true;
    }

193 194 195 196
    /**
     * Default provider should be cached.
     * @see org.jivesoftware.openfire.lockout.LockOutProvider#shouldNotBeCached()
     */
197
    @Override
198 199 200 201
    public boolean shouldNotBeCached() {
        return false;
    }

202
}