source ~/mcp-env/bin/activate



export ANTHROPIC_API_KEY="sk-ant-SEU-TOKEN-AQUI"



# Confirma que exportou corretamente

echo $ANTHROPIC_API_KEY

python3 << 'EOF'

import anthropic

client = anthropic.Anthropic()

msg = client.messages.create(

    model="claude-haiku-4-5-20251001",

    max_tokens=50,

    messages=[{"role": "user", "content": "responda só: ok"}]

)

print("✅ API OK:", msg.content[0].text)

EOF

import asyncio

import sys

from mcp import ClientSession, StdioServerParameters

from mcp.client.stdio import stdio_client

import anthropic



# ─────────────────────────────────────────

# CONFIGURAÇÃO

# ─────────────────────────────────────────



SERVER_SCRIPT = "/usr/local/scripts/pentest_server_v2.py"

MODEL = "claude-opus-4-5"



# ─────────────────────────────────────────

# CONVERTE TOOLS MCP -> FORMATO ANTHROPIC

# ─────────────────────────────────────────



def mcp_tools_to_anthropic(tools):

    result = []

    for tool in tools:

        result.append({

            "name": tool.name,

            "description": tool.description,

            "input_schema": tool.inputSchema

        })

    return result



# ─────────────────────────────────────────

# LOOP PRINCIPAL

# ─────────────────────────────────────────



async def chat_loop(session: ClientSession, anthropic_client: anthropic.Anthropic):

    tools_response = await session.list_tools()

    tools = mcp_tools_to_anthropic(tools_response.tools)



    print(f"\n✅ Conectado ao PentestServer")

    print(f"🔧 {len(tools)} tools disponíveis: {', '.join(t['name'] for t in tools)}")

    print(f"🤖 Modelo: {MODEL}")

    print("\n" + "─"*60)

    print("Digite sua pergunta ou comando. 'sair' para encerrar.")

    print("─"*60 + "\n")



    messages = []



    while True:

        try:

            user_input = input("Você: ").strip()

        except (KeyboardInterrupt, EOFError):

            print("\nEncerrando...")

            break



        if user_input.lower() in ("sair", "exit", "quit"):

            print("Até mais!")

            break



        if not user_input:

            continue



        messages.append({"role": "user", "content": user_input})



        response = anthropic_client.messages.create(

            model=MODEL,

            max_tokens=4096,

            system="""Você é um assistente especializado em pentest e segurança ofensiva.

Você tem acesso a tools de reconhecimento e busca de vulnerabilidades.

SEMPRE leia o resource 'file://escopo' antes de executar qualquer scan.

Seja objetivo e técnico. Apresente resultados em formato estruturado.""",

            tools=tools,

            messages=messages

        )



        while response.stop_reason == "tool_use":

            tool_calls = [b for b in response.content if b.type == "tool_use"]

            text_blocks = [b for b in response.content if b.type == "text"]



            for tb in text_blocks:

                if tb.text.strip():

                    print(f"\nClaude: {tb.text}")



            messages.append({"role": "assistant", "content": response.content})



            tool_results = []

            for tc in tool_calls:

                print(f"\n🔧 Chamando tool: {tc.name}")

                print(f"   Parâmetros: {tc.input}")



                try:

                    result = await session.call_tool(tc.name, tc.input)

                    result_text = result.content[0].text if result.content else "Sem resultado"

                    print(f"   ✅ Resultado obtido ({len(result_text)} chars)")

                except Exception as e:

                    result_text = f"Erro ao executar {tc.name}: {str(e)}"

                    print(f"   ❌ Erro: {e}")



                tool_results.append({

                    "type": "tool_result",

                    "tool_use_id": tc.id,

                    "content": result_text

                })



            messages.append({"role": "user", "content": tool_results})



            response = anthropic_client.messages.create(

                model=MODEL,

                max_tokens=4096,

                system="""Você é um assistente especializado em pentest e segurança ofensiva.

Você tem acesso a tools de reconhecimento e busca de vulnerabilidades.

SEMPRE leia o resource 'file://escopo' antes de executar qualquer scan.

Seja objetivo e técnico. Apresente resultados em formato estruturado.""",

                tools=tools,

                messages=messages

            )



        final_text = ""

        for block in response.content:

            if hasattr(block, "text"):

                final_text += block.text



        print(f"\nClaude: {final_text}\n")

        messages.append({"role": "assistant", "content": response.content})





# ─────────────────────────────────────────

# ENTRYPOINT

# ─────────────────────────────────────────



async def main():

    anthropic_client = anthropic.Anthropic()



    server_params = StdioServerParameters(

        command="python3",

        args=[SERVER_SCRIPT]

    )



    async with stdio_client(server_params) as (read, write):

        async with ClientSession(read, write) as session:

            await session.initialize()

            await chat_loop(session, anthropic_client)





if __name__ == "__main__":

    asyncio.run(main())

source ~/mcp-env/bin/activate

pip install anthropic

cd /usr/local/scripts

python3 mcp_client.py

faz um dns lookup no google.com e depois verifica se a porta 443 esta aberta

🔧 Chamando tool: dns_lookup

   Parâmetros: {'domain': 'google.com'}

   ✅ Resultado obtido (97 chars)



🔧 Chamando tool: check_open_port

   Parâmetros: {'host': '142.250.218.238', 'port': 443}

   ✅ Resultado obtido (37 chars)



Claude:

| Tipo | Endereço          |

|------|-------------------|

| IPv4 | 142.250.218.238   |

| IPv6 | 2800:3f0:4004:... |



| Host            | Porta    | Status    |

|-----------------|----------|-----------|

| 142.250.218.238 | 443/TCP  | ✅ ABERTA |

busca CVEs do apache 2.4.49 e salva os criticos no loot