fce5b180f7588e65b553a470822bb023a4a68270
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / aa / arp / provider / FileSystemArpRepository.java
1 /* 
2  * The Shibboleth License, Version 1. 
3  * Copyright (c) 2002 
4  * University Corporation for Advanced Internet Development, Inc. 
5  * All rights reserved
6  * 
7  * 
8  * Redistribution and use in source and binary forms, with or without 
9  * modification, are permitted provided that the following conditions are met:
10  * 
11  * Redistributions of source code must retain the above copyright notice, this 
12  * list of conditions and the following disclaimer.
13  * 
14  * Redistributions in binary form must reproduce the above copyright notice, 
15  * this list of conditions and the following disclaimer in the documentation 
16  * and/or other materials provided with the distribution, if any, must include 
17  * the following acknowledgment: "This product includes software developed by 
18  * the University Corporation for Advanced Internet Development 
19  * <http://www.ucaid.edu>Internet2 Project. Alternately, this acknowledegement 
20  * may appear in the software itself, if and wherever such third-party 
21  * acknowledgments normally appear.
22  * 
23  * Neither the name of Shibboleth nor the names of its contributors, nor 
24  * Internet2, nor the University Corporation for Advanced Internet Development, 
25  * Inc., nor UCAID may be used to endorse or promote products derived from this 
26  * software without specific prior written permission. For written permission, 
27  * please contact shibboleth@shibboleth.org
28  * 
29  * Products derived from this software may not be called Shibboleth, Internet2, 
30  * UCAID, or the University Corporation for Advanced Internet Development, nor 
31  * may Shibboleth appear in their name, without prior written permission of the 
32  * University Corporation for Advanced Internet Development.
33  * 
34  * 
35  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
36  * AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 
37  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A 
38  * PARTICULAR PURPOSE, AND NON-INFRINGEMENT ARE DISCLAIMED AND THE ENTIRE RISK 
39  * OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS WITH LICENSEE. 
40  * IN NO EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY 
41  * CORPORATION FOR ADVANCED INTERNET DEVELOPMENT, INC. BE LIABLE FOR ANY DIRECT, 
42  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
43  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
44  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 
45  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
46  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 
47  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
48  */
49
50 package edu.internet2.middleware.shibboleth.aa.arp.provider;
51
52 import java.io.File;
53 import java.io.FileNotFoundException;
54 import java.io.IOException;
55 import java.io.InputStream;
56 import java.security.Principal;
57 import java.util.Properties;
58
59 import org.apache.log4j.Logger;
60 import org.apache.xerces.parsers.DOMParser;
61 import org.w3c.dom.Element;
62 import org.xml.sax.InputSource;
63 import org.xml.sax.SAXException;
64
65 import edu.internet2.middleware.shibboleth.aa.arp.Arp;
66 import edu.internet2.middleware.shibboleth.aa.arp.ArpRepository;
67 import edu.internet2.middleware.shibboleth.aa.arp.ArpRepositoryException;
68 import edu.internet2.middleware.shibboleth.common.ShibResource;
69
70 /**
71  * Simple <code>ArpRepository</code> implementation that uses a filesystem for storage.
72  * 
73  * @author Walter Hoehn (wassa@columbia.edu)
74  */
75
76 public class FileSystemArpRepository extends BaseArpRepository implements ArpRepository {
77
78         private static Logger log = Logger.getLogger(FileSystemArpRepository.class.getName());
79         private final String siteArpFileName = "arp.site.xml";
80
81         private String dataStorePath;
82
83         public FileSystemArpRepository(Properties props) throws ArpRepositoryException {
84                 super(props);
85                 if (props
86                         .getProperty(
87                                 "edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path",
88                                 null)
89                         == null) {
90                         log.error(
91                                 "Cannot initialize FileSystemArpRepository: attribute (edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path) not specified");
92                         throw new ArpRepositoryException("Cannot initialize FileSystemArpRepository");
93                 }
94
95                 try {
96                         File givenPath =
97                                 new ShibResource(
98                                         props.getProperty(
99                                                 "edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path"),
100                                         this.getClass())
101                                         .getFile();
102
103                         if (!givenPath.isDirectory()) {
104                                 log.error(
105                                         "Cannot initialize FileSystemArpRepository: specified path is not a directory: ("
106                                                 + givenPath.getPath()
107                                                 + ").");
108                                 throw new ArpRepositoryException("Cannot initialize FileSystemArpRepository");
109                         }
110
111                         dataStorePath =
112                                 props.getProperty(
113                                         "edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path");
114                         if (!dataStorePath.endsWith("/")) {
115                                 dataStorePath += "/";
116                         }
117                         log.info("Initializing File System Arp Repository with a root of (" + dataStorePath + ").");
118                 } catch (Exception e) {
119                         log.error(
120                                 "Cannot initialize FileSystemArpRepository: error accessing path: ("
121                                         + props.getProperty(
122                                                 "edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path")
123                                         + "): "
124                                         + e);
125                         throw new ArpRepositoryException("Cannot initialize FileSystemArpRepository");
126                 }
127         }
128         /**
129          * @see edu.internet2.middleware.shibboleth.aa.arp.ArpRepository#remove(Arp)
130          */
131         public void remove(Arp arp) throws ArpRepositoryException {
132                 throw new ArpRepositoryException("Remove not implemented for FileSystemArpRepository.");
133         }
134
135         /**
136          * @see edu.internet2.middleware.shibboleth.aa.arp.ArpRepository#update(Arp)
137          */
138         public void update(Arp arp) throws ArpRepositoryException {
139                 throw new ArpRepositoryException("Update not implemented for FileSystemArpRepository.");
140         }
141
142         /**
143          * @see edu.internet2.middleware.shibboleth.aa.arp.provider.BaseArpRepository#retrieveSiteArpXml()
144          */
145         protected Element retrieveSiteArpXml() throws IOException, SAXException {
146
147                 String fileName = dataStorePath + siteArpFileName;
148                 log.debug("Attempting to load site ARP from: (" + fileName + ").");
149                 return retrieveArpXml(fileName);
150
151         }
152
153         private Element retrieveArpXml(String fileName) throws FileNotFoundException, SAXException, IOException {
154
155                 File arpFile = new ShibResource(fileName, this.getClass()).getFile();
156                 if (!arpFile.exists()) {
157                         log.debug("No ARP found.");
158                         return null;
159                 }
160
161                 InputStream inStream = new ShibResource(fileName).getInputStream();
162                 DOMParser parser = new DOMParser();
163                 parser.parse(new InputSource(inStream));
164                 inStream.close();
165                 return parser.getDocument().getDocumentElement();
166         }
167
168         /**
169          * @see edu.internet2.middleware.shibboleth.aa.arp.provider.BaseArpRepository#retrieveUserArpXml(Principal)
170          */
171         protected Element retrieveUserArpXml(Principal principal) throws IOException, SAXException {
172                 String fileName =
173                         dataStorePath
174                                 + "arp.user."
175                                 + principal.getName()
176                                 + ".xml";
177                 log.debug(
178                         "Attempting to load user (" + principal.getName() + ") ARP from: (" + fileName + ").");
179                 return retrieveArpXml(fileName);
180         }
181
182 }