tplinkrouterc6u 4.2.3__tar.gz → 5.0.1__tar.gz

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tplinkrouterc6u
-Version: 4.2.3
+Version: 5.0.1
 Summary: TP-Link Router API
 Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
 Author: Alex Erohin
@@ -42,6 +42,7 @@ from tplinkrouterc6u import (
     TplinkRouterProvider,
     TplinkRouter,
     TplinkC1200Router,
+    TplinkC5400XRouter,
     TPLinkMRClient,
     TPLinkDecoClient,
     Connection
@@ -53,10 +54,10 @@ router = TplinkRouterProvider.get_client('http://192.168.0.1', 'password')
 # router = TplinkRouter('http://192.168.0.1', 'password')
 # You may also pass username if it is different and a logger to log errors as
 # router = TplinkRouter('http://192.168.0.1','password','admin2', Logger('test'))
-# If you have the TP-link C1200 V2 or similar, you can use the TplinkC1200Router class instead of the TplinkRouter class.
+# If you have the TP-link C5400X or similar, you can use the TplinkC5400XRouter class instead of the TplinkRouter class.
 # Remember that the password for this router is different, here you need to use the web encrypted password.
 # To get web encrypted password, read Web Encrypted Password section
-# router = TplinkC1200Router('http://192.168.0.1','WebEncryptedPassword', Logger('test'))
+# router = TplinkC5400XRouter('http://192.168.0.1','WebEncryptedPassword', Logger('test'))
 
 try:
     router.authorize()  # authorizing
@@ -87,8 +88,8 @@ The TP-Link Web Interface only supports upto 1 user logged in at a time (for sec
 So before action you need to authorize and after logout
 
 ### <a id="encrypted_pass">Web Encrypted Password</a>
-If you got exception - `You need to use web encrypted password instead. Check the documentation!`
-or you have TP-link C1200 V2 or similar router you need to get web encrypted password by these actions:
+If you got exception - `use web encrypted password instead. Check the documentation!`
+or you have TP-link C5400X or similar router you need to get web encrypted password by these actions:
 1. Go to the login page of your router. (default: 192.168.0.1).
 2. Type in the password you use to login into the password field.
 3. Click somewhere else on the page so that the password field is not selected anymore.
@@ -246,8 +247,9 @@ or you have TP-link C1200 V2 or similar router you need to get web encrypted pas
 - Archer AX11000 V1
 - Archer BE800 v1.0
 - Archer BE805 v1.0
-- Archer C1200 (v1.0, v2.0) (You need to use [web encrypted password](#encrypted_pass))
-- Archer C2300 v1.0 (You need to use [web encrypted password](#encrypted_pass))
+- Archer BE3600 1.6
+- Archer C1200 (v1.0, v2.0)
+- Archer C2300 (v1.0, v2.0)
 - Archer C6 (v2.0, v3.0)
 - Archer C6U v1.0
 - Archer C7 (v4.0, v5.0)

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/README.md

@@ -20,6 +20,7 @@ from tplinkrouterc6u import (
     TplinkRouterProvider,
     TplinkRouter,
     TplinkC1200Router,
+    TplinkC5400XRouter,
     TPLinkMRClient,
     TPLinkDecoClient,
     Connection
@@ -31,10 +32,10 @@ router = TplinkRouterProvider.get_client('http://192.168.0.1', 'password')
 # router = TplinkRouter('http://192.168.0.1', 'password')
 # You may also pass username if it is different and a logger to log errors as
 # router = TplinkRouter('http://192.168.0.1','password','admin2', Logger('test'))
-# If you have the TP-link C1200 V2 or similar, you can use the TplinkC1200Router class instead of the TplinkRouter class.
+# If you have the TP-link C5400X or similar, you can use the TplinkC5400XRouter class instead of the TplinkRouter class.
 # Remember that the password for this router is different, here you need to use the web encrypted password.
 # To get web encrypted password, read Web Encrypted Password section
-# router = TplinkC1200Router('http://192.168.0.1','WebEncryptedPassword', Logger('test'))
+# router = TplinkC5400XRouter('http://192.168.0.1','WebEncryptedPassword', Logger('test'))
 
 try:
     router.authorize()  # authorizing
@@ -65,8 +66,8 @@ The TP-Link Web Interface only supports upto 1 user logged in at a time (for sec
 So before action you need to authorize and after logout
 
 ### <a id="encrypted_pass">Web Encrypted Password</a>
-If you got exception - `You need to use web encrypted password instead. Check the documentation!`
-or you have TP-link C1200 V2 or similar router you need to get web encrypted password by these actions:
+If you got exception - `use web encrypted password instead. Check the documentation!`
+or you have TP-link C5400X or similar router you need to get web encrypted password by these actions:
 1. Go to the login page of your router. (default: 192.168.0.1).
 2. Type in the password you use to login into the password field.
 3. Click somewhere else on the page so that the password field is not selected anymore.
@@ -224,8 +225,9 @@ or you have TP-link C1200 V2 or similar router you need to get web encrypted pas
 - Archer AX11000 V1
 - Archer BE800 v1.0
 - Archer BE805 v1.0
-- Archer C1200 (v1.0, v2.0) (You need to use [web encrypted password](#encrypted_pass))
-- Archer C2300 v1.0 (You need to use [web encrypted password](#encrypted_pass))
+- Archer BE3600 1.6
+- Archer C1200 (v1.0, v2.0)
+- Archer C2300 (v1.0, v2.0)
 - Archer C6 (v2.0, v3.0)
 - Archer C6U v1.0
 - Archer C7 (v4.0, v5.0)

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="tplinkrouterc6u",
-    version="4.2.3",
+    version="5.0.1",
     author="Alex Erohin",
     author_email="alexanderErohin@yandex.ru",
     description="TP-Link Router API",

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/tplinkrouterc6u/__init__.py

@@ -2,6 +2,7 @@ from tplinkrouterc6u.client import (
     TplinkRouter,
     TplinkRouterProvider,
     TplinkC1200Router,
+    TplinkC5400XRouter,
     TPLinkMRClient,
     AbstractRouter,
     TPLinkDecoClient,

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/tplinkrouterc6u/client.py

@@ -14,7 +14,7 @@ from tplinkrouterc6u.helper import get_ip
 from tplinkrouterc6u.encryption import EncryptionWrapper, EncryptionWrapperMR
 from tplinkrouterc6u.package_enum import Connection
 from tplinkrouterc6u.dataclass import Firmware, Status, Device, IPv4Reservation, IPv4DHCPLease, IPv4Status
-from tplinkrouterc6u.exception import ClientException, ClientError
+from tplinkrouterc6u.exception import ClientException, ClientError, AuthorizeError
 from abc import ABC, abstractmethod
 
 
@@ -108,7 +108,7 @@ class TplinkRequest:
         error = ('TplinkRouter - {} - Response with error; Request {} - Response {}'
                  .format(self.__class__.__name__, path, data)) if not error else error
         if self._logger:
-            self._logger.error(error)
+            self._logger.debug(error)
         raise ClientError(error)
 
     def _is_valid_response(self, data: dict) -> bool:
@@ -178,7 +178,7 @@ class TplinkEncryption(TplinkRequest):
             error = ("TplinkRouter - {} - Cannot authorize! Error - {}; Response - {}"
                      .format(self.__class__.__name__, e, data))
             if self._logger:
-                self._logger.error(error)
+                self._logger.debug(error)
             raise ClientException(error)
 
     def _request_pwd(self) -> None:
@@ -203,7 +203,7 @@ class TplinkEncryption(TplinkRequest):
             error = ('TplinkRouter - {} - Unknown error for pwd! Error - {}; Response - {}'
                      .format(self.__class__.__name__, e, response.text))
             if self._logger:
-                self._logger.error(error)
+                self._logger.debug(error)
             raise ClientException(error)
 
     def _request_seq(self) -> None:
@@ -230,7 +230,7 @@ class TplinkEncryption(TplinkRequest):
             error = ('TplinkRouter - {} - Unknown error for seq! Error - {}; Response - {}'
                      .format(self.__class__.__name__, e, response.text))
             if self._logger:
-                self._logger.error(error)
+                self._logger.debug(error)
             raise ClientException(error)
 
     def _try_login(self) -> Response:
@@ -658,7 +658,7 @@ class TplinkC6V4Router(AbstractRouter):
             return False
         if response.status_code == 401 and response.text.startswith('00'):
             raise ClientException(('Your router is not supported. Please add your router support to '
-                                   'https://github.com/AlexandrErohin/TP-Link-Archer-C6U'
+                                   'https://github.com/AlexandrErohin/TP-Link-Archer-C6U '
                                    'by implementing methods for TplinkC6V4Router class'
                                    ))
         return False
@@ -682,9 +682,9 @@ class TplinkC6V4Router(AbstractRouter):
         raise ClientException('Not Implemented')
 
 
-class TplinkC1200Router(TplinkBaseRouter):
+class TplinkC5400XRouter(TplinkBaseRouter):
     def supports(self) -> bool:
-        return True
+        return len(self.password) >= 200
 
     def authorize(self) -> None:
         if len(self.password) < 200:
@@ -708,9 +708,9 @@ class TplinkC1200Router(TplinkBaseRouter):
             self._smart_network = False
 
         except Exception as e:
-            error = "TplinkRouter - C1200 - Cannot authorize! Error - {}; Response - {}".format(e, response.text)
+            error = "TplinkRouter - C5400X - Cannot authorize! Error - {}; Response - {}".format(e, response.text)
             if self._logger:
-                self._logger.error(error)
+                self._logger.debug(error)
             raise ClientException(error)
 
     def set_led(self, enable: bool) -> None:
@@ -786,6 +786,103 @@ class TplinkC1200Router(TplinkBaseRouter):
         self.request(path, data)
 
 
+class TplinkC1200Router(TplinkC5400XRouter):
+    username = ''
+    password = ''
+    _pwdNN = ''
+    _pwdEE = ''
+    _encryption = EncryptionWrapper()
+
+    def supports(self) -> bool:
+        if len(self.password) > 125:
+            return False
+
+        try:
+            self._request_pwd()
+            return True
+        except ClientException:
+            return False
+
+    def authorize(self) -> None:
+        if self._pwdNN == '':
+            self._request_pwd()
+
+        response = self._try_login()
+
+        is_valid_json = False
+        try:
+            response.json()
+            is_valid_json = True
+        except BaseException:
+            """Ignore"""
+
+        if is_valid_json is False or response.status_code == 403:
+            self._logged = False
+            self._request_pwd()
+            response = self._try_login()
+
+        data = response.text
+        try:
+            data = response.json()
+            data = self._decrypt_response(data)
+
+            self._stok = data[self._data_block]['stok']
+            regex_result = search(
+                'sysauth=(.*);', response.headers['set-cookie'])
+            self._sysauth = regex_result.group(1)
+            self._logged = True
+
+        except Exception as e:
+            error = ("TplinkRouter - C1200 - Cannot authorize! Error - {}; Response - {}".format(e, data))
+            if self._logger:
+                self._logger.debug(error)
+            if 'data' in vars() and data.get('errorcode') == 'login failed':
+                raise AuthorizeError(error)
+            raise ClientException(error)
+
+    def _request_pwd(self) -> None:
+        url = '{}/cgi-bin/luci/;stok=/login?form=login'.format(self.host)
+        response = post(
+            url, params={'operation': 'read'},
+            timeout=self.timeout,
+            verify=self._verify_ssl,
+        )
+
+        try:
+            data = response.json()
+
+            args = data[self._data_block]['password']
+
+            self._pwdNN = args[0]
+            self._pwdEE = args[1]
+
+        except Exception as e:
+            error = ('TplinkRouter - C1200 - {} - Unknown error for pwd! Error - {}; Response - {}'
+                     .format(self.__class__.__name__, e, response.text))
+            if self._logger:
+                self._logger.debug(error)
+            raise ClientException(error)
+
+    def _try_login(self) -> Response:
+        url = '{}/cgi-bin/luci/;stok=/login?form=login'.format(self.host)
+
+        crypted_pwd = self._encryption.encrypt_password_C1200(self.password, self._pwdNN, self._pwdEE)
+
+        body = self._get_login_data(crypted_pwd)
+
+        return post(
+            url,
+            data=body,
+            headers=self._headers_login,
+            timeout=self.timeout,
+            verify=self._verify_ssl,
+        )
+
+    @staticmethod
+    def _get_login_data(crypted_pwd: str) -> str:
+        return 'operation=login&password={}'.format(crypted_pwd)
+
+
 class TPLinkMRClient(AbstractRouter):
     REQUEST_RETRIES = 3
 
@@ -1104,7 +1201,7 @@ class TPLinkMRClient(AbstractRouter):
         if code != 200:
             error = 'TplinkRouter - MR -  Response with error; Request {} - Response {}'.format(data, response)
             if self._logger:
-                self._logger.error(error)
+                self._logger.debug(error)
             raise ClientError(error)
 
         result = self._merge_response(response)
@@ -1255,7 +1352,7 @@ class TPLinkMRClient(AbstractRouter):
 
         if error:
             if self._logger:
-                self._logger.error(error)
+                self._logger.debug(error)
             raise ClientException(error)
 
     def _request(self, url, method='POST', data_str=None, encrypt=False):
@@ -1334,10 +1431,19 @@ class TPLinkMRClient(AbstractRouter):
 class TplinkRouterProvider:
     @staticmethod
     def get_client(host: str, password: str, username: str = 'admin', logger: Logger = None,
-                   verify_ssl: bool = True, timeout: int = 30) -> AbstractRouter | None:
-        for client in [TPLinkMRClient, TplinkC6V4Router, TPLinkDecoClient, TplinkRouter, TplinkC1200Router]:
+                   verify_ssl: bool = True, timeout: int = 30) -> AbstractRouter:
+        for client in [TplinkC5400XRouter, TPLinkMRClient, TplinkC6V4Router, TPLinkDecoClient, TplinkRouter]:
             router = client(host, password, username, logger, verify_ssl, timeout)
             if router.supports():
                 return router
 
-        return None
+        router = TplinkC1200Router(host, password, username, logger, verify_ssl, timeout)
+        try:
+            router.authorize()
+        except AuthorizeError as e:
+            logger.error(e.__str__())
+            raise ClientException(('Login failed! Please check if your router local password is correct or '
+                                   'try to use web encrypted password instead. Check the documentation!'
+                                   ))
+
+        raise ClientException('You need to use web encrypted password instead. Check the documentation!')

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/tplinkrouterc6u/encryption.py

@@ -64,6 +64,24 @@ class EncryptionWrapper:
     def _get_aes_string(self) -> str:
         return 'k={}&i={}'.format(self._key.decode(), self._iv.decode())
 
+    @staticmethod
+    def encrypt_password_C1200(password: str, nn: str, ee: str) -> str:
+        n = int(nn, 16)
+        e = int(ee, 16)
+        key = RSA.construct((n, e))
+
+        modulus_byte_length = (key.size_in_bits() + 7) // 8
+        password_bytes = password.encode('utf-8')
+        if len(password_bytes) > modulus_byte_length:
+            raise ValueError("Password too long for the RSA key size.")
+
+        padded_password = password_bytes.ljust(modulus_byte_length, b'\x00')
+        message_int = bytes_to_long(padded_password)
+        encrypted_int = pow(message_int, e, n)
+        encrypted_hex = format(encrypted_int, 'x').zfill(256)
+
+        return encrypted_hex
+
 
 class EncryptionWrapperMR:
     RSA_USE_PKCS_V1_5 = False

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/tplinkrouterc6u/exception.py

@@ -4,3 +4,7 @@ class ClientException(Exception):
 
 class ClientError(ClientException):
     pass
+
+
+class AuthorizeError(ClientException):
+    pass

{tplinkrouterc6u-4.2.3 → tplinkrouterc6u-5.0.1}/tplinkrouterc6u.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tplinkrouterc6u
-Version: 4.2.3
+Version: 5.0.1
 Summary: TP-Link Router API
 Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
 Author: Alex Erohin
@@ -42,6 +42,7 @@ from tplinkrouterc6u import (
     TplinkRouterProvider,
     TplinkRouter,
     TplinkC1200Router,
+    TplinkC5400XRouter,
     TPLinkMRClient,
     TPLinkDecoClient,
     Connection
@@ -53,10 +54,10 @@ router = TplinkRouterProvider.get_client('http://192.168.0.1', 'password')
 # router = TplinkRouter('http://192.168.0.1', 'password')
 # You may also pass username if it is different and a logger to log errors as
 # router = TplinkRouter('http://192.168.0.1','password','admin2', Logger('test'))
-# If you have the TP-link C1200 V2 or similar, you can use the TplinkC1200Router class instead of the TplinkRouter class.
+# If you have the TP-link C5400X or similar, you can use the TplinkC5400XRouter class instead of the TplinkRouter class.
 # Remember that the password for this router is different, here you need to use the web encrypted password.
 # To get web encrypted password, read Web Encrypted Password section
-# router = TplinkC1200Router('http://192.168.0.1','WebEncryptedPassword', Logger('test'))
+# router = TplinkC5400XRouter('http://192.168.0.1','WebEncryptedPassword', Logger('test'))
 
 try:
     router.authorize()  # authorizing
@@ -87,8 +88,8 @@ The TP-Link Web Interface only supports upto 1 user logged in at a time (for sec
 So before action you need to authorize and after logout
 
 ### <a id="encrypted_pass">Web Encrypted Password</a>
-If you got exception - `You need to use web encrypted password instead. Check the documentation!`
-or you have TP-link C1200 V2 or similar router you need to get web encrypted password by these actions:
+If you got exception - `use web encrypted password instead. Check the documentation!`
+or you have TP-link C5400X or similar router you need to get web encrypted password by these actions:
 1. Go to the login page of your router. (default: 192.168.0.1).
 2. Type in the password you use to login into the password field.
 3. Click somewhere else on the page so that the password field is not selected anymore.
@@ -246,8 +247,9 @@ or you have TP-link C1200 V2 or similar router you need to get web encrypted pas
 - Archer AX11000 V1
 - Archer BE800 v1.0
 - Archer BE805 v1.0
-- Archer C1200 (v1.0, v2.0) (You need to use [web encrypted password](#encrypted_pass))
-- Archer C2300 v1.0 (You need to use [web encrypted password](#encrypted_pass))
+- Archer BE3600 1.6
+- Archer C1200 (v1.0, v2.0)
+- Archer C2300 (v1.0, v2.0)
 - Archer C6 (v2.0, v3.0)
 - Archer C6U v1.0
 - Archer C7 (v4.0, v5.0)