@agentunion/kite 1.3.2 → 1.5.0

package/extensions/services/proxy/agentcp/base/auth_client.py

@@ -0,0 +1,257 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 AgentUnion Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from typing import Union
+import uuid
+import time
+import requests
+from cryptography import x509
+from cryptography.hazmat.primitives import serialization, hashes
+from cryptography.hazmat.primitives.asymmetric import ec
+from cryptography.x509.oid import NameOID
+from agentcp.ca.ca_root import CARoot
+from agentcp.base.log import log_error, log_exception, log_info, log_warning
+from cryptography.hazmat.backends import default_backend
+import os
+
+
+class AuthClient:
+
+    success_crt_list = []
+
+    # HTTP 请求超时配置 (连接超时, 读取超时)
+    HTTP_TIMEOUT = (3, 10)
+
+    def __init__(self, agent_id: str, server_url: str, aid_path: str, seed_password: str):
+        """认证客户端类
+        Args:
+            agent_id: 代理ID
+            server_url: 服务器URL
+        """
+        self.agent_id = agent_id
+        self.server_url = server_url
+        self.signature = None
+        self.aid_path = aid_path
+        self.seed_password = seed_password
+
+    def sign_in(self, max_retry_num: int = 10) -> Union[dict, None]:
+        """登录方法，使用循环重试，失败返回 None"""
+        for retry_count in range(max_retry_num + 1):
+            try:
+                if retry_count > 0:
+                    # 指数退避：2s, 4s, 6s, ..., 最大 30s
+                    backoff = min(2 * retry_count, 30)
+                    log_info(f"Sign in retry {retry_count}/{max_retry_num}, waiting {backoff}s...")
+                    time.sleep(backoff)
+
+                hb_url = self.server_url + "/sign_in"
+                log_info(f"Sign in: {hb_url}")
+                request_id = uuid.uuid4().hex
+                data = {
+                    "agent_id": self.agent_id,
+                    "request_id": request_id,
+                }
+                headers = {
+                    'User-Agent': f'AgentCP/{__import__("agentcp").__version__} (AuthClient; {self.agent_id})'
+                }
+                response = requests.post(hb_url, json=data, verify=False, headers=headers, proxies={}, timeout=self.HTTP_TIMEOUT)
+
+                if response.status_code == 200:
+                    log_info(f"Sign in url: {hb_url}, response: {response.json()}")
+                    aid_path = os.path.join(self.aid_path, self.agent_id + ".key")
+                    private_key = self.__load_private_key(aid_path)
+                    if private_key is None:
+                        raise Exception("私钥加载失败,请检查加密种子是否一致")
+
+                    aid_path = os.path.join(self.aid_path, self.agent_id + ".crt")
+                    with open(aid_path, "rb") as f:
+                        certificate_pem = f.read().decode('utf-8')
+
+                    cert = x509.load_pem_x509_certificate(certificate_pem.encode('utf-8'))
+                    public_key = cert.public_key()
+                    public_key_pem = public_key.public_bytes(
+                        encoding=serialization.Encoding.PEM,
+                        format=serialization.PublicFormat.SubjectPublicKeyInfo
+                    ).decode('utf-8')
+
+                    if "nonce" in response.json():
+                        nonce = response.json()["nonce"]
+                        server_cert_valid = True
+
+                        if "cert" in response.json() and "signature" in response.json():
+                            server_cert_valid = False
+                            try:
+                                cert_data = response.json()["cert"].encode('utf-8')
+                                server_cert = x509.load_pem_x509_certificate(cert_data, default_backend())
+                                server_public_key = server_cert.public_key()
+                                data_to_verify = (self.agent_id + str(request_id)).lower().encode('utf-8')
+                                signature_bytes = bytes.fromhex(response.json()["signature"])
+                                server_public_key.verify(
+                                    signature_bytes,
+                                    data_to_verify,
+                                    ec.ECDSA(hashes.SHA256())
+                                )
+                                log_info("服务器签名验证成功")
+                                server_cert_valid = self.__check_server_cert(server_cert)
+                                if not server_cert_valid:
+                                    raise Exception("服务器证书验证失败")
+                            except Exception as e:
+                                server_cert_valid = False
+                                raise Exception(f"{e}")
+
+                            if not server_cert_valid:
+                                raise Exception("服务器证书是无效证书，请注意通信安全")
+
+                        if nonce:
+                            signature = private_key.sign(
+                                nonce.encode('utf-8'),
+                                ec.ECDSA(hashes.SHA256())
+                            )
+                            data = {
+                                "agent_id": self.agent_id,
+                                "request_id": request_id,
+                                "nonce": nonce,
+                                "public_key": public_key_pem,
+                                "cert": certificate_pem,
+                                "signature": signature.hex(),
+                            }
+                            response = requests.post(hb_url, json=data, verify=False, headers=headers, proxies={}, timeout=self.HTTP_TIMEOUT)
+                            if response.status_code == 200:
+                                result = response.json()
+                                self.signature = result.get("signature")
+                                log_info("Sign in successful")
+                                return result
+                            else:
+                                log_error(f"Sign in FAILED: {response.status_code} - {response.json().get('error', '')}")
+                else:
+                    log_error(f"Sign in failed: {response.status_code} - {response.json().get('error', '')}")
+
+            except Exception as e:
+                log_warning(f"Sign in exception (retry {retry_count}/{max_retry_num}): {e}")
+
+        # 所有重试都失败
+        log_error(f"Sign in failed after {max_retry_num} retries, please check network connection")
+        return None
+            
+            
+    def __load_private_key(self, aid_path):
+        try:
+            with open(aid_path, "rb") as f:
+                private_key = serialization.load_pem_private_key(
+                    f.read(),
+                    password=self.seed_password.encode('utf-8'),
+                )
+            return private_key
+        except Exception as e:
+            # 兼容性代码，按照不加密获取 private_key
+            return None
+
+    def sign_out(self) -> None:
+        """登出方法"""
+        try:
+            if self.signature is None:
+                return
+            hb_url = self.server_url + "/sign_out"
+            data = {
+                "agent_id": self.agent_id,
+                "signature": self.signature,
+            }
+            headers = {
+                'User-Agent': f'AgentCP/{__import__("agentcp").__version__} (AuthClient; {self.agent_id})'
+            }
+            response = requests.post(hb_url, json=data, verify=False, headers=headers, proxies={}, timeout=self.HTTP_TIMEOUT)
+            if response.status_code == 200:
+                log_info(f"Sign out OK: {response.json()}")
+            else:
+                log_error(f"Sign out failed: {response.json()}")
+        except Exception as e:
+            log_exception("Sign out exception")
+            
+    def __check_server_cert(self, server_cert):
+        try:
+            # 尝试获取主体的组织及 common name
+            try:
+                organization = server_cert.subject.get_attributes_for_oid(NameOID.ORGANIZATION_NAME)
+                common_name = server_cert.subject.get_attributes_for_oid(NameOID.COMMON_NAME)
+                if organization:
+                    log_info(f"__check_server_cert 证书的组织名称为: {organization[0].value}")
+                else:
+                    log_info("__check_server_cert 未找到证书的组织名称")
+                if common_name:
+                    log_info(f"__check_server_cert 证书的通用名称为: {common_name[0].value}")
+                else:
+                    log_info("__check_server_cert 未找到证书的通用名称")
+            except AttributeError:
+                log_error("获取证书主体信息时出错，可能证书对象不存在或格式不正确。")
+                return False
+            aia_ext = server_cert.extensions.get_extension_for_class(x509.AuthorityInformationAccess)
+            log_info(f"aia_ext:{aia_ext}")
+            issuer_url = None
+            for desc in aia_ext.value:
+                if desc.access_method == x509.oid.AuthorityInformationAccessOID.CA_ISSUERS:
+                    issuer_url = desc.access_location.value
+                    break
+    
+            if issuer_url:
+                log_info(f"证书颁发者 URL: {issuer_url}")
+                if issuer_url in AuthClient.success_crt_list:
+                    log_info(f"证书之前验证成功 {issuer_url}")
+                    return True
+                try:
+                    issuer_response = requests.get(issuer_url, verify=False, proxies={}, timeout=self.HTTP_TIMEOUT)
+                    issuer_response.raise_for_status()
+                    #TODO:将证书内存以issuer_url为键值缓存在本地，避免重复下载和验证证书
+                    issuer_cert = x509.load_pem_x509_certificate(issuer_response.content, default_backend())
+                    # 验证服务器证书的有效性
+                    issuer_public_key = issuer_cert.public_key()
+                    issuer_public_key.verify(
+                        server_cert.signature,
+                        server_cert.tbs_certificate_bytes,
+                        ec.ECDSA(server_cert.signature_hash_algorithm)
+                    )
+                    #TODO标记证书是受信任的
+                    log_info(f"证书验证成功 {issuer_url}")
+                    AuthClient.success_crt_list.append(issuer_url)
+                    return self.__check_server_cert(issuer_cert)
+                except requests.RequestException as e:
+                    log_error(f"下载证书颁发者证书时出错: {e}")
+                    return False
+                except Exception as e:
+                    log_error(f"验证证书时出错: {e}")
+                    return False
+            else:
+                log_error("未找到证书颁发者信息")
+                return False
+        except x509.ExtensionNotFound:
+            log_error("证书中未包含 AIA 扩展信息")
+            try:
+                root_cert_pem = CARoot().get_ca_root_crt()
+                root_cert = x509.load_pem_x509_certificate(root_cert_pem.encode('utf-8'), default_backend())
+                root_public_key = root_cert.public_key()
+                root_public_key.verify(
+                    server_cert.signature,
+                    server_cert.tbs_certificate_bytes,
+                    ec.ECDSA(server_cert.signature_hash_algorithm)
+                )
+                log_info("用root.crt 证书进行验证成功")
+                return True
+            except FileNotFoundError:
+                log_error("未找到 root.crt 文件，请检查文件路径。")
+                return False
+            except Exception as e:
+                log_error(f"加载 root.crt 证书时出错: {e}")
+                return False
+        except Exception as e:
+            log_error(f"服务器证书验证失败: {e}")
+            return False

