ni_nitdoc: added fast copy past utility to signatures.
[nit.git] / src / debugger.nit
index af3f467..87c9a17 100644 (file)
@@ -23,11 +23,14 @@ intrude import naive_interpreter
 redef class ToolContext
        # -d
        var opt_debugger_mode: OptionBool = new OptionBool("Launches the target program with the debugger attached to it", "-d")
+       # -c
+       var opt_debugger_autorun: OptionBool = new OptionBool("Launches the target program with the interpreter, such as when the program fails, the debugging prompt is summoned", "-c")
 
        redef init
        do
                super
                self.option_context.add_option(self.opt_debugger_mode)
+               self.option_context.add_option(self.opt_debugger_autorun)
        end
 end
 
@@ -50,6 +53,20 @@ redef class ModelBuilder
                var time1 = get_time
                self.toolcontext.info("*** END INTERPRETING: {time1-time0} ***", 2)
        end
+
+       fun run_debugger_autorun(mainmodule: MModule, arguments: Array[String])
+       do
+               var time0 = get_time
+               self.toolcontext.info("*** START INTERPRETING ***", 1)
+
+               var interpreter = new Debugger(self, mainmodule, arguments)
+               interpreter.autocontinue = true
+
+               init_naive_interpreter(interpreter, mainmodule)
+
+               var time1 = get_time
+               self.toolcontext.info("*** END INTERPRETING: {time1-time0} ***", 2)
+       end
 end
 
 # The class extending NaiveInterpreter by adding debugging methods
@@ -79,6 +96,23 @@ class Debugger
        # Aliases hashmap (maps an alias to a variable name)
        var aliases = new HashMap[String, String]
 
+       # Set containing all the traced variables and their related frame
+       private var traces = new HashSet[TraceObject]
+
+       # Map containing all the positions for the positions of the arguments traced
+       # In a function call
+       private var fun_call_arguments_positions = new HashMap[Int, TraceObject]
+
+       # Triggers the remapping of a trace object in the local context after a function call
+       var aftermath = false
+
+       # Used to prevent the case when the body of the function called is empty
+       # If it is not, then, the remapping won't be happening
+       var frame_count_aftermath = 1
+
+       # Auto continues the execution until the end or until an error is encountered
+       var autocontinue = false
+
        #######################################################################
        ##                  Execution of statement function                  ##
        #######################################################################
@@ -92,9 +126,19 @@ class Debugger
                var old = frame.current_node
                frame.current_node = n
 
-               steps_fun_call(n)
+               if not self.autocontinue then
+                       if not n isa ABlockExpr then
+                               steps_fun_call(n)
+
+                               breakpoint_check(n)
 
-               breakpoint_check(n)
+                               check_funcall_and_traced_args(n)
+
+                               remap(n)
+
+                               check_if_vars_are_traced(n)
+                       end
+               end
 
                n.stmt(self)
                frame.current_node = old
@@ -142,6 +186,65 @@ class Debugger
                end
        end
 
+       # Check if a variable of current expression is traced
+       # Then prints and/or breaks for command prompt
+       private fun check_if_vars_are_traced(n: AExpr)
+       do
+               var identifiers_in_instruction = get_identifiers_in_current_instruction(n.location.text)
+
+               for i in identifiers_in_instruction do
+                       var variable = seek_variable(i, frame)
+                       for j in self.traces do
+                               if j.is_variable_traced_in_frame(i, frame) then
+                                       n.debug("Traced variable {i} used")
+                                       if j.break_on_encounter then while process_debug_command(gets) do end
+                                       break
+                               end
+                       end
+               end
+       end
+
+       # Function remapping all the traced objects to match their name in the local context
+       private fun remap(n: AExpr)
+       do
+               if self.aftermath then
+
+                       # Trace every argument variable pre-specified
+                       if self.frame_count_aftermath < frames.length and fun_call_arguments_positions.length > 0 then
+
+                               var ids_in_fun_def = get_identifiers_in_current_instruction(get_function_arguments(frame.mpropdef.location.text))
+
+                               for i in fun_call_arguments_positions.keys do
+                                       self.fun_call_arguments_positions[i].add_frame_variable(frame, ids_in_fun_def[i])
+                               end
+                       end
+
+                       self.aftermath = false
+               end
+       end
+
+       # If the current instruction is a function call
+       # We analyse its signature and the position of traced arguments if the call
+       # For future remapping when inside the function
+       private fun check_funcall_and_traced_args(n: AExpr) do
+               # If we have a function call, we need to see if any of the arguments is traced (including the caller)
+               # if it is, next time we face an instruction, we'll trace the local version on the traced variable in the next frame
+               if n isa ACallExpr then
+                       self.aftermath = true
+                       self.frame_count_aftermath = frames.length
+                       fun_call_arguments_positions.clear
+                       var fun_arguments = get_identifiers_in_current_instruction(get_function_arguments(n.location.text))
+
+                       for i in self.traces do
+                               for j in [0 .. fun_arguments.length - 1] do
+                                       if i.is_variable_traced_in_frame(fun_arguments[j],frame) then
+                                               fun_call_arguments_positions[j] = i
+                                       end
+                               end
+                       end
+               end
+       end
+
        #######################################################################
        ##                   Processing commands functions                   ##
        #######################################################################
