summaryrefslogtreecommitdiff
path: root/TAO/CIAO/tests/RTCCM/DiffServ/Controller/Controller.cpp
blob: 197444ec43df03242b2e93809a619eda2f6aa49b (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
// $Id$

#include "SenderC.h"
#include "ReceiverC.h"
#include "ace/Get_Opt.h"

static const char *sender_ior = "file://sender.ior";
static const char *receiver_ior = "file://receiver.ior";
static int iterations = 1;
static int corba_priority = 0;

int
parse_args (int argc, char *argv[])
{
  ACE_Get_Opt get_opts (argc, argv, "s:r:n:p:");
  int c;

  while ((c = get_opts ()) != -1)
    switch (c)
      {
        // sender ior
      case 's':
        sender_ior = get_opts.opt_arg ();
        break;

        // receiver ior
      case 'r':
        receiver_ior = get_opts.opt_arg ();
        break;

        // number of itarations
      case 'n':
        iterations = ACE_OS::atoi (get_opts.opt_arg ());
        break;

        // corba priority
      case 'p':
        corba_priority = ACE_OS::atoi (get_opts.opt_arg ());
        break;

      default:
        ACE_ERROR_RETURN ((LM_ERROR,
                           "usage:  %s\n"
                           "\t-s <sender ior> [default is %s]\n"
                           "\t-r <receiver ior> [default is %s]\n"
                           "\t-n <number of invocations> [defaults to %d]\n"
                           "\t-p <corba priority> [defaults to %d]\n"
                           "\n",
                           argv [0],
                           sender_ior,
                           receiver_ior,
                           iterations,
                           corba_priority),
                          -1);
      }

  return 0;
}

int
main (int argc, char *argv[])
{
  try
    {
      CORBA::ORB_var orb =
        CORBA::ORB_init (argc, argv, "");

      int result =
        parse_args (argc, argv);
      if (result != 0)
        return result;

      CORBA::Object_var object =
        orb->string_to_object (sender_ior);

      DiffServ::Sender_var sender =
        DiffServ::Sender::_narrow (object.in ());

      object =
        orb->string_to_object (receiver_ior);

      DiffServ::Receiver_var receiver =
        DiffServ::Receiver::_narrow (object.in ());

      sender->start (iterations,
                     corba_priority);

      sender->shutdown ();

      receiver->shutdown ();

      return 0;
    }
  catch (CORBA::Exception &exception)
    {
      ACE_ERROR ((LM_ERROR,
                  "Unexpected exception caught by client: %s (%s)\n",
                  exception._name (),
                  exception._rep_id ()));

      return -1;
    }
}