b54e5d5d80bd577e86b1c8a1a2838d7aa9e11394
[shibboleth/sp.git] / shibsp / Application.cpp
1 /*
2  *  Copyright 2001-2009 Internet2
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 /**
18  * Application.cpp
19  *
20  * Interface to a Shibboleth Application instance.
21  */
22
23 #include "internal.h"
24 #include "Application.h"
25 #include "SPRequest.h"
26 #include "ServiceProvider.h"
27 #include "attribute/Attribute.h"
28 #include "remoting/ListenerService.h"
29
30 #include <algorithm>
31 #include <xmltooling/util/Threads.h>
32
33 using namespace shibsp;
34 using namespace xmltooling;
35 using namespace std;
36
37 Application::Application(const ServiceProvider* sp) : m_sp(sp), m_lock(RWLock::create())
38 {
39 }
40
41 Application::~Application()
42 {
43     delete m_lock;
44 }
45
46 const ServiceProvider& Application::getServiceProvider() const
47 {
48     return *m_sp;
49 }
50
51 const char* Application::getId() const
52 {
53     return getString("id").second;
54 }
55
56 pair<string,const char*> Application::getCookieNameProps(const char* prefix, time_t* lifetime) const
57 {
58     static const char* defProps="; path=/";
59
60     if (lifetime)
61         *lifetime = 0;
62     const PropertySet* props=getPropertySet("Sessions");
63     if (props) {
64         if (lifetime) {
65             pair<bool,unsigned int> lt = props->getUnsignedInt("cookieLifetime");
66             if (lt.first)
67                 *lifetime = lt.second;
68         }
69         pair<bool,const char*> p=props->getString("cookieProps");
70         if (!p.first)
71             p.second=defProps;
72         pair<bool,const char*> p2=props->getString("cookieName");
73         if (p2.first)
74             return make_pair(string(prefix) + p2.second,p.second);
75         return make_pair(string(prefix) + getHash(),p.second);
76     }
77
78     // Shouldn't happen, but just in case..
79     return pair<string,const char*>(prefix,defProps);
80 }
81
82 void Application::clearHeader(SPRequest& request, const char* rawname, const char* cginame) const
83 {
84     request.clearHeader(rawname, cginame);
85 }
86
87 void Application::setHeader(SPRequest& request, const char* name, const char* value) const
88 {
89     request.setHeader(name, value);
90 }
91
92 string Application::getSecureHeader(const SPRequest& request, const char* name) const
93 {
94     return request.getSecureHeader(name);
95 }
96
97 void Application::clearAttributeHeaders(SPRequest& request) const
98 {
99     if (SPConfig::getConfig().isEnabled(SPConfig::OutOfProcess)) {
100         for (vector< pair<string,string> >::const_iterator i = m_unsetHeaders.begin(); i!=m_unsetHeaders.end(); ++i)
101             request.clearHeader(i->first.c_str(), i->second.c_str());
102         return;
103     }
104
105     m_lock->rdlock();
106     if (m_unsetHeaders.empty()) {
107         // No headers yet, so we have to request them from the remote half.
108         m_lock->unlock();
109         m_lock->wrlock();
110         if (m_unsetHeaders.empty()) {
111             SharedLock wrlock(m_lock, false);
112             string addr=string(getId()) + "::getHeaders::Application";
113             DDF out,in = DDF(addr.c_str());
114             DDFJanitor jin(in),jout(out);
115             out = getServiceProvider().getListenerService()->send(in);
116             if (out.islist()) {
117                 DDF header = out.first();
118                 while (header.isstring()) {
119                     m_unsetHeaders.push_back(pair<string,string>(header.name(),header.string()));
120                     header = out.next();
121                 }
122             }
123         }
124         else {
125             m_lock->unlock();
126         }
127         m_lock->rdlock();
128     }
129
130     // Now holding read lock.
131     SharedLock unsetLock(m_lock, false);
132     for (vector< pair<string,string> >::const_iterator i = m_unsetHeaders.begin(); i!=m_unsetHeaders.end(); ++i)
133         request.clearHeader(i->first.c_str(), i->second.c_str());
134 }