def _extractCSRFtoken(self, soup):
CSRFtoken = soup.find('meta', attrs={'name': 'CSRFtoken'})
if CSRFtoken:
self.CSRFtoken = CSRFtoken['content']
else:
raise tsm_errors.TSMModemError(
"Expected CSRFtoken but didn't find one")
def _tryGet(self, modalFunction, errorMessage):
try:
successful, data = modalFunction()
if not successful:
self._authenticate(data)
successful, data = modalFunction()
if not successful:
raise tsm_errors.TSMModemError(errorMessage)
return data
except tsm_errors.TSMBase:
raise
except Exception as e:
raise tsm_errors.TSMModemError(f"{errorMessage} {type(e)} | {e}")
def _logout(self):
def doLogout():
response = self.session.post(f"{self.base_url}/login.lp", {
"do_signout": 1,
"CSRFtoken": self.CSRFtoken
},
timeout=HTTP_TIMEOUT)
self._extractCSRFtoken_html(response.text)
try:
try:
doLogout()
except requests.exceptions.HTTPError as e:
status = e.response.status_code
if status == 403:
self._getCSRFtoken()
doLogout()
else:
raise
except Exception as e:
raise tsm_errors.TSMModemError(
f"Error during logout: {type(e)} | {e}")
def _authenticate(self, soup=None):
def authRequest(data):
r = self.session.post(f"{self.base_url}/authenticate", {
**{
"CSRFtoken": self.CSRFtoken
},
**data
},
timeout=HTTP_TIMEOUT)
return r.json()
def firstAuthRequest(A):
def doFirstAuth():
return authRequest({"I": self.username, "A": A})
try:
r = doFirstAuth()
except requests.exceptions.HTTPError as ex:
status = ex.response.status_code
if status == 403:
self._getCSRFtoken()
r = doFirstAuth()
else:
raise
s = r.get('s')
B = r.get('B')
if not (s and B):
raise tsm_errors.TSMUsernameIncorrect('Username is incorrect')
return s, B
def secondAuthRequest(M):
r = authRequest({"M": M})
M = r.get('M')
if not M:
raise tsm_errors.TSMPasswordIncorrect('Password is incorrect')
def doAuth():
srp = tsm_srp.User(self.username, self.password)
A = srp.start_authentication()
s, B = firstAuthRequest(A)
M = srp.process_challenge(s, B)
secondAuthRequest(M)
try:
if soup:
self._extractCSRFtoken(soup)
elif not self.CSRFtoken:
self._getCSRFtoken()
try:
doAuth()
except tsm_errors.TSMPasswordIncorrect:
# Attempt authentication one more time if failed.
# This is due to an extremely low probability error that occurs in srp.py.
doAuth()
except tsm_errors.TSMPasswordIncorrect as e:
raise e from None
except tsm_errors.TSMBase:
raise
except Exception as e:
raise tsm_errors.TSMModemError(
f"Error during authentication: {type(e)} | {e}")
def _authenticate(self, soup=None):
def authRequest(data):
r = self.session.post(
f"{self.base_url}/authenticate",
{**{"CSRFtoken": self.CSRFtoken}, **data},
timeout=HTTP_TIMEOUT
)
return r.json()
def firstAuthRequest(A):
def doFirstAuth():
return authRequest({"I": self.username, "A": A})
try:
r = doFirstAuth()
except requests.exceptions.HTTPError as ex:
status = ex.response.status_code
if status == 403:
self._getCSRFtoken()
r = doFirstAuth()
else:
raise
s = r.get('s')
B = r.get('B')
if not (s and B):
raise tsm_errors.TSMUsernameIncorrect('Username is incorrect')
return s, B
def secondAuthRequest(M):
r = authRequest({"M": M})
M = r.get('M')
if not M:
raise tsm_errors.TSMPasswordIncorrect('Password is incorrect')
def doAuth():
srp = tsm_srp.User(self.username, self.password)
A = srp.start_authentication()
s, B = firstAuthRequest(A)
M = srp.process_challenge(s, B)
secondAuthRequest(M)
def _utf8_decode(utftext):
s = ''
i = 0
while i < len(utftext):
c = ord(utftext[i])
if c < 128:
s = s + chr(c)
i = i + 1
elif c > 191 and c < 224:
c2 = ord(utftext[i + 1])
s = s + chr(((c & 31) << 6) | (c2 & 63))
i = i + 2
else:
c2 = ord(utftext[i + 1])
c3 = ord(utftext[i + 2])
s = s + chr(((c & 15) << 12) | ((c2 & 63) << 6) | (c3 & 63))
i = i + 3
return s
def decode(s):
_keyStr = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/='
output = ''
i = 0
#replace all chars not in _keyStr
s = re.sub(r'[^' + _keyStr + ']', '', s)
while i < len(s):
enc1 = _keyStr.find(s[i:i+1])
i = i + 1
enc2 = _keyStr.find(s[i:i+1])
i = i + 1
enc3 = _keyStr.find(s[i:i+1])
i = i + 1
enc4 = _keyStr.find(s[i:i+1])
i = i + 1
chr1 = (enc1 << 2) | (enc2 >> 4)
chr2 = ((enc2 & 15) << 4) | (enc3 >> 2)
chr3 = ((enc3 & 3) << 6) | enc4
output = output + chr(chr1)
if enc3 != 64:
output = output + chr(chr2)
if enc4 != 64:
output = output + chr(chr3)
output = _utf8_decode(output)
return output
def getLH1000Token():
response = self.session.get(f"{self.base_url}/login.htm")
if response.status_code == 200:
bs = bs4.BeautifulSoup(response.text, 'html.parser')
self.LH1000Post['httoken'] = decode(bs.find('img', src=re.compile('data:'))['src'][78:])
try:
if soup:
self._extractCSRFtoken(soup)
elif not self.CSRFtoken:
self._getCSRFtoken()
try:
doAuth()
except tsm_errors.TSMPasswordIncorrect:
# Attempt authentication one more time if failed.
# This is due to an extremely low probability error that occurs in srp.py.
doAuth()
except tsm_errors.TSMPasswordIncorrect as e:
raise e from None
except tsm_errors.TSMNoToken:
getLH1000Token()
except tsm_errors.TSMBase:
raise
except Exception as e:
raise tsm_errors.TSMModemError(f"Error during authentication: {type(e)} | {e}")