?? ssl-server.cpp
字號(hào):
// SSL-server.cpp,v 1.10 2003/11/26 14:52:09 jwillemsen Exp
// This example tests the features of the <ACE_SSL_SOCK_Acceptor>,
// <ACE_SSL_SOCK_Stream>, and <ACE_Svc_Handler> classes. If the platform
// supports threads it uses a thread-per-connection concurrency model.
// Otherwise, it uses a single-threaded iterative server model.
#include "ace/Thread_Manager.h"
#include "ace/Handle_Set.h"
#include "ace/Profile_Timer.h"
#include "ace/OS_NS_sys_select.h"
#include "ace/SSL/SSL_SOCK_Acceptor.h"
ACE_RCSID(SSL_SAP, SSL_server, "SSL-server.cpp,v 1.10 2003/11/26 14:52:09 jwillemsen Exp")
// Are we running verbosely?
static int verbose = 0;
static void
run_server (ACE_THR_FUNC server,
ACE_SSL_SOCK_Stream * new_stream)
{
#if defined (ACE_HAS_THREADS)
// Spawn a new thread and run the new connection in that thread of
// control using the <server> function as the entry point.
if (ACE_Thread_Manager::instance ()->spawn (server,
(void *) new_stream,
THR_DETACHED) == -1)
ACE_ERROR ((LM_ERROR,
"(%P|%t) %p\n",
"spawn"));
#else
(*server) ((void *) new_stream);
#endif /* ACE_HAS_THREADS */
}
// Function entry point into the twoway server task.
static ACE_THR_FUNC_RETURN
twoway_server (void *arg)
{
ACE_INET_Addr cli_addr;
ACE_SSL_SOCK_Stream * new_stream = (ACE_SSL_SOCK_Stream *) arg;
// Make sure we're not in non-blocking mode.
if (new_stream->disable (ACE_NONBLOCK) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
"%p\n",
"disable"),
0);
else if (new_stream->get_remote_addr (cli_addr) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
"%p\n",
"get_remote_addr"),
0);
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) client %s connected from %d\n",
cli_addr.get_host_name (),
cli_addr.get_port_number ()));
size_t total_bytes = 0;
size_t message_count = 0;
char *request = 0;
// Read data from client (terminate on error).
for (;;)
{
ACE_INT32 len;
ssize_t r_bytes = new_stream->recv_n ((void *) &len,
sizeof (ACE_INT32));
if (r_bytes == -1)
{
ACE_ERROR ((LM_ERROR,
"%p\n",
"recv"));
break;
}
else if (r_bytes == 0)
{
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) reached end of input, connection closed by client\n"));
break;
}
else if (r_bytes != sizeof (ACE_INT32))
{
ACE_ERROR ((LM_ERROR,
"(%P|%t) %p\n",
"recv_n failed"));
break;
}
else
{
len = ntohl (len);
ACE_NEW_RETURN (request,
char [len],
0);
}
// Subtract off the sizeof the length prefix.
r_bytes = new_stream->recv_n (request,
len - sizeof (ACE_UINT32));
if (r_bytes == -1)
{
ACE_ERROR ((LM_ERROR,
"%p\n",
"recv"));
break;
}
else if (r_bytes == 0)
{
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) reached end of input, "
"connection closed by client\n"));
break;
}
else if (verbose
&& ACE::write_n (ACE_STDOUT,
request,
r_bytes) != r_bytes)
ACE_ERROR ((LM_ERROR,
"%p\n",
"ACE::write_n"));
else if (new_stream->send_n (request,
r_bytes) != r_bytes)
ACE_ERROR ((LM_ERROR,
"%p\n",
"send_n"));
total_bytes += size_t (r_bytes);
message_count++;
delete [] request;
request = 0;
}
// Close new endpoint (listening endpoint stays open).
new_stream->close ();
delete new_stream;
delete [] request;
return 0;
}
// Function entry point into the oneway server task.
static ACE_THR_FUNC_RETURN
oneway_server (void *arg)
{
ACE_INET_Addr cli_addr;
ACE_SSL_SOCK_Stream * new_stream = (ACE_SSL_SOCK_Stream *) arg;
// Make sure we're not in non-blocking mode.
if (new_stream->disable (ACE_NONBLOCK) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
"%p\n",
"disable"),
0);
else if (new_stream->get_remote_addr (cli_addr) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
"%p\n",
"get_remote_addr"),
0);
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) client %s connected from %d\n",
cli_addr.get_host_name (),
cli_addr.get_port_number ()));
// Timer business
ACE_Profile_Timer timer;
timer.start ();
size_t total_bytes = 0;
size_t message_count = 0;
char *request = 0;
// Read data from client (terminate on error).
for (;;)
{
ACE_INT32 len;
ssize_t r_bytes = new_stream->recv_n ((void *) &len,
sizeof (ACE_INT32));
if (r_bytes == -1)
{
ACE_ERROR ((LM_ERROR,
"%p\n",
"recv"));
break;
}
else if (r_bytes == 0)
{
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) reached end of input, connection closed by client\n"));
break;
}
else if (r_bytes != sizeof (ACE_INT32))
{
ACE_ERROR ((LM_ERROR,
"(%P|%t) %p\n",
"recv_n failed"));
break;
}
else
{
len = ntohl (len);
ACE_NEW_RETURN (request,
char [len],
0);
}
// Subtract off the sizeof the length prefix.
r_bytes = new_stream->recv_n (request,
len - sizeof (ACE_UINT32));
if (r_bytes == -1)
{
ACE_ERROR ((LM_ERROR,
"%p\n",
"recv"));
break;
}
else if (r_bytes == 0)
{
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) reached end of input, connection closed by client\n"));
break;
}
else if (verbose
&& ACE::write_n (ACE_STDOUT, request, r_bytes) != r_bytes)
ACE_ERROR ((LM_ERROR,
"%p\n",
"ACE::write_n"));
total_bytes += size_t (r_bytes);
message_count++;
delete [] request;
request = 0;
}
timer.stop ();
ACE_Profile_Timer::ACE_Elapsed_Time et;
timer.elapsed_time (et);
ACE_DEBUG ((LM_DEBUG,
ACE_TEXT ("\t\treal time = %f secs \n")
ACE_TEXT ("\t\tuser time = %f secs \n")
ACE_TEXT ("\t\tsystem time = %f secs\n"),
et.real_time,
et.user_time,
et.system_time));
double messages_per_sec = double (message_count) / et.real_time;
ACE_DEBUG ((LM_DEBUG,
ACE_TEXT ("\t\tmessages = %d\n")
ACE_TEXT ("\t\ttotal bytes = %d\n")
ACE_TEXT ("\t\tmbits/sec = %f\n")
ACE_TEXT ("\t\tusec-per-message = %f\n")
ACE_TEXT ("\t\tmessages-per-second = %0.00f\n"),
message_count,
total_bytes,
(((double) total_bytes * 8) / et.real_time) / (double) (1024 * 1024),
(et.real_time / (double) message_count) * 1000000,
messages_per_sec < 0 ? 0 : messages_per_sec));
// Close new endpoint (listening endpoint stays open).
new_stream->close ();
delete new_stream;
delete [] request;
return 0;
}
static int
run_event_loop (u_short port)
{
// Raise the socket handle limit to the maximum.
ACE::set_handle_limit ();
// Create the oneway and twoway acceptors.
ACE_SSL_SOCK_Acceptor twoway_acceptor;
ACE_SSL_SOCK_Acceptor oneway_acceptor;
// Create the oneway and twoway server addresses.
ACE_INET_Addr twoway_server_addr (port);
ACE_INET_Addr oneway_server_addr (port + 1);
// Create acceptors, reuse the address.
if (twoway_acceptor.open (twoway_server_addr, 1) == -1
|| oneway_acceptor.open (oneway_server_addr, 1) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
"%p\n",
"open"),
1);
else if (twoway_acceptor.get_local_addr (twoway_server_addr) == -1
|| oneway_acceptor.get_local_addr (oneway_server_addr) == -1)
ACE_ERROR_RETURN ((LM_ERROR,
"%p\n",
"get_local_addr"),
1);
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) starting twoway server at port %d and oneway server at port %d\n",
twoway_server_addr.get_port_number (),
oneway_server_addr.get_port_number ()));
// Keep these objects out here to prevent excessive constructor
// calls within the loop.
ACE_Handle_Set handle_set;
handle_set.set_bit (twoway_acceptor.get_handle ());
handle_set.set_bit (oneway_acceptor.get_handle ());
ACE_SSL_SOCK_Stream * new_stream = 0;
// Performs the iterative server activities.
for (;;)
{
ACE_Time_Value timeout (ACE_DEFAULT_TIMEOUT);
ACE_Handle_Set temp = handle_set;
int result = ACE_OS::select (int (oneway_acceptor.get_handle ()) + 1,
(fd_set *) temp,
0,
0,
timeout);
if (result == -1)
ACE_ERROR ((LM_ERROR,
"(%P|%t) %p\n",
"select"));
else if (result == 0 && verbose)
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) select timed out\n"));
else
{
// A new ACE_SSL_SOCK_Stream must be initialized for each
// connection. However, it retains (SSL) state so simply
// initializing a new ACE_SSL_SOCK_Stream by passing it a
// handle isn't enough, nor is it allowed. Such a scheme is
// is sometimes done with the non-SSL aware
// ACE_SOCK_Stream. An ACE_SSL_SOCK_Stream should only be
// initialized by an ACE_SSL_SOCK_Acceptor (server side), or an
// ACE_SSL_SOCK_Connector (client side).
//
// It is also possible to copy or assign an
// ACE_SSL_SOCK_Stream since it implements both
// methods/operators. However, the user must ensure that
// the copy or assignment is atomic.
if (temp.is_set (twoway_acceptor.get_handle ()))
{
ACE_NEW_RETURN (new_stream,
ACE_SSL_SOCK_Stream,
-1);
if (twoway_acceptor.accept (*new_stream) == -1)
{
ACE_ERROR ((LM_ERROR,
"%p\n",
"accept"));
delete new_stream;
continue;
}
else
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) spawning twoway server\n"));
// Run the twoway server.
run_server (twoway_server,
new_stream);
}
if (temp.is_set (oneway_acceptor.get_handle ()))
{
ACE_NEW_RETURN (new_stream,
ACE_SSL_SOCK_Stream,
-1);
if (oneway_acceptor.accept (*new_stream) == -1)
{
ACE_ERROR ((LM_ERROR, "%p\n", "accept"));
delete new_stream;
continue;
}
else
ACE_DEBUG ((LM_DEBUG,
"(%P|%t) spawning oneway server\n"));
// Run the oneway server.
run_server (oneway_server,
new_stream);
}
}
}
/* NOTREACHED */
}
int
main (int argc, char *argv[])
{
ACE_SSL_Context *context = ACE_SSL_Context::instance ();
context->certificate ("./dummy.pem", SSL_FILETYPE_PEM);
context->private_key ("./key.pem", SSL_FILETYPE_PEM);
u_short port = ACE_DEFAULT_SERVER_PORT;
if (argc > 1)
port = ACE_OS::atoi (argv[1]);
return run_event_loop (port);
}
?? 快捷鍵說(shuō)明
復(fù)制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號(hào)
Ctrl + =
減小字號(hào)
Ctrl + -