clientctx.h

Go to the documentation of this file.

#ifndef CVMFS_CLIENTCTX_H_
#define CVMFS_CLIENTCTX_H_

#include <pthread.h>
#include <unistd.h>

#include <cassert>
#include <vector>

class InterruptCue;

class ClientCtx {
 public:
  struct ThreadLocalStorage {
    ThreadLocalStorage(uid_t u, gid_t g, pid_t p, InterruptCue *ic)
      : uid(u), gid(g), pid(p), interrupt_cue(ic), is_set(true) { }
    uid_t uid;
    gid_t gid;
    pid_t pid;
    InterruptCue *interrupt_cue;  
    bool is_set;  
  };

  static ClientCtx *GetInstance();
  static void CleanupInstance();
  ~ClientCtx();

  void Set(uid_t uid, gid_t gid, pid_t pid, InterruptCue *ic);
  void Unset();
  void Get(uid_t *uid, gid_t *gid, pid_t *pid, InterruptCue **ic);
  bool IsSet();

 private:
  static ClientCtx *instance_;
  static void TlsDestructor(void *data);

  ClientCtx();

  pthread_key_t thread_local_storage_;
  pthread_mutex_t *lock_tls_blocks_;
  std::vector<ThreadLocalStorage *> tls_blocks_;
};

class ClientCtxGuard {
 public:
  ClientCtxGuard(uid_t uid, gid_t gid, pid_t pid, InterruptCue *ic)
    : set_on_construction_(false)
    , old_uid_(-1)
    , old_gid_(-1)
    , old_pid_(-1)
    , old_interrupt_cue_(NULL)
  {
    // Implementation guarantees old_ctx is not null.
    ClientCtx *old_ctx = ClientCtx::GetInstance();
    assert(old_ctx);
    if (old_ctx->IsSet()) {
      set_on_construction_ = true;
      old_ctx->Get(&old_uid_, &old_gid_, &old_pid_, &old_interrupt_cue_);
    }
    old_ctx->Set(uid, gid, pid, ic);
  }

  ~ClientCtxGuard() {
    ClientCtx *ctx = ClientCtx::GetInstance();
    if (set_on_construction_) {
      ctx->Set(old_uid_, old_gid_, old_pid_, old_interrupt_cue_);
    } else {
      ctx->Unset();
    }
  }

 private:
  bool set_on_construction_;
  uid_t old_uid_;
  gid_t old_gid_;
  pid_t old_pid_;
  InterruptCue *old_interrupt_cue_;
};

#endif  // CVMFS_CLIENTCTX_H_