package/extensions/services/proxy/agentcp/base/client.py

@@ -0,0 +1,112 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 AgentUnion Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import abc
+import requests
+
+from agentcp.base.log import log_error, log_info
+from agentcp.utils.proxy_bypass import ensure_no_proxy_for_local_env, get_requests_proxies
+
+ensure_no_proxy_for_local_env()
+
+
+class IClient(abc.ABC):
+    def __init__(self):
+        self._agent_id_ref = None
+
+    def set_agent_id_ref(self, agent_id_ref):
+        """设置AgentID引用，用于获取代理配置"""
+        self._agent_id_ref = agent_id_ref
+
+    def _get_proxies(self, url: str):
+        """获取 requests 代理配置；agentcp 内部本地/远程一律直连。"""
+        return get_requests_proxies(False, url)
+
+    @abc.abstractmethod
+    def sign_in(self) -> bool:
+        pass
+
+    @abc.abstractmethod
+    def get_headers(self) -> dict:
+        pass
+
+    @abc.abstractmethod
+    def sign_out(self):
+        pass
+
+
+    def get_request(self,url: str, params: dict = None, headers: dict = None,is_retry=True) -> requests.Response:
+        """
+        发送GET请求的通用方法
+        Args:
+            url (str): 请求的URL
+            params (dict, optional): 请求参数. Defaults to None.
+            headers (dict, optional): 请求头. Defaults to None.
+        Returns:
+            requests.Response: 响应对象
+        """
+        try:
+            common_headers = self.get_headers()
+            if headers:
+                common_headers = {**common_headers, **headers}
+            proxies = self._get_proxies(url)
+            response = requests.get(url, params=params, headers=common_headers, verify=False, proxies=proxies)
+            response.raise_for_status()
+            if response.status_code == 200:
+                return response
+            else:
+                error = response.json().get("error", "")
+                log_error(f"请求失败，错误详情: {error}")
+                if response.status_code == 401 and is_retry:
+                    log_info("尝试重新登录...")
+                    if self.sign_in():
+                        return self.get_request(url, params=params, headers=common_headers, is_retry=False)
+                return response
+        except requests.exceptions.RequestException as e:
+            log_error(f"请求失败: {e}")
+            return None
+
+
+    def post_request(self, url: str, data: dict = None, json: dict = None, headers: dict = None, is_retry=True) -> requests.Response:
+        """
+        发送POST请求的通用方法
+
+        Args:
+            url (str): 请求的URL
+            data (dict, optional): 表单数据. Defaults to None.
+            json (dict, optional): JSON数据. Defaults to None.
+            headers (dict, optional): 请求头. Defaults to None.
+            is_retry (bool, optional): 是否重试. Defaults to True.
+        Returns:
+            requests.Response: 响应对象
+        """
+        try:
+            common_headers = self.get_headers()
+            if headers:
+                common_headers = {**common_headers, **headers}
+            proxies = self._get_proxies(url)
+            response = requests.post(url, data=data, json=json, headers=common_headers, verify=False, proxies=proxies)
+            if response.status_code == 200:
+                return response
+            else:
+                error = response.json().get("error", "")
+                log_error(f"请求失败，错误详情: {error}")
+                if response.status_code == 401 and is_retry:
+                    log_info("尝试重新登录...")
+                    if self.sign_in():
+                        return self.post_request(url, data=data, json=json, headers=common_headers, is_retry=False)
+            return response
+        except requests.exceptions.RequestException as e:
+            log_error(f"请求失败: {e}")
+            raise

package/extensions/services/proxy/agentcp/base/env.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Copyright 2025 AgentUnion Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import enum
+import os
+
+class Environ(enum.Enum):
+    """
+    Environment for the agent.
+    """
+    # Define the environments
+    ENTRY_SERVER = "ENTRY_SERVER"
+    LOG_LEVEL = "LOG_LEVEL"
+    CA_SERVER = "CA_SERVER"
+    
+    def __str__(self):
+        return self.value
+
+    def get(self, default=None):
+        """
+        Get the environment variable value.
+        """
+        return os.environ.get(self.value, default)