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