shibsp/Application.cpp

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