Method
writeFileWithLimit
(fileHeader *multipart.FileHeader, dst string)
Source from the content-addressed store, hash-verified
| 78 | } |
| 79 | |
| 80 | func (s *HTMLHostService) writeFileWithLimit(fileHeader *multipart.FileHeader, dst string) error { |
| 81 | src, err := fileHeader.Open() |
| 82 | if err != nil { |
| 83 | return err |
| 84 | } |
| 85 | defer src.Close() |
| 86 | |
| 87 | tmpFile, err := os.Create(dst) |
| 88 | if err != nil { |
| 89 | return err |
| 90 | } |
| 91 | defer tmpFile.Close() |
| 92 | |
| 93 | limit := s.maxSize + 1 |
| 94 | n, err := io.Copy(tmpFile, io.LimitReader(src, limit)) |
| 95 | if err != nil { |
| 96 | os.Remove(dst) |
| 97 | return err |
| 98 | } |
| 99 | if n > s.maxSize { |
| 100 | os.Remove(dst) |
| 101 | return ErrHTMLTooLarge |
| 102 | } |
| 103 | return nil |
| 104 | } |
Tested by
no test coverage detected