1 /*
   2  * Copyright (c) 2011, 2012, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package sun.security.krb5;
  27 
  28 import java.io.IOException;
  29 import java.util.Collection;
  30 import java.util.Hashtable;
  31 import java.util.Vector;
  32 
  33 
  34 public class SCDynamicStoreConfig {
  35     private static native void installNotificationCallback();
  36     private static native Hashtable<String, Object> getKerberosConfig();
  37     private static boolean DEBUG = sun.security.krb5.internal.Krb5.DEBUG;
  38 
  39     static {
  40         boolean isMac = java.security.AccessController.doPrivileged(
  41             new java.security.PrivilegedAction<Boolean>() {
  42                 public Boolean run() {
  43                     String osname = System.getProperty("os.name");
  44                     if (osname.contains("OS X")) {
  45                         System.loadLibrary("osxkrb5");
  46                         return true;
  47                     }
  48                     return false;
  49                 }
  50             });
  51         if (isMac) installNotificationCallback();
  52     }
  53 
  54     private static Vector<String> unwrapHost(
  55             Collection<Hashtable<String, String>> c) {
  56         Vector<String> vector = new Vector<String>();
  57         for (Hashtable<String, String> m : c) {
  58             vector.add(m.get("host"));
  59         }
  60         return vector;
  61     }
  62 
  63     /**
  64      * convertRealmConfigs: Maps the Object graph that we get from JNI to the
  65      * object graph that Config expects. Also the items inside the kdc array
  66      * are wrapped inside Hashtables
  67      */
  68     @SuppressWarnings("unchecked")
  69     private static Hashtable<String, Object>
  70             convertRealmConfigs(Hashtable<String, ?> configs) {
  71         Hashtable<String, Object> realmsTable = new Hashtable<String, Object>();
  72 
  73         for (String realm : configs.keySet()) {
  74             // get the kdc
  75             Hashtable<String, Collection<?>> map =
  76                     (Hashtable<String, Collection<?>>) configs.get(realm);
  77             Hashtable<String, Vector<String>> realmMap =
  78                     new Hashtable<String, Vector<String>>();
  79 
  80             // put the kdc into the realmMap
  81             Collection<Hashtable<String, String>> kdc =
  82                     (Collection<Hashtable<String, String>>) map.get("kdc");
  83             if (kdc != null) realmMap.put("kdc", unwrapHost(kdc));
  84 
  85             // put the admin server into the realmMap
  86             Collection<Hashtable<String, String>> kadmin =
  87                     (Collection<Hashtable<String, String>>) map.get("kadmin");
  88             if (kadmin != null) realmMap.put("admin_server", unwrapHost(kadmin));
  89 
  90             // add the full entry to the realmTable
  91             realmsTable.put(realm, realmMap);
  92         }
  93 
  94         return realmsTable;
  95     }
  96 
  97     /**
  98      * Calls down to JNI to get the raw Kerberos Config and maps the object
  99      * graph to the one that Kerberos Config in Java expects
 100      *
 101      * @return
 102      * @throws IOException
 103      */
 104     public static Hashtable<String, Object> getConfig() throws IOException {
 105         Hashtable<String, Object> stanzaTable = getKerberosConfig();
 106         if (stanzaTable == null) {
 107             throw new IOException(
 108                     "Could not load configuration from SCDynamicStore");
 109         }
 110         if (DEBUG) System.out.println("Raw map from JNI: " + stanzaTable);
 111         return convertNativeConfig(stanzaTable);
 112     }
 113 
 114     @SuppressWarnings("unchecked")
 115     private static Hashtable<String, Object> convertNativeConfig(
 116             Hashtable<String, Object> stanzaTable) {
 117         // convert SCDynamicStore realm structure to Java realm structure
 118         Hashtable<String, ?> realms =
 119                 (Hashtable<String, ?>) stanzaTable.get("realms");
 120         if (realms != null) {
 121             stanzaTable.remove("realms");
 122             Hashtable<String, Object> realmsTable = convertRealmConfigs(realms);
 123             stanzaTable.put("realms", realmsTable);
 124         }
 125         WrapAllStringInVector(stanzaTable);
 126         if (DEBUG) System.out.println("stanzaTable : " + stanzaTable);
 127         return stanzaTable;
 128     }
 129 
 130     @SuppressWarnings("unchecked")
 131     private static void WrapAllStringInVector(
 132             Hashtable<String, Object> stanzaTable) {
 133         for (String s: stanzaTable.keySet()) {
 134             Object v = stanzaTable.get(s);
 135             if (v instanceof Hashtable) {
 136                 WrapAllStringInVector((Hashtable<String,Object>)v);
 137             } else if (v instanceof String) {
 138                 Vector<String> vec = new Vector<>();
 139                 vec.add((String)v);
 140                 stanzaTable.put(s, vec);
 141             }
 142         }
 143     }
 144 }