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 "MpdEventManager.h"
 
#include "FairTask.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