summaryrefslogtreecommitdiff
path: root/examples/IOStream/server/iostream_server.cpp
blob: 82b46cd0928c484e81d59b4654dacb64a07c1f6d (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
128
129
130
131
132
133
134
135
136
137
138
139
// This is a simple example of using the ACE_IOStream and
// ACE_streambuf templates to create an object based on ACE_*_Stream
// classes which mimic a C++ iostream.

#include "ace/Acceptor.h"
#include "ace/SOCK_Acceptor.h"
#include "ace/Service_Config.h"
#include "ace/IOStream.h"

// Declare a new type which will case an ACE_SOCK_Stream to behave
// like an iostream.  The new type (ACE_SOCK_IOStream) can be used
// anywhere an ACE_SOCK_Stream is used.

typedef ACE_IOStream<ACE_SOCK_Stream> ACE_SOCK_IOStream ;

// Create a service handler object based on our new
// iostream/SOCK_Stream hybrid.

typedef ACE_Svc_Handler<ACE_SOCK_IOStream, ACE_INET_Addr, ACE_NULL_SYNCH>       
	Service_Handler;

class Handler : public Service_Handler 
  // = TITLE
  //     Extend the <Service_Handler> object to do our bidding.  All of
  //     this is fairly standard until we get to the <handle_input>
  //     where we begin using the iostream characteristics of the
  //     peer.
{
public:
  Handler (void) {}
		
  virtual int open (void *)
  {
    if (this->reactor () ->register_handler 
	(this, ACE_Event_Handler::READ_MASK) == - 1) 
      ACE_ERROR_RETURN ((LM_ERROR, 
			 "registering connection handler with ACE_Reactor\n"), 
			- 1);

    return 0;
  }
		
  virtual void destroy (void)
  {
    this->peer ().close ();
    delete this ;
  }
		
  virtual int close (u_long)
  {
    this->destroy ();
    return 0 ;
  }
		
  virtual int handle_input (ACE_HANDLE)
  {
    int i;
    float f;

#if defined (__GNUC__) 
    String s;
		
    if (!(this -> peer () >> i >> f >> s))
      {
	cerr << "Error getting data" << endl ;
	return - 1 ;
      }

    cerr << "Received (" << i << ") (" << f << ") (" << s << ")" << endl ;

    if (!(this -> peer () << "Received: " << i << " " << f << " " << s << endl))
      {
	cerr << __LINE__ << "Error sending data" << endl ;
	return - 1 ;
      }
#else
    if (!(this -> peer () >> i >> f))
      {
	cerr << "Error getting data" << endl ;
	return - 1 ;
      }

    cerr << "Received (" << i << ") (" << f << ")" << endl;

    if (!(this -> peer () << i << " " << f << endl))
      {
	cerr << __LINE__ << "Error sending data" << endl ;
	return - 1 ;
      }
#endif /* __GNUC__ */
    // In order to flush the output to the peer, we have to use the
    // sync () function.  Some iostreams implementations let us use a
    // 'flush' function much like the 'endl' function.

    // this->peer ().sync ();
    return 0;
  }
};

// Create an object which will accept new connection requests and
// create handler objects for us.

typedef ACE_Acceptor<Handler, ACE_SOCK_ACCEPTOR> Logging_Acceptor;

int 
main (int argc, char *argv [])
{
  ACE_Service_Config daemon;

  // Create an adapter to end the event loop.
  ACE_Sig_Adapter sa ((ACE_Sig_Handler_Ex) ACE_Service_Config::end_reactor_event_loop);

  ACE_Sig_Set sig_set;
  sig_set.sig_add (SIGINT);
  sig_set.sig_add (SIGQUIT);

  // Register ourselves to receive SIGINT and SIGQUIT so we can shut
  // down gracefully via signals.
  if (ACE_Service_Config::reactor ()->register_handler (sig_set, &sa) == -1)
    ACE_ERROR_RETURN ((LM_ERROR, "%p\n"), -1);

  Logging_Acceptor peer_acceptor ;
	
  if (peer_acceptor.open (ACE_INET_Addr (argc > 1 ? atoi (argv[1]) : ACE_DEFAULT_SERVER_PORT)) == - 1)
    ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "open"), - 1);

  else if (ACE_Service_Config::reactor ()->register_handler 
	   (&peer_acceptor, ACE_Event_Handler::READ_MASK) == - 1) 
    ACE_ERROR_RETURN ((LM_ERROR, "registering service with ACE_Reactor\n"), - 1);
	
  ACE_DEBUG ((LM_DEBUG, " (%P|%t) starting up daemon\n"));
	
  daemon.run_reactor_event_loop ();

  ACE_DEBUG ((LM_DEBUG, " (%P|%t) shutting down server daemon\n"));

  return 0;
}