1 #include <opensrf/osrf_stack.h>
2 #include <opensrf/osrf_application.h>
4 osrf_message* _do_client( osrf_app_session*, osrf_message* );
5 osrf_message* _do_server( osrf_app_session*, osrf_message* );
7 /* tell osrf_app_session where the stack entry is */
8 int (*osrf_stack_entry_point) (transport_client*, int, int*) = &osrf_stack_process;
10 int osrf_stack_process( transport_client* client, int timeout, int* msg_received ) {
11 if( !client ) return -1;
12 transport_message* msg = NULL;
13 if(msg_received) *msg_received = 0;
15 while( (msg = client_recv( client, timeout )) ) {
16 if(msg_received) *msg_received = 1;
17 osrfLogDebug( OSRF_LOG_MARK, "Received message from transport code from %s", msg->sender );
18 osrf_stack_transport_handler( msg, NULL );
23 osrfLogWarning(OSRF_LOG_MARK, "transport_client had trouble reading from the socket..");
27 if( ! client_connected( client ) ) return -1;
34 // -----------------------------------------------------------------------------
35 // Entry point into the stack
36 // -----------------------------------------------------------------------------
37 osrfAppSession* osrf_stack_transport_handler( transport_message* msg, char* my_service ) {
41 osrfLogSetXid(msg->osrf_xid);
43 osrfLogDebug( OSRF_LOG_MARK, "Transport handler received new message \nfrom %s "
44 "to %s with body \n\n%s\n", msg->sender, msg->recipient, msg->body );
46 if( msg->is_error && ! msg->thread ) {
47 osrfLogWarning( OSRF_LOG_MARK, "!! Received jabber layer error for %s ... exiting\n", msg->sender );
52 if(! msg->thread && ! msg->is_error ) {
53 osrfLogWarning( OSRF_LOG_MARK, "Received a non-error message with no thread trace... dropping");
58 osrf_app_session* session = osrf_app_session_find_session( msg->thread );
60 if( !session && my_service )
61 session = osrf_app_server_session_init( msg->thread, my_service, msg->sender);
63 if( !session ) return NULL;
66 osrfLogDebug( OSRF_LOG_MARK, "Session [%s] found or built", session->session_id );
68 osrf_app_session_set_remote( session, msg->sender );
69 osrf_message* arr[OSRF_MAX_MSGS_PER_PACKET];
70 memset(arr, 0, OSRF_MAX_MSGS_PER_PACKET );
71 int num_msgs = osrf_message_deserialize(msg->body, arr, OSRF_MAX_MSGS_PER_PACKET);
73 osrfLogDebug( OSRF_LOG_MARK, "We received %d messages from %s", num_msgs, msg->sender );
75 double starttime = get_timestamp_millis();
78 for( i = 0; i != num_msgs; i++ ) {
80 /* if we've received a jabber layer error message (probably talking to
81 someone who no longer exists) and we're not talking to the original
82 remote id for this server, consider it a redirect and pass it up */
84 osrfLogWarning( OSRF_LOG_MARK, " !!! Received Jabber layer error message" );
86 if(strcmp(session->remote_id,session->orig_remote_id)) {
87 osrfLogWarning( OSRF_LOG_MARK, "Treating jabber error as redirect for tt [%d] "
88 "and session [%s]", arr[i]->thread_trace, session->session_id );
90 arr[i]->m_type = STATUS;
91 arr[i]->status_code = OSRF_STATUS_REDIRECTED;
94 osrfLogWarning( OSRF_LOG_MARK, " * Jabber Error is for top level remote id [%s], no one "
95 "to send my message too!!!", session->remote_id );
99 osrf_stack_message_handler( session, arr[i] );
102 double duration = get_timestamp_millis() - starttime;
103 osrfLogInfo(OSRF_LOG_MARK, "Message processing duration %f", duration);
106 osrfLogDebug( OSRF_LOG_MARK, "after msg delete");
111 int osrf_stack_message_handler( osrf_app_session* session, osrf_message* msg ) {
112 if(session == NULL || msg == NULL)
115 osrf_message* ret_msg = NULL;
117 if( session->type == OSRF_SESSION_CLIENT )
118 ret_msg = _do_client( session, msg );
120 ret_msg= _do_server( session, msg );
123 osrfLogDebug( OSRF_LOG_MARK, "passing message %d / session %s to app handler",
124 msg->thread_trace, session->session_id );
125 osrf_stack_application_handler( session, ret_msg );
127 osrf_message_free(msg);
133 /** If we return a message, that message should be passed up the stack,
134 * if we return NULL, we're finished for now...
136 osrf_message* _do_client( osrf_app_session* session, osrf_message* msg ) {
137 if(session == NULL || msg == NULL)
140 osrf_message* new_msg;
142 if( msg->m_type == STATUS ) {
144 switch( msg->status_code ) {
147 osrfLogDebug( OSRF_LOG_MARK, "We connected successfully");
148 session->state = OSRF_SESSION_CONNECTED;
149 osrfLogDebug( OSRF_LOG_MARK, "State: %x => %s => %d", session, session->session_id, session->state );
152 case OSRF_STATUS_COMPLETE:
153 osrf_app_session_set_complete( session, msg->thread_trace );
156 case OSRF_STATUS_CONTINUE:
157 osrf_app_session_request_reset_timeout( session, msg->thread_trace );
160 case OSRF_STATUS_REDIRECTED:
161 osrf_app_session_reset_remote( session );
162 session->state = OSRF_SESSION_DISCONNECTED;
163 osrf_app_session_request_resend( session, msg->thread_trace );
166 case OSRF_STATUS_EXPFAILED:
167 osrf_app_session_reset_remote( session );
168 session->state = OSRF_SESSION_DISCONNECTED;
171 case OSRF_STATUS_TIMEOUT:
172 osrf_app_session_reset_remote( session );
173 session->state = OSRF_SESSION_DISCONNECTED;
174 osrf_app_session_request_resend( session, msg->thread_trace );
179 new_msg = osrf_message_init( RESULT, msg->thread_trace, msg->protocol );
180 osrf_message_set_status_info( new_msg,
181 msg->status_name, msg->status_text, msg->status_code );
182 osrfLogWarning( OSRF_LOG_MARK, "The stack doesn't know what to do with "
183 "the provided message code: %d, name %s. Passing UP.",
184 msg->status_code, msg->status_name );
185 new_msg->is_exception = 1;
186 osrf_app_session_set_complete( session, msg->thread_trace );
187 osrf_message_free(msg);
193 } else if( msg->m_type == RESULT )
201 /** If we return a message, that message should be passed up the stack,
202 * if we return NULL, we're finished for now...
204 osrf_message* _do_server( osrf_app_session* session, osrf_message* msg ) {
206 if(session == NULL || msg == NULL) return NULL;
208 osrfLogDebug( OSRF_LOG_MARK, "Server received message of type %d", msg->m_type );
210 switch( msg->m_type ) {
216 /* session will be freed by the forker */
217 osrfLogDebug(OSRF_LOG_MARK, "Client sent explicit disconnect");
218 session->state = OSRF_SESSION_DISCONNECTED;
222 osrfAppSessionStatus( session, OSRF_STATUS_OK,
223 "osrfConnectStatus", msg->thread_trace, "Connection Successful" );
224 session->state = OSRF_SESSION_CONNECTED;
229 osrfLogDebug( OSRF_LOG_MARK, "server passing message %d to application handler "
230 "for session %s", msg->thread_trace, session->session_id );
234 osrfLogWarning( OSRF_LOG_MARK, "Server cannot handle message of type %d", msg->m_type );
235 session->state = OSRF_SESSION_DISCONNECTED;
243 int osrf_stack_application_handler( osrf_app_session* session, osrf_message* msg ) {
244 if(session == NULL || msg == NULL) return 0;
246 if(msg->m_type == RESULT && session->type == OSRF_SESSION_CLIENT) {
247 osrf_app_session_push_queue( session, msg );
251 if(msg->m_type != REQUEST) return 1;
253 char* method = msg->method_name;
254 char* app = session->remote_service;
255 jsonObject* params = msg->_params;
257 osrfAppRunMethod( app, method, session, msg->thread_trace, params );
258 osrfMessageFree(msg);