summaryrefslogtreecommitdiff
path: root/examples/IPC_SAP/SOCK_SAP/CPP-inserver-poll.cpp
blob: 9ba1fa02ad4086e9b8d4c901bc261692d3d47a35 (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
// IPC_SAP/poll server, which illustrates how to integrate the ACE
// $Id$

// socket wrappers with the SVR4 poll() system call to create a
// single-threaded concurrent server.

#include "ace/SOCK_Acceptor.h"                             
#include "ace/SOCK_Stream.h"
#include "ace/INET_Addr.h"                                       

#if defined (ACE_HAS_SVR4_POLL)

// Maximum per-process open I/O descriptors.
const int MAX_FDS = 200;

int
main (void)
{
  // Create a server end-point.
  ACE_INET_Addr addr (ACE_DEFAULT_SERVER_PORT);
  ACE_SOCK_Acceptor peer_acceptor (addr);
  ACE_SOCK_Stream new_stream;
  ACE_HANDLE s_handle = peer_acceptor.get_handle ();
  struct pollfd poll_array[MAX_FDS];

  for (int i = 0; i < MAX_FDS; i++)
    {
      poll_array[i].fd = ACE::INVALID_HANDLE;
      poll_array[i].events = POLLIN;
    }

  poll_array[0].fd = s_handle;

  for (int n_handles = 1;;) 
    {
      // Wait for client I/O events (handle interrupts).
      while (ACE_OS::poll (poll_array, n_handles) == -1
             && errno == EINTR)
        continue;

      // Handle pending logging messages first (s_handle + 1 is
      // guaranteed to be lowest client descriptor).

      for (i = 1; i < n_handles; i++)
	{
	  if (poll_array[i].revents & POLLIN) 
	    {
	      char buf[BUFSIZ];
	      int  n;
	      // recv will not block in this case!
	      if ((n = ACE_OS::recv (poll_array[i].fd, buf, sizeof buf, 0)) == -1)
		ACE_OS::perror ("read failed");
	      else if (n == 0) 
		{
		  // Handle client connection shutdown.
		  if (ACE_OS::close (poll_array[i].fd) == -1)
		    ACE_OS::perror ("close");
		  poll_array[i].fd = poll_array[--n_handles].fd;

		  // Send handshake back to client to unblock it.
		  if (ACE_OS::send (poll_array[i].fd, "", 1) != 1)
		    ACE_ERROR ((LM_ERROR, "%p\n", "send_n"));
		}
	      else 
		ACE_OS::printf ("%*s", n, buf), fflush (stdout);
	    } 
	  ACE_OS::fflush (stdout);
	}
      if (poll_array[0].revents & POLLIN)
	{
	  ACE_INET_Addr client;
	  ACE_Time_Value nonblock (0, 0);

	  // Handle all pending connection requests (note use of
	  // "polling" feature that doesn't block).

	  while (ACE_OS::poll (poll_array, 1, nonblock) > 0)
	    if (peer_acceptor.accept (new_stream, &client) == -1)
	      ACE_OS::perror ("accept");
	    else 
	      {
		const char *s = client.get_host_name ();

		ACE_ASSERT (s != 0);
		ACE_OS::printf ("client %s\n", s);
		ACE_OS::fflush (stdout);
		poll_array[n_handles++].fd = new_stream.get_handle ();
	      }
	}
    }
  /* NOTREACHED */
  return 0;
}
#else
#include <stdio.h>
int main (void)
{
  ACE_OS::fprintf (stderr, "This feature is not supported\n");
  return 0;
}
#endif /* ACE_HAS_SVR4_POLL */