root/trunk/pdns/pdns/utility.hh @ 76

/*
    PowerDNS Versatile Database Driven Nameserver
    Copyright (C) 2002  PowerDNS.COM BV

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
// Utility class specification.

#ifndef UTILITY_HH
#define UTILITY_HH

#ifndef WIN32
# include "config.h"
#endif // WIN32

#ifdef NEED_POSIX_TYPEDEF
typedef unsigned short int u_int16_t;
typedef unsigned int u_int32_t;
#endif


#ifndef WIN32
# include <arpa/inet.h>
# include <netinet/in.h>
# include <sys/socket.h>
# include <sys/time.h>
# include <sys/uio.h>
# include <signal.h>
# include <pthread.h>
# include <semaphore.h>
# include <signal.h>
# include <errno.h>
#else
// Disable debug info truncation warning.
# pragma warning ( disable: 4786 )
# pragma warning ( disable: 4503 )
# pragma warning ( disable: 4101 )

# define WINDOWS_LEAN_AND_MEAN
# include <windows.h>
# include <signal.h>
# include <map>

// For scope fix.
# define for if ( false ) {} else for

# define ETIMEDOUT    WSAETIMEDOUT
# define EINPROGRESS  WSAEWOULDBLOCK

# define AF_INET6 -1

# define VERSION "2.9.2-WIN32"

# define snprintf _snprintf

// Custom bittypes.
typedef unsigned char int8_t;
typedef unsigned int  int16_t;
typedef unsigned long int32_t;
typedef unsigned char u_int8_t;
typedef unsigned int  u_int16_t;
typedef unsigned long u_int32_t;

struct in6_addr {
        unsigned char s6_addr[16]; /* IPv6 address */
};

struct sockaddr_in6 {
        unsigned short sin6_family; /* AF_INET6 */
        unsigned short sin6_port; /* transport layer port # */
        unsigned long sin6_flowinfo; /* IPv6 flow information */
        struct in6_addr sin6_addr; /* IPv6 address */
};

#endif // WIN32

#include <semaphore.h>
#include <string>

using namespace std;


//! A semaphore class.
class Semaphore
{
private:
  sem_t *m_pSemaphore;
#ifdef WIN32
  typedef int sem_value_t;

  //! The semaphore.



  //! Semaphore counter.
  long m_counter;

#else
  typedef int sem_value_t;

  u_int32_t       m_magic;
  pthread_mutex_t m_lock;
  pthread_cond_t  m_gtzero;
  sem_value_t     m_count;
  u_int32_t       m_nwaiters;
#endif

protected:
public:
  //! Default constructor.
  Semaphore( unsigned int value = 0 );

  //! Destructor.
  ~Semaphore( void );
  
  //! Posts to a semaphore.
  int post( void );

  //! Waits for a semaphore.
  int wait( void );

  //! Tries to wait for a semaphore.
  int tryWait( void );
  
  //! Retrieves the semaphore value.
  int getValue( Semaphore::sem_value_t *sval );
  
};


//! This is a utility class used for platform independant abstraction.
class Utility
{
#ifdef WIN32
private:
  static int inet_pton4( const char *src, void *dst );
  static int inet_pton6( const char *src, void *dst );

  static const char *inet_ntop4( const char *src, char *dst, size_t size );
  static const char *inet_ntop6( const char *src, char *dst, size_t size );

#endif // WIN32

public:
#ifdef WIN32

  //! iovec structure for windows.
  typedef struct 
  {
    void  *iov_base;  //!< Base address.
    size_t iov_len;   //!< Number of bytes.
  } iovec;

  // A few type defines.
  typedef DWORD     pid_t;
  typedef SOCKET    sock_t;
  typedef int       socklen_t;
  
#else
  typedef ::iovec iovec;
  typedef ::pid_t     pid_t;
  typedef int       sock_t;
  typedef ::socklen_t        socklen_t;
  
#endif // WIN32

  //! Closes a socket.
  static int closesocket( sock_t socket );

  //! Returns the process id of the current process.
  static pid_t getpid( void );

  //! Gets the current time.
  static int gettimeofday( struct timeval *tv, void *tz = NULL );

  //! Converts an address from dot and numbers format to binary data.
  static int inet_aton( const char *cp, struct in_addr *inp );

  //! Converts an address from presentation format to network format.
  static int inet_pton( int af, const char *src, void *dst );

  //! The inet_ntop() function converts an address from network format (usually a struct in_addr or some other binary form, in network byte order) to presentation format.
  static const char *inet_ntop( int af, const char *src, char *dst, size_t size );

  //! Retrieves a gid using a groupname.
  static int makeGidNumeric( const string & group );
  
  //! Retrieves an uid using an username.
  static int makeUidNumeric( const string & username );

  //! Writes a vector.
  static int writev( Utility::sock_t socket, const iovec *vector, size_t count );
  //! Returns a random number.
  static long int random( void );

  //! Sets the random seed.
  static void srandom( unsigned int seed );

  //! Compares two strings and ignores case.
  static int strcasecmp( const char *s1, const char *s2 );

  //! Drops the program's privileges.
  static void dropPrivs( int uid, int gid );
  
  //! Sets the socket into blocking mode.
  static bool setBlocking( Utility::sock_t socket );

  //! Sets the socket into non-blocking mode.
  static bool setNonBlocking( Utility::sock_t socket );
  
  //! Sleeps for a number of seconds.
  static unsigned int sleep( unsigned int seconds );
  
  //! Sleeps for a number of microseconds.
  static void usleep( unsigned long usec );
  
};


#endif // UTILITY_HH