2 * Copyright 2001-2005 Internet2
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
18 * shib-mlp.cpp -- The ShibTarget Markup Language processor
20 * Created by: Derek Atkins <derek@ihtfp.com>
29 #include <xercesc/util/XercesDefs.hpp>
33 using namespace shibboleth;
34 using namespace shibtarget;
35 using namespace shibtarget::logging;
37 class shibtarget::ShibMLPPriv {
43 static void html_encode(string& os, const char* start);
47 void ShibMLPPriv::html_encode(string& os, const char* start)
49 while (start && *start) {
51 case '<': os += "<"; break;
52 case '>': os += ">"; break;
53 case '"': os += """; break;
54 case '#': os += "#"; break;
55 case '%': os += "%"; break;
56 case '&': os += "&"; break;
57 case '\'': os += "'"; break;
58 case '(': os += "("; break;
59 case ')': os += ")"; break;
60 case ':': os += ":"; break;
61 case '[': os += "["; break;
62 case '\\': os += "\"; break;
63 case ']': os += "]"; break;
64 case '`': os += "`"; break;
65 case '{': os += "{"; break;
66 case '}': os += "}"; break;
67 default: os += *start;
73 ShibMLPPriv::ShibMLPPriv() : log(&(Category::getInstance("shibtarget.ShibMLP"))) {}
75 static void trimspace (string& s)
77 int end = s.size() - 1, start = 0;
79 // Trim stuff on right.
80 while (end > 0 && !isgraph(s[end])) end--;
82 // Trim stuff on left.
83 while (start < end && !isgraph(s[start])) start++;
86 s = s.substr(start, end - start + 1);
91 m_priv = new ShibMLPPriv ();
99 const char* ShibMLP::run(const string& is, const IPropertySet* props, std::string* output)
101 // Create a timestamp
102 time_t now = time(NULL);
105 insert("now", ctime_r(&now,timebuf));
107 insert("now", ctime(&now));
112 const char* line = is.c_str();
113 const char* lastpos = line;
116 m_priv->log->debug("Processing string");
119 // Search for SHIBMLP tags. These are of the form:
121 // <shibmlpif key> stuff </shibmlpif>
122 // <shibmlpifnot key> stuff </shibmlpifnot>
123 // Note that there MUST be white-space after "<shibmlp" but
124 // there does not need to be white space between the key and
127 while ((thispos = strchr(lastpos, '<')) != NULL) {
128 // save the string up to this token
129 *output += is.substr(lastpos-line, thispos-lastpos);
131 // Make sure this token matches our tokens.
132 #ifdef HAVE_STRCASECMP
133 if (!strncasecmp(thispos, "<shibmlp ", 9))
135 if (!strnicmp(thispos, "<shibmlp ", 9))
138 // Save this position off.
139 lastpos = thispos + 9; // strlen("<shibmlp ")
141 // search for the end-tag
142 if ((thispos = strstr(lastpos, "/>")) != NULL) {
143 string key = is.substr(lastpos-line, thispos-lastpos);
146 map<string,string>::const_iterator i=m_map.find(key);
147 if (i != m_map.end()) {
148 m_priv->html_encode(*output,i->second.c_str());
151 pair<bool,const char*> p=props ? props->getString(key.c_str()) : pair<bool,const char*>(false,NULL);
153 m_priv->html_encode(*output,p.second);
156 static const char* s1 = "<!-- Unknown SHIBMLP key: ";
157 static const char* s2 = "/>";
162 lastpos = thispos + 2; // strlen("/>")
165 #ifdef HAVE_STRCASECMP
166 else if (!strncasecmp(thispos, "<shibmlpif ", 11))
168 else if (!strnicmp(thispos, "<shibmlpif ", 11))
171 // Save this position off.
172 lastpos = thispos + 11; // strlen("<shibmlpif ")
174 // search for the end of this tag
175 if ((thispos = strchr(lastpos, '>')) != NULL) {
176 string key = is.substr(lastpos-line, thispos-lastpos);
179 map<string,string>::const_iterator i=m_map.find(key);
180 if (i != m_map.end() && !i->second.empty()) {
184 pair<bool,const char*> p=props ? props->getString(key.c_str()) : pair<bool,const char*>(false,NULL);
189 lastpos = thispos + 1; // strlen(">")
191 // Search for the closing tag.
192 const char* frontpos=lastpos;
193 while ((thispos = strstr(lastpos, "</")) != NULL) {
194 #ifdef HAVE_STRCASECMP
195 if (!strncasecmp(thispos, "</shibmlpif>", 12))
197 if (!strnicmp(thispos, "</shibmlpif>", 12))
200 // We found our terminator. Process the string in between.
202 run(is.substr(frontpos-line, thispos-frontpos),props,&segment);
205 lastpos = thispos + 12; // strlen("</shibmlpif>")
210 lastpos = thispos + 2;
215 #ifdef HAVE_STRCASECMP
216 else if (!strncasecmp(thispos, "<shibmlpifnot ", 14))
218 else if (!strnicmp(thispos, "<shibmlpifnot ", 14))
221 // Save this position off.
222 lastpos = thispos + 14; // strlen("<shibmlpifnot ")
224 // search for the end of this tag
225 if ((thispos = strchr(lastpos, '>')) != NULL) {
226 string key = is.substr(lastpos-line, thispos-lastpos);
229 map<string,string>::const_iterator i=m_map.find(key);
230 if (i != m_map.end() && !i->second.empty()) {
234 pair<bool,const char*> p=props ? props->getString(key.c_str()) : pair<bool,const char*>(false,NULL);
239 lastpos = thispos + 1; // strlen(">")
241 // Search for the closing tag.
242 const char* frontpos=lastpos;
243 while ((thispos = strstr(lastpos, "</")) != NULL) {
244 #ifdef HAVE_STRCASECMP
245 if (!strncasecmp(thispos, "</shibmlpifnot>", 15))
247 if (!strnicmp(thispos, "</shibmlpifnot>", 15))
250 // We found our terminator. Process the string in between.
252 run(is.substr(frontpos-line, thispos-frontpos),props,&segment);
255 lastpos = thispos + 15; // strlen("</shibmlpifnot>")
260 lastpos = thispos + 2;
268 lastpos = thispos + 1;
271 *output += is.substr(lastpos-line);
273 return output->c_str();
276 const char* ShibMLP::run(istream& is, const IPropertySet* props, std::string* output)
278 static string eol = "\r\n";
281 m_priv->log->debug("processing stream");
283 while (getline(is, line))
286 return run(str,props,output);
289 void ShibMLP::insert(SAMLException& e)
291 insert("errorType", e.classname());
292 if (typeid(e)==typeid(ContentTypeException))
293 insert("errorText", "A problem was detected with your identity provider's software configuration.");
295 insert("errorText", e.getMessage() ? e.getMessage() : "No Message");
296 if (e.getProperty("errorURL"))
297 insert("originErrorURL", e.getProperty("errorURL"));
298 if (e.getProperty("contactName"))
299 insert("originContactName", e.getProperty("contactName"));
300 const char* email=e.getProperty("contactEmail");
302 if (!strncmp(email,"mailto:",7) && strlen(email)>7)
303 insert("originContactEmail", email+7);
305 insert("originContactEmail", email);
309 void ShibMLP::insert (const std::string& key, const std::string& value)
311 m_priv->log->debug("inserting %s -> %s", key.c_str(), value.c_str());