1 files changed, 226 insertions, 0 deletions
diff --git a/base/util/src/netscape/security/x509/Extensions.java b/base/util/src/netscape/security/x509/Extensions.java
new file mode 100644
index 000000000..622367ab6
--- /dev/null
+++ b/base/util/src/netscape/security/x509/Extensions.java
@@ -0,0 +1,226 @@
+// --- BEGIN COPYRIGHT BLOCK ---
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; version 2 of the License.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License along
+// with this program; if not, write to the Free Software Foundation, Inc.,
+// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+//
+// (C) 2007 Red Hat, Inc.
+// All rights reserved.
+// --- END COPYRIGHT BLOCK ---
+package netscape.security.x509;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.lang.reflect.Array;
+import java.lang.reflect.Constructor;
+import java.lang.reflect.InvocationTargetException;
+import java.security.cert.CertificateException;
+import java.util.Enumeration;
+import java.util.Hashtable;
+import java.util.Vector;
+
+import netscape.security.util.DerInputStream;
+import netscape.security.util.DerOutputStream;
+import netscape.security.util.DerValue;
+
+/**
+ * This class defines the Extensions attribute for the Certificate.
+ * 
+ * @author Amit Kapoor
+ * @author Hemma Prafullchandra
+ * @version 1.11
+ * @see CertAttrSet
+ */
+public class Extensions extends Vector<Extension>
+        implements CertAttrSet {
+    /**
+     *
+     */
+    private static final long serialVersionUID = 4597917347772057433L;
+    /**
+     * Identifier for this attribute, to be used with the
+     * get, set, delete methods of Certificate, x509 type.
+     */
+    public static final String IDENT = "x509.info.extensions";
+    /**
+     * name
+     */
+    public static final String NAME = "extensions";
+
+    private Hashtable<String, Extension> map;
+
+    // Parse the encoded extension
+    public void parseExtension(Extension ext) throws IOException {
+        try {
+            @SuppressWarnings("unchecked")
+            Class<CertAttrSet> extClass = (Class<CertAttrSet>) OIDMap.getClass(ext.getExtensionId());
+            if (extClass == null) { // Unsupported extension
+                if (ext.isCritical()) {
+                    throw new IOException("Unsupported CRITICAL extension: "
+                                          + ext.getExtensionId());
+                } else {
+                    map.put(ext.getExtensionId().toString(), ext);
+                    addElement(ext);
+                    return;
+                }
+            }
+            Class<?>[] params = { Boolean.class, Object.class };
+            Constructor<CertAttrSet> cons = extClass.getConstructor(params);
+
+            byte[] extData = ext.getExtensionValue();
+            int extLen = extData.length;
+            Object value = Array.newInstance(byte.class, extLen);
+
+            for (int i = 0; i < extLen; i++) {
+                Array.setByte(value, i, extData[i]);
+            }
+            Object[] passed = new Object[] { new Boolean(ext.isCritical()),
+                                                        value };
+            CertAttrSet certExt = cons.newInstance(passed);
+            map.put(certExt.getName(), (Extension) certExt);
+            addElement((Extension) certExt);
+
+        } catch (NoSuchMethodException nosuch) {
+            throw new IOException(nosuch.toString());
+        } catch (InvocationTargetException invk) {
+            throw new IOException(invk.getTargetException().toString());
+        } catch (Exception e) {
+            throw new IOException(e.toString());
+        }
+    }
+
+    /**
+     * Default constructor for the certificate attribute.
+     */
+    public Extensions() {
+        map = new Hashtable<String, Extension>();
+    }
+
+    /**
+     * Create the object, decoding the values from the passed DER stream.
+     * 
+     * @param in the DerInputStream to read the Extension from.
+     * @exception IOException on decoding errors.
+     */
+    public Extensions(DerInputStream in)
+            throws IOException {
+
+        map = new Hashtable<String, Extension>();
+        DerValue[] exts = in.getSequence(5);
+
+        for (int i = 0; i < exts.length; i++) {
+            Extension ext = new Extension(exts[i]);
+            parseExtension(ext);
+        }
+    }
+
+    /**
+     * Decode the extensions from the InputStream.
+     * 
+     * @param in the InputStream to unmarshal the contents from.
+     * @exception IOException on decoding or validity errors.
+     */
+    public void decode(InputStream in) throws IOException {
+        DerValue val = new DerValue(in);
+        DerInputStream str = val.toDerInputStream();
+
+        map = new Hashtable<String, Extension>();
+        DerValue[] exts = str.getSequence(5);
+
+        for (int i = 0; i < exts.length; i++) {
+            Extension ext = new Extension(exts[i]);
+            parseExtension(ext);
+        }
+    }
+
+    /**
+     * Encode the extensions in DER form to the stream.
+     * 
+     * @param out the DerOutputStream to marshal the contents to.
+     * @exception CertificateException on encoding errors.
+     * @exception IOException on errors.
+     */
+    public void encode(OutputStream out)
+            throws CertificateException, IOException {
+        DerOutputStream extOut = new DerOutputStream();
+        for (int i = 0; i < size(); i++) {
+            Object thisOne = elementAt(i);
+            if (thisOne instanceof CertAttrSet)
+                ((CertAttrSet) thisOne).encode(extOut);
+            else if (thisOne instanceof Extension)
+                ((Extension) thisOne).encode(extOut);
+            else
+                throw new CertificateException("Invalid extension object");
+        }
+
+        DerOutputStream seq = new DerOutputStream();
+        seq.write(DerValue.tag_Sequence, extOut);
+
+        out.write(seq.toByteArray());
+    }
+
+    /**
+     * Set the attribute value.
+     * 
+     * @param name the extension name used in the cache.
+     * @param obj the object to set.
+     * @exception IOException if the object could not be cached.
+     */
+    public void set(String name, Object obj) throws IOException {
+        map.put(name, (Extension) obj);
+        addElement((Extension) obj);
+    }
+
+    /**
+     * Get the attribute value.
+     * 
+     * @param name the extension name used in the lookup.
+     * @exception IOException if named extension is not found.
+     */
+    public Object get(String name) throws IOException {
+        Object obj = map.get(name);
+        if (obj == null) {
+            throw new IOException("No extension found with name " + name);
+        }
+        return (obj);
+    }
+
+    /**
+     * Delete the attribute value.
+     * 
+     * @param name the extension name used in the lookup.
+     * @exception IOException if named extension is not found.
+     */
+    public void delete(String name) throws IOException {
+        Object obj = map.get(name);
+        if (obj == null) {
+            throw new IOException("No extension found with name " + name);
+        }
+        map.remove(name);
+        removeElement(obj);
+    }
+
+    /**
+     * Return an enumeration of names of attributes existing within this
+     * attribute.
+     */
+    public Enumeration<String> getAttributeNames() {
+        return map.keys();
+    }
+
+    /**
+     * Return the name of this attribute.
+     */
+    public String getName() {
+        return (NAME);
+    }
+}