Bug [#2110]. Limit text returned from WM_GETTEXT to length specified in wParam.

Changed GetTextLength to use same logic as GetText to ensure they agree.

1 files changed, 162 insertions, 0 deletions

diff --git a/test/win32Tests.py b/test/win32Tests.py
new file mode 100644
index 000000000..21a89dbed
--- /dev/null
+++ b/test/win32Tests.py
@@ -0,0 +1,162 @@
+# -*- coding: utf-8 -*-
+# Requires Python 2.7 or later
+
+# These are tests that run only on Win32 as they use Win32 SendMessage call
+# to send WM_* messages to Scintilla that are not implemented on other platforms.
+# These help Scintilla behave like a Win32 text control and can help screen readers,
+# for example.
+
+from __future__ import with_statement
+from __future__ import unicode_literals
+
+import codecs, ctypes, os, sys, unittest
+
+from MessageNumbers import msgs, sgsm
+
+import ctypes
+user32 = ctypes.windll.user32
+
+import XiteWin as Xite
+
+class TestWins(unittest.TestCase):
+
+	def setUp(self):
+		self.xite = Xite.xiteFrame
+		self.ed = self.xite.ed
+		self.sciHwnd = self.xite.sciHwnd
+		self.ed.ClearAll()
+		self.ed.EmptyUndoBuffer()
+		self.ed.SetCodePage(0)
+		self.ed.SetStatus(0)
+
+	# Helper methods
+
+	def Send(self, msg, w, l):
+		return user32.SendMessageW(self.sciHwnd, msgs[msg], w, l)
+
+	def GetTextLength(self):
+		return self.Send("WM_GETTEXTLENGTH", 0, 0)
+
+	def GetText(self, n, s):
+		# n = The maximum number of characters to be copied, including the terminating null character.
+		# returns the number of characters copied, not including the terminating null character
+		return self.Send("WM_GETTEXT", n, s)
+
+	def TextValue(self):
+		self.assertEquals(self.ed.GetStatus(), 0)
+		lenValue = self.GetTextLength()
+		lenValueWithNUL = lenValue + 1
+		value = ctypes.create_unicode_buffer(lenValueWithNUL)
+		lenData = self.GetText(lenValueWithNUL, value)
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.assertEquals(lenData, lenValue)
+		return value.value
+
+	def SetText(self, s):
+		return self.Send("WM_SETTEXT", 0, s)
+
+	# Tests
+
+	def testSetText(self):
+		self.SetText(b"ab")
+		self.assertEquals(self.ed.Length, 2)
+
+	def testGetTextLength(self):
+		self.SetText(b"ab")
+		self.assertEquals(self.GetTextLength(), 2)
+
+	def testGetText(self):
+		self.SetText(b"ab")
+		data = ctypes.create_unicode_buffer(100)
+		lenData = self.GetText(100, data)
+		self.assertEquals(lenData, 2)
+		self.assertEquals(len(data.value), 2)
+		self.assertEquals(data.value, "ab")
+
+	def testGetUTF8Text(self):
+		self.ed.SetCodePage(65001)
+		t = "å"
+		tu8 = t.encode("UTF-8")
+		self.SetText(tu8)
+		value = self.TextValue()
+		self.assertEquals(value, t)
+
+	def testGetBadUTF8Text(self):
+		self.ed.SetCodePage(65001)
+		tu8 = b't\xc2'
+		t = "t\xc2"
+		self.SetText(tu8)
+		value = self.TextValue()
+		self.assertEquals(len(value), 2)
+		self.assertEquals(value, t)
+
+	def testGetJISText(self):
+		self.ed.SetCodePage(932)
+		t = "\N{HIRAGANA LETTER KA}"
+		tu8 = t.encode("shift-jis")
+		self.SetText(tu8)
+		value = self.TextValue()
+		self.assertEquals(len(value), 1)
+		self.assertEquals(value, t)
+
+	def testGetBadJISText(self):
+		self.ed.SetCodePage(932)
+		# This is invalid Shift-JIS, surrounded by []
+		tu8 = b'[\x85\xff]'
+		# Win32 uses Katakana Middle Dot to indicate some invalid Shift-JIS text
+		# At other times \uF8F3 is used which is a private use area character
+		# See https://unicodebook.readthedocs.io/operating_systems.html
+		katakanaMiddleDot = '[\N{KATAKANA MIDDLE DOT}]'
+		privateBad = '[\uf8f3]'
+		self.SetText(tu8)
+		value = self.TextValue()
+		self.assertEquals(len(value), 3)
+		self.assertEquals(value, katakanaMiddleDot)
+		
+		# This is even less valid Shift-JIS
+		tu8 = b'[\xff]'
+		self.SetText(tu8)
+		value = self.TextValue()
+		self.assertEquals(len(value), 3)
+		self.assertEquals(value, privateBad)
+
+	def testGetTextLong(self):
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.SetText(b"ab")
+		data = ctypes.create_unicode_buffer(100)
+		lenData = self.GetText(4, data)
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.assertEquals(lenData, 2)
+		self.assertEquals(data.value, "ab")
+
+	def testGetTextShort(self):
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.SetText(b"ab")
+		data = ctypes.create_unicode_buffer(100)
+		lenData = self.GetText(2, data)
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.assertEquals(lenData, 1)
+		self.assertEquals(data.value, "a")
+
+	def testGetTextJustNUL(self):
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.SetText(b"ab")
+		data = ctypes.create_unicode_buffer(100)
+		lenData = self.GetText(1, data)
+		self.assertEquals(self.ed.GetStatus(), 0)
+		#~ print(data)
+		self.assertEquals(lenData, 0)
+		self.assertEquals(data.value, "")
+
+	def testGetTextZeroLength(self):
+		self.assertEquals(self.ed.GetStatus(), 0)
+		self.SetText(b"ab")
+		data = ctypes.create_unicode_buffer(100)
+		lenData = self.GetText(0, data)
+		self.assertEquals(self.ed.GetStatus(), 0)
+		#~ print(data)
+		self.assertEquals(lenData, 0)
+		self.assertEquals(data.value, "")
+
+if __name__ == '__main__':
+	uu = Xite.main("win32Tests")