484
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9 #include <ngx_http.h>
|
488
|
10
|
|
11 #include <gd.h>
|
484
|
12
|
|
13
|
|
14 #define NGX_HTTP_IMAGE_OFF 0
|
|
15 #define NGX_HTTP_IMAGE_TEST 1
|
|
16 #define NGX_HTTP_IMAGE_SIZE 2
|
|
17 #define NGX_HTTP_IMAGE_RESIZE 3
|
|
18 #define NGX_HTTP_IMAGE_CROP 4
|
|
19
|
|
20
|
|
21 #define NGX_HTTP_IMAGE_START 0
|
|
22 #define NGX_HTTP_IMAGE_READ 1
|
|
23 #define NGX_HTTP_IMAGE_PROCESS 2
|
488
|
24 #define NGX_HTTP_IMAGE_PASS 3
|
|
25 #define NGX_HTTP_IMAGE_DONE 4
|
484
|
26
|
|
27
|
|
28 #define NGX_HTTP_IMAGE_NONE 0
|
|
29 #define NGX_HTTP_IMAGE_JPEG 1
|
|
30 #define NGX_HTTP_IMAGE_GIF 2
|
|
31 #define NGX_HTTP_IMAGE_PNG 3
|
|
32
|
|
33
|
|
34 #define NGX_HTTP_IMAGE_BUFFERED 0x08
|
|
35
|
|
36
|
|
37 typedef struct {
|
|
38 ngx_uint_t filter;
|
|
39 ngx_uint_t width;
|
|
40 ngx_uint_t height;
|
492
|
41 ngx_int_t jpeg_quality;
|
484
|
42
|
|
43 size_t buffer_size;
|
|
44 } ngx_http_image_filter_conf_t;
|
|
45
|
|
46
|
|
47 typedef struct {
|
|
48 u_char *image;
|
|
49 u_char *last;
|
|
50
|
|
51 size_t length;
|
|
52
|
|
53 ngx_uint_t width;
|
|
54 ngx_uint_t height;
|
|
55
|
|
56 ngx_uint_t phase;
|
|
57 ngx_uint_t type;
|
|
58 } ngx_http_image_filter_ctx_t;
|
|
59
|
|
60
|
488
|
61 static ngx_int_t ngx_http_image_send(ngx_http_request_t *r,
|
|
62 ngx_http_image_filter_ctx_t *ctx, ngx_chain_t *in);
|
484
|
63 static ngx_uint_t ngx_http_image_test(ngx_http_request_t *r, ngx_chain_t *in);
|
|
64 static ngx_int_t ngx_http_image_read(ngx_http_request_t *r, ngx_chain_t *in);
|
|
65 static ngx_buf_t *ngx_http_image_process(ngx_http_request_t *r);
|
|
66 static ngx_buf_t *ngx_http_image_json(ngx_http_request_t *r,
|
|
67 ngx_http_image_filter_ctx_t *ctx);
|
|
68 static ngx_buf_t *ngx_http_image_asis(ngx_http_request_t *r,
|
|
69 ngx_http_image_filter_ctx_t *ctx);
|
|
70 static void ngx_http_image_length(ngx_http_request_t *r, ngx_buf_t *b);
|
|
71 static ngx_int_t ngx_http_image_size(ngx_http_request_t *r,
|
|
72 ngx_http_image_filter_ctx_t *ctx);
|
|
73
|
|
74 static ngx_buf_t *ngx_http_image_resize(ngx_http_request_t *r,
|
|
75 ngx_http_image_filter_ctx_t *ctx);
|
|
76 static gdImagePtr ngx_http_image_source(ngx_http_request_t *r,
|
|
77 ngx_http_image_filter_ctx_t *ctx);
|
|
78 static gdImagePtr ngx_http_image_new(ngx_http_request_t *r, int w, int h,
|
|
79 int colors);
|
|
80 static u_char *ngx_http_image_out(ngx_http_request_t *r, ngx_uint_t type,
|
|
81 gdImagePtr img, int *size);
|
|
82 static void ngx_http_image_cleanup(void *data);
|
|
83
|
|
84
|
|
85 static void *ngx_http_image_filter_create_conf(ngx_conf_t *cf);
|
|
86 static char *ngx_http_image_filter_merge_conf(ngx_conf_t *cf, void *parent,
|
|
87 void *child);
|
|
88 static char *ngx_http_image_filter(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
89 void *conf);
|
|
90 static ngx_int_t ngx_http_image_filter_init(ngx_conf_t *cf);
|
|
91
|
|
92
|
|
93 static ngx_command_t ngx_http_image_filter_commands[] = {
|
|
94
|
|
95 { ngx_string("image_filter"),
|
|
96 NGX_HTTP_LOC_CONF|NGX_CONF_TAKE13,
|
|
97 ngx_http_image_filter,
|
|
98 NGX_HTTP_LOC_CONF_OFFSET,
|
|
99 0,
|
|
100 NULL },
|
|
101
|
492
|
102 { ngx_string("image_filter_jpeg_quality"),
|
|
103 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
|
|
104 ngx_conf_set_num_slot,
|
|
105 NGX_HTTP_LOC_CONF_OFFSET,
|
|
106 offsetof(ngx_http_image_filter_conf_t, jpeg_quality),
|
|
107 NULL },
|
|
108
|
484
|
109 { ngx_string("image_filter_buffer"),
|
|
110 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
|
|
111 ngx_conf_set_size_slot,
|
|
112 NGX_HTTP_LOC_CONF_OFFSET,
|
|
113 offsetof(ngx_http_image_filter_conf_t, buffer_size),
|
|
114 NULL },
|
|
115
|
|
116 ngx_null_command
|
|
117 };
|
|
118
|
|
119
|
|
120 static ngx_http_module_t ngx_http_image_filter_module_ctx = {
|
|
121 NULL, /* preconfiguration */
|
|
122 ngx_http_image_filter_init, /* postconfiguration */
|
|
123
|
|
124 NULL, /* create main configuration */
|
|
125 NULL, /* init main configuration */
|
|
126
|
|
127 NULL, /* create server configuration */
|
|
128 NULL, /* merge server configuration */
|
|
129
|
|
130 ngx_http_image_filter_create_conf, /* create location configuration */
|
|
131 ngx_http_image_filter_merge_conf /* merge location configuration */
|
|
132 };
|
|
133
|
|
134
|
|
135 ngx_module_t ngx_http_image_filter_module = {
|
|
136 NGX_MODULE_V1,
|
|
137 &ngx_http_image_filter_module_ctx, /* module context */
|
|
138 ngx_http_image_filter_commands, /* module directives */
|
|
139 NGX_HTTP_MODULE, /* module type */
|
|
140 NULL, /* init master */
|
|
141 NULL, /* init module */
|
|
142 NULL, /* init process */
|
|
143 NULL, /* init thread */
|
|
144 NULL, /* exit thread */
|
|
145 NULL, /* exit process */
|
|
146 NULL, /* exit master */
|
|
147 NGX_MODULE_V1_PADDING
|
|
148 };
|
|
149
|
|
150
|
|
151 static ngx_http_output_header_filter_pt ngx_http_next_header_filter;
|
|
152 static ngx_http_output_body_filter_pt ngx_http_next_body_filter;
|
|
153
|
|
154
|
|
155 static ngx_str_t ngx_http_image_types[] = {
|
|
156 ngx_string("image/jpeg"),
|
|
157 ngx_string("image/gif"),
|
|
158 ngx_string("image/png")
|
|
159 };
|
|
160
|
|
161
|
|
162 static ngx_int_t
|
|
163 ngx_http_image_header_filter(ngx_http_request_t *r)
|
|
164 {
|
|
165 off_t len;
|
|
166 ngx_http_image_filter_ctx_t *ctx;
|
|
167 ngx_http_image_filter_conf_t *conf;
|
|
168
|
|
169 if (r->headers_out.status == NGX_HTTP_NOT_MODIFIED) {
|
|
170 return ngx_http_next_header_filter(r);
|
|
171 }
|
|
172
|
490
|
173 ctx = ngx_http_get_module_ctx(r, ngx_http_image_filter_module);
|
|
174
|
|
175 if (ctx) {
|
|
176 ngx_http_set_ctx(r, NULL, ngx_http_image_filter_module);
|
|
177 return ngx_http_next_header_filter(r);
|
|
178 }
|
|
179
|
484
|
180 conf = ngx_http_get_module_loc_conf(r, ngx_http_image_filter_module);
|
|
181
|
|
182 if (conf->filter == NGX_HTTP_IMAGE_OFF) {
|
|
183 return ngx_http_next_header_filter(r);
|
|
184 }
|
|
185
|
|
186 if (r->headers_out.content_type.len
|
|
187 >= sizeof("multipart/x-mixed-replace") - 1
|
|
188 && ngx_strncasecmp(r->headers_out.content_type.data,
|
|
189 (u_char *) "multipart/x-mixed-replace",
|
|
190 sizeof("multipart/x-mixed-replace") - 1)
|
|
191 == 0)
|
|
192 {
|
|
193 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
|
|
194 "image filter: multipart/x-mixed-replace response");
|
|
195
|
|
196 return NGX_ERROR;
|
|
197 }
|
|
198
|
|
199 ctx = ngx_pcalloc(r->pool, sizeof(ngx_http_image_filter_ctx_t));
|
|
200 if (ctx == NULL) {
|
|
201 return NGX_ERROR;
|
|
202 }
|
|
203
|
|
204 ngx_http_set_ctx(r, ctx, ngx_http_image_filter_module);
|
|
205
|
|
206 len = r->headers_out.content_length_n;
|
|
207
|
486
|
208 if (len != -1 && len > (off_t) conf->buffer_size) {
|
484
|
209 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
|
|
210 "image filter: too big response: %O", len);
|
|
211
|
|
212 return NGX_ERROR;
|
|
213 }
|
|
214
|
|
215 if (len == -1) {
|
|
216 ctx->length = conf->buffer_size;
|
|
217
|
|
218 } else {
|
|
219 ctx->length = (size_t) len;
|
|
220 }
|
|
221
|
|
222 if (r->headers_out.refresh) {
|
|
223 r->headers_out.refresh->hash = 0;
|
|
224 }
|
|
225
|
|
226 r->main_filter_need_in_memory = 1;
|
|
227 r->allow_ranges = 0;
|
|
228
|
|
229 return NGX_OK;
|
|
230 }
|
|
231
|
|
232
|
|
233 static ngx_int_t
|
|
234 ngx_http_image_body_filter(ngx_http_request_t *r, ngx_chain_t *in)
|
|
235 {
|
|
236 ngx_int_t rc;
|
|
237 ngx_str_t *ct;
|
|
238 ngx_chain_t out;
|
|
239 ngx_http_image_filter_ctx_t *ctx;
|
|
240 ngx_http_image_filter_conf_t *conf;
|
|
241
|
|
242 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, "image filter");
|
|
243
|
|
244 if (in == NULL) {
|
|
245 return ngx_http_next_body_filter(r, in);
|
|
246 }
|
|
247
|
|
248 ctx = ngx_http_get_module_ctx(r, ngx_http_image_filter_module);
|
|
249
|
|
250 if (ctx == NULL) {
|
|
251 return ngx_http_next_body_filter(r, in);
|
|
252 }
|
|
253
|
|
254 switch (ctx->phase) {
|
|
255
|
|
256 case NGX_HTTP_IMAGE_START:
|
|
257
|
|
258 ctx->type = ngx_http_image_test(r, in);
|
|
259
|
|
260 conf = ngx_http_get_module_loc_conf(r, ngx_http_image_filter_module);
|
|
261
|
|
262 if (ctx->type == NGX_HTTP_IMAGE_NONE) {
|
|
263
|
|
264 if (conf->filter == NGX_HTTP_IMAGE_SIZE) {
|
|
265 out.buf = ngx_http_image_json(r, NULL);
|
|
266
|
|
267 if (out.buf) {
|
|
268 out.next = NULL;
|
488
|
269 ctx->phase = NGX_HTTP_IMAGE_DONE;
|
484
|
270
|
488
|
271 return ngx_http_image_send(r, ctx, &out);
|
484
|
272 }
|
|
273 }
|
|
274
|
|
275 return ngx_http_filter_finalize_request(r,
|
488
|
276 &ngx_http_image_filter_module,
|
484
|
277 NGX_HTTP_UNSUPPORTED_MEDIA_TYPE);
|
|
278 }
|
|
279
|
|
280 /* override content type */
|
|
281
|
|
282 ct = &ngx_http_image_types[ctx->type - 1];
|
|
283 r->headers_out.content_type_len = ct->len;
|
|
284 r->headers_out.content_type = *ct;
|
494
|
285 r->headers_out.content_type_lowcase = NULL;
|
484
|
286
|
|
287 if (conf->filter == NGX_HTTP_IMAGE_TEST) {
|
488
|
288 ctx->phase = NGX_HTTP_IMAGE_PASS;
|
|
289
|
|
290 return ngx_http_image_send(r, ctx, in);
|
484
|
291 }
|
|
292
|
|
293 ctx->phase = NGX_HTTP_IMAGE_READ;
|
|
294
|
|
295 /* fall through */
|
|
296
|
|
297 case NGX_HTTP_IMAGE_READ:
|
|
298
|
|
299 rc = ngx_http_image_read(r, in);
|
|
300
|
|
301 if (rc == NGX_AGAIN) {
|
|
302 return NGX_OK;
|
|
303 }
|
|
304
|
|
305 if (rc == NGX_ERROR) {
|
|
306 return ngx_http_filter_finalize_request(r,
|
488
|
307 &ngx_http_image_filter_module,
|
484
|
308 NGX_HTTP_UNSUPPORTED_MEDIA_TYPE);
|
|
309 }
|
|
310
|
|
311 /* fall through */
|
|
312
|
|
313 case NGX_HTTP_IMAGE_PROCESS:
|
|
314
|
|
315 out.buf = ngx_http_image_process(r);
|
|
316
|
|
317 if (out.buf == NULL) {
|
|
318 return ngx_http_filter_finalize_request(r,
|
488
|
319 &ngx_http_image_filter_module,
|
484
|
320 NGX_HTTP_UNSUPPORTED_MEDIA_TYPE);
|
|
321 }
|
|
322
|
|
323 out.next = NULL;
|
488
|
324 ctx->phase = NGX_HTTP_IMAGE_PASS;
|
|
325
|
|
326 return ngx_http_image_send(r, ctx, &out);
|
484
|
327
|
488
|
328 case NGX_HTTP_IMAGE_PASS:
|
|
329
|
|
330 return ngx_http_next_body_filter(r, in);
|
484
|
331
|
|
332 default: /* NGX_HTTP_IMAGE_DONE */
|
|
333
|
488
|
334 rc = ngx_http_next_body_filter(r, NULL);
|
|
335
|
|
336 /* NGX_ERROR resets any pending data */
|
|
337 return (rc == NGX_OK) ? NGX_ERROR : rc;
|
484
|
338 }
|
488
|
339 }
|
484
|
340
|
488
|
341
|
|
342 static ngx_int_t
|
|
343 ngx_http_image_send(ngx_http_request_t *r, ngx_http_image_filter_ctx_t *ctx,
|
|
344 ngx_chain_t *in)
|
|
345 {
|
|
346 ngx_int_t rc;
|
484
|
347
|
|
348 rc = ngx_http_next_header_filter(r);
|
|
349
|
|
350 if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
|
488
|
351 return NGX_ERROR;
|
484
|
352 }
|
|
353
|
488
|
354 rc = ngx_http_next_body_filter(r, in);
|
|
355
|
|
356 if (ctx->phase == NGX_HTTP_IMAGE_DONE) {
|
|
357 /* NGX_ERROR resets any pending data */
|
|
358 return (rc == NGX_OK) ? NGX_ERROR : rc;
|
|
359 }
|
|
360
|
|
361 return rc;
|
484
|
362 }
|
|
363
|
|
364
|
|
365 static ngx_uint_t
|
|
366 ngx_http_image_test(ngx_http_request_t *r, ngx_chain_t *in)
|
|
367 {
|
|
368 u_char *p;
|
|
369
|
|
370 p = in->buf->pos;
|
|
371
|
|
372 if (in->buf->last - p < 16) {
|
|
373 return NGX_HTTP_IMAGE_NONE;
|
|
374 }
|
|
375
|
|
376 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
377 "image filter: \"%c%c\"", p[0], p[1]);
|
|
378
|
|
379 if (p[0] == 0xff && p[1] == 0xd8) {
|
|
380
|
|
381 /* JPEG */
|
|
382
|
|
383 return NGX_HTTP_IMAGE_JPEG;
|
|
384
|
|
385 } else if (p[0] == 'G' && p[1] == 'I' && p[2] == 'F' && p[3] == '8'
|
|
386 && p[4] == '9' && p[5] == 'a')
|
|
387 {
|
|
388 /* GIF */
|
|
389
|
|
390 return NGX_HTTP_IMAGE_GIF;
|
|
391
|
|
392 } else if (p[0] == 0x89 && p[1] == 'P' && p[2] == 'N' && p[3] == 'G'
|
|
393 && p[4] == 0x0d && p[5] == 0x0a && p[6] == 0x1a && p[7] == 0x0a)
|
|
394 {
|
|
395 /* PNG */
|
|
396
|
|
397 return NGX_HTTP_IMAGE_PNG;
|
|
398 }
|
|
399
|
|
400 return NGX_HTTP_IMAGE_NONE;
|
|
401 }
|
|
402
|
|
403
|
|
404 static ngx_int_t
|
|
405 ngx_http_image_read(ngx_http_request_t *r, ngx_chain_t *in)
|
|
406 {
|
|
407 u_char *p;
|
|
408 size_t size, rest;
|
|
409 ngx_buf_t *b;
|
|
410 ngx_chain_t *cl;
|
|
411 ngx_http_image_filter_ctx_t *ctx;
|
|
412
|
|
413 ctx = ngx_http_get_module_ctx(r, ngx_http_image_filter_module);
|
|
414
|
|
415 if (ctx->image == NULL) {
|
|
416 ctx->image = ngx_palloc(r->pool, ctx->length);
|
|
417 if (ctx->image == NULL) {
|
|
418 return NGX_ERROR;
|
|
419 }
|
|
420
|
|
421 ctx->last = ctx->image;
|
|
422 }
|
|
423
|
|
424 p = ctx->last;
|
|
425
|
|
426 for (cl = in; cl; cl = cl->next) {
|
|
427
|
|
428 b = cl->buf;
|
|
429 size = b->last - b->pos;
|
|
430
|
|
431 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
432 "image buf: %uz", size);
|
|
433
|
|
434 rest = ctx->image + ctx->length - p;
|
|
435 size = (rest < size) ? rest : size;
|
|
436
|
|
437 p = ngx_cpymem(p, b->pos, size);
|
|
438 b->pos += size;
|
|
439
|
|
440 if (b->last_buf) {
|
|
441 ctx->last = p;
|
|
442 return NGX_OK;
|
|
443 }
|
|
444 }
|
|
445
|
|
446 ctx->last = p;
|
|
447 r->connection->buffered |= NGX_HTTP_IMAGE_BUFFERED;
|
|
448
|
|
449 return NGX_AGAIN;
|
|
450 }
|
|
451
|
|
452
|
|
453 static ngx_buf_t *
|
|
454 ngx_http_image_process(ngx_http_request_t *r)
|
|
455 {
|
|
456 ngx_buf_t *b;
|
|
457 ngx_int_t rc;
|
|
458 ngx_http_image_filter_ctx_t *ctx;
|
|
459 ngx_http_image_filter_conf_t *conf;
|
|
460
|
|
461 r->connection->buffered &= ~NGX_HTTP_IMAGE_BUFFERED;
|
|
462
|
|
463 ctx = ngx_http_get_module_ctx(r, ngx_http_image_filter_module);
|
|
464
|
|
465 rc = ngx_http_image_size(r, ctx);
|
|
466
|
|
467 conf = ngx_http_get_module_loc_conf(r, ngx_http_image_filter_module);
|
|
468
|
|
469 if (conf->filter == NGX_HTTP_IMAGE_SIZE) {
|
|
470
|
|
471 b = ngx_http_image_json(r, rc == NGX_OK ? ctx : NULL);
|
|
472
|
|
473 } else if (rc == NGX_OK
|
|
474 && ctx->width <= conf->width
|
|
475 && ctx->height <= conf->height)
|
|
476 {
|
|
477 b = ngx_http_image_asis(r, ctx);
|
|
478
|
|
479 } else {
|
|
480 b = ngx_http_image_resize(r, ctx);
|
|
481 }
|
|
482
|
|
483 return b;
|
|
484 }
|
|
485
|
|
486
|
|
487 static ngx_buf_t *
|
|
488 ngx_http_image_json(ngx_http_request_t *r, ngx_http_image_filter_ctx_t *ctx)
|
|
489 {
|
|
490 size_t len;
|
|
491 ngx_buf_t *b;
|
|
492
|
|
493 b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t));
|
|
494 if (b == NULL) {
|
|
495 return NULL;
|
|
496 }
|
|
497
|
|
498 b->memory = 1;
|
|
499 b->last_buf = 1;
|
|
500
|
|
501 ngx_http_clean_header(r);
|
|
502
|
|
503 r->headers_out.status = NGX_HTTP_OK;
|
|
504 r->headers_out.content_type.len = sizeof("text/plain") - 1;
|
|
505 r->headers_out.content_type.data = (u_char *) "text/plain";
|
494
|
506 r->headers_out.content_type_lowcase = NULL;
|
484
|
507
|
|
508 if (ctx == NULL) {
|
|
509 b->pos = (u_char *) "{}" CRLF;
|
|
510 b->last = b->pos + sizeof("{}" CRLF) - 1;
|
|
511
|
|
512 ngx_http_image_length(r, b);
|
|
513
|
|
514 return b;
|
|
515 }
|
|
516
|
|
517 len = sizeof("{ \"img\" : "
|
|
518 "{ \"width\": , \"height\": , \"type\": \"jpeg\" } }" CRLF) - 1
|
|
519 + 2 * NGX_SIZE_T_LEN;
|
|
520
|
|
521 b->pos = ngx_pnalloc(r->pool, len);
|
|
522 if (b->pos == NULL) {
|
|
523 return NULL;
|
|
524 }
|
|
525
|
|
526 b->last = ngx_sprintf(b->pos,
|
|
527 "{ \"img\" : "
|
|
528 "{ \"width\": %uz,"
|
|
529 " \"height\": %uz,"
|
|
530 " \"type\": \"%s\" } }" CRLF,
|
|
531 ctx->width, ctx->height,
|
|
532 ngx_http_image_types[ctx->type - 1].data + 6);
|
|
533
|
|
534 ngx_http_image_length(r, b);
|
|
535
|
|
536 return b;
|
|
537 }
|
|
538
|
|
539
|
|
540 static ngx_buf_t *
|
|
541 ngx_http_image_asis(ngx_http_request_t *r, ngx_http_image_filter_ctx_t *ctx)
|
|
542 {
|
|
543 ngx_buf_t *b;
|
|
544
|
|
545 b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t));
|
|
546 if (b == NULL) {
|
|
547 return NULL;
|
|
548 }
|
|
549
|
|
550 b->pos = ctx->image;
|
|
551 b->last = ctx->last;
|
|
552 b->memory = 1;
|
|
553 b->last_buf = 1;
|
|
554
|
|
555 ngx_http_image_length(r, b);
|
|
556
|
|
557 return b;
|
|
558 }
|
|
559
|
|
560
|
|
561 static void
|
|
562 ngx_http_image_length(ngx_http_request_t *r, ngx_buf_t *b)
|
|
563 {
|
|
564 r->headers_out.content_length_n = b->last - b->pos;
|
|
565
|
|
566 if (r->headers_out.content_length) {
|
|
567 r->headers_out.content_length->hash = 0;
|
|
568 }
|
|
569
|
|
570 r->headers_out.content_length = NULL;
|
|
571 }
|
|
572
|
|
573
|
|
574 static ngx_int_t
|
|
575 ngx_http_image_size(ngx_http_request_t *r, ngx_http_image_filter_ctx_t *ctx)
|
|
576 {
|
|
577 u_char *p, *last;
|
|
578 ngx_uint_t width, height;
|
|
579
|
|
580 p = ctx->image;
|
|
581
|
|
582 switch (ctx->type) {
|
|
583
|
|
584 case NGX_HTTP_IMAGE_JPEG:
|
|
585
|
|
586 p += 2;
|
|
587 last = ctx->image + ctx->length - 10;
|
|
588
|
|
589 while (p < last) {
|
|
590
|
|
591 if (p[0] == 0xff && p[1] != 0xff) {
|
|
592
|
|
593 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
594 "JPEG: %02xd %02xd", *p, *(p + 1));
|
|
595
|
|
596 p++;
|
|
597
|
|
598 if (*p == 0xc0 || *p == 0xc1 || *p == 0xc2 || *p == 0xc3
|
|
599 || *p == 0xc9 || *p == 0xca || *p == 0xcb)
|
|
600 {
|
|
601 goto found;
|
|
602 }
|
|
603
|
|
604 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
605 "JPEG: %02xd %02xd", p[1], p[2]);
|
|
606
|
|
607 p += p[1] * 256 + p[2];
|
|
608
|
|
609 continue;
|
|
610 }
|
|
611
|
|
612 p++;
|
|
613 }
|
|
614
|
|
615 return NGX_DECLINED;
|
|
616
|
|
617 found:
|
|
618
|
|
619 width = p[6] * 256 + p[7];
|
|
620 height = p[4] * 256 + p[5];
|
|
621
|
|
622 break;
|
|
623
|
|
624 case NGX_HTTP_IMAGE_GIF:
|
|
625
|
|
626 if (ctx->length < 10) {
|
|
627 return NGX_DECLINED;
|
|
628 }
|
|
629
|
|
630 width = p[7] * 256 + p[6];
|
|
631 height = p[9] * 256 + p[8];
|
|
632
|
|
633 break;
|
|
634
|
|
635 case NGX_HTTP_IMAGE_PNG:
|
|
636
|
|
637 if (ctx->length < 24) {
|
|
638 return NGX_DECLINED;
|
|
639 }
|
|
640
|
|
641 width = p[18] * 256 + p[19];
|
|
642 height = p[22] * 256 + p[23];
|
|
643
|
|
644 break;
|
|
645
|
|
646 default:
|
|
647
|
|
648 return NGX_DECLINED;
|
|
649 }
|
|
650
|
|
651 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
652 "image size: %d x %d", width, height);
|
|
653
|
|
654 ctx->width = width;
|
|
655 ctx->height = height;
|
|
656
|
|
657 return NGX_OK;
|
|
658 }
|
|
659
|
|
660
|
|
661 static ngx_buf_t *
|
|
662 ngx_http_image_resize(ngx_http_request_t *r, ngx_http_image_filter_ctx_t *ctx)
|
|
663 {
|
|
664 int sx, sy, dx, dy, ox, oy,
|
|
665 colors, transparent, size;
|
|
666 u_char *out;
|
|
667 ngx_buf_t *b;
|
|
668 ngx_uint_t resize;
|
|
669 gdImagePtr src, dst;
|
|
670 ngx_pool_cleanup_t *cln;
|
|
671 ngx_http_image_filter_conf_t *conf;
|
|
672
|
|
673 src = ngx_http_image_source(r, ctx);
|
|
674
|
|
675 if (src == NULL) {
|
|
676 return NULL;
|
|
677 }
|
|
678
|
|
679 sx = gdImageSX(src);
|
|
680 sy = gdImageSY(src);
|
|
681
|
|
682 conf = ngx_http_get_module_loc_conf(r, ngx_http_image_filter_module);
|
|
683
|
|
684 if ((ngx_uint_t) sx <= conf->width && (ngx_uint_t) sy <= conf->height) {
|
|
685 gdImageDestroy(src);
|
|
686 return ngx_http_image_asis(r, ctx);
|
|
687 }
|
|
688
|
|
689 colors = gdImageColorsTotal(src);
|
|
690 transparent = gdImageGetTransparent(src);
|
|
691
|
|
692 dx = sx;
|
|
693 dy = sy;
|
|
694
|
|
695 if (conf->filter == NGX_HTTP_IMAGE_RESIZE) {
|
|
696
|
|
697 if ((ngx_uint_t) dx > conf->width) {
|
|
698 dy = dy * conf->width / dx;
|
|
699 dy = dy ? dy : 1;
|
|
700 dx = conf->width;
|
|
701 }
|
|
702
|
|
703 if ((ngx_uint_t) dy > conf->height) {
|
|
704 dx = dx * conf->height / dy;
|
|
705 dx = dx ? dx : 1;
|
|
706 dy = conf->height;
|
|
707 }
|
|
708
|
|
709 resize = 1;
|
|
710
|
|
711 } else { /* NGX_HTTP_IMAGE_CROP */
|
|
712
|
|
713 resize = 0;
|
|
714
|
|
715 if ((ngx_uint_t) (dx * 100 / dy) < conf->width * 100 / conf->height) {
|
|
716
|
|
717 if ((ngx_uint_t) dx > conf->width) {
|
|
718 dy = dy * conf->width / dx;
|
|
719 dy = dy ? dy : 1;
|
|
720 dx = conf->width;
|
|
721 resize = 1;
|
|
722 }
|
|
723
|
|
724 } else {
|
|
725 if ((ngx_uint_t) dy > conf->height) {
|
|
726 dx = dx * conf->height / dy;
|
|
727 dx = dx ? dx : 1;
|
|
728 dy = conf->height;
|
|
729 resize = 1;
|
|
730 }
|
|
731 }
|
|
732 }
|
|
733
|
|
734 if (resize) {
|
|
735 dst = ngx_http_image_new(r, dx, dy, colors);
|
|
736 if (dst == NULL) {
|
|
737 gdImageDestroy(src);
|
|
738 return NULL;
|
|
739 }
|
|
740
|
|
741 gdImageCopyResampled(dst, src, 0, 0, 0, 0, dx, dy, sx, sy);
|
|
742
|
|
743 gdImageDestroy(src);
|
|
744
|
|
745 } else {
|
|
746 dst = src;
|
|
747 }
|
|
748
|
|
749 if (conf->filter == NGX_HTTP_IMAGE_CROP) {
|
|
750
|
|
751 src = dst;
|
|
752
|
|
753 if ((ngx_uint_t) dx > conf->width) {
|
|
754 ox = dx - conf->width;
|
|
755
|
|
756 } else {
|
|
757 ox = 0;
|
|
758 }
|
|
759
|
|
760 if ((ngx_uint_t) dy > conf->height) {
|
|
761 oy = dy - conf->height;
|
|
762
|
|
763 } else {
|
|
764 oy = 0;
|
|
765 }
|
|
766
|
|
767 if (ox || oy) {
|
|
768
|
|
769 dst = ngx_http_image_new(r, dx - ox, dy - oy, colors);
|
|
770
|
|
771 if (dst == NULL) {
|
|
772 gdImageDestroy(src);
|
|
773 return NULL;
|
|
774 }
|
|
775
|
|
776 ox /= 2;
|
|
777 oy /= 2;
|
|
778
|
|
779 ngx_log_debug4(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
780 "image crop: %d x %d @ %d x %d",
|
|
781 dx, dy, ox, oy);
|
|
782
|
|
783 gdImageCopy(dst, src, 0, 0, ox, oy, dx - ox, dy - oy);
|
|
784
|
|
785 gdImageDestroy(src);
|
|
786 }
|
|
787 }
|
|
788
|
|
789 gdImageColorTransparent(dst, transparent);
|
|
790
|
|
791 out = ngx_http_image_out(r, ctx->type, dst, &size);
|
|
792
|
|
793 ngx_log_debug3(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
794 "image: %d x %d %d", sx, sy, colors);
|
|
795
|
|
796 gdImageDestroy(dst);
|
|
797 ngx_pfree(r->pool, ctx->image);
|
|
798
|
|
799 if (out == NULL) {
|
|
800 return NULL;
|
|
801 }
|
|
802
|
|
803 cln = ngx_pool_cleanup_add(r->pool, 0);
|
|
804 if (cln == NULL) {
|
|
805 gdFree(out);
|
|
806 return NULL;
|
|
807 }
|
|
808
|
|
809 b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t));
|
|
810 if (b == NULL) {
|
|
811 gdFree(out);
|
|
812 return NULL;
|
|
813 }
|
|
814
|
|
815 cln->handler = ngx_http_image_cleanup;
|
|
816 cln->data = out;
|
|
817
|
|
818 b->pos = out;
|
|
819 b->last = out + size;
|
|
820 b->memory = 1;
|
|
821 b->last_buf = 1;
|
|
822
|
|
823 ngx_http_image_length(r, b);
|
|
824
|
|
825 return b;
|
|
826 }
|
|
827
|
|
828
|
|
829 static gdImagePtr
|
|
830 ngx_http_image_source(ngx_http_request_t *r, ngx_http_image_filter_ctx_t *ctx)
|
|
831 {
|
|
832 char *failed;
|
|
833 gdImagePtr img;
|
|
834
|
|
835 img = NULL;
|
|
836
|
|
837 switch (ctx->type) {
|
|
838
|
|
839 case NGX_HTTP_IMAGE_JPEG:
|
|
840 img = gdImageCreateFromJpegPtr(ctx->length, ctx->image);
|
|
841 failed = "gdImageCreateFromJpegPtr() failed";
|
|
842 break;
|
|
843
|
|
844 case NGX_HTTP_IMAGE_GIF:
|
|
845 img = gdImageCreateFromGifPtr(ctx->length, ctx->image);
|
|
846 failed = "gdImageCreateFromGifPtr() failed";
|
|
847 break;
|
|
848
|
|
849 case NGX_HTTP_IMAGE_PNG:
|
|
850 img = gdImageCreateFromPngPtr(ctx->length, ctx->image);
|
|
851 failed = "gdImageCreateFromPngPtr() failed";
|
|
852 break;
|
|
853
|
|
854 default:
|
|
855 failed = "unknown image type";
|
|
856 break;
|
|
857 }
|
|
858
|
|
859 if (img == NULL) {
|
|
860 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, failed);
|
|
861 }
|
|
862
|
|
863 return img;
|
|
864 }
|
|
865
|
|
866
|
|
867 static gdImagePtr
|
|
868 ngx_http_image_new(ngx_http_request_t *r, int w, int h, int colors)
|
|
869 {
|
|
870 gdImagePtr img;
|
|
871
|
|
872 if (colors == 0) {
|
|
873 img = gdImageCreateTrueColor(w, h);
|
|
874
|
|
875 if (img == NULL) {
|
|
876 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
|
|
877 "gdImageCreateTrueColor() failed");
|
|
878 return NULL;
|
|
879 }
|
|
880
|
|
881 } else {
|
|
882 img = gdImageCreate(w, h);
|
|
883
|
|
884 if (img == NULL) {
|
|
885 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
|
|
886 "gdImageCreate() failed");
|
|
887 return NULL;
|
|
888 }
|
|
889 }
|
|
890
|
|
891 return img;
|
|
892 }
|
|
893
|
|
894
|
|
895 static u_char *
|
|
896 ngx_http_image_out(ngx_http_request_t *r, ngx_uint_t type, gdImagePtr img,
|
|
897 int *size)
|
|
898 {
|
492
|
899 char *failed;
|
|
900 u_char *out;
|
|
901 ngx_http_image_filter_conf_t *conf;
|
484
|
902
|
|
903 out = NULL;
|
|
904
|
|
905 switch (type) {
|
|
906
|
|
907 case NGX_HTTP_IMAGE_JPEG:
|
492
|
908 conf = ngx_http_get_module_loc_conf(r, ngx_http_image_filter_module);
|
|
909 out = gdImageJpegPtr(img, size, conf->jpeg_quality);
|
484
|
910 failed = "gdImageJpegPtr() failed";
|
|
911 break;
|
|
912
|
|
913 case NGX_HTTP_IMAGE_GIF:
|
|
914 out = gdImageGifPtr(img, size);
|
|
915 failed = "gdImageGifPtr() failed";
|
|
916 break;
|
|
917
|
|
918 case NGX_HTTP_IMAGE_PNG:
|
|
919 out = gdImagePngPtr(img, size);
|
|
920 failed = "gdImagePngPtr() failed";
|
|
921 break;
|
|
922
|
|
923 default:
|
|
924 failed = "unknown image type";
|
|
925 break;
|
|
926 }
|
|
927
|
|
928 if (out == NULL) {
|
|
929 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, failed);
|
|
930 }
|
|
931
|
|
932 return out;
|
|
933 }
|
|
934
|
|
935
|
|
936 static void
|
|
937 ngx_http_image_cleanup(void *data)
|
|
938 {
|
|
939 gdFree(data);
|
|
940 }
|
|
941
|
|
942
|
|
943 static void *
|
|
944 ngx_http_image_filter_create_conf(ngx_conf_t *cf)
|
|
945 {
|
|
946 ngx_http_image_filter_conf_t *conf;
|
|
947
|
|
948 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_image_filter_conf_t));
|
|
949 if (conf == NULL) {
|
|
950 return NGX_CONF_ERROR;
|
|
951 }
|
|
952
|
|
953 conf->filter = NGX_CONF_UNSET_UINT;
|
492
|
954 conf->jpeg_quality = NGX_CONF_UNSET;
|
484
|
955 conf->buffer_size = NGX_CONF_UNSET_SIZE;
|
|
956
|
|
957 return conf;
|
|
958 }
|
|
959
|
|
960
|
|
961 static char *
|
|
962 ngx_http_image_filter_merge_conf(ngx_conf_t *cf, void *parent, void *child)
|
|
963 {
|
|
964 ngx_http_image_filter_conf_t *prev = parent;
|
|
965 ngx_http_image_filter_conf_t *conf = child;
|
|
966
|
|
967 if (conf->filter == NGX_CONF_UNSET_UINT) {
|
|
968
|
|
969 if (prev->filter == NGX_CONF_UNSET_UINT) {
|
|
970 conf->filter = NGX_HTTP_IMAGE_OFF;
|
|
971
|
|
972 } else {
|
|
973 conf->filter = prev->filter;
|
|
974 conf->width = prev->width;
|
|
975 conf->height = prev->height;
|
|
976 }
|
|
977 }
|
|
978
|
492
|
979 /* 75 is libjpeg default quality */
|
|
980 ngx_conf_merge_value(conf->jpeg_quality, prev->jpeg_quality, 75);
|
|
981
|
484
|
982 ngx_conf_merge_size_value(conf->buffer_size, prev->buffer_size,
|
|
983 1 * 1024 * 1024);
|
|
984
|
|
985 return NGX_CONF_OK;
|
|
986 }
|
|
987
|
|
988
|
|
989 static char *
|
|
990 ngx_http_image_filter(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
991 {
|
|
992 ngx_http_image_filter_conf_t *imcf = conf;
|
|
993
|
|
994 ngx_str_t *value;
|
|
995 ngx_int_t n;
|
|
996 ngx_uint_t i;
|
|
997
|
|
998 value = cf->args->elts;
|
|
999
|
|
1000 i = 1;
|
|
1001
|
|
1002 if (cf->args->nelts == 2) {
|
|
1003 if (ngx_strcmp(value[i].data, "off") == 0) {
|
|
1004 imcf->filter = NGX_HTTP_IMAGE_OFF;
|
|
1005
|
|
1006 } else if (ngx_strcmp(value[i].data, "test") == 0) {
|
|
1007 imcf->filter = NGX_HTTP_IMAGE_TEST;
|
|
1008
|
|
1009 } else if (ngx_strcmp(value[i].data, "size") == 0) {
|
|
1010 imcf->filter = NGX_HTTP_IMAGE_SIZE;
|
|
1011
|
|
1012 } else {
|
|
1013 goto failed;
|
|
1014 }
|
|
1015
|
|
1016 return NGX_CONF_OK;
|
|
1017 }
|
|
1018
|
|
1019 if (ngx_strcmp(value[i].data, "resize") == 0) {
|
|
1020 imcf->filter = NGX_HTTP_IMAGE_RESIZE;
|
|
1021
|
|
1022 } else if (ngx_strcmp(value[i].data, "crop") == 0) {
|
|
1023 imcf->filter = NGX_HTTP_IMAGE_CROP;
|
|
1024
|
|
1025 } else {
|
|
1026 goto failed;
|
|
1027 }
|
|
1028
|
|
1029 i++;
|
|
1030
|
|
1031 if (value[i].len == 1 && value[i].data[0] == '-') {
|
|
1032 imcf->width = (ngx_uint_t) -1;
|
|
1033
|
|
1034 } else {
|
|
1035 n = ngx_atoi(value[i].data, value[i].len);
|
|
1036 if (n == NGX_ERROR) {
|
|
1037 goto failed;
|
|
1038 }
|
|
1039
|
|
1040 imcf->width = (ngx_uint_t) n;
|
|
1041 }
|
|
1042
|
|
1043 i++;
|
|
1044
|
|
1045 if (value[i].len == 1 && value[i].data[0] == '-') {
|
|
1046 imcf->height = (ngx_uint_t) -1;
|
|
1047
|
|
1048 } else {
|
|
1049 n = ngx_atoi(value[i].data, value[i].len);
|
|
1050 if (n == NGX_ERROR) {
|
|
1051 goto failed;
|
|
1052 }
|
|
1053
|
|
1054 imcf->height = (ngx_uint_t) n;
|
|
1055 }
|
|
1056
|
|
1057 return NGX_CONF_OK;
|
|
1058
|
|
1059 failed:
|
|
1060
|
|
1061 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "invalid parameter \"%V\"",
|
|
1062 &value[i]);
|
|
1063
|
|
1064 return NGX_CONF_ERROR;
|
|
1065 }
|
|
1066
|
|
1067
|
|
1068 static ngx_int_t
|
|
1069 ngx_http_image_filter_init(ngx_conf_t *cf)
|
|
1070 {
|
|
1071 ngx_http_next_header_filter = ngx_http_top_header_filter;
|
|
1072 ngx_http_top_header_filter = ngx_http_image_header_filter;
|
|
1073
|
|
1074 ngx_http_next_body_filter = ngx_http_top_body_filter;
|
|
1075 ngx_http_top_body_filter = ngx_http_image_body_filter;
|
|
1076
|
|
1077 return NGX_OK;
|
|
1078 }
|