XrdThrottle.hh

Go to the documentation of this file.

#ifndef __XRDTHROTTLE_H_
#define __XRDTHROTTLE_H_
 
#include <memory>
#include <string>
 
#include "XrdVersion.hh"
#include "XrdSys/XrdSysError.hh"
#include "XrdSfs/XrdSfsInterface.hh"
 
#include "XrdThrottle/XrdThrottleTrace.hh"
#include "XrdThrottle/XrdThrottleManager.hh"
 
class XrdSysLogger;
class XrdOucStream;
 
 
namespace XrdThrottle {
 
#if __cplusplus >= 201103L
typedef std::unique_ptr<XrdSfsFile> unique_sfs_ptr;
#else
typedef std::auto_ptr<XrdSfsFile> unique_sfs_ptr;
#endif
 
class FileSystem;
 
class File : public XrdSfsFile {
 
friend class FileSystem;
 
public:
 
   virtual int
   open(const char                *fileName,
              XrdSfsFileOpenMode   openMode,
              mode_t               createMode,
        const XrdSecEntity        *client,
        const char                *opaque = 0);
 
   virtual int
   close();
 
   using XrdSfsFile::fctl;
   virtual int
   fctl(const int               cmd,
        const char             *args,
              XrdOucErrInfo    &out_error);
 
   virtual const char *
   FName();
 
   virtual int
   getMmap(void **Addr, off_t &Size);
 
   virtual int
   read(XrdSfsFileOffset   fileOffset,   // Preread only
        XrdSfsXferSize     amount);
 
   virtual XrdSfsXferSize
   read(XrdSfsFileOffset   fileOffset,
        char              *buffer,
        XrdSfsXferSize     buffer_size);
 
   virtual int
   read(XrdSfsAio *aioparm);
 
   virtual XrdSfsXferSize
   write(XrdSfsFileOffset   fileOffset,
         const char        *buffer,
         XrdSfsXferSize     buffer_size);
 
   virtual int
   write(XrdSfsAio *aioparm);
 
   virtual int
   sync();
 
   virtual int
   sync(XrdSfsAio *aiop);
 
   virtual int
   stat(struct stat *buf);
 
   virtual int
   truncate(XrdSfsFileOffset   fileOffset);
 
   virtual int
   getCXinfo(char cxtype[4], int &cxrsz);
 
   virtual int
   SendData(XrdSfsDio         *sfDio,
            XrdSfsFileOffset   offset,
            XrdSfsXferSize     size);
 
private:
   File(const char *user, int monid, unique_sfs_ptr, XrdThrottleManager &throttle, XrdSysError &eroute);
 
   virtual
   ~File();
 
   unique_sfs_ptr m_sfs;
   int m_uid; // A unique identifier for this user; has no meaning except for the fairshare.
   std::string m_loadshed;
   std::string m_user;
   XrdThrottleManager &m_throttle;
   XrdSysError &m_eroute;
};
 
class FileSystem : public XrdSfsFileSystem
{
 
friend XrdSfsFileSystem * XrdSfsGetFileSystem_Internal(XrdSfsFileSystem *, XrdSysLogger *, const char *);
 
public:
 
   virtual XrdSfsDirectory *
   newDir(char *user=0, int monid=0);
 
   virtual XrdSfsFile *
   newFile(char *user=0, int monid=0);
 
   virtual int
   chksum(      csFunc         Func,
          const char          *csName,
          const char          *path,
                XrdOucErrInfo &eInfo,
          const XrdSecEntity  *client = 0,
          const char          *opaque = 0);
 
   virtual int
   chmod(const char             *Name,
               XrdSfsMode        Mode,
               XrdOucErrInfo    &out_error,
         const XrdSecEntity     *client,
         const char             *opaque = 0);
 
   virtual void
   Disc(const XrdSecEntity   *client = 0);
 
   virtual void
   EnvInfo(XrdOucEnv *envP);
 
   virtual int
   exists(const char                *fileName,
                XrdSfsFileExistence &exists_flag,
                XrdOucErrInfo       &out_error,
          const XrdSecEntity        *client,
          const char                *opaque = 0);
 
   virtual int
   fsctl(const int               cmd,
         const char             *args,
               XrdOucErrInfo    &out_error,
         const XrdSecEntity     *client);
 
   virtual int
   getStats(char *buff, int blen);
 
   virtual const char *
   getVersion();
 
   virtual int
   mkdir(const char             *dirName,
               XrdSfsMode        Mode,
               XrdOucErrInfo    &out_error,
         const XrdSecEntity     *client,
         const char             *opaque = 0);
 
   virtual int
   prepare(      XrdSfsPrep       &pargs,
                 XrdOucErrInfo    &out_error,
           const XrdSecEntity     *client = 0);
 
   virtual int
   rem(const char             *path,
             XrdOucErrInfo    &out_error,
       const XrdSecEntity     *client,
       const char             *info = 0);
 
   virtual int
   remdir(const char             *dirName,
                XrdOucErrInfo    &out_error,
          const XrdSecEntity     *client,
          const char             *info = 0);
 
   virtual int
   rename(const char             *oldFileName,
          const char             *newFileName,
                XrdOucErrInfo    &out_error,
          const XrdSecEntity     *client,
          const char             *infoO = 0,
          const char             *infoN = 0);
 
   virtual int
   stat(const char             *Name,
              struct stat      *buf,
              XrdOucErrInfo    &out_error,
        const XrdSecEntity     *client,
        const char             *opaque = 0);
 
   virtual int
   stat(const char             *Name,
              mode_t           &mode,
              XrdOucErrInfo    &out_error,
        const XrdSecEntity     *client,
        const char             *opaque = 0);
 
   virtual int
   truncate(const char             *Name,
                  XrdSfsFileOffset fileOffset,
                  XrdOucErrInfo    &out_error,
            const XrdSecEntity     *client = 0,
            const char             *opaque = 0);
 
   virtual int
   Configure(XrdSysError &, XrdSfsFileSystem *native_fs);
 
private:
   static void
   Initialize(      FileSystem      *&fs,
                    XrdSfsFileSystem *native_fs,
                    XrdSysLogger     *lp,
              const char             *config_file);
 
   FileSystem();
 
   virtual
  ~FileSystem();
 
   int
   xthrottle(XrdOucStream &Config);
 
   int
   xloadshed(XrdOucStream &Config);
 
   int
   xtrace(XrdOucStream &Config);
 
   static FileSystem  *m_instance;
   XrdSysError         m_eroute;
   XrdOucTrace         m_trace;
   std::string         m_config_file;
   XrdSfsFileSystem   *m_sfs_ptr;
   bool                m_initialized;
   XrdThrottleManager  m_throttle;
   XrdVersionInfo     *myVersion;
 
};
 
}
 
#endif