diff options
Diffstat (limited to 'src/main/java/com/gitblit/auth/AuthenticationProvider.java')
-rw-r--r-- | src/main/java/com/gitblit/auth/AuthenticationProvider.java | 82 |
1 files changed, 22 insertions, 60 deletions
diff --git a/src/main/java/com/gitblit/auth/AuthenticationProvider.java b/src/main/java/com/gitblit/auth/AuthenticationProvider.java index 6205fd75..f6511bad 100644 --- a/src/main/java/com/gitblit/auth/AuthenticationProvider.java +++ b/src/main/java/com/gitblit/auth/AuthenticationProvider.java @@ -15,13 +15,8 @@ */ package com.gitblit.auth; -import java.io.ByteArrayOutputStream; import java.io.File; -import java.io.IOException; -import java.io.ObjectOutputStream; import java.math.BigInteger; -import java.security.MessageDigest; -import java.security.NoSuchAlgorithmException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -33,6 +28,7 @@ import com.gitblit.manager.IUserManager; import com.gitblit.models.TeamModel; import com.gitblit.models.UserModel; import com.gitblit.utils.ArrayUtils; +import com.gitblit.utils.DeepCopier; import com.gitblit.utils.StringUtils; public abstract class AuthenticationProvider { @@ -83,69 +79,35 @@ public abstract class AuthenticationProvider { } } - /** - * Utility method to calculate the checksum of an object. - * @param sourceObject The object from which to establish the checksum. - * @return The checksum - * @throws IOException - * @throws NoSuchAlgorithmException - */ - private BigInteger checksum(Object sourceObject) throws IOException, NoSuchAlgorithmException { - - if (sourceObject == null) { - return BigInteger.ZERO; - } + protected void updateUser(UserModel userModel) { + final UserModel userLocalDB = userManager.getUserModel(userModel.getName()); - ByteArrayOutputStream baos = new ByteArrayOutputStream(); - ObjectOutputStream oos = new ObjectOutputStream(baos); - oos.writeObject(sourceObject); - oos.close(); + // Establish the checksum of the current version of the user + final BigInteger userCurrentCheck = DeepCopier.checksum(userModel); - MessageDigest m = MessageDigest.getInstance("SHA1"); - m.update(baos.toByteArray()); + // Establish the checksum of the stored version of the user + final BigInteger userLocalDBcheck = DeepCopier.checksum(userLocalDB); - return new BigInteger(1, m.digest()); - } - - protected void updateUser(UserModel userModel) { - final UserModel userLocalDB = userManager.getUserModel(userModel.getName()); - try { - // Establish the checksum of the current version of the user - final BigInteger userCurrentCheck = checksum(userModel); - // Establish the checksum of the stored version of the user - final BigInteger userLocalDBcheck = checksum(userLocalDB); - // Compare the checksums - if (!userCurrentCheck.equals(userLocalDBcheck)) - { - // If mismatch, save the new instance. - userManager.updateUserModel(userModel); - } - } catch (NoSuchAlgorithmException | IOException e) { - // Trace any potential error. - if (logger.isErrorEnabled()) { - logger.error(e.getMessage()); - } + // Compare the checksums + if (!userCurrentCheck.equals(userLocalDBcheck)) { + // If mismatch, save the new instance. + userManager.updateUserModel(userModel); } } protected void updateTeam(TeamModel teamModel) { final TeamModel teamLocalDB = userManager.getTeamModel(teamModel.name); - try { - // Establish the checksum of the current version of the team - final BigInteger teamCurrentCheck = checksum(teamModel); - // Establish the checksum of the stored version of the team - final BigInteger teamLocalDBcheck = checksum(teamLocalDB); - // Compare the checksums - if (!teamCurrentCheck.equals(teamLocalDBcheck)) - { - // If mismatch, save the new instance. - userManager.updateTeamModel(teamModel); - } - } catch (NoSuchAlgorithmException | IOException e) { - // Trace any potential error. - if (logger.isErrorEnabled()) { - logger.error(e.getMessage()); - } + + // Establish the checksum of the current version of the team + final BigInteger teamCurrentCheck = DeepCopier.checksum(teamModel); + + // Establish the checksum of the stored version of the team + final BigInteger teamLocalDBcheck = DeepCopier.checksum(teamLocalDB); + + // Compare the checksums + if (!teamCurrentCheck.equals(teamLocalDBcheck)) { + // If mismatch, save the new instance. + userManager.updateTeamModel(teamModel); } } |