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

流媒体/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 281421,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 RtspTcpConnection_cxx_version =
  51.     "$Id: RtspTcpConnection.cxx,v 1.20 2001/06/12 22:39:13 kle Exp $";
  52. const int KEEP_ALIVE_COUNTER = 10;
  53. #include "RtspTcpConnection.hxx"
  54. #include <sys/types.h>
  55. #include <sys/wait.h>
  56. RtspTcpConnection::TcpConnectionMap RtspTcpConnection::myTcpConnectionMap;
  57. RtspTcpConnection::RtspTcpConnection( Fifo< Sptr<RtspMsg> >& recvFifo, int listenPort )
  58.     : myTcpStack( listenPort ),
  59.       myMsgFifo( recvFifo ),
  60.       myShutdown( false )
  61. {
  62.     // setup signal handlers for TCP connections
  63.     signal( SIGTERM, RtspTcpConnection::sigTerm );
  64.     //signal( SIGINT, RtspTcpConnection::sigTerm );
  65.     //signal( SIGHUP, RtspTcpConnection::sigTerm );
  66.     //signal( SIGQUIT, RtspTcpConnection::sigTerm );
  67.     myRecvTcpThread.spawn( recvTcpThreadWrapper, this );
  68. }
  69. RtspTcpConnection::~RtspTcpConnection()
  70. {
  71.     cpLog( LOG_DEBUG, "~RtspTcpConnection" );
  72.     destory();
  73. }
  74. void
  75. RtspTcpConnection::destory()
  76. {
  77.     // close server tcp connection
  78.     if( myTcpStack.getServerConn().isLive() )
  79.     {
  80.         closeTcpConnection();
  81.     }
  82.     // close all tcp connections
  83.     while( !myTcpConnectionMap.empty() )
  84.     {
  85.         Sptr<RtspTcpBuffer> tcpBufferObj = (*myTcpConnectionMap.begin()).second;
  86.         tcpBufferObj->closeConnection();
  87.         myTcpConnectionMap.erase( tcpBufferObj->getMapIndex() );
  88.     }
  89. }
  90. void
  91. RtspTcpConnection::sigTerm( int signo )
  92. {
  93.     cpLog( LOG_ALERT, "Signal term caught" );
  94.     //destory();
  95. }
  96. void
  97. RtspTcpConnection::closeTcpConnection()
  98. {
  99.     cpLog( LOG_DEBUG, "Closing server tcp connection" );
  100.     // stop recvTcpThread
  101.     myShutdownMutex.lock();
  102.     myShutdown = true;
  103.     myShutdownMutex.unlock();
  104.     // wait until receive thread completed
  105.     myRecvTcpThread.join();
  106.     // close server tcp connection
  107.     myTcpStack.close();
  108. }
  109. void*
  110. RtspTcpConnection::recvTcpThreadWrapper( void* p )
  111. {
  112.     return static_cast<RtspTcpConnection*>(p)->recvTcpThread();
  113. }
  114. void*
  115. RtspTcpConnection::recvTcpThread()
  116. {
  117.     cpLog( LOG_DEBUG, "RtspTcpConnection thread is running" );
  118.     int maxFd;
  119.     fd_set baseReadFd;
  120.     struct timeval timeout;
  121.     int selectResult;
  122.     FD_ZERO( &baseReadFd );
  123.     FD_SET( myTcpStack.getServerConn().getConnId(), &baseReadFd );
  124.     maxFd = myTcpStack.getServerConn().getConnId();
  125.     // process all incoming TCP messages
  126.     while( 1 )
  127.     {
  128.         // setup select parameters
  129.         fd_set readFd = baseReadFd;
  130.         timeout.tv_sec = 3;
  131.         timeout.tv_usec = 0;
  132.         
  133.         selectResult = select( maxFd+1, &readFd, 0, 0, &timeout );
  134.         if( cpLogGetPriority() >= LOG_DEBUG_HB )
  135.         {
  136.             cerr<<"tcp";
  137.         }
  138.         if( selectResult < 0 )
  139.         {
  140.             cpLog( LOG_DEBUG, "select() error in recvTcpThread" );
  141.         }
  142.         else if( selectResult == 0 )
  143.         {
  144.             // check if stop accepting new network connections
  145.             myShutdownMutex.lock();
  146.             bool shutdownNow = myShutdown;
  147.             myShutdownMutex.unlock();
  148.             if( shutdownNow )
  149.             {
  150.                 break;
  151.             }
  152.             else
  153.             {
  154.                 continue;
  155.             }
  156.         }
  157.         // check for new connections
  158.         if( FD_ISSET( myTcpStack.getServerConn().getConnId(), &readFd ) )
  159.         {
  160.             // accept new connection
  161.             bool successAccept = true;
  162.             Connection conn;
  163.             try
  164.             {
  165.                 myTcpStack.accept( conn );
  166.             }
  167.             catch( VException& e )
  168.             {
  169.                 cpLog( LOG_DEBUG, e.getDescription().c_str() );
  170.                 successAccept = false;
  171.             }
  172.             if( successAccept )
  173.             {
  174.                 // create new session
  175.                 Sptr<RtspTcpBuffer> newTcpBufferObj= new RtspTcpBuffer( conn,
  176.                                     KEEP_ALIVE_COUNTER, myMsgFifo );
  177.                 // added new session to myTcpConnectionMap
  178.                 cpLog( LOG_DEBUG, "Adding %s to myTcpConnectionMap",
  179.                        newTcpBufferObj->getDescription().c_str() );
  180.                 myTcpConnectionMap[ newTcpBufferObj->getMapIndex() ] =
  181.                                     newTcpBufferObj;
  182.                 // add to select map
  183.                 int fd = conn.getConnId();
  184.                 FD_SET( fd, &baseReadFd );
  185.                 if( fd > maxFd )  maxFd = fd;
  186.             }
  187.         }
  188.         // process some connections
  189.         for( TcpConnectionMap::iterator itr = myTcpConnectionMap.begin();
  190.              itr != myTcpConnectionMap.end(); itr++ )
  191.         {
  192.             Sptr<RtspTcpBuffer> tcpBufferObj = (*itr).second;
  193.             assert(tcpBufferObj != 0);
  194.             Connection conn = tcpBufferObj->getConnection();
  195.             if( FD_ISSET( conn.getConnId(), &readFd ) )
  196.             {
  197.                 if( !tcpBufferObj->processConnection( tcpBufferObj ) )
  198.                 {
  199.                     // connection failure, delete from map
  200.                     cpLog( LOG_DEBUG,
  201.                            "Tcp connnection closed, deleting %s",
  202.                            tcpBufferObj->getDescription().c_str() );
  203.                     // delete from select map
  204.                     FD_CLR( conn.getConnId(), &baseReadFd );
  205.                     conn.close();
  206.                     myTcpConnectionMap.erase( tcpBufferObj->getMapIndex() );
  207.                 }
  208.             }
  209.         }
  210.     }
  211.     myRecvTcpThread.exit();
  212.     return 0;
  213. }
  214. /* Local Variables: */
  215. /* c-file-style: "stroustrup" */
  216. /* indent-tabs-mode: nil */
  217. /* c-file-offsets: ((access-label . -) (inclass . ++)) */
  218. /* c-basic-offset: 4 */
  219. /* End: */