8e41cab615a04a0599c7f7654c13d235ceb523c8
[shibboleth/sp.git] / shibsp / Application.cpp
1 /*
2  *  Copyright 2001-2007 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
32 using namespace shibsp;
33 using namespace xmltooling;
34 using namespace std;
35
36 Application::Application(const ServiceProvider* sp) : m_sp(sp), m_lock(RWLock::create())
37 {
38 }
39
40 Application::~Application()
41 {
42     delete m_lock;
43 }
44
45 pair<string,const char*> Application::getCookieNameProps(const char* prefix, time_t* lifetime) const
46 {
47     static const char* defProps="; path=/";
48
49     if (lifetime)
50         *lifetime = 0;
51     const PropertySet* props=getPropertySet("Sessions");
52     if (props) {
53         if (lifetime) {
54             pair<bool,unsigned int> lt = props->getUnsignedInt("cookieLifetime");
55             if (lt.first)
56                 *lifetime = lt.second;
57         }
58         pair<bool,const char*> p=props->getString("cookieProps");
59         if (!p.first)
60             p.second=defProps;
61         pair<bool,const char*> p2=props->getString("cookieName");
62         if (p2.first)
63             return make_pair(string(prefix) + p2.second,p.second);
64         return make_pair(string(prefix) + getHash(),p.second);
65     }
66
67     // Shouldn't happen, but just in case..
68     return pair<string,const char*>(prefix,defProps);
69 }
70
71 void Application::clearAttributeHeaders(SPRequest& request) const
72 {
73     if (SPConfig::getConfig().isEnabled(SPConfig::OutOfProcess)) {
74         for (vector< pair<string,string> >::const_iterator i = m_unsetHeaders.begin(); i!=m_unsetHeaders.end(); ++i)
75             request.clearHeader(i->first.c_str(), i->second.c_str());
76         return;
77     }
78
79     m_lock->rdlock();
80     if (m_unsetHeaders.empty()) {
81         // No headers yet, so we have to request them from the remote half.
82         m_lock->unlock();
83         m_lock->wrlock();
84         if (m_unsetHeaders.empty()) {
85             SharedLock wrlock(m_lock, false);
86             string addr=string(getId()) + "::getHeaders::Application";
87             DDF out,in = DDF(addr.c_str());
88             DDFJanitor jin(in),jout(out);
89             out = getServiceProvider().getListenerService()->send(in);
90             if (out.islist()) {
91                 DDF header = out.first();
92                 while (header.isstring()) {
93                     m_unsetHeaders.push_back(pair<string,string>(header.name(),header.string()));
94                     header = out.next();
95                 }
96             }
97         }
98         else {
99             m_lock->unlock();
100         }
101         m_lock->rdlock();
102     }
103
104     // Now holding read lock.
105     SharedLock unsetLock(m_lock, false);
106     for (vector< pair<string,string> >::const_iterator i = m_unsetHeaders.begin(); i!=m_unsetHeaders.end(); ++i)
107         request.clearHeader(i->first.c_str(), i->second.c_str());
108 }