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 <pthread.h>
  36 #include <semaphore.h>
  37
  38 class Mutex
  39 {
  40 public:
  41   Mutex()
  42   {
  43     if(pthread_mutex_init(&mutex,NULL))
  44       throw std::runtime_error("can't create mutex");
  45   }
  46
  47   ~Mutex()
  48   {
  49     pthread_mutex_destroy(&mutex);
  50   }
  51
  52 private:
  53   Mutex(const Mutex& src);
  54   void operator=(const Mutex& src);
  55
  56   void lock()
  57   {
  58     if(pthread_mutex_lock(&mutex))
  59       throw std::runtime_error("can't lock mutex");
  60   }
  61
  62   void unlock()
  63   {
  64     if(pthread_mutex_unlock(&mutex))
  65       throw std::runtime_error("can't unlock mutex");
  66   }
  67   friend class Lock;
  68   friend class Condition;
  69   pthread_mutex_t mutex;
  70 };
  71
  72
  73 class Lock
  74 {
  75 public:
  76   Lock(Mutex &m) : mutex(m)
  77   {
  78     mutex.lock();
  79   }
  80
  81   ~Lock()
  82   {
  83     mutex.unlock();
  84   }
  85
  86 private:
  87   Lock(const Lock& src);
  88   void operator=(const Lock& src);
  89
  90   Mutex &mutex;
  91 };
  92
  93 class Condition
  94 {
  95 public:
  96   Condition()
  97   {
  98     if(pthread_cond_init(&cond, NULL))
  99       throw std::runtime_error("can't create condition");
 100   }
 101
 102   ~Condition()
 103   {
 104     pthread_cond_destroy(&cond);
 105   }
 106
 107   void wait()
 108   {
 109     mutex.lock();
 110     if(pthread_cond_wait(&cond, &mutex.mutex))
 111     {
 112       mutex.unlock();
 113       throw std::runtime_error("error on waiting for condition");
 114     }
 115     mutex.unlock();
 116   }
 117
 118   void signal()
 119   {
 120     mutex.lock();
 121     if(pthread_cond_signal(&cond))
 122     {
 123       mutex.unlock();
 124       throw std::runtime_error("can't signal condition");
 125     }
 126     mutex.unlock();
 127   }
 128
 129   void broadcast()
 130   {
 131     mutex.lock();
 132     if(pthread_cond_broadcast(&cond))
 133     {
 134       mutex.unlock();
 135       throw std::runtime_error("can't broadcast condition");
 136     }
 137     mutex.unlock();
 138   }
 139
 140 private:
 141   pthread_cond_t cond;
 142   Mutex mutex;
 143 };
 144
 145 class Semaphore
 146 {
 147 public:
 148   Semaphore(unsigned int initVal=0)
 149   {
 150     if(sem_init(&sem, 0, initVal))
 151       throw std::runtime_error("can't create semaphore");
 152   }
 153
 154   ~Semaphore()
 155   {
 156     sem_destroy(&sem);
 157   }
 158
 159   void down()
 160   {
 161     if(sem_wait(&sem))
 162       throw std::runtime_error("error on semaphore down");
 163   }
 164
 165   void up()
 166   {
 167     if(sem_post(&sem))
 168       throw std::runtime_error("error on semaphore up");
 169   }
 170
 171 private:
 172   sem_t sem;
 173 };
 174
 175 #endif