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
|
// This tests the non-blocking features of the ACE_SOCK_Connector class.
// @(#)CPP-inclient.cpp 1.1 10/18/96
#include "ace/Log_Msg.h"
#include "ace/SOCK_Connector.h"
#include "ace/INET_Addr.h"
// ACE SOCK_SAP client.
int main (int argc, char *argv[])
{
char *host = argc > 1 ? argv[1] : ACE_DEFAULT_SERVER_HOST;
u_short r_port = argc > 2 ? ACE_OS::atoi (argv[2]) : ACE_DEFAULT_SERVER_PORT;
int timeout = argc > 3 ? ACE_OS::atoi (argv[3]) : ACE_DEFAULT_TIMEOUT;
u_short l_port = argc > 4 ? ACE_OS::atoi (argv[4]) : ACE_DEFAULT_LOCAL_PORT;
char buf[BUFSIZ];
ACE_SOCK_Stream cli_stream;
ACE_INET_Addr remote_addr (r_port, host);
ACE_INET_Addr local_addr (l_port);
ACE_DEBUG ((LM_DEBUG, "starting non-blocking connect\n"));
// Initiate timed, non-blocking connection with server.
ACE_SOCK_Connector con;
// Attempt a non-blocking connect to the server, reusing the local
// addr if necessary.
if (con.connect (cli_stream, remote_addr,
(ACE_Time_Value *) &ACE_Time_Value::zero,
local_addr, 1) == -1)
{
if (errno != EWOULDBLOCK)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "connection failed"), 1);
ACE_DEBUG ((LM_DEBUG, "starting timed connect\n"));
// Check if non-blocking connection is in progress,
// and wait up to timeout seconds for it to complete.
ACE_Time_Value tv (timeout);
if (con.complete (cli_stream, &remote_addr, &tv) == -1)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "connection failed"), 1);
else
ACE_DEBUG ((LM_DEBUG, "connected to %s\n", remote_addr.get_host_name ()));
}
if (cli_stream.disable (ACE_NONBLOCK) == -1)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "disable"), 1);
// Send data to server (correctly handles "incomplete writes").
for (ssize_t r_bytes;
(r_bytes = ACE_OS::read (ACE_STDIN, buf, sizeof buf)) > 0; )
if (ACE_OS::strcmp (buf, "quit\n") == 0)
break;
else if (cli_stream.send_n (buf, r_bytes) == -1)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "send_n"), 1);
// Explicitly close the writer-side of the connection.
if (cli_stream.close_writer () == -1)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "close_writer"), 1);
// Wait for handshake with server.
if (cli_stream.recv_n (buf, 1) != 1)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "recv_n"), 1);
// Close the connection completely.
if (cli_stream.close () == -1)
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "close"), 1);
return 0;
}
|