shiro/src/main/java/org/onap/aaf/cadi/shiro/AAFAuthenticationInfo.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21 package org.onap.aaf.cadi.shiro;
  22
  23 import java.nio.ByteBuffer;
  24 import java.security.NoSuchAlgorithmException;
  25 import java.security.SecureRandom;
  26
  27 import org.apache.shiro.authc.AuthenticationInfo;
  28 import org.apache.shiro.authc.AuthenticationToken;
  29 import org.apache.shiro.authc.UsernamePasswordToken;
  30 import org.apache.shiro.subject.PrincipalCollection;
  31 import org.onap.aaf.cadi.Access;
  32 import org.onap.aaf.cadi.Access.Level;
  33 import org.onap.aaf.cadi.Hash;
  34
  35 public class AAFAuthenticationInfo implements AuthenticationInfo {
  36         private static final long serialVersionUID = -1502704556864321020L;
  37
  38         // We assume that Shiro is doing Memory Only, and this salt is not needed cross process
  39         private final static int salt = new SecureRandom().nextInt();
  40
  41         private final AAFPrincipalCollection apc;
  42         private final byte[] hash;
  43         private Access access;
  44
  45         public AAFAuthenticationInfo(Access access, String username, String password) {
  46                 this.access = access;
  47                 apc = new AAFPrincipalCollection(username);
  48                 hash = getSaltedCred(password);
  49         }
  50
  51         @Override
  52         public byte[] getCredentials() {
  53                 access.log(Level.DEBUG, "AAFAuthenticationInfo.getCredentials");
  54                 return hash;
  55         }
  56
  57         @Override
  58         public PrincipalCollection getPrincipals() {
  59                 access.log(Level.DEBUG, "AAFAuthenticationInfo.getPrincipals");
  60                 return apc;
  61         }
  62
  63         public boolean matches(AuthenticationToken atoken) {
  64                 if(atoken instanceof UsernamePasswordToken) {
  65                         UsernamePasswordToken upt = (UsernamePasswordToken)atoken;
  66                         if(apc.getPrimaryPrincipal().getName().equals(upt.getPrincipal())) {
  67                                 byte[] newhash = getSaltedCred(new String(upt.getPassword()));
  68                                 access.printf(Level.INFO,"Successful authentication attempt by %s",upt.getPrincipal());
  69                                 if(newhash.length==hash.length) {
  70                                         for(int i=0;i<hash.length;++i) {
  71                                                 if(hash[i]!=newhash[i]) {
  72                                                         return false;
  73                                                 }
  74                                         }
  75                                         return true;
  76                                 }
  77                         }
  78                 } else {
  79                         access.printf(Level.ERROR, "AAFAuthenticationInfo received non-AAF token %s (%s)",atoken.getPrincipal(),atoken.getClass().getName());
  80                 }
  81                 return false;
  82         }
  83
  84         private byte[] getSaltedCred(String password) {
  85                 byte[] pbytes = password.getBytes();
  86                 ByteBuffer bb = ByteBuffer.allocate(pbytes.length+Integer.SIZE/8);
  87                 bb.asIntBuffer().put(salt);
  88                 bb.put(password.getBytes());
  89                 try {
  90                         return Hash.hashSHA256(bb.array());
  91                 } catch (NoSuchAlgorithmException e) {
  92                         return new byte[0]; // should never get here
  93                 }
  94         }
  95 }