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) const
  46 {
  47     static const char* defProps="; path=/";
  48
  49     const PropertySet* props=getPropertySet("Sessions");
  50     if (props) {
  51         pair<bool,const char*> p=props->getString("cookieProps");
  52         if (!p.first)
  53             p.second=defProps;
  54         pair<bool,const char*> p2=props->getString("cookieName");
  55         if (p2.first)
  56             return make_pair(string(prefix) + p2.second,p.second);
  57         return make_pair(string(prefix) + getHash(),p.second);
  58     }
  59
  60     // Shouldn't happen, but just in case..
  61     return pair<string,const char*>(prefix,defProps);
  62 }
  63
  64 void Application::clearAttributeHeaders(SPRequest& request) const
  65 {
  66     if (SPConfig::getConfig().isEnabled(SPConfig::OutOfProcess)) {
  67         for (vector< pair<string,string> >::const_iterator i = m_unsetHeaders.begin(); i!=m_unsetHeaders.end(); ++i)
  68             request.clearHeader(i->first.c_str(), i->second.c_str());
  69         return;
  70     }
  71
  72     m_lock->rdlock();
  73     if (m_unsetHeaders.empty()) {
  74         // No headers yet, so we have to request them from the remote half.
  75         m_lock->unlock();
  76         m_lock->wrlock();
  77         if (m_unsetHeaders.empty()) {
  78             SharedLock wrlock(m_lock, false);
  79             string addr=string(getId()) + "::getHeaders::Application";
  80             DDF out,in = DDF(addr.c_str());
  81             DDFJanitor jin(in),jout(out);
  82             out = getServiceProvider().getListenerService()->send(in);
  83             if (out.islist()) {
  84                 DDF header = out.first();
  85                 while (header.isstring()) {
  86                     m_unsetHeaders.push_back(pair<string,string>(header.name(),header.string()));
  87                     header = out.next();
  88                 }
  89             }
  90         }
  91         else {
  92             m_lock->unlock();
  93         }
  94         m_lock->rdlock();
  95     }
  96
  97     // Now holding read lock.
  98     SharedLock unsetLock(m_lock, false);
  99     for (vector< pair<string,string> >::const_iterator i = m_unsetHeaders.begin(); i!=m_unsetHeaders.end(); ++i)
 100         request.clearHeader(i->first.c_str(), i->second.c_str());
 101 }