bool
init_http_client() {
debug( "Initializaing HTTP client." );
assert( http_client_thread == NULL );
assert( http_client_efd == -1 );
assert( main_efd == -1 );
assert( transactions == NULL );
http_client_efd = create_event_fd();
assert( http_client_efd >= 0 );
http_client_notify_count = 0;
set_fd_handler( http_client_efd, NULL, NULL, notify_http_client_actually, &http_client_notify_count );
set_writable( http_client_efd, false );
main_efd = create_event_fd();
assert( main_efd >= 0 );
set_fd_handler( main_efd, retrieve_http_transactions_from_http_client, NULL, NULL, NULL );
set_readable( main_efd, true );
assert( main_to_http_client_queue == NULL );
assert( http_client_to_main_queue == NULL );
main_to_http_client_queue = create_queue();
assert( main_to_http_client_queue != NULL );
http_client_to_main_queue = create_queue();
assert( http_client_to_main_queue != NULL );
create_http_transaction_db();
debug( "Initialization completed." );
return true;
}
static bool
init_syslog_relay( int *argc, char **argv[] ) {
parse_options( argc, argv );
if ( syslog_fd >= 0 ) {
close( syslog_fd );
syslog_fd = -1;
}
syslog_fd = socket( PF_INET, SOCK_DGRAM, 0 );
if ( syslog_fd < 0 ) {
error( "Failed to create socket ( errno = %s [%d] ).", strerror( errno ), errno );
return false;
}
struct sockaddr_in addr;
memset( &addr, 0, sizeof( struct sockaddr_in ) );
addr.sin_family = AF_INET;
addr.sin_port = htons( listen_port );
addr.sin_addr.s_addr = htonl( INADDR_ANY );
int ret = bind( syslog_fd, ( struct sockaddr * ) &addr, sizeof( struct sockaddr_in ) );
if ( ret < 0 ) {
error( "Failed to bind socket ( errno = %s [%d] ).", strerror( errno ), errno );
close( syslog_fd );
syslog_fd = -1;
return false;
}
set_fd_handler( syslog_fd, recv_syslog_message, NULL, NULL, NULL );
set_readable( syslog_fd, true );
return true;
}
static bool
init_stdin_relay( int *argc, char **argv[] ) {
parse_options( argc, argv );
stdin_read_buffer = alloc_buffer_with_length( 1024 );
set_fd_handler( STDIN_FILENO, read_stdin, NULL, NULL, NULL );
set_readable( STDIN_FILENO, true );
return true;
}
static void
connected() {
transit_state( CONNECTED );
set_fd_handler( connection.fd, recv_from_secure_channel, NULL, flush_send_queue, NULL );
set_readable( connection.fd, true );
set_writable( connection.fd, false );
if ( connection.connected_callback != NULL ) {
connection.connected_callback();
}
}
int
main( int argc, char *argv[] ) {
bool ret;
const char *switch_daemon = NULL;
char *startup_dir;
// get startup directory using absolute_path()
startup_dir = get_current_dir_name();
if ( startup_dir == NULL ) {
die( "Failed to get_current_dir_name." );
}
init_trema( &argc, &argv ); // changes the current working directory
init_listener_info( &listener_info );
ret = parse_argument( &listener_info, argc, argv );
if ( !ret ) {
finalize_listener_info( &listener_info );
exit( EXIT_FAILURE );
}
init_dpid_table();
start_service_management();
start_switch_management();
switch_daemon = listener_info.switch_daemon;
listener_info.switch_daemon = absolute_path( startup_dir, switch_daemon );
xfree( ( void * ) ( uintptr_t ) switch_daemon );
// free returned buffer of get_current_dir_name()
free( startup_dir );
catch_sigchild();
// listener start (listen socket binding and listen)
ret = secure_channel_listen_start( &listener_info );
if ( !ret ) {
finalize_listener_info( &listener_info );
exit( EXIT_FAILURE );
}
set_fd_handler( listener_info.listen_fd, secure_channel_accept, &listener_info, NULL, NULL );
set_readable( listener_info.listen_fd, true );
start_trema();
finalize_listener_info( &listener_info );
stop_switch_management();
stop_service_management();
finalize_dpid_table();
return 0;
}
static void
check_connected( void *user_data ) {
UNUSED( user_data );
debug( "Checking a connection ( fd = %d, ip = %#x, port = %u ).", connection.fd, connection.ip, connection.port );
assert( secure_channel_initialized );
assert( connection.fd >= 0 );
set_writable( connection.fd, false );
delete_fd_handler( connection.fd );
int err = 0;
socklen_t length = sizeof( error );
int ret = getsockopt( connection.fd, SOL_SOCKET, SO_ERROR, &err, &length );
if ( ret < 0 ) {
error( "Failed to retrieve error code ( fd = %d, ret = %d, errno = %s [%d] ).",
connection.fd, ret, strerror( errno ), errno );
return;
}
switch ( err ) {
case 0:
connected();
break;
case EINTR:
case EAGAIN:
case ECONNREFUSED:
case ENETUNREACH:
case ETIMEDOUT:
warn( "Failed to connect ( fd = %d, errno = %s [%d] ).", connection.fd, strerror( err ), err );
backoff();
return;
case EINPROGRESS:
set_fd_handler( connection.fd, NULL, NULL, ( event_fd_callback ) check_connected, NULL );
set_writable( connection.fd, true );
break;
default:
error( "Failed to connect ( fd = %d, errno = %s [%d] ).", connection.fd, strerror( err ), err );
clear_connection();
return;
}
}
bool
init_redirector() {
if ( !init_tun( TUN_INTERFACE ) ) {
error( "Cannot create a tun interface." );
return false;
}
if ( host_db != NULL ) {
error( "Host database is already created." );
return false;
}
host_db = create_hash( compare_ip_address, hash_ip_address );
set_fd_handler( fd, read_tun_fd, NULL, NULL, NULL );
set_readable( fd, true );
add_periodic_event_callback( HOST_DB_AGING_INTERVAL, age_host_db, NULL );
return true;
}
/* A non-blocking call returned EAGAIN, so yield, ensuring the
* handlers are set up so that we'll be rescheduled when there is an
* interesting event on the socket.
*/
static coroutine_fn void co_yield(BDRVSSHState *s, BlockDriverState *bs)
{
set_fd_handler(s, bs);
qemu_coroutine_yield();
clear_fd_handler(s, bs);
}
static bool
try_connect() {
assert( connection.state != CONNECTED );
int fd = socket( PF_INET, SOCK_STREAM, 0 );
if ( fd < 0 ) {
error( "Failed to create a socket ( ret = %d, errno = %s [%d] ).",
fd, strerror( errno ), errno );
return false;
}
int flag = 1;
int ret = setsockopt( fd, IPPROTO_TCP, TCP_NODELAY, &flag, sizeof( flag ) );
if ( ret < 0 ) {
error( "Failed to set socket options ( fd = %d, ret = %d, errno = %s [%d] ).",
fd, ret, strerror( errno ), errno );
return false;
}
ret = fcntl( fd, F_SETFL, O_NONBLOCK );
if ( ret < 0 ) {
error( "Failed to enable non-blocking mode ( fd = %d, ret = %d, errno = %s [%d] ).",
fd, ret, strerror( errno ), errno );
close( fd );
return false;
}
connection.fd = fd;
struct sockaddr_in addr;
memset( &addr, 0, sizeof( struct sockaddr_in ) );
addr.sin_family = AF_INET;
addr.sin_port = htons( connection.port );
addr.sin_addr.s_addr = htonl( connection.ip );
transit_state( CONNECTING );
ret = connect( connection.fd, ( struct sockaddr * ) &addr, sizeof( struct sockaddr_in ) );
if ( ret < 0 ) {
switch ( errno ) {
case EINTR:
case EAGAIN:
case ECONNREFUSED:
case ENETUNREACH:
case ETIMEDOUT:
warn( "Failed to connect ( fd = %d, ret = %d, errno = %s [%d] ).",
connection.fd, ret, strerror( errno ), errno );
backoff();
return true;
case EINPROGRESS:
break;
default:
error( "Failed to connect ( fd = %d, ret = %d, errno = %s [%d] ).",
connection.fd, ret, strerror( errno ), errno );
clear_connection();
return false;
}
}
set_fd_handler( connection.fd, NULL, NULL, ( event_fd_callback ) check_connected, NULL );
set_writable( connection.fd, true );
return true;
}
int
main( int argc, char *argv[] ) {
int ret;
int i;
char *service_name;
char management_service_name[ MESSENGER_SERVICE_NAME_LENGTH ];
init_trema( &argc, &argv );
option_parser( argc, argv );
create_list( &switch_info.vendor_service_name_list );
create_list( &switch_info.packetin_service_name_list );
create_list( &switch_info.portstatus_service_name_list );
create_list( &switch_info.state_service_name_list );
// FIXME
#define VENDER_PREFIX "vendor::"
#define PACKET_IN_PREFIX "packet_in::"
#define PORTSTATUS_PREFIX "port_status::"
#define STATE_PREFIX "state_notify::"
for ( i = optind; i < argc; i++ ) {
if ( strncmp( argv[i], VENDER_PREFIX, strlen( VENDER_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[i] + strlen( VENDER_PREFIX ) );
insert_in_front( &switch_info.vendor_service_name_list, service_name );
}
else if ( strncmp( argv[i], PACKET_IN_PREFIX, strlen( PACKET_IN_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[i] + strlen( PACKET_IN_PREFIX ) );
insert_in_front( &switch_info.packetin_service_name_list, service_name );
}
else if ( strncmp( argv[i], PORTSTATUS_PREFIX, strlen( PORTSTATUS_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[i] + strlen( PORTSTATUS_PREFIX ) );
insert_in_front( &switch_info.portstatus_service_name_list, service_name );
}
else if ( strncmp( argv[i], STATE_PREFIX, strlen( STATE_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[i] + strlen( STATE_PREFIX ) );
insert_in_front( &switch_info.state_service_name_list, service_name );
}
}
fcntl( switch_info.secure_channel_fd, F_SETFL, O_NONBLOCK );
set_fd_handler( switch_info.secure_channel_fd, secure_channel_read, NULL, secure_channel_write, NULL );
set_readable( switch_info.secure_channel_fd, true );
set_writable( switch_info.secure_channel_fd, false );
// default switch configuration
switch_info.config_flags = OFPC_FRAG_NORMAL;
switch_info.miss_send_len = UINT16_MAX;
switch_info.fragment_buf = NULL;
switch_info.send_queue = create_message_queue();
switch_info.recv_queue = create_message_queue();
switch_info.running_timer = false;
init_xid_table();
init_cookie_table();
add_message_received_callback( get_trema_name(), service_recv );
snprintf( management_service_name , MESSENGER_SERVICE_NAME_LENGTH,
"%s.m", get_trema_name() );
management_service_name[ MESSENGER_SERVICE_NAME_LENGTH - 1 ] = '\0';
add_message_received_callback( management_service_name, management_recv );
ret = switch_event_connected( &switch_info );
if ( ret < 0 ) {
error( "Failed to set connected state." );
return -1;
}
flush_secure_channel( &switch_info );
start_trema();
finalize_xid_table();
finalize_cookie_table();
if ( switch_info.secure_channel_fd >= 0 ) {
delete_fd_handler( switch_info.secure_channel_fd );
}
return 0;
}
/* A non-blocking call returned EAGAIN, so yield, ensuring the
* handlers are set up so that we'll be rescheduled when there is an
* interesting event on the socket.
*/
static coroutine_fn void co_yield(BDRVSSHState *s)
{
set_fd_handler(s);
qemu_coroutine_yield();
clear_fd_handler(s);
}
int
main( int argc, char *argv[] ) {
int ret;
int i;
char *service_name;
init_trema( &argc, &argv );
option_parser( argc, argv );
create_list( &switch_info.vendor_service_name_list );
create_list( &switch_info.packetin_service_name_list );
create_list( &switch_info.portstatus_service_name_list );
create_list( &switch_info.state_service_name_list );
for ( i = optind; i < argc; i++ ) {
if ( strncmp( argv[ i ], VENDOR_PREFIX, strlen( VENDOR_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[ i ] + strlen( VENDOR_PREFIX ) );
append_to_tail( &switch_info.vendor_service_name_list, service_name );
}
else if ( strncmp( argv[ i ], PACKET_IN_PREFIX, strlen( PACKET_IN_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[ i ] + strlen( PACKET_IN_PREFIX ) );
append_to_tail( &switch_info.packetin_service_name_list, service_name );
}
else if ( strncmp( argv[ i ], PORTSTATUS_PREFIX, strlen( PORTSTATUS_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[ i ] + strlen( PORTSTATUS_PREFIX ) );
append_to_tail( &switch_info.portstatus_service_name_list, service_name );
}
else if ( strncmp( argv[ i ], STATE_PREFIX, strlen( STATE_PREFIX ) ) == 0 ) {
service_name = xstrdup( argv[ i ] + strlen( STATE_PREFIX ) );
append_to_tail( &switch_info.state_service_name_list, service_name );
}
}
struct sigaction signal_exit;
memset( &signal_exit, 0, sizeof( struct sigaction ) );
signal_exit.sa_handler = handle_sigterm;
sigaction( SIGINT, &signal_exit, NULL );
sigaction( SIGTERM, &signal_exit, NULL );
fcntl( switch_info.secure_channel_fd, F_SETFL, O_NONBLOCK );
set_fd_handler( switch_info.secure_channel_fd, secure_channel_read, NULL, secure_channel_write, NULL );
set_readable( switch_info.secure_channel_fd, true );
set_writable( switch_info.secure_channel_fd, false );
// default switch configuration
switch_info.config_flags = OFPC_FRAG_NORMAL;
switch_info.miss_send_len = UINT16_MAX;
switch_info.fragment_buf = NULL;
switch_info.send_queue = create_message_queue();
switch_info.recv_queue = create_message_queue();
switch_info.running_timer = false;
switch_info.echo_request_xid = 0;
init_xid_table();
if ( switch_info.cookie_translation ) {
init_cookie_table();
}
add_message_received_callback( get_trema_name(), service_recv );
set_management_application_request_handler( management_recv, NULL );
ret = switch_event_connected( &switch_info );
if ( ret < 0 ) {
error( "Failed to set connected state." );
return -1;
}
ret = flush_secure_channel( &switch_info );
if ( ret < 0 ) {
error( "Failed to flush secure channel. Terminating %s.", argv[ 0 ] );
return -1;
}
start_trema();
// Note: init_event_forward_interface will be called on feature_reply.
finalize_event_forward_interface();
finalize_xid_table();
if ( switch_info.cookie_translation ) {
finalize_cookie_table();
}
if ( switch_info.secure_channel_fd >= 0 ) {
delete_fd_handler( switch_info.secure_channel_fd );
}
return 0;
}
