source: nginx/src/http/modules/ngx_http_mp4_module.c @ 5621:345e4fd4bb64

Last change on this file since 5621:345e4fd4bb64 was 5621:345e4fd4bb64, checked in by Maxim Dounin <mdounin@…>, 4 years ago

Range filter: single_range flag in request.

If set, it means that response body is going to be in more than one buffer,
hence only range requests with a single range should be honored.

The flag is now used by mp4 and cacheable upstream responses, thus allowing
range requests of mp4 files with start/end, as well as range processing
on a first request to a not-yet-cached files with proxy_cache.

Notably this makes it possible to play mp4 files (with proxy_cache, or with
mp4 module) on iOS devices, as byte-range support is required by Apple.

File size: 99.4 KB
Line 
1
2/*
3 * Copyright (C) Igor Sysoev
4 * Copyright (C) Nginx, Inc.
5 */
6
7#include <ngx_config.h>
8#include <ngx_core.h>
9#include <ngx_http.h>
10
11
12#define NGX_HTTP_MP4_TRAK_ATOM     0
13#define NGX_HTTP_MP4_TKHD_ATOM     1
14#define NGX_HTTP_MP4_MDIA_ATOM     2
15#define NGX_HTTP_MP4_MDHD_ATOM     3
16#define NGX_HTTP_MP4_HDLR_ATOM     4
17#define NGX_HTTP_MP4_MINF_ATOM     5
18#define NGX_HTTP_MP4_VMHD_ATOM     6
19#define NGX_HTTP_MP4_SMHD_ATOM     7
20#define NGX_HTTP_MP4_DINF_ATOM     8
21#define NGX_HTTP_MP4_STBL_ATOM     9
22#define NGX_HTTP_MP4_STSD_ATOM    10
23#define NGX_HTTP_MP4_STTS_ATOM    11
24#define NGX_HTTP_MP4_STTS_DATA    12
25#define NGX_HTTP_MP4_STSS_ATOM    13
26#define NGX_HTTP_MP4_STSS_DATA    14
27#define NGX_HTTP_MP4_CTTS_ATOM    15
28#define NGX_HTTP_MP4_CTTS_DATA    16
29#define NGX_HTTP_MP4_STSC_ATOM    17
30#define NGX_HTTP_MP4_STSC_START   18
31#define NGX_HTTP_MP4_STSC_DATA    19
32#define NGX_HTTP_MP4_STSC_END     20
33#define NGX_HTTP_MP4_STSZ_ATOM    21
34#define NGX_HTTP_MP4_STSZ_DATA    22
35#define NGX_HTTP_MP4_STCO_ATOM    23
36#define NGX_HTTP_MP4_STCO_DATA    24
37#define NGX_HTTP_MP4_CO64_ATOM    25
38#define NGX_HTTP_MP4_CO64_DATA    26
39
40#define NGX_HTTP_MP4_LAST_ATOM    NGX_HTTP_MP4_CO64_DATA
41
42
43typedef struct {
44    size_t                buffer_size;
45    size_t                max_buffer_size;
46} ngx_http_mp4_conf_t;
47
48
49typedef struct {
50    u_char                chunk[4];
51    u_char                samples[4];
52    u_char                id[4];
53} ngx_mp4_stsc_entry_t;
54
55
56typedef struct {
57    uint32_t              timescale;
58    uint32_t              time_to_sample_entries;
59    uint32_t              sample_to_chunk_entries;
60    uint32_t              sync_samples_entries;
61    uint32_t              composition_offset_entries;
62    uint32_t              sample_sizes_entries;
63    uint32_t              chunks;
64
65    ngx_uint_t            start_sample;
66    ngx_uint_t            end_sample;
67    ngx_uint_t            start_chunk;
68    ngx_uint_t            end_chunk;
69    ngx_uint_t            start_chunk_samples;
70    ngx_uint_t            end_chunk_samples;
71    uint64_t              start_chunk_samples_size;
72    uint64_t              end_chunk_samples_size;
73    off_t                 start_offset;
74    off_t                 end_offset;
75
76    size_t                tkhd_size;
77    size_t                mdhd_size;
78    size_t                hdlr_size;
79    size_t                vmhd_size;
80    size_t                smhd_size;
81    size_t                dinf_size;
82    size_t                size;
83
84    ngx_chain_t           out[NGX_HTTP_MP4_LAST_ATOM + 1];
85
86    ngx_buf_t             trak_atom_buf;
87    ngx_buf_t             tkhd_atom_buf;
88    ngx_buf_t             mdia_atom_buf;
89    ngx_buf_t             mdhd_atom_buf;
90    ngx_buf_t             hdlr_atom_buf;
91    ngx_buf_t             minf_atom_buf;
92    ngx_buf_t             vmhd_atom_buf;
93    ngx_buf_t             smhd_atom_buf;
94    ngx_buf_t             dinf_atom_buf;
95    ngx_buf_t             stbl_atom_buf;
96    ngx_buf_t             stsd_atom_buf;
97    ngx_buf_t             stts_atom_buf;
98    ngx_buf_t             stts_data_buf;
99    ngx_buf_t             stss_atom_buf;
100    ngx_buf_t             stss_data_buf;
101    ngx_buf_t             ctts_atom_buf;
102    ngx_buf_t             ctts_data_buf;
103    ngx_buf_t             stsc_atom_buf;
104    ngx_buf_t             stsc_start_chunk_buf;
105    ngx_buf_t             stsc_end_chunk_buf;
106    ngx_buf_t             stsc_data_buf;
107    ngx_buf_t             stsz_atom_buf;
108    ngx_buf_t             stsz_data_buf;
109    ngx_buf_t             stco_atom_buf;
110    ngx_buf_t             stco_data_buf;
111    ngx_buf_t             co64_atom_buf;
112    ngx_buf_t             co64_data_buf;
113
114    ngx_mp4_stsc_entry_t  stsc_start_chunk_entry;
115    ngx_mp4_stsc_entry_t  stsc_end_chunk_entry;
116} ngx_http_mp4_trak_t;
117
118
119typedef struct {
120    ngx_file_t            file;
121
122    u_char               *buffer;
123    u_char               *buffer_start;
124    u_char               *buffer_pos;
125    u_char               *buffer_end;
126    size_t                buffer_size;
127
128    off_t                 offset;
129    off_t                 end;
130    off_t                 content_length;
131    ngx_uint_t            start;
132    ngx_uint_t            length;
133    uint32_t              timescale;
134    ngx_http_request_t   *request;
135    ngx_array_t           trak;
136    ngx_http_mp4_trak_t   traks[2];
137
138    size_t                ftyp_size;
139    size_t                moov_size;
140
141    ngx_chain_t          *out;
142    ngx_chain_t           ftyp_atom;
143    ngx_chain_t           moov_atom;
144    ngx_chain_t           mvhd_atom;
145    ngx_chain_t           mdat_atom;
146    ngx_chain_t           mdat_data;
147
148    ngx_buf_t             ftyp_atom_buf;
149    ngx_buf_t             moov_atom_buf;
150    ngx_buf_t             mvhd_atom_buf;
151    ngx_buf_t             mdat_atom_buf;
152    ngx_buf_t             mdat_data_buf;
153
154    u_char                moov_atom_header[8];
155    u_char                mdat_atom_header[16];
156} ngx_http_mp4_file_t;
157
158
159typedef struct {
160    char                 *name;
161    ngx_int_t           (*handler)(ngx_http_mp4_file_t *mp4,
162                                   uint64_t atom_data_size);
163} ngx_http_mp4_atom_handler_t;
164
165
166#define ngx_mp4_atom_header(mp4)   (mp4->buffer_pos - 8)
167#define ngx_mp4_atom_data(mp4)     mp4->buffer_pos
168#define ngx_mp4_atom_data_size(t)  (uint64_t) (sizeof(t) - 8)
169
170
171#define ngx_mp4_atom_next(mp4, n)                                             \
172    mp4->buffer_pos += (size_t) n;                                            \
173    mp4->offset += n
174
175
176#define ngx_mp4_set_atom_name(p, n1, n2, n3, n4)                              \
177    ((u_char *) (p))[4] = n1;                                                 \
178    ((u_char *) (p))[5] = n2;                                                 \
179    ((u_char *) (p))[6] = n3;                                                 \
180    ((u_char *) (p))[7] = n4
181
182#define ngx_mp4_get_32value(p)                                                \
183    ( ((uint32_t) ((u_char *) (p))[0] << 24)                                  \
184    + (           ((u_char *) (p))[1] << 16)                                  \
185    + (           ((u_char *) (p))[2] << 8)                                   \
186    + (           ((u_char *) (p))[3]) )
187
188#define ngx_mp4_set_32value(p, n)                                             \
189    ((u_char *) (p))[0] = (u_char) ((n) >> 24);                               \
190    ((u_char *) (p))[1] = (u_char) ((n) >> 16);                               \
191    ((u_char *) (p))[2] = (u_char) ((n) >> 8);                                \
192    ((u_char *) (p))[3] = (u_char)  (n)
193
194#define ngx_mp4_get_64value(p)                                                \
195    ( ((uint64_t) ((u_char *) (p))[0] << 56)                                  \
196    + ((uint64_t) ((u_char *) (p))[1] << 48)                                  \
197    + ((uint64_t) ((u_char *) (p))[2] << 40)                                  \
198    + ((uint64_t) ((u_char *) (p))[3] << 32)                                  \
199    + ((uint64_t) ((u_char *) (p))[4] << 24)                                  \
200    + (           ((u_char *) (p))[5] << 16)                                  \
201    + (           ((u_char *) (p))[6] << 8)                                   \
202    + (           ((u_char *) (p))[7]) )
203
204#define ngx_mp4_set_64value(p, n)                                             \
205    ((u_char *) (p))[0] = (u_char) ((uint64_t) (n) >> 56);                    \
206    ((u_char *) (p))[1] = (u_char) ((uint64_t) (n) >> 48);                    \
207    ((u_char *) (p))[2] = (u_char) ((uint64_t) (n) >> 40);                    \
208    ((u_char *) (p))[3] = (u_char) ((uint64_t) (n) >> 32);                    \
209    ((u_char *) (p))[4] = (u_char) (           (n) >> 24);                    \
210    ((u_char *) (p))[5] = (u_char) (           (n) >> 16);                    \
211    ((u_char *) (p))[6] = (u_char) (           (n) >> 8);                     \
212    ((u_char *) (p))[7] = (u_char)             (n)
213
214#define ngx_mp4_last_trak(mp4)                                                \
215    &((ngx_http_mp4_trak_t *) mp4->trak.elts)[mp4->trak.nelts - 1]
216
217
218static ngx_int_t ngx_http_mp4_handler(ngx_http_request_t *r);
219
220static ngx_int_t ngx_http_mp4_process(ngx_http_mp4_file_t *mp4);
221static ngx_int_t ngx_http_mp4_read_atom(ngx_http_mp4_file_t *mp4,
222    ngx_http_mp4_atom_handler_t *atom, uint64_t atom_data_size);
223static ngx_int_t ngx_http_mp4_read(ngx_http_mp4_file_t *mp4, size_t size);
224static ngx_int_t ngx_http_mp4_read_ftyp_atom(ngx_http_mp4_file_t *mp4,
225    uint64_t atom_data_size);
226static ngx_int_t ngx_http_mp4_read_moov_atom(ngx_http_mp4_file_t *mp4,
227    uint64_t atom_data_size);
228static ngx_int_t ngx_http_mp4_read_mdat_atom(ngx_http_mp4_file_t *mp4,
229    uint64_t atom_data_size);
230static size_t ngx_http_mp4_update_mdat_atom(ngx_http_mp4_file_t *mp4,
231    off_t start_offset, off_t end_offset);
232static ngx_int_t ngx_http_mp4_read_mvhd_atom(ngx_http_mp4_file_t *mp4,
233    uint64_t atom_data_size);
234static ngx_int_t ngx_http_mp4_read_trak_atom(ngx_http_mp4_file_t *mp4,
235    uint64_t atom_data_size);
236static void ngx_http_mp4_update_trak_atom(ngx_http_mp4_file_t *mp4,
237    ngx_http_mp4_trak_t *trak);
238static ngx_int_t ngx_http_mp4_read_cmov_atom(ngx_http_mp4_file_t *mp4,
239    uint64_t atom_data_size);
240static ngx_int_t ngx_http_mp4_read_tkhd_atom(ngx_http_mp4_file_t *mp4,
241    uint64_t atom_data_size);
242static ngx_int_t ngx_http_mp4_read_mdia_atom(ngx_http_mp4_file_t *mp4,
243    uint64_t atom_data_size);
244static void ngx_http_mp4_update_mdia_atom(ngx_http_mp4_file_t *mp4,
245    ngx_http_mp4_trak_t *trak);
246static ngx_int_t ngx_http_mp4_read_mdhd_atom(ngx_http_mp4_file_t *mp4,
247    uint64_t atom_data_size);
248static ngx_int_t ngx_http_mp4_read_hdlr_atom(ngx_http_mp4_file_t *mp4,
249    uint64_t atom_data_size);
250static ngx_int_t ngx_http_mp4_read_minf_atom(ngx_http_mp4_file_t *mp4,
251    uint64_t atom_data_size);
252static void ngx_http_mp4_update_minf_atom(ngx_http_mp4_file_t *mp4,
253    ngx_http_mp4_trak_t *trak);
254static ngx_int_t ngx_http_mp4_read_dinf_atom(ngx_http_mp4_file_t *mp4,
255    uint64_t atom_data_size);
256static ngx_int_t ngx_http_mp4_read_vmhd_atom(ngx_http_mp4_file_t *mp4,
257    uint64_t atom_data_size);
258static ngx_int_t ngx_http_mp4_read_smhd_atom(ngx_http_mp4_file_t *mp4,
259    uint64_t atom_data_size);
260static ngx_int_t ngx_http_mp4_read_stbl_atom(ngx_http_mp4_file_t *mp4,
261    uint64_t atom_data_size);
262static void ngx_http_mp4_update_stbl_atom(ngx_http_mp4_file_t *mp4,
263    ngx_http_mp4_trak_t *trak);
264static ngx_int_t ngx_http_mp4_read_stsd_atom(ngx_http_mp4_file_t *mp4,
265    uint64_t atom_data_size);
266static ngx_int_t ngx_http_mp4_read_stts_atom(ngx_http_mp4_file_t *mp4,
267    uint64_t atom_data_size);
268static ngx_int_t ngx_http_mp4_update_stts_atom(ngx_http_mp4_file_t *mp4,
269    ngx_http_mp4_trak_t *trak);
270static ngx_int_t ngx_http_mp4_crop_stts_data(ngx_http_mp4_file_t *mp4,
271    ngx_http_mp4_trak_t *trak, ngx_uint_t start);
272static ngx_int_t ngx_http_mp4_read_stss_atom(ngx_http_mp4_file_t *mp4,
273    uint64_t atom_data_size);
274static ngx_int_t ngx_http_mp4_update_stss_atom(ngx_http_mp4_file_t *mp4,
275    ngx_http_mp4_trak_t *trak);
276static void ngx_http_mp4_crop_stss_data(ngx_http_mp4_file_t *mp4,
277    ngx_http_mp4_trak_t *trak, ngx_uint_t start);
278static ngx_int_t ngx_http_mp4_read_ctts_atom(ngx_http_mp4_file_t *mp4,
279    uint64_t atom_data_size);
280static void ngx_http_mp4_update_ctts_atom(ngx_http_mp4_file_t *mp4,
281    ngx_http_mp4_trak_t *trak);
282static void ngx_http_mp4_crop_ctts_data(ngx_http_mp4_file_t *mp4,
283    ngx_http_mp4_trak_t *trak, ngx_uint_t start);
284static ngx_int_t ngx_http_mp4_read_stsc_atom(ngx_http_mp4_file_t *mp4,
285    uint64_t atom_data_size);
286static ngx_int_t ngx_http_mp4_update_stsc_atom(ngx_http_mp4_file_t *mp4,
287    ngx_http_mp4_trak_t *trak);
288static ngx_int_t ngx_http_mp4_crop_stsc_data(ngx_http_mp4_file_t *mp4,
289    ngx_http_mp4_trak_t *trak, ngx_uint_t start);
290static ngx_int_t ngx_http_mp4_read_stsz_atom(ngx_http_mp4_file_t *mp4,
291    uint64_t atom_data_size);
292static ngx_int_t ngx_http_mp4_update_stsz_atom(ngx_http_mp4_file_t *mp4,
293    ngx_http_mp4_trak_t *trak);
294static ngx_int_t ngx_http_mp4_read_stco_atom(ngx_http_mp4_file_t *mp4,
295    uint64_t atom_data_size);
296static ngx_int_t ngx_http_mp4_update_stco_atom(ngx_http_mp4_file_t *mp4,
297    ngx_http_mp4_trak_t *trak);
298static void ngx_http_mp4_adjust_stco_atom(ngx_http_mp4_file_t *mp4,
299    ngx_http_mp4_trak_t *trak, int32_t adjustment);
300static ngx_int_t ngx_http_mp4_read_co64_atom(ngx_http_mp4_file_t *mp4,
301    uint64_t atom_data_size);
302static ngx_int_t ngx_http_mp4_update_co64_atom(ngx_http_mp4_file_t *mp4,
303    ngx_http_mp4_trak_t *trak);
304static void ngx_http_mp4_adjust_co64_atom(ngx_http_mp4_file_t *mp4,
305    ngx_http_mp4_trak_t *trak, off_t adjustment);
306
307static char *ngx_http_mp4(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
308static void *ngx_http_mp4_create_conf(ngx_conf_t *cf);
309static char *ngx_http_mp4_merge_conf(ngx_conf_t *cf, void *parent, void *child);
310
311
312static ngx_command_t  ngx_http_mp4_commands[] = {
313
314    { ngx_string("mp4"),
315      NGX_HTTP_LOC_CONF|NGX_CONF_NOARGS,
316      ngx_http_mp4,
317      0,
318      0,
319      NULL },
320
321    { ngx_string("mp4_buffer_size"),
322      NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
323      ngx_conf_set_size_slot,
324      NGX_HTTP_LOC_CONF_OFFSET,
325      offsetof(ngx_http_mp4_conf_t, buffer_size),
326      NULL },
327
328    { ngx_string("mp4_max_buffer_size"),
329      NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
330      ngx_conf_set_size_slot,
331      NGX_HTTP_LOC_CONF_OFFSET,
332      offsetof(ngx_http_mp4_conf_t, max_buffer_size),
333      NULL },
334
335      ngx_null_command
336};
337
338
339static ngx_http_module_t  ngx_http_mp4_module_ctx = {
340    NULL,                          /* preconfiguration */
341    NULL,                          /* postconfiguration */
342
343    NULL,                          /* create main configuration */
344    NULL,                          /* init main configuration */
345
346    NULL,                          /* create server configuration */
347    NULL,                          /* merge server configuration */
348
349    ngx_http_mp4_create_conf,      /* create location configuration */
350    ngx_http_mp4_merge_conf        /* merge location configuration */
351};
352
353
354ngx_module_t  ngx_http_mp4_module = {
355    NGX_MODULE_V1,
356    &ngx_http_mp4_module_ctx,      /* module context */
357    ngx_http_mp4_commands,         /* module directives */
358    NGX_HTTP_MODULE,               /* module type */
359    NULL,                          /* init master */
360    NULL,                          /* init module */
361    NULL,                          /* init process */
362    NULL,                          /* init thread */
363    NULL,                          /* exit thread */
364    NULL,                          /* exit process */
365    NULL,                          /* exit master */
366    NGX_MODULE_V1_PADDING
367};
368
369
370static ngx_http_mp4_atom_handler_t  ngx_http_mp4_atoms[] = {
371    { "ftyp", ngx_http_mp4_read_ftyp_atom },
372    { "moov", ngx_http_mp4_read_moov_atom },
373    { "mdat", ngx_http_mp4_read_mdat_atom },
374    { NULL, NULL }
375};
376
377static ngx_http_mp4_atom_handler_t  ngx_http_mp4_moov_atoms[] = {
378    { "mvhd", ngx_http_mp4_read_mvhd_atom },
379    { "trak", ngx_http_mp4_read_trak_atom },
380    { "cmov", ngx_http_mp4_read_cmov_atom },
381    { NULL, NULL }
382};
383
384static ngx_http_mp4_atom_handler_t  ngx_http_mp4_trak_atoms[] = {
385    { "tkhd", ngx_http_mp4_read_tkhd_atom },
386    { "mdia", ngx_http_mp4_read_mdia_atom },
387    { NULL, NULL }
388};
389
390static ngx_http_mp4_atom_handler_t  ngx_http_mp4_mdia_atoms[] = {
391    { "mdhd", ngx_http_mp4_read_mdhd_atom },
392    { "hdlr", ngx_http_mp4_read_hdlr_atom },
393    { "minf", ngx_http_mp4_read_minf_atom },
394    { NULL, NULL }
395};
396
397static ngx_http_mp4_atom_handler_t  ngx_http_mp4_minf_atoms[] = {
398    { "vmhd", ngx_http_mp4_read_vmhd_atom },
399    { "smhd", ngx_http_mp4_read_smhd_atom },
400    { "dinf", ngx_http_mp4_read_dinf_atom },
401    { "stbl", ngx_http_mp4_read_stbl_atom },
402    { NULL, NULL }
403};
404
405static ngx_http_mp4_atom_handler_t  ngx_http_mp4_stbl_atoms[] = {
406    { "stsd", ngx_http_mp4_read_stsd_atom },
407    { "stts", ngx_http_mp4_read_stts_atom },
408    { "stss", ngx_http_mp4_read_stss_atom },
409    { "ctts", ngx_http_mp4_read_ctts_atom },
410    { "stsc", ngx_http_mp4_read_stsc_atom },
411    { "stsz", ngx_http_mp4_read_stsz_atom },
412    { "stco", ngx_http_mp4_read_stco_atom },
413    { "co64", ngx_http_mp4_read_co64_atom },
414    { NULL, NULL }
415};
416
417
418static ngx_int_t
419ngx_http_mp4_handler(ngx_http_request_t *r)
420{
421    u_char                    *last;
422    size_t                     root;
423    ngx_int_t                  rc, start, end;
424    ngx_uint_t                 level, length;
425    ngx_str_t                  path, value;
426    ngx_log_t                 *log;
427    ngx_buf_t                 *b;
428    ngx_chain_t                out;
429    ngx_http_mp4_file_t       *mp4;
430    ngx_open_file_info_t       of;
431    ngx_http_core_loc_conf_t  *clcf;
432
433    if (!(r->method & (NGX_HTTP_GET|NGX_HTTP_HEAD))) {
434        return NGX_HTTP_NOT_ALLOWED;
435    }
436
437    if (r->uri.data[r->uri.len - 1] == '/') {
438        return NGX_DECLINED;
439    }
440
441    rc = ngx_http_discard_request_body(r);
442
443    if (rc != NGX_OK) {
444        return rc;
445    }
446
447    last = ngx_http_map_uri_to_path(r, &path, &root, 0);
448    if (last == NULL) {
449        return NGX_HTTP_INTERNAL_SERVER_ERROR;
450    }
451
452    log = r->connection->log;
453
454    path.len = last - path.data;
455
456    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, log, 0,
457                   "http mp4 filename: \"%V\"", &path);
458
459    clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
460
461    ngx_memzero(&of, sizeof(ngx_open_file_info_t));
462
463    of.read_ahead = clcf->read_ahead;
464    of.directio = NGX_MAX_OFF_T_VALUE;
465    of.valid = clcf->open_file_cache_valid;
466    of.min_uses = clcf->open_file_cache_min_uses;
467    of.errors = clcf->open_file_cache_errors;
468    of.events = clcf->open_file_cache_events;
469
470    if (ngx_http_set_disable_symlinks(r, clcf, &path, &of) != NGX_OK) {
471        return NGX_HTTP_INTERNAL_SERVER_ERROR;
472    }
473
474    if (ngx_open_cached_file(clcf->open_file_cache, &path, &of, r->pool)
475        != NGX_OK)
476    {
477        switch (of.err) {
478
479        case 0:
480            return NGX_HTTP_INTERNAL_SERVER_ERROR;
481
482        case NGX_ENOENT:
483        case NGX_ENOTDIR:
484        case NGX_ENAMETOOLONG:
485
486            level = NGX_LOG_ERR;
487            rc = NGX_HTTP_NOT_FOUND;
488            break;
489
490        case NGX_EACCES:
491#if (NGX_HAVE_OPENAT)
492        case NGX_EMLINK:
493        case NGX_ELOOP:
494#endif
495
496            level = NGX_LOG_ERR;
497            rc = NGX_HTTP_FORBIDDEN;
498            break;
499
500        default:
501
502            level = NGX_LOG_CRIT;
503            rc = NGX_HTTP_INTERNAL_SERVER_ERROR;
504            break;
505        }
506
507        if (rc != NGX_HTTP_NOT_FOUND || clcf->log_not_found) {
508            ngx_log_error(level, log, of.err,
509                          "%s \"%s\" failed", of.failed, path.data);
510        }
511
512        return rc;
513    }
514
515    if (!of.is_file) {
516
517        if (ngx_close_file(of.fd) == NGX_FILE_ERROR) {
518            ngx_log_error(NGX_LOG_ALERT, log, ngx_errno,
519                          ngx_close_file_n " \"%s\" failed", path.data);
520        }
521
522        return NGX_DECLINED;
523    }
524
525    r->root_tested = !r->error_page;
526    r->allow_ranges = 1;
527
528    start = -1;
529    length = 0;
530    r->headers_out.content_length_n = of.size;
531    mp4 = NULL;
532    b = NULL;
533
534    if (r->args.len) {
535
536        if (ngx_http_arg(r, (u_char *) "start", 5, &value) == NGX_OK) {
537
538            /*
539             * A Flash player may send start value with a lot of digits
540             * after dot so strtod() is used instead of atofp().  NaNs and
541             * infinities become negative numbers after (int) conversion.
542             */
543
544            ngx_set_errno(0);
545            start = (int) (strtod((char *) value.data, NULL) * 1000);
546
547            if (ngx_errno != 0) {
548                start = -1;
549            }
550        }
551
552        if (ngx_http_arg(r, (u_char *) "end", 3, &value) == NGX_OK) {
553
554            ngx_set_errno(0);
555            end = (int) (strtod((char *) value.data, NULL) * 1000);
556
557            if (ngx_errno != 0) {
558                end = -1;
559            }
560
561            if (end > 0) {
562                if (start < 0) {
563                    start = 0;
564                }
565
566                if (end > start) {
567                    length = end - start;
568                }
569            }
570        }
571    }
572
573    if (start >= 0) {
574        r->single_range = 1;
575
576        mp4 = ngx_pcalloc(r->pool, sizeof(ngx_http_mp4_file_t));
577        if (mp4 == NULL) {
578            return NGX_HTTP_INTERNAL_SERVER_ERROR;
579        }
580
581        mp4->file.fd = of.fd;
582        mp4->file.name = path;
583        mp4->file.log = r->connection->log;
584        mp4->end = of.size;
585        mp4->start = (ngx_uint_t) start;
586        mp4->length = length;
587        mp4->request = r;
588
589        switch (ngx_http_mp4_process(mp4)) {
590
591        case NGX_DECLINED:
592            if (mp4->buffer) {
593                ngx_pfree(r->pool, mp4->buffer);
594            }
595
596            ngx_pfree(r->pool, mp4);
597            mp4 = NULL;
598
599            break;
600
601        case NGX_OK:
602            r->headers_out.content_length_n = mp4->content_length;
603            break;
604
605        default: /* NGX_ERROR */
606            if (mp4->buffer) {
607                ngx_pfree(r->pool, mp4->buffer);
608            }
609
610            ngx_pfree(r->pool, mp4);
611
612            return NGX_HTTP_INTERNAL_SERVER_ERROR;
613        }
614    }
615
616    log->action = "sending mp4 to client";
617
618    if (clcf->directio <= of.size) {
619
620        /*
621         * DIRECTIO is set on transfer only
622         * to allow kernel to cache "moov" atom
623         */
624
625        if (ngx_directio_on(of.fd) == NGX_FILE_ERROR) {
626            ngx_log_error(NGX_LOG_ALERT, log, ngx_errno,
627                          ngx_directio_on_n " \"%s\" failed", path.data);
628        }
629
630        of.is_directio = 1;
631
632        if (mp4) {
633            mp4->file.directio = 1;
634        }
635    }
636
637    r->headers_out.status = NGX_HTTP_OK;
638    r->headers_out.last_modified_time = of.mtime;
639
640    if (ngx_http_set_etag(r) != NGX_OK) {
641        return NGX_HTTP_INTERNAL_SERVER_ERROR;
642    }
643
644    if (ngx_http_set_content_type(r) != NGX_OK) {
645        return NGX_HTTP_INTERNAL_SERVER_ERROR;
646    }
647
648    if (mp4 == NULL) {
649        b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t));
650        if (b == NULL) {
651            return NGX_HTTP_INTERNAL_SERVER_ERROR;
652        }
653
654        b->file = ngx_pcalloc(r->pool, sizeof(ngx_file_t));
655        if (b->file == NULL) {
656            return NGX_HTTP_INTERNAL_SERVER_ERROR;
657        }
658    }
659
660    rc = ngx_http_send_header(r);
661
662    if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
663        return rc;
664    }
665
666    if (mp4) {
667        return ngx_http_output_filter(r, mp4->out);
668    }
669
670    b->file_pos = 0;
671    b->file_last = of.size;
672
673    b->in_file = b->file_last ? 1 : 0;
674    b->last_buf = (r == r->main) ? 1 : 0;
675    b->last_in_chain = 1;
676
677    b->file->fd = of.fd;
678    b->file->name = path;
679    b->file->log = log;
680    b->file->directio = of.is_directio;
681
682    out.buf = b;
683    out.next = NULL;
684
685    return ngx_http_output_filter(r, &out);
686}
687
688
689static ngx_int_t
690ngx_http_mp4_process(ngx_http_mp4_file_t *mp4)
691{
692    off_t                  start_offset, end_offset, adjustment;
693    ngx_int_t              rc;
694    ngx_uint_t             i, j;
695    ngx_chain_t          **prev;
696    ngx_http_mp4_trak_t   *trak;
697    ngx_http_mp4_conf_t   *conf;
698
699    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
700                   "mp4 start:%ui, length:%ui", mp4->start, mp4->length);
701
702    conf = ngx_http_get_module_loc_conf(mp4->request, ngx_http_mp4_module);
703
704    mp4->buffer_size = conf->buffer_size;
705
706    rc = ngx_http_mp4_read_atom(mp4, ngx_http_mp4_atoms, mp4->end);
707    if (rc != NGX_OK) {
708        return rc;
709    }
710
711    if (mp4->trak.nelts == 0) {
712        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
713                      "no mp4 trak atoms were found in \"%s\"",
714                      mp4->file.name.data);
715        return NGX_ERROR;
716    }
717
718    if (mp4->mdat_atom.buf == NULL) {
719        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
720                      "no mp4 mdat atom was found in \"%s\"",
721                      mp4->file.name.data);
722        return NGX_ERROR;
723    }
724
725    prev = &mp4->out;
726
727    if (mp4->ftyp_atom.buf) {
728        *prev = &mp4->ftyp_atom;
729        prev = &mp4->ftyp_atom.next;
730    }
731
732    *prev = &mp4->moov_atom;
733    prev = &mp4->moov_atom.next;
734
735    if (mp4->mvhd_atom.buf) {
736        mp4->moov_size += mp4->mvhd_atom_buf.last - mp4->mvhd_atom_buf.pos;
737        *prev = &mp4->mvhd_atom;
738        prev = &mp4->mvhd_atom.next;
739    }
740
741    start_offset = mp4->end;
742    end_offset = 0;
743    trak = mp4->trak.elts;
744
745    for (i = 0; i < mp4->trak.nelts; i++) {
746
747        if (ngx_http_mp4_update_stts_atom(mp4, &trak[i]) != NGX_OK) {
748            return NGX_ERROR;
749        }
750
751        if (ngx_http_mp4_update_stss_atom(mp4, &trak[i]) != NGX_OK) {
752            return NGX_ERROR;
753        }
754
755        ngx_http_mp4_update_ctts_atom(mp4, &trak[i]);
756
757        if (ngx_http_mp4_update_stsc_atom(mp4, &trak[i]) != NGX_OK) {
758            return NGX_ERROR;
759        }
760
761        if (ngx_http_mp4_update_stsz_atom(mp4, &trak[i]) != NGX_OK) {
762            return NGX_ERROR;
763        }
764
765        if (trak[i].out[NGX_HTTP_MP4_CO64_DATA].buf) {
766            if (ngx_http_mp4_update_co64_atom(mp4, &trak[i]) != NGX_OK) {
767                return NGX_ERROR;
768            }
769
770        } else {
771            if (ngx_http_mp4_update_stco_atom(mp4, &trak[i]) != NGX_OK) {
772                return NGX_ERROR;
773            }
774        }
775
776        ngx_http_mp4_update_stbl_atom(mp4, &trak[i]);
777        ngx_http_mp4_update_minf_atom(mp4, &trak[i]);
778        trak[i].size += trak[i].mdhd_size;
779        trak[i].size += trak[i].hdlr_size;
780        ngx_http_mp4_update_mdia_atom(mp4, &trak[i]);
781        trak[i].size += trak[i].tkhd_size;
782        ngx_http_mp4_update_trak_atom(mp4, &trak[i]);
783
784        mp4->moov_size += trak[i].size;
785
786        if (start_offset > trak[i].start_offset) {
787            start_offset = trak[i].start_offset;
788        }
789
790        if (end_offset < trak[i].end_offset) {
791            end_offset = trak[i].end_offset;
792        }
793
794        *prev = &trak[i].out[NGX_HTTP_MP4_TRAK_ATOM];
795        prev = &trak[i].out[NGX_HTTP_MP4_TRAK_ATOM].next;
796
797        for (j = 0; j < NGX_HTTP_MP4_LAST_ATOM + 1; j++) {
798            if (trak[i].out[j].buf) {
799                *prev = &trak[i].out[j];
800                prev = &trak[i].out[j].next;
801            }
802        }
803    }
804
805    if (end_offset < start_offset) {
806        end_offset = start_offset;
807    }
808
809    mp4->moov_size += 8;
810
811    ngx_mp4_set_32value(mp4->moov_atom_header, mp4->moov_size);
812    ngx_mp4_set_atom_name(mp4->moov_atom_header, 'm', 'o', 'o', 'v');
813    mp4->content_length += mp4->moov_size;
814
815    *prev = &mp4->mdat_atom;
816
817    if (start_offset > mp4->mdat_data.buf->file_last) {
818        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
819                      "start time is out mp4 mdat atom in \"%s\"",
820                      mp4->file.name.data);
821        return NGX_ERROR;
822    }
823
824    adjustment = mp4->ftyp_size + mp4->moov_size
825                 + ngx_http_mp4_update_mdat_atom(mp4, start_offset, end_offset)
826                 - start_offset;
827
828    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
829                   "mp4 adjustment:%O", adjustment);
830
831    for (i = 0; i < mp4->trak.nelts; i++) {
832        if (trak[i].out[NGX_HTTP_MP4_CO64_DATA].buf) {
833            ngx_http_mp4_adjust_co64_atom(mp4, &trak[i], adjustment);
834        } else {
835            ngx_http_mp4_adjust_stco_atom(mp4, &trak[i], (int32_t) adjustment);
836        }
837    }
838
839    return NGX_OK;
840}
841
842
843typedef struct {
844    u_char    size[4];
845    u_char    name[4];
846} ngx_mp4_atom_header_t;
847
848typedef struct {
849    u_char    size[4];
850    u_char    name[4];
851    u_char    size64[8];
852} ngx_mp4_atom_header64_t;
853
854
855static ngx_int_t
856ngx_http_mp4_read_atom(ngx_http_mp4_file_t *mp4,
857    ngx_http_mp4_atom_handler_t *atom, uint64_t atom_data_size)
858{
859    off_t        end;
860    size_t       atom_header_size;
861    u_char      *atom_header, *atom_name;
862    uint64_t     atom_size;
863    ngx_int_t    rc;
864    ngx_uint_t   n;
865
866    end = mp4->offset + atom_data_size;
867
868    while (mp4->offset < end) {
869
870        if (ngx_http_mp4_read(mp4, sizeof(uint32_t)) != NGX_OK) {
871            return NGX_ERROR;
872        }
873
874        atom_header = mp4->buffer_pos;
875        atom_size = ngx_mp4_get_32value(atom_header);
876        atom_header_size = sizeof(ngx_mp4_atom_header_t);
877
878        if (atom_size == 0) {
879            ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
880                           "mp4 atom end");
881            return NGX_OK;
882        }
883
884        if (atom_size < sizeof(ngx_mp4_atom_header_t)) {
885
886            if (atom_size == 1) {
887
888                if (ngx_http_mp4_read(mp4, sizeof(ngx_mp4_atom_header64_t))
889                    != NGX_OK)
890                {
891                    return NGX_ERROR;
892                }
893
894                /* 64-bit atom size */
895                atom_header = mp4->buffer_pos;
896                atom_size = ngx_mp4_get_64value(atom_header + 8);
897                atom_header_size = sizeof(ngx_mp4_atom_header64_t);
898
899            } else {
900                ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
901                              "\"%s\" mp4 atom is too small:%uL",
902                              mp4->file.name.data, atom_size);
903                return NGX_ERROR;
904            }
905        }
906
907        if (ngx_http_mp4_read(mp4, sizeof(ngx_mp4_atom_header_t)) != NGX_OK) {
908            return NGX_ERROR;
909        }
910
911        atom_header = mp4->buffer_pos;
912        atom_name = atom_header + sizeof(uint32_t);
913
914        ngx_log_debug4(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
915                       "mp4 atom: %*s @%O:%uL",
916                       4, atom_name, mp4->offset, atom_size);
917
918        if (atom_size > (uint64_t) (NGX_MAX_OFF_T_VALUE - mp4->offset)
919            || mp4->offset + (off_t) atom_size > end)
920        {
921            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
922                          "\"%s\" mp4 atom too large:%uL",
923                          mp4->file.name.data, atom_size);
924            return NGX_ERROR;
925        }
926
927        for (n = 0; atom[n].name; n++) {
928
929            if (ngx_strncmp(atom_name, atom[n].name, 4) == 0) {
930
931                ngx_mp4_atom_next(mp4, atom_header_size);
932
933                rc = atom[n].handler(mp4, atom_size - atom_header_size);
934                if (rc != NGX_OK) {
935                    return rc;
936                }
937
938                goto next;
939            }
940        }
941
942        ngx_mp4_atom_next(mp4, atom_size);
943
944    next:
945        continue;
946    }
947
948    return NGX_OK;
949}
950
951
952static ngx_int_t
953ngx_http_mp4_read(ngx_http_mp4_file_t *mp4, size_t size)
954{
955    ssize_t  n;
956
957    if (mp4->buffer_pos + size <= mp4->buffer_end) {
958        return NGX_OK;
959    }
960
961    if (mp4->offset + (off_t) mp4->buffer_size > mp4->end) {
962        mp4->buffer_size = (size_t) (mp4->end - mp4->offset);
963    }
964
965    if (mp4->buffer_size < size) {
966        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
967                      "\"%s\" mp4 file truncated", mp4->file.name.data);
968        return NGX_ERROR;
969    }
970
971    if (mp4->buffer == NULL) {
972        mp4->buffer = ngx_palloc(mp4->request->pool, mp4->buffer_size);
973        if (mp4->buffer == NULL) {
974            return NGX_ERROR;
975        }
976
977        mp4->buffer_start = mp4->buffer;
978    }
979
980    n = ngx_read_file(&mp4->file, mp4->buffer_start, mp4->buffer_size,
981                      mp4->offset);
982
983    if (n == NGX_ERROR) {
984        return NGX_ERROR;
985    }
986
987    if ((size_t) n != mp4->buffer_size) {
988        ngx_log_error(NGX_LOG_CRIT, mp4->file.log, 0,
989                      ngx_read_file_n " read only %z of %z from \"%s\"",
990                      n, mp4->buffer_size, mp4->file.name.data);
991        return NGX_ERROR;
992    }
993
994    mp4->buffer_pos = mp4->buffer_start;
995    mp4->buffer_end = mp4->buffer_start + mp4->buffer_size;
996
997    return NGX_OK;
998}
999
1000
1001static ngx_int_t
1002ngx_http_mp4_read_ftyp_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1003{
1004    u_char     *ftyp_atom;
1005    size_t      atom_size;
1006    ngx_buf_t  *atom;
1007
1008    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 ftyp atom");
1009
1010    if (atom_data_size > 1024
1011        || ngx_mp4_atom_data(mp4) + (size_t) atom_data_size > mp4->buffer_end)
1012    {
1013        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1014                      "\"%s\" mp4 ftyp atom is too large:%uL",
1015                      mp4->file.name.data, atom_data_size);
1016        return NGX_ERROR;
1017    }
1018
1019    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1020
1021    ftyp_atom = ngx_palloc(mp4->request->pool, atom_size);
1022    if (ftyp_atom == NULL) {
1023        return NGX_ERROR;
1024    }
1025
1026    ngx_mp4_set_32value(ftyp_atom, atom_size);
1027    ngx_mp4_set_atom_name(ftyp_atom, 'f', 't', 'y', 'p');
1028
1029    /*
1030     * only moov atom content is guaranteed to be in mp4->buffer
1031     * during sending response, so ftyp atom content should be copied
1032     */
1033    ngx_memcpy(ftyp_atom + sizeof(ngx_mp4_atom_header_t),
1034               ngx_mp4_atom_data(mp4), (size_t) atom_data_size);
1035
1036    atom = &mp4->ftyp_atom_buf;
1037    atom->temporary = 1;
1038    atom->pos = ftyp_atom;
1039    atom->last = ftyp_atom + atom_size;
1040
1041    mp4->ftyp_atom.buf = atom;
1042    mp4->ftyp_size = atom_size;
1043    mp4->content_length = atom_size;
1044
1045    ngx_mp4_atom_next(mp4, atom_data_size);
1046
1047    return NGX_OK;
1048}
1049
1050
1051/*
1052 * Small excess buffer to process atoms after moov atom, mp4->buffer_start
1053 * will be set to this buffer part after moov atom processing.
1054 */
1055#define NGX_HTTP_MP4_MOOV_BUFFER_EXCESS  (4 * 1024)
1056
1057static ngx_int_t
1058ngx_http_mp4_read_moov_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1059{
1060    ngx_int_t             rc;
1061    ngx_uint_t            no_mdat;
1062    ngx_buf_t            *atom;
1063    ngx_http_mp4_conf_t  *conf;
1064
1065    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 moov atom");
1066
1067    no_mdat = (mp4->mdat_atom.buf == NULL);
1068
1069    if (no_mdat && mp4->start == 0 && mp4->length == 0) {
1070        /*
1071         * send original file if moov atom resides before
1072         * mdat atom and client requests integral file
1073         */
1074        return NGX_DECLINED;
1075    }
1076
1077    conf = ngx_http_get_module_loc_conf(mp4->request, ngx_http_mp4_module);
1078
1079    if (atom_data_size > mp4->buffer_size) {
1080
1081        if (atom_data_size > conf->max_buffer_size) {
1082            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1083                          "\"%s\" mp4 moov atom is too large:%uL, "
1084                          "you may want to increase mp4_max_buffer_size",
1085                          mp4->file.name.data, atom_data_size);
1086            return NGX_ERROR;
1087        }
1088
1089        ngx_pfree(mp4->request->pool, mp4->buffer);
1090        mp4->buffer = NULL;
1091        mp4->buffer_pos = NULL;
1092        mp4->buffer_end = NULL;
1093
1094        mp4->buffer_size = (size_t) atom_data_size
1095                         + NGX_HTTP_MP4_MOOV_BUFFER_EXCESS * no_mdat;
1096    }
1097
1098    if (ngx_http_mp4_read(mp4, (size_t) atom_data_size) != NGX_OK) {
1099        return NGX_ERROR;
1100    }
1101
1102    mp4->trak.elts = &mp4->traks;
1103    mp4->trak.size = sizeof(ngx_http_mp4_trak_t);
1104    mp4->trak.nalloc = 2;
1105    mp4->trak.pool = mp4->request->pool;
1106
1107    atom = &mp4->moov_atom_buf;
1108    atom->temporary = 1;
1109    atom->pos = mp4->moov_atom_header;
1110    atom->last = mp4->moov_atom_header + 8;
1111
1112    mp4->moov_atom.buf = &mp4->moov_atom_buf;
1113
1114    rc = ngx_http_mp4_read_atom(mp4, ngx_http_mp4_moov_atoms, atom_data_size);
1115
1116    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 moov atom done");
1117
1118    if (no_mdat) {
1119        mp4->buffer_start = mp4->buffer_pos;
1120        mp4->buffer_size = NGX_HTTP_MP4_MOOV_BUFFER_EXCESS;
1121
1122        if (mp4->buffer_start + mp4->buffer_size > mp4->buffer_end) {
1123            mp4->buffer = NULL;
1124            mp4->buffer_pos = NULL;
1125            mp4->buffer_end = NULL;
1126        }
1127
1128    } else {
1129        /* skip atoms after moov atom */
1130        mp4->offset = mp4->end;
1131    }
1132
1133    return rc;
1134}
1135
1136
1137static ngx_int_t
1138ngx_http_mp4_read_mdat_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1139{
1140    ngx_buf_t  *data;
1141
1142    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 mdat atom");
1143
1144    data = &mp4->mdat_data_buf;
1145    data->file = &mp4->file;
1146    data->in_file = 1;
1147    data->last_buf = 1;
1148    data->last_in_chain = 1;
1149    data->file_last = mp4->offset + atom_data_size;
1150
1151    mp4->mdat_atom.buf = &mp4->mdat_atom_buf;
1152    mp4->mdat_atom.next = &mp4->mdat_data;
1153    mp4->mdat_data.buf = data;
1154
1155    if (mp4->trak.nelts) {
1156        /* skip atoms after mdat atom */
1157        mp4->offset = mp4->end;
1158
1159    } else {
1160        ngx_mp4_atom_next(mp4, atom_data_size);
1161    }
1162
1163    return NGX_OK;
1164}
1165
1166
1167static size_t
1168ngx_http_mp4_update_mdat_atom(ngx_http_mp4_file_t *mp4, off_t start_offset,
1169    off_t end_offset)
1170{
1171    off_t       atom_data_size;
1172    u_char     *atom_header;
1173    uint32_t    atom_header_size;
1174    uint64_t    atom_size;
1175    ngx_buf_t  *atom;
1176
1177    atom_data_size = end_offset - start_offset;
1178    mp4->mdat_data.buf->file_pos = start_offset;
1179    mp4->mdat_data.buf->file_last = end_offset;
1180
1181    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1182                   "mdat new offset @%O:%O", start_offset, atom_data_size);
1183
1184    atom_header = mp4->mdat_atom_header;
1185
1186    if ((uint64_t) atom_data_size > (uint64_t) 0xffffffff) {
1187        atom_size = 1;
1188        atom_header_size = sizeof(ngx_mp4_atom_header64_t);
1189        ngx_mp4_set_64value(atom_header + sizeof(ngx_mp4_atom_header_t),
1190                            sizeof(ngx_mp4_atom_header64_t) + atom_data_size);
1191    } else {
1192        atom_size = sizeof(ngx_mp4_atom_header_t) + atom_data_size;
1193        atom_header_size = sizeof(ngx_mp4_atom_header_t);
1194    }
1195
1196    mp4->content_length += atom_header_size + atom_data_size;
1197
1198    ngx_mp4_set_32value(atom_header, atom_size);
1199    ngx_mp4_set_atom_name(atom_header, 'm', 'd', 'a', 't');
1200
1201    atom = &mp4->mdat_atom_buf;
1202    atom->temporary = 1;
1203    atom->pos = atom_header;
1204    atom->last = atom_header + atom_header_size;
1205
1206    return atom_header_size;
1207}
1208
1209
1210typedef struct {
1211    u_char    size[4];
1212    u_char    name[4];
1213    u_char    version[1];
1214    u_char    flags[3];
1215    u_char    creation_time[4];
1216    u_char    modification_time[4];
1217    u_char    timescale[4];
1218    u_char    duration[4];
1219    u_char    rate[4];
1220    u_char    volume[2];
1221    u_char    reserved[10];
1222    u_char    matrix[36];
1223    u_char    preview_time[4];
1224    u_char    preview_duration[4];
1225    u_char    poster_time[4];
1226    u_char    selection_time[4];
1227    u_char    selection_duration[4];
1228    u_char    current_time[4];
1229    u_char    next_track_id[4];
1230} ngx_mp4_mvhd_atom_t;
1231
1232typedef struct {
1233    u_char    size[4];
1234    u_char    name[4];
1235    u_char    version[1];
1236    u_char    flags[3];
1237    u_char    creation_time[8];
1238    u_char    modification_time[8];
1239    u_char    timescale[4];
1240    u_char    duration[8];
1241    u_char    rate[4];
1242    u_char    volume[2];
1243    u_char    reserved[10];
1244    u_char    matrix[36];
1245    u_char    preview_time[4];
1246    u_char    preview_duration[4];
1247    u_char    poster_time[4];
1248    u_char    selection_time[4];
1249    u_char    selection_duration[4];
1250    u_char    current_time[4];
1251    u_char    next_track_id[4];
1252} ngx_mp4_mvhd64_atom_t;
1253
1254
1255static ngx_int_t
1256ngx_http_mp4_read_mvhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1257{
1258    u_char                 *atom_header;
1259    size_t                  atom_size;
1260    uint32_t                timescale;
1261    uint64_t                duration, start_time, length_time;
1262    ngx_buf_t              *atom;
1263    ngx_mp4_mvhd_atom_t    *mvhd_atom;
1264    ngx_mp4_mvhd64_atom_t  *mvhd64_atom;
1265
1266    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 mvhd atom");
1267
1268    atom_header = ngx_mp4_atom_header(mp4);
1269    mvhd_atom = (ngx_mp4_mvhd_atom_t *) atom_header;
1270    mvhd64_atom = (ngx_mp4_mvhd64_atom_t *) atom_header;
1271    ngx_mp4_set_atom_name(atom_header, 'm', 'v', 'h', 'd');
1272
1273    if (ngx_mp4_atom_data_size(ngx_mp4_mvhd_atom_t) > atom_data_size) {
1274        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1275                      "\"%s\" mp4 mvhd atom too small", mp4->file.name.data);
1276        return NGX_ERROR;
1277    }
1278
1279    if (mvhd_atom->version[0] == 0) {
1280        /* version 0: 32-bit duration */
1281        timescale = ngx_mp4_get_32value(mvhd_atom->timescale);
1282        duration = ngx_mp4_get_32value(mvhd_atom->duration);
1283
1284    } else {
1285        /* version 1: 64-bit duration */
1286
1287        if (ngx_mp4_atom_data_size(ngx_mp4_mvhd64_atom_t) > atom_data_size) {
1288            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1289                          "\"%s\" mp4 mvhd atom too small",
1290                          mp4->file.name.data);
1291            return NGX_ERROR;
1292        }
1293
1294        timescale = ngx_mp4_get_32value(mvhd64_atom->timescale);
1295        duration = ngx_mp4_get_64value(mvhd64_atom->duration);
1296    }
1297
1298    mp4->timescale = timescale;
1299
1300    ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1301                   "mvhd timescale:%uD, duration:%uL, time:%.3fs",
1302                   timescale, duration, (double) duration / timescale);
1303
1304    start_time = (uint64_t) mp4->start * timescale / 1000;
1305
1306    if (duration < start_time) {
1307        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1308                      "\"%s\" mp4 start time exceeds file duration",
1309                      mp4->file.name.data);
1310        return NGX_ERROR;
1311    }
1312
1313    duration -= start_time;
1314
1315    if (mp4->length) {
1316        length_time = (uint64_t) mp4->length * timescale / 1000;
1317
1318        if (duration > length_time) {
1319            duration = length_time;
1320        }
1321    }
1322
1323    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1324                   "mvhd new duration:%uL, time:%.3fs",
1325                   duration, (double) duration / timescale);
1326
1327    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1328    ngx_mp4_set_32value(mvhd_atom->size, atom_size);
1329
1330    if (mvhd_atom->version[0] == 0) {
1331        ngx_mp4_set_32value(mvhd_atom->duration, duration);
1332
1333    } else {
1334        ngx_mp4_set_64value(mvhd64_atom->duration, duration);
1335    }
1336
1337    atom = &mp4->mvhd_atom_buf;
1338    atom->temporary = 1;
1339    atom->pos = atom_header;
1340    atom->last = atom_header + atom_size;
1341
1342    mp4->mvhd_atom.buf = atom;
1343
1344    ngx_mp4_atom_next(mp4, atom_data_size);
1345
1346    return NGX_OK;
1347}
1348
1349
1350static ngx_int_t
1351ngx_http_mp4_read_trak_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1352{
1353    u_char               *atom_header, *atom_end;
1354    off_t                 atom_file_end;
1355    ngx_int_t             rc;
1356    ngx_buf_t            *atom;
1357    ngx_http_mp4_trak_t  *trak;
1358
1359    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 trak atom");
1360
1361    trak = ngx_array_push(&mp4->trak);
1362    if (trak == NULL) {
1363        return NGX_ERROR;
1364    }
1365
1366    ngx_memzero(trak, sizeof(ngx_http_mp4_trak_t));
1367
1368    atom_header = ngx_mp4_atom_header(mp4);
1369    ngx_mp4_set_atom_name(atom_header, 't', 'r', 'a', 'k');
1370
1371    atom = &trak->trak_atom_buf;
1372    atom->temporary = 1;
1373    atom->pos = atom_header;
1374    atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);
1375
1376    trak->out[NGX_HTTP_MP4_TRAK_ATOM].buf = atom;
1377
1378    atom_end = mp4->buffer_pos + (size_t) atom_data_size;
1379    atom_file_end = mp4->offset + atom_data_size;
1380
1381    rc = ngx_http_mp4_read_atom(mp4, ngx_http_mp4_trak_atoms, atom_data_size);
1382
1383    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1384                   "mp4 trak atom: %i", rc);
1385
1386    if (rc == NGX_DECLINED) {
1387        /* skip this trak */
1388        ngx_memzero(trak, sizeof(ngx_http_mp4_trak_t));
1389        mp4->trak.nelts--;
1390        mp4->buffer_pos = atom_end;
1391        mp4->offset = atom_file_end;
1392        return NGX_OK;
1393    }
1394
1395    return rc;
1396}
1397
1398
1399static void
1400ngx_http_mp4_update_trak_atom(ngx_http_mp4_file_t *mp4,
1401    ngx_http_mp4_trak_t *trak)
1402{
1403    ngx_buf_t  *atom;
1404
1405    trak->size += sizeof(ngx_mp4_atom_header_t);
1406    atom = &trak->trak_atom_buf;
1407    ngx_mp4_set_32value(atom->pos, trak->size);
1408}
1409
1410
1411static ngx_int_t
1412ngx_http_mp4_read_cmov_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1413{
1414    ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1415                  "\"%s\" mp4 compressed moov atom (cmov) is not supported",
1416                  mp4->file.name.data);
1417
1418    return NGX_ERROR;
1419}
1420
1421
1422typedef struct {
1423    u_char    size[4];
1424    u_char    name[4];
1425    u_char    version[1];
1426    u_char    flags[3];
1427    u_char    creation_time[4];
1428    u_char    modification_time[4];
1429    u_char    track_id[4];
1430    u_char    reserved1[4];
1431    u_char    duration[4];
1432    u_char    reserved2[8];
1433    u_char    layer[2];
1434    u_char    group[2];
1435    u_char    volume[2];
1436    u_char    reverved3[2];
1437    u_char    matrix[36];
1438    u_char    width[4];
1439    u_char    heigth[4];
1440} ngx_mp4_tkhd_atom_t;
1441
1442typedef struct {
1443    u_char    size[4];
1444    u_char    name[4];
1445    u_char    version[1];
1446    u_char    flags[3];
1447    u_char    creation_time[8];
1448    u_char    modification_time[8];
1449    u_char    track_id[4];
1450    u_char    reserved1[4];
1451    u_char    duration[8];
1452    u_char    reserved2[8];
1453    u_char    layer[2];
1454    u_char    group[2];
1455    u_char    volume[2];
1456    u_char    reverved3[2];
1457    u_char    matrix[36];
1458    u_char    width[4];
1459    u_char    heigth[4];
1460} ngx_mp4_tkhd64_atom_t;
1461
1462
1463static ngx_int_t
1464ngx_http_mp4_read_tkhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1465{
1466    u_char                 *atom_header;
1467    size_t                  atom_size;
1468    uint64_t                duration, start_time, length_time;
1469    ngx_buf_t              *atom;
1470    ngx_http_mp4_trak_t    *trak;
1471    ngx_mp4_tkhd_atom_t    *tkhd_atom;
1472    ngx_mp4_tkhd64_atom_t  *tkhd64_atom;
1473
1474    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 tkhd atom");
1475
1476    atom_header = ngx_mp4_atom_header(mp4);
1477    tkhd_atom = (ngx_mp4_tkhd_atom_t *) atom_header;
1478    tkhd64_atom = (ngx_mp4_tkhd64_atom_t *) atom_header;
1479    ngx_mp4_set_atom_name(tkhd_atom, 't', 'k', 'h', 'd');
1480
1481    if (ngx_mp4_atom_data_size(ngx_mp4_tkhd_atom_t) > atom_data_size) {
1482        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1483                      "\"%s\" mp4 tkhd atom too small", mp4->file.name.data);
1484        return NGX_ERROR;
1485    }
1486
1487    if (tkhd_atom->version[0] == 0) {
1488        /* version 0: 32-bit duration */
1489        duration = ngx_mp4_get_32value(tkhd_atom->duration);
1490
1491    } else {
1492        /* version 1: 64-bit duration */
1493
1494        if (ngx_mp4_atom_data_size(ngx_mp4_tkhd64_atom_t) > atom_data_size) {
1495            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1496                          "\"%s\" mp4 tkhd atom too small",
1497                          mp4->file.name.data);
1498            return NGX_ERROR;
1499        }
1500
1501        duration = ngx_mp4_get_64value(tkhd64_atom->duration);
1502    }
1503
1504    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1505                   "tkhd duration:%uL, time:%.3fs",
1506                   duration, (double) duration / mp4->timescale);
1507
1508    start_time = (uint64_t) mp4->start * mp4->timescale / 1000;
1509
1510    if (duration < start_time) {
1511        ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1512                       "tkhd duration is less than start time");
1513        return NGX_DECLINED;
1514    }
1515
1516    duration -= start_time;
1517
1518    if (mp4->length) {
1519        length_time = (uint64_t) mp4->length * mp4->timescale / 1000;
1520
1521        if (duration > length_time) {
1522            duration = length_time;
1523        }
1524    }
1525
1526    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1527                   "tkhd new duration:%uL, time:%.3fs",
1528                   duration, (double) duration / mp4->timescale);
1529
1530    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1531
1532    trak = ngx_mp4_last_trak(mp4);
1533    trak->tkhd_size = atom_size;
1534
1535    ngx_mp4_set_32value(tkhd_atom->size, atom_size);
1536
1537    if (tkhd_atom->version[0] == 0) {
1538        ngx_mp4_set_32value(tkhd_atom->duration, duration);
1539
1540    } else {
1541        ngx_mp4_set_64value(tkhd64_atom->duration, duration);
1542    }
1543
1544    atom = &trak->tkhd_atom_buf;
1545    atom->temporary = 1;
1546    atom->pos = atom_header;
1547    atom->last = atom_header + atom_size;
1548
1549    trak->out[NGX_HTTP_MP4_TKHD_ATOM].buf = atom;
1550
1551    ngx_mp4_atom_next(mp4, atom_data_size);
1552
1553    return NGX_OK;
1554}
1555
1556
1557static ngx_int_t
1558ngx_http_mp4_read_mdia_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1559{
1560    u_char               *atom_header;
1561    ngx_buf_t            *atom;
1562    ngx_http_mp4_trak_t  *trak;
1563
1564    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "process mdia atom");
1565
1566    atom_header = ngx_mp4_atom_header(mp4);
1567    ngx_mp4_set_atom_name(atom_header, 'm', 'd', 'i', 'a');
1568
1569    trak = ngx_mp4_last_trak(mp4);
1570
1571    atom = &trak->mdia_atom_buf;
1572    atom->temporary = 1;
1573    atom->pos = atom_header;
1574    atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);
1575
1576    trak->out[NGX_HTTP_MP4_MDIA_ATOM].buf = atom;
1577
1578    return ngx_http_mp4_read_atom(mp4, ngx_http_mp4_mdia_atoms, atom_data_size);
1579}
1580
1581
1582static void
1583ngx_http_mp4_update_mdia_atom(ngx_http_mp4_file_t *mp4,
1584    ngx_http_mp4_trak_t *trak)
1585{
1586    ngx_buf_t  *atom;
1587
1588    trak->size += sizeof(ngx_mp4_atom_header_t);
1589    atom = &trak->mdia_atom_buf;
1590    ngx_mp4_set_32value(atom->pos, trak->size);
1591}
1592
1593
1594typedef struct {
1595    u_char    size[4];
1596    u_char    name[4];
1597    u_char    version[1];
1598    u_char    flags[3];
1599    u_char    creation_time[4];
1600    u_char    modification_time[4];
1601    u_char    timescale[4];
1602    u_char    duration[4];
1603    u_char    language[2];
1604    u_char    quality[2];
1605} ngx_mp4_mdhd_atom_t;
1606
1607typedef struct {
1608    u_char    size[4];
1609    u_char    name[4];
1610    u_char    version[1];
1611    u_char    flags[3];
1612    u_char    creation_time[8];
1613    u_char    modification_time[8];
1614    u_char    timescale[4];
1615    u_char    duration[8];
1616    u_char    language[2];
1617    u_char    quality[2];
1618} ngx_mp4_mdhd64_atom_t;
1619
1620
1621static ngx_int_t
1622ngx_http_mp4_read_mdhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1623{
1624    u_char                 *atom_header;
1625    size_t                  atom_size;
1626    uint32_t                timescale;
1627    uint64_t                duration, start_time, length_time;
1628    ngx_buf_t              *atom;
1629    ngx_http_mp4_trak_t    *trak;
1630    ngx_mp4_mdhd_atom_t    *mdhd_atom;
1631    ngx_mp4_mdhd64_atom_t  *mdhd64_atom;
1632
1633    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 mdhd atom");
1634
1635    atom_header = ngx_mp4_atom_header(mp4);
1636    mdhd_atom = (ngx_mp4_mdhd_atom_t *) atom_header;
1637    mdhd64_atom = (ngx_mp4_mdhd64_atom_t *) atom_header;
1638    ngx_mp4_set_atom_name(mdhd_atom, 'm', 'd', 'h', 'd');
1639
1640    if (ngx_mp4_atom_data_size(ngx_mp4_mdhd_atom_t) > atom_data_size) {
1641        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1642                      "\"%s\" mp4 mdhd atom too small", mp4->file.name.data);
1643        return NGX_ERROR;
1644    }
1645
1646    if (mdhd_atom->version[0] == 0) {
1647        /* version 0: everything is 32-bit */
1648        timescale = ngx_mp4_get_32value(mdhd_atom->timescale);
1649        duration = ngx_mp4_get_32value(mdhd_atom->duration);
1650
1651    } else {
1652        /* version 1: 64-bit duration and 32-bit timescale */
1653
1654        if (ngx_mp4_atom_data_size(ngx_mp4_mdhd64_atom_t) > atom_data_size) {
1655            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1656                          "\"%s\" mp4 mdhd atom too small",
1657                          mp4->file.name.data);
1658            return NGX_ERROR;
1659        }
1660
1661        timescale = ngx_mp4_get_32value(mdhd64_atom->timescale);
1662        duration = ngx_mp4_get_64value(mdhd64_atom->duration);
1663    }
1664
1665    ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1666                   "mdhd timescale:%uD, duration:%uL, time:%.3fs",
1667                   timescale, duration, (double) duration / timescale);
1668
1669    start_time = (uint64_t) mp4->start * timescale / 1000;
1670
1671    if (duration < start_time) {
1672        ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1673                       "mdhd duration is less than start time");
1674        return NGX_DECLINED;
1675    }
1676
1677    duration -= start_time;
1678
1679    if (mp4->length) {
1680        length_time = (uint64_t) mp4->length * timescale / 1000;
1681
1682        if (duration > length_time) {
1683            duration = length_time;
1684        }
1685    }
1686
1687    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1688                   "mdhd new duration:%uL, time:%.3fs",
1689                   duration, (double) duration / timescale);
1690
1691    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1692
1693    trak = ngx_mp4_last_trak(mp4);
1694    trak->mdhd_size = atom_size;
1695    trak->timescale = timescale;
1696
1697    ngx_mp4_set_32value(mdhd_atom->size, atom_size);
1698
1699    if (mdhd_atom->version[0] == 0) {
1700        ngx_mp4_set_32value(mdhd_atom->duration, duration);
1701
1702    } else {
1703        ngx_mp4_set_64value(mdhd64_atom->duration, duration);
1704    }
1705
1706    atom = &trak->mdhd_atom_buf;
1707    atom->temporary = 1;
1708    atom->pos = atom_header;
1709    atom->last = atom_header + atom_size;
1710
1711    trak->out[NGX_HTTP_MP4_MDHD_ATOM].buf = atom;
1712
1713    ngx_mp4_atom_next(mp4, atom_data_size);
1714
1715    return NGX_OK;
1716}
1717
1718
1719static ngx_int_t
1720ngx_http_mp4_read_hdlr_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1721{
1722    u_char              *atom_header;
1723    size_t               atom_size;
1724    ngx_buf_t            *atom;
1725    ngx_http_mp4_trak_t  *trak;
1726
1727    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 hdlr atom");
1728
1729    atom_header = ngx_mp4_atom_header(mp4);
1730    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1731    ngx_mp4_set_32value(atom_header, atom_size);
1732    ngx_mp4_set_atom_name(atom_header, 'h', 'd', 'l', 'r');
1733
1734    trak = ngx_mp4_last_trak(mp4);
1735
1736    atom = &trak->hdlr_atom_buf;
1737    atom->temporary = 1;
1738    atom->pos = atom_header;
1739    atom->last = atom_header + atom_size;
1740
1741    trak->hdlr_size = atom_size;
1742    trak->out[NGX_HTTP_MP4_HDLR_ATOM].buf = atom;
1743
1744    ngx_mp4_atom_next(mp4, atom_data_size);
1745
1746    return NGX_OK;
1747}
1748
1749
1750static ngx_int_t
1751ngx_http_mp4_read_minf_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1752{
1753    u_char               *atom_header;
1754    ngx_buf_t            *atom;
1755    ngx_http_mp4_trak_t  *trak;
1756
1757    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "process minf atom");
1758
1759    atom_header = ngx_mp4_atom_header(mp4);
1760    ngx_mp4_set_atom_name(atom_header, 'm', 'i', 'n', 'f');
1761
1762    trak = ngx_mp4_last_trak(mp4);
1763
1764    atom = &trak->minf_atom_buf;
1765    atom->temporary = 1;
1766    atom->pos = atom_header;
1767    atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);
1768
1769    trak->out[NGX_HTTP_MP4_MINF_ATOM].buf = atom;
1770
1771    return ngx_http_mp4_read_atom(mp4, ngx_http_mp4_minf_atoms, atom_data_size);
1772}
1773
1774
1775static void
1776ngx_http_mp4_update_minf_atom(ngx_http_mp4_file_t *mp4,
1777    ngx_http_mp4_trak_t *trak)
1778{
1779    ngx_buf_t  *atom;
1780
1781    trak->size += sizeof(ngx_mp4_atom_header_t)
1782               + trak->vmhd_size
1783               + trak->smhd_size
1784               + trak->dinf_size;
1785    atom = &trak->minf_atom_buf;
1786    ngx_mp4_set_32value(atom->pos, trak->size);
1787}
1788
1789
1790static ngx_int_t
1791ngx_http_mp4_read_vmhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1792{
1793    u_char              *atom_header;
1794    size_t               atom_size;
1795    ngx_buf_t            *atom;
1796    ngx_http_mp4_trak_t  *trak;
1797
1798    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 vmhd atom");
1799
1800    atom_header = ngx_mp4_atom_header(mp4);
1801    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1802    ngx_mp4_set_32value(atom_header, atom_size);
1803    ngx_mp4_set_atom_name(atom_header, 'v', 'm', 'h', 'd');
1804
1805    trak = ngx_mp4_last_trak(mp4);
1806
1807    atom = &trak->vmhd_atom_buf;
1808    atom->temporary = 1;
1809    atom->pos = atom_header;
1810    atom->last = atom_header + atom_size;
1811
1812    trak->vmhd_size += atom_size;
1813    trak->out[NGX_HTTP_MP4_VMHD_ATOM].buf = atom;
1814
1815    ngx_mp4_atom_next(mp4, atom_data_size);
1816
1817    return NGX_OK;
1818}
1819
1820
1821static ngx_int_t
1822ngx_http_mp4_read_smhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1823{
1824    u_char              *atom_header;
1825    size_t               atom_size;
1826    ngx_buf_t            *atom;
1827    ngx_http_mp4_trak_t  *trak;
1828
1829    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 smhd atom");
1830
1831    atom_header = ngx_mp4_atom_header(mp4);
1832    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1833    ngx_mp4_set_32value(atom_header, atom_size);
1834    ngx_mp4_set_atom_name(atom_header, 's', 'm', 'h', 'd');
1835
1836    trak = ngx_mp4_last_trak(mp4);
1837
1838    atom = &trak->smhd_atom_buf;
1839    atom->temporary = 1;
1840    atom->pos = atom_header;
1841    atom->last = atom_header + atom_size;
1842
1843    trak->vmhd_size += atom_size;
1844    trak->out[NGX_HTTP_MP4_SMHD_ATOM].buf = atom;
1845
1846    ngx_mp4_atom_next(mp4, atom_data_size);
1847
1848    return NGX_OK;
1849}
1850
1851
1852static ngx_int_t
1853ngx_http_mp4_read_dinf_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1854{
1855    u_char              *atom_header;
1856    size_t               atom_size;
1857    ngx_buf_t            *atom;
1858    ngx_http_mp4_trak_t  *trak;
1859
1860    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 dinf atom");
1861
1862    atom_header = ngx_mp4_atom_header(mp4);
1863    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1864    ngx_mp4_set_32value(atom_header, atom_size);
1865    ngx_mp4_set_atom_name(atom_header, 'd', 'i', 'n', 'f');
1866
1867    trak = ngx_mp4_last_trak(mp4);
1868
1869    atom = &trak->dinf_atom_buf;
1870    atom->temporary = 1;
1871    atom->pos = atom_header;
1872    atom->last = atom_header + atom_size;
1873
1874    trak->dinf_size += atom_size;
1875    trak->out[NGX_HTTP_MP4_DINF_ATOM].buf = atom;
1876
1877    ngx_mp4_atom_next(mp4, atom_data_size);
1878
1879    return NGX_OK;
1880}
1881
1882
1883static ngx_int_t
1884ngx_http_mp4_read_stbl_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1885{
1886    u_char               *atom_header;
1887    ngx_buf_t            *atom;
1888    ngx_http_mp4_trak_t  *trak;
1889
1890    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "process stbl atom");
1891
1892    atom_header = ngx_mp4_atom_header(mp4);
1893    ngx_mp4_set_atom_name(atom_header, 's', 't', 'b', 'l');
1894
1895    trak = ngx_mp4_last_trak(mp4);
1896
1897    atom = &trak->stbl_atom_buf;
1898    atom->temporary = 1;
1899    atom->pos = atom_header;
1900    atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);
1901
1902    trak->out[NGX_HTTP_MP4_STBL_ATOM].buf = atom;
1903
1904    return ngx_http_mp4_read_atom(mp4, ngx_http_mp4_stbl_atoms, atom_data_size);
1905}
1906
1907
1908static void
1909ngx_http_mp4_update_stbl_atom(ngx_http_mp4_file_t *mp4,
1910    ngx_http_mp4_trak_t *trak)
1911{
1912    ngx_buf_t  *atom;
1913
1914    trak->size += sizeof(ngx_mp4_atom_header_t);
1915    atom = &trak->stbl_atom_buf;
1916    ngx_mp4_set_32value(atom->pos, trak->size);
1917}
1918
1919
1920typedef struct {
1921    u_char    size[4];
1922    u_char    name[4];
1923    u_char    version[1];
1924    u_char    flags[3];
1925    u_char    entries[4];
1926
1927    u_char    media_size[4];
1928    u_char    media_name[4];
1929} ngx_mp4_stsd_atom_t;
1930
1931
1932static ngx_int_t
1933ngx_http_mp4_read_stsd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1934{
1935    u_char               *atom_header, *atom_table;
1936    size_t                atom_size;
1937    ngx_buf_t            *atom;
1938    ngx_mp4_stsd_atom_t  *stsd_atom;
1939    ngx_http_mp4_trak_t  *trak;
1940
1941    /* sample description atom */
1942
1943    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsd atom");
1944
1945    atom_header = ngx_mp4_atom_header(mp4);
1946    stsd_atom = (ngx_mp4_stsd_atom_t *) atom_header;
1947    atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
1948    atom_table = atom_header + atom_size;
1949    ngx_mp4_set_32value(stsd_atom->size, atom_size);
1950    ngx_mp4_set_atom_name(stsd_atom, 's', 't', 's', 'd');
1951
1952    if (ngx_mp4_atom_data_size(ngx_mp4_stsd_atom_t) > atom_data_size) {
1953        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
1954                      "\"%s\" mp4 stsd atom too small", mp4->file.name.data);
1955        return NGX_ERROR;
1956    }
1957
1958    ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
1959                   "stsd entries:%uD, media:%*s",
1960                   ngx_mp4_get_32value(stsd_atom->entries),
1961                   4, stsd_atom->media_name);
1962
1963    trak = ngx_mp4_last_trak(mp4);
1964
1965    atom = &trak->stsd_atom_buf;
1966    atom->temporary = 1;
1967    atom->pos = atom_header;
1968    atom->last = atom_table;
1969
1970    trak->out[NGX_HTTP_MP4_STSD_ATOM].buf = atom;
1971    trak->size += atom_size;
1972
1973    ngx_mp4_atom_next(mp4, atom_data_size);
1974
1975    return NGX_OK;
1976}
1977
1978
1979typedef struct {
1980    u_char    size[4];
1981    u_char    name[4];
1982    u_char    version[1];
1983    u_char    flags[3];
1984    u_char    entries[4];
1985} ngx_mp4_stts_atom_t;
1986
1987typedef struct {
1988    u_char    count[4];
1989    u_char    duration[4];
1990} ngx_mp4_stts_entry_t;
1991
1992
1993static ngx_int_t
1994ngx_http_mp4_read_stts_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
1995{
1996    u_char               *atom_header, *atom_table, *atom_end;
1997    uint32_t              entries;
1998    ngx_buf_t            *atom, *data;
1999    ngx_mp4_stts_atom_t  *stts_atom;
2000    ngx_http_mp4_trak_t  *trak;
2001
2002    /* time-to-sample atom */
2003
2004    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stts atom");
2005
2006    atom_header = ngx_mp4_atom_header(mp4);
2007    stts_atom = (ngx_mp4_stts_atom_t *) atom_header;
2008    ngx_mp4_set_atom_name(stts_atom, 's', 't', 't', 's');
2009
2010    if (ngx_mp4_atom_data_size(ngx_mp4_stts_atom_t) > atom_data_size) {
2011        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2012                      "\"%s\" mp4 stts atom too small", mp4->file.name.data);
2013        return NGX_ERROR;
2014    }
2015
2016    entries = ngx_mp4_get_32value(stts_atom->entries);
2017
2018    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2019                   "mp4 time-to-sample entries:%uD", entries);
2020
2021    if (ngx_mp4_atom_data_size(ngx_mp4_stts_atom_t)
2022        + entries * sizeof(ngx_mp4_stts_entry_t) > atom_data_size)
2023    {
2024        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2025                      "\"%s\" mp4 stts atom too small", mp4->file.name.data);
2026        return NGX_ERROR;
2027    }
2028
2029    atom_table = atom_header + sizeof(ngx_mp4_stts_atom_t);
2030    atom_end = atom_table + entries * sizeof(ngx_mp4_stts_entry_t);
2031
2032    trak = ngx_mp4_last_trak(mp4);
2033    trak->time_to_sample_entries = entries;
2034
2035    atom = &trak->stts_atom_buf;
2036    atom->temporary = 1;
2037    atom->pos = atom_header;
2038    atom->last = atom_table;
2039
2040    data = &trak->stts_data_buf;
2041    data->temporary = 1;
2042    data->pos = atom_table;
2043    data->last = atom_end;
2044
2045    trak->out[NGX_HTTP_MP4_STTS_ATOM].buf = atom;
2046    trak->out[NGX_HTTP_MP4_STTS_DATA].buf = data;
2047
2048    ngx_mp4_atom_next(mp4, atom_data_size);
2049
2050    return NGX_OK;
2051}
2052
2053
2054static ngx_int_t
2055ngx_http_mp4_update_stts_atom(ngx_http_mp4_file_t *mp4,
2056    ngx_http_mp4_trak_t *trak)
2057{
2058    size_t                atom_size;
2059    ngx_buf_t            *atom, *data;
2060    ngx_mp4_stts_atom_t  *stts_atom;
2061
2062    /*
2063     * mdia.minf.stbl.stts updating requires trak->timescale
2064     * from mdia.mdhd atom which may reside after mdia.minf
2065     */
2066
2067    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2068                   "mp4 stts atom update");
2069
2070    data = trak->out[NGX_HTTP_MP4_STTS_DATA].buf;
2071
2072    if (data == NULL) {
2073        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2074                      "no mp4 stts atoms were found in \"%s\"",
2075                      mp4->file.name.data);
2076        return NGX_ERROR;
2077    }
2078
2079    if (ngx_http_mp4_crop_stts_data(mp4, trak, 1) != NGX_OK) {
2080        return NGX_ERROR;
2081    }
2082
2083    if (ngx_http_mp4_crop_stts_data(mp4, trak, 0) != NGX_OK) {
2084        return NGX_ERROR;
2085    }
2086
2087    atom_size = sizeof(ngx_mp4_stts_atom_t) + (data->last - data->pos);
2088    trak->size += atom_size;
2089
2090    atom = trak->out[NGX_HTTP_MP4_STTS_ATOM].buf;
2091    stts_atom = (ngx_mp4_stts_atom_t *) atom->pos;
2092    ngx_mp4_set_32value(stts_atom->size, atom_size);
2093    ngx_mp4_set_32value(stts_atom->entries, trak->time_to_sample_entries);
2094
2095    return NGX_OK;
2096}
2097
2098
2099static ngx_int_t
2100ngx_http_mp4_crop_stts_data(ngx_http_mp4_file_t *mp4,
2101    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2102{
2103    uint32_t               count, duration, rest;
2104    uint64_t               start_time;
2105    ngx_buf_t             *data;
2106    ngx_uint_t             start_sample, entries, start_sec;
2107    ngx_mp4_stts_entry_t  *entry, *end;
2108
2109    if (start) {
2110        start_sec = mp4->start;
2111
2112    } else if (mp4->length) {
2113        start_sec = mp4->length;
2114
2115    } else {
2116        return NGX_OK;
2117    }
2118
2119    data = trak->out[NGX_HTTP_MP4_STTS_DATA].buf;
2120
2121    start_time = (uint64_t) start_sec * trak->timescale / 1000;
2122
2123    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2124                   "time-to-sample start_time:%uL", start_time);
2125
2126    entries = trak->time_to_sample_entries;
2127    start_sample = 0;
2128    entry = (ngx_mp4_stts_entry_t *) data->pos;
2129    end = (ngx_mp4_stts_entry_t *) data->last;
2130
2131    while (entry < end) {
2132        count = ngx_mp4_get_32value(entry->count);
2133        duration = ngx_mp4_get_32value(entry->duration);
2134
2135        ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2136                       "count:%uD, duration:%uD", count, duration);
2137
2138        if (start_time < (uint64_t) count * duration) {
2139            start_sample += (ngx_uint_t) (start_time / duration);
2140            rest = (uint32_t) (start_time / duration);
2141            goto found;
2142        }
2143
2144        start_sample += count;
2145        start_time -= count * duration;
2146        entries--;
2147        entry++;
2148    }
2149
2150    ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2151                  "start time is out mp4 stts samples in \"%s\"",
2152                  mp4->file.name.data);
2153
2154    return NGX_ERROR;
2155
2156found:
2157
2158    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2159                   "start_sample:%ui, new count:%uD", start_sample, count);
2160
2161    if (start) {
2162        ngx_mp4_set_32value(entry->count, count - rest);
2163        data->pos = (u_char *) entry;
2164        trak->time_to_sample_entries = entries;
2165        trak->start_sample = start_sample;
2166
2167    } else {
2168        ngx_mp4_set_32value(entry->count, rest);
2169        data->last = (u_char *) (entry + 1);
2170        trak->time_to_sample_entries -= entries - 1;
2171        trak->end_sample = trak->start_sample + start_sample;
2172    }
2173
2174    return NGX_OK;
2175}
2176
2177
2178typedef struct {
2179    u_char    size[4];
2180    u_char    name[4];
2181    u_char    version[1];
2182    u_char    flags[3];
2183    u_char    entries[4];
2184} ngx_http_mp4_stss_atom_t;
2185
2186
2187static ngx_int_t
2188ngx_http_mp4_read_stss_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2189{
2190    u_char                    *atom_header, *atom_table, *atom_end;
2191    uint32_t                   entries;
2192    ngx_buf_t                 *atom, *data;
2193    ngx_http_mp4_trak_t       *trak;
2194    ngx_http_mp4_stss_atom_t  *stss_atom;
2195
2196    /* sync samples atom */
2197
2198    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stss atom");
2199
2200    atom_header = ngx_mp4_atom_header(mp4);
2201    stss_atom = (ngx_http_mp4_stss_atom_t *) atom_header;
2202    ngx_mp4_set_atom_name(stss_atom, 's', 't', 's', 's');
2203
2204    if (ngx_mp4_atom_data_size(ngx_http_mp4_stss_atom_t) > atom_data_size) {
2205        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2206                      "\"%s\" mp4 stss atom too small", mp4->file.name.data);
2207        return NGX_ERROR;
2208    }
2209
2210    entries = ngx_mp4_get_32value(stss_atom->entries);
2211
2212    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2213                   "sync sample entries:%uD", entries);
2214
2215    trak = ngx_mp4_last_trak(mp4);
2216    trak->sync_samples_entries = entries;
2217
2218    atom_table = atom_header + sizeof(ngx_http_mp4_stss_atom_t);
2219
2220    atom = &trak->stss_atom_buf;
2221    atom->temporary = 1;
2222    atom->pos = atom_header;
2223    atom->last = atom_table;
2224
2225    if (ngx_mp4_atom_data_size(ngx_http_mp4_stss_atom_t)
2226        + entries * sizeof(uint32_t) > atom_data_size)
2227    {
2228        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2229                      "\"%s\" mp4 stss atom too small", mp4->file.name.data);
2230        return NGX_ERROR;
2231    }
2232
2233    atom_end = atom_table + entries * sizeof(uint32_t);
2234
2235    data = &trak->stss_data_buf;
2236    data->temporary = 1;
2237    data->pos = atom_table;
2238    data->last = atom_end;
2239
2240    trak->out[NGX_HTTP_MP4_STSS_ATOM].buf = atom;
2241    trak->out[NGX_HTTP_MP4_STSS_DATA].buf = data;
2242
2243    ngx_mp4_atom_next(mp4, atom_data_size);
2244
2245    return NGX_OK;
2246}
2247
2248
2249static ngx_int_t
2250ngx_http_mp4_update_stss_atom(ngx_http_mp4_file_t *mp4,
2251    ngx_http_mp4_trak_t *trak)
2252{
2253    size_t                     atom_size;
2254    uint32_t                   sample, start_sample, *entry, *end;
2255    ngx_buf_t                 *atom, *data;
2256    ngx_http_mp4_stss_atom_t  *stss_atom;
2257
2258    /*
2259     * mdia.minf.stbl.stss updating requires trak->start_sample
2260     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2261     * atom which may reside after mdia.minf
2262     */
2263
2264    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2265                   "mp4 stss atom update");
2266
2267    data = trak->out[NGX_HTTP_MP4_STSS_DATA].buf;
2268
2269    if (data == NULL) {
2270        return NGX_OK;
2271    }
2272
2273    ngx_http_mp4_crop_stss_data(mp4, trak, 1);
2274    ngx_http_mp4_crop_stss_data(mp4, trak, 0);
2275
2276    entry = (uint32_t *) data->pos;
2277    end = (uint32_t *) data->last;
2278
2279    start_sample = trak->start_sample;
2280
2281    while (entry < end) {
2282        sample = ngx_mp4_get_32value(entry);
2283        sample -= start_sample;
2284        ngx_mp4_set_32value(entry, sample);
2285        entry++;
2286    }
2287
2288    atom_size = sizeof(ngx_http_mp4_stss_atom_t) + (data->last - data->pos);
2289    trak->size += atom_size;
2290
2291    atom = trak->out[NGX_HTTP_MP4_STSS_ATOM].buf;
2292    stss_atom = (ngx_http_mp4_stss_atom_t *) atom->pos;
2293
2294    ngx_mp4_set_32value(stss_atom->size, atom_size);
2295    ngx_mp4_set_32value(stss_atom->entries, trak->sync_samples_entries);
2296
2297    return NGX_OK;
2298}
2299
2300
2301static void
2302ngx_http_mp4_crop_stss_data(ngx_http_mp4_file_t *mp4,
2303    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2304{
2305    uint32_t     sample, start_sample, *entry, *end;
2306    ngx_buf_t   *data;
2307    ngx_uint_t   entries;
2308
2309    /* sync samples starts from 1 */
2310
2311    if (start) {
2312        start_sample = trak->start_sample + 1;
2313
2314    } else if (mp4->length) {
2315        start_sample = trak->end_sample + 1;
2316
2317    } else {
2318        return;
2319    }
2320
2321    data = trak->out[NGX_HTTP_MP4_STSS_DATA].buf;
2322
2323    entries = trak->sync_samples_entries;
2324    entry = (uint32_t *) data->pos;
2325    end = (uint32_t *) data->last;
2326
2327    while (entry < end) {
2328        sample = ngx_mp4_get_32value(entry);
2329
2330        ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2331                       "start:%uD, sync:%uD", start_sample, sample);
2332
2333        if (sample >= start_sample) {
2334            goto found;
2335        }
2336
2337        entries--;
2338        entry++;
2339    }
2340
2341    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2342                   "start sample is out of mp4 stss atom");
2343
2344found:
2345
2346    if (start) {
2347        data->pos = (u_char *) entry;
2348        trak->sync_samples_entries = entries;
2349
2350    } else {
2351        data->last = (u_char *) entry;
2352        trak->sync_samples_entries -= entries;
2353    }
2354}
2355
2356
2357typedef struct {
2358    u_char    size[4];
2359    u_char    name[4];
2360    u_char    version[1];
2361    u_char    flags[3];
2362    u_char    entries[4];
2363} ngx_mp4_ctts_atom_t;
2364
2365typedef struct {
2366    u_char    count[4];
2367    u_char    offset[4];
2368} ngx_mp4_ctts_entry_t;
2369
2370
2371static ngx_int_t
2372ngx_http_mp4_read_ctts_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2373{
2374    u_char               *atom_header, *atom_table, *atom_end;
2375    uint32_t              entries;
2376    ngx_buf_t            *atom, *data;
2377    ngx_mp4_ctts_atom_t  *ctts_atom;
2378    ngx_http_mp4_trak_t  *trak;
2379
2380    /* composition offsets atom */
2381
2382    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 ctts atom");
2383
2384    atom_header = ngx_mp4_atom_header(mp4);
2385    ctts_atom = (ngx_mp4_ctts_atom_t *) atom_header;
2386    ngx_mp4_set_atom_name(ctts_atom, 'c', 't', 't', 's');
2387
2388    if (ngx_mp4_atom_data_size(ngx_mp4_ctts_atom_t) > atom_data_size) {
2389        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2390                      "\"%s\" mp4 ctts atom too small", mp4->file.name.data);
2391        return NGX_ERROR;
2392    }
2393
2394    entries = ngx_mp4_get_32value(ctts_atom->entries);
2395
2396    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2397                   "composition offset entries:%uD", entries);
2398
2399    trak = ngx_mp4_last_trak(mp4);
2400    trak->composition_offset_entries = entries;
2401
2402    atom_table = atom_header + sizeof(ngx_mp4_ctts_atom_t);
2403
2404    atom = &trak->ctts_atom_buf;
2405    atom->temporary = 1;
2406    atom->pos = atom_header;
2407    atom->last = atom_table;
2408
2409    if (ngx_mp4_atom_data_size(ngx_mp4_ctts_atom_t)
2410        + entries * sizeof(ngx_mp4_ctts_entry_t) > atom_data_size)
2411    {
2412        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2413                      "\"%s\" mp4 ctts atom too small", mp4->file.name.data);
2414        return NGX_ERROR;
2415    }
2416
2417    atom_end = atom_table + entries * sizeof(ngx_mp4_ctts_entry_t);
2418
2419    data = &trak->ctts_data_buf;
2420    data->temporary = 1;
2421    data->pos = atom_table;
2422    data->last = atom_end;
2423
2424    trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf = atom;
2425    trak->out[NGX_HTTP_MP4_CTTS_DATA].buf = data;
2426
2427    ngx_mp4_atom_next(mp4, atom_data_size);
2428
2429    return NGX_OK;
2430}
2431
2432
2433static void
2434ngx_http_mp4_update_ctts_atom(ngx_http_mp4_file_t *mp4,
2435    ngx_http_mp4_trak_t *trak)
2436{
2437    size_t                atom_size;
2438    ngx_buf_t            *atom, *data;
2439    ngx_mp4_ctts_atom_t  *ctts_atom;
2440
2441    /*
2442     * mdia.minf.stbl.ctts updating requires trak->start_sample
2443     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2444     * atom which may reside after mdia.minf
2445     */
2446
2447    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2448                   "mp4 ctts atom update");
2449
2450    data = trak->out[NGX_HTTP_MP4_CTTS_DATA].buf;
2451
2452    if (data == NULL) {
2453        return;
2454    }
2455
2456    ngx_http_mp4_crop_ctts_data(mp4, trak, 1);
2457    ngx_http_mp4_crop_ctts_data(mp4, trak, 0);
2458
2459    if (trak->composition_offset_entries == 0) {
2460        trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf = NULL;
2461        trak->out[NGX_HTTP_MP4_CTTS_DATA].buf = NULL;
2462        return;
2463    }
2464
2465    atom_size = sizeof(ngx_mp4_ctts_atom_t) + (data->last - data->pos);
2466    trak->size += atom_size;
2467
2468    atom = trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf;
2469    ctts_atom = (ngx_mp4_ctts_atom_t *) atom->pos;
2470
2471    ngx_mp4_set_32value(ctts_atom->size, atom_size);
2472    ngx_mp4_set_32value(ctts_atom->entries, trak->composition_offset_entries);
2473
2474    return;
2475}
2476
2477
2478static void
2479ngx_http_mp4_crop_ctts_data(ngx_http_mp4_file_t *mp4,
2480    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2481{
2482    uint32_t               count, start_sample, rest;
2483    ngx_buf_t             *data;
2484    ngx_uint_t             entries;
2485    ngx_mp4_ctts_entry_t  *entry, *end;
2486
2487    /* sync samples starts from 1 */
2488
2489    if (start) {
2490        start_sample = trak->start_sample + 1;
2491
2492    } else if (mp4->length) {
2493        start_sample = trak->end_sample - trak->start_sample + 1;
2494
2495    } else {
2496        return;
2497    }
2498
2499    data = trak->out[NGX_HTTP_MP4_CTTS_DATA].buf;
2500
2501    entries = trak->composition_offset_entries;
2502    entry = (ngx_mp4_ctts_entry_t *) data->pos;
2503    end = (ngx_mp4_ctts_entry_t *) data->last;
2504
2505    while (entry < end) {
2506        count = ngx_mp4_get_32value(entry->count);
2507
2508        ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2509                       "start:%uD, count:%uD, offset:%uD",
2510                       start_sample, count, ngx_mp4_get_32value(entry->offset));
2511
2512         if (start_sample <= count) {
2513             rest = start_sample - 1;
2514             goto found;
2515         }
2516
2517         start_sample -= count;
2518         entries--;
2519         entry++;
2520    }
2521
2522    if (start) {
2523        data->pos = (u_char *) end;
2524        trak->composition_offset_entries = 0;
2525    }
2526
2527    return;
2528
2529found:
2530
2531    if (start) {
2532        ngx_mp4_set_32value(entry->count, count - rest);
2533        data->pos = (u_char *) entry;
2534        trak->composition_offset_entries = entries;
2535
2536    } else {
2537        ngx_mp4_set_32value(entry->count, rest);
2538        data->last = (u_char *) (entry + 1);
2539        trak->composition_offset_entries -= entries - 1;
2540    }
2541}
2542
2543
2544typedef struct {
2545    u_char    size[4];
2546    u_char    name[4];
2547    u_char    version[1];
2548    u_char    flags[3];
2549    u_char    entries[4];
2550} ngx_mp4_stsc_atom_t;
2551
2552
2553static ngx_int_t
2554ngx_http_mp4_read_stsc_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2555{
2556    u_char               *atom_header, *atom_table, *atom_end;
2557    uint32_t              entries;
2558    ngx_buf_t            *atom, *data;
2559    ngx_mp4_stsc_atom_t  *stsc_atom;
2560    ngx_http_mp4_trak_t  *trak;
2561
2562    /* sample-to-chunk atom */
2563
2564    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsc atom");
2565
2566    atom_header = ngx_mp4_atom_header(mp4);
2567    stsc_atom = (ngx_mp4_stsc_atom_t *) atom_header;
2568    ngx_mp4_set_atom_name(stsc_atom, 's', 't', 's', 'c');
2569
2570    if (ngx_mp4_atom_data_size(ngx_mp4_stsc_atom_t) > atom_data_size) {
2571        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2572                      "\"%s\" mp4 stsc atom too small", mp4->file.name.data);
2573        return NGX_ERROR;
2574    }
2575
2576    entries = ngx_mp4_get_32value(stsc_atom->entries);
2577
2578    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2579                   "sample-to-chunk entries:%uD", entries);
2580
2581    if (ngx_mp4_atom_data_size(ngx_mp4_stsc_atom_t)
2582        + entries * sizeof(ngx_mp4_stsc_entry_t) > atom_data_size)
2583    {
2584        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2585                      "\"%s\" mp4 stsc atom too small", mp4->file.name.data);
2586        return NGX_ERROR;
2587    }
2588
2589    atom_table = atom_header + sizeof(ngx_mp4_stsc_atom_t);
2590    atom_end = atom_table + entries * sizeof(ngx_mp4_stsc_entry_t);
2591
2592    trak = ngx_mp4_last_trak(mp4);
2593    trak->sample_to_chunk_entries = entries;
2594
2595    atom = &trak->stsc_atom_buf;
2596    atom->temporary = 1;
2597    atom->pos = atom_header;
2598    atom->last = atom_table;
2599
2600    data = &trak->stsc_data_buf;
2601    data->temporary = 1;
2602    data->pos = atom_table;
2603    data->last = atom_end;
2604
2605    trak->out[NGX_HTTP_MP4_STSC_ATOM].buf = atom;
2606    trak->out[NGX_HTTP_MP4_STSC_DATA].buf = data;
2607
2608    ngx_mp4_atom_next(mp4, atom_data_size);
2609
2610    return NGX_OK;
2611}
2612
2613
2614static ngx_int_t
2615ngx_http_mp4_update_stsc_atom(ngx_http_mp4_file_t *mp4,
2616    ngx_http_mp4_trak_t *trak)
2617{
2618    size_t                 atom_size;
2619    uint32_t               chunk;
2620    ngx_buf_t             *atom, *data;
2621    ngx_mp4_stsc_atom_t   *stsc_atom;
2622    ngx_mp4_stsc_entry_t  *entry, *end;
2623
2624    /*
2625     * mdia.minf.stbl.stsc updating requires trak->start_sample
2626     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2627     * atom which may reside after mdia.minf
2628     */
2629
2630    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2631                   "mp4 stsc atom update");
2632
2633    data = trak->out[NGX_HTTP_MP4_STSC_DATA].buf;
2634
2635    if (data == NULL) {
2636        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2637                      "no mp4 stsc atoms were found in \"%s\"",
2638                      mp4->file.name.data);
2639        return NGX_ERROR;
2640    }
2641
2642    if (trak->sample_to_chunk_entries == 0) {
2643        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2644                      "zero number of entries in stsc atom in \"%s\"",
2645                      mp4->file.name.data);
2646        return NGX_ERROR;
2647    }
2648
2649    if (ngx_http_mp4_crop_stsc_data(mp4, trak, 1) != NGX_OK) {
2650        return NGX_ERROR;
2651    }
2652
2653    if (ngx_http_mp4_crop_stsc_data(mp4, trak, 0) != NGX_OK) {
2654        return NGX_ERROR;
2655    }
2656
2657    entry = (ngx_mp4_stsc_entry_t *) data->pos;
2658    end = (ngx_mp4_stsc_entry_t *) data->last;
2659
2660    while (entry < end) {
2661        chunk = ngx_mp4_get_32value(entry->chunk);
2662        chunk -= trak->start_chunk;
2663        ngx_mp4_set_32value(entry->chunk, chunk);
2664        entry++;
2665    }
2666
2667    atom_size = sizeof(ngx_mp4_stsc_atom_t)
2668                + trak->sample_to_chunk_entries * sizeof(ngx_mp4_stsc_entry_t);
2669
2670    trak->size += atom_size;
2671
2672    atom = trak->out[NGX_HTTP_MP4_STSC_ATOM].buf;
2673    stsc_atom = (ngx_mp4_stsc_atom_t *) atom->pos;
2674
2675    ngx_mp4_set_32value(stsc_atom->size, atom_size);
2676    ngx_mp4_set_32value(stsc_atom->entries, trak->sample_to_chunk_entries);
2677
2678    return NGX_OK;
2679}
2680
2681
2682static ngx_int_t
2683ngx_http_mp4_crop_stsc_data(ngx_http_mp4_file_t *mp4,
2684    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2685{
2686    uint32_t               start_sample, chunk, samples, id, next_chunk, n,
2687                           prev_samples;
2688    ngx_buf_t             *data, *buf;
2689    ngx_uint_t             entries, target_chunk, chunk_samples;
2690    ngx_mp4_stsc_entry_t  *entry, *end, *first;
2691
2692    entries = trak->sample_to_chunk_entries - 1;
2693
2694    if (start) {
2695        start_sample = (uint32_t) trak->start_sample;
2696
2697    } else if (mp4->length) {
2698        start_sample = (uint32_t) (trak->end_sample - trak->start_sample);
2699
2700        data = trak->out[NGX_HTTP_MP4_STSC_START].buf;
2701
2702        if (data) {
2703            entry = (ngx_mp4_stsc_entry_t *) data->pos;
2704            samples = ngx_mp4_get_32value(entry->samples);
2705            entries--;
2706
2707            if (samples > start_sample) {
2708                samples = start_sample;
2709                ngx_mp4_set_32value(entry->samples, samples);
2710            }
2711
2712            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2713                           "mp4 stsc using %uD start samples", samples);
2714
2715            start_sample -= samples;
2716        }
2717
2718    } else {
2719        return NGX_OK;
2720    }
2721
2722    data = trak->out[NGX_HTTP_MP4_STSC_DATA].buf;
2723
2724    entry = (ngx_mp4_stsc_entry_t *) data->pos;
2725    end = (ngx_mp4_stsc_entry_t *) data->last;
2726
2727    chunk = ngx_mp4_get_32value(entry->chunk);
2728    samples = ngx_mp4_get_32value(entry->samples);
2729    id = ngx_mp4_get_32value(entry->id);
2730    prev_samples = 0;
2731    entry++;
2732
2733    while (entry < end) {
2734
2735        next_chunk = ngx_mp4_get_32value(entry->chunk);
2736
2737        ngx_log_debug5(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2738                       "start_sample:%uD, chunk:%uD, chunks:%uD, "
2739                       "samples:%uD, id:%uD",
2740                       start_sample, chunk, next_chunk - chunk, samples, id);
2741
2742        n = (next_chunk - chunk) * samples;
2743
2744        if (start_sample < n) {
2745            goto found;
2746        }
2747
2748        start_sample -= n;
2749
2750        prev_samples = samples;
2751        chunk = next_chunk;
2752        samples = ngx_mp4_get_32value(entry->samples);
2753        id = ngx_mp4_get_32value(entry->id);
2754        entries--;
2755        entry++;
2756    }
2757
2758    next_chunk = trak->chunks + 1;
2759
2760    ngx_log_debug4(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2761                   "start_sample:%uD, chunk:%uD, chunks:%uD, samples:%uD",
2762                   start_sample, chunk, next_chunk - chunk, samples);
2763
2764    n = (next_chunk - chunk) * samples;
2765
2766    if (start_sample > n) {
2767        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2768                      "start time is out mp4 stsc chunks in \"%s\"",
2769                      mp4->file.name.data);
2770        return NGX_ERROR;
2771    }
2772
2773found:
2774
2775    entries++;
2776    entry--;
2777
2778    if (samples == 0) {
2779        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2780                      "zero number of samples in \"%s\"",
2781                      mp4->file.name.data);
2782        return NGX_ERROR;
2783    }
2784
2785    target_chunk = chunk - 1;
2786    target_chunk += start_sample / samples;
2787    chunk_samples = start_sample % samples;
2788
2789    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2790                   "start chunk:%ui, samples:%uD",
2791                   target_chunk, chunk_samples);
2792
2793    if (start) {
2794        data->pos = (u_char *) entry;
2795
2796        trak->sample_to_chunk_entries = entries;
2797        trak->start_chunk = target_chunk;
2798        trak->start_chunk_samples = chunk_samples;
2799
2800        ngx_mp4_set_32value(entry->chunk, trak->start_chunk + 1);
2801
2802        samples -= chunk_samples;
2803
2804    } else {
2805        if (start_sample) {
2806            data->last = (u_char *) (entry + 1);
2807            trak->sample_to_chunk_entries -= entries - 1;
2808            trak->end_chunk_samples = samples;
2809
2810        } else {
2811            data->last = (u_char *) entry;
2812            trak->sample_to_chunk_entries -= entries;
2813            trak->end_chunk_samples = prev_samples;
2814        }
2815
2816        if (chunk_samples) {
2817            trak->end_chunk = target_chunk + 1;
2818            trak->end_chunk_samples = chunk_samples;
2819
2820        } else {
2821            trak->end_chunk = target_chunk;
2822        }
2823
2824        samples = chunk_samples;
2825        next_chunk = chunk + 1;
2826    }
2827
2828    if (chunk_samples && next_chunk - target_chunk == 2) {
2829
2830        ngx_mp4_set_32value(entry->samples, samples);
2831
2832    } else if (chunk_samples && start) {
2833
2834        first = &trak->stsc_start_chunk_entry;
2835        ngx_mp4_set_32value(first->chunk, 1);
2836        ngx_mp4_set_32value(first->samples, samples);
2837        ngx_mp4_set_32value(first->id, id);
2838
2839        buf = &trak->stsc_start_chunk_buf;
2840        buf->temporary = 1;
2841        buf->pos = (u_char *) first;
2842        buf->last = (u_char *) first + sizeof(ngx_mp4_stsc_entry_t);
2843
2844        trak->out[NGX_HTTP_MP4_STSC_START].buf = buf;
2845
2846        ngx_mp4_set_32value(entry->chunk, trak->start_chunk + 2);
2847
2848        trak->sample_to_chunk_entries++;
2849
2850    } else if (chunk_samples) {
2851
2852        first = &trak->stsc_end_chunk_entry;
2853        ngx_mp4_set_32value(first->chunk, trak->end_chunk - trak->start_chunk);
2854        ngx_mp4_set_32value(first->samples, samples);
2855        ngx_mp4_set_32value(first->id, id);
2856
2857        buf = &trak->stsc_end_chunk_buf;
2858        buf->temporary = 1;
2859        buf->pos = (u_char *) first;
2860        buf->last = (u_char *) first + sizeof(ngx_mp4_stsc_entry_t);
2861
2862        trak->out[NGX_HTTP_MP4_STSC_END].buf = buf;
2863
2864        trak->sample_to_chunk_entries++;
2865    }
2866
2867    return NGX_OK;
2868}
2869
2870
2871typedef struct {
2872    u_char    size[4];
2873    u_char    name[4];
2874    u_char    version[1];
2875    u_char    flags[3];
2876    u_char    uniform_size[4];
2877    u_char    entries[4];
2878} ngx_mp4_stsz_atom_t;
2879
2880
2881static ngx_int_t
2882ngx_http_mp4_read_stsz_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2883{
2884    u_char               *atom_header, *atom_table, *atom_end;
2885    size_t                atom_size;
2886    uint32_t              entries, size;
2887    ngx_buf_t            *atom, *data;
2888    ngx_mp4_stsz_atom_t  *stsz_atom;
2889    ngx_http_mp4_trak_t  *trak;
2890
2891    /* sample sizes atom */
2892
2893    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsz atom");
2894
2895    atom_header = ngx_mp4_atom_header(mp4);
2896    stsz_atom = (ngx_mp4_stsz_atom_t *) atom_header;
2897    ngx_mp4_set_atom_name(stsz_atom, 's', 't', 's', 'z');
2898
2899    if (ngx_mp4_atom_data_size(ngx_mp4_stsz_atom_t) > atom_data_size) {
2900        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2901                      "\"%s\" mp4 stsz atom too small", mp4->file.name.data);
2902        return NGX_ERROR;
2903    }
2904
2905    size = ngx_mp4_get_32value(stsz_atom->uniform_size);
2906    entries = ngx_mp4_get_32value(stsz_atom->entries);
2907
2908    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2909                   "sample uniform size:%uD, entries:%uD", size, entries);
2910
2911    trak = ngx_mp4_last_trak(mp4);
2912    trak->sample_sizes_entries = entries;
2913
2914    atom_table = atom_header + sizeof(ngx_mp4_stsz_atom_t);
2915
2916    atom = &trak->stsz_atom_buf;
2917    atom->temporary = 1;
2918    atom->pos = atom_header;
2919    atom->last = atom_table;
2920
2921    trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf = atom;
2922
2923    if (size == 0) {
2924        if (ngx_mp4_atom_data_size(ngx_mp4_stsz_atom_t)
2925            + entries * sizeof(uint32_t) > atom_data_size)
2926        {
2927            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2928                          "\"%s\" mp4 stsz atom too small",
2929                          mp4->file.name.data);
2930            return NGX_ERROR;
2931        }
2932
2933        atom_end = atom_table + entries * sizeof(uint32_t);
2934
2935        data = &trak->stsz_data_buf;
2936        data->temporary = 1;
2937        data->pos = atom_table;
2938        data->last = atom_end;
2939
2940        trak->out[NGX_HTTP_MP4_STSZ_DATA].buf = data;
2941
2942    } else {
2943        /* if size != 0 then all samples are the same size */
2944        /* TODO : chunk samples */
2945        atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
2946        ngx_mp4_set_32value(atom_header, atom_size);
2947        trak->size += atom_size;
2948    }
2949
2950    ngx_mp4_atom_next(mp4, atom_data_size);
2951
2952    return NGX_OK;
2953}
2954
2955
2956static ngx_int_t
2957ngx_http_mp4_update_stsz_atom(ngx_http_mp4_file_t *mp4,
2958    ngx_http_mp4_trak_t *trak)
2959{
2960    size_t                atom_size;
2961    uint32_t             *pos, *end, entries;
2962    ngx_buf_t            *atom, *data;
2963    ngx_mp4_stsz_atom_t  *stsz_atom;
2964
2965    /*
2966     * mdia.minf.stbl.stsz updating requires trak->start_sample
2967     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2968     * atom which may reside after mdia.minf
2969     */
2970
2971    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2972                   "mp4 stsz atom update");
2973
2974    data = trak->out[NGX_HTTP_MP4_STSZ_DATA].buf;
2975
2976    if (data) {
2977        entries = trak->sample_sizes_entries;
2978
2979        if (trak->start_sample > entries) {
2980            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2981                          "start time is out mp4 stsz samples in \"%s\"",
2982                          mp4->file.name.data);
2983            return NGX_ERROR;
2984        }
2985
2986        entries -= trak->start_sample;
2987        data->pos += trak->start_sample * sizeof(uint32_t);
2988        end = (uint32_t *) data->pos;
2989
2990        for (pos = end - trak->start_chunk_samples; pos < end; pos++) {
2991            trak->start_chunk_samples_size += ngx_mp4_get_32value(pos);
2992        }
2993
2994        ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2995                       "chunk samples sizes:%uL",
2996                       trak->start_chunk_samples_size);
2997
2998        if (mp4->length) {
2999            if (trak->end_sample - trak->start_sample > entries) {
3000                ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3001                              "end time is out mp4 stsz samples in \"%s\"",
3002                              mp4->file.name.data);
3003                return NGX_ERROR;
3004            }
3005
3006            entries = trak->end_sample - trak->start_sample;
3007            data->last = data->pos + entries * sizeof(uint32_t);
3008            end = (uint32_t *) data->last;
3009
3010            for (pos = end - trak->end_chunk_samples; pos < end; pos++) {
3011                trak->end_chunk_samples_size += ngx_mp4_get_32value(pos);
3012            }
3013
3014            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3015                           "mp4 stsz end_chunk_samples_size:%uL",
3016                           trak->end_chunk_samples_size);
3017        }
3018
3019        atom_size = sizeof(ngx_mp4_stsz_atom_t) + (data->last - data->pos);
3020        trak->size += atom_size;
3021
3022        atom = trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf;
3023        stsz_atom = (ngx_mp4_stsz_atom_t *) atom->pos;
3024
3025        ngx_mp4_set_32value(stsz_atom->size, atom_size);
3026        ngx_mp4_set_32value(stsz_atom->entries, entries);
3027    }
3028
3029    return NGX_OK;
3030}
3031
3032
3033typedef struct {
3034    u_char    size[4];
3035    u_char    name[4];
3036    u_char    version[1];
3037    u_char    flags[3];
3038    u_char    entries[4];
3039} ngx_mp4_stco_atom_t;
3040
3041
3042static ngx_int_t
3043ngx_http_mp4_read_stco_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
3044{
3045    u_char               *atom_header, *atom_table, *atom_end;
3046    uint32_t              entries;
3047    ngx_buf_t            *atom, *data;
3048    ngx_mp4_stco_atom_t  *stco_atom;
3049    ngx_http_mp4_trak_t  *trak;
3050
3051    /* chunk offsets atom */
3052
3053    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stco atom");
3054
3055    atom_header = ngx_mp4_atom_header(mp4);
3056    stco_atom = (ngx_mp4_stco_atom_t *) atom_header;
3057    ngx_mp4_set_atom_name(stco_atom, 's', 't', 'c', 'o');
3058
3059    if (ngx_mp4_atom_data_size(ngx_mp4_stco_atom_t) > atom_data_size) {
3060        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3061                      "\"%s\" mp4 stco atom too small", mp4->file.name.data);
3062        return NGX_ERROR;
3063    }
3064
3065    entries = ngx_mp4_get_32value(stco_atom->entries);
3066
3067    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "chunks:%uD", entries);
3068
3069    if (ngx_mp4_atom_data_size(ngx_mp4_stco_atom_t)
3070        + entries * sizeof(uint32_t) > atom_data_size)
3071    {
3072        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3073                      "\"%s\" mp4 stco atom too small", mp4->file.name.data);
3074        return NGX_ERROR;
3075    }
3076
3077    atom_table = atom_header + sizeof(ngx_mp4_stco_atom_t);
3078    atom_end = atom_table + entries * sizeof(uint32_t);
3079
3080    trak = ngx_mp4_last_trak(mp4);
3081    trak->chunks = entries;
3082
3083    atom = &trak->stco_atom_buf;
3084    atom->temporary = 1;
3085    atom->pos = atom_header;
3086    atom->last = atom_table;
3087
3088    data = &trak->stco_data_buf;
3089    data->temporary = 1;
3090    data->pos = atom_table;
3091    data->last = atom_end;
3092
3093    trak->out[NGX_HTTP_MP4_STCO_ATOM].buf = atom;
3094    trak->out[NGX_HTTP_MP4_STCO_DATA].buf = data;
3095
3096    ngx_mp4_atom_next(mp4, atom_data_size);
3097
3098    return NGX_OK;
3099}
3100
3101
3102static ngx_int_t
3103ngx_http_mp4_update_stco_atom(ngx_http_mp4_file_t *mp4,
3104    ngx_http_mp4_trak_t *trak)
3105{
3106    size_t                atom_size;
3107    uint32_t              entries;
3108    ngx_buf_t            *atom, *data;
3109    ngx_mp4_stco_atom_t  *stco_atom;
3110
3111    /*
3112     * mdia.minf.stbl.stco updating requires trak->start_chunk
3113     * from mdia.minf.stbl.stsc which depends on value from mdia.mdhd
3114     * atom which may reside after mdia.minf
3115     */
3116
3117    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3118                   "mp4 stco atom update");
3119
3120    data = trak->out[NGX_HTTP_MP4_STCO_DATA].buf;
3121
3122    if (data == NULL) {
3123        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3124                      "no mp4 stco atoms were found in \"%s\"",
3125                      mp4->file.name.data);
3126        return NGX_ERROR;
3127    }
3128
3129    if (trak->start_chunk > trak->chunks) {
3130        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3131                      "start time is out mp4 stco chunks in \"%s\"",
3132                      mp4->file.name.data);
3133        return NGX_ERROR;
3134    }
3135
3136    data->pos += trak->start_chunk * sizeof(uint32_t);
3137
3138    trak->start_offset = ngx_mp4_get_32value(data->pos);
3139    trak->start_offset += trak->start_chunk_samples_size;
3140    ngx_mp4_set_32value(data->pos, trak->start_offset);
3141
3142    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3143                   "start chunk offset:%uD", trak->start_offset);
3144
3145    if (mp4->length) {
3146
3147        if (trak->end_chunk > trak->chunks) {
3148            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3149                          "end time is out mp4 stco chunks in \"%s\"",
3150                          mp4->file.name.data);
3151            return NGX_ERROR;
3152        }
3153
3154        entries = trak->end_chunk - trak->start_chunk;
3155        data->last = data->pos + entries * sizeof(uint32_t);
3156
3157        if (entries) {
3158            trak->end_offset =
3159                            ngx_mp4_get_32value(data->last - sizeof(uint32_t));
3160            trak->end_offset += trak->end_chunk_samples_size;
3161
3162            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3163                           "end chunk offset:%O", trak->end_offset);
3164        }
3165
3166    } else {
3167        entries = trak->chunks - trak->start_chunk;
3168        trak->end_offset = mp4->mdat_data.buf->file_last;
3169    }
3170
3171    if (entries == 0) {
3172        trak->start_offset = mp4->end;
3173        trak->end_offset = 0;
3174    }
3175
3176    atom_size = sizeof(ngx_mp4_stco_atom_t) + (data->last - data->pos);
3177    trak->size += atom_size;
3178
3179    atom = trak->out[NGX_HTTP_MP4_STCO_ATOM].buf;
3180    stco_atom = (ngx_mp4_stco_atom_t *) atom->pos;
3181
3182    ngx_mp4_set_32value(stco_atom->size, atom_size);
3183    ngx_mp4_set_32value(stco_atom->entries, entries);
3184
3185    return NGX_OK;
3186}
3187
3188
3189static void
3190ngx_http_mp4_adjust_stco_atom(ngx_http_mp4_file_t *mp4,
3191    ngx_http_mp4_trak_t *trak, int32_t adjustment)
3192{
3193    uint32_t    offset, *entry, *end;
3194    ngx_buf_t  *data;
3195
3196    /*
3197     * moov.trak.mdia.minf.stbl.stco adjustment requires
3198     * minimal start offset of all traks and new moov atom size
3199     */
3200
3201    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3202                   "mp4 stco atom adjustment");
3203
3204    data = trak->out[NGX_HTTP_MP4_STCO_DATA].buf;
3205    entry = (uint32_t *) data->pos;
3206    end = (uint32_t *) data->last;
3207
3208    while (entry < end) {
3209        offset = ngx_mp4_get_32value(entry);
3210        offset += adjustment;
3211        ngx_mp4_set_32value(entry, offset);
3212        entry++;
3213    }
3214}
3215
3216
3217typedef struct {
3218    u_char    size[4];
3219    u_char    name[4];
3220    u_char    version[1];
3221    u_char    flags[3];
3222    u_char    entries[4];
3223} ngx_mp4_co64_atom_t;
3224
3225
3226static ngx_int_t
3227ngx_http_mp4_read_co64_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
3228{
3229    u_char               *atom_header, *atom_table, *atom_end;
3230    uint32_t              entries;
3231    ngx_buf_t            *atom, *data;
3232    ngx_mp4_co64_atom_t  *co64_atom;
3233    ngx_http_mp4_trak_t  *trak;
3234
3235    /* chunk offsets atom */
3236
3237    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 co64 atom");
3238
3239    atom_header = ngx_mp4_atom_header(mp4);
3240    co64_atom = (ngx_mp4_co64_atom_t *) atom_header;
3241    ngx_mp4_set_atom_name(co64_atom, 'c', 'o', '6', '4');
3242
3243    if (ngx_mp4_atom_data_size(ngx_mp4_co64_atom_t) > atom_data_size) {
3244        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3245                      "\"%s\" mp4 co64 atom too small", mp4->file.name.data);
3246        return NGX_ERROR;
3247    }
3248
3249    entries = ngx_mp4_get_32value(co64_atom->entries);
3250
3251    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "chunks:%uD", entries);
3252
3253    if (ngx_mp4_atom_data_size(ngx_mp4_co64_atom_t)
3254        + entries * sizeof(uint64_t) > atom_data_size)
3255    {
3256        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3257                      "\"%s\" mp4 co64 atom too small", mp4->file.name.data);
3258        return NGX_ERROR;
3259    }
3260
3261    atom_table = atom_header + sizeof(ngx_mp4_co64_atom_t);
3262    atom_end = atom_table + entries * sizeof(uint64_t);
3263
3264    trak = ngx_mp4_last_trak(mp4);
3265    trak->chunks = entries;
3266
3267    atom = &trak->co64_atom_buf;
3268    atom->temporary = 1;
3269    atom->pos = atom_header;
3270    atom->last = atom_table;
3271
3272    data = &trak->co64_data_buf;
3273    data->temporary = 1;
3274    data->pos = atom_table;
3275    data->last = atom_end;
3276
3277    trak->out[NGX_HTTP_MP4_CO64_ATOM].buf = atom;
3278    trak->out[NGX_HTTP_MP4_CO64_DATA].buf = data;
3279
3280    ngx_mp4_atom_next(mp4, atom_data_size);
3281
3282    return NGX_OK;
3283}
3284
3285
3286static ngx_int_t
3287ngx_http_mp4_update_co64_atom(ngx_http_mp4_file_t *mp4,
3288    ngx_http_mp4_trak_t *trak)
3289{
3290    size_t                atom_size;
3291    uint64_t              entries;
3292    ngx_buf_t            *atom, *data;
3293    ngx_mp4_co64_atom_t  *co64_atom;
3294
3295    /*
3296     * mdia.minf.stbl.co64 updating requires trak->start_chunk
3297     * from mdia.minf.stbl.stsc which depends on value from mdia.mdhd
3298     * atom which may reside after mdia.minf
3299     */
3300
3301    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3302                   "mp4 co64 atom update");
3303
3304    data = trak->out[NGX_HTTP_MP4_CO64_DATA].buf;
3305
3306    if (data == NULL) {
3307        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3308                      "no mp4 co64 atoms were found in \"%s\"",
3309                      mp4->file.name.data);
3310        return NGX_ERROR;
3311    }
3312
3313    if (trak->start_chunk > trak->chunks) {
3314        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3315                      "start time is out mp4 co64 chunks in \"%s\"",
3316                      mp4->file.name.data);
3317        return NGX_ERROR;
3318    }
3319
3320    data->pos += trak->start_chunk * sizeof(uint64_t);
3321
3322    trak->start_offset = ngx_mp4_get_64value(data->pos);
3323    trak->start_offset += trak->start_chunk_samples_size;
3324    ngx_mp4_set_64value(data->pos, trak->start_offset);
3325
3326    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3327                   "start chunk offset:%uL", trak->start_offset);
3328
3329    if (mp4->length) {
3330
3331        if (trak->end_chunk > trak->chunks) {
3332            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3333                          "end time is out mp4 co64 chunks in \"%s\"",
3334                          mp4->file.name.data);
3335            return NGX_ERROR;
3336        }
3337
3338        entries = trak->end_chunk - trak->start_chunk;
3339        data->last = data->pos + entries * sizeof(uint64_t);
3340
3341        if (entries) {
3342            trak->end_offset =
3343                            ngx_mp4_get_64value(data->last - sizeof(uint64_t));
3344            trak->end_offset += trak->end_chunk_samples_size;
3345
3346            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3347                           "end chunk offset:%O", trak->end_offset);
3348        }
3349
3350    } else {
3351        entries = trak->chunks - trak->start_chunk;
3352        trak->end_offset = mp4->mdat_data.buf->file_last;
3353    }
3354
3355    if (entries == 0) {
3356        trak->start_offset = mp4->end;
3357        trak->end_offset = 0;
3358    }
3359
3360    atom_size = sizeof(ngx_mp4_co64_atom_t) + (data->last - data->pos);
3361    trak->size += atom_size;
3362
3363    atom = trak->out[NGX_HTTP_MP4_CO64_ATOM].buf;
3364    co64_atom = (ngx_mp4_co64_atom_t *) atom->pos;
3365
3366    ngx_mp4_set_32value(co64_atom->size, atom_size);
3367    ngx_mp4_set_32value(co64_atom->entries, entries);
3368
3369    return NGX_OK;
3370}
3371
3372
3373static void
3374ngx_http_mp4_adjust_co64_atom(ngx_http_mp4_file_t *mp4,
3375    ngx_http_mp4_trak_t *trak, off_t adjustment)
3376{
3377    uint64_t    offset, *entry, *end;
3378    ngx_buf_t  *data;
3379
3380    /*
3381     * moov.trak.mdia.minf.stbl.co64 adjustment requires
3382     * minimal start offset of all traks and new moov atom size
3383     */
3384
3385    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3386                   "mp4 co64 atom adjustment");
3387
3388    data = trak->out[NGX_HTTP_MP4_CO64_DATA].buf;
3389    entry = (uint64_t *) data->pos;
3390    end = (uint64_t *) data->last;
3391
3392    while (entry < end) {
3393        offset = ngx_mp4_get_64value(entry);
3394        offset += adjustment;
3395        ngx_mp4_set_64value(entry, offset);
3396        entry++;
3397    }
3398}
3399
3400
3401static char *
3402ngx_http_mp4(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
3403{
3404    ngx_http_core_loc_conf_t  *clcf;
3405
3406    clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
3407    clcf->handler = ngx_http_mp4_handler;
3408
3409    return NGX_CONF_OK;
3410}
3411
3412
3413static void *
3414ngx_http_mp4_create_conf(ngx_conf_t *cf)
3415{
3416    ngx_http_mp4_conf_t  *conf;
3417
3418    conf = ngx_palloc(cf->pool, sizeof(ngx_http_mp4_conf_t));
3419    if (conf == NULL) {
3420        return NULL;
3421    }
3422
3423    conf->buffer_size = NGX_CONF_UNSET_SIZE;
3424    conf->max_buffer_size = NGX_CONF_UNSET_SIZE;
3425
3426    return conf;
3427}
3428
3429
3430static char *
3431ngx_http_mp4_merge_conf(ngx_conf_t *cf, void *parent, void *child)
3432{
3433    ngx_http_mp4_conf_t *prev = parent;
3434    ngx_http_mp4_conf_t *conf = child;
3435
3436    ngx_conf_merge_size_value(conf->buffer_size, prev->buffer_size, 512 * 1024);
3437    ngx_conf_merge_size_value(conf->max_buffer_size, prev->max_buffer_size,
3438                              10 * 1024 * 1024);
3439
3440    return NGX_CONF_OK;
3441}
Note: See TracBrowser for help on using the repository browser.