// server.cpp,v 1.17 2002/07/28 21:28:16 crodrigu Exp
#include "server.h"
FTP_Server_FlowEndPoint::FTP_Server_FlowEndPoint (void)
:TAO_FlowConsumer ("Data",FTP_SERVER::instance ()->protocols (), FTP_SERVER::instance ()->format ())
{
AVStreams::protocolSpec protocols (2);
protocols.length (3);
protocols [0] = CORBA::string_dup ("TCP");
protocols [1] = CORBA::string_dup ("UDP");
protocols [2] = CORBA::string_dup ("RTP/UDP");
ACE_DECLARE_NEW_CORBA_ENV;
ACE_TRY
{
this->set_protocol_restriction (protocols
ACE_ENV_ARG_PARAMETER);
ACE_TRY_CHECK;
}
ACE_CATCHANY
{
ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,"FTP_Server_FlowEndPoint::FTP_Server_FlowEndPoint\n");
}
ACE_ENDTRY;
ACE_CHECK;
}
int
FTP_Server_FlowEndPoint::get_callback (const char *,
TAO_AV_Callback *&callback)
{
ACE_NEW_RETURN (callback,
FTP_Server_Callback,
-1);
return 0;
}
FTP_Server_Callback::FTP_Server_Callback (void)
{
}
int
FTP_Server_Callback::handle_stop (void)
{
ACE_DEBUG ((LM_DEBUG,"FTP_Server_Callback::stop\n"));
ACE_OS::fclose (FTP_SERVER::instance ()->file ());
TAO_AV_CORE::instance ()->orb ()->shutdown ();
return 0;
}
int
FTP_Server_Callback::receive_frame (ACE_Message_Block *frame,
TAO_AV_frame_info *,
const ACE_Addr &)
{
ACE_DEBUG ((LM_DEBUG,"FTP_Server_Callback::receive_frame\n"));
while (frame != 0)
{
int result = ACE_OS::fwrite (frame->rd_ptr (),
frame->length (),
1,
FTP_SERVER::instance ()->file ());
if (result == 0)
ACE_ERROR_RETURN ((LM_ERROR,"FTP_Server_Flow_Handler::fwrite failed\n"),-1);
frame = frame->cont ();
}
return 0;
}
int
FTP_Server_Callback::handle_end_stream (void)
{
CORBA::ORB_var orb = TAO_AV_CORE::instance ()->orb ();
orb->shutdown ();
return 0;
}
Server::Server (void)
:orb_ (TAO_AV_CORE::instance ()->orb ()),
poa_ (TAO_AV_CORE::instance ()->poa ())
{
reactive_strategy_.init (TAO_AV_CORE::instance ()->orb (),
TAO_AV_CORE::instance ()->poa ());
}
AVStreams::protocolSpec
Server::protocols (void)
{
AVStreams::protocolSpec protocols (2);
protocols.length (2);
protocols [0] = CORBA::string_dup ("TCP");
protocols [1] = CORBA::string_dup ("UDP");
return protocols;
}
const char*
Server::format (void)
{
return "UNS:ftp";
}
int
Server::init (int argc,
char **argv)
{
ACE_DECLARE_NEW_CORBA_ENV;
ACE_TRY
{
PortableServer::POAManager_var mgr
= this->poa_->the_POAManager ();
mgr->activate ();
int result = this->parse_args (argc,argv);
if (result == -1)
ACE_ERROR_RETURN ((LM_ERROR,"parse args failed\n"),-1);
// Initialize the naming services
if (my_naming_client_.init (this->orb_.in ()) != 0)
ACE_ERROR_RETURN ((LM_ERROR,
" (%P|%t) Unable to initialize "
"the TAO_Naming_Client. \n"),
-1);
ACE_NEW_RETURN (this->streamendpoint_b_, TAO_StreamEndPoint_B, -1);
ACE_NEW_RETURN (this->fep_b_, FTP_Server_FlowEndPoint, -1);
sep_b_ = this->streamendpoint_b_->_this (ACE_ENV_SINGLE_ARG_PARAMETER);
ACE_TRY_CHECK;
fep_b_obj_ = this->fep_b_->_this (ACE_ENV_SINGLE_ARG_PARAMETER);
ACE_TRY_CHECK;
CORBA::String_var s1 = sep_b_->add_fep( fep_b_obj_.in() ACE_ENV_ARG_PARAMETER );
ACE_TRY_CHECK;
ACE_DEBUG ((LM_DEBUG, "(%N,%l) Added flowendpoint named: %s\n", s1.in() ));
// Register the mmdevice with the naming service.
CosNaming::Name server_sep_b_name (1);
server_sep_b_name.length (1);
server_sep_b_name [0].id = CORBA::string_dup ("Server_StreamEndPoint_b");
// Register the video control object with the naming server.
this->my_naming_client_->rebind (server_sep_b_name,
sep_b_.in ()
ACE_ENV_ARG_PARAMETER);
ACE_TRY_CHECK;
}
ACE_CATCHANY
{
ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,"server::init");
return -1;
}
ACE_ENDTRY;
ACE_CHECK_RETURN (-1);
return 0;
}
int
Server::run (void)
{
ACE_DECLARE_NEW_CORBA_ENV;
ACE_TRY
{
this->orb_->run (ACE_ENV_SINGLE_ARG_PARAMETER);
ACE_TRY_CHECK;
}
ACE_CATCHANY
{
ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,"server::init\n");
return -1;
}
ACE_ENDTRY;
ACE_CHECK_RETURN (-1);
return 0;
}
int
Server::parse_args (int argc,char **argv)
{
ACE_Get_Opt opts (argc,argv,"f:p:");
int c;
while ((c = opts ()) != -1)
{
switch (c)
{
case 'f':
this->fp_ = ACE_OS::fopen (opts.opt_arg (),"w");
if (this->fp_ != 0)
{
ACE_DEBUG ((LM_DEBUG,"file opened successfully\n"));
}
break;
case 'p':
this->protocol_ = ACE_OS::strdup (opts.opt_arg ());
break;
default:
ACE_ERROR_RETURN ((LM_ERROR,"Usage: server -f filename\n"),-1);
}
}
if( ! this->fp_ )
{
ACE_ERROR_RETURN ((LM_ERROR, "Invalid file!\nUsage: server -f filename\n"),-1);
}
return 0;
}
FILE*
Server::file (void)
{
return this->fp_;
}
int
main (int argc,
char **argv)
{
int result = 0;
ACE_DECLARE_NEW_CORBA_ENV;
ACE_TRY
{
CORBA::ORB_var orb = CORBA::ORB_init (argc,
argv);
CORBA::Object_var obj
= orb->resolve_initial_references ("RootPOA" ACE_ENV_ARG_PARAMETER);
ACE_TRY_CHECK;
PortableServer::POA_var poa
= PortableServer::POA::_narrow (obj.in ());
TAO_AV_CORE::instance ()->init (orb.in (),
poa.in ()
ACE_ENV_ARG_PARAMETER);
ACE_TRY_CHECK;
}
ACE_CATCHANY
{
ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,"server::init");
return -1;
}
ACE_ENDTRY;
ACE_CHECK_RETURN (-1);
result = FTP_SERVER::instance ()->init (argc,argv);
if (result < 0)
ACE_ERROR_RETURN ((LM_ERROR,"SERVER::init failed\n"),1);
result = FTP_SERVER::instance ()->run ();
if (result < 0)
ACE_ERROR_RETURN ((LM_ERROR,"SERVER::run failed\n"),1);
return result;
}
#if defined (ACE_HAS_EXPLICIT_TEMPLATE_INSTANTIATION)
template class ACE_Singleton ;
template class TAO_AV_Endpoint_Reactive_Strategy_B ;
template class TAO_AV_Endpoint_Reactive_Strategy ;
#elif defined (ACE_HAS_TEMPLATE_INSTANTIATION_PRAGMA)
s#pragma instantiate ACE_Singleton
#pragma instantiate TAO_AV_Endpoint_Reactive_Strategy_B
#pragma instantiate TAO_AV_Endpoint_Reactive_Strategy
#endif /* ACE_HAS_EXPLICIT_TEMPLATE_INSTANTIATION */