@@ -192,6 +295,15 @@ class Debugger
                        else if parts_of_command.length == 3 and parts_of_command[1] == "as"
                        then
                                add_alias(parts_of_command[0], parts_of_command[2])
+                       # Modifies the value of a variable in the current frame
+                       else if parts_of_command.length >= 3 and parts_of_command[1] == "=" then
+                               process_mod_function(parts_of_command)
+                       # Traces the modifications on a variable
+                       else if parts_of_command.length >= 2 and parts_of_command[0] == "trace" then
+                               process_trace_command(parts_of_command)
+                       # Untraces the modifications on a variable
+                       else if parts_of_command.length == 2 and parts_of_command[0] == "untrace" then
+                               process_untrace_command(parts_of_command)
                        # Lists all the commands available
                        else
                                list_commands
@@ -259,6 +371,8 @@ class Debugger
                        end
 
                        print "\nEnd of current instruction \n"
+               else if parts_of_command[1] == "stack" then
+                       print self.stack_trace
                else if parts_of_command[1].has('[') and parts_of_command[1].has(']') then
                        process_array_command(parts_of_command)
                else
@@ -338,6 +452,220 @@ class Debugger
                end
        end
 
+       # Processes the modification function to modify a variable dynamically
+       #
+       # Command of type variable = value
+       fun process_mod_function(parts_of_command: Array[String])
+       do
+               parts_of_command[0] = get_real_variable_name(parts_of_command[0])
+               var parts_of_variable = parts_of_command[0].split_with(".")
+
+               if parts_of_variable.length > 1 then
+                       var last_part = parts_of_variable.pop
+                       var first_part = parts_of_command[0].substring(0,parts_of_command[0].length - last_part.length - 1)
+                       var papa = seek_variable(first_part, frame)
+
+                       if papa != null and papa isa MutableInstance then
+                               var attribute = get_attribute_in_mutable_instance(papa, last_part)
+
+                               if attribute != null then
+                                       modify_argument_of_complex_type(papa, attribute, parts_of_command[2])
+                               end
+                       end
+               else
+                       var target = seek_variable(parts_of_variable[0], frame)
+                       if target != null then
+                               modify_in_frame(target, parts_of_command[2])
+                       end
+               end
+       end
+
+       # Processes the untrace variable command
+       #
+       # Command pattern : "untrace variable"
+       fun process_untrace_command(parts_of_command: Array[String])
+       do
+               var variable_name = get_real_variable_name(parts_of_command[1])
+               if untrace_variable(variable_name) then
+                       print "Untraced variable {parts_of_command[1]}"
+               else
+                       print "{parts_of_command[1]} is not traced"
+               end
+       end
+
+       # Processes the trace variable command
+       #
+       # Command pattern : "trace variable [break/print]"
+       fun process_trace_command(parts_of_command: Array[String])
+       do
+               var variable_name = get_real_variable_name(parts_of_command[1])
+               var breaker:Bool
+
+               if seek_variable(variable_name, frame) == null then
+                       print "Cannot find a variable called {parts_of_command[1]}"
+                       return
+               end
+
+               if parts_of_command.length == 3 then
+                       if parts_of_command[2] == "break" then
+                               breaker = true
+                       else
+                               breaker = false
+                       end
+               else
+                       breaker = false
+               end
+
+               trace_variable(variable_name, breaker)
+
+               print "Successfully tracing {parts_of_command[1]}"
+       end
+
+       #######################################################################
+       ##                    Trace Management functions                     ##
+       #######################################################################
+
+       # Effectively untraces the variable called *variable_name*
+       #
+       # Returns true if the variable exists, false otherwise
+       private fun untrace_variable(variable_name: String): Bool
+       do
+               var to_remove: nullable TraceObject = null
+               for i in self.traces do
+                       if i.is_variable_traced_in_frame(variable_name, frame) then
+                               to_remove = i
+                       end
+               end
+
+               if to_remove != null then
+                       self.traces.remove(to_remove)
+                       return true
+               else
+                       return false
+               end
+       end
+
+       # Effectively traces the variable *variable_name* either in print or break mode depending on the value of breaker (break if true, print if false)
+       #
+       private fun trace_variable(variable_name: String, breaker: Bool)
+       do
+               for i in self.traces do
+                       if i.is_variable_traced_in_frame(variable_name, frame) then
+                               print "This variable is already traced"
+                               return
+                       end
+               end
+
+               var trace_object: TraceObject
+
+               if breaker then
+                       trace_object = new TraceObject(true)
+               else
+                       trace_object = new TraceObject(false)
+               end
+
+               # We trace the current variable found for the current frame
+               trace_object.add_frame_variable(self.frame, variable_name)
+
+               var position_of_variable_in_arguments = get_position_of_variable_in_arguments(frame, variable_name)
+
+               # Start parsing the frames starting with the parent of the current one, until the highest
+               # When the variable traced is declared locally, the loop stops
+               for i in [1 .. frames.length-1] do
+
+                       # If the variable was reported to be an argument of the previous frame
+                       if position_of_variable_in_arguments != -1 then
+
+                               var local_name = get_identifiers_in_current_instruction(get_function_arguments(frames[i].current_node.location.text))[position_of_variable_in_arguments]
+
+                               position_of_variable_in_arguments = get_position_of_variable_in_arguments(frames[i], local_name)
+
+                               trace_object.add_frame_variable(frames[i], local_name)
+                       else
+                               break
+                       end
+               end
+
+               self.traces.add(trace_object)
+       end
+
+       # If the variable *variable_name* is an argument of the function being executed in the frame *frame*
+       # The function returns its position in the arguments
+       # Else, it returns -1
+       private fun get_position_of_variable_in_arguments(frame: Frame, variable_name: String): Int
+       do
+               var identifiers = get_identifiers_in_current_instruction(get_function_arguments(frame.mpropdef.location.text))
+               for i in [0 .. identifiers.length-1] do
+                       # If the current traced variable is an argument of the current function, we trace its parent (at least)
+                       if identifiers[i] == variable_name then return i
+               end
+               return -1
+       end
+
+       # Gets all the identifiers of an instruction (uses the rules of Nit as of Mar 05 2013)
+       #
+       fun get_identifiers_in_current_instruction(instruction: AbstractString): Array[String]
+       do
+               var result_array = new Array[String]
+               var instruction_buffer = new Buffer
+
+               var trigger_char_escape = false
+               var trigger_string_escape = false
+               var trigger_concat_in_string = false
+
+               for i in instruction do
+                       if trigger_char_escape then
+                               if i == '\'' then trigger_char_escape = false
+                       else if trigger_string_escape then
+                               if i == '{' then
+                                       trigger_concat_in_string = true
+                                       trigger_string_escape = false
+                               else if i == '\"' then trigger_string_escape = false
+                       else
+                               if i.is_alphanumeric or i == '_' then
+                                       instruction_buffer.add(i)
+                               else if i == '.' then
+                                       if instruction_buffer.is_numeric or (instruction_buffer[0] >= 'A' and instruction_buffer[0] <= 'Z') then
+                                               instruction_buffer.clear
+                                       else
+                                               result_array.push(instruction_buffer.to_s)
+                                               instruction_buffer.add(i)
+                                       end
+                               else if i == '\'' then
+                                       trigger_char_escape = true
+                               else if i == '\"' then
+                                       trigger_string_escape = true
+                               else if i == '}' then
+                                       trigger_concat_in_string = false
+                                       trigger_string_escape = true
+                               else
+                                       if instruction_buffer.length > 0 and not instruction_buffer.is_numeric and not (instruction_buffer[0] >= 'A' and instruction_buffer[0] <= 'Z') then result_array.push(instruction_buffer.to_s)
+                                       instruction_buffer.clear
+                               end
+                       end
+               end
+
+               if instruction_buffer.length > 0 and not instruction_buffer.is_numeric and not (instruction_buffer[0] >= 'A' and instruction_buffer[0] <= 'Z') then result_array.push(instruction_buffer.to_s)
+
+               return result_array
+       end
+
+       # Takes a function call or declaration and strips all but the arguments
+       #
+       fun get_function_arguments(function: AbstractString): String
+       do
+               var buf = new Buffer
+               var trigger_copy = false
+
+               for i in function do
+                       if i == ')' then break
+                       if trigger_copy then buf.add(i)
+                       if i == '(' then trigger_copy = true
+               end
+
+               return buf.to_s
+       end
+
        #######################################################################
        ##                    Alias management functions                     ##
        #######################################################################
