mtrudel-adhearsion 0.8.3

data/lib/adhearsion/voip/asterisk/manager_interface/ami_lexer.rl.rb

@@ -0,0 +1,286 @@
+require File.join(File.dirname(__FILE__), 'ami_messages.rb')
+
+module Adhearsion
+  module VoIP
+    module Asterisk
+      module Manager
+        class AbstractAsteriskManagerInterfaceStreamLexer
+
+          BUFFER_SIZE = 8.kilobytes unless defined? BUFFER_SIZE
+
+          %%{
+          	machine ami_protocol_parser;
+          
+            # All required Ragel actions are implemented as Ruby methods.
+          
+            # Executed after a "Respone: Success" or "Response: Pong"
+            action init_success { init_success }
+          
+            action init_response_follows { init_response_follows }
+          
+            action init_error { init_error }
+            
+            action message_received { message_received @current_message }
+            action error_received   {   error_received @current_message }
+              
+            action version_starts { version_starts }
+            action version_stops  { version_stops  }
+          
+            action key_starts { key_starts }
+            action key_stops  { key_stops  }
+          
+            action value_starts { value_starts }
+            action value_stops  { value_stops  }
+          
+            action error_reason_starts { error_reason_starts }
+            action error_reason_stops  { error_reason_stops  }
+          
+            action syntax_error_starts { syntax_error_starts }
+            action syntax_error_stops  { syntax_error_stops  }
+          
+            action immediate_response_starts { immediate_response_starts }
+            action immediate_response_stops  { immediate_response_stops  }
+          
+            action follows_text_starts { follows_text_starts }
+            action follows_text_stops  { follows_text_stops  }
+          
+            action event_name_starts { event_name_starts }
+            action event_name_stops  { event_name_stops  }
+          
+            include ami_protocol_parser_machine "ami_protocol_lexer_machine.rl";
+    
+          }%%##
+
+          attr_accessor(:ami_version)
+          def initialize
+    
+            @data = ""
+            @current_pointer = 0
+            @ragel_stack = []
+            
+            %%{
+              # All other variables become local, letting Ruby garbage collect them. This
+              # prevents us from having to manually reset them.
+      
+        			variable data  @data;
+              variable p     @current_pointer;
+        			variable pe    @data_ending_pointer;
+        			variable cs    @current_state;
+        			variable ts    @token_start;
+        			variable te    @token_end;
+        			variable stack @stack;
+        			variable act   @ragel_act;
+        			variable eof   @eof;
+			        variable stack @ragel_stack;
+			        variable top   @ragel_stack_top;
+			        
+        			write data;
+              write init;
+            }%%##
+            
+          end
+  
+          def <<(new_data)
+            extend_buffer_with new_data
+            resume!
+          end
+        
+          def resume!
+            %%{ write exec; }%%##
+          end
+        
+          def extend_buffer_with(new_data)
+            if new_data.size + @data.size > BUFFER_SIZE
+              @data.slice! 0...new_data.size
+              # TODO: What if the current_pointer wasn't at the end of the data for some reason?
+              @current_pointer = @data.size
+            end
+            @data << new_data
+            @data_ending_pointer = @data.size
+          end
+        
+          protected
+                
+          ##
+          # Called after a response or event has been successfully parsed.
+          #
+          # @param [ManagerInterfaceResponse, ManagerInterfaceEvent] message The message just received
+          #
+          def message_received(message)
+            raise NotImplementedError, "Must be implemented in subclass!"
+          end
+
+          ##
+          # Called when there is an Error: stanza on the socket. Could be caused by executing an unrecognized command, trying
+          # to originate into an invalid priority, etc. Note: many errors' responses are actually tightly coupled to a
+          # ManagerInterfaceEvent which comes directly after it. Often the message will say something like "Channel status
+          # will follow".
+          #
+          # @param [String] reason The reason given in the Message: header for the error stanza.
+          #
+          def error_received(reason)
+            raise NotImplementedError, "Must be implemented in subclass!"
+          end
+
+          ##
+          # Called when there's a syntax error on the socket. This doesn't happen as often as it should because, in many cases,
+          # it's impossible to distinguish between a syntax error and an immediate packet.
+          #
+          # @param [String] ignored_chunk The offending text which caused the syntax error.
+          def syntax_error_encountered(ignored_chunk)
+            raise NotImplementedError, "Must be implemented in subclass!"
+          end
+        
+          def init_success
+            @current_message = ManagerInterfaceResponse.new
+          end
+        
+          def init_response_follows
+            @current_message = ManagerInterfaceResponse.new
+          end
+          
+          def init_error
+            @current_message = ManagerInterfaceError.new()
+          end
+          
+          def version_starts
+            @start_of_version = @current_pointer
+          end
+  
+          def version_stops
+            self.ami_version = @data[@start_of_version...@current_pointer].to_f
+            @start_of_version = nil
+          end
+  
+          def event_name_starts
+            @event_name_start = @current_pointer
+          end
+  
+          def event_name_stops
+            event_name = @data[@event_name_start...@current_pointer]
+            @event_name_start = nil
+            @current_message = ManagerInterfaceEvent.new(event_name)
+          end
+  
+          def key_starts
+            @current_key_position = @current_pointer
+          end
+  
+          def key_stops
+            @current_key = @data[@current_key_position...@current_pointer]
+          end
+  
+          def value_starts
+            @current_value_position = @current_pointer
+          end
+  
+          def value_stops
+            @current_value = @data[@current_value_position...@current_pointer]
+            @last_seen_value_end = @current_pointer + 2 # 2 for \r\n
+            add_pair_to_current_message
+          end
+          
+          def error_reason_starts
+            @error_reason_start = @current_pointer
+          end
+          
+          def error_reason_stops
+            @current_message.message = @data[@error_reason_start...@current_pointer]
+          end
+          
+          def follows_text_starts
+            @follows_text_start = @current_pointer
+          end
+  
+          def follows_text_stops
+            text = @data[@last_seen_value_end..(@current_pointer - "\r\n--END COMMAND--".size)]
+            @current_message.text_body = text
+            @follows_text_start = nil
+          end
+  
+          def add_pair_to_current_message
+            @current_message[@current_key] = @current_value
+            reset_key_and_value_positions
+          end
+  
+          def reset_key_and_value_positions
+            @current_key, @current_value, @current_key_position, @current_value_position = nil
+          end
+  
+          def syntax_error_starts
+            @current_syntax_error_start = @current_pointer # Adding 1 since the pointer is still set to the last successful match
+          end
+  
+          def syntax_error_stops
+            # Subtracting 3 from @current_pointer below for "\r\n" which separates a stanza
+            offending_data = @data[@current_syntax_error_start...@current_pointer - 1]
+            syntax_error_encountered offending_data
+            @current_syntax_error_start = nil
+          end
+  
+          def immediate_response_starts
+            @immediate_response_start = @current_pointer
+          end
+  
+          def immediate_response_stops
+            message = @data[@immediate_response_start...(@current_pointer -1)]
+            message_received ManagerInterfaceResponse.from_immediate_response(message)
+          end
+  
+          ##
+          # This method is used primarily in debugging.
+          #
+          def view_buffer(message=nil)
+    
+            message ||= "Viewing the buffer"
+    
+            buffer = @data.clone
+            buffer.insert(@current_pointer, "\033[0;31m\033[1;31m^\033[0m")
+    
+            buffer.gsub!("\r", "\\\\r")
+            buffer.gsub!("\n", "\\n\n")
+    
+            puts <<-INSPECTION
+VVVVVVVVVVVVVVVVVVVVVVVVVVVVV
+####  #{message}
+#############################
+#{buffer}
+#############################
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+            INSPECTION
+    
+          end
+        end
+        class DelegatingAsteriskManagerInterfaceLexer < AbstractAsteriskManagerInterfaceStreamLexer
+          
+          def initialize(delegate, method_delegation_map=nil)
+            super()
+            @delegate = delegate
+            
+            @message_received_method = method_delegation_map && method_delegation_map.has_key?(:message_received) ?
+                method_delegation_map[:message_received] : :message_received
+            
+            @error_received_method = method_delegation_map && method_delegation_map.has_key?(:error_received) ?
+                method_delegation_map[:error_received] : :error_received
+            
+            @syntax_error_method = method_delegation_map && method_delegation_map.has_key?(:syntax_error_encountered) ?
+                method_delegation_map[:syntax_error_encountered] : :syntax_error_encountered
+          end
+          
+          def message_received(message)
+            @delegate.send(@message_received_method, message)
+          end
+          
+          def error_received(message)
+            @delegate.send(@error_received_method, message)
+          end
+          
+          def syntax_error_encountered(ignored_chunk)
+            @delegate.send(@syntax_error_method, ignored_chunk)
+          end
+          
+        end
+      end
+    end
+  end
+end

