fix(security): allow in-cluster service hosts for FastMCP transport validation
This commit is contained in:
19
server.py
19
server.py
@@ -1,6 +1,7 @@
|
|||||||
import os
|
import os
|
||||||
import httpx
|
import httpx
|
||||||
from mcp.server.fastmcp import FastMCP
|
from mcp.server.fastmcp import FastMCP
|
||||||
|
from mcp.server.transport_security import TransportSecuritySettings
|
||||||
import psycopg
|
import psycopg
|
||||||
from pgvector.psycopg import register_vector
|
from pgvector.psycopg import register_vector
|
||||||
import uuid
|
import uuid
|
||||||
@@ -16,7 +17,23 @@ EMBEDDING_DIM = 768 # BAAI/bge-base-en-v1.5
|
|||||||
|
|
||||||
# Initialize
|
# Initialize
|
||||||
logging.basicConfig(level=logging.INFO)
|
logging.basicConfig(level=logging.INFO)
|
||||||
mcp = FastMCP("knowledge-mcp")
|
mcp = FastMCP(
|
||||||
|
"knowledge-mcp",
|
||||||
|
host="0.0.0.0",
|
||||||
|
port=8000,
|
||||||
|
sse_path="/sse",
|
||||||
|
transport_security=TransportSecuritySettings(
|
||||||
|
enable_dns_rebinding_protection=True,
|
||||||
|
allowed_hosts=[
|
||||||
|
"localhost:*",
|
||||||
|
"127.0.0.1:*",
|
||||||
|
"knowledge-mcp:*",
|
||||||
|
"knowledge-mcp.knowledge-mcp.svc:*",
|
||||||
|
"knowledge-mcp.knowledge-mcp.svc.cluster.local:*",
|
||||||
|
],
|
||||||
|
allowed_origins=[],
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
def get_db(init=False):
|
def get_db(init=False):
|
||||||
|
|||||||
Reference in New Issue
Block a user