OpenSRF/src/libstack/osrf_stack.c

   1 #include "osrf_stack.h"
   2
   3 osrf_message* _do_client( osrf_app_session*, osrf_message* );
   4 osrf_message* _do_server( osrf_app_session*, osrf_message* );
   5
   6 int osrf_stack_process( transport_client* client, int timeout ) {
   7         transport_message* msg = client_recv( client, timeout );
   8         if(msg == NULL) return 0;
   9         debug_handler( "Received message from transport code from %s", msg->sender );
  10         int status = osrf_stack_transport_handler( msg );
  11
  12         while(1) {
  13                 transport_message* m = client_recv( client, 0 );
  14                 if(m) {
  15                         debug_handler( "Received additional message from transport code");
  16                         status = osrf_stack_transport_handler( m );
  17                 } else  {
  18                         debug_handler( "osrf_stack_process returning with only 1 received message" );
  19                         break;
  20                 }
  21         }
  22
  23         return status;
  24 }
  25
  26
  27
  28 // -----------------------------------------------------------------------------
  29 // Entry point into the stack
  30 // -----------------------------------------------------------------------------
  31 int osrf_stack_transport_handler( transport_message* msg ) {
  32
  33         debug_handler( "Transport handler received new message \nfrom %s "
  34                         "to %s with body \n\n%s\n", msg->sender, msg->recipient, msg->body );
  35
  36         osrf_app_session* session = osrf_app_session_find_session( msg->thread );
  37
  38         if( session == NULL ) {  /* we must be a server, build a new session */
  39                 info_handler( "Received message for nonexistant session. Dropping..." );
  40                 //osrf_app_server_session_init( msg->thread,
  41                 message_free( msg );
  42                 return 1;
  43         }
  44
  45         debug_handler("Session [%s] found, building message", msg->thread );
  46
  47         osrf_app_session_set_remote( session, msg->sender );
  48         osrf_message* arr[OSRF_MAX_MSGS_PER_PACKET];
  49         memset(arr, 0, OSRF_MAX_MSGS_PER_PACKET );
  50         int num_msgs = osrf_message_deserialize(msg->body, arr, OSRF_MAX_MSGS_PER_PACKET);
  51
  52         debug_handler( "We received %d messages from %s", num_msgs, msg->sender );
  53
  54         /* XXX ERROR CHECKING, BAD JSON, ETC... */
  55         int i;
  56         for( i = 0; i != num_msgs; i++ ) {
  57
  58                 /* if we've received a jabber layer error message (probably talking to
  59                         someone who no longer exists) and we're not talking to the original
  60                         remote id for this server, consider it a redirect and pass it up */
  61                 if(msg->is_error) {
  62                         warning_handler( " !!! Received Jabber layer error message" );
  63
  64                         if(strcmp(session->remote_id,session->orig_remote_id)) {
  65                                 warning_handler( "Treating jabber error as redirect for tt [%d] "
  66                                         "and session [%s]", arr[i]->thread_trace, session->session_id );
  67
  68                                 arr[i]->m_type = STATUS;
  69                                 arr[i]->status_code = OSRF_STATUS_REDIRECTED;
  70
  71                         } else {
  72                                 warning_handler(" * Jabber Error is for top level remote id [%s], no one "
  73                                                 "to send my message too!!!", session->remote_id );
  74                         }
  75                 }
  76
  77                 osrf_stack_message_handler( session, arr[i] );
  78         }
  79
  80         message_free( msg );
  81         debug_handler("after msg delete");
  82
  83         return 1;
  84 }
  85
  86 int osrf_stack_message_handler( osrf_app_session* session, osrf_message* msg ) {
  87         if(session == NULL || msg == NULL)
  88                 return 0;
  89
  90         osrf_message* ret_msg = NULL;
  91         if( session->type ==  OSRF_SESSION_CLIENT )
  92                  ret_msg = _do_client( session, msg );
  93         else
  94                 ret_msg= _do_server( session, msg );
  95
  96         if(ret_msg)
  97                 osrf_stack_application_handler( session, ret_msg );
  98         else
  99                 osrf_message_free(msg);
 100
 101         return 1;
 102
 103 }
 104
 105 /** If we return a message, that message should be passed up the stack,
 106   * if we return NULL, we're finished for now...
 107   */
 108 osrf_message* _do_client( osrf_app_session* session, osrf_message* msg ) {
 109         if(session == NULL || msg == NULL)
 110                 return NULL;
 111
 112         osrf_message* new_msg;
 113
 114         if( msg->m_type == STATUS ) {
 115
 116                 switch( msg->status_code ) {
 117
 118                         case OSRF_STATUS_OK:
 119                                 debug_handler("We connected successfully");
 120                                 session->state = OSRF_SESSION_CONNECTED;
 121                                 debug_handler( "State: %x => %s => %d", session, session->session_id, session->state );
 122                                 return NULL;
 123
 124                         case OSRF_STATUS_COMPLETE:
 125                                 osrf_app_session_set_complete( session, msg->thread_trace );
 126                                 return NULL;
 127
 128                         case OSRF_STATUS_CONTINUE:
 129                                 osrf_app_session_request_reset_timeout( session, msg->thread_trace );
 130                                 return NULL;
 131
 132                         case OSRF_STATUS_REDIRECTED:
 133                                 osrf_app_session_reset_remote( session );
 134                                 session->state = OSRF_SESSION_DISCONNECTED;
 135                                 osrf_app_session_request_resend( session, msg->thread_trace );
 136                                 return NULL;
 137
 138                         case OSRF_STATUS_EXPFAILED:
 139                                 osrf_app_session_reset_remote( session );
 140                                 session->state = OSRF_SESSION_DISCONNECTED;
 141                                 osrf_app_session_request_resend( session, msg->thread_trace );
 142                                 return NULL;
 143
 144                         case OSRF_STATUS_TIMEOUT:
 145                                 osrf_app_session_reset_remote( session );
 146                                 session->state = OSRF_SESSION_DISCONNECTED;
 147                                 osrf_app_session_request_resend( session, msg->thread_trace );
 148                                 return NULL;
 149
 150
 151                         default:
 152                                 new_msg = osrf_message_init( RESULT, msg->thread_trace, msg->protocol );
 153                                 osrf_message_set_status_info( new_msg,
 154                                                 msg->status_name, msg->status_text, msg->status_code );
 155                                 warning_handler("The stack doesn't know what to do with "
 156                                                 "the provided message code: %d, name %s. Passing UP.",
 157                                                 msg->status_code, msg->status_name );
 158                                 new_msg->is_exception = 1;
 159                                 osrf_app_session_set_complete( session, msg->thread_trace );
 160                                 osrf_message_free(msg);
 161                                 return new_msg;
 162                 }
 163
 164                 return NULL;
 165
 166         } else if( msg->m_type == RESULT )
 167                 return msg;
 168
 169         return NULL;
 170
 171 }
 172
 173
 174 /** If we return a message, that message should be passed up the stack,
 175   * if we return NULL, we're finished for now...
 176   */
 177 osrf_message* _do_server( osrf_app_session* session, osrf_message* msg ) {
 178         if(session == NULL || msg == NULL)
 179                 return NULL;
 180
 181
 182         if( msg->m_type == STATUS ) { return NULL; }
 183
 184         warning_handler( "We dont' do servers yet !!" );
 185
 186         return msg;
 187 }
 188
 189
 190
 191
 192 int osrf_stack_application_handler( osrf_app_session* session, osrf_message* msg ) {
 193         if(session == NULL || msg == NULL)
 194                 return 0;
 195
 196         if(msg->m_type == RESULT) {
 197                 osrf_app_session_push_queue( session, msg );
 198                 return 1;
 199         }
 200
 201         warning_handler( "application_handler can't handle whatever you sent, type %d", msg->m_type);
 202
 203         return 1;
 204
 205 }