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
|
|
15 int ngx_write_chain_to_temp_file(ngx_temp_file_t *tf, ngx_chain_t *chain)
|
|
16 {
|
|
17 int rc;
|
|
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
|
|
36 int ngx_create_temp_file(ngx_file_t *file, ngx_path_t *path,
|
|
37 ngx_pool_t *pool, int persistent)
|
|
38 {
|
|
39 int num;
|
|
40 ngx_err_t err;
|
|
41
|
|
42 file->name.len = path->name.len + 1 + path->len + 10;
|
|
43
|
|
44 ngx_test_null(file->name.data, ngx_palloc(pool, file->name.len + 1),
|
|
45 NGX_ERROR);
|
|
46
|
|
47 #if 0
|
|
48 for (i = 0; i < file->name.len; i++) {
|
|
49 file->name.data[i] = 'X';
|
|
50 }
|
|
51 #endif
|
|
52
|
|
53 ngx_memcpy(file->name.data, path->name.data, path->name.len);
|
|
54
|
|
55 num = ngx_next_temp_number(0);
|
|
56
|
|
57 for ( ;; ) {
|
10
|
58 ngx_sprintf(file->name.data + path->name.len + 1 + path->len,
|
|
59 "%010ud%Z", num);
|
0
|
60
|
|
61 ngx_create_hashed_filename(file, path);
|
|
62
|
|
63 #if 1
|
|
64 file->fd = ngx_open_tempfile(file->name.data, persistent);
|
|
65 #else
|
|
66 file->fd = ngx_open_tempfile(file->name.data, 1);
|
|
67 #endif
|
|
68
|
|
69 ngx_log_debug1(NGX_LOG_DEBUG_CORE, file->log, 0,
|
|
70 "temp fd:%d", file->fd);
|
|
71
|
|
72 if (file->fd != NGX_INVALID_FILE) {
|
|
73 return NGX_OK;
|
|
74 }
|
|
75
|
|
76 err = ngx_errno;
|
|
77
|
|
78 if (err == NGX_EEXIST) {
|
|
79 num = ngx_next_temp_number(1);
|
|
80 continue;
|
|
81 }
|
|
82
|
|
83 if ((path->level[0] == 0)
|
|
84 || (err != NGX_ENOENT
|
10
|
85 #if (NGX_WIN32)
|
0
|
86 && err != NGX_ENOTDIR
|
|
87 #endif
|
|
88 )) {
|
|
89 ngx_log_error(NGX_LOG_CRIT, file->log, err,
|
|
90 ngx_open_tempfile_n " \"%s\" failed",
|
|
91 file->name.data);
|
|
92 return NGX_ERROR;
|
|
93 }
|
|
94
|
|
95 if (ngx_create_path(file, path) == NGX_ERROR) {
|
|
96 return NGX_ERROR;
|
|
97 }
|
|
98 }
|
|
99 }
|
|
100
|
|
101
|
|
102 void ngx_create_hashed_filename(ngx_file_t *file, ngx_path_t *path)
|
|
103 {
|
|
104 int i, name, pos;
|
|
105 size_t level;
|
|
106
|
|
107 name = file->name.len;
|
|
108 pos = path->name.len + 1;
|
|
109
|
|
110 file->name.data[path->name.len + path->len] = '/';
|
|
111
|
|
112 for (i = 0; i < 3; i++) {
|
|
113 level = path->level[i];
|
|
114
|
|
115 if (level == 0) {
|
|
116 break;
|
|
117 }
|
|
118
|
|
119 name -= level;
|
|
120 file->name.data[pos - 1] = '/';
|
|
121 ngx_memcpy(&file->name.data[pos], &file->name.data[name], level);
|
|
122 pos += level + 1;
|
|
123 }
|
|
124
|
|
125 ngx_log_debug1(NGX_LOG_DEBUG_CORE, file->log, 0,
|
|
126 "hashed path: %s", file->name.data);
|
|
127 }
|
|
128
|
|
129
|
|
130 int ngx_create_path(ngx_file_t *file, ngx_path_t *path)
|
|
131 {
|
|
132 int i, pos;
|
|
133 ngx_err_t err;
|
|
134
|
|
135 pos = path->name.len;
|
|
136
|
|
137 for (i = 0; i < 3; i++) {
|
|
138 if (path->level[i] == 0) {
|
|
139 break;
|
|
140 }
|
|
141
|
|
142 pos += path->level[i] + 1;
|
|
143
|
|
144 file->name.data[pos] = '\0';
|
|
145
|
|
146 ngx_log_debug1(NGX_LOG_DEBUG_CORE, file->log, 0,
|
|
147 "temp file: \"%s\"", file->name.data);
|
|
148
|
|
149 if (ngx_create_dir(file->name.data) == NGX_FILE_ERROR) {
|
|
150 err = ngx_errno;
|
|
151 if (err != NGX_EEXIST) {
|
|
152 ngx_log_error(NGX_LOG_CRIT, file->log, err,
|
|
153 ngx_create_dir_n " \"%s\" failed",
|
|
154 file->name.data);
|
|
155 return NGX_ERROR;
|
|
156 }
|
|
157 }
|
|
158
|
|
159 file->name.data[pos] = '/';
|
|
160 }
|
|
161
|
|
162 return NGX_OK;
|
|
163 }
|
|
164
|
|
165
|
|
166 void ngx_init_temp_number()
|
|
167 {
|
|
168 ngx_random = 0;
|
|
169
|
|
170 ngx_temp_number = ngx_random;
|
|
171
|
|
172 while (ngx_random < 10000) {
|
|
173 ngx_random = 123456;
|
|
174 }
|
|
175 }
|
|
176
|
|
177
|
|
178 ngx_uint_t ngx_next_temp_number(ngx_uint_t collision)
|
|
179 {
|
|
180 if (collision) {
|
|
181 ngx_temp_number += ngx_random;
|
|
182 }
|
|
183
|
|
184 return ngx_temp_number++;
|
|
185 }
|
|
186
|
|
187
|
|
188 char *ngx_conf_set_path_slot(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
189 {
|
|
190 char *p = conf;
|
|
191
|
|
192 ngx_int_t level;
|
|
193 ngx_uint_t i, n;
|
|
194 ngx_str_t *value;
|
|
195 ngx_path_t *path, **pp;
|
|
196
|
|
197 pp = (ngx_path_t **) (p + cmd->offset);
|
|
198
|
|
199 if (*pp) {
|
|
200 return "is duplicate";
|
|
201 }
|
|
202
|
|
203 /* TODO: check duplicate in cf->cycle->pathes */
|
|
204
|
|
205 ngx_test_null(path, ngx_pcalloc(cf->pool, sizeof(ngx_path_t)),
|
|
206 NGX_CONF_ERROR);
|
|
207
|
|
208 *pp = path;
|
|
209
|
|
210 ngx_test_null(pp, ngx_push_array(&cf->cycle->pathes), NGX_CONF_ERROR);
|
|
211 *pp = path;
|
|
212
|
|
213 value = (ngx_str_t *) cf->args->elts;
|
|
214
|
|
215 path->name = value[1];
|
|
216
|
|
217 path->len = 0;
|
|
218
|
|
219 for (i = 0, n = 2; n < cf->args->nelts; i++, n++) {
|
|
220 level = ngx_atoi(value[n].data, value[n].len);
|
|
221 if (level == NGX_ERROR || level == 0) {
|
|
222 return "invalid value";
|
|
223 }
|
|
224
|
|
225 path->level[i] = level;
|
|
226 path->len += level + 1;
|
|
227 }
|
|
228
|
|
229 while (i < 3) {
|
|
230 path->level[i++] = 0;
|
|
231 }
|
|
232
|
|
233 path->gc_handler = (ngx_gc_handler_pt) cmd->post;
|
|
234
|
|
235 return NGX_CONF_OK;
|
|
236 }
|