threadUtils.hpp

   1 /*
   2  *  anytun
   3  *
   4  *  The secure anycast tunneling protocol (satp) defines a protocol used
   5  *  for communication between any combination of unicast and anycast
   6  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
   7  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
   8  *  ethernet, ip, arp ...). satp directly includes cryptography and
   9  *  message authentication based on the methodes used by SRTP.  It is
  10  *  intended to deliver a generic, scaleable and secure solution for
  11  *  tunneling and relaying of packets of any protocol.
  12  *
  13  *
  14  *  Copyright (C) 2007 anytun.org <satp@wirdorange.org>
  15  *
  16  *  This program is free software; you can redistribute it and/or modify
  17  *  it under the terms of the GNU General Public License version 2
  18  *  as published by the Free Software Foundation.
  19  *
  20  *  This program is distributed in the hope that it will be useful,
  21  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  22  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23  *  GNU General Public License for more details.
  24  *
  25  *  You should have received a copy of the GNU General Public License
  26  *  along with this program (see the file COPYING included with this
  27  *  distribution); if not, write to the Free Software Foundation, Inc.,
  28  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  29  */
  30
  31 #ifndef _THREADUTILS_HPP_
  32 #define _THREADUTILS_HPP_
  33
  34 #include <stdexcept>
  35 #include <semaphore.h>
  36
  37 class Mutex
  38 {
  39 public:
  40   Mutex()
  41   {
  42     if(pthread_mutex_init(&mutex,NULL))
  43       throw std::runtime_error("can't create mutex");
  44   }
  45
  46   ~Mutex()
  47   {
  48     pthread_mutex_destroy(&mutex);
  49   }
  50
  51 private:
  52   Mutex(const Mutex& src);
  53   void operator=(const Mutex& src);
  54
  55   void lock()
  56   {
  57     if(pthread_mutex_lock(&mutex))
  58       throw std::runtime_error("can't lock mutex");
  59   }
  60
  61   void unlock()
  62   {
  63     if(pthread_mutex_unlock(&mutex))
  64       throw std::runtime_error("can't unlock mutex");
  65   }
  66   friend class Lock;
  67   friend class Condition;
  68   pthread_mutex_t mutex;
  69 };
  70
  71
  72 class Lock
  73 {
  74 public:
  75   Lock(Mutex &m) : mutex(m)
  76   {
  77     mutex.lock();
  78   }
  79
  80   ~Lock()
  81   {
  82     mutex.unlock();
  83   }
  84
  85 private:
  86   Lock(const Lock& src);
  87   void operator=(const Lock& src);
  88
  89   Mutex &mutex;
  90 };
  91
  92 class Condition
  93 {
  94 public:
  95   Condition()
  96   {
  97     if(pthread_cond_init(&cond, NULL))
  98       throw std::runtime_error("can't create condition");
  99   }
 100
 101   ~Condition()
 102   {
 103     pthread_cond_destroy(&cond);
 104   }
 105
 106   void wait()
 107   {
 108     mutex.lock();
 109     if(pthread_cond_wait(&cond, &mutex.mutex))
 110     {
 111       mutex.unlock();
 112       throw std::runtime_error("error on waiting for condition");
 113     }
 114     mutex.unlock();
 115   }
 116
 117   void signal()
 118   {
 119     mutex.lock();
 120     if(pthread_cond_signal(&cond))
 121     {
 122       mutex.unlock();
 123       throw std::runtime_error("can't signal condition");
 124     }
 125     mutex.unlock();
 126   }
 127
 128   void broadcast()
 129   {
 130     mutex.lock();
 131     if(pthread_cond_broadcast(&cond))
 132     {
 133       mutex.unlock();
 134       throw std::runtime_error("can't broadcast condition");
 135     }
 136     mutex.unlock();
 137   }
 138
 139 private:
 140   pthread_cond_t cond;
 141   Mutex mutex;
 142 };
 143
 144 class Semaphore
 145 {
 146 public:
 147   Semaphore(unsigned int initVal=0)
 148   {
 149     if(sem_init(&sem, 0, initVal))
 150       throw std::runtime_error("can't create semaphore");
 151   }
 152
 153   ~Semaphore()
 154   {
 155     sem_destroy(&sem);
 156   }
 157
 158   void down()
 159   {
 160     if(sem_wait(&sem))
 161       throw std::runtime_error("error on semaphore down");
 162   }
 163
 164   void up()
 165   {
 166     if(sem_post(&sem))
 167       throw std::runtime_error("error on semaphore up");
 168   }
 169
 170 private:
 171   sem_t sem;
 172 };
 173
 174 #endif