Logo Search packages:      
Sourcecode: tcpreen version File versions  Download package

sockprot.h

/*
 * sockprot.h - C++ header for sockprot.cpp
 * socket addresses handling
 * $Id: sockprot.h,v 1.2 2004/06/05 15:15:17 rdenisc Exp $
 */

/***********************************************************************
 *  Copyright (C) 2002-2003 Remi Denis-Courmont.                       *
 *  This program is free software; you can redistribute and/or modify  *
 *  it under the terms of the GNU General Public License as published  *
 *  by the Free Software Foundation; version 2 of the license.         *
 *                                                                     *
 *  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, you can get it from:              *
 *  http://www.gnu.org/copyleft/gpl.html                               *
 ***********************************************************************/

#ifndef __TCPREEN_SOCKPROT_H
# define __TCPREEN_SOCKPROT_H

# include <stddef.h>
# include <sys/types.h>
# if HAVE_SYS_SOCKET_H
#  include <sys/socket.h> // SOCK_STREAM, PF_UNSPEC
# endif
# include "getaddrinfo.h" // struct addrinfo

# ifndef AF_UNSPEC
#  define AF_UNSPEC 0
#  define PF_UNSPEC 0
# endif

class SocketAddress
{
      private:
            int err_ai, err_sys;
            char myhost[1024], myserv[128];
            struct addrinfo *res;

            int SetFromSocket (int fd, int flags, int side);

      public:
            SocketAddress (void) : err_ai (0), err_sys (0), res (NULL)
            {
            }

            int GetSockName (int fd, int flags = 0)
            {
                  return SetFromSocket (fd, flags, 0);
            }

            int GetPeerName (int fd, int flags = 0)
            {
                  return SetFromSocket (fd, flags, 1);
            }
            /* Creates a socket address from a bound socket. */

            int SetByName (const char *host, const char *service,
                        int flags = 0, int af = AF_UNSPEC,
                        int type = SOCK_STREAM, int proto = 0);
            /* Resolves a socket address. */

            SocketAddress& operator= (const SocketAddress& src);
            SocketAddress (const SocketAddress& src);
            /* Copy constructor. */

            ~SocketAddress (void);

            int SetError (int ai = EAI_SYSTEM);
            void ClearError (void)
            {
                  SetError (0);
            }

            int GetError (void) const
            {
                  return err_ai;
            }

            int Bind (void);
            int Connect (int nonblock = 0);
            void CleanUp (void) const;

            int operator! (void) const
            {
                  return err_ai;
            }

            const char *StrError (void) const;
            const char *Node (void) const
            {
                  return myhost;
            }
            const char *Service (void) const
            {
                  return myserv;
            }
};

#endif

Generated by  Doxygen 1.6.0   Back to index