static void
unregistration_retry( void *user_data ) {
struct switch_info *sw_info = user_data;
if ( sw_info->retry_count == 0 ) {
stop_switch();
return;
}
sw_info->retry_count--;
sw_info->running_timer = false;
debug( "Checking switch manager's switch list." );
set_list_switches_reply_handler( confirm_self_dpid_is_unregistered );
if ( send_list_switches_request( sw_info ) ) {
switch_set_timeout( UNREGISTRATION_TIMEOUT, unregistration_timeout, sw_info );
}
else {
error( "Failed to send switch list request to switch manager." );
stop_switch();
}
}
int
main( int argc, char *argv[] ) {
init_trema( &argc, &argv );
show_desc show_desc;
show_desc.db = create_hash( compare_datapath_id, hash_datapath_id );
show_desc.count = 0;
set_list_switches_reply_handler( handle_list_switches_reply );
#ifdef TREMA_EDGE
set_multipart_reply_handler( handle_stats_reply, &show_desc );
#else
set_stats_reply_handler( handle_stats_reply, &show_desc );
set_features_reply_handler( handle_features_reply, &show_desc );
#endif
add_periodic_event_callback( 10, timed_out, NULL );
send_list_switches_request( &show_desc );
start_trema();
return 0;
}
int
switch_event_recv_featuresreply( struct switch_info *sw_info, uint64_t *dpid ) {
int ret;
char new_service_name[ SWITCH_MANAGER_PREFIX_STR_LEN + SWITCH_MANAGER_DPID_STR_LEN + 1 ];
const uint16_t new_service_name_len = SWITCH_MANAGER_PREFIX_STR_LEN + SWITCH_MANAGER_DPID_STR_LEN + 1;
switch ( sw_info->state ) {
case SWITCH_STATE_WAIT_FEATURES_REPLY:
sw_info->datapath_id = *dpid;
sw_info->state = SWITCH_STATE_WAIT_REGISTRATION;
// cancel to features_reply_wait-timeout timer
switch_unset_timeout( switch_event_timeout_features_reply, NULL );
// TODO: set keepalive-timeout
snprintf( new_service_name, new_service_name_len, "%s%#" PRIx64, SWITCH_MANAGER_PREFIX, sw_info->datapath_id );
// checking duplicate service
pid_t pid = get_pid_by_trema_name( new_service_name );
if ( pid > 0 ) {
debug( "duplicated datapath-id %#" PRIx64, sw_info->datapath_id );
if ( !terminate_trema_process( pid ) ) {
return -1;
}
}
// rename service_name of messenger
rename_message_received_callback( get_trema_name(), new_service_name );
// rename management service name
char *management_service_name = xstrdup( get_management_service_name( get_trema_name() ) );
char *new_management_service_name = xstrdup( get_management_service_name( new_service_name ) );
rename_message_requested_callback( management_service_name, new_management_service_name );
xfree( management_service_name );
xfree( new_management_service_name );
debug( "Rename service name from %s to %s.", get_trema_name(), new_service_name );
if ( messenger_dump_enabled() ) {
stop_messenger_dump();
start_messenger_dump( new_service_name, DEFAULT_DUMP_SERVICE_NAME );
}
set_trema_name( new_service_name );
// reset to default config
ret = ofpmsg_send_setconfig( sw_info );
if ( ret < 0 ) {
error( "Failed to send setconfig." );
return ret;
}
if ( switch_info.flow_cleanup ) {
debug( "Deleting all flows." );
ret = ofpmsg_send_delete_all_flows( sw_info );
if ( ret < 0 ) {
error( "Failed to send delete all flows." );
return ret;
}
}
// switch_ready to switch_manager
debug( "Notify switch_ready to switch manager." );
char switch_manager[] = SWITCH_MANAGER;
list_element switch_manager_only_list;
switch_manager_only_list.next = NULL;
switch_manager_only_list.data = switch_manager;
service_send_to_application( &switch_manager_only_list, MESSENGER_OPENFLOW_READY, &sw_info->datapath_id, NULL );
init_event_forward_interface();
// Check switch_manager registration
debug( "Checking switch manager's switch list." );
sw_info->retry_count = REGISTRATION_RETRY_COUNT;
set_list_switches_reply_handler( confirm_self_dpid_is_registered );
if ( send_list_switches_request( sw_info ) ) {
switch_set_timeout( REGISTRATION_TIMEOUT, registration_timeout, sw_info );
}
else {
error( "Failed to send switch list request to switch manager." );
return -1;
}
break;
case SWITCH_STATE_WAIT_REGISTRATION:
case SWITCH_STATE_COMPLETED:
// NOP
break;
default:
notice( "Invalid event 'features reply' from a switch." );
return -1;
break;
}
return 0;
}
int
switch_event_disconnected( struct switch_info *sw_info ) {
int old_state = sw_info->state;
if ( sw_info->state == SWITCH_STATE_DISCONNECTED ||
sw_info->state == SWITCH_STATE_CONNECTION_FAILED ) {
debug( "already disconnected" );
return -1;
}
if ( sw_info->state == SWITCH_STATE_COMPLETED ) {
sw_info->state = SWITCH_STATE_DISCONNECTED;
}
else {
sw_info->state = SWITCH_STATE_CONNECTION_FAILED;
}
if ( old_state == SWITCH_STATE_COMPLETED ) {
switch_unset_timeout( echo_reply_timeout, NULL );
}
if ( sw_info->fragment_buf != NULL ) {
free_buffer( sw_info->fragment_buf );
sw_info->fragment_buf = NULL;
}
if ( sw_info->send_queue != NULL ) {
delete_message_queue( sw_info->send_queue );
sw_info->send_queue = NULL;
}
if ( sw_info->recv_queue != NULL ) {
delete_message_queue( sw_info->recv_queue );
sw_info->recv_queue = NULL;
}
if ( sw_info->secure_channel_fd >= 0 ) {
set_readable( switch_info.secure_channel_fd, false );
set_writable( switch_info.secure_channel_fd, false );
delete_fd_handler( switch_info.secure_channel_fd );
}
uint8_t state = MESSENGER_OPENFLOW_DISCONNECTED;
if ( sw_info->state == SWITCH_STATE_CONNECTION_FAILED ) {
state = MESSENGER_OPENFLOW_FAILD_TO_CONNECT;
}
debug( "Notify switch_disconnected to switch manager." );
char switch_manager[] = SWITCH_MANAGER;
list_element switch_manager_only_list;
switch_manager_only_list.next = NULL;
switch_manager_only_list.data = switch_manager;
service_send_to_application( &switch_manager_only_list, state, &sw_info->datapath_id, NULL );
// Check switch_manager registration
debug( "Checking switch manager's switch list." );
sw_info->retry_count = UNREGISTRATION_RETRY_COUNT;
set_list_switches_reply_handler( confirm_self_dpid_is_unregistered );
if ( send_list_switches_request( sw_info ) ) {
switch_set_timeout( UNREGISTRATION_TIMEOUT, unregistration_timeout, sw_info );
}
else {
error( "Failed to send switch list request to switch manager." );
stop_switch();
}
return 0;
}
static VALUE
controller_send_list_switches_request( VALUE self ) {
send_list_switches_request( ( void * ) self );
return self;
}
