codeant-ai-for-open-source[bot] commented on code in PR #44310:
URL: https://github.com/apache/superset/pull/44310#discussion_r4016706957
##########
superset/utils/core.py:
##########
@@ -2088,6 +2088,60 @@ def create_zip(files: dict[str, Any]) -> BytesIO:
return buf
+# Matches a safe, opaque token suitable for use as a cookie name. Restricting
the
+# allowed characters prevents client-controlled input from injecting unexpected
+# cookie attributes or control characters.
+COOKIE_TOKEN_RE = re.compile(r"^[A-Za-z0-9_-]{1,128}$")
+
+
+def sanitize_cookie_token(token: str | None) -> str | None:
+ """Return the token if it is a valid cookie name, otherwise None.
+
+ The export endpoints echo a client-provided ``token`` query parameter back
as
+ a cookie name to signal download completion. Validate it against a strict
+ allow-list before trusting it.
+
+ :param token: the client-provided token value
+ :return: the token if valid, else None
+ """
+ if token and COOKIE_TOKEN_RE.match(token):
+ return token
+ return None
+
+
+def send_export_zip(buf: BytesIO, filename: str) -> Response:
+ """Build a non-cacheable ZIP attachment response for the export endpoints.
+
+ Export bundles are generated per request from live metadata, so they must
never
+ be cached. Flask applies ``SEND_FILE_MAX_AGE_DEFAULT`` (one year in
Superset's
+ config) to every ``send_file`` response that does not opt out, which made
+ browsers and intermediate proxies serve stale export archives. Passing
+ ``max_age=0`` and marking the response ``no-store``/``no-cache`` keeps the
+ behavior of genuine static assets untouched while forcing exports to be
fetched
+ fresh every time.
+
+ The optional client-provided ``token`` query parameter is echoed back as a
+ cookie so the UI can detect that the download finished.
+
+ :param buf: an in-memory ZIP archive, positioned at the start
+ :param filename: the download file name advertised to the client
+ :return: the response to return from the export endpoint
+ """
+ response = send_file(
+ buf,
+ mimetype="application/zip",
+ as_attachment=True,
+ download_name=filename,
+ max_age=0,
+ )
+ response.cache_control.no_store = True
+ response.cache_control.no_cache = True
+ response.cache_control.must_revalidate = True
+ if token := sanitize_cookie_token(request.args.get("token")):
+ response.set_cookie(token, "done", max_age=600)
Review Comment:
**Suggestion:** A caller can choose `session` or another application cookie
name, so this response overwrites that cookie with `done` and can log users
out. [security]
**Assessment:** ๐ `Major` ยท ๐ `Occurrence: Sometimes`
[](https://docs.codeant.ai/cli/resolve-pr-comments-skill)
[](https://app.codeant.ai/fix-in-ide?tool=cursor&prompt_id=03c721080e4546fa9b9ead6df63f6d76&service=github&base_url=https%3A%2F%2Fgithub.com&org=apache&repo=apache%2Fsuperset)
[](https://app.codeant.ai/fix-in-ide?tool=vscode-claude&prompt_id=03c721080e4546fa9b9ead6df63f6d76&service=github&base_url=https%3A%2F%2Fgithub.com&org=apache&repo=apache%2Fsuperset)
<details>
<summary><b>Prompt for AI Agent ๐ค </b></summary>
```mdx
This is a comment left during a code review.
**Path:** superset/utils/core.py
**Line:** 2141:2141
**Comment:**
*Security: A caller can choose `session` or another application cookie
name, so this response overwrites that cookie with `done` and can log users out.
Validate the correctness of the flagged issue. If correct, How can I resolve
this? If you propose a fix, implement it and please make it concise.
Once fix is implemented, also check other comments on the same PR, and ask
user if the user wants to fix the rest of the comments as well. if said yes,
then fetch all the comments validate the correctness and implement a minimal fix
```
</details>
<a
href='https://app.codeant.ai/feedback?pr_url=https%3A%2F%2Fgithub.com%2Fapache%2Fsuperset%2Fpull%2F44310&comment_hash=6da26fadb93ab27addd5a0e060a0266877d63825fe76b078afb5fe6c1d48530f&reaction=like'>๐</a>
| <a
href='https://app.codeant.ai/feedback?pr_url=https%3A%2F%2Fgithub.com%2Fapache%2Fsuperset%2Fpull%2F44310&comment_hash=6da26fadb93ab27addd5a0e060a0266877d63825fe76b078afb5fe6c1d48530f&reaction=dislike'>๐</a>
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
