2 * Copyright 2001-2007 Internet2
\r
4 * Licensed under the Apache License, Version 2.0 (the "License");
\r
5 * you may not use this file except in compliance with the License.
\r
6 * You may obtain a copy of the License at
\r
8 * http://www.apache.org/licenses/LICENSE-2.0
\r
10 * Unless required by applicable law or agreed to in writing, software
\r
11 * distributed under the License is distributed on an "AS IS" BASIS,
\r
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
13 * See the License for the specific language governing permissions and
\r
14 * limitations under the License.
\r
17 /* shibresponder.cpp - Shibboleth FastCGI Responder/Handler
\r
23 #include "config_win32.h"
\r
25 #define _CRT_NONSTDC_NO_DEPRECATE 1
\r
26 #define _CRT_SECURE_NO_DEPRECATE 1
\r
27 #define _SCL_SECURE_NO_WARNINGS 1
\r
29 #include <shibsp/AbstractSPRequest.h>
\r
30 #include <shibsp/SPConfig.h>
\r
31 #include <shibsp/ServiceProvider.h>
\r
32 #include <xmltooling/unicode.h>
\r
33 #include <xmltooling/XMLToolingConfig.h>
\r
34 #include <xmltooling/util/NDC.h>
\r
35 #include <xmltooling/util/XMLConstants.h>
\r
36 #include <xmltooling/util/XMLHelper.h>
\r
37 #include <xercesc/util/XMLUniDefs.hpp>
\r
39 #include <stdexcept>
\r
41 #ifdef HAVE_UNISTD_H
\r
42 # include <unistd.h>
\r
43 # include <sys/mman.h>
\r
47 using namespace shibsp;
\r
48 using namespace xmltooling;
\r
49 using namespace xercesc;
\r
50 using namespace std;
\r
52 static const XMLCh path[] = UNICODE_LITERAL_4(p,a,t,h);
\r
53 static const XMLCh validate[] = UNICODE_LITERAL_8(v,a,l,i,d,a,t,e);
\r
61 class ShibTargetFCGI : public AbstractSPRequest
\r
63 FCGX_Request* m_req;
\r
65 multimap<string,string> m_headers;
\r
67 string m_scheme,m_hostname;
\r
70 ShibTargetFCGI(FCGX_Request* req, char* post_data, const char* scheme=NULL, const char* hostname=NULL, int port=0)
\r
71 : m_req(req), m_body(post_data) {
\r
73 const char* server_name_str = hostname;
\r
74 if (!server_name_str || !*server_name_str)
\r
75 server_name_str = FCGX_GetParam("SERVER_NAME", req->envp);
\r
76 m_hostname = server_name_str;
\r
80 char* server_port_str = FCGX_GetParam("SERVER_PORT", req->envp);
\r
81 m_port = strtol(server_port_str, &server_port_str, 10);
\r
82 if (*server_port_str) {
\r
83 cerr << "can't parse SERVER_PORT (" << FCGX_GetParam("SERVER_PORT", req->envp) << ")" << endl;
\r
84 throw runtime_error("Unable to determine server port.");
\r
88 const char* server_scheme_str = scheme;
\r
89 if (!server_scheme_str || !*server_scheme_str)
\r
90 server_scheme_str = (m_port == 443 || m_port == 8443) ? "https" : "http";
\r
91 m_scheme = server_scheme_str;
\r
94 ~ShibTargetFCGI() { }
\r
96 const char* getScheme() const {
\r
97 return m_scheme.c_str();
\r
99 const char* getHostname() const {
\r
100 return m_hostname.c_str();
\r
102 int getPort() const {
\r
105 const char* getRequestURI() const {
\r
106 return FCGX_GetParam("REQUEST_URI", m_req->envp);
\r
108 const char* getMethod() const {
\r
109 return FCGX_GetParam("REQUEST_METHOD", m_req->envp);
\r
111 string getContentType() const {
\r
112 const char* s = FCGX_GetParam("CONTENT_TYPE", m_req->envp);
\r
115 long getContentLength() const {
\r
116 const char* s = FCGX_GetParam("CONTENT_LENGTH", m_req->envp);
\r
117 return s ? atol(s) : 0;
\r
119 string getRemoteUser() const {
\r
120 const char* s = FCGX_GetParam("REMOTE_USER", m_req->envp);
\r
123 string getRemoteAddr() const {
\r
124 const char* s = FCGX_GetParam("REMOTE_ADDR", m_req->envp);
\r
127 void log(SPLogLevel level, const string& msg) const {
\r
128 AbstractSPRequest::log(level,msg);
\r
129 if (level >= SPError)
\r
130 cerr << "shib: " << msg;
\r
133 string getHeader(const char* name) const {
\r
134 string hdr("HTTP_");
\r
135 for (; *name; ++name) {
\r
139 hdr += toupper(*name);
\r
141 char* s = FCGX_GetParam(hdr.c_str(), m_req->envp);
\r
145 void setResponseHeader(const char* name, const char* value) {
\r
148 m_headers.insert(make_pair(name,value));
\r
150 m_headers.erase(name);
\r
153 const char* getQueryString() const {
\r
154 return FCGX_GetParam("QUERY_STRING", m_req->envp);
\r
157 const char* getRequestBody() const {
\r
161 long sendResponse(istream& in, long status) {
\r
162 string hdr = string("Connection: close\r\n");
\r
163 for (multimap<string,string>::const_iterator i=m_headers.begin(); i!=m_headers.end(); ++i)
\r
164 hdr += i->first + ": " + i->second + "\r\n";
\r
166 const char* codestr="Status: 200 OK";
\r
168 case XMLTOOLING_HTTP_STATUS_ERROR: codestr="Status: 500 Server Error"; break;
\r
169 case XMLTOOLING_HTTP_STATUS_FORBIDDEN:codestr="Status: 403 Forbidden"; break;
\r
170 case XMLTOOLING_HTTP_STATUS_NOTFOUND: codestr="Status: 404 Not Found"; break;
\r
172 cout << codestr << "\r\n" << hdr << "\r\n";
\r
176 cout.write(buf, in.gcount());
\r
178 return SHIB_RETURN_DONE;
\r
181 long sendRedirect(const char* url) {
\r
182 string hdr=string("Status: 302 Please Wait\r\nLocation: ") + url + "\r\n"
\r
183 "Content-Type: text/html\r\n"
\r
184 "Content-Length: 40\r\n"
\r
185 "Expires: 01-Jan-1997 12:00:00 GMT\r\n"
\r
186 "Cache-Control: private,no-store,no-cache\r\n";
\r
187 for (multimap<string,string>::const_iterator i=m_headers.begin(); i!=m_headers.end(); ++i)
\r
188 hdr += i->first + ": " + i->second + "\r\n";
\r
191 cout << hdr << "<HTML><BODY>Redirecting...</BODY></HTML>";
\r
192 return SHIB_RETURN_DONE;
\r
195 long returnDecline() {
\r
196 return SHIB_RETURN_KO;
\r
199 return SHIB_RETURN_OK;
\r
202 const vector<string>& getClientCertificates() const {
\r
203 static vector<string> g_NoCerts;
\r
207 // Not used in the extension.
\r
209 virtual void clearHeader(const char* rawname, const char* cginame) {
\r
210 throw runtime_error("clearHeader not implemented by FastCGI responder.");
\r
213 virtual void setHeader(const char* name, const char* value) {
\r
214 throw runtime_error("setHeader not implemented by FastCGI responder.");
\r
217 virtual void setRemoteUser(const char* user) {
\r
218 throw runtime_error("setRemoteUser not implemented by FastCGI responder.");
\r
222 // Maximum number of bytes allowed to be read from stdin
\r
223 static const unsigned long STDIN_MAX = 1000000;
\r
225 static long gstdin(FCGX_Request* request, char** content)
\r
227 char* clenstr = FCGX_GetParam("CONTENT_LENGTH", request->envp);
\r
228 unsigned long clen = STDIN_MAX;
\r
231 clen = strtol(clenstr, &clenstr, 10);
\r
233 cerr << "can't parse CONTENT_LENGTH (" << FCGX_GetParam("CONTENT_LENGTH", request->envp) << ")" << endl;
\r
237 // *always* put a cap on the amount of data that will be read
\r
238 if (clen > STDIN_MAX)
\r
241 *content = new char[clen];
\r
243 cin.read(*content, clen);
\r
244 clen = cin.gcount();
\r
247 // *never* read stdin when CONTENT_LENGTH is missing or unparsable
\r
252 // Chew up any remaining stdin - this shouldn't be necessary
\r
253 // but is because mod_fastcgi doesn't handle it correctly.
\r
255 // ignore() doesn't set the eof bit in some versions of glibc++
\r
256 // so use gcount() instead of eof()...
\r
257 do cin.ignore(1024); while (cin.gcount() == 1024);
\r
262 static void print_ok() {
\r
263 cout << "Status: 200 OK" << "\r\n\r\n";
\r
266 static void print_error(const char* msg) {
\r
267 cout << "Status: 500 Server Error" << "\r\n\r\n" << msg;
\r
272 const char* schemadir=getenv("SHIBSP_SCHEMAS");
\r
274 schemadir=SHIBSP_SCHEMAS;
\r
275 const char* config=getenv("SHIBSP_CONFIG");
\r
277 config=SHIBSP_CONFIG;
\r
279 cerr << "SHIBSP_CONFIG = " << config << endl
\r
280 << "SHIBSP_SCHEMAS = " << schemadir << endl;
\r
282 SPConfig* g_Config=&SPConfig::getConfig();
\r
283 g_Config->setFeatures(
\r
284 SPConfig::Listener |
\r
285 SPConfig::Caching |
\r
286 SPConfig::RequestMapping |
\r
287 SPConfig::InProcess |
\r
288 SPConfig::Logging |
\r
291 if (!g_Config->init(schemadir)) {
\r
292 cerr << "failed to initialize Shibboleth libraries" << endl;
\r
297 DOMDocument* dummydoc=XMLToolingConfig::getConfig().getParser().newDocument();
\r
298 XercesJanitor<DOMDocument> docjanitor(dummydoc);
\r
299 DOMElement* dummy = dummydoc->createElementNS(NULL,path);
\r
300 auto_ptr_XMLCh src(config);
\r
301 dummy->setAttributeNS(NULL,path,src.get());
\r
302 dummy->setAttributeNS(NULL,validate,xmlconstants::XML_ONE);
\r
304 g_Config->setServiceProvider(g_Config->ServiceProviderManager.newPlugin(XML_SERVICE_PROVIDER,dummy));
\r
305 g_Config->getServiceProvider()->init();
\r
307 catch (exception& ex) {
\r
309 cerr << "exception while initializing Shibboleth configuration: " << ex.what() << endl;
\r
313 string g_ServerScheme;
\r
314 string g_ServerName;
\r
315 int g_ServerPort=0;
\r
317 // Load "authoritative" URL fields.
\r
318 char* var = getenv("SHIBSP_SERVER_NAME");
\r
320 g_ServerName = var;
\r
321 var = getenv("SHIBSP_SERVER_SCHEME");
\r
323 g_ServerScheme = var;
\r
324 var = getenv("SHIBSP_SERVER_PORT");
\r
326 g_ServerPort = atoi(var);
\r
328 streambuf* cin_streambuf = cin.rdbuf();
\r
329 streambuf* cout_streambuf = cout.rdbuf();
\r
330 streambuf* cerr_streambuf = cerr.rdbuf();
\r
332 FCGX_Request request;
\r
335 FCGX_InitRequest(&request, 0, 0);
\r
337 cout << "Shibboleth initialization complete. Starting request loop." << endl;
\r
338 while (FCGX_Accept_r(&request) == 0) {
\r
339 // Note that the default bufsize (0) will cause the use of iostream
\r
340 // methods that require positioning (such as peek(), seek(),
\r
341 // unget() and putback()) to fail (in favour of more efficient IO).
\r
342 fcgi_streambuf cin_fcgi_streambuf(request.in);
\r
343 fcgi_streambuf cout_fcgi_streambuf(request.out);
\r
344 fcgi_streambuf cerr_fcgi_streambuf(request.err);
\r
346 cin.rdbuf(&cin_fcgi_streambuf);
\r
347 cout.rdbuf(&cout_fcgi_streambuf);
\r
348 cerr.rdbuf(&cerr_fcgi_streambuf);
\r
350 // Although FastCGI supports writing before reading,
\r
351 // many http clients (browsers) don't support it (so
\r
352 // the connection deadlocks until a timeout expires!).
\r
354 gstdin(&request, &content);
\r
357 xmltooling::NDC ndc("FastCGI shibresponder");
\r
358 ShibTargetFCGI stf(&request, content, g_ServerScheme.c_str(), g_ServerName.c_str(), g_ServerPort);
\r
360 pair<bool,long> res = stf.getServiceProvider().doHandler(stf);
\r
363 cerr << "shib: doHandler handled the request" << endl;
\r
365 switch(res.second) {
\r
366 case SHIB_RETURN_OK:
\r
370 case SHIB_RETURN_KO:
\r
371 cerr << "shib: doHandler failed to handle the request" << endl;
\r
372 print_error("<html><body>FastCGI Shibboleth responder should only be used for Shibboleth protocol requests.</body></html>");
\r
375 case SHIB_RETURN_DONE:
\r
376 // response already handled
\r
380 cerr << "shib: doHandler returned an unexpected result: " << res.second << endl;
\r
381 print_error("<html><body>FastCGI Shibboleth responder returned an unexpected result.</body></html>");
\r
386 cerr << "shib: doHandler failed to handle request." << endl;
\r
387 print_error("<html><body>FastCGI Shibboleth responder failed to process request.</body></html>");
\r
391 catch (exception& e) {
\r
392 cerr << "shib: FastCGI responder caught an exception: " << e.what() << endl;
\r
393 print_error("<html><body>FastCGI Shibboleth responder caught an exception, check log for details.</body></html>");
\r
398 // If the output streambufs had non-zero bufsizes and
\r
399 // were constructed outside of the accept loop (i.e.
\r
400 // their destructor won't be called here), they would
\r
401 // have to be flushed here.
\r
404 cout << "Request loop ended." << endl;
\r
406 cin.rdbuf(cin_streambuf);
\r
407 cout.rdbuf(cout_streambuf);
\r
408 cerr.rdbuf(cerr_streambuf);
\r