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

流媒体/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 ,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 testLoadBalance_cxx_Version =
  51.     "$Id: testDnsResolver.cxx,v 1.2 2001/05/15 15:51:21 wbreu Exp $";
  52. /*
  53.  * testLoadBalance -- test prog to retrieve the DNS SRV records, sort   
  54.  *              the records, and select one of the records based on
  55.  *              the preference (prioirty) and weight
  56.  *
  57.  */
  58. // #include <cstdlib> 
  59. #include <stdlib.h>
  60. #include <stdio.h>
  61. #include <unistd.h>
  62. #ifdef __linux__
  63. #include <getopt.h>
  64. #endif
  65. #include "DnsResolver.hxx"
  66. extern char *optarg;
  67. extern int optind;
  68. /* local function prototypes */
  69. void usage(void);
  70. /* Max number of name server to be checked: */
  71. const int MAX_NS = 20;
  72. char *progname;
  73. void
  74. usage( void )
  75. {
  76.     fprintf( stderr, "usage: %s domainn", progname );
  77.     exit(2);
  78. }
  79. void do_test0( DnsResolver& srv )
  80. {
  81.     // test 0:
  82.     cout << "nTest case 0:n" << endl;
  83.     // provide function to dynamically reorder the srv records
  84.     // based on random selection given a range of the srv
  85.     // records in the same priority
  86.     if (srv.isReady())
  87.        srv.reOrder();
  88. }   // do_test0()
  89. // single record:- passed
  90. void do_test1( DnsResolver& srv )
  91. {
  92.    srv.clear();
  93.    srv.addSrvRecord(2, 2, 5060, "vanilla.private.vovida.com");
  94.    srv.sortByPreference();
  95.    srv.reOrder();
  96. }   // do_test1()
  97. // unique records:
  98. void do_test2( DnsResolver& srv )
  99. {
  100.    srv.clear();
  101.    srv.addSrvRecord(2, 2, 5060, "vanilla.private.vovida.com");
  102.    srv.addSrvRecord(3, 0, 5060, "chocolate.private.vovida.com");
  103.    srv.addSrvRecord(1, 2, 5060, "strawberry.private.vovida.com");
  104.    srv.sortByPreference();
  105.    srv.reOrder();
  106. }   // do_test2()
  107. // 
  108. void do_test3( DnsResolver& srv )
  109. {
  110.    cout << "nTest Case 3:n" << endl;
  111.    srv.clear();
  112.    srv.addSrvRecord(5, 20, 5060, "vanilla.private.vovida.com");
  113.    srv.addSrvRecord(2, 256, 5060, "maple.private.vovida.com");
  114.    srv.addSrvRecord(2, 128, 5060, "blueberry.private.vovida.com");
  115.    srv.addSrvRecord(2, 64, 5060, "bubblegum.private.vovida.com");
  116.    srv.addSrvRecord(2, 32, 5060, "mango.private.vovida.com");
  117.    srv.addSrvRecord(2, 32, 5060, "neopolitan.private.vovida.com");
  118.    srv.addSrvRecord(3, 5, 5060, "chocolate.private.vovida.com");
  119.    srv.addSrvRecord(1, 5, 5060, "strawberry.private.vovida.com");
  120.    srv.sortByPreference();
  121.    srv.reOrder();
  122.    srv.reOrder();
  123.    srv.reOrder();
  124. }  // do_test3()
  125. // 
  126. void do_test4( DnsResolver& srv )
  127. {
  128.    cout << "nTest Case 4:n" << endl;
  129.    srv.clear();
  130.    srv.addSrvRecord(1, 8, 5060, "vanilla.private.vovida.com");
  131.    srv.addSrvRecord(1, 256, 5060, "coffee.private.vovida.com");
  132.    srv.addSrvRecord(1, 128, 5060, "blueberry.private.vovida.com");
  133.    srv.addSrvRecord(1, 64, 5060, "bubblegum.private.vovida.com");
  134.    srv.addSrvRecord(1, 32, 5060, "mango.private.vovida.com");
  135.    srv.addSrvRecord(1, 32, 5060, "neopolitan.private.vovida.com");
  136.    srv.addSrvRecord(1, 16, 5060, "chocolate.private.vovida.com");
  137.    srv.addSrvRecord(1, 8, 5060, "strawberry.private.vovida.com");
  138.    srv.sortByPreference();
  139.    for (int i = 0; i <= 9999; i++)
  140.    {
  141.        srv.reOrder();
  142.        if ((i%50) == 0)
  143.    usleep(10);
  144.    }
  145.    srv.printTally();
  146. }  // do_test4()
  147. // main body
  148. main(int argc, char* argv[])
  149. {
  150.     progname = *argv;
  151.     if (argc < 2)
  152.     {
  153.         usage();
  154.     }
  155. cpLogSetPriority( LOG_DEBUG );
  156. cpLogSetLabel("dns");
  157. // test A record retrieval: e.g. www.yahoo.com
  158. // cpLog(LOG_DEBUG, "Testing buildARecords()");
  159.     // DnsResolver AResolver( argv[1], ns_t_a, NULL, NULL );
  160. // test SRV record retrieval: e.g. _sip._udp.cisco.com
  161. cpLog(LOG_DEBUG, "nnTesting buildSRVRecords()");
  162.     const char* st = 0;
  163.     const char* proto = 0;
  164.     if (argc >= 4)
  165.     {
  166.         st = argv[2];
  167.         proto = argv[3];
  168.     } 
  169.     DnsResolver srv( argv[1], ns_t_srv, st, proto );
  170.  
  171.     cpLog( LOG_DEBUG, "nnReorder the list again:");
  172.     int num = srv.getNumDnsRecords();
  173.     for (int j=0; j<3; j++)
  174.     {  
  175.         for (int i=0; i<num; i++)
  176.         {
  177.            Sptr<DnsRecord>* rec = srv.getRecord( i );
  178.            cpLog( LOG_DEBUG, "Getting record #%d, target=%s",
  179.                   i,(*rec)->getTarget() );
  180.         }
  181.         cpLog( LOG_DEBUG, "********* Reordering the SRV list");
  182.         srv.reOrder();
  183.        
  184.     }
  185.     // test 0:
  186.     // do_test0(srv);
  187.     // do_test1(srv);
  188.     // do_test2(srv);
  189.     //
  190.     // do_test3(srv);
  191.     //
  192.     // do_test4(srv);
  193.     exit( 0 );
  194. }   // end of main()