source: nginx/src/http/modules/ngx_http_mp4_module.c @ 5622:a23c35496c2f

Last change on this file since 5622:a23c35496c2f was 5622:a23c35496c2f, checked in by Sergey Kandaurov <pluknet@…>, 3 years ago

Mp4: skipped empty stss atom table in output.

The atom may have no data entries after cropping.

This fixes "zero size buf in output" alerts.

File size: 99.6 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    if (trak->sync_samples_entries) {
2277        entry = (uint32_t *) data->pos;
2278        end = (uint32_t *) data->last;
2279
2280        start_sample = trak->start_sample;
2281
2282        while (entry < end) {
2283            sample = ngx_mp4_get_32value(entry);
2284            sample -= start_sample;
2285            ngx_mp4_set_32value(entry, sample);
2286            entry++;
2287        }
2288
2289    } else {
2290        trak->out[NGX_HTTP_MP4_STSS_DATA].buf = NULL;
2291    }
2292
2293    atom_size = sizeof(ngx_http_mp4_stss_atom_t) + (data->last - data->pos);
2294    trak->size += atom_size;
2295
2296    atom = trak->out[NGX_HTTP_MP4_STSS_ATOM].buf;
2297    stss_atom = (ngx_http_mp4_stss_atom_t *) atom->pos;
2298
2299    ngx_mp4_set_32value(stss_atom->size, atom_size);
2300    ngx_mp4_set_32value(stss_atom->entries, trak->sync_samples_entries);
2301
2302    return NGX_OK;
2303}
2304
2305
2306static void
2307ngx_http_mp4_crop_stss_data(ngx_http_mp4_file_t *mp4,
2308    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2309{
2310    uint32_t     sample, start_sample, *entry, *end;
2311    ngx_buf_t   *data;
2312    ngx_uint_t   entries;
2313
2314    /* sync samples starts from 1 */
2315
2316    if (start) {
2317        start_sample = trak->start_sample + 1;
2318
2319    } else if (mp4->length) {
2320        start_sample = trak->end_sample + 1;
2321
2322    } else {
2323        return;
2324    }
2325
2326    data = trak->out[NGX_HTTP_MP4_STSS_DATA].buf;
2327
2328    entries = trak->sync_samples_entries;
2329    entry = (uint32_t *) data->pos;
2330    end = (uint32_t *) data->last;
2331
2332    while (entry < end) {
2333        sample = ngx_mp4_get_32value(entry);
2334
2335        ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2336                       "start:%uD, sync:%uD", start_sample, sample);
2337
2338        if (sample >= start_sample) {
2339            goto found;
2340        }
2341
2342        entries--;
2343        entry++;
2344    }
2345
2346    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2347                   "start sample is out of mp4 stss atom");
2348
2349found:
2350
2351    if (start) {
2352        data->pos = (u_char *) entry;
2353        trak->sync_samples_entries = entries;
2354
2355    } else {
2356        data->last = (u_char *) entry;
2357        trak->sync_samples_entries -= entries;
2358    }
2359}
2360
2361
2362typedef struct {
2363    u_char    size[4];
2364    u_char    name[4];
2365    u_char    version[1];
2366    u_char    flags[3];
2367    u_char    entries[4];
2368} ngx_mp4_ctts_atom_t;
2369
2370typedef struct {
2371    u_char    count[4];
2372    u_char    offset[4];
2373} ngx_mp4_ctts_entry_t;
2374
2375
2376static ngx_int_t
2377ngx_http_mp4_read_ctts_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2378{
2379    u_char               *atom_header, *atom_table, *atom_end;
2380    uint32_t              entries;
2381    ngx_buf_t            *atom, *data;
2382    ngx_mp4_ctts_atom_t  *ctts_atom;
2383    ngx_http_mp4_trak_t  *trak;
2384
2385    /* composition offsets atom */
2386
2387    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 ctts atom");
2388
2389    atom_header = ngx_mp4_atom_header(mp4);
2390    ctts_atom = (ngx_mp4_ctts_atom_t *) atom_header;
2391    ngx_mp4_set_atom_name(ctts_atom, 'c', 't', 't', 's');
2392
2393    if (ngx_mp4_atom_data_size(ngx_mp4_ctts_atom_t) > atom_data_size) {
2394        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2395                      "\"%s\" mp4 ctts atom too small", mp4->file.name.data);
2396        return NGX_ERROR;
2397    }
2398
2399    entries = ngx_mp4_get_32value(ctts_atom->entries);
2400
2401    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2402                   "composition offset entries:%uD", entries);
2403
2404    trak = ngx_mp4_last_trak(mp4);
2405    trak->composition_offset_entries = entries;
2406
2407    atom_table = atom_header + sizeof(ngx_mp4_ctts_atom_t);
2408
2409    atom = &trak->ctts_atom_buf;
2410    atom->temporary = 1;
2411    atom->pos = atom_header;
2412    atom->last = atom_table;
2413
2414    if (ngx_mp4_atom_data_size(ngx_mp4_ctts_atom_t)
2415        + entries * sizeof(ngx_mp4_ctts_entry_t) > atom_data_size)
2416    {
2417        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2418                      "\"%s\" mp4 ctts atom too small", mp4->file.name.data);
2419        return NGX_ERROR;
2420    }
2421
2422    atom_end = atom_table + entries * sizeof(ngx_mp4_ctts_entry_t);
2423
2424    data = &trak->ctts_data_buf;
2425    data->temporary = 1;
2426    data->pos = atom_table;
2427    data->last = atom_end;
2428
2429    trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf = atom;
2430    trak->out[NGX_HTTP_MP4_CTTS_DATA].buf = data;
2431
2432    ngx_mp4_atom_next(mp4, atom_data_size);
2433
2434    return NGX_OK;
2435}
2436
2437
2438static void
2439ngx_http_mp4_update_ctts_atom(ngx_http_mp4_file_t *mp4,
2440    ngx_http_mp4_trak_t *trak)
2441{
2442    size_t                atom_size;
2443    ngx_buf_t            *atom, *data;
2444    ngx_mp4_ctts_atom_t  *ctts_atom;
2445
2446    /*
2447     * mdia.minf.stbl.ctts updating requires trak->start_sample
2448     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2449     * atom which may reside after mdia.minf
2450     */
2451
2452    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2453                   "mp4 ctts atom update");
2454
2455    data = trak->out[NGX_HTTP_MP4_CTTS_DATA].buf;
2456
2457    if (data == NULL) {
2458        return;
2459    }
2460
2461    ngx_http_mp4_crop_ctts_data(mp4, trak, 1);
2462    ngx_http_mp4_crop_ctts_data(mp4, trak, 0);
2463
2464    if (trak->composition_offset_entries == 0) {
2465        trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf = NULL;
2466        trak->out[NGX_HTTP_MP4_CTTS_DATA].buf = NULL;
2467        return;
2468    }
2469
2470    atom_size = sizeof(ngx_mp4_ctts_atom_t) + (data->last - data->pos);
2471    trak->size += atom_size;
2472
2473    atom = trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf;
2474    ctts_atom = (ngx_mp4_ctts_atom_t *) atom->pos;
2475
2476    ngx_mp4_set_32value(ctts_atom->size, atom_size);
2477    ngx_mp4_set_32value(ctts_atom->entries, trak->composition_offset_entries);
2478
2479    return;
2480}
2481
2482
2483static void
2484ngx_http_mp4_crop_ctts_data(ngx_http_mp4_file_t *mp4,
2485    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2486{
2487    uint32_t               count, start_sample, rest;
2488    ngx_buf_t             *data;
2489    ngx_uint_t             entries;
2490    ngx_mp4_ctts_entry_t  *entry, *end;
2491
2492    /* sync samples starts from 1 */
2493
2494    if (start) {
2495        start_sample = trak->start_sample + 1;
2496
2497    } else if (mp4->length) {
2498        start_sample = trak->end_sample - trak->start_sample + 1;
2499
2500    } else {
2501        return;
2502    }
2503
2504    data = trak->out[NGX_HTTP_MP4_CTTS_DATA].buf;
2505
2506    entries = trak->composition_offset_entries;
2507    entry = (ngx_mp4_ctts_entry_t *) data->pos;
2508    end = (ngx_mp4_ctts_entry_t *) data->last;
2509
2510    while (entry < end) {
2511        count = ngx_mp4_get_32value(entry->count);
2512
2513        ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2514                       "start:%uD, count:%uD, offset:%uD",
2515                       start_sample, count, ngx_mp4_get_32value(entry->offset));
2516
2517         if (start_sample <= count) {
2518             rest = start_sample - 1;
2519             goto found;
2520         }
2521
2522         start_sample -= count;
2523         entries--;
2524         entry++;
2525    }
2526
2527    if (start) {
2528        data->pos = (u_char *) end;
2529        trak->composition_offset_entries = 0;
2530    }
2531
2532    return;
2533
2534found:
2535
2536    if (start) {
2537        ngx_mp4_set_32value(entry->count, count - rest);
2538        data->pos = (u_char *) entry;
2539        trak->composition_offset_entries = entries;
2540
2541    } else {
2542        ngx_mp4_set_32value(entry->count, rest);
2543        data->last = (u_char *) (entry + 1);
2544        trak->composition_offset_entries -= entries - 1;
2545    }
2546}
2547
2548
2549typedef struct {
2550    u_char    size[4];
2551    u_char    name[4];
2552    u_char    version[1];
2553    u_char    flags[3];
2554    u_char    entries[4];
2555} ngx_mp4_stsc_atom_t;
2556
2557
2558static ngx_int_t
2559ngx_http_mp4_read_stsc_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2560{
2561    u_char               *atom_header, *atom_table, *atom_end;
2562    uint32_t              entries;
2563    ngx_buf_t            *atom, *data;
2564    ngx_mp4_stsc_atom_t  *stsc_atom;
2565    ngx_http_mp4_trak_t  *trak;
2566
2567    /* sample-to-chunk atom */
2568
2569    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsc atom");
2570
2571    atom_header = ngx_mp4_atom_header(mp4);
2572    stsc_atom = (ngx_mp4_stsc_atom_t *) atom_header;
2573    ngx_mp4_set_atom_name(stsc_atom, 's', 't', 's', 'c');
2574
2575    if (ngx_mp4_atom_data_size(ngx_mp4_stsc_atom_t) > atom_data_size) {
2576        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2577                      "\"%s\" mp4 stsc atom too small", mp4->file.name.data);
2578        return NGX_ERROR;
2579    }
2580
2581    entries = ngx_mp4_get_32value(stsc_atom->entries);
2582
2583    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2584                   "sample-to-chunk entries:%uD", entries);
2585
2586    if (ngx_mp4_atom_data_size(ngx_mp4_stsc_atom_t)
2587        + entries * sizeof(ngx_mp4_stsc_entry_t) > atom_data_size)
2588    {
2589        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2590                      "\"%s\" mp4 stsc atom too small", mp4->file.name.data);
2591        return NGX_ERROR;
2592    }
2593
2594    atom_table = atom_header + sizeof(ngx_mp4_stsc_atom_t);
2595    atom_end = atom_table + entries * sizeof(ngx_mp4_stsc_entry_t);
2596
2597    trak = ngx_mp4_last_trak(mp4);
2598    trak->sample_to_chunk_entries = entries;
2599
2600    atom = &trak->stsc_atom_buf;
2601    atom->temporary = 1;
2602    atom->pos = atom_header;
2603    atom->last = atom_table;
2604
2605    data = &trak->stsc_data_buf;
2606    data->temporary = 1;
2607    data->pos = atom_table;
2608    data->last = atom_end;
2609
2610    trak->out[NGX_HTTP_MP4_STSC_ATOM].buf = atom;
2611    trak->out[NGX_HTTP_MP4_STSC_DATA].buf = data;
2612
2613    ngx_mp4_atom_next(mp4, atom_data_size);
2614
2615    return NGX_OK;
2616}
2617
2618
2619static ngx_int_t
2620ngx_http_mp4_update_stsc_atom(ngx_http_mp4_file_t *mp4,
2621    ngx_http_mp4_trak_t *trak)
2622{
2623    size_t                 atom_size;
2624    uint32_t               chunk;
2625    ngx_buf_t             *atom, *data;
2626    ngx_mp4_stsc_atom_t   *stsc_atom;
2627    ngx_mp4_stsc_entry_t  *entry, *end;
2628
2629    /*
2630     * mdia.minf.stbl.stsc updating requires trak->start_sample
2631     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2632     * atom which may reside after mdia.minf
2633     */
2634
2635    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2636                   "mp4 stsc atom update");
2637
2638    data = trak->out[NGX_HTTP_MP4_STSC_DATA].buf;
2639
2640    if (data == NULL) {
2641        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2642                      "no mp4 stsc atoms were found in \"%s\"",
2643                      mp4->file.name.data);
2644        return NGX_ERROR;
2645    }
2646
2647    if (trak->sample_to_chunk_entries == 0) {
2648        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2649                      "zero number of entries in stsc atom in \"%s\"",
2650                      mp4->file.name.data);
2651        return NGX_ERROR;
2652    }
2653
2654    if (ngx_http_mp4_crop_stsc_data(mp4, trak, 1) != NGX_OK) {
2655        return NGX_ERROR;
2656    }
2657
2658    if (ngx_http_mp4_crop_stsc_data(mp4, trak, 0) != NGX_OK) {
2659        return NGX_ERROR;
2660    }
2661
2662    entry = (ngx_mp4_stsc_entry_t *) data->pos;
2663    end = (ngx_mp4_stsc_entry_t *) data->last;
2664
2665    while (entry < end) {
2666        chunk = ngx_mp4_get_32value(entry->chunk);
2667        chunk -= trak->start_chunk;
2668        ngx_mp4_set_32value(entry->chunk, chunk);
2669        entry++;
2670    }
2671
2672    atom_size = sizeof(ngx_mp4_stsc_atom_t)
2673                + trak->sample_to_chunk_entries * sizeof(ngx_mp4_stsc_entry_t);
2674
2675    trak->size += atom_size;
2676
2677    atom = trak->out[NGX_HTTP_MP4_STSC_ATOM].buf;
2678    stsc_atom = (ngx_mp4_stsc_atom_t *) atom->pos;
2679
2680    ngx_mp4_set_32value(stsc_atom->size, atom_size);
2681    ngx_mp4_set_32value(stsc_atom->entries, trak->sample_to_chunk_entries);
2682
2683    return NGX_OK;
2684}
2685
2686
2687static ngx_int_t
2688ngx_http_mp4_crop_stsc_data(ngx_http_mp4_file_t *mp4,
2689    ngx_http_mp4_trak_t *trak, ngx_uint_t start)
2690{
2691    uint32_t               start_sample, chunk, samples, id, next_chunk, n,
2692                           prev_samples;
2693    ngx_buf_t             *data, *buf;
2694    ngx_uint_t             entries, target_chunk, chunk_samples;
2695    ngx_mp4_stsc_entry_t  *entry, *end, *first;
2696
2697    entries = trak->sample_to_chunk_entries - 1;
2698
2699    if (start) {
2700        start_sample = (uint32_t) trak->start_sample;
2701
2702    } else if (mp4->length) {
2703        start_sample = (uint32_t) (trak->end_sample - trak->start_sample);
2704
2705        data = trak->out[NGX_HTTP_MP4_STSC_START].buf;
2706
2707        if (data) {
2708            entry = (ngx_mp4_stsc_entry_t *) data->pos;
2709            samples = ngx_mp4_get_32value(entry->samples);
2710            entries--;
2711
2712            if (samples > start_sample) {
2713                samples = start_sample;
2714                ngx_mp4_set_32value(entry->samples, samples);
2715            }
2716
2717            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2718                           "mp4 stsc using %uD start samples", samples);
2719
2720            start_sample -= samples;
2721        }
2722
2723    } else {
2724        return NGX_OK;
2725    }
2726
2727    data = trak->out[NGX_HTTP_MP4_STSC_DATA].buf;
2728
2729    entry = (ngx_mp4_stsc_entry_t *) data->pos;
2730    end = (ngx_mp4_stsc_entry_t *) data->last;
2731
2732    chunk = ngx_mp4_get_32value(entry->chunk);
2733    samples = ngx_mp4_get_32value(entry->samples);
2734    id = ngx_mp4_get_32value(entry->id);
2735    prev_samples = 0;
2736    entry++;
2737
2738    while (entry < end) {
2739
2740        next_chunk = ngx_mp4_get_32value(entry->chunk);
2741
2742        ngx_log_debug5(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2743                       "start_sample:%uD, chunk:%uD, chunks:%uD, "
2744                       "samples:%uD, id:%uD",
2745                       start_sample, chunk, next_chunk - chunk, samples, id);
2746
2747        n = (next_chunk - chunk) * samples;
2748
2749        if (start_sample < n) {
2750            goto found;
2751        }
2752
2753        start_sample -= n;
2754
2755        prev_samples = samples;
2756        chunk = next_chunk;
2757        samples = ngx_mp4_get_32value(entry->samples);
2758        id = ngx_mp4_get_32value(entry->id);
2759        entries--;
2760        entry++;
2761    }
2762
2763    next_chunk = trak->chunks + 1;
2764
2765    ngx_log_debug4(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2766                   "start_sample:%uD, chunk:%uD, chunks:%uD, samples:%uD",
2767                   start_sample, chunk, next_chunk - chunk, samples);
2768
2769    n = (next_chunk - chunk) * samples;
2770
2771    if (start_sample > n) {
2772        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2773                      "start time is out mp4 stsc chunks in \"%s\"",
2774                      mp4->file.name.data);
2775        return NGX_ERROR;
2776    }
2777
2778found:
2779
2780    entries++;
2781    entry--;
2782
2783    if (samples == 0) {
2784        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2785                      "zero number of samples in \"%s\"",
2786                      mp4->file.name.data);
2787        return NGX_ERROR;
2788    }
2789
2790    target_chunk = chunk - 1;
2791    target_chunk += start_sample / samples;
2792    chunk_samples = start_sample % samples;
2793
2794    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2795                   "start chunk:%ui, samples:%uD",
2796                   target_chunk, chunk_samples);
2797
2798    if (start) {
2799        data->pos = (u_char *) entry;
2800
2801        trak->sample_to_chunk_entries = entries;
2802        trak->start_chunk = target_chunk;
2803        trak->start_chunk_samples = chunk_samples;
2804
2805        ngx_mp4_set_32value(entry->chunk, trak->start_chunk + 1);
2806
2807        samples -= chunk_samples;
2808
2809    } else {
2810        if (start_sample) {
2811            data->last = (u_char *) (entry + 1);
2812            trak->sample_to_chunk_entries -= entries - 1;
2813            trak->end_chunk_samples = samples;
2814
2815        } else {
2816            data->last = (u_char *) entry;
2817            trak->sample_to_chunk_entries -= entries;
2818            trak->end_chunk_samples = prev_samples;
2819        }
2820
2821        if (chunk_samples) {
2822            trak->end_chunk = target_chunk + 1;
2823            trak->end_chunk_samples = chunk_samples;
2824
2825        } else {
2826            trak->end_chunk = target_chunk;
2827        }
2828
2829        samples = chunk_samples;
2830        next_chunk = chunk + 1;
2831    }
2832
2833    if (chunk_samples && next_chunk - target_chunk == 2) {
2834
2835        ngx_mp4_set_32value(entry->samples, samples);
2836
2837    } else if (chunk_samples && start) {
2838
2839        first = &trak->stsc_start_chunk_entry;
2840        ngx_mp4_set_32value(first->chunk, 1);
2841        ngx_mp4_set_32value(first->samples, samples);
2842        ngx_mp4_set_32value(first->id, id);
2843
2844        buf = &trak->stsc_start_chunk_buf;
2845        buf->temporary = 1;
2846        buf->pos = (u_char *) first;
2847        buf->last = (u_char *) first + sizeof(ngx_mp4_stsc_entry_t);
2848
2849        trak->out[NGX_HTTP_MP4_STSC_START].buf = buf;
2850
2851        ngx_mp4_set_32value(entry->chunk, trak->start_chunk + 2);
2852
2853        trak->sample_to_chunk_entries++;
2854
2855    } else if (chunk_samples) {
2856
2857        first = &trak->stsc_end_chunk_entry;
2858        ngx_mp4_set_32value(first->chunk, trak->end_chunk - trak->start_chunk);
2859        ngx_mp4_set_32value(first->samples, samples);
2860        ngx_mp4_set_32value(first->id, id);
2861
2862        buf = &trak->stsc_end_chunk_buf;
2863        buf->temporary = 1;
2864        buf->pos = (u_char *) first;
2865        buf->last = (u_char *) first + sizeof(ngx_mp4_stsc_entry_t);
2866
2867        trak->out[NGX_HTTP_MP4_STSC_END].buf = buf;
2868
2869        trak->sample_to_chunk_entries++;
2870    }
2871
2872    return NGX_OK;
2873}
2874
2875
2876typedef struct {
2877    u_char    size[4];
2878    u_char    name[4];
2879    u_char    version[1];
2880    u_char    flags[3];
2881    u_char    uniform_size[4];
2882    u_char    entries[4];
2883} ngx_mp4_stsz_atom_t;
2884
2885
2886static ngx_int_t
2887ngx_http_mp4_read_stsz_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
2888{
2889    u_char               *atom_header, *atom_table, *atom_end;
2890    size_t                atom_size;
2891    uint32_t              entries, size;
2892    ngx_buf_t            *atom, *data;
2893    ngx_mp4_stsz_atom_t  *stsz_atom;
2894    ngx_http_mp4_trak_t  *trak;
2895
2896    /* sample sizes atom */
2897
2898    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsz atom");
2899
2900    atom_header = ngx_mp4_atom_header(mp4);
2901    stsz_atom = (ngx_mp4_stsz_atom_t *) atom_header;
2902    ngx_mp4_set_atom_name(stsz_atom, 's', 't', 's', 'z');
2903
2904    if (ngx_mp4_atom_data_size(ngx_mp4_stsz_atom_t) > atom_data_size) {
2905        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2906                      "\"%s\" mp4 stsz atom too small", mp4->file.name.data);
2907        return NGX_ERROR;
2908    }
2909
2910    size = ngx_mp4_get_32value(stsz_atom->uniform_size);
2911    entries = ngx_mp4_get_32value(stsz_atom->entries);
2912
2913    ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2914                   "sample uniform size:%uD, entries:%uD", size, entries);
2915
2916    trak = ngx_mp4_last_trak(mp4);
2917    trak->sample_sizes_entries = entries;
2918
2919    atom_table = atom_header + sizeof(ngx_mp4_stsz_atom_t);
2920
2921    atom = &trak->stsz_atom_buf;
2922    atom->temporary = 1;
2923    atom->pos = atom_header;
2924    atom->last = atom_table;
2925
2926    trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf = atom;
2927
2928    if (size == 0) {
2929        if (ngx_mp4_atom_data_size(ngx_mp4_stsz_atom_t)
2930            + entries * sizeof(uint32_t) > atom_data_size)
2931        {
2932            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2933                          "\"%s\" mp4 stsz atom too small",
2934                          mp4->file.name.data);
2935            return NGX_ERROR;
2936        }
2937
2938        atom_end = atom_table + entries * sizeof(uint32_t);
2939
2940        data = &trak->stsz_data_buf;
2941        data->temporary = 1;
2942        data->pos = atom_table;
2943        data->last = atom_end;
2944
2945        trak->out[NGX_HTTP_MP4_STSZ_DATA].buf = data;
2946
2947    } else {
2948        /* if size != 0 then all samples are the same size */
2949        /* TODO : chunk samples */
2950        atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
2951        ngx_mp4_set_32value(atom_header, atom_size);
2952        trak->size += atom_size;
2953    }
2954
2955    ngx_mp4_atom_next(mp4, atom_data_size);
2956
2957    return NGX_OK;
2958}
2959
2960
2961static ngx_int_t
2962ngx_http_mp4_update_stsz_atom(ngx_http_mp4_file_t *mp4,
2963    ngx_http_mp4_trak_t *trak)
2964{
2965    size_t                atom_size;
2966    uint32_t             *pos, *end, entries;
2967    ngx_buf_t            *atom, *data;
2968    ngx_mp4_stsz_atom_t  *stsz_atom;
2969
2970    /*
2971     * mdia.minf.stbl.stsz updating requires trak->start_sample
2972     * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
2973     * atom which may reside after mdia.minf
2974     */
2975
2976    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
2977                   "mp4 stsz atom update");
2978
2979    data = trak->out[NGX_HTTP_MP4_STSZ_DATA].buf;
2980
2981    if (data) {
2982        entries = trak->sample_sizes_entries;
2983
2984        if (trak->start_sample > entries) {
2985            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
2986                          "start time is out mp4 stsz samples in \"%s\"",
2987                          mp4->file.name.data);
2988            return NGX_ERROR;
2989        }
2990
2991        entries -= trak->start_sample;
2992        data->pos += trak->start_sample * sizeof(uint32_t);
2993        end = (uint32_t *) data->pos;
2994
2995        for (pos = end - trak->start_chunk_samples; pos < end; pos++) {
2996            trak->start_chunk_samples_size += ngx_mp4_get_32value(pos);
2997        }
2998
2999        ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3000                       "chunk samples sizes:%uL",
3001                       trak->start_chunk_samples_size);
3002
3003        if (mp4->length) {
3004            if (trak->end_sample - trak->start_sample > entries) {
3005                ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3006                              "end time is out mp4 stsz samples in \"%s\"",
3007                              mp4->file.name.data);
3008                return NGX_ERROR;
3009            }
3010
3011            entries = trak->end_sample - trak->start_sample;
3012            data->last = data->pos + entries * sizeof(uint32_t);
3013            end = (uint32_t *) data->last;
3014
3015            for (pos = end - trak->end_chunk_samples; pos < end; pos++) {
3016                trak->end_chunk_samples_size += ngx_mp4_get_32value(pos);
3017            }
3018
3019            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3020                           "mp4 stsz end_chunk_samples_size:%uL",
3021                           trak->end_chunk_samples_size);
3022        }
3023
3024        atom_size = sizeof(ngx_mp4_stsz_atom_t) + (data->last - data->pos);
3025        trak->size += atom_size;
3026
3027        atom = trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf;
3028        stsz_atom = (ngx_mp4_stsz_atom_t *) atom->pos;
3029
3030        ngx_mp4_set_32value(stsz_atom->size, atom_size);
3031        ngx_mp4_set_32value(stsz_atom->entries, entries);
3032    }
3033
3034    return NGX_OK;
3035}
3036
3037
3038typedef struct {
3039    u_char    size[4];
3040    u_char    name[4];
3041    u_char    version[1];
3042    u_char    flags[3];
3043    u_char    entries[4];
3044} ngx_mp4_stco_atom_t;
3045
3046
3047static ngx_int_t
3048ngx_http_mp4_read_stco_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
3049{
3050    u_char               *atom_header, *atom_table, *atom_end;
3051    uint32_t              entries;
3052    ngx_buf_t            *atom, *data;
3053    ngx_mp4_stco_atom_t  *stco_atom;
3054    ngx_http_mp4_trak_t  *trak;
3055
3056    /* chunk offsets atom */
3057
3058    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stco atom");
3059
3060    atom_header = ngx_mp4_atom_header(mp4);
3061    stco_atom = (ngx_mp4_stco_atom_t *) atom_header;
3062    ngx_mp4_set_atom_name(stco_atom, 's', 't', 'c', 'o');
3063
3064    if (ngx_mp4_atom_data_size(ngx_mp4_stco_atom_t) > atom_data_size) {
3065        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3066                      "\"%s\" mp4 stco atom too small", mp4->file.name.data);
3067        return NGX_ERROR;
3068    }
3069
3070    entries = ngx_mp4_get_32value(stco_atom->entries);
3071
3072    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "chunks:%uD", entries);
3073
3074    if (ngx_mp4_atom_data_size(ngx_mp4_stco_atom_t)
3075        + entries * sizeof(uint32_t) > atom_data_size)
3076    {
3077        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3078                      "\"%s\" mp4 stco atom too small", mp4->file.name.data);
3079        return NGX_ERROR;
3080    }
3081
3082    atom_table = atom_header + sizeof(ngx_mp4_stco_atom_t);
3083    atom_end = atom_table + entries * sizeof(uint32_t);
3084
3085    trak = ngx_mp4_last_trak(mp4);
3086    trak->chunks = entries;
3087
3088    atom = &trak->stco_atom_buf;
3089    atom->temporary = 1;
3090    atom->pos = atom_header;
3091    atom->last = atom_table;
3092
3093    data = &trak->stco_data_buf;
3094    data->temporary = 1;
3095    data->pos = atom_table;
3096    data->last = atom_end;
3097
3098    trak->out[NGX_HTTP_MP4_STCO_ATOM].buf = atom;
3099    trak->out[NGX_HTTP_MP4_STCO_DATA].buf = data;
3100
3101    ngx_mp4_atom_next(mp4, atom_data_size);
3102
3103    return NGX_OK;
3104}
3105
3106
3107static ngx_int_t
3108ngx_http_mp4_update_stco_atom(ngx_http_mp4_file_t *mp4,
3109    ngx_http_mp4_trak_t *trak)
3110{
3111    size_t                atom_size;
3112    uint32_t              entries;
3113    ngx_buf_t            *atom, *data;
3114    ngx_mp4_stco_atom_t  *stco_atom;
3115
3116    /*
3117     * mdia.minf.stbl.stco updating requires trak->start_chunk
3118     * from mdia.minf.stbl.stsc which depends on value from mdia.mdhd
3119     * atom which may reside after mdia.minf
3120     */
3121
3122    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3123                   "mp4 stco atom update");
3124
3125    data = trak->out[NGX_HTTP_MP4_STCO_DATA].buf;
3126
3127    if (data == NULL) {
3128        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3129                      "no mp4 stco atoms were found in \"%s\"",
3130                      mp4->file.name.data);
3131        return NGX_ERROR;
3132    }
3133
3134    if (trak->start_chunk > trak->chunks) {
3135        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3136                      "start time is out mp4 stco chunks in \"%s\"",
3137                      mp4->file.name.data);
3138        return NGX_ERROR;
3139    }
3140
3141    data->pos += trak->start_chunk * sizeof(uint32_t);
3142
3143    trak->start_offset = ngx_mp4_get_32value(data->pos);
3144    trak->start_offset += trak->start_chunk_samples_size;
3145    ngx_mp4_set_32value(data->pos, trak->start_offset);
3146
3147    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3148                   "start chunk offset:%uD", trak->start_offset);
3149
3150    if (mp4->length) {
3151
3152        if (trak->end_chunk > trak->chunks) {
3153            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3154                          "end time is out mp4 stco chunks in \"%s\"",
3155                          mp4->file.name.data);
3156            return NGX_ERROR;
3157        }
3158
3159        entries = trak->end_chunk - trak->start_chunk;
3160        data->last = data->pos + entries * sizeof(uint32_t);
3161
3162        if (entries) {
3163            trak->end_offset =
3164                            ngx_mp4_get_32value(data->last - sizeof(uint32_t));
3165            trak->end_offset += trak->end_chunk_samples_size;
3166
3167            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3168                           "end chunk offset:%O", trak->end_offset);
3169        }
3170
3171    } else {
3172        entries = trak->chunks - trak->start_chunk;
3173        trak->end_offset = mp4->mdat_data.buf->file_last;
3174    }
3175
3176    if (entries == 0) {
3177        trak->start_offset = mp4->end;
3178        trak->end_offset = 0;
3179    }
3180
3181    atom_size = sizeof(ngx_mp4_stco_atom_t) + (data->last - data->pos);
3182    trak->size += atom_size;
3183
3184    atom = trak->out[NGX_HTTP_MP4_STCO_ATOM].buf;
3185    stco_atom = (ngx_mp4_stco_atom_t *) atom->pos;
3186
3187    ngx_mp4_set_32value(stco_atom->size, atom_size);
3188    ngx_mp4_set_32value(stco_atom->entries, entries);
3189
3190    return NGX_OK;
3191}
3192
3193
3194static void
3195ngx_http_mp4_adjust_stco_atom(ngx_http_mp4_file_t *mp4,
3196    ngx_http_mp4_trak_t *trak, int32_t adjustment)
3197{
3198    uint32_t    offset, *entry, *end;
3199    ngx_buf_t  *data;
3200
3201    /*
3202     * moov.trak.mdia.minf.stbl.stco adjustment requires
3203     * minimal start offset of all traks and new moov atom size
3204     */
3205
3206    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3207                   "mp4 stco atom adjustment");
3208
3209    data = trak->out[NGX_HTTP_MP4_STCO_DATA].buf;
3210    entry = (uint32_t *) data->pos;
3211    end = (uint32_t *) data->last;
3212
3213    while (entry < end) {
3214        offset = ngx_mp4_get_32value(entry);
3215        offset += adjustment;
3216        ngx_mp4_set_32value(entry, offset);
3217        entry++;
3218    }
3219}
3220
3221
3222typedef struct {
3223    u_char    size[4];
3224    u_char    name[4];
3225    u_char    version[1];
3226    u_char    flags[3];
3227    u_char    entries[4];
3228} ngx_mp4_co64_atom_t;
3229
3230
3231static ngx_int_t
3232ngx_http_mp4_read_co64_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
3233{
3234    u_char               *atom_header, *atom_table, *atom_end;
3235    uint32_t              entries;
3236    ngx_buf_t            *atom, *data;
3237    ngx_mp4_co64_atom_t  *co64_atom;
3238    ngx_http_mp4_trak_t  *trak;
3239
3240    /* chunk offsets atom */
3241
3242    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 co64 atom");
3243
3244    atom_header = ngx_mp4_atom_header(mp4);
3245    co64_atom = (ngx_mp4_co64_atom_t *) atom_header;
3246    ngx_mp4_set_atom_name(co64_atom, 'c', 'o', '6', '4');
3247
3248    if (ngx_mp4_atom_data_size(ngx_mp4_co64_atom_t) > atom_data_size) {
3249        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3250                      "\"%s\" mp4 co64 atom too small", mp4->file.name.data);
3251        return NGX_ERROR;
3252    }
3253
3254    entries = ngx_mp4_get_32value(co64_atom->entries);
3255
3256    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "chunks:%uD", entries);
3257
3258    if (ngx_mp4_atom_data_size(ngx_mp4_co64_atom_t)
3259        + entries * sizeof(uint64_t) > atom_data_size)
3260    {
3261        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3262                      "\"%s\" mp4 co64 atom too small", mp4->file.name.data);
3263        return NGX_ERROR;
3264    }
3265
3266    atom_table = atom_header + sizeof(ngx_mp4_co64_atom_t);
3267    atom_end = atom_table + entries * sizeof(uint64_t);
3268
3269    trak = ngx_mp4_last_trak(mp4);
3270    trak->chunks = entries;
3271
3272    atom = &trak->co64_atom_buf;
3273    atom->temporary = 1;
3274    atom->pos = atom_header;
3275    atom->last = atom_table;
3276
3277    data = &trak->co64_data_buf;
3278    data->temporary = 1;
3279    data->pos = atom_table;
3280    data->last = atom_end;
3281
3282    trak->out[NGX_HTTP_MP4_CO64_ATOM].buf = atom;
3283    trak->out[NGX_HTTP_MP4_CO64_DATA].buf = data;
3284
3285    ngx_mp4_atom_next(mp4, atom_data_size);
3286
3287    return NGX_OK;
3288}
3289
3290
3291static ngx_int_t
3292ngx_http_mp4_update_co64_atom(ngx_http_mp4_file_t *mp4,
3293    ngx_http_mp4_trak_t *trak)
3294{
3295    size_t                atom_size;
3296    uint64_t              entries;
3297    ngx_buf_t            *atom, *data;
3298    ngx_mp4_co64_atom_t  *co64_atom;
3299
3300    /*
3301     * mdia.minf.stbl.co64 updating requires trak->start_chunk
3302     * from mdia.minf.stbl.stsc which depends on value from mdia.mdhd
3303     * atom which may reside after mdia.minf
3304     */
3305
3306    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3307                   "mp4 co64 atom update");
3308
3309    data = trak->out[NGX_HTTP_MP4_CO64_DATA].buf;
3310
3311    if (data == NULL) {
3312        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3313                      "no mp4 co64 atoms were found in \"%s\"",
3314                      mp4->file.name.data);
3315        return NGX_ERROR;
3316    }
3317
3318    if (trak->start_chunk > trak->chunks) {
3319        ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3320                      "start time is out mp4 co64 chunks in \"%s\"",
3321                      mp4->file.name.data);
3322        return NGX_ERROR;
3323    }
3324
3325    data->pos += trak->start_chunk * sizeof(uint64_t);
3326
3327    trak->start_offset = ngx_mp4_get_64value(data->pos);
3328    trak->start_offset += trak->start_chunk_samples_size;
3329    ngx_mp4_set_64value(data->pos, trak->start_offset);
3330
3331    ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3332                   "start chunk offset:%uL", trak->start_offset);
3333
3334    if (mp4->length) {
3335
3336        if (trak->end_chunk > trak->chunks) {
3337            ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
3338                          "end time is out mp4 co64 chunks in \"%s\"",
3339                          mp4->file.name.data);
3340            return NGX_ERROR;
3341        }
3342
3343        entries = trak->end_chunk - trak->start_chunk;
3344        data->last = data->pos + entries * sizeof(uint64_t);
3345
3346        if (entries) {
3347            trak->end_offset =
3348                            ngx_mp4_get_64value(data->last - sizeof(uint64_t));
3349            trak->end_offset += trak->end_chunk_samples_size;
3350
3351            ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3352                           "end chunk offset:%O", trak->end_offset);
3353        }
3354
3355    } else {
3356        entries = trak->chunks - trak->start_chunk;
3357        trak->end_offset = mp4->mdat_data.buf->file_last;
3358    }
3359
3360    if (entries == 0) {
3361        trak->start_offset = mp4->end;
3362        trak->end_offset = 0;
3363    }
3364
3365    atom_size = sizeof(ngx_mp4_co64_atom_t) + (data->last - data->pos);
3366    trak->size += atom_size;
3367
3368    atom = trak->out[NGX_HTTP_MP4_CO64_ATOM].buf;
3369    co64_atom = (ngx_mp4_co64_atom_t *) atom->pos;
3370
3371    ngx_mp4_set_32value(co64_atom->size, atom_size);
3372    ngx_mp4_set_32value(co64_atom->entries, entries);
3373
3374    return NGX_OK;
3375}
3376
3377
3378static void
3379ngx_http_mp4_adjust_co64_atom(ngx_http_mp4_file_t *mp4,
3380    ngx_http_mp4_trak_t *trak, off_t adjustment)
3381{
3382    uint64_t    offset, *entry, *end;
3383    ngx_buf_t  *data;
3384
3385    /*
3386     * moov.trak.mdia.minf.stbl.co64 adjustment requires
3387     * minimal start offset of all traks and new moov atom size
3388     */
3389
3390    ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
3391                   "mp4 co64 atom adjustment");
3392
3393    data = trak->out[NGX_HTTP_MP4_CO64_DATA].buf;
3394    entry = (uint64_t *) data->pos;
3395    end = (uint64_t *) data->last;
3396
3397    while (entry < end) {
3398        offset = ngx_mp4_get_64value(entry);
3399        offset += adjustment;
3400        ngx_mp4_set_64value(entry, offset);
3401        entry++;
3402    }
3403}
3404
3405
3406static char *
3407ngx_http_mp4(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
3408{
3409    ngx_http_core_loc_conf_t  *clcf;
3410
3411    clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
3412    clcf->handler = ngx_http_mp4_handler;
3413
3414    return NGX_CONF_OK;
3415}
3416
3417
3418static void *
3419ngx_http_mp4_create_conf(ngx_conf_t *cf)
3420{
3421    ngx_http_mp4_conf_t  *conf;
3422
3423    conf = ngx_palloc(cf->pool, sizeof(ngx_http_mp4_conf_t));
3424    if (conf == NULL) {
3425        return NULL;
3426    }
3427
3428    conf->buffer_size = NGX_CONF_UNSET_SIZE;
3429    conf->max_buffer_size = NGX_CONF_UNSET_SIZE;
3430
3431    return conf;
3432}
3433
3434
3435static char *
3436ngx_http_mp4_merge_conf(ngx_conf_t *cf, void *parent, void *child)
3437{
3438    ngx_http_mp4_conf_t *prev = parent;
3439    ngx_http_mp4_conf_t *conf = child;
3440
3441    ngx_conf_merge_size_value(conf->buffer_size, prev->buffer_size, 512 * 1024);
3442    ngx_conf_merge_size_value(conf->max_buffer_size, prev->max_buffer_size,
3443                              10 * 1024 * 1024);
3444
3445    return NGX_CONF_OK;
3446}
Note: See TracBrowser for help on using the repository browser.