anytun: threadUtils.hpp Source File

00001 /*
-00002  *  anytun
-00003  *
-00004  *  The secure anycast tunneling protocol (satp) defines a protocol used
-00005  *  for communication between any combination of unicast and anycast
-00006  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
-00007  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
-00008  *  ethernet, ip, arp ...). satp directly includes cryptography and
-00009  *  message authentication based on the methodes used by SRTP.  It is
-00010  *  intended to deliver a generic, scaleable and secure solution for
-00011  *  tunneling and relaying of packets of any protocol.
-00012  *
-00013  *
-00014  *  Copyright (C) 2007 anytun.org <satp@wirdorange.org>
-00015  *
-00016  *  This program is free software; you can redistribute it and/or modify
-00017  *  it under the terms of the GNU General Public License version 2
-00018  *  as published by the Free Software Foundation.
-00019  *
-00020  *  This program is distributed in the hope that it will be useful,
-00021  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
-00022  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-00023  *  GNU General Public License for more details.
-00024  *
-00025  *  You should have received a copy of the GNU General Public License
-00026  *  along with this program (see the file COPYING included with this
-00027  *  distribution); if not, write to the Free Software Foundation, Inc.,
-00028  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
-00029  */
-00030 
-00031 #ifndef _THREADUTILS_HPP_
-00032 #define _THREADUTILS_HPP_
-00033 
-00034 #include <stdexcept>
-00035 #include <semaphore.h>
-00036 
-00037 class Mutex 
-00038 {
-00039 public:
-00040   Mutex() 
-00041   { 
-00042     if(pthread_mutex_init(&mutex,NULL)) 
-00043       throw std::runtime_error("can't create mutex");
-00044   }
-00045 
-00046   ~Mutex()
-00047   {
-00048     pthread_mutex_destroy(&mutex);
-00049   }
-00050   
-00051 private:
-00052   Mutex(const Mutex& src);
-00053   void operator=(const Mutex& src);
-00054   
-00055   void lock()
-00056   {
-00057     if(pthread_mutex_lock(&mutex)) 
-00058       throw std::runtime_error("can't lock mutex");
-00059   }
-00060   
-00061   void unlock()
-00062   {
-00063     if(pthread_mutex_unlock(&mutex)) 
-00064       throw std::runtime_error("can't unlock mutex");
-00065   }
-00066   friend class Lock;
-00067   friend class Condition;
-00068   pthread_mutex_t mutex;
-00069 };
-00070 
-00071 
-00072 class Lock
-00073 {
-00074 public:
-00075   Lock(Mutex &m) : mutex(m)
-00076   {
-00077     mutex.lock();
-00078   }
-00079   
-00080   ~Lock()
-00081   {
-00082     mutex.unlock();
-00083   }
-00084 
-00085 private:
-00086   Lock(const Lock& src);
-00087   void operator=(const Lock& src);
-00088 
-00089   Mutex &mutex;
-00090 };
-00091 
-00092 class Condition
-00093 {
-00094 public:
-00095   Condition()
-00096   {
-00097     if(pthread_cond_init(&cond, NULL)) 
-00098       throw std::runtime_error("can't create condition");
-00099   }
-00100 
-00101   ~Condition()
-00102   {
-00103     pthread_cond_destroy(&cond);
-00104   }
-00105   
-00106   void wait()
-00107   {
-00108     mutex.lock();
-00109     if(pthread_cond_wait(&cond, &mutex.mutex)) 
-00110     {
-00111       mutex.unlock();
-00112       throw std::runtime_error("error on waiting for condition");
-00113     }
-00114     mutex.unlock();
-00115   }
-00116 
-00117   void signal()
-00118   {
-00119     mutex.lock();
-00120     if(pthread_cond_signal(&cond)) 
-00121     {
-00122       mutex.unlock();
-00123       throw std::runtime_error("can't signal condition");
-00124     }
-00125     mutex.unlock();
-00126   }
-00127 
-00128   void broadcast()
-00129   {
-00130     mutex.lock();
-00131     if(pthread_cond_broadcast(&cond)) 
-00132     {
-00133       mutex.unlock();
-00134       throw std::runtime_error("can't broadcast condition");
-00135     }
-00136     mutex.unlock();
-00137   }
-00138   
-00139 private:
-00140   pthread_cond_t cond;
-00141   Mutex mutex;
-00142 };
-00143 
-00144 class Semaphore
-00145 {
-00146 public:
-00147   Semaphore(unsigned int initVal=0)
-00148   {
-00149     if(sem_init(&sem, 0, initVal))
-00150       throw std::runtime_error("can't create semaphore");
-00151   }
-00152 
-00153   ~Semaphore()
-00154   {
-00155     sem_destroy(&sem);
-00156   }
-00157   
-00158   void down()
-00159   {
-00160     if(sem_wait(&sem)) 
-00161       throw std::runtime_error("error on semaphore down");
-00162   }
-00163 
-00164   void up()
-00165   {
-00166     if(sem_post(&sem)) 
-00167       throw std::runtime_error("error on semaphore up");
-00168   }
-00169 
-00170 private:
-00171   sem_t sem;
-00172 };
-00173 
-00174 #endif
-