NetworkAddress.cxx
上传用户:sy_wanhua
上传日期:2013-07-25
资源大小:3048k
文件大小:13k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

C/C++

  1. /* ====================================================================
  2.  * The Vovida Software License, Version 1.0 
  3.  * 
  4.  * Copyright (c) 2000 Vovida Networks, Inc.  All rights reserved.
  5.  * 
  6.  * Redistribution and use in source and binary forms, with or without
  7.  * modification, are permitted provided that the following conditions
  8.  * are met:
  9.  * 
  10.  * 1. Redistributions of source code must retain the above copyright
  11.  *    notice, this list of conditions and the following disclaimer.
  12.  * 
  13.  * 2. Redistributions in binary form must reproduce the above copyright
  14.  *    notice, this list of conditions and the following disclaimer in
  15.  *    the documentation and/or other materials provided with the
  16.  *    distribution.
  17.  * 
  18.  * 3. The names "VOCAL", "Vovida Open Communication Application Library",
  19.  *    and "Vovida Open Communication Application Library (VOCAL)" must
  20.  *    not be used to endorse or promote products derived from this
  21.  *    software without prior written permission. For written
  22.  *    permission, please contact vocal@vovida.org.
  23.  *
  24.  * 4. Products derived from this software may not be called "VOCAL", nor
  25.  *    may "VOCAL" appear in their name, without prior written
  26.  *    permission of Vovida Networks, Inc.
  27.  * 
  28.  * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
  29.  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  30.  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
  31.  * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL VOVIDA
  32.  * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
  33.  * IN EXCESS OF $1,000, NOR FOR ANY INDIRECT, INCIDENTAL, SPECIAL,
  34.  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  35.  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  36.  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  37.  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  38.  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
  39.  * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
  40.  * DAMAGE.
  41.  * 
  42.  * ====================================================================
  43.  * 
  44.  * This software consists of voluntary contributions made by Vovida
  45.  * Networks, Inc. and many individuals on behalf of Vovida Networks,
  46.  * Inc.  For more information on Vovida Networks, Inc., please see
  47.  * <http://www.vovida.org/>.
  48.  *
  49.  */
  50. static const char* const NetworkAddress_cxx_Version =
  51.     "$Id: NetworkAddress.cxx,v 1.8 2001/06/27 01:43:47 bko Exp $";
  52. #include <string>
  53. #if defined(__FreeBSD__)
  54. #include <sys/types.h>
  55. #include "netdb_r.h"
  56. #endif
  57. #include "errno.h"
  58. #include <netinet/in.h>
  59. #include <arpa/inet.h>
  60. #ifndef WIN32
  61. #include <strstream.h>
  62. #else
  63. #include <strstream>
  64. #endif
  65. #include <netdb.h>
  66. #include "NetworkAddress.h"
  67. #include "cpLog.h"
  68. #include "vsock.hxx"
  69. #include <iostream>
  70. #ifdef WIN32
  71. const int NetworkAddress::getHostLookupFailed = 1;
  72. const int NetworkAddress::getHostLookupOK = 0;
  73. #endif
  74. NetworkAddress::NetworkAddress ( int port /* = -1 */) :
  75.    aPort(port),
  76.    rawHostName("localhost"),
  77.    ipAddressSet(false)
  78. {
  79. }
  80. NetworkAddress::NetworkAddress ( const string& hostname, int port /* = -1 */) :
  81.    aPort(port),
  82.    rawHostName(hostname),
  83.    ipAddressSet(false)
  84. {
  85.    string::size_type pos = hostname.find(":");
  86.    rawHostName = hostname.substr(0,pos);
  87.    if (pos != string::npos)
  88.    {
  89.       aPort = atoi(hostname.substr(pos+1, string::npos).c_str());
  90.    }
  91.    //cpLog (LOG_INFO, "Address[%s]", rawHostName.c_str());
  92.    //cpLog (LOG_INFO, "Port[%d]", aPort);
  93.    
  94.    ipAddressSet = false;
  95. }
  96. void
  97. NetworkAddress::setHostName ( const string& hostname )
  98. {
  99.    string::size_type pos = hostname.find(":");
  100.    rawHostName = hostname.substr(0,pos);
  101.    if (pos != string::npos)
  102.    {
  103.       aPort = atoi(hostname.substr(pos+1, string::npos).c_str());
  104.    }
  105.    //cpLog (LOG_INFO, "Address[%s]", rawHostName.c_str());
  106.    //cpLog (LOG_INFO, "Port[%d]", aPort);
  107.    
  108.    ipAddressSet = false;
  109. }
  110. void
  111. NetworkAddress::setPort( int iPort )
  112. {
  113.     ipAddressSet = false;
  114.     aPort = iPort;
  115. }
  116. string
  117. NetworkAddress::getHostName( ) const
  118. {
  119.    cpLog(LOG_DEBUG_STACK, "NetworkAddress::getHostName()");
  120.    initIpAddress();
  121.    return hostName;
  122. }
  123. string
  124. NetworkAddress::getIpName () const
  125. {
  126.     initIpAddress();
  127.     char hostname[1024];
  128.     string ipName = inet_ntop(AF_INET, &ipAddress, hostname, sizeof(hostname));
  129.     return ipName;
  130. }
  131. u_int32_t
  132. NetworkAddress::getIp4Address () const
  133. {
  134.     u_int32_t lTmp;
  135.     initIpAddress();
  136.     memcpy((void *) &lTmp, ipAddress, IPV4_LENGTH);
  137.     return lTmp;
  138. }
  139. void
  140. NetworkAddress::getSockAddr (struct sockaddr & socka) const
  141. {
  142.     short tmp_s = htons (aPort);
  143.     char * tmp_p;
  144.     tmp_p = (char*) &tmp_s;
  145.     socka.sa_family = AF_INET;
  146.     socka.sa_data[0] = tmp_p[0];
  147.     socka.sa_data[1] = tmp_p[1];
  148.     initIpAddress();
  149.     memcpy((void *)&socka.sa_data[2], ipAddress, IPV4_LENGTH);
  150.     return ;
  151. }
  152. int
  153. NetworkAddress::getPort () const
  154. {
  155.     return aPort;
  156. }
  157. bool
  158. operator < ( const NetworkAddress & xAddress,
  159.              const NetworkAddress & yAddress )
  160. {
  161.     xAddress.initIpAddress();
  162.     yAddress.initIpAddress();
  163.     for (int i = 0 ; i <= IPV4_LENGTH; i++ )
  164.     {
  165.         if ( xAddress.ipAddress[i] < yAddress.ipAddress[i])
  166.         {
  167.             return true;
  168.         }
  169.         if ( xAddress.ipAddress[i] > yAddress.ipAddress[i])
  170.         {
  171.             return false;
  172.         }
  173.     }
  174.     return xAddress.aPort < yAddress.aPort ;
  175. }
  176. bool
  177. operator == ( const NetworkAddress & xAddress,
  178.               const NetworkAddress & yAddress )
  179. {
  180.     xAddress.initIpAddress();
  181.     yAddress.initIpAddress();
  182.     for (int i = 0 ; i < IPV4_LENGTH ; i++ )
  183.     {
  184.         if ( xAddress.ipAddress[i] != yAddress.ipAddress[i])
  185.         {
  186.             return false;
  187.         }
  188.     }
  189.     return xAddress.aPort == yAddress.aPort ;
  190. }
  191. bool
  192. operator != ( const NetworkAddress & xAddress,
  193.               const NetworkAddress & yAddress )
  194. {
  195.     return !(xAddress == yAddress);
  196. }
  197. NetworkAddress& 
  198. NetworkAddress::operator=( const NetworkAddress& x )
  199. {
  200.     this->aPort = x.aPort;
  201.     memcpy( this->ipAddress, x.ipAddress, sizeof(this->ipAddress) );
  202.     ipAddressSet = x.ipAddressSet;
  203.     rawHostName = x.rawHostName;
  204.     hostName = x.hostName;
  205.     
  206.     //   strncpy( this->ipAddress, x.ipAddress, IPV4_LENGTH );
  207.     return ( *this );
  208. }
  209. NetworkAddress::operator string ()
  210. {
  211.     string xStr;
  212.     strstream xStream;
  213.     xStream << (*this) << ends;
  214.     xStr = xStream.str();
  215.     xStream.freeze(false);
  216.     return xStr;
  217. }
  218. ostream &
  219. NetworkAddress::print ( ostream & xStr ) const
  220. {
  221.    initIpAddress();
  222.    char buffer[1024];
  223.    const char* cpBuf=inet_ntop(AF_INET, &ipAddress, buffer, sizeof(buffer));
  224.    xStr << hostName << "(" << cpBuf << ")";
  225.    
  226.    if ( aPort )
  227.    {
  228.       xStr << ":" << aPort;
  229.    }
  230.    return xStr;
  231. }
  232. u_int32_t
  233. NetworkAddress::hashIpPort( )
  234. {
  235.     u_int32_t ipAddressLocal = getIp4Address ();
  236.     int port = getPort ();
  237.     return (hashIpPort( ipAddressLocal, port )) ;
  238. }
  239. u_int32_t
  240. NetworkAddress::hashIpPort( const string& hostname, const string& port )
  241. {
  242.     NetworkAddress net;
  243.     net.setHostName ( hostname );
  244.     net.setPort ( atoi(port.c_str()) );
  245.     
  246.     return ( net.hashIpPort() );
  247. }
  248. u_int32_t
  249. NetworkAddress::hashIpPort( const u_int32_t ipAddress, const int port )
  250. {
  251.     u_int32_t hashKey = 0x0;
  252.     hashKey = (ipAddress << 16) | (port & 0xFF);
  253.     return hashKey;
  254. }
  255. void
  256. NetworkAddress::initIpAddress() const
  257. {
  258.     if(!ipAddressSet)
  259.     {
  260.         if (true == is_valid_ip_addr(rawHostName) )
  261.         {
  262.            hostName = rawHostName;
  263.            ipAddressSet = true;
  264.         }
  265.         else
  266.         {
  267.            struct hostent hostbuf, *hp;
  268.            size_t hstbuflen;
  269.            char *tmphstbuf;
  270.            int res=0;
  271.            int herr;
  272.            
  273.           
  274. #if defined(__linux__)
  275.            hstbuflen = 1024;
  276.            tmphstbuf = (char*)malloc (hstbuflen);
  277.            while ((res = gethostbyname_r (rawHostName.c_str(), &hostbuf, tmphstbuf, hstbuflen, &hp, &herr)) == ERANGE)
  278.            {
  279.               /* Enlarge the buffer.  */
  280.               hstbuflen *= 2;
  281.               tmphstbuf = (char*)realloc (tmphstbuf, hstbuflen);
  282.            }
  283. #else
  284.            hstbuflen = 4024;
  285.            tmphstbuf = (char*)malloc (hstbuflen);
  286.            hp = gethostbyname_r (rawHostName.c_str(), &hostbuf, tmphstbuf, hstbuflen, &herr);
  287. #endif
  288.            
  289.            /*  Check for errors.  */
  290.            if (res == 0 && hp)
  291.            {
  292.               memcpy((void *)&ipAddress,(void *)hp->h_addr,hp->h_length);
  293.               hostName = hp->h_name;
  294.               ipAddressSet = true;
  295.               free(tmphstbuf);
  296.            }
  297.            else
  298.            {
  299.               cpLog (LOG_ERR, "Could not resolve Address for (%s)", rawHostName.c_str());
  300.            }
  301.         }
  302.     }
  303. }
  304. NetworkAddress::NetworkAddress( const NetworkAddress& x)
  305. {
  306.     aPort = x.aPort;
  307.     ipAddressSet = x.ipAddressSet;
  308.     rawHostName = x.rawHostName;
  309.     hostName = x.hostName;
  310.     memcpy( ipAddress, x.ipAddress, sizeof(ipAddress) );
  311. }
  312. int
  313. NetworkAddress::getHostByName(const char* hostName,
  314.                               struct hostent* hEnt,
  315.                               char* buf,
  316.                               int buflen,
  317.                               int* thrErrno)
  318. {
  319. #if defined(__linux__)
  320.     struct hostent* x;
  321.     int retval = gethostbyname_r(hostName, hEnt, buf, buflen, &x, thrErrno);
  322.     if (retval == 0)
  323.         return getHostLookupOK;
  324.     cpLog(LOG_DEBUG, "********* Failed to get host by name:%s", hostName);
  325.     return getHostLookupFailed;
  326. #else
  327. #if defined(__svr4__) || defined(__FreeBSD__)|| defined (__SUNPRO_CC)
  328.     struct hostent* x;
  329.     x = gethostbyname_r(hostName, hEnt, buf, buflen, thrErrno);
  330.     if (x == 0)
  331.     {
  332.         cpLog(LOG_DEBUG, "********* Failed to get host by name:%s", hostName);
  333.         return getHostLookupFailed;
  334.     }
  335.     return getHostLookupOK;
  336. #else
  337. #if defined (WIN32)
  338.     struct hostent* x;
  339.     
  340.     x = gethostbyname(hostName);
  341.     if (x == 0) 
  342.     {
  343. cpLog(LOG_DEBUG, "********* Failed to get host by name:%s", hostName); 
  344. return getHostLookupFailed;
  345.     }
  346.     *hEnt = *x;
  347.     return getHostLookupOK;
  348.     
  349. #else
  350. #error no implementation for critical function
  351. #endif
  352. #endif
  353. #endif
  354. }
  355. #if defined (WIN32)
  356. struct hostent FAR *_gethostbyname( const char FAR *hostName )
  357. {
  358.     struct hostent* x;
  359. #undef gethostbyname
  360.     x = gethostbyname(hostName);
  361.     if (x == 0) 
  362.     {
  363.         int address = inet_addr(hostName);
  364.         if( address != 0xFFFFFFFF )
  365.         {
  366.             // Tried gethostbyaddr, but it did not seem to work...
  367. //          x = gethostbyaddr((char *)&address, IPV4_LENGTH, AF_INET);
  368.     if (x == 0) 
  369.             {
  370.                 struct hostentex
  371.                     : hostent 
  372.                 {
  373.                   char FAR *       hex_aliases;
  374.                   char             hex_alias;
  375.                   char FAR *       hex_addr_list;
  376.                   char             hex_addr[IPV4_LENGTH];
  377.                   char             hex_name[IPV4_LENGTH * 4];
  378.                 };
  379.                 static struct hostentex entex;
  380.                 entex.h_name        = entex.hex_name;
  381.                 entex.h_aliases     = & entex.hex_aliases;
  382.                 entex.h_addrtype    = AF_INET;
  383.                 entex.h_length      = IPV4_LENGTH;
  384.                 entex.h_addr_list   = & entex.hex_addr_list;
  385.                 entex.hex_aliases   = & entex.hex_alias;
  386.                 entex.hex_alias     = '';
  387.                 entex.hex_addr_list = & entex.hex_addr[0];
  388.                 memcpy((void *)entex.hex_addr, (void*)& address, IPV4_LENGTH);
  389.                 strcpy(entex.hex_name, hostName);
  390.                 x = & entex;
  391.             }
  392.         }
  393.     }
  394.     return x;
  395. }
  396. #endif
  397. /**
  398.   * Return TRUE if the address is a valid IP address (dot-decimal form)
  399.   * such as 128.128.128.128. Checks for invalid or ill formed addresses 
  400.   * If the address is valid, copy it into ipAddress
  401.   */
  402. bool
  403. NetworkAddress::is_valid_ip_addr(const string& addr) const
  404. {
  405. #ifdef WIN32
  406. typedef ULONG ulong;
  407. #endif
  408.     unsigned long maskcheck = ~255; // mask to check if 'tween 0 and 255
  409.     const char *advancer = addr.c_str();
  410.     unsigned long octet;
  411.     char *nextchar;
  412.     // always check for spaces and right number
  413.     // first and last fields must be 1-255, middle two 0-255
  414.     
  415.     if ((*(advancer) == 0) || (*(advancer) == ' ') || (*(advancer) == 't'))
  416.     {
  417.         return false;
  418.     }
  419.     octet = strtoul(advancer, &nextchar, 10);
  420.     if((*nextchar != '.') || (octet & maskcheck) || (octet == 0))
  421.     {
  422.         return false;
  423.     }
  424.     ipAddress[0] = octet;
  425.     advancer = nextchar+1;
  426.     if ((*(advancer) == 0) || (*(advancer) == ' ') || (*(advancer) == 't'))
  427.     {
  428.         return false;
  429.     }
  430.     octet = strtoul(advancer, &nextchar, 10);
  431.     if((*nextchar != '.') || (octet & maskcheck))
  432.     {
  433.         return false;
  434.     }
  435.     ipAddress[1] = octet;
  436.     advancer = nextchar+1;
  437.     if ((*(advancer) == 0) || (*(advancer) == ' ') || (*(advancer) == 't'))
  438.     {
  439.         return false;
  440.     }
  441.     octet = strtoul(advancer, &nextchar, 10);
  442.     if((*nextchar != '.') || (octet & maskcheck))
  443.     {
  444.         return false;
  445.     }
  446.     ipAddress[2] = octet;
  447.     
  448.     advancer = nextchar+1;
  449.     if ((*(advancer) == 0) || (*(advancer) == ' ') || (*(advancer) == 't'))
  450.     {
  451.         return false;
  452.     }
  453.     octet = strtoul(advancer, &nextchar, 10);
  454.     if((*nextchar) || (octet & maskcheck) || (octet == 0))
  455.     {
  456.         return false;
  457.     }
  458.     ipAddress[3] = octet;
  459.     
  460.     return true;
  461. }
  462. // End of File