ynerant/squinnondation
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ensure that a DataTLV is not too long

Browse Source 
Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
This commit is contained in:
Yohann D'ANELLO
2020-12-21 16:04:13 +01:00
parent b96ff488e7
commit 7a25d24ba3
3 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
squinnondation/hazel.py
View File

@ -109,7 +109,10 @@ class Squirrel(Hazelnut):
self.refresh_input()
if not self.squinnondation.no_emoji:
self.refresh_emoji_pad()
key = self.squinnondation.screen.getkey(curses.LINES - 1, 3 + len(self.nickname) + self.input_index)
try:
key = self.squinnondation.screen.getkey(curses.LINES - 1, 3 + len(self.nickname) + self.input_index)
except curses.error:
continue
if key == "KEY_MOUSE":
try:
@ -169,7 +172,12 @@ class Squirrel(Hazelnut):
return
elif key != "\n":
# Insert the pressed key in the current message
self.input_buffer = self.input_buffer[:self.input_index] + key + self.input_buffer[self.input_index:]
new_buffer = self.input_buffer[:self.input_index] + key + self.input_buffer[self.input_index:]
if len(DataTLV.construct(f"<{self.nickname}> {new_buffer}")) > 255 - 8 - 4:
# The message is too long to be sent once. We don't allow the user to type any other character.
curses.beep()
return
self.input_buffer = new_buffer
self.input_index += 1
return