Commit 40b8ae6
committed
Fix missing bounds check in MCP handlers
The getAttachment and exportAttachment MCP handlers had att.Size (database metadata) available but never checked it against
maxAttachmentSize before calling readAttachmentFile. This meant:
- Unnecessary file I/O (open + stat) for attachments already known to be too large
- Misleading error messages ("file not available" instead of "too large")
- If the file existed, the full stat + LimitReader pipeline would execute before rejecting
The Fix (2 lines of logic, added to 2 call sites)
Added a pre-flight att.Size > maxAttachmentSize check in both getAttachment (line 153) and exportAttachment (line 219), immediately
after confirming the attachments directory is configured and before any file I/O.
Changes Made
- internal/mcp/handlers.go: Added pre-flight size check in both handlers (2x 3 lines)
- internal/mcp/server_test.go: Added TestGetAttachment_RejectsOversizedBeforeFileIO and
TestExportAttachment_RejectsOversizedBeforeFileIO (44 lines)1 parent 84f254a commit 40b8ae6
2 files changed
+67
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
151 | 151 | | |
152 | 152 | | |
153 | 153 | | |
| 154 | + | |
| 155 | + | |
| 156 | + | |
| 157 | + | |
154 | 158 | | |
155 | 159 | | |
156 | 160 | | |
| |||
213 | 217 | | |
214 | 218 | | |
215 | 219 | | |
| 220 | + | |
| 221 | + | |
| 222 | + | |
| 223 | + | |
216 | 224 | | |
217 | 225 | | |
218 | 226 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
677 | 677 | | |
678 | 678 | | |
679 | 679 | | |
| 680 | + | |
| 681 | + | |
| 682 | + | |
| 683 | + | |
| 684 | + | |
| 685 | + | |
| 686 | + | |
| 687 | + | |
| 688 | + | |
| 689 | + | |
| 690 | + | |
| 691 | + | |
| 692 | + | |
| 693 | + | |
| 694 | + | |
| 695 | + | |
| 696 | + | |
| 697 | + | |
| 698 | + | |
| 699 | + | |
| 700 | + | |
| 701 | + | |
| 702 | + | |
| 703 | + | |
| 704 | + | |
| 705 | + | |
| 706 | + | |
| 707 | + | |
| 708 | + | |
| 709 | + | |
| 710 | + | |
| 711 | + | |
| 712 | + | |
| 713 | + | |
| 714 | + | |
| 715 | + | |
| 716 | + | |
| 717 | + | |
| 718 | + | |
| 719 | + | |
| 720 | + | |
| 721 | + | |
| 722 | + | |
| 723 | + | |
| 724 | + | |
| 725 | + | |
| 726 | + | |
| 727 | + | |
| 728 | + | |
| 729 | + | |
| 730 | + | |
| 731 | + | |
| 732 | + | |
| 733 | + | |
| 734 | + | |
| 735 | + | |
| 736 | + | |
| 737 | + | |
| 738 | + | |
680 | 739 | | |
681 | 740 | | |
682 | 741 | | |
| |||
0 commit comments