include/root/THttpServer.h

0001 // $Id$
0002 // Author: Sergey Linev   21/12/2013
0003
0004 /*************************************************************************
0005  * Copyright (C) 1995-2013, Rene Brun and Fons Rademakers.               *
0006  * All rights reserved.                                                  *
0007  *                                                                       *
0008  * For the licensing terms see $ROOTSYS/LICENSE.                         *
0009  * For the list of contributors see $ROOTSYS/README/CREDITS.             *
0010  *************************************************************************/
0011
0012 #ifndef ROOT_THttpServer
0013 #define ROOT_THttpServer
0014
0015 #include "TNamed.h"
0016 #include "TList.h"
0017 #include "THttpCallArg.h"
0018
0019 #include <mutex>
0020 #include <map>
0021 #include <string>
0022 #include <memory>
0023 #include <queue>
0024 #include <thread>
0025 #include <vector>
0026
0027 class THttpEngine;
0028 class THttpTimer;
0029 class TRootSniffer;
0030
0031 class THttpServer : public TNamed {
0032
0033 protected:
0034    TList fEngines;                      ///<! engines which runs http server
0035    std::unique_ptr<THttpTimer> fTimer;   ///<! timer used to access main thread
0036    std::unique_ptr<TRootSniffer> fSniffer; ///<! sniffer provides access to ROOT objects hierarchy
0037    Bool_t fTerminated{kFALSE};          ///<! termination flag, disables all requests processing
0038    Long_t fMainThrdId{0};               ///<! id of the thread for processing requests
0039    Long_t fProcessingThrdId{0};         ///<! id of the thread where events are recently processing
0040    Bool_t fOwnThread{kFALSE};           ///<! true when specialized thread allocated for processing requests
0041    std::thread fThrd;                   ///<! own thread
0042    Bool_t fWSOnly{kFALSE};              ///<! when true, handle only websockets / longpoll engine
0043
0044    TString fJSROOTSYS;       ///<! location of local JSROOT files
0045    TString fTopName{"ROOT"}; ///<! name of top folder, default - "ROOT"
0046    TString fJSROOT;          ///<! location of external JSROOT files
0047
0048    std::map<std::string, std::string> fLocations; ///<! list of local directories, which could be accessed via server
0049
0050    std::string fDefaultPage;     ///<! file name for default page name
0051    std::string fDefaultPageCont; ///<! content of default html page
0052    std::string fDrawPage;        ///<! file name for drawing of single element
0053    std::string fDrawPageCont;    ///<! content of draw html page
0054    std::string fCors;            ///<! CORS: sets Access-Control-Allow-Origin header for ProcessRequest responses
0055    std::string fCorsCredentials; ///<! CORS: add Access-Control-Allow-Credentials: true response header
0056
0057    std::mutex fMutex;                                        ///<! mutex to protect list with arguments
0058    std::queue<std::shared_ptr<THttpCallArg>> fArgs;          ///<! submitted arguments
0059
0060    std::mutex fWSMutex;                                      ///<! mutex to protect WS handler lists
0061    std::vector<std::shared_ptr<THttpWSHandler>> fWSHandlers; ///<! list of WS handlers
0062
0063    virtual void MissedRequest(THttpCallArg *arg);
0064
0065    virtual void ProcessRequest(std::shared_ptr<THttpCallArg> arg);
0066
0067    virtual void ProcessBatchHolder(std::shared_ptr<THttpCallArg> &arg);
0068
0069    void StopServerThread();
0070
0071    std::string BuildWSEntryPage();
0072
0073    void ReplaceJSROOTLinks(std::shared_ptr<THttpCallArg> &arg, const std::string &version = "");
0074
0075    static Bool_t VerifyFilePath(const char *fname);
0076
0077    THttpServer(const THttpServer &) = delete;
0078    THttpServer &operator=(const THttpServer &) = delete;
0079
0080 public:
0081    THttpServer(const char *engine = "http:8080");
0082    virtual ~THttpServer();
0083
0084    virtual Bool_t CreateEngine(const char *engine);
0085
0086    Bool_t IsAnyEngine() const { return fEngines.GetSize() > 0; }
0087
0088    /** returns pointer on objects sniffer */
0089    TRootSniffer *GetSniffer() const { return fSniffer.get(); }
0090
0091    void SetSniffer(TRootSniffer *sniff);
0092
0093    Bool_t IsReadOnly() const;
0094
0095    void SetReadOnly(Bool_t readonly = kTRUE);
0096
0097    Bool_t IsAllowPostObject() const;
0098
0099    void SetAllowPostObject(Bool_t allow_post_obj);
0100
0101    Bool_t IsWSOnly() const;
0102
0103    void SetWSOnly(Bool_t on = kTRUE);
0104
0105    /** set termination flag, no any further requests will be processed */
0106    void SetTerminate();
0107
0108    /** returns kTRUE, if server was terminated */
0109    Bool_t IsTerminated() const { return fTerminated; }
0110
0111    /** Enable CORS header to ProcessRequests() responses
0112     * Specified location (typically "*") add as "Access-Control-Allow-Origin" header */
0113    void SetCors(const std::string &domain = "*") { fCors = domain; }
0114
0115    /** Returns kTRUE if CORS was configured */
0116    Bool_t IsCors() const { return !fCors.empty(); }
0117
0118    /** Returns specified CORS domain */
0119    const char *GetCors() const { return fCors.c_str(); }
0120
0121    /** Enable/disable usage Access-Control-Allow-Credentials response header */
0122    void SetCorsCredentials(const std::string &value = "true") { fCorsCredentials = value; }
0123
0124    /** Returns kTRUE if Access-Control-Allow-Credentials header should be used */
0125    Bool_t IsCorsCredentials() const { return !fCorsCredentials.empty(); }
0126
0127    /** Returns specified CORS credentials value - if any */
0128    const char *GetCorsCredentials() const { return fCorsCredentials.c_str(); }
0129
0130    /** set name of top item in objects hierarchy */
0131    void SetTopName(const char *top) { fTopName = top; }
0132
0133    /** returns name of top item in objects hierarchy */
0134    const char *GetTopName() const { return fTopName.Data(); }
0135
0136    void SetJSROOT(const char *location);
0137
0138    void AddLocation(const char *prefix, const char *path);
0139
0140    void SetDefaultPage(const std::string &filename = "");
0141
0142    void SetDrawPage(const std::string &filename = "");
0143
0144    void SetTimer(Long_t milliSec = 100, Bool_t mode = kTRUE);
0145
0146    void CreateServerThread();
0147
0148    /** Check if file is requested, thread safe */
0149    Bool_t IsFileRequested(const char *uri, TString &res) const;
0150
0151    /** Execute HTTP request */
0152    Bool_t ExecuteHttp(std::shared_ptr<THttpCallArg> arg);
0153
0154    /** Submit HTTP request */
0155    Bool_t SubmitHttp(std::shared_ptr<THttpCallArg> arg, Bool_t can_run_immediately = kFALSE);
0156
0157    /** Process submitted requests, must be called from appropriate thread */
0158    Int_t ProcessRequests();
0159
0160    /** Register object in subfolder */
0161    Bool_t Register(const char *subfolder, TObject *obj);
0162
0163    /** Unregister object */
0164    Bool_t Unregister(TObject *obj);
0165
0166    /** Register WS handler*/
0167    void RegisterWS(std::shared_ptr<THttpWSHandler> ws);
0168
0169    /** Unregister WS handler*/
0170    void UnregisterWS(std::shared_ptr<THttpWSHandler> ws);
0171
0172    /** Find web-socket handler with given name */
0173    std::shared_ptr<THttpWSHandler> FindWS(const char *name);
0174
0175    /** Execute WS request */
0176    Bool_t ExecuteWS(std::shared_ptr<THttpCallArg> &arg, Bool_t external_thrd = kFALSE, Bool_t wait_process = kFALSE);
0177
0178    /** Restrict access to specified object */
0179    void Restrict(const char *path, const char *options);
0180
0181    Bool_t RegisterCommand(const char *cmdname, const char *method, const char *icon = nullptr);
0182
0183    Bool_t Hide(const char *fullname, Bool_t hide = kTRUE);
0184
0185    Bool_t SetIcon(const char *fullname, const char *iconname);
0186
0187    Bool_t CreateItem(const char *fullname, const char *title);
0188
0189    Bool_t SetItemField(const char *fullname, const char *name, const char *value);
0190
0191    const char *GetItemField(const char *fullname, const char *name);
0192
0193    /** Guess mime type base on file extension */
0194    static const char *GetMimeType(const char *path);
0195
0196    /** Reads content of file from the disk */
0197    static char *ReadFileContent(const char *filename, Int_t &len);
0198
0199    /** Reads content of file from the disk, use std::string in return value */
0200    static std::string ReadFileContent(const std::string &filename);
0201
0202    ClassDefOverride(THttpServer, 0) // HTTP server for ROOT analysis
0203 };
0204
0205 #endif