0
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9
|
|
10
|
|
11 static ngx_uint_t ngx_temp_number;
|
|
12 static ngx_uint_t ngx_random;
|
|
13
|
|
14
|
16
|
15 ssize_t ngx_write_chain_to_temp_file(ngx_temp_file_t *tf, ngx_chain_t *chain)
|
0
|
16 {
|
16
|
17 ngx_int_t rc;
|
0
|
18
|
|
19 if (tf->file.fd == NGX_INVALID_FILE) {
|
|
20 rc = ngx_create_temp_file(&tf->file, tf->path, tf->pool,
|
|
21 tf->persistent);
|
|
22
|
|
23 if (rc == NGX_ERROR || rc == NGX_AGAIN) {
|
|
24 return rc;
|
|
25 }
|
|
26
|
|
27 if (!tf->persistent && tf->warn) {
|
|
28 ngx_log_error(NGX_LOG_WARN, tf->file.log, 0, tf->warn);
|
|
29 }
|
|
30 }
|
|
31
|
|
32 return ngx_write_chain_to_file(&tf->file, chain, tf->offset, tf->pool);
|
|
33 }
|
|
34
|
|
35
|
16
|
36 ngx_int_t ngx_create_temp_file(ngx_file_t *file, ngx_path_t *path,
|
|
37 ngx_pool_t *pool, int persistent)
|
0
|
38 {
|
16
|
39 ngx_err_t err;
|
|
40 uint32_t num;
|
0
|
41
|
|
42 file->name.len = path->name.len + 1 + path->len + 10;
|
|
43
|
16
|
44 if (!(file->name.data = ngx_palloc(pool, file->name.len + 1))) {
|
|
45 return NGX_ERROR;
|
|
46 }
|
0
|
47
|
|
48 #if 0
|
|
49 for (i = 0; i < file->name.len; i++) {
|
|
50 file->name.data[i] = 'X';
|
|
51 }
|
|
52 #endif
|
|
53
|
|
54 ngx_memcpy(file->name.data, path->name.data, path->name.len);
|
|
55
|
16
|
56 num = (uint32_t) ngx_next_temp_number(0);
|
0
|
57
|
|
58 for ( ;; ) {
|
10
|
59 ngx_sprintf(file->name.data + path->name.len + 1 + path->len,
|
16
|
60 "%010ui%Z", num);
|
0
|
61
|
|
62 ngx_create_hashed_filename(file, path);
|
|
63
|
|
64 #if 1
|
|
65 file->fd = ngx_open_tempfile(file->name.data, persistent);
|
|
66 #else
|
|
67 file->fd = ngx_open_tempfile(file->name.data, 1);
|
|
68 #endif
|
|
69
|
|
70 ngx_log_debug1(NGX_LOG_DEBUG_CORE, file->log, 0,
|
|
71 "temp fd:%d", file->fd);
|
|
72
|
|
73 if (file->fd != NGX_INVALID_FILE) {
|
|
74 return NGX_OK;
|
|
75 }
|
|
76
|
|
77 err = ngx_errno;
|
|
78
|
|
79 if (err == NGX_EEXIST) {
|
|
80 num = ngx_next_temp_number(1);
|
|
81 continue;
|
|
82 }
|
|
83
|
|
84 if ((path->level[0] == 0)
|
|
85 || (err != NGX_ENOENT
|
10
|
86 #if (NGX_WIN32)
|
0
|
87 && err != NGX_ENOTDIR
|
|
88 #endif
|
16
|
89 ))
|
|
90 {
|
0
|
91 ngx_log_error(NGX_LOG_CRIT, file->log, err,
|
|
92 ngx_open_tempfile_n " \"%s\" failed",
|
|
93 file->name.data);
|
|
94 return NGX_ERROR;
|
|
95 }
|
|
96
|
|
97 if (ngx_create_path(file, path) == NGX_ERROR) {
|
|
98 return NGX_ERROR;
|
|
99 }
|
|
100 }
|
|
101 }
|
|
102
|
|
103
|
|
104 void ngx_create_hashed_filename(ngx_file_t *file, ngx_path_t *path)
|
|
105 {
|
16
|
106 ngx_uint_t i, name, pos, level;
|
0
|
107
|
|
108 name = file->name.len;
|
|
109 pos = path->name.len + 1;
|
|
110
|
|
111 file->name.data[path->name.len + path->len] = '/';
|
|
112
|
|
113 for (i = 0; i < 3; i++) {
|
|
114 level = path->level[i];
|
|
115
|
|
116 if (level == 0) {
|
|
117 break;
|
|
118 }
|
|
119
|
|
120 name -= level;
|
|
121 file->name.data[pos - 1] = '/';
|
|
122 ngx_memcpy(&file->name.data[pos], &file->name.data[name], level);
|
|
123 pos += level + 1;
|
|
124 }
|
|
125
|
|
126 ngx_log_debug1(NGX_LOG_DEBUG_CORE, file->log, 0,
|
|
127 "hashed path: %s", file->name.data);
|
|
128 }
|
|
129
|
|
130
|
16
|
131 ngx_int_t ngx_create_path(ngx_file_t *file, ngx_path_t *path)
|
0
|
132 {
|
|
133 int i, pos;
|
|
134 ngx_err_t err;
|
|
135
|
|
136 pos = path->name.len;
|
|
137
|
|
138 for (i = 0; i < 3; i++) {
|
|
139 if (path->level[i] == 0) {
|
|
140 break;
|
|
141 }
|
|
142
|
|
143 pos += path->level[i] + 1;
|
|
144
|
|
145 file->name.data[pos] = '\0';
|
|
146
|
|
147 ngx_log_debug1(NGX_LOG_DEBUG_CORE, file->log, 0,
|
|
148 "temp file: \"%s\"", file->name.data);
|
|
149
|
|
150 if (ngx_create_dir(file->name.data) == NGX_FILE_ERROR) {
|
|
151 err = ngx_errno;
|
|
152 if (err != NGX_EEXIST) {
|
|
153 ngx_log_error(NGX_LOG_CRIT, file->log, err,
|
|
154 ngx_create_dir_n " \"%s\" failed",
|
|
155 file->name.data);
|
|
156 return NGX_ERROR;
|
|
157 }
|
|
158 }
|
|
159
|
|
160 file->name.data[pos] = '/';
|
|
161 }
|
|
162
|
|
163 return NGX_OK;
|
|
164 }
|
|
165
|
|
166
|
|
167 void ngx_init_temp_number()
|
|
168 {
|
|
169 ngx_random = 0;
|
|
170
|
|
171 ngx_temp_number = ngx_random;
|
|
172
|
|
173 while (ngx_random < 10000) {
|
|
174 ngx_random = 123456;
|
|
175 }
|
|
176 }
|
|
177
|
|
178
|
|
179 ngx_uint_t ngx_next_temp_number(ngx_uint_t collision)
|
|
180 {
|
|
181 if (collision) {
|
|
182 ngx_temp_number += ngx_random;
|
|
183 }
|
|
184
|
|
185 return ngx_temp_number++;
|
|
186 }
|
|
187
|
|
188
|
|
189 char *ngx_conf_set_path_slot(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
190 {
|
|
191 char *p = conf;
|
|
192
|
16
|
193 ssize_t level;
|
0
|
194 ngx_uint_t i, n;
|
|
195 ngx_str_t *value;
|
28
|
196 ngx_path_t *path, **slot;
|
0
|
197
|
16
|
198 slot = (ngx_path_t **) (p + cmd->offset);
|
0
|
199
|
16
|
200 if (*slot) {
|
0
|
201 return "is duplicate";
|
|
202 }
|
|
203
|
16
|
204 if (!(path = ngx_pcalloc(cf->pool, sizeof(ngx_path_t)))) {
|
|
205 return NGX_CONF_ERROR;
|
|
206 }
|
0
|
207
|
16
|
208 value = cf->args->elts;
|
0
|
209
|
|
210 path->name = value[1];
|
|
211 path->len = 0;
|
28
|
212 path->cleaner = (ngx_gc_handler_pt) cmd->post;
|
16
|
213 path->conf_file = cf->conf_file->file.name.data;
|
|
214 path->line = cf->conf_file->line;
|
0
|
215
|
|
216 for (i = 0, n = 2; n < cf->args->nelts; i++, n++) {
|
|
217 level = ngx_atoi(value[n].data, value[n].len);
|
|
218 if (level == NGX_ERROR || level == 0) {
|
|
219 return "invalid value";
|
|
220 }
|
|
221
|
|
222 path->level[i] = level;
|
|
223 path->len += level + 1;
|
|
224 }
|
|
225
|
|
226 while (i < 3) {
|
|
227 path->level[i++] = 0;
|
|
228 }
|
|
229
|
28
|
230 *slot = path;
|
16
|
231
|
28
|
232 if (ngx_add_path(cf, slot) == NGX_ERROR) {
|
|
233 return NGX_CONF_ERROR;
|
|
234 }
|
|
235
|
|
236 return NGX_CONF_OK;
|
|
237 }
|
|
238
|
|
239
|
|
240 ngx_int_t ngx_add_path(ngx_conf_t *cf, ngx_path_t **slot)
|
|
241 {
|
|
242 ngx_uint_t i, n;
|
|
243 ngx_path_t *path, **p;
|
|
244
|
|
245 path = *slot;
|
|
246
|
|
247 p = cf->cycle->pathes.elts;
|
16
|
248 for (i = 0; i < cf->cycle->pathes.nelts; i++) {
|
28
|
249 if (p[i]->name.len == path->name.len
|
|
250 && ngx_strcmp(p[i]->name.data, path->name.data) == 0)
|
16
|
251 {
|
|
252 for (n = 0; n < 3; n++) {
|
28
|
253 if (p[i]->level[n] != path->level[n]) {
|
|
254 if (path->conf_file == NULL) {
|
|
255 ngx_log_error(NGX_LOG_EMERG, cf->log, 0,
|
|
256 "the path name \"%V\" in %s:%ui has "
|
|
257 "the same name as default path but "
|
|
258 "the different levels, you need to "
|
|
259 "define default path in http section",
|
|
260 &p[i]->name, p[i]->conf_file, p[i]->line);
|
|
261 return NGX_ERROR;
|
|
262 }
|
|
263
|
16
|
264 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
28
|
265 "the same path name \"%V\" in %s:%ui "
|
|
266 "has the different levels than",
|
|
267 &p[i]->name, p[i]->conf_file, p[i]->line);
|
|
268 return NGX_ERROR;
|
16
|
269 }
|
|
270
|
28
|
271 if (p[i]->level[n] == 0) {
|
16
|
272 break;
|
|
273 }
|
|
274 }
|
|
275
|
28
|
276 *slot = p[i];
|
16
|
277
|
28
|
278 return NGX_OK;
|
16
|
279 }
|
|
280 }
|
|
281
|
28
|
282 if (!(p = ngx_array_push(&cf->cycle->pathes))) {
|
|
283 return NGX_ERROR;
|
16
|
284 }
|
|
285
|
28
|
286 *p = path;
|
0
|
287
|
28
|
288 return NGX_OK;
|
0
|
289 }
|
16
|
290
|
|
291
|
|
292 ngx_int_t ngx_create_pathes(ngx_cycle_t *cycle, ngx_uid_t user)
|
|
293 {
|
|
294 ngx_err_t err;
|
|
295 ngx_uint_t i;
|
|
296 ngx_path_t **path;
|
|
297 #if !(NGX_WIN32)
|
|
298 ngx_file_info_t fi;
|
|
299 #endif
|
|
300
|
|
301 path = cycle->pathes.elts;
|
|
302 for (i = 0; i < cycle->pathes.nelts; i++) {
|
|
303
|
|
304 if (ngx_create_dir(path[i]->name.data) == NGX_FILE_ERROR) {
|
|
305 err = ngx_errno;
|
|
306 if (err != NGX_EEXIST) {
|
|
307 ngx_log_error(NGX_LOG_EMERG, cycle->log, err,
|
|
308 ngx_create_dir_n " \"%s\" failed",
|
|
309 path[i]->name.data);
|
|
310 return NGX_ERROR;
|
|
311 }
|
|
312 }
|
|
313
|
|
314 if (user == (ngx_uid_t) -1) {
|
|
315 continue;
|
|
316 }
|
|
317
|
|
318 #if !(NGX_WIN32)
|
|
319
|
|
320 if (ngx_file_info((const char *) path[i]->name.data, &fi) == -1) {
|
|
321 ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno,
|
|
322 ngx_file_info_n " \"%s\" failed", path[i]->name.data);
|
|
323 return NGX_ERROR;
|
|
324 }
|
|
325
|
|
326 if (fi.st_uid != user) {
|
|
327 if (chown((const char *) path[i]->name.data, user, -1) == -1) {
|
|
328 ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno,
|
|
329 "chown(\"%s\", %d) failed",
|
|
330 path[i]->name.data, user);
|
|
331 return NGX_ERROR;
|
|
332 }
|
|
333 }
|
|
334
|
|
335 if ((fi.st_mode & (S_IRUSR|S_IWUSR|S_IXUSR))
|
|
336 != (S_IRUSR|S_IWUSR|S_IXUSR))
|
|
337 {
|
|
338 fi.st_mode |= (S_IRUSR|S_IWUSR|S_IXUSR);
|
|
339
|
|
340 if (chmod((const char *) path[i]->name.data, fi.st_mode) == -1) {
|
|
341 ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno,
|
|
342 "chmod() \"%s\" failed", path[i]->name.data);
|
|
343 return NGX_ERROR;
|
|
344 }
|
|
345 }
|
|
346
|
|
347 #endif
|
|
348 }
|
|
349
|
|
350 return NGX_OK;
|
|
351 }
|