Main Page | Namespace List | Class Hierarchy | Alphabetical List | Class List | Directories | File List | Namespace Members | Class Members | File Members

home » brain » inspircd-cvs » inspircd » src » modules

httpd.h

Go to the documentation of this file.

/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
 * See: http://www.inspircd.org/wiki/index.php/Credits
 *
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#include "base.h"

#ifndef __HTTPD_H__
#define __HTTPD_H__

#include <string>
#include <sstream>
#include <map>

class HTTPHeaders : public classbase
{
 protected:
        std::map<std::string,std::string> headers;
 public:

        void SetHeader(const std::string &name, const std::string &data)
        {
                headers[name] = data;
        }

        void CreateHeader(const std::string &name, const std::string &data)
        {
                if (!IsSet(name))
                        SetHeader(name, data);
        }

        void RemoveHeader(const std::string &name)
        {
                headers.erase(name);
        }

        void Clear()
        {
                headers.clear();
        }

        std::string GetHeader(const std::string &name)
        {
                std::map<std::string,std::string>::iterator it = headers.find(name);
                if (it == headers.end())
                        return std::string();

                return it->second;
        }

        bool IsSet(const std::string &name)
        {
                std::map<std::string,std::string>::iterator it = headers.find(name);
                return (it != headers.end());
        }

        std::string GetFormattedHeaders()
        {
                std::string re;

                for (std::map<std::string,std::string>::iterator i = headers.begin(); i != headers.end(); i++)
                        re += i->first + ": " + i->second + "\r\n";

                return re;
        }
};

class HTTPRequest : public classbase
{
 protected:
        std::string type;
        std::string document;
        std::string ipaddr;
        std::string postdata;

 public:

        HTTPHeaders *headers;
        int errorcode;

        void* sock;

        HTTPRequest(const std::string &request_type, const std::string &uri, HTTPHeaders* hdr, void* opaque, const std::string &ip, const std::string &pdata)
                : type(request_type), document(uri), ipaddr(ip), postdata(pdata), headers(hdr), sock(opaque)
        {
        }

        std::string& GetPostData()
        {
                return postdata;
        }

        std::string& GetType()
        {
                return type;
        }

        std::string& GetURI()
        {
                return document;
        }

        std::string& GetIP()
        {
                return ipaddr;
        }
};

class HTTPDocument : public classbase
{
 protected:

        std::stringstream* document;
        int responsecode;

 public:

        HTTPHeaders headers;

        void* sock;

        HTTPDocument(void* opaque, std::stringstream* doc, int response) : document(doc), responsecode(response), sock(opaque)
        {
        }

        std::stringstream* GetDocument()
        {
                return this->document;
        }

        unsigned long GetDocumentSize()
        {
                return this->document->str().length();
        }

        int GetResponseCode()
        {
                return this->responsecode;
        }
};

#endif