server.h

Go to the documentation of this file.

// Copyright (C) 2010-2011 by Konrad Rosenbaum <konrad@silmor.de>
// protected under the GNU LGPL version 3 or at your option any newer.
// See COPYING.LGPL file that comes with this distribution.
//

#ifndef WOB_SERVER_H
#define WOB_SERVER_H

#include <QObject>
#include <QHostAddress>
#include <QMap>
#include <QMetaType>

class QTcpServer;
class QLocalServer;
class WInterface;

#ifndef WOLF_BASE_EXPORT
#define WOLF_BASE_EXPORT Q_DECL_IMPORT
#endif

class WOLF_BASE_EXPORT WServerRequest
{
        public:
                WServerRequest();
                WServerRequest(const WServerRequest&);
                
                WServerRequest& operator=(const WServerRequest&);
                
                bool hasHeader(const QString&)const;
                bool hasCgiHeader(const QString&)const;
                QByteArray header(const QString&)const;
                QByteArray cgiHeader(const QString&)const;
                QByteArray bodyData()const;
                
                bool isValid()const;
                
                QByteArray dump()const;
                
                QString pathInfo()const;
                QString hostInfo()const;
                
        protected:
                WServerRequest(const QByteArray&);
                void setBody(const QByteArray&);
                friend class WServer;
                friend class WServerReceiver;
                
        private:
                QMap<QByteArray,QByteArray>mhead;
                QByteArray mbody;
};
Q_DECLARE_METATYPE(WServerRequest);

class WOLF_BASE_EXPORT WServerReply
{
        public:
                WServerReply();
                WServerReply(const WServerReply&)=default;
                WServerReply& operator=(const WServerReply&)=default;
                
                bool setStatus(short code,QString str);
                bool setHeader(const QString&,const QString&);
                void setBody(const QString&);
                void setBody(const QByteArray&);
                
        protected:
                friend class WServer;
                QByteArray toWireFormat()const;
        private:
                short mStatCode;
                QByteArray mStatStr,mBody;
                QMap<QString,QPair<QString,QString> >mHead;
};

class WOLF_BASE_EXPORT WServer:public QObject
{
        Q_OBJECT
        public:
                WServer(const QString&path,bool removeIfExist=true,QObject*parent=0);
                WServer(const QHostAddress&addr,unsigned short port,QObject*parent=0);
                virtual ~WServer()=default;
                
                bool isActive()const{return tcpserv!=0 || locserv!=0;}
                
                int receiveTimeout()const{return mrecvtimeout;}
                
                quint16 tcpPort()const;
                
        public slots:
                void registerInterface(const QString&path,WInterface*ifc);
                void registerStatic(const QString&path,const QString&content);
                void unregisterPath(const QString&);
                void register404(const QString&);
                void register400(const QString&);
                void setReceiveTimeout(int);
                void restrictSourceHosts(const QPair<QHostAddress,int>&);
                void enableDebugUrl(bool enable=true);
                
        private slots:
                void newConnection();
                void handleConnection(QIODevice*);
                void handleRequest(WServerRequest,QIODevice*);
        private:
                void handleRequestHelper(WServerRequest,QIODevice*);
                QTcpServer*tcpserv;
                QLocalServer*locserv;
                QMap<QString,WInterface*>iface;
                QMap<QString,QString>statcontent;
                QString err404,err400;
                int mrecvtimeout;
                QPair<QHostAddress,int>sourcehost;
                bool mdebugenabled;
};

#endif