jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Check private key size for the final value

pull/26/head
Sheng 2018-09-09 15:33:59 +08:00
parent 8eb636d73b
commit 704d2b68e6
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
webssh/handler.py
View File

@ -90,16 +90,18 @@ class IndexHandler(MixinHandler, tornado.web.RequestHandler):
def get_privatekey(self): def get_privatekey(self):
name = 'privatekey' name = 'privatekey'
lst = self.request.files.get(name) # multipart form lst = self.request.files.get(name) # multipart form
if not lst: if lst:
return self.get_argument(name, u'') # urlencoded form
else:
self.filename = lst[0]['filename'] self.filename = lst[0]['filename']
data = lst[0]['body'] data = lst[0]['body']
if len(data) > KEY_MAX_SIZE: value = self.decode_argument(data, name=name).strip()
raise InvalidValueError( else:
'Invalid private key: {}'.format(self.filename) value = self.get_argument(name, u'') # urlencoded form
)
return self.decode_argument(data, name=name) if len(value) > KEY_MAX_SIZE:
raise InvalidValueError(
'Invalid private key: {}'.format(self.filename)
)
return value
@classmethod @classmethod
def get_specific_pkey(cls, pkeycls, privatekey, password): def get_specific_pkey(cls, pkeycls, privatekey, password):