@@ -719,9 +1047,58 @@ class Debugger
        end
 
        #######################################################################
+       ##                  Runtime modification functions                   ##
+       #######################################################################
+
+       # Modifies the value of a variable contained in a frame
+       fun modify_in_frame(variable: Instance, value: String)
+       do
+               var new_variable = get_variable_of_type_with_value(variable.mtype.to_s, value)
+               if new_variable != null
+               then
+                       var keys = frame.map.keys
+                       for key in keys
+                       do
+                               if frame.map[key] == variable
+                               then
+                                       frame.map[key] = new_variable
+                               end
+                       end
+               end
+       end
+
+       # Modifies the value of a variable contained in a MutableInstance
+       fun modify_argument_of_complex_type(papa: MutableInstance, attribute: MAttribute, value: String)
+       do
+               var final_variable = papa.attributes[attribute]
+               var type_of_variable = final_variable.mtype.to_s
+               var new_variable = get_variable_of_type_with_value(type_of_variable, value)
+               if new_variable != null
+               then
+                       papa.attributes[attribute] = new_variable
+               end
+       end
+
+       #######################################################################
        ##                   Variable generator functions                    ##
        #######################################################################
 
+       # Returns a new variable of the type 'type_of_variable' with a value of 'value'
+       fun get_variable_of_type_with_value(type_of_variable: String, value: String): nullable Instance
+       do
+               if type_of_variable == "Int" then
+                       return get_int(value)
+               else if type_of_variable == "Float" then
+                       return get_float(value)
+               else if type_of_variable == "Bool" then
+                       return get_bool(value)
+               else if type_of_variable == "Char" then
+                       return get_char(value)
+               end
+
+               return null
+       end
+
        # Returns a new int instance with value 'value'
        fun get_int(value: String): nullable Instance
        do
