Scrub past history input before split

ruby · Dec 1, 2023 · c2c3bb7 · c2c3bb7
1 parent 4fedce9
commit c2c3bb7
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/lib/irb/history.rb b/lib/irb/history.rb
@@ -60,7 +60,7 @@ def save_history
         end
 
         File.open(history_file, (append_history ? 'a' : 'w'), 0o600, encoding: IRB.conf[:LC_MESSAGES]&.encoding) do |f|
-          hist = history.map{ |l| l.split("\n").join("\\\n") }
+          hist = history.map{ |l| l.scrub.split("\n").join("\\\n") }
           unless append_history
             begin
               hist = hist.last(num) if hist.size > num and num > 0

diff --git a/test/irb/test_history.rb b/test/irb/test_history.rb
@@ -148,6 +148,25 @@ def test_history_concurrent_use_not_present
       ENV["IRBRC"] = backup_irbrc
     end
 
+    def test_history_different_encodings
+      backup_default_external = Encoding.default_external
+      backup_lang = ENV["LANG"]
+      IRB.conf[:SAVE_HISTORY] = 2
+      ENV["LANG"] = "C"
+      Encoding.default_external = Encoding::US_ASCII
+      assert_history(<<~EXPECTED_HISTORY.encode(Encoding::US_ASCII), <<~INITIAL_HISTORY.encode(Encoding::UTF_8), <<~INPUT)
+        ????
+        exit
+      EXPECTED_HISTORY
+        😀
+      INITIAL_HISTORY
+        exit
+      INPUT
+    ensure
+      ENV["LANG"] = backup_lang
+      Encoding.default_external = backup_default_external
+    end
+
     private
 
     def history_concurrent_use_for_input_method(input_method)