001    /*
002     * Cumulus4j - Securing your data in the cloud - http://cumulus4j.org
003     * Copyright (C) 2011 NightLabs Consulting GmbH
004     *
005     * This program is free software: you can redistribute it and/or modify
006     * it under the terms of the GNU Affero General Public License as
007     * published by the Free Software Foundation, either version 3 of the
008     * License, or (at your option) any later version.
009     *
010     * This program is distributed in the hope that it will be useful,
011     * but WITHOUT ANY WARRANTY; without even the implied warranty of
012     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
013     * GNU Affero General Public License for more details.
014     *
015     * You should have received a copy of the GNU Affero General Public License
016     * along with this program.  If not, see <http://www.gnu.org/licenses/>.
017     */
018    package org.cumulus4j.store;
019    
020    import java.io.IOException;
021    import java.io.InputStream;
022    import java.io.ObjectInputStream;
023    import java.io.ObjectStreamClass;
024    
025    import org.datanucleus.ClassLoaderResolver;
026    
027    /**
028     * A subclass of {@link ObjectInputStream} using the {@link ClassLoaderResolver} to load classes.
029     *
030     * @author Marco หงุ่ยตระกูล-Schulze - marco at nightlabs dot de
031     */
032    public class DataNucleusObjectInputStream extends java.io.ObjectInputStream
033    {
034            private ClassLoaderResolver classLoaderResolver;
035    
036            public DataNucleusObjectInputStream(InputStream in, ClassLoaderResolver classLoaderResolver) throws IOException {
037                    super(in);
038    
039                    if (in == null) // in case the super(in) didn't throw an exception, yet, we do it now (fast failure)
040                            throw new IllegalArgumentException("in == null");
041    
042                    if (classLoaderResolver == null)
043                            throw new IllegalArgumentException("classLoaderResolver == null");
044    
045                    this.classLoaderResolver = classLoaderResolver;
046            }
047    
048            @Override
049            protected Class<?> resolveClass(ObjectStreamClass desc)
050            throws IOException, ClassNotFoundException
051            {
052                    return classLoaderResolver.classForName(desc.getName());
053            }
054    }