@ai-sdk/mcp 2.0.48 → 2.0.50

package/CHANGELOG.md

@@ -1,5 +1,26 @@
 # @ai-sdk/mcp
 
+## 2.0.50
+
+### Patch Changes
+
+- Updated dependencies [5c0054d]
+- Updated dependencies [39535af]
+  - @ai-sdk/provider@4.0.15
+  - @ai-sdk/provider-utils@5.0.41
+
+## 2.0.49
+
+### Patch Changes
+
+- c43e4b7: fix(mcp): prevent SSRF in OAuth metadata discovery
+- Updated dependencies [5ec21a6]
+- Updated dependencies [7469a3b]
+- Updated dependencies [813bb36]
+- Updated dependencies [c43e4b7]
+  - @ai-sdk/provider@4.0.14
+  - @ai-sdk/provider-utils@5.0.40
+
 ## 2.0.48
 
 ### Patch Changes

package/dist/index.js

@@ -570,6 +570,8 @@ function checkResourceAllowed({
 
 // src/tool/oauth.ts
 import {
+  fetchWithValidatedEndpoint,
+  fetchWithValidatedRedirects,
   parseJSON as parseJSON2,
   validateDownloadUrl
 } from "@ai-sdk/provider-utils";
@@ -586,8 +588,8 @@ function isOAuthLoopbackHost(hostname) {
   const normalized = hostname.toLowerCase().replace(/\.+$/, "");
   return normalized === "localhost" || normalized.endsWith(".localhost") || normalized === "127.0.0.1" || normalized === "[::1]" || normalized === "::1";
 }
-function assertSafeOAuthEndpoint(endpointUrl) {
-  if ((endpointUrl.protocol === "http:" || endpointUrl.protocol === "https:") && isOAuthLoopbackHost(endpointUrl.hostname)) {
+function assertSafeOAuthEndpoint(endpointUrl, { allowLoopback = false } = {}) {
+  if (allowLoopback && (endpointUrl.protocol === "http:" || endpointUrl.protocol === "https:") && isOAuthLoopbackHost(endpointUrl.hostname)) {
     return;
   }
   try {
@@ -599,6 +601,10 @@ function assertSafeOAuthEndpoint(endpointUrl) {
     });
   }
 }
+function getTrustedLoopbackOrigin(authorizationServerUrl) {
+  const url = new URL(authorizationServerUrl);
+  return isOAuthLoopbackHost(url.hostname) ? url.origin : void 0;
+}
 function validateAuthorizationResponseIssuer({
   callbackIssuer,
   expectedIssuer
@@ -754,13 +760,27 @@ function buildWellKnownPath(wellKnownPrefix, pathname = "", options = {}) {
   }
   return options.prependPathname ? `${pathname}/.well-known/${wellKnownPrefix}` : `/.well-known/${wellKnownPrefix}${pathname}`;
 }
-async function fetchWithCorsRetry(url, headers, fetchFn = fetch) {
+async function fetchWithCorsRetry(url, headers, fetchFn = fetch, trustedOrigin) {
   try {
-    return await fetchFn(url, { headers });
+    return await fetchWithValidatedRedirects({
+      url: url.href,
+      fetch: async (input, init) => fetchWithValidatedEndpoint({
+        url: input,
+        init: {
+          ...init,
+          headers: new Headers(init == null ? void 0 : init.headers).has("MCP-Protocol-Version") ? headers : void 0
+        },
+        fetch: fetchFn,
+        trustedOrigin,
+        redirect: "manual"
+      }),
+      headers,
+      trustedOrigin
+    });
   } catch (error) {
     if (error instanceof TypeError) {
       if (headers) {
-        return fetchWithCorsRetry(url, void 0, fetchFn);
+        return fetchWithCorsRetry(url, void 0, fetchFn, trustedOrigin);
       } else {
         return void 0;
       }
@@ -768,11 +788,11 @@ async function fetchWithCorsRetry(url, headers, fetchFn = fetch) {
     throw error;
   }
 }
-async function tryMetadataDiscovery(url, protocolVersion, fetchFn = fetch) {
+async function tryMetadataDiscovery(url, protocolVersion, fetchFn = fetch, trustedOrigin) {
   const headers = {
     "MCP-Protocol-Version": protocolVersion
   };
-  return await fetchWithCorsRetry(url, headers, fetchFn);
+  return await fetchWithCorsRetry(url, headers, fetchFn, trustedOrigin);
 }
 function shouldAttemptFallback(response, pathname) {
   return !response || response.status >= 400 && response.status < 500 && pathname !== "/";
@@ -789,10 +809,20 @@ async function discoverMetadataWithFallback(serverUrl, wellKnownType, fetchFn, o
     url = new URL(wellKnownPath, (_b3 = opts == null ? void 0 : opts.metadataServerUrl) != null ? _b3 : issuer);
     url.search = issuer.search;
   }
-  let response = await tryMetadataDiscovery(url, protocolVersion, fetchFn);
+  let response = await tryMetadataDiscovery(
+    url,
+    protocolVersion,
+    fetchFn,
+    opts == null ? void 0 : opts.trustedOrigin
+  );
   if (!(opts == null ? void 0 : opts.metadataUrl) && shouldAttemptFallback(response, issuer.pathname)) {
     const rootUrl = new URL(`/.well-known/${wellKnownType}`, issuer);
-    response = await tryMetadataDiscovery(rootUrl, protocolVersion, fetchFn);
+    response = await tryMetadataDiscovery(
+      rootUrl,
+      protocolVersion,
+      fetchFn,
+      opts == null ? void 0 : opts.trustedOrigin
+    );
   }
   return response;
 }
@@ -803,7 +833,10 @@ async function discoverOAuthProtectedResourceMetadata(serverUrl, opts, fetchFn =
     fetchFn,
     {
       protocolVersion: opts == null ? void 0 : opts.protocolVersion,
-      metadataUrl: opts == null ? void 0 : opts.resourceMetadataUrl
+      metadataUrl: opts == null ? void 0 : opts.resourceMetadataUrl,
+      // The configured MCP server is trusted as a request target. Redirects
+      // crossing its origin are still validated before they are followed.
+      trustedOrigin: new URL(serverUrl).origin
     }
   );
   if (!response || response.status === 404) {
@@ -876,13 +909,19 @@ function assertMetadataIssuerMatches(metadata, expectedIssuer) {
 }
 async function discoverAuthorizationServerMetadata(authorizationServerUrl, {
   fetchFn = fetch,
-  protocolVersion = LATEST_PROTOCOL_VERSION
+  protocolVersion = LATEST_PROTOCOL_VERSION,
+  trustedOrigin
 } = {}) {
   var _a3;
   const headers = { "MCP-Protocol-Version": protocolVersion };
   const urlsToTry = buildDiscoveryUrls(authorizationServerUrl);
   for (const { url: endpointUrl, type, expectedIssuer } of urlsToTry) {
-    const response = await fetchWithCorsRetry(endpointUrl, headers, fetchFn);
+    const response = await fetchWithCorsRetry(
+      endpointUrl,
+      headers,
+      fetchFn,
+      trustedOrigin
+    );
     if (!response) {
       continue;
     }
@@ -1048,7 +1087,10 @@ async function exchangeAuthorization(authorizationServerUrl, {
   var _a3;
   const grantType = "authorization_code";
   const tokenUrl = (metadata == null ? void 0 : metadata.token_endpoint) ? new URL(metadata.token_endpoint) : new URL("/token", authorizationServerUrl);
-  assertSafeOAuthEndpoint(tokenUrl);
+  const trustedOrigin = getTrustedLoopbackOrigin(authorizationServerUrl);
+  assertSafeOAuthEndpoint(tokenUrl, {
+    allowLoopback: tokenUrl.origin === trustedOrigin
+  });
   if ((metadata == null ? void 0 : metadata.grant_types_supported) && !metadata.grant_types_supported.includes(grantType)) {
     throw new Error(
       `Incompatible auth server: does not support grant type ${grantType}`
@@ -1082,11 +1124,15 @@ async function exchangeAuthorization(authorizationServerUrl, {
   if (resource) {
     params.set("resource", resourceUrlStripSlash(resource));
   }
-  const response = await (fetchFn != null ? fetchFn : fetch)(tokenUrl, {
-    method: "POST",
-    headers,
-    body: params,
-    redirect: "error"
+  const response = await fetchWithValidatedEndpoint({
+    url: tokenUrl,
+    init: {
+      method: "POST",
+      headers,
+      body: params
+    },
+    fetch: fetchFn,
+    trustedOrigin
   });
   if (!response.ok) {
     throw await parseErrorResponse(response);
@@ -1114,7 +1160,10 @@ async function refreshAuthorization(authorizationServerUrl, {
   } else {
     tokenUrl = new URL("/token", authorizationServerUrl);
   }
-  assertSafeOAuthEndpoint(tokenUrl);
+  const trustedOrigin = getTrustedLoopbackOrigin(authorizationServerUrl);
+  assertSafeOAuthEndpoint(tokenUrl, {
+    allowLoopback: tokenUrl.origin === trustedOrigin
+  });
   const headers = new Headers({
     "Content-Type": "application/x-www-form-urlencoded",
     Accept: "application/json"
@@ -1141,11 +1190,15 @@ async function refreshAuthorization(authorizationServerUrl, {
   if (resource) {
     params.set("resource", resourceUrlStripSlash(resource));
   }
-  const response = await (fetchFn != null ? fetchFn : fetch)(tokenUrl, {
-    method: "POST",
-    headers,
-    body: params,
-    redirect: "error"
+  const response = await fetchWithValidatedEndpoint({
+    url: tokenUrl,
+    init: {
+      method: "POST",
+      headers,
+      body: params
+    },
+    fetch: fetchFn,
+    trustedOrigin
   });
   if (!response.ok) {
     throw await parseErrorResponse(response);
@@ -1172,18 +1225,25 @@ async function registerClient(authorizationServerUrl, {
   } else {
     registrationUrl = new URL("/register", authorizationServerUrl);
   }
-  assertSafeOAuthEndpoint(registrationUrl);
+  const trustedOrigin = getTrustedLoopbackOrigin(authorizationServerUrl);
+  assertSafeOAuthEndpoint(registrationUrl, {
+    allowLoopback: registrationUrl.origin === trustedOrigin
+  });
   const applicationType = (_a3 = clientMetadata.application_type) != null ? _a3 : inferOAuthApplicationType(clientMetadata.redirect_uris);
-  const response = await (fetchFn != null ? fetchFn : fetch)(registrationUrl, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json"
+  const response = await fetchWithValidatedEndpoint({
+    url: registrationUrl,
+    init: {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        ...clientMetadata,
+        application_type: applicationType
+      })
     },
-    body: JSON.stringify({
-      ...clientMetadata,
-      application_type: applicationType
-    }),
-    redirect: "error"
+    fetch: fetchFn,
+    trustedOrigin
   });
   if (!response.ok) {
     throw await parseErrorResponse(response);
@@ -1260,6 +1320,14 @@ async function authInternal(provider, {
   if (!authorizationServerUrl) {
     authorizationServerUrl = serverUrl;
   }
+  const parsedServerUrl = new URL(serverUrl);
+  const parsedAuthorizationServerUrl = new URL(authorizationServerUrl);
+  const serverOrigin = parsedServerUrl.origin;
+  const authorizationServerOrigin = parsedAuthorizationServerUrl.origin;
+  const trustedAuthorizationServerOrigin = authorizationServerOrigin === serverOrigin || isOAuthLoopbackHost(parsedServerUrl.hostname) && isOAuthLoopbackHost(parsedAuthorizationServerUrl.hostname) ? authorizationServerOrigin : void 0;
+  if (!trustedAuthorizationServerOrigin) {
+    assertSafeOAuthEndpoint(new URL(authorizationServerUrl));
+  }
   const resource = await selectResourceURL(
     serverUrl,
     provider,
@@ -1273,7 +1341,8 @@ async function authInternal(provider, {
   const metadata = await discoverAuthorizationServerMetadata(
     authorizationServerUrl,
     {
-      fetchFn
+      fetchFn,
+      trustedOrigin: trustedAuthorizationServerOrigin
     }
   );
   const currentAuthorizationServerInformation = createAuthorizationServerInformation(authorizationServerUrl, metadata);