summaryrefslogtreecommitdiff
path: root/TAO/orbsvcs/tests/AVStreams/sfp/client.cpp
blob: aed0cf49330a2875ffac2616fb027e556e0ffef0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
// $Id$

#include "common.h"
#include "ace/Event_Handler.h"
#include "ace/Get_Opt.h"

static char *data;
static char *addr;
int fragment = 0;

class frame_handler: public ACE_Event_Handler
{
  public:

  frame_handler (TAO_SFP *sfp,
                 CORBA_ORB_ptr orb,
                 ACE_Reactor *reactor)
    :sfp_ (sfp),
     orb_ (orb)
    {
      reactor->schedule_timer (this,
                               0,
                               10);
    }

  // Called when timer expires.  
  int handle_timeout (const ACE_Time_Value &/* tv */,
                      const void */* arg */)
    {
      //       ACE_DEBUG ((LM_DEBUG,"frame_handler:handle_timeout\n"));
       if (fragment)
         {
           size_t mb_len = SFP_MAX_PACKET_SIZE*2;
           ACE_DEBUG ((LM_DEBUG,"message block size = %d\n",mb_len));
           ACE_Message_Block mb (mb_len);
           int i =0;
           char c = 'a';
           for (char *ptr = mb.rd_ptr ();i<2;i++,c++,ptr+=SFP_MAX_PACKET_SIZE)
             ACE_OS::memset (ptr,c,SFP_MAX_PACKET_SIZE);
           
           mb.wr_ptr (mb_len);
           int result = this->sfp_->send_frame (&mb);
           if (result != -1)
             ACE_DEBUG ((LM_DEBUG,"simple_frame sent\n"));
           else
             ACE_DEBUG ((LM_DEBUG,"simple_frame send failed\n"));
         }
       else
         {
           ACE_Message_Block mb (data,
                                 ACE_OS::strlen (data));
           mb.wr_ptr (ACE_OS::strlen (data));
           int result = this->sfp_->send_frame (&mb);
           if (result != -1)
             ACE_DEBUG ((LM_DEBUG,"simple_frame sent\n"));
           else
             ACE_DEBUG ((LM_DEBUG,"simple_frame send failed\n"));
         }
       this->sfp_->end_stream ();
       this->orb_->shutdown ();
       return 0;
    }

private:
  TAO_SFP *sfp_;
  CORBA_ORB_ptr orb_;
};

int
parse_args (int argc,char **argv)
{
  ACE_Get_Opt opts (argc,argv,"fa:");
  int c;
  while ((c = opts ()) != -1)
    {	
      switch (c)
        {
        case 'f':
          fragment = 1;
          break;
        case 'a':
          addr = ACE_OS::strdup (opts.optarg);
          break;
        default:
          ACE_ERROR_RETURN ((LM_ERROR,"Usage:client [-f] [-a addr]"),-1); 
        }	
    }
  return 0;	
}


int
main (int argc, char **argv)
{
  server_addr = ACE_OS::strdup ("localhost:10000");
  addr = server_addr;
  data = ACE_OS::strdup ("Hello how are you");
  TAO_ORB_Manager orb_manager;
  
  orb_manager.init (argc,argv);

  int result = parse_args (argc,argv);    
  if (result < 0)
    return 1;
  ACE_Time_Value timeout1 (5),timeout2 (50);

  TAO_SFP sfp (orb_manager.orb (),
           TAO_ORB_Core_instance ()->reactor (),
           timeout1,
           timeout2,
           0);

  result = sfp.start_stream (addr);
  if (result != 0)
    ACE_ERROR_RETURN ((LM_ERROR,"sfp start failed\n"),1);

  frame_handler handler (&sfp,
                         orb_manager.orb (),
                         TAO_ORB_Core_instance ()->reactor ());

  result = orb_manager.run ();
//   if (result == 0)
//     ACE_DEBUG ((LM_DEBUG,"ORB run timed out\n"));
//   else if (result == -1)
//     ACE_DEBUG ((LM_DEBUG,"ORB run error\n"));
  return 0;
}