celogeek/photos-api
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

check checksum, cleanup

Browse Source
This commit is contained in:
celogeek 2022-05-08 17:03:46 +02:00
parent 3e21d30f62
commit 9020415e74
Signed by: celogeek
GPG Key ID: E6B7BDCFC446233A
1 changed files with 25 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
internal/photos/api/upload.go
View File

@ -18,8 +18,9 @@ const (
)
var (
ErrUploadNotExists = errors.New("upload id doesn't exists")
ErrUploadTooLarge = fmt.Errorf("upload part too large ( > %d B)", MaxUploadPartSize)
ErrUploadNotExists = errors.New("upload id doesn't exists")
ErrUploadPartTooLarge = fmt.Errorf("upload part too large (> %d B)", MaxUploadPartSize)
ErrUploadPartWrongSha256 = errors.New("upload part wrong sha256")
)
func (s *Service) UploadCreate(c *gin.Context) {
@ -65,25 +66,39 @@ func (s *Service) UploadPart(c *gin.Context) {
}
if c.Request.ContentLength > MaxUploadPartSize {
c.AbortWithError(http.StatusRequestEntityTooLarge, ErrUploadTooLarge)
c.AbortWithError(http.StatusRequestEntityTooLarge, ErrUploadPartTooLarge)
return
}
f, err := os.Create(
s.StorageTmp.Join(upload.Id, fmt.Sprint(uploadPart.Part)),
)
tmp_file := s.StorageTmp.Join(upload.Id, fmt.Sprintf("._tmp_%d", uploadPart.Part))
file := s.StorageTmp.Join(upload.Id, fmt.Sprint(uploadPart.Part))
f, err := os.Create(tmp_file)
if err != nil {
c.AbortWithError(http.StatusInternalServerError, err)
return
}
defer f.Close()
defer c.Request.Body.Close()
sha := sha256.New()
t := io.TeeReader(c.Request.Body, sha)
w, err := io.Copy(f, t)
if err != nil {
f.Close()
os.Remove(tmp_file)
c.AbortWithError(http.StatusInternalServerError, err)
return
}
f.Close()
shastr := hex.EncodeToString(sha.Sum(nil))
if shastr != uploadPart.PartSha256 {
os.Remove(tmp_file)
c.AbortWithError(http.StatusBadRequest, ErrUploadPartWrongSha256)
return
}
if err = os.Rename(tmp_file, file); err != nil {
os.Remove(tmp_file)
c.AbortWithError(http.StatusInternalServerError, err)
return
}
@ -92,7 +107,7 @@ func (s *Service) UploadPart(c *gin.Context) {
"upload_id": upload.Id,
"part": uploadPart.Part,
"size": w,
"sha256": hex.EncodeToString(sha.Sum(nil)),
"sha256": uploadPart.PartSha256,
"status": "success",
})
}