MpdWebScreenshots.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 MpdWebScreenshots header file                 -----
// -----                Created 11/12/15  by K. Smirnov                -----
// -------------------------------------------------------------------------
 
#ifndef MPDWEBSCREENSHOTS_H
#define MPDWEBSCREENSHOTS_H
 
#include "FairTask.h"
#include "MpdEventManager.h"
#include "TString.h"
 
///////////////////////////////////////////////////////////////////////////////
#define BUFFER_SIZE 512
#define MAX_FILE_SIZE 1024 * 1024
#define MAX_CONNECTIONS 3
///////////////////////////////////////////////////////////////////////////////
 
struct www_thread_par
{
    int web_port;
    TString outputDir;
    int daemon;
};
 
class MpdWebScreenshots : public FairTask
{
  public:
    // Standard constructor
    //*@param name        Name of task
    //*@outputDir         Output directory
    //*@param iVerbose    Verbosity level
    MpdWebScreenshots(const char* name, char* output_dir, bool isWebServer = false, Int_t iVerbose = 0);
 
    // Destructor
    virtual ~MpdWebScreenshots();
 
    // Set verbosity level. For this task and all of the subtasks.
    void SetVerbose(Int_t iVerbose) { fVerbose = iVerbose; }
    // Executed task
    virtual void Exec(Option_t* option);
    virtual InitStatus Init();
    virtual void SetParContainers();
    // Action after each event
    virtual void Finish();
 
    // Set format of saved files
    void SetFormatFiles(int i_format_files) { iFormatFiles = i_format_files; }
    // Set quantity of files
    void SetMultiFiles(bool is_multi_files) { isMultiFiles = is_multi_files; }
    // Set Number port
    void SetPort(int NumberPort) { web_port = NumberPort; }
 
  private:
    // Default constructor
    MpdWebScreenshots();
    MpdWebScreenshots(const MpdWebScreenshots&) = delete;
    MpdWebScreenshots& operator=(const MpdWebScreenshots&) = delete;
 
    static int daemonize();
    static int sendString(const char* message, int socket);
    static void sendHeader(const char* Status_code, char* Content_Type, int TotalSize, int socket);
    static void sendHTML(char* statusCode, char* contentType, char* content, int size, int socket);
    static void sendFile(FILE* fp, int connecting_socket);
    static int scan(char* input, char* output, size_t start, size_t max);
    static int checkMime(char* extension, char* mime_type);
    static int getHttpVersion(char* input, char* output);
    static int GetExtension(char* input, char* output, size_t max);
    static int handleHttpGET(char* input, TString output_dir, int connecting_socket);
    static int getRequestType(char* input);
    static int receive(int connecting_socket, TString output_dir);
    static int acceptConnection(int currentSocket, TString output_dir);
    static int start(int webPort, TString output_dir);
    static int start_server(void* ptr);
 
    MpdEventManager* fMan;
 
    // 0 - PNG, 1 -JPG, 2 - both types
    int iFormatFiles;
    // 0 - one (the same) file event.png; 1 - multiple files event_nnn.png (event1.png and etc.)
    bool isMultiFiles;
 
    TString outputDir;
    int web_port;
    int daemon;
 
    bool isWebStarted;
    bool isWeb;
 
    ClassDef(MpdWebScreenshots, 1);
};
 
#endif