usertable.h

Go to the documentation of this file.

#ifndef _USERTABLE_H_
#define _USERTABLE_H_

#include <map>
#include <deque>
#include <sys/poll.h>

#include "inotify-cxx.h"
#include "incrontab.h"


class UserTable;

typedef std::map<std::string, UserTable*> SUT_MAP;

typedef void (*proc_done_cb)(InotifyWatch*);

typedef struct
{
  proc_done_cb onDone;  
  InotifyWatch* pWatch; 
} ProcData_t;

typedef std::map<int, UserTable*> FDUT_MAP;

typedef std::map<InotifyWatch*, IncronTabEntry*> IWCE_MAP;

typedef std::map<pid_t, ProcData_t> PROC_MAP;


class EventDispatcher
{
public:

  EventDispatcher(int iPipeFd, Inotify* pIn, InotifyWatch* pSys, InotifyWatch* pUser);
  
  ~EventDispatcher();


  bool ProcessEvents();
  

  void Register(UserTable* pTab);
  

  void Unregister(UserTable* pTab);
  

  inline size_t GetSize() const
  {
    return m_size;
  }
  

  inline struct pollfd* GetPollData()
  {
    return m_pPoll;
  }
  
  void Rebuild();
  

  inline void Clear()
  {
    m_maps.clear();
  }
  
private:
  int m_iPipeFd;    
  int m_iMgmtFd;    
  Inotify* m_pIn;   
  InotifyWatch* m_pSys;   
  InotifyWatch* m_pUser;  
  FDUT_MAP m_maps;  
  size_t m_size;    
  struct pollfd* m_pPoll; 
  
  void ProcessMgmtEvents();
};



class UserTable
{
public:

  UserTable(EventDispatcher* pEd, const std::string& rUser, bool fSysTable);
  
  virtual ~UserTable();
  

  void Load();
  

  void Dispose();
  

  void OnEvent(InotifyEvent& rEvt);
  

  static void FinishDone();
  

  bool MayAccess(const std::string& rPath, bool fNoFollow) const;
  

  bool IsSystem() const;
  

  Inotify* GetInotify()
  {
    return &m_in;
  }
  

  inline static bool CheckUser(const char* user)
  {
    struct passwd* pw = getpwnam(user);
    if (pw == NULL)
      return false;
      
    return IncronTab::CheckUser(user);
  }
  

  void RunAsUser(char* const* argv) const;
  
private:
  Inotify m_in;           
  EventDispatcher* m_pEd; 
  std::string m_user;     
  bool m_fSysTable;       
  IncronTab m_tab;        
  IWCE_MAP m_map;         

  static PROC_MAP s_procMap;  
  

  IncronTabEntry* FindEntry(InotifyWatch* pWatch);
  

  bool PrepareArgs(const std::string& rCmd, int& argc, char**& argv);
  

  void CleanupArgs(int argc, char** argv);
  
};

#endif //_USERTABLE_H_