data/lib/adhearsion/voip/asterisk/manager_interface/ami_messages.rb

@@ -0,0 +1,78 @@
+module Adhearsion
+  module VoIP
+    module Asterisk
+      module Manager
+
+
+        ##
+        # This is the object containing a response from Asterisk.
+        #
+        # Note: not all responses have an ActionID!
+        #
+        class ManagerInterfaceResponse
+        
+          class << self
+            def from_immediate_response(text)
+              returning new do |instance|
+                instance.text_body = text
+              end
+            end
+          end
+          
+          attr_accessor :action,
+                        :action_id,
+                        :text_body  # For "Response: Follows" sections
+          attr_reader   :events
+          
+          def initialize
+            @headers = HashWithIndifferentAccess.new
+          end
+          
+          def has_text_body?
+            !! @text_body
+          end
+          
+          def headers
+            @headers.clone
+          end
+        
+          def [](arg)
+            @headers[arg]
+          end
+  
+          def []=(key,value)
+            @headers[key] = value
+          end
+          
+        end
+      
+        class ManagerInterfaceError < Exception
+          
+          attr_accessor :message
+          def initialize
+            @headers = HashWithIndifferentAccess.new
+          end
+          
+          def [](key)
+            @headers[key]
+          end
+          
+          def []=(key,value)
+            @headers[key] = value
+          end
+          
+        end
+        
+        class ManagerInterfaceEvent < ManagerInterfaceResponse
+  
+          attr_reader :name
+          def initialize(name)
+            super()
+            @name = name
+          end
+          
+        end
+      end
+    end
+  end
+end

data/lib/adhearsion/voip/asterisk/manager_interface/ami_protocol_lexer_machine.rl

@@ -0,0 +1,87 @@
+%%{ #%
+
+#########
+## This file is written with the Ragel programming language and parses the Asterisk Manager Interface protocol. It depends
+## upon Ragel actions which should be implemented in another Ragel-parsed file which includes this file.
+##
+## Ragel was used because the AMI protocol is extremely non-deterministic and, in the edge cases, requires something both
+## very robust and something which can recover from syntax errors.
+##
+## Note: This file is language agnostic. From this AMI parsers in many other languages can be generated.
+#########
+
+machine ami_protocol_parser_machine;
+
+cr = "\r";              # A carriage return. Used before (almost) every newline character.
+lf = "\n";              # Newline. Used (with cr) to separate key/value pairs and stanzas.
+crlf = cr lf;           # Means "carriage return and line feed". Used to separate key/value pairs and stanzas
+loose_newline = cr? lf; # Used sometimes when the AMI protocol is nondeterministic about the delimiter
+
+white = [\t ];                 # Single whitespace character, either a tab or a space
+colon = ":" [ ]**;             # Separates keys from values. "A colon followed by any number of spaces"
+stanza_break = crlf crlf;      # The seperator between two stanzas.
+rest_of_line = (any* -- crlf); # Match all characters until the next line seperator.
+
+Prompt = "Asterisk Call Manager/" digit+ >version_starts "." digit+ %version_stops crlf;
+
+Key = ((alnum | print) -- (cr | lf | ":"))+;
+KeyValuePair = Key >key_starts %key_stops colon rest_of_line >value_starts %value_stops crlf;
+
+FollowsDelimiter = crlf "--END COMMAND--";
+
+Response = "Response"i colon;
+
+Success	 = Response "Success"i %init_success crlf @{ fgoto success; };
+Pong     = Response "Pong"i %init_success crlf @{ fgoto success; };
+Event    = "Event"i colon %event_name_starts rest_of_line %event_name_stops crlf @{ fgoto success; };
+Error    = Response "Error"i %init_error crlf (("Message"i colon rest_of_line >error_reason_starts crlf >error_reason_stops) | KeyValuePair)+ crlf @error_received;
+Follows  = Response "Follows"i crlf @init_response_follows @{ fgoto response_follows; };
+
+# For "Response: Follows"
+FollowsBody = (any* -- FollowsDelimiter) >follows_text_starts FollowsDelimiter @follows_text_stops crlf;
+
+ImmediateResponse = (any+ -- (loose_newline | ":")) >immediate_response_starts loose_newline @immediate_response_stops @{fret;};
+SyntaxError       = (any+ -- crlf) >syntax_error_starts crlf @syntax_error_stops;
+
+irregularity := |*
+  ImmediateResponse; # Performs the fret in the ImmediateResponse FSM
+  SyntaxError       => { fret; };
+*|;
+
+# When a new socket is established, Asterisk will send the version of the protocol per the Prompt machine. Because it's
+# tedious for unit tests to always send this, we'll put some intelligence into this parser to support going straight into
+# the protocol-parsing machine. It's also conceivable that a variant of AMI would not send this initial information.
+main := |*
+  Prompt => { fgoto protocol; };
+  any => {
+    # If this scanner's look-ahead capability didn't match the prompt, let's ignore the need for a prompt
+    fhold;
+    fgoto protocol;
+  };
+*|;
+
+protocol := |*
+  Prompt;
+  Success;
+  Pong;
+  Event;
+  Error;
+  Follows crlf;
+  crlf => { fgoto protocol; }; # If we get a crlf out of place, let's just ignore it.
+  any  => {
+    # If NONE of the above patterns match, we consider this a syntax error. The irregularity machine can recover gracefully.
+    fhold;
+    fcall irregularity;
+  };
+*|;
+
+success := KeyValuePair* crlf @message_received @{fgoto protocol;};
+
+# For the "Response: Follows" protocol abnormality. What happens if there's a protocol irregularity in this state???
+response_follows := |*
+    KeyValuePair+;
+    FollowsBody;
+    crlf @{ message_received; fgoto protocol; };
+*|;
+ 
+}%%