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
|
// $Id$
#include "ace/OS_main.h"
#include "ace/OS_NS_string.h"
#include "ace/OS_NS_sys_socket.h"
#include "ace/OS_NS_unistd.h"
#include "ace/OS_NS_stdlib.h"
#include "ace/OS_NS_stdio.h"
#include "ace/OS_NS_netdb.h"
#include "ace/Default_Constants.h"
ACE_RCSID(SOCK_SAP, C_inclient, "$Id$")
/* BSD socket client */
int
ACE_TMAIN (int argc, ACE_TCHAR *argv[])
{
// Initialize WinSock DLL on Win32...
ACE_OS::socket_init (ACE_WSOCK_VERSION);
struct sockaddr_in saddr;
struct hostent *hp;
const ACE_TCHAR *host = argc > 1 ? argv[1] : ACE_DEFAULT_SERVER_HOST;
u_short port_num =
htons (argc > 2 ? ACE_OS::atoi (argv[2]) : ACE_DEFAULT_SERVER_PORT);
int sockbufsize = argc > 3 ? ACE_OS::atoi (argv[3]) : 0;
char buf[BUFSIZ];
ACE_HANDLE s_handle;
int w_bytes;
int r_bytes;
int n;
// Create a local endpoint of communication.
if ((s_handle = ACE_OS::socket (PF_INET, SOCK_STREAM, 0)) == ACE_INVALID_HANDLE)
ACE_OS::perror (ACE_TEXT("socket")), ACE_OS::exit (1);
// If a sockbufsize was specified, set it for both send and receive.
if (sockbufsize > 0)
{
if (ACE_OS::setsockopt (s_handle, SOL_SOCKET, SO_SNDBUF,
(const char *) &sockbufsize,
sizeof (sockbufsize)) != 0)
ACE_OS::perror (ACE_TEXT("SO_SNDBUF")), ACE_OS::exit (1);
if (ACE_OS::setsockopt (s_handle, SOL_SOCKET, SO_RCVBUF,
(const char *) &sockbufsize,
sizeof (sockbufsize)) != 0)
ACE_OS::perror (ACE_TEXT("SO_RCVBUF")), ACE_OS::exit (1);
}
// Determine IP address of the server.
if ((hp = ACE_OS::gethostbyname (ACE_TEXT_TO_CHAR_IN(host))) == 0)
ACE_OS::perror (ACE_TEXT("gethostbyname")), ACE_OS::exit (1);
// Set up the address information to contact the server.
ACE_OS::memset ((void *) &saddr, 0, sizeof saddr);
saddr.sin_family = AF_INET;
saddr.sin_port = port_num;
ACE_OS::memcpy (&saddr.sin_addr, hp->h_addr, hp->h_length);
// Establish connection with remote server.
if (ACE_OS::connect (s_handle,
reinterpret_cast<sockaddr *> (&saddr),
sizeof saddr) == -1)
ACE_OS::perror (ACE_TEXT("connect")), ACE_OS::exit (1);
// Send data to server (correctly handles "incomplete writes" due to
// flow control).
while ((r_bytes = ACE_OS::read (ACE_STDIN, buf, sizeof buf)) > 0)
for (w_bytes = 0; w_bytes < r_bytes; w_bytes += n)
if ((n = ACE_OS::send (s_handle, buf + w_bytes,
r_bytes - w_bytes)) < 0)
ACE_OS::perror (ACE_TEXT("write")), ACE_OS::exit (1);
if (ACE_OS::recv (s_handle, buf, 1) == 1)
ACE_OS::write (ACE_STDOUT, buf, 1);
// Explicitly close the connection.
if (ACE_OS::closesocket (s_handle) == -1)
ACE_OS::perror (ACE_TEXT("close")), ACE_OS::exit (1);
return 0;
}
|
