FCGIResponse.cc

Go to the documentation of this file.

/*
 * Copyright 2006-2009 Savarese Software Research Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.savarese.com/software/ApacheLicense-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <ssrc/wispers/fcgi/FCGIResponse.h>

#include <sstream>

__BEGIN_NS_SSRC_WSPR_FCGI

// TODO: Provide more general cookie handling (as well as session cookie
// configuration in relay).  Need a generalized header manipulation
// system and a Cookie class.
void FCGIResponse::set_session_id(const sid_type & session_id,
                                  const unsigned int max_age,
                                  const bool secure)
{
  std::ostringstream ostr;

  _session_id = session_id;

  ostr << "Set-Cookie: WSPRSID=" << session_id << "; Path=/; Version=1";

  if(max_age == 1) {
    ostr << "; Discard";
  } else {
    ostr << "; Max-Age=" << max_age;
  }

  if(secure) {
    ostr << "; Secure";
  }

  _session_cookie = std::move(ostr.str());
}

// TODO: Move header output and buffer output into two separate functions.
void FCGIResponse::fcgi_output(const char *content,
                               const unsigned int content_length,
                               bool cache_disable)
{
  FCGI::FCGX_Stream * const out = _request->_fcgx_request->out;
  if(cache_disable) {
    FCGI::FCGX_FPrintF(out,
                       "Pragma: no-cache\r\n"
                       "Expires: Fri, 01 Jan 1990 00:00:00 GMT\r\n"
                       "Cache-control: no-cache, must-revalidate, no-cache=\"Set-Cookie\", private\r\n");
  }

  if(!_session_cookie.empty()) {
    FCGI::FCGX_FPrintF(out, "%s\r\n", _session_cookie.c_str());
  }

  FCGI::FCGX_FPrintF(out,
                     "Status: %d\r\n"
                     "Content-Type: %s\r\n\r\n",
                     status(), content_type().c_str());

  if(content && content_length > 0) {
    FCGI::FCGX_PutStr(content, content_length, out);
  }
}

void FCGIResponse::send_redirect(const string & redirect) {
  try {
    const string && location(_request->to_absolute_url(redirect));

    if(!_session_cookie.empty())
      FCGI::FCGX_FPrintF(_request->_fcgx_request->out,
                         "Status: %d\r\nLocation: %s\r\n%s\r\n\r\n",
                         StatusFound, location.c_str(),
                         _session_cookie.c_str());
    else
      FCGI::FCGX_FPrintF(_request->_fcgx_request->out,
                         "Status: %d\r\nLocation: %s\r\n\r\n",
                         StatusFound, location.c_str());

    fcgx_finish(StatusFound);
  } catch(const std::invalid_argument & ia) {
    send_error(StatusInternalServerError);
  }
}

__END_NS_SSRC_WSPR_FCGI