1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
package http
import (
"encoding/json"
"io"
"io/ioutil"
"mime/multipart"
"os"
"sync"
)
const (
// UploadErrorOK - no error, the file uploaded with success.
UploadErrorOK = 0
// UploadErrorNoFile - no file was uploaded.
UploadErrorNoFile = 4
// UploadErrorNoTmpDir - missing a temporary folder.
UploadErrorNoTmpDir = 5
// UploadErrorCantWrite - failed to write file to disk.
UploadErrorCantWrite = 6
// UploadErrorExtension - forbidden file extension.
UploadErrorExtension = 7
)
// Uploads tree manages uploaded files tree and temporary files.
type Uploads struct {
// associated temp directory and forbidden extensions.
cfg *UploadsConfig
// pre processed data tree for Uploads.
tree fileTree
// flat list of all file Uploads.
list []*FileUpload
}
// MarshalJSON marshal tree tree into JSON.
func (u *Uploads) MarshalJSON() ([]byte, error) {
return json.Marshal(u.tree)
}
// Open moves all uploaded files to temp directory, return error in case of issue with temp directory. File errors
// will be handled individually.
func (u *Uploads) Open() error {
var wg sync.WaitGroup
for _, f := range u.list {
wg.Add(1)
go func(f *FileUpload) {
defer wg.Done()
f.Open(u.cfg)
}(f)
}
wg.Wait()
return nil
}
// Clear deletes all temporary files.
func (u *Uploads) Clear() {
for _, f := range u.list {
if f.TempFilename != "" && exists(f.TempFilename) {
os.Remove(f.TempFilename)
}
}
}
// FileUpload represents singular file NewUpload.
type FileUpload struct {
// ID contains filename specified by the client.
Name string `json:"name"`
// Mime contains mime-type provided by the client.
Mime string `json:"mime"`
// Size of the uploaded file.
Size int64 `json:"size"`
// Error indicates file upload error (if any). See http://php.net/manual/en/features.file-upload.errors.php
Error int `json:"error"`
// TempFilename points to temporary file location.
TempFilename string `json:"tmpName"`
// associated file header
header *multipart.FileHeader
}
// NewUpload wraps net/http upload into PRS-7 compatible structure.
func NewUpload(f *multipart.FileHeader) *FileUpload {
return &FileUpload{
Name: f.Filename,
Mime: f.Header.Get("Content-Type"),
Error: UploadErrorOK,
header: f,
}
}
// Open moves file content into temporary file available for PHP.
func (f *FileUpload) Open(cfg *UploadsConfig) error {
if cfg.Forbids(f.Name) {
f.Error = UploadErrorExtension
return nil
}
file, err := f.header.Open()
if err != nil {
f.Error = UploadErrorNoFile
return err
}
defer file.Close()
tmp, err := ioutil.TempFile(cfg.TmpDir(), "upload")
if err != nil {
// most likely cause of this issue is missing tmp dir
f.Error = UploadErrorNoTmpDir
return err
}
f.TempFilename = tmp.Name()
defer tmp.Close()
if f.Size, err = io.Copy(tmp, file); err != nil {
f.Error = UploadErrorCantWrite
}
return err
}
// exists if file exists.
func exists(path string) bool {
_, err := os.Stat(path)
if err == nil {
return true
}
return false
}
|
