cff05b6a2be1185262d89484dff424da0eef8f82
[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.FileInputStream;
54 import java.io.FileNotFoundException;
55 import java.io.IOException;
56 import java.io.InputStream;
57 import java.security.Principal;
58 import java.util.Properties;
59
60 import org.apache.log4j.Logger;
61 import org.apache.xerces.parsers.DOMParser;
62 import org.w3c.dom.Element;
63 import org.xml.sax.InputSource;
64 import org.xml.sax.SAXException;
65
66 import edu.internet2.middleware.shibboleth.aa.arp.Arp;
67 import edu.internet2.middleware.shibboleth.aa.arp.ArpRepository;
68 import edu.internet2.middleware.shibboleth.aa.arp.ArpRepositoryException;
69 import edu.internet2.middleware.shibboleth.common.ShibResource;
70
71 /**
72  * Simple <code>ArpRepository</code> implementation that uses a filesystem for storage.
73  * 
74  * @author Walter Hoehn (wassa@columbia.edu)
75  */
76
77 public class FileSystemArpRepository extends BaseArpRepository implements ArpRepository {
78
79         private static Logger log = Logger.getLogger(FileSystemArpRepository.class.getName());
80         private final String siteArpFileName = "arp.site.xml";
81
82         private String dataStorePath;
83
84         public FileSystemArpRepository(Properties props) throws ArpRepositoryException {
85                 super(props);
86                 if (props.getProperty("edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path", null)
87                         == null) {
88                         log.error(
89                                 "Cannot initialize FileSystemArpRepository: attribute (edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path) not specified");
90                         throw new ArpRepositoryException("Cannot initialize FileSystemArpRepository");
91                 }
92
93                 try {
94                         File givenPath =
95                                 new ShibResource(
96                                         props.getProperty(
97                                                 "edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path"),
98                                         this.getClass())
99                                         .getFile();
100
101                         if (!givenPath.isDirectory()) {
102                                 log.error(
103                                         "Cannot initialize FileSystemArpRepository: specified path is not a directory: ("
104                                                 + givenPath.getPath()
105                                                 + ").");
106                                 throw new ArpRepositoryException("Cannot initialize FileSystemArpRepository");
107                         }
108
109                         dataStorePath =
110                                 props.getProperty("edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path");
111                         if (!dataStorePath.endsWith("/")) {
112                                 dataStorePath += "/";
113                         }
114                         log.info("Initializing File System Arp Repository with a root of (" + dataStorePath + ").");
115                 } catch (IOException e) {
116                         log.error(
117                                 "Cannot initialize FileSystemArpRepository: error accessing path: ("
118                                         + props.getProperty(
119                                                 "edu.internet2.middleware.shibboleth.aa.arp.provider.FileSystemArpRepository.Path")
120                                         + ").");
121                         throw new ArpRepositoryException("Cannot initialize FileSystemArpRepository");
122                 }
123         }
124
125         /**
126          * @see edu.internet2.middleware.shibboleth.aa.arp.ArpRepository#remove(Arp)
127          */
128         public void remove(Arp arp) throws ArpRepositoryException {
129                 throw new ArpRepositoryException("Remove not implemented for FileSystemArpRepository.");
130         }
131
132         /**
133          * @see edu.internet2.middleware.shibboleth.aa.arp.ArpRepository#update(Arp)
134          */
135         public void update(Arp arp) throws ArpRepositoryException {
136                 throw new ArpRepositoryException("Update not implemented for FileSystemArpRepository.");
137         }
138
139         /**
140          * @see edu.internet2.middleware.shibboleth.aa.arp.provider.BaseArpRepository#retrieveSiteArpXml()
141          */
142         protected Element retrieveSiteArpXml() throws IOException, SAXException {
143
144                 String fileName = dataStorePath + siteArpFileName;
145                 log.debug("Attempting to load site ARP from: (" + fileName + ").");
146                 return retrieveArpXml(fileName);
147
148         }
149
150         private Element retrieveArpXml(String fileName) throws FileNotFoundException, SAXException, IOException {
151
152                 File arpFile = new ShibResource(fileName, this.getClass()).getFile();
153                 if (!arpFile.exists()) {
154                         log.debug("No ARP found.");
155                         return null;
156                 }
157
158                 InputStream inStream = new FileInputStream(fileName);
159                 DOMParser parser = new DOMParser();
160                 parser.parse(new InputSource(inStream));
161                 return parser.getDocument().getDocumentElement();
162         }
163
164         /**
165          * @see edu.internet2.middleware.shibboleth.aa.arp.provider.BaseArpRepository#retrieveUserArpXml(Principal)
166          */
167         protected Element retrieveUserArpXml(Principal principal) throws IOException, SAXException {
168                 String fileName =
169                         dataStorePath
170                                 + "arp.user."
171                                 + principal.getName()
172                                 + ".xml";
173                 log.debug(
174                         "Attempting to load user (" + principal.getName() + ") ARP from: (" + fileName + ").");
175                 return retrieveArpXml(fileName);
176         }
177
178 }