@@ -752,6 +1129,19 @@ class Debugger
                end
        end
 
+       # Returns a new bool instance with value 'value'
+       fun get_bool(value: String): nullable Instance
+       do
+               if value.to_lower == "true" then
+                       return self.true_instance
+               else if value.to_lower == "false" then
+                       return self.false_instance
+               else
+                       print "Invalid value, a boolean must be set at \"true\" or \"false\""
+                       return null
+               end
+       end
+
        #######################################################################
        ##                     Command listing function                      ##
        #######################################################################
@@ -764,14 +1154,71 @@ class Debugger
                print "[break/b] line : Adds a breakpoint on line *line_nb* of the current file\n"
                print "[break/b] file_name line_nb : Adds a breakpoint on line *line_nb* of file *file_name* \n"
                print "[p/print] variable : [p/print] * shows the status of all the variables\n"
+               print "[p/print] variable[i] : Prints the value of the variable contained at position *i* in SequenceRead collection *variable*\n"
+               print "[p/print] variable[i..j]: Prints the value of all the variables contained between positions *i* and *j* in SequenceRead collection *variable*\n"
+               print "[p/print] stack: Prints a stack trace at current instruction\n"
+               print "Note : The arrays can be multi-dimensional (Ex : variable[i..j][k] will print all the values at position *k* of all the SequenceRead collections contained between positions *i* and *j* in SequenceRead collection *variable*)\n"
                print "s : steps in on the current function\n"
                print "n : steps-over the current instruction\n"
                print "finish : steps out of the current function\n"
                print "variable as alias : Adds an alias called *alias* for the variable *variable*"
                print "An alias can reference another alias\n"
+               print "variable = value : Sets the value of *variable* to *value*\n"
                print "[d/delete] line_nb : Removes a breakpoint on line *line_nb* of the current file \n"
                print "[d/delete] file_name line_nb : Removes a breakpoint on line *line_nb* of file *file_name* \n"
+               print "trace variable_name [break/print] : Traces the uses of the variable you chose to trace by printing the statement it appears in or by breaking on each use."
+               print "untrace variable_name : Removes the trace on the variable you chose to trace earlier in the program"
                print "kill : kills the current program (Exits with an error and stack trace)\n"
        end
 
 end
+
+# Traces the modifications of an object linked to a certain frame
+private class TraceObject
+
+       # Map of the local names bound to a frame
+       var trace_map: HashMap[Frame, String]
+       # Decides if breaking or printing statement when the variable is encountered
+       var break_on_encounter: Bool
+
+       init(break_on_encounter: Bool)
+       do
+               trace_map = new HashMap[Frame, String]
+               self.break_on_encounter = break_on_encounter
+       end
+
+       # Adds the local alias for a variable and the frame bound to it
+       fun add_frame_variable(frame: Frame, variable_name: String)
+       do
+               self.trace_map[frame] = variable_name
+       end
+
+       # Checks if the prompted variable is traced in the specified frame
+       fun is_variable_traced_in_frame(variable_name: String, frame: Frame): Bool
+       do
+               if self.trace_map.has_key(frame) then
+                       if self.trace_map[frame] == variable_name then
+                               return true
+                       end
+               end
+
+               return false
+       end
+
+end
+
+redef class ANode
+
+       # Breaks automatically when encountering an error
+       # Permits the injunction of commands before crashing
+       redef private fun fatal(v: NaiveInterpreter, message: String)
+       do
+               if v isa Debugger then
+                       print "An error was encountered, the program will stop now."
+                       self.debug(message)
+                       while v.process_debug_command(gets) do end
+               end
+
+               super
+       end
+end