James Moger
2015-11-19 e41e8f8c3bc9f5edab1d271464364f95620ece8c
src/main/java/com/gitblit/auth/AuthenticationProvider.java
@@ -15,24 +15,21 @@
 */
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;
import com.gitblit.Constants.AccountType;
import com.gitblit.Constants.Role;
import com.gitblit.IStoredSettings;
import com.gitblit.manager.IRuntimeManager;
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 +80,35 @@
      }
   }
   /**
    * 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;
       }
       ByteArrayOutputStream baos = new ByteArrayOutputStream();
       ObjectOutputStream oos = new ObjectOutputStream(baos);
       oos.writeObject(sourceObject);
       oos.close();
       MessageDigest m = MessageDigest.getInstance("SHA1");
       m.update(baos.toByteArray());
       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());
         }
      // Establish the checksum of the current version of the user
      final BigInteger userCurrentCheck = DeepCopier.checksum(userModel);
      // Establish the checksum of the stored version of the user
      final BigInteger userLocalDBcheck = DeepCopier.checksum(userLocalDB);
      // 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);
      }
   }
@@ -188,6 +151,24 @@
    * @return true if the user service supports team membership changes
    */
   public abstract boolean supportsTeamMembershipChanges();
   /**
    * Returns true if the user's role can be changed.
    *
    * @param user
    * @param role
    * @return true if the user's role can be changed
    */
   public abstract boolean supportsRoleChanges(UserModel user, Role role);
   /**
    * Returns true if the team's role can be changed.
    *
    * @param user
    * @param role
    * @return true if the team's role can be changed
    */
   public abstract boolean supportsRoleChanges(TeamModel team, Role role);
    @Override
    public String toString() {
@@ -250,5 +231,16 @@
      public boolean supportsTeamMembershipChanges() {
         return true;
      }
      @Override
      public boolean supportsRoleChanges(UserModel user, Role role) {
         return true;
      }
      @Override
      public boolean supportsRoleChanges(TeamModel team, Role role) {
         return true;
      }
    }
}