src/http/modules/ngx_http_mp4_module.c - nginx source code

Global variables defined

Data types defined

Functions defined

Macros defined

Source code


  1. /*
  2. * Copyright (C) Igor Sysoev
  3. * Copyright (C) Nginx, Inc.
  4. */

  5. #include <ngx_config.h>
  6. #include <ngx_core.h>
  7. #include <ngx_http.h>


  8. #define NGX_HTTP_MP4_TRAK_ATOM     0
  9. #define NGX_HTTP_MP4_TKHD_ATOM     1
  10. #define NGX_HTTP_MP4_EDTS_ATOM     2
  11. #define NGX_HTTP_MP4_ELST_ATOM     3
  12. #define NGX_HTTP_MP4_MDIA_ATOM     4
  13. #define NGX_HTTP_MP4_MDHD_ATOM     5
  14. #define NGX_HTTP_MP4_HDLR_ATOM     6
  15. #define NGX_HTTP_MP4_MINF_ATOM     7
  16. #define NGX_HTTP_MP4_VMHD_ATOM     8
  17. #define NGX_HTTP_MP4_SMHD_ATOM     9
  18. #define NGX_HTTP_MP4_DINF_ATOM    10
  19. #define NGX_HTTP_MP4_STBL_ATOM    11
  20. #define NGX_HTTP_MP4_STSD_ATOM    12
  21. #define NGX_HTTP_MP4_STTS_ATOM    13
  22. #define NGX_HTTP_MP4_STTS_DATA    14
  23. #define NGX_HTTP_MP4_STSS_ATOM    15
  24. #define NGX_HTTP_MP4_STSS_DATA    16
  25. #define NGX_HTTP_MP4_CTTS_ATOM    17
  26. #define NGX_HTTP_MP4_CTTS_DATA    18
  27. #define NGX_HTTP_MP4_STSC_ATOM    19
  28. #define NGX_HTTP_MP4_STSC_START   20
  29. #define NGX_HTTP_MP4_STSC_DATA    21
  30. #define NGX_HTTP_MP4_STSC_END     22
  31. #define NGX_HTTP_MP4_STSZ_ATOM    23
  32. #define NGX_HTTP_MP4_STSZ_DATA    24
  33. #define NGX_HTTP_MP4_STCO_ATOM    25
  34. #define NGX_HTTP_MP4_STCO_DATA    26
  35. #define NGX_HTTP_MP4_CO64_ATOM    27
  36. #define NGX_HTTP_MP4_CO64_DATA    28

  37. #define NGX_HTTP_MP4_LAST_ATOM    NGX_HTTP_MP4_CO64_DATA


  38. typedef struct {
  39.     size_t                buffer_size;
  40.     size_t                max_buffer_size;
  41.     ngx_flag_t            start_key_frame;
  42. } ngx_http_mp4_conf_t;


  43. typedef struct {
  44.     u_char                chunk[4];
  45.     u_char                samples[4];
  46.     u_char                id[4];
  47. } ngx_mp4_stsc_entry_t;


  48. typedef struct {
  49.     u_char                size[4];
  50.     u_char                name[4];
  51. } ngx_mp4_edts_atom_t;


  52. typedef struct {
  53.     u_char                size[4];
  54.     u_char                name[4];
  55.     u_char                version[1];
  56.     u_char                flags[3];
  57.     u_char                entries[4];
  58.     u_char                duration[8];
  59.     u_char                media_time[8];
  60.     u_char                media_rate[2];
  61.     u_char                reserved[2];
  62. } ngx_mp4_elst_atom_t;


  63. typedef struct {
  64.     uint32_t              timescale;
  65.     uint32_t              time_to_sample_entries;
  66.     uint32_t              sample_to_chunk_entries;
  67.     uint32_t              sync_samples_entries;
  68.     uint32_t              composition_offset_entries;
  69.     uint32_t              sample_sizes_entries;
  70.     uint32_t              chunks;

  71.     ngx_uint_t            start_sample;
  72.     ngx_uint_t            end_sample;
  73.     ngx_uint_t            start_chunk;
  74.     ngx_uint_t            end_chunk;
  75.     ngx_uint_t            start_chunk_samples;
  76.     ngx_uint_t            end_chunk_samples;
  77.     uint64_t              start_chunk_samples_size;
  78.     uint64_t              end_chunk_samples_size;
  79.     uint64_t              duration;
  80.     uint64_t              prefix;
  81.     uint64_t              movie_duration;
  82.     off_t                 start_offset;
  83.     off_t                 end_offset;

  84.     size_t                tkhd_size;
  85.     size_t                mdhd_size;
  86.     size_t                hdlr_size;
  87.     size_t                vmhd_size;
  88.     size_t                smhd_size;
  89.     size_t                dinf_size;
  90.     size_t                size;

  91.     ngx_chain_t           out[NGX_HTTP_MP4_LAST_ATOM + 1];

  92.     ngx_buf_t             trak_atom_buf;
  93.     ngx_buf_t             tkhd_atom_buf;
  94.     ngx_buf_t             edts_atom_buf;
  95.     ngx_buf_t             elst_atom_buf;
  96.     ngx_buf_t             mdia_atom_buf;
  97.     ngx_buf_t             mdhd_atom_buf;
  98.     ngx_buf_t             hdlr_atom_buf;
  99.     ngx_buf_t             minf_atom_buf;
  100.     ngx_buf_t             vmhd_atom_buf;
  101.     ngx_buf_t             smhd_atom_buf;
  102.     ngx_buf_t             dinf_atom_buf;
  103.     ngx_buf_t             stbl_atom_buf;
  104.     ngx_buf_t             stsd_atom_buf;
  105.     ngx_buf_t             stts_atom_buf;
  106.     ngx_buf_t             stts_data_buf;
  107.     ngx_buf_t             stss_atom_buf;
  108.     ngx_buf_t             stss_data_buf;
  109.     ngx_buf_t             ctts_atom_buf;
  110.     ngx_buf_t             ctts_data_buf;
  111.     ngx_buf_t             stsc_atom_buf;
  112.     ngx_buf_t             stsc_start_chunk_buf;
  113.     ngx_buf_t             stsc_end_chunk_buf;
  114.     ngx_buf_t             stsc_data_buf;
  115.     ngx_buf_t             stsz_atom_buf;
  116.     ngx_buf_t             stsz_data_buf;
  117.     ngx_buf_t             stco_atom_buf;
  118.     ngx_buf_t             stco_data_buf;
  119.     ngx_buf_t             co64_atom_buf;
  120.     ngx_buf_t             co64_data_buf;

  121.     ngx_mp4_edts_atom_t   edts_atom;
  122.     ngx_mp4_elst_atom_t   elst_atom;
  123.     ngx_mp4_stsc_entry_t  stsc_start_chunk_entry;
  124.     ngx_mp4_stsc_entry_t  stsc_end_chunk_entry;
  125. } ngx_http_mp4_trak_t;


  126. typedef struct {
  127.     ngx_file_t            file;

  128.     u_char               *buffer;
  129.     u_char               *buffer_start;
  130.     u_char               *buffer_pos;
  131.     u_char               *buffer_end;
  132.     size_t                buffer_size;

  133.     off_t                 offset;
  134.     off_t                 end;
  135.     off_t                 content_length;
  136.     ngx_uint_t            start;
  137.     ngx_uint_t            length;
  138.     uint32_t              timescale;
  139.     ngx_http_request_t   *request;
  140.     ngx_array_t           trak;
  141.     ngx_http_mp4_trak_t   traks[2];

  142.     size_t                ftyp_size;
  143.     size_t                moov_size;

  144.     ngx_chain_t          *out;
  145.     ngx_chain_t           ftyp_atom;
  146.     ngx_chain_t           moov_atom;
  147.     ngx_chain_t           mvhd_atom;
  148.     ngx_chain_t           mdat_atom;
  149.     ngx_chain_t           mdat_data;

  150.     ngx_buf_t             ftyp_atom_buf;
  151.     ngx_buf_t             moov_atom_buf;
  152.     ngx_buf_t             mvhd_atom_buf;
  153.     ngx_buf_t             mdat_atom_buf;
  154.     ngx_buf_t             mdat_data_buf;

  155.     u_char                moov_atom_header[8];
  156.     u_char                mdat_atom_header[16];
  157. } ngx_http_mp4_file_t;


  158. typedef struct {
  159.     char                 *name;
  160.     ngx_int_t           (*handler)(ngx_http_mp4_file_t *mp4,
  161.                                    uint64_t atom_data_size);
  162. } ngx_http_mp4_atom_handler_t;


  163. #define ngx_mp4_atom_header(mp4)   (mp4->buffer_pos - 8)
  164. #define ngx_mp4_atom_data(mp4)     mp4->buffer_pos
  165. #define ngx_mp4_atom_data_size(t)  (uint64_t) (sizeof(t) - 8)


  166. #define ngx_mp4_atom_next(mp4, n)                                             \
  167.                                                                               \
  168.     if (n > (size_t) (mp4->buffer_end - mp4->buffer_pos)) {                   \
  169.         mp4->buffer_pos = mp4->buffer_end;                                    \
  170.                                                                               \
  171.     } else {                                                                  \
  172.         mp4->buffer_pos += (size_t) n;                                        \
  173.     }                                                                         \
  174.                                                                               \
  175.     mp4->offset += n


  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. #define ngx_mp4_get_16value(p)                                                \
  182.     ( ((uint16_t) ((u_char *) (p))[0] << 8)                                   \
  183.     + (           ((u_char *) (p))[1]) )

  184. #define ngx_mp4_set_16value(p, n)                                             \
  185.     ((u_char *) (p))[0] = (u_char) ((n) >> 8);                                \
  186.     ((u_char *) (p))[1] = (u_char)  (n)

  187. #define ngx_mp4_get_32value(p)                                                \
  188.     ( ((uint32_t) ((u_char *) (p))[0] << 24)                                  \
  189.     + (           ((u_char *) (p))[1] << 16)                                  \
  190.     + (           ((u_char *) (p))[2] << 8)                                   \
  191.     + (           ((u_char *) (p))[3]) )

  192. #define ngx_mp4_set_32value(p, n)                                             \
  193.     ((u_char *) (p))[0] = (u_char) ((n) >> 24);                               \
  194.     ((u_char *) (p))[1] = (u_char) ((n) >> 16);                               \
  195.     ((u_char *) (p))[2] = (u_char) ((n) >> 8);                                \
  196.     ((u_char *) (p))[3] = (u_char)  (n)

  197. #define ngx_mp4_get_64value(p)                                                \
  198.     ( ((uint64_t) ((u_char *) (p))[0] << 56)                                  \
  199.     + ((uint64_t) ((u_char *) (p))[1] << 48)                                  \
  200.     + ((uint64_t) ((u_char *) (p))[2] << 40)                                  \
  201.     + ((uint64_t) ((u_char *) (p))[3] << 32)                                  \
  202.     + ((uint64_t) ((u_char *) (p))[4] << 24)                                  \
  203.     + (           ((u_char *) (p))[5] << 16)                                  \
  204.     + (           ((u_char *) (p))[6] << 8)                                   \
  205.     + (           ((u_char *) (p))[7]) )

  206. #define ngx_mp4_set_64value(p, n)                                             \
  207.     ((u_char *) (p))[0] = (u_char) ((uint64_t) (n) >> 56);                    \
  208.     ((u_char *) (p))[1] = (u_char) ((uint64_t) (n) >> 48);                    \
  209.     ((u_char *) (p))[2] = (u_char) ((uint64_t) (n) >> 40);                    \
  210.     ((u_char *) (p))[3] = (u_char) ((uint64_t) (n) >> 32);                    \
  211.     ((u_char *) (p))[4] = (u_char) (           (n) >> 24);                    \
  212.     ((u_char *) (p))[5] = (u_char) (           (n) >> 16);                    \
  213.     ((u_char *) (p))[6] = (u_char) (           (n) >> 8);                     \
  214.     ((u_char *) (p))[7] = (u_char)             (n)

  215. #define ngx_mp4_last_trak(mp4)                                                \
  216.     &((ngx_http_mp4_trak_t *) mp4->trak.elts)[mp4->trak.nelts - 1]


  217. static ngx_int_t ngx_http_mp4_handler(ngx_http_request_t *r);
  218. static ngx_int_t ngx_http_mp4_atofp(u_char *line, size_t n, size_t point);

  219. static ngx_int_t ngx_http_mp4_process(ngx_http_mp4_file_t *mp4);
  220. static ngx_int_t ngx_http_mp4_read_atom(ngx_http_mp4_file_t *mp4,
  221.     ngx_http_mp4_atom_handler_t *atom, uint64_t atom_data_size);
  222. static ngx_int_t ngx_http_mp4_read(ngx_http_mp4_file_t *mp4, size_t size);
  223. static ngx_int_t ngx_http_mp4_read_ftyp_atom(ngx_http_mp4_file_t *mp4,
  224.     uint64_t atom_data_size);
  225. static ngx_int_t ngx_http_mp4_read_moov_atom(ngx_http_mp4_file_t *mp4,
  226.     uint64_t atom_data_size);
  227. static ngx_int_t ngx_http_mp4_read_mdat_atom(ngx_http_mp4_file_t *mp4,
  228.     uint64_t atom_data_size);
  229. static size_t ngx_http_mp4_update_mdat_atom(ngx_http_mp4_file_t *mp4,
  230.     off_t start_offset, off_t end_offset);
  231. static ngx_int_t ngx_http_mp4_read_mvhd_atom(ngx_http_mp4_file_t *mp4,
  232.     uint64_t atom_data_size);
  233. static ngx_int_t ngx_http_mp4_read_trak_atom(ngx_http_mp4_file_t *mp4,
  234.     uint64_t atom_data_size);
  235. static void ngx_http_mp4_update_trak_atom(ngx_http_mp4_file_t *mp4,
  236.     ngx_http_mp4_trak_t *trak);
  237. static ngx_int_t ngx_http_mp4_read_cmov_atom(ngx_http_mp4_file_t *mp4,
  238.     uint64_t atom_data_size);
  239. static ngx_int_t ngx_http_mp4_read_tkhd_atom(ngx_http_mp4_file_t *mp4,
  240.     uint64_t atom_data_size);
  241. static ngx_int_t ngx_http_mp4_read_mdia_atom(ngx_http_mp4_file_t *mp4,
  242.     uint64_t atom_data_size);
  243. static void ngx_http_mp4_update_mdia_atom(ngx_http_mp4_file_t *mp4,
  244.     ngx_http_mp4_trak_t *trak);
  245. static ngx_int_t ngx_http_mp4_read_mdhd_atom(ngx_http_mp4_file_t *mp4,
  246.     uint64_t atom_data_size);
  247. static void ngx_http_mp4_update_mdhd_atom(ngx_http_mp4_file_t *mp4,
  248.     ngx_http_mp4_trak_t *trak);
  249. static ngx_int_t ngx_http_mp4_read_hdlr_atom(ngx_http_mp4_file_t *mp4,
  250.     uint64_t atom_data_size);
  251. static ngx_int_t ngx_http_mp4_read_minf_atom(ngx_http_mp4_file_t *mp4,
  252.     uint64_t atom_data_size);
  253. static void ngx_http_mp4_update_minf_atom(ngx_http_mp4_file_t *mp4,
  254.     ngx_http_mp4_trak_t *trak);
  255. static ngx_int_t ngx_http_mp4_read_dinf_atom(ngx_http_mp4_file_t *mp4,
  256.     uint64_t atom_data_size);
  257. static ngx_int_t ngx_http_mp4_read_vmhd_atom(ngx_http_mp4_file_t *mp4,
  258.     uint64_t atom_data_size);
  259. static ngx_int_t ngx_http_mp4_read_smhd_atom(ngx_http_mp4_file_t *mp4,
  260.     uint64_t atom_data_size);
  261. static ngx_int_t ngx_http_mp4_read_stbl_atom(ngx_http_mp4_file_t *mp4,
  262.     uint64_t atom_data_size);
  263. static void ngx_http_mp4_update_edts_atom(ngx_http_mp4_file_t *mp4,
  264.     ngx_http_mp4_trak_t *trak);
  265. static void ngx_http_mp4_update_stbl_atom(ngx_http_mp4_file_t *mp4,
  266.     ngx_http_mp4_trak_t *trak);
  267. static ngx_int_t ngx_http_mp4_read_stsd_atom(ngx_http_mp4_file_t *mp4,
  268.     uint64_t atom_data_size);
  269. static ngx_int_t ngx_http_mp4_read_stts_atom(ngx_http_mp4_file_t *mp4,
  270.     uint64_t atom_data_size);
  271. static ngx_int_t ngx_http_mp4_update_stts_atom(ngx_http_mp4_file_t *mp4,
  272.     ngx_http_mp4_trak_t *trak);
  273. static ngx_int_t ngx_http_mp4_crop_stts_data(ngx_http_mp4_file_t *mp4,
  274.     ngx_http_mp4_trak_t *trak, ngx_uint_t start);
  275. static uint32_t ngx_http_mp4_seek_key_frame(ngx_http_mp4_file_t *mp4,
  276.     ngx_http_mp4_trak_t *trak, uint32_t start_sample);
  277. static ngx_int_t ngx_http_mp4_read_stss_atom(ngx_http_mp4_file_t *mp4,
  278.     uint64_t atom_data_size);
  279. static ngx_int_t ngx_http_mp4_update_stss_atom(ngx_http_mp4_file_t *mp4,
  280.     ngx_http_mp4_trak_t *trak);
  281. static void ngx_http_mp4_crop_stss_data(ngx_http_mp4_file_t *mp4,
  282.     ngx_http_mp4_trak_t *trak, ngx_uint_t start);
  283. static ngx_int_t ngx_http_mp4_read_ctts_atom(ngx_http_mp4_file_t *mp4,
  284.     uint64_t atom_data_size);
  285. static void ngx_http_mp4_update_ctts_atom(ngx_http_mp4_file_t *mp4,
  286.     ngx_http_mp4_trak_t *trak);
  287. static void ngx_http_mp4_crop_ctts_data(ngx_http_mp4_file_t *mp4,
  288.     ngx_http_mp4_trak_t *trak, ngx_uint_t start);
  289. static ngx_int_t ngx_http_mp4_read_stsc_atom(ngx_http_mp4_file_t *mp4,
  290.     uint64_t atom_data_size);
  291. static ngx_int_t ngx_http_mp4_update_stsc_atom(ngx_http_mp4_file_t *mp4,
  292.     ngx_http_mp4_trak_t *trak);
  293. static ngx_int_t ngx_http_mp4_crop_stsc_data(ngx_http_mp4_file_t *mp4,
  294.     ngx_http_mp4_trak_t *trak, ngx_uint_t start);
  295. static ngx_int_t ngx_http_mp4_read_stsz_atom(ngx_http_mp4_file_t *mp4,
  296.     uint64_t atom_data_size);
  297. static ngx_int_t ngx_http_mp4_update_stsz_atom(ngx_http_mp4_file_t *mp4,
  298.     ngx_http_mp4_trak_t *trak);
  299. static ngx_int_t ngx_http_mp4_read_stco_atom(ngx_http_mp4_file_t *mp4,
  300.     uint64_t atom_data_size);
  301. static ngx_int_t ngx_http_mp4_update_stco_atom(ngx_http_mp4_file_t *mp4,
  302.     ngx_http_mp4_trak_t *trak);
  303. static void ngx_http_mp4_adjust_stco_atom(ngx_http_mp4_file_t *mp4,
  304.     ngx_http_mp4_trak_t *trak, int32_t adjustment);
  305. static ngx_int_t ngx_http_mp4_read_co64_atom(ngx_http_mp4_file_t *mp4,
  306.     uint64_t atom_data_size);
  307. static ngx_int_t ngx_http_mp4_update_co64_atom(ngx_http_mp4_file_t *mp4,
  308.     ngx_http_mp4_trak_t *trak);
  309. static void ngx_http_mp4_adjust_co64_atom(ngx_http_mp4_file_t *mp4,
  310.     ngx_http_mp4_trak_t *trak, off_t adjustment);

  311. static char *ngx_http_mp4(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
  312. static void *ngx_http_mp4_create_conf(ngx_conf_t *cf);
  313. static char *ngx_http_mp4_merge_conf(ngx_conf_t *cf, void *parent, void *child);


  314. static ngx_command_t  ngx_http_mp4_commands[] = {

  315.     { ngx_string("mp4"),
  316.       NGX_HTTP_LOC_CONF|NGX_CONF_NOARGS,
  317.       ngx_http_mp4,
  318.       0,
  319.       0,
  320.       NULL },

  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.     { ngx_string("mp4_max_buffer_size"),
  328.       NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
  329.       ngx_conf_set_size_slot,
  330.       NGX_HTTP_LOC_CONF_OFFSET,
  331.       offsetof(ngx_http_mp4_conf_t, max_buffer_size),
  332.       NULL },

  333.     { ngx_string("mp4_start_key_frame"),
  334.       NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_FLAG,
  335.       ngx_conf_set_flag_slot,
  336.       NGX_HTTP_LOC_CONF_OFFSET,
  337.       offsetof(ngx_http_mp4_conf_t, start_key_frame),
  338.       NULL },

  339.       ngx_null_command
  340. };


  341. static ngx_http_module_t  ngx_http_mp4_module_ctx = {
  342.     NULL,                          /* preconfiguration */
  343.     NULL,                          /* postconfiguration */

  344.     NULL,                          /* create main configuration */
  345.     NULL,                          /* init main configuration */

  346.     NULL,                          /* create server configuration */
  347.     NULL,                          /* merge server configuration */

  348.     ngx_http_mp4_create_conf,      /* create location configuration */
  349.     ngx_http_mp4_merge_conf        /* merge location configuration */
  350. };


  351. ngx_module_t  ngx_http_mp4_module = {
  352.     NGX_MODULE_V1,
  353.     &ngx_http_mp4_module_ctx,      /* module context */
  354.     ngx_http_mp4_commands,         /* module directives */
  355.     NGX_HTTP_MODULE,               /* module type */
  356.     NULL,                          /* init master */
  357.     NULL,                          /* init module */
  358.     NULL,                          /* init process */
  359.     NULL,                          /* init thread */
  360.     NULL,                          /* exit thread */
  361.     NULL,                          /* exit process */
  362.     NULL,                          /* exit master */
  363.     NGX_MODULE_V1_PADDING
  364. };


  365. static ngx_http_mp4_atom_handler_t  ngx_http_mp4_atoms[] = {
  366.     { "ftyp", ngx_http_mp4_read_ftyp_atom },
  367.     { "moov", ngx_http_mp4_read_moov_atom },
  368.     { "mdat", ngx_http_mp4_read_mdat_atom },
  369.     { NULL, NULL }
  370. };

  371. static ngx_http_mp4_atom_handler_t  ngx_http_mp4_moov_atoms[] = {
  372.     { "mvhd", ngx_http_mp4_read_mvhd_atom },
  373.     { "trak", ngx_http_mp4_read_trak_atom },
  374.     { "cmov", ngx_http_mp4_read_cmov_atom },
  375.     { NULL, NULL }
  376. };

  377. static ngx_http_mp4_atom_handler_t  ngx_http_mp4_trak_atoms[] = {
  378.     { "tkhd", ngx_http_mp4_read_tkhd_atom },
  379.     { "mdia", ngx_http_mp4_read_mdia_atom },
  380.     { NULL, NULL }
  381. };

  382. static ngx_http_mp4_atom_handler_t  ngx_http_mp4_mdia_atoms[] = {
  383.     { "mdhd", ngx_http_mp4_read_mdhd_atom },
  384.     { "hdlr", ngx_http_mp4_read_hdlr_atom },
  385.     { "minf", ngx_http_mp4_read_minf_atom },
  386.     { NULL, NULL }
  387. };

  388. static ngx_http_mp4_atom_handler_t  ngx_http_mp4_minf_atoms[] = {
  389.     { "vmhd", ngx_http_mp4_read_vmhd_atom },
  390.     { "smhd", ngx_http_mp4_read_smhd_atom },
  391.     { "dinf", ngx_http_mp4_read_dinf_atom },
  392.     { "stbl", ngx_http_mp4_read_stbl_atom },
  393.     { NULL, NULL }
  394. };

  395. static ngx_http_mp4_atom_handler_t  ngx_http_mp4_stbl_atoms[] = {
  396.     { "stsd", ngx_http_mp4_read_stsd_atom },
  397.     { "stts", ngx_http_mp4_read_stts_atom },
  398.     { "stss", ngx_http_mp4_read_stss_atom },
  399.     { "ctts", ngx_http_mp4_read_ctts_atom },
  400.     { "stsc", ngx_http_mp4_read_stsc_atom },
  401.     { "stsz", ngx_http_mp4_read_stsz_atom },
  402.     { "stco", ngx_http_mp4_read_stco_atom },
  403.     { "co64", ngx_http_mp4_read_co64_atom },
  404.     { NULL, NULL }
  405. };


  406. static ngx_int_t
  407. ngx_http_mp4_handler(ngx_http_request_t *r)
  408. {
  409.     u_char                    *last;
  410.     size_t                     root;
  411.     ngx_int_t                  rc, start, end;
  412.     ngx_uint_t                 level, length;
  413.     ngx_str_t                  path, value;
  414.     ngx_log_t                 *log;
  415.     ngx_buf_t                 *b;
  416.     ngx_chain_t                out;
  417.     ngx_http_mp4_file_t       *mp4;
  418.     ngx_open_file_info_t       of;
  419.     ngx_http_core_loc_conf_t  *clcf;

  420.     if (!(r->method & (NGX_HTTP_GET|NGX_HTTP_HEAD))) {
  421.         return NGX_HTTP_NOT_ALLOWED;
  422.     }

  423.     if (r->uri.data[r->uri.len - 1] == '/') {
  424.         return NGX_DECLINED;
  425.     }

  426.     rc = ngx_http_discard_request_body(r);

  427.     if (rc != NGX_OK) {
  428.         return rc;
  429.     }

  430.     last = ngx_http_map_uri_to_path(r, &path, &root, 0);
  431.     if (last == NULL) {
  432.         return NGX_HTTP_INTERNAL_SERVER_ERROR;
  433.     }

  434.     log = r->connection->log;

  435.     path.len = last - path.data;

  436.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, log, 0,
  437.                    "http mp4 filename: \"%V\"", &path);

  438.     clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);

  439.     ngx_memzero(&of, sizeof(ngx_open_file_info_t));

  440.     of.read_ahead = clcf->read_ahead;
  441.     of.directio = NGX_MAX_OFF_T_VALUE;
  442.     of.valid = clcf->open_file_cache_valid;
  443.     of.min_uses = clcf->open_file_cache_min_uses;
  444.     of.errors = clcf->open_file_cache_errors;
  445.     of.events = clcf->open_file_cache_events;

  446.     if (ngx_http_set_disable_symlinks(r, clcf, &path, &of) != NGX_OK) {
  447.         return NGX_HTTP_INTERNAL_SERVER_ERROR;
  448.     }

  449.     if (ngx_open_cached_file(clcf->open_file_cache, &path, &of, r->pool)
  450.         != NGX_OK)
  451.     {
  452.         switch (of.err) {

  453.         case 0:
  454.             return NGX_HTTP_INTERNAL_SERVER_ERROR;

  455.         case NGX_ENOENT:
  456.         case NGX_ENOTDIR:
  457.         case NGX_ENAMETOOLONG:

  458.             level = NGX_LOG_ERR;
  459.             rc = NGX_HTTP_NOT_FOUND;
  460.             break;

  461.         case NGX_EACCES:
  462. #if (NGX_HAVE_OPENAT)
  463.         case NGX_EMLINK:
  464.         case NGX_ELOOP:
  465. #endif

  466.             level = NGX_LOG_ERR;
  467.             rc = NGX_HTTP_FORBIDDEN;
  468.             break;

  469.         default:

  470.             level = NGX_LOG_CRIT;
  471.             rc = NGX_HTTP_INTERNAL_SERVER_ERROR;
  472.             break;
  473.         }

  474.         if (rc != NGX_HTTP_NOT_FOUND || clcf->log_not_found) {
  475.             ngx_log_error(level, log, of.err,
  476.                           "%s \"%s\" failed", of.failed, path.data);
  477.         }

  478.         return rc;
  479.     }

  480.     if (!of.is_file) {
  481.         return NGX_DECLINED;
  482.     }

  483.     r->root_tested = !r->error_page;
  484.     r->allow_ranges = 1;

  485.     start = -1;
  486.     length = 0;
  487.     r->headers_out.content_length_n = of.size;
  488.     mp4 = NULL;
  489.     b = NULL;

  490.     if (r->args.len) {

  491.         if (ngx_http_arg(r, (u_char *) "start", 5, &value) == NGX_OK) {

  492.             /*
  493.              * A Flash player may send start value with a lot of digits
  494.              * after dot so a custom function is used instead of ngx_atofp().
  495.              */

  496.             start = ngx_http_mp4_atofp(value.data, value.len, 3);
  497.         }

  498.         if (ngx_http_arg(r, (u_char *) "end", 3, &value) == NGX_OK) {

  499.             end = ngx_http_mp4_atofp(value.data, value.len, 3);

  500.             if (end > 0) {
  501.                 if (start < 0) {
  502.                     start = 0;
  503.                 }

  504.                 if (end > start) {
  505.                     length = end - start;
  506.                 }
  507.             }
  508.         }
  509.     }

  510.     if (start >= 0) {
  511.         r->single_range = 1;

  512.         mp4 = ngx_pcalloc(r->pool, sizeof(ngx_http_mp4_file_t));
  513.         if (mp4 == NULL) {
  514.             return NGX_HTTP_INTERNAL_SERVER_ERROR;
  515.         }

  516.         mp4->file.fd = of.fd;
  517.         mp4->file.name = path;
  518.         mp4->file.log = r->connection->log;
  519.         mp4->end = of.size;
  520.         mp4->start = (ngx_uint_t) start;
  521.         mp4->length = length;
  522.         mp4->request = r;

  523.         switch (ngx_http_mp4_process(mp4)) {

  524.         case NGX_DECLINED:
  525.             if (mp4->buffer) {
  526.                 ngx_pfree(r->pool, mp4->buffer);
  527.             }

  528.             ngx_pfree(r->pool, mp4);
  529.             mp4 = NULL;

  530.             break;

  531.         case NGX_OK:
  532.             r->headers_out.content_length_n = mp4->content_length;
  533.             break;

  534.         default: /* NGX_ERROR */
  535.             if (mp4->buffer) {
  536.                 ngx_pfree(r->pool, mp4->buffer);
  537.             }

  538.             ngx_pfree(r->pool, mp4);

  539.             return NGX_HTTP_INTERNAL_SERVER_ERROR;
  540.         }
  541.     }

  542.     log->action = "sending mp4 to client";

  543.     if (clcf->directio <= of.size) {

  544.         /*
  545.          * DIRECTIO is set on transfer only
  546.          * to allow kernel to cache "moov" atom
  547.          */

  548.         if (ngx_directio_on(of.fd) == NGX_FILE_ERROR) {
  549.             ngx_log_error(NGX_LOG_ALERT, log, ngx_errno,
  550.                           ngx_directio_on_n " \"%s\" failed", path.data);
  551.         }

  552.         of.is_directio = 1;

  553.         if (mp4) {
  554.             mp4->file.directio = 1;
  555.         }
  556.     }

  557.     r->headers_out.status = NGX_HTTP_OK;
  558.     r->headers_out.last_modified_time = of.mtime;

  559.     if (ngx_http_set_etag(r) != NGX_OK) {
  560.         return NGX_HTTP_INTERNAL_SERVER_ERROR;
  561.     }

  562.     if (ngx_http_set_content_type(r) != NGX_OK) {
  563.         return NGX_HTTP_INTERNAL_SERVER_ERROR;
  564.     }

  565.     if (mp4 == NULL) {
  566.         b = ngx_calloc_buf(r->pool);
  567.         if (b == NULL) {
  568.             return NGX_HTTP_INTERNAL_SERVER_ERROR;
  569.         }

  570.         b->file = ngx_pcalloc(r->pool, sizeof(ngx_file_t));
  571.         if (b->file == NULL) {
  572.             return NGX_HTTP_INTERNAL_SERVER_ERROR;
  573.         }
  574.     }

  575.     rc = ngx_http_send_header(r);

  576.     if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
  577.         return rc;
  578.     }

  579.     if (mp4) {
  580.         return ngx_http_output_filter(r, mp4->out);
  581.     }

  582.     b->file_pos = 0;
  583.     b->file_last = of.size;

  584.     b->in_file = b->file_last ? 1 : 0;
  585.     b->last_buf = (r == r->main) ? 1 : 0;
  586.     b->last_in_chain = 1;
  587.     b->sync = (b->last_buf || b->in_file) ? 0 : 1;

  588.     b->file->fd = of.fd;
  589.     b->file->name = path;
  590.     b->file->log = log;
  591.     b->file->directio = of.is_directio;

  592.     out.buf = b;
  593.     out.next = NULL;

  594.     return ngx_http_output_filter(r, &out);
  595. }


  596. static ngx_int_t
  597. ngx_http_mp4_atofp(u_char *line, size_t n, size_t point)
  598. {
  599.     ngx_int_t   value, cutoff, cutlim;
  600.     ngx_uint_t  dot;

  601.     /* same as ngx_atofp(), but allows additional digits */

  602.     if (n == 0) {
  603.         return NGX_ERROR;
  604.     }

  605.     cutoff = NGX_MAX_INT_T_VALUE / 10;
  606.     cutlim = NGX_MAX_INT_T_VALUE % 10;

  607.     dot = 0;

  608.     for (value = 0; n--; line++) {

  609.         if (*line == '.') {
  610.             if (dot) {
  611.                 return NGX_ERROR;
  612.             }

  613.             dot = 1;
  614.             continue;
  615.         }

  616.         if (*line < '0' || *line > '9') {
  617.             return NGX_ERROR;
  618.         }

  619.         if (point == 0) {
  620.             continue;
  621.         }

  622.         if (value >= cutoff && (value > cutoff || *line - '0' > cutlim)) {
  623.             return NGX_ERROR;
  624.         }

  625.         value = value * 10 + (*line - '0');
  626.         point -= dot;
  627.     }

  628.     while (point--) {
  629.         if (value > cutoff) {
  630.             return NGX_ERROR;
  631.         }

  632.         value = value * 10;
  633.     }

  634.     return value;
  635. }


  636. static ngx_int_t
  637. ngx_http_mp4_process(ngx_http_mp4_file_t *mp4)
  638. {
  639.     off_t                  start_offset, end_offset, adjustment;
  640.     ngx_int_t              rc;
  641.     ngx_uint_t             i, j;
  642.     ngx_chain_t          **prev;
  643.     ngx_http_mp4_trak_t   *trak;
  644.     ngx_http_mp4_conf_t   *conf;

  645.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  646.                    "mp4 start:%ui, length:%ui", mp4->start, mp4->length);

  647.     conf = ngx_http_get_module_loc_conf(mp4->request, ngx_http_mp4_module);

  648.     mp4->buffer_size = conf->buffer_size;

  649.     rc = ngx_http_mp4_read_atom(mp4, ngx_http_mp4_atoms, mp4->end);
  650.     if (rc != NGX_OK) {
  651.         return rc;
  652.     }

  653.     if (mp4->trak.nelts == 0) {
  654.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  655.                       "no mp4 trak atoms were found in \"%s\"",
  656.                       mp4->file.name.data);
  657.         return NGX_ERROR;
  658.     }

  659.     if (mp4->mdat_atom.buf == NULL) {
  660.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  661.                       "no mp4 mdat atom was found in \"%s\"",
  662.                       mp4->file.name.data);
  663.         return NGX_ERROR;
  664.     }

  665.     prev = &mp4->out;

  666.     if (mp4->ftyp_atom.buf) {
  667.         *prev = &mp4->ftyp_atom;
  668.         prev = &mp4->ftyp_atom.next;
  669.     }

  670.     *prev = &mp4->moov_atom;
  671.     prev = &mp4->moov_atom.next;

  672.     if (mp4->mvhd_atom.buf) {
  673.         mp4->moov_size += mp4->mvhd_atom_buf.last - mp4->mvhd_atom_buf.pos;
  674.         *prev = &mp4->mvhd_atom;
  675.         prev = &mp4->mvhd_atom.next;
  676.     }

  677.     start_offset = mp4->end;
  678.     end_offset = 0;
  679.     trak = mp4->trak.elts;

  680.     for (i = 0; i < mp4->trak.nelts; i++) {

  681.         if (ngx_http_mp4_update_stts_atom(mp4, &trak[i]) != NGX_OK) {
  682.             return NGX_ERROR;
  683.         }

  684.         if (ngx_http_mp4_update_stss_atom(mp4, &trak[i]) != NGX_OK) {
  685.             return NGX_ERROR;
  686.         }

  687.         ngx_http_mp4_update_ctts_atom(mp4, &trak[i]);

  688.         if (ngx_http_mp4_update_stsc_atom(mp4, &trak[i]) != NGX_OK) {
  689.             return NGX_ERROR;
  690.         }

  691.         if (ngx_http_mp4_update_stsz_atom(mp4, &trak[i]) != NGX_OK) {
  692.             return NGX_ERROR;
  693.         }

  694.         if (trak[i].out[NGX_HTTP_MP4_CO64_DATA].buf) {
  695.             if (ngx_http_mp4_update_co64_atom(mp4, &trak[i]) != NGX_OK) {
  696.                 return NGX_ERROR;
  697.             }

  698.         } else {
  699.             if (ngx_http_mp4_update_stco_atom(mp4, &trak[i]) != NGX_OK) {
  700.                 return NGX_ERROR;
  701.             }
  702.         }

  703.         ngx_http_mp4_update_stbl_atom(mp4, &trak[i]);
  704.         ngx_http_mp4_update_minf_atom(mp4, &trak[i]);
  705.         ngx_http_mp4_update_mdhd_atom(mp4, &trak[i]);
  706.         trak[i].size += trak[i].hdlr_size;
  707.         ngx_http_mp4_update_mdia_atom(mp4, &trak[i]);
  708.         trak[i].size += trak[i].tkhd_size;
  709.         ngx_http_mp4_update_edts_atom(mp4, &trak[i]);
  710.         ngx_http_mp4_update_trak_atom(mp4, &trak[i]);

  711.         mp4->moov_size += trak[i].size;

  712.         if (start_offset > trak[i].start_offset) {
  713.             start_offset = trak[i].start_offset;
  714.         }

  715.         if (end_offset < trak[i].end_offset) {
  716.             end_offset = trak[i].end_offset;
  717.         }

  718.         *prev = &trak[i].out[NGX_HTTP_MP4_TRAK_ATOM];
  719.         prev = &trak[i].out[NGX_HTTP_MP4_TRAK_ATOM].next;

  720.         for (j = 0; j < NGX_HTTP_MP4_LAST_ATOM + 1; j++) {
  721.             if (trak[i].out[j].buf) {
  722.                 *prev = &trak[i].out[j];
  723.                 prev = &trak[i].out[j].next;
  724.             }
  725.         }
  726.     }

  727.     if (end_offset < start_offset) {
  728.         end_offset = start_offset;
  729.     }

  730.     mp4->moov_size += 8;

  731.     ngx_mp4_set_32value(mp4->moov_atom_header, mp4->moov_size);
  732.     ngx_mp4_set_atom_name(mp4->moov_atom_header, 'm', 'o', 'o', 'v');
  733.     mp4->content_length += mp4->moov_size;

  734.     *prev = &mp4->mdat_atom;

  735.     if (start_offset > mp4->mdat_data.buf->file_last) {
  736.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  737.                       "start time is out mp4 mdat atom in \"%s\"",
  738.                       mp4->file.name.data);
  739.         return NGX_ERROR;
  740.     }

  741.     adjustment = mp4->ftyp_size + mp4->moov_size
  742.                  + ngx_http_mp4_update_mdat_atom(mp4, start_offset, end_offset)
  743.                  - start_offset;

  744.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  745.                    "mp4 adjustment:%O", adjustment);

  746.     for (i = 0; i < mp4->trak.nelts; i++) {
  747.         if (trak[i].out[NGX_HTTP_MP4_CO64_DATA].buf) {
  748.             ngx_http_mp4_adjust_co64_atom(mp4, &trak[i], adjustment);
  749.         } else {
  750.             ngx_http_mp4_adjust_stco_atom(mp4, &trak[i], (int32_t) adjustment);
  751.         }
  752.     }

  753.     return NGX_OK;
  754. }


  755. typedef struct {
  756.     u_char    size[4];
  757.     u_char    name[4];
  758. } ngx_mp4_atom_header_t;

  759. typedef struct {
  760.     u_char    size[4];
  761.     u_char    name[4];
  762.     u_char    size64[8];
  763. } ngx_mp4_atom_header64_t;


  764. static ngx_int_t
  765. ngx_http_mp4_read_atom(ngx_http_mp4_file_t *mp4,
  766.     ngx_http_mp4_atom_handler_t *atom, uint64_t atom_data_size)
  767. {
  768.     off_t        end;
  769.     size_t       atom_header_size;
  770.     u_char      *atom_header, *atom_name;
  771.     uint64_t     atom_size;
  772.     ngx_int_t    rc;
  773.     ngx_uint_t   n;

  774.     end = mp4->offset + atom_data_size;

  775.     while (mp4->offset < end) {

  776.         if (ngx_http_mp4_read(mp4, sizeof(uint32_t)) != NGX_OK) {
  777.             return NGX_ERROR;
  778.         }

  779.         atom_header = mp4->buffer_pos;
  780.         atom_size = ngx_mp4_get_32value(atom_header);
  781.         atom_header_size = sizeof(ngx_mp4_atom_header_t);

  782.         if (atom_size == 0) {
  783.             ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  784.                            "mp4 atom end");
  785.             return NGX_OK;
  786.         }

  787.         if (atom_size < sizeof(ngx_mp4_atom_header_t)) {

  788.             if (atom_size == 1) {

  789.                 if (ngx_http_mp4_read(mp4, sizeof(ngx_mp4_atom_header64_t))
  790.                     != NGX_OK)
  791.                 {
  792.                     return NGX_ERROR;
  793.                 }

  794.                 /* 64-bit atom size */
  795.                 atom_header = mp4->buffer_pos;
  796.                 atom_size = ngx_mp4_get_64value(atom_header + 8);
  797.                 atom_header_size = sizeof(ngx_mp4_atom_header64_t);

  798.                 if (atom_size < sizeof(ngx_mp4_atom_header64_t)) {
  799.                     ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  800.                                   "\"%s\" mp4 atom is too small:%uL",
  801.                                   mp4->file.name.data, atom_size);
  802.                     return NGX_ERROR;
  803.                 }

  804.             } else {
  805.                 ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  806.                               "\"%s\" mp4 atom is too small:%uL",
  807.                               mp4->file.name.data, atom_size);
  808.                 return NGX_ERROR;
  809.             }
  810.         }

  811.         if (ngx_http_mp4_read(mp4, sizeof(ngx_mp4_atom_header_t)) != NGX_OK) {
  812.             return NGX_ERROR;
  813.         }

  814.         atom_header = mp4->buffer_pos;
  815.         atom_name = atom_header + sizeof(uint32_t);

  816.         ngx_log_debug4(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  817.                        "mp4 atom: %*s @%O:%uL",
  818.                        (size_t) 4, atom_name, mp4->offset, atom_size);

  819.         if (atom_size > (uint64_t) (NGX_MAX_OFF_T_VALUE - mp4->offset)
  820.             || mp4->offset + (off_t) atom_size > end)
  821.         {
  822.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  823.                           "\"%s\" mp4 atom too large:%uL",
  824.                           mp4->file.name.data, atom_size);
  825.             return NGX_ERROR;
  826.         }

  827.         for (n = 0; atom[n].name; n++) {

  828.             if (ngx_strncmp(atom_name, atom[n].name, 4) == 0) {

  829.                 ngx_mp4_atom_next(mp4, atom_header_size);

  830.                 rc = atom[n].handler(mp4, atom_size - atom_header_size);
  831.                 if (rc != NGX_OK) {
  832.                     return rc;
  833.                 }

  834.                 goto next;
  835.             }
  836.         }

  837.         ngx_mp4_atom_next(mp4, atom_size);

  838.     next:
  839.         continue;
  840.     }

  841.     return NGX_OK;
  842. }


  843. static ngx_int_t
  844. ngx_http_mp4_read(ngx_http_mp4_file_t *mp4, size_t size)
  845. {
  846.     ssize_t  n;

  847.     if (mp4->buffer_pos + size <= mp4->buffer_end) {
  848.         return NGX_OK;
  849.     }

  850.     if (mp4->offset + (off_t) mp4->buffer_size > mp4->end) {
  851.         mp4->buffer_size = (size_t) (mp4->end - mp4->offset);
  852.     }

  853.     if (mp4->buffer_size < size) {
  854.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  855.                       "\"%s\" mp4 file truncated", mp4->file.name.data);
  856.         return NGX_ERROR;
  857.     }

  858.     if (mp4->buffer == NULL) {
  859.         mp4->buffer = ngx_palloc(mp4->request->pool, mp4->buffer_size);
  860.         if (mp4->buffer == NULL) {
  861.             return NGX_ERROR;
  862.         }

  863.         mp4->buffer_start = mp4->buffer;
  864.     }

  865.     n = ngx_read_file(&mp4->file, mp4->buffer_start, mp4->buffer_size,
  866.                       mp4->offset);

  867.     if (n == NGX_ERROR) {
  868.         return NGX_ERROR;
  869.     }

  870.     if ((size_t) n != mp4->buffer_size) {
  871.         ngx_log_error(NGX_LOG_CRIT, mp4->file.log, 0,
  872.                       ngx_read_file_n " read only %z of %z from \"%s\"",
  873.                       n, mp4->buffer_size, mp4->file.name.data);
  874.         return NGX_ERROR;
  875.     }

  876.     mp4->buffer_pos = mp4->buffer_start;
  877.     mp4->buffer_end = mp4->buffer_start + mp4->buffer_size;

  878.     return NGX_OK;
  879. }


  880. static ngx_int_t
  881. ngx_http_mp4_read_ftyp_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  882. {
  883.     u_char     *ftyp_atom;
  884.     size_t      atom_size;
  885.     ngx_buf_t  *atom;

  886.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 ftyp atom");

  887.     if (atom_data_size > 1024
  888.         || ngx_mp4_atom_data(mp4) + (size_t) atom_data_size > mp4->buffer_end)
  889.     {
  890.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  891.                       "\"%s\" mp4 ftyp atom is too large:%uL",
  892.                       mp4->file.name.data, atom_data_size);
  893.         return NGX_ERROR;
  894.     }

  895.     if (mp4->ftyp_atom.buf) {
  896.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  897.                       "duplicate mp4 ftyp atom in \"%s\"", mp4->file.name.data);
  898.         return NGX_ERROR;
  899.     }

  900.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;

  901.     ftyp_atom = ngx_palloc(mp4->request->pool, atom_size);
  902.     if (ftyp_atom == NULL) {
  903.         return NGX_ERROR;
  904.     }

  905.     ngx_mp4_set_32value(ftyp_atom, atom_size);
  906.     ngx_mp4_set_atom_name(ftyp_atom, 'f', 't', 'y', 'p');

  907.     /*
  908.      * only moov atom content is guaranteed to be in mp4->buffer
  909.      * during sending response, so ftyp atom content should be copied
  910.      */
  911.     ngx_memcpy(ftyp_atom + sizeof(ngx_mp4_atom_header_t),
  912.                ngx_mp4_atom_data(mp4), (size_t) atom_data_size);

  913.     atom = &mp4->ftyp_atom_buf;
  914.     atom->temporary = 1;
  915.     atom->pos = ftyp_atom;
  916.     atom->last = ftyp_atom + atom_size;

  917.     mp4->ftyp_atom.buf = atom;
  918.     mp4->ftyp_size = atom_size;
  919.     mp4->content_length = atom_size;

  920.     ngx_mp4_atom_next(mp4, atom_data_size);

  921.     return NGX_OK;
  922. }


  923. /*
  924. * Small excess buffer to process atoms after moov atom, mp4->buffer_start
  925. * will be set to this buffer part after moov atom processing.
  926. */
  927. #define NGX_HTTP_MP4_MOOV_BUFFER_EXCESS  (4 * 1024)

  928. static ngx_int_t
  929. ngx_http_mp4_read_moov_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  930. {
  931.     ngx_int_t             rc;
  932.     ngx_uint_t            no_mdat;
  933.     ngx_buf_t            *atom;
  934.     ngx_http_mp4_conf_t  *conf;

  935.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 moov atom");

  936.     no_mdat = (mp4->mdat_atom.buf == NULL);

  937.     if (no_mdat && mp4->start == 0 && mp4->length == 0) {
  938.         /*
  939.          * send original file if moov atom resides before
  940.          * mdat atom and client requests integral file
  941.          */
  942.         return NGX_DECLINED;
  943.     }

  944.     if (mp4->moov_atom.buf) {
  945.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  946.                       "duplicate mp4 moov atom in \"%s\"", mp4->file.name.data);
  947.         return NGX_ERROR;
  948.     }

  949.     conf = ngx_http_get_module_loc_conf(mp4->request, ngx_http_mp4_module);

  950.     if (atom_data_size > mp4->buffer_size) {

  951.         if (atom_data_size > conf->max_buffer_size) {
  952.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  953.                           "\"%s\" mp4 moov atom is too large:%uL, "
  954.                           "you may want to increase mp4_max_buffer_size",
  955.                           mp4->file.name.data, atom_data_size);
  956.             return NGX_ERROR;
  957.         }

  958.         ngx_pfree(mp4->request->pool, mp4->buffer);
  959.         mp4->buffer = NULL;
  960.         mp4->buffer_pos = NULL;
  961.         mp4->buffer_end = NULL;

  962.         mp4->buffer_size = (size_t) atom_data_size
  963.                          + NGX_HTTP_MP4_MOOV_BUFFER_EXCESS * no_mdat;
  964.     }

  965.     if (ngx_http_mp4_read(mp4, (size_t) atom_data_size) != NGX_OK) {
  966.         return NGX_ERROR;
  967.     }

  968.     mp4->trak.elts = &mp4->traks;
  969.     mp4->trak.size = sizeof(ngx_http_mp4_trak_t);
  970.     mp4->trak.nalloc = 2;
  971.     mp4->trak.pool = mp4->request->pool;

  972.     atom = &mp4->moov_atom_buf;
  973.     atom->temporary = 1;
  974.     atom->pos = mp4->moov_atom_header;
  975.     atom->last = mp4->moov_atom_header + 8;

  976.     mp4->moov_atom.buf = &mp4->moov_atom_buf;

  977.     rc = ngx_http_mp4_read_atom(mp4, ngx_http_mp4_moov_atoms, atom_data_size);

  978.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 moov atom done");

  979.     if (no_mdat) {
  980.         mp4->buffer_start = mp4->buffer_pos;
  981.         mp4->buffer_size = NGX_HTTP_MP4_MOOV_BUFFER_EXCESS;

  982.         if (mp4->buffer_start + mp4->buffer_size > mp4->buffer_end) {
  983.             mp4->buffer = NULL;
  984.             mp4->buffer_pos = NULL;
  985.             mp4->buffer_end = NULL;
  986.         }

  987.     } else {
  988.         /* skip atoms after moov atom */
  989.         mp4->offset = mp4->end;
  990.     }

  991.     return rc;
  992. }


  993. static ngx_int_t
  994. ngx_http_mp4_read_mdat_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  995. {
  996.     ngx_buf_t  *data;

  997.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 mdat atom");

  998.     if (mp4->mdat_atom.buf) {
  999.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1000.                       "duplicate mp4 mdat atom in \"%s\"", mp4->file.name.data);
  1001.         return NGX_ERROR;
  1002.     }

  1003.     data = &mp4->mdat_data_buf;
  1004.     data->file = &mp4->file;
  1005.     data->in_file = 1;
  1006.     data->last_buf = (mp4->request == mp4->request->main) ? 1 : 0;
  1007.     data->last_in_chain = 1;
  1008.     data->file_last = mp4->offset + atom_data_size;

  1009.     mp4->mdat_atom.buf = &mp4->mdat_atom_buf;
  1010.     mp4->mdat_atom.next = &mp4->mdat_data;
  1011.     mp4->mdat_data.buf = data;

  1012.     if (mp4->trak.nelts) {
  1013.         /* skip atoms after mdat atom */
  1014.         mp4->offset = mp4->end;

  1015.     } else {
  1016.         ngx_mp4_atom_next(mp4, atom_data_size);
  1017.     }

  1018.     return NGX_OK;
  1019. }


  1020. static size_t
  1021. ngx_http_mp4_update_mdat_atom(ngx_http_mp4_file_t *mp4, off_t start_offset,
  1022.     off_t end_offset)
  1023. {
  1024.     off_t       atom_data_size;
  1025.     u_char     *atom_header;
  1026.     uint32_t    atom_header_size;
  1027.     uint64_t    atom_size;
  1028.     ngx_buf_t  *atom;

  1029.     atom_data_size = end_offset - start_offset;
  1030.     mp4->mdat_data.buf->file_pos = start_offset;
  1031.     mp4->mdat_data.buf->file_last = end_offset;

  1032.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1033.                    "mdat new offset @%O:%O", start_offset, atom_data_size);

  1034.     atom_header = mp4->mdat_atom_header;

  1035.     if ((uint64_t) atom_data_size
  1036.         > (uint64_t) 0xffffffff - sizeof(ngx_mp4_atom_header_t))
  1037.     {
  1038.         atom_size = 1;
  1039.         atom_header_size = sizeof(ngx_mp4_atom_header64_t);
  1040.         ngx_mp4_set_64value(atom_header + sizeof(ngx_mp4_atom_header_t),
  1041.                             sizeof(ngx_mp4_atom_header64_t) + atom_data_size);
  1042.     } else {
  1043.         atom_size = sizeof(ngx_mp4_atom_header_t) + atom_data_size;
  1044.         atom_header_size = sizeof(ngx_mp4_atom_header_t);
  1045.     }

  1046.     mp4->content_length += atom_header_size + atom_data_size;

  1047.     ngx_mp4_set_32value(atom_header, atom_size);
  1048.     ngx_mp4_set_atom_name(atom_header, 'm', 'd', 'a', 't');

  1049.     atom = &mp4->mdat_atom_buf;
  1050.     atom->temporary = 1;
  1051.     atom->pos = atom_header;
  1052.     atom->last = atom_header + atom_header_size;

  1053.     return atom_header_size;
  1054. }


  1055. typedef struct {
  1056.     u_char    size[4];
  1057.     u_char    name[4];
  1058.     u_char    version[1];
  1059.     u_char    flags[3];
  1060.     u_char    creation_time[4];
  1061.     u_char    modification_time[4];
  1062.     u_char    timescale[4];
  1063.     u_char    duration[4];
  1064.     u_char    rate[4];
  1065.     u_char    volume[2];
  1066.     u_char    reserved[10];
  1067.     u_char    matrix[36];
  1068.     u_char    preview_time[4];
  1069.     u_char    preview_duration[4];
  1070.     u_char    poster_time[4];
  1071.     u_char    selection_time[4];
  1072.     u_char    selection_duration[4];
  1073.     u_char    current_time[4];
  1074.     u_char    next_track_id[4];
  1075. } ngx_mp4_mvhd_atom_t;

  1076. typedef struct {
  1077.     u_char    size[4];
  1078.     u_char    name[4];
  1079.     u_char    version[1];
  1080.     u_char    flags[3];
  1081.     u_char    creation_time[8];
  1082.     u_char    modification_time[8];
  1083.     u_char    timescale[4];
  1084.     u_char    duration[8];
  1085.     u_char    rate[4];
  1086.     u_char    volume[2];
  1087.     u_char    reserved[10];
  1088.     u_char    matrix[36];
  1089.     u_char    preview_time[4];
  1090.     u_char    preview_duration[4];
  1091.     u_char    poster_time[4];
  1092.     u_char    selection_time[4];
  1093.     u_char    selection_duration[4];
  1094.     u_char    current_time[4];
  1095.     u_char    next_track_id[4];
  1096. } ngx_mp4_mvhd64_atom_t;


  1097. static ngx_int_t
  1098. ngx_http_mp4_read_mvhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1099. {
  1100.     u_char                 *atom_header;
  1101.     size_t                  atom_size;
  1102.     uint32_t                timescale;
  1103.     uint64_t                duration, start_time, length_time;
  1104.     ngx_buf_t              *atom;
  1105.     ngx_mp4_mvhd_atom_t    *mvhd_atom;
  1106.     ngx_mp4_mvhd64_atom_t  *mvhd64_atom;

  1107.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 mvhd atom");

  1108.     if (mp4->mvhd_atom.buf) {
  1109.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1110.                       "duplicate mp4 mvhd atom in \"%s\"", mp4->file.name.data);
  1111.         return NGX_ERROR;
  1112.     }

  1113.     atom_header = ngx_mp4_atom_header(mp4);
  1114.     mvhd_atom = (ngx_mp4_mvhd_atom_t *) atom_header;
  1115.     mvhd64_atom = (ngx_mp4_mvhd64_atom_t *) atom_header;
  1116.     ngx_mp4_set_atom_name(atom_header, 'm', 'v', 'h', 'd');

  1117.     if (ngx_mp4_atom_data_size(ngx_mp4_mvhd_atom_t) > atom_data_size) {
  1118.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1119.                       "\"%s\" mp4 mvhd atom too small", mp4->file.name.data);
  1120.         return NGX_ERROR;
  1121.     }

  1122.     if (mvhd_atom->version[0] == 0) {
  1123.         /* version 0: 32-bit duration */
  1124.         timescale = ngx_mp4_get_32value(mvhd_atom->timescale);
  1125.         duration = ngx_mp4_get_32value(mvhd_atom->duration);

  1126.     } else {
  1127.         /* version 1: 64-bit duration */

  1128.         if (ngx_mp4_atom_data_size(ngx_mp4_mvhd64_atom_t) > atom_data_size) {
  1129.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1130.                           "\"%s\" mp4 mvhd atom too small",
  1131.                           mp4->file.name.data);
  1132.             return NGX_ERROR;
  1133.         }

  1134.         timescale = ngx_mp4_get_32value(mvhd64_atom->timescale);
  1135.         duration = ngx_mp4_get_64value(mvhd64_atom->duration);
  1136.     }

  1137.     mp4->timescale = timescale;

  1138.     ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1139.                    "mvhd timescale:%uD, duration:%uL, time:%.3fs",
  1140.                    timescale, duration, (double) duration / timescale);

  1141.     start_time = (uint64_t) mp4->start * timescale / 1000;

  1142.     if (duration < start_time) {
  1143.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1144.                       "\"%s\" mp4 start time exceeds file duration",
  1145.                       mp4->file.name.data);
  1146.         return NGX_ERROR;
  1147.     }

  1148.     duration -= start_time;

  1149.     if (mp4->length) {
  1150.         length_time = (uint64_t) mp4->length * timescale / 1000;

  1151.         if (duration > length_time) {
  1152.             duration = length_time;
  1153.         }
  1154.     }

  1155.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1156.                    "mvhd new duration:%uL, time:%.3fs",
  1157.                    duration, (double) duration / timescale);

  1158.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  1159.     ngx_mp4_set_32value(mvhd_atom->size, atom_size);

  1160.     if (mvhd_atom->version[0] == 0) {
  1161.         ngx_mp4_set_32value(mvhd_atom->duration, duration);

  1162.     } else {
  1163.         ngx_mp4_set_64value(mvhd64_atom->duration, duration);
  1164.     }

  1165.     atom = &mp4->mvhd_atom_buf;
  1166.     atom->temporary = 1;
  1167.     atom->pos = atom_header;
  1168.     atom->last = atom_header + atom_size;

  1169.     mp4->mvhd_atom.buf = atom;

  1170.     ngx_mp4_atom_next(mp4, atom_data_size);

  1171.     return NGX_OK;
  1172. }


  1173. static ngx_int_t
  1174. ngx_http_mp4_read_trak_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1175. {
  1176.     u_char               *atom_header, *atom_end;
  1177.     off_t                 atom_file_end;
  1178.     ngx_int_t             rc;
  1179.     ngx_buf_t            *atom;
  1180.     ngx_http_mp4_trak_t  *trak;

  1181.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 trak atom");

  1182.     trak = ngx_array_push(&mp4->trak);
  1183.     if (trak == NULL) {
  1184.         return NGX_ERROR;
  1185.     }

  1186.     ngx_memzero(trak, sizeof(ngx_http_mp4_trak_t));

  1187.     atom_header = ngx_mp4_atom_header(mp4);
  1188.     ngx_mp4_set_atom_name(atom_header, 't', 'r', 'a', 'k');

  1189.     atom = &trak->trak_atom_buf;
  1190.     atom->temporary = 1;
  1191.     atom->pos = atom_header;
  1192.     atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);

  1193.     trak->out[NGX_HTTP_MP4_TRAK_ATOM].buf = atom;

  1194.     atom_end = mp4->buffer_pos + (size_t) atom_data_size;
  1195.     atom_file_end = mp4->offset + atom_data_size;

  1196.     rc = ngx_http_mp4_read_atom(mp4, ngx_http_mp4_trak_atoms, atom_data_size);

  1197.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1198.                    "mp4 trak atom: %i", rc);

  1199.     if (rc == NGX_DECLINED) {
  1200.         /* skip this trak */
  1201.         ngx_memzero(trak, sizeof(ngx_http_mp4_trak_t));
  1202.         mp4->trak.nelts--;
  1203.         mp4->buffer_pos = atom_end;
  1204.         mp4->offset = atom_file_end;
  1205.         return NGX_OK;
  1206.     }

  1207.     return rc;
  1208. }


  1209. static void
  1210. ngx_http_mp4_update_trak_atom(ngx_http_mp4_file_t *mp4,
  1211.     ngx_http_mp4_trak_t *trak)
  1212. {
  1213.     ngx_buf_t  *atom;

  1214.     trak->size += sizeof(ngx_mp4_atom_header_t);
  1215.     atom = &trak->trak_atom_buf;
  1216.     ngx_mp4_set_32value(atom->pos, trak->size);
  1217. }


  1218. static ngx_int_t
  1219. ngx_http_mp4_read_cmov_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1220. {
  1221.     ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1222.                   "\"%s\" mp4 compressed moov atom (cmov) is not supported",
  1223.                   mp4->file.name.data);

  1224.     return NGX_ERROR;
  1225. }


  1226. typedef struct {
  1227.     u_char    size[4];
  1228.     u_char    name[4];
  1229.     u_char    version[1];
  1230.     u_char    flags[3];
  1231.     u_char    creation_time[4];
  1232.     u_char    modification_time[4];
  1233.     u_char    track_id[4];
  1234.     u_char    reserved1[4];
  1235.     u_char    duration[4];
  1236.     u_char    reserved2[8];
  1237.     u_char    layer[2];
  1238.     u_char    group[2];
  1239.     u_char    volume[2];
  1240.     u_char    reserved3[2];
  1241.     u_char    matrix[36];
  1242.     u_char    width[4];
  1243.     u_char    height[4];
  1244. } ngx_mp4_tkhd_atom_t;

  1245. typedef struct {
  1246.     u_char    size[4];
  1247.     u_char    name[4];
  1248.     u_char    version[1];
  1249.     u_char    flags[3];
  1250.     u_char    creation_time[8];
  1251.     u_char    modification_time[8];
  1252.     u_char    track_id[4];
  1253.     u_char    reserved1[4];
  1254.     u_char    duration[8];
  1255.     u_char    reserved2[8];
  1256.     u_char    layer[2];
  1257.     u_char    group[2];
  1258.     u_char    volume[2];
  1259.     u_char    reserved3[2];
  1260.     u_char    matrix[36];
  1261.     u_char    width[4];
  1262.     u_char    height[4];
  1263. } ngx_mp4_tkhd64_atom_t;


  1264. static ngx_int_t
  1265. ngx_http_mp4_read_tkhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1266. {
  1267.     u_char                 *atom_header;
  1268.     size_t                  atom_size;
  1269.     uint64_t                duration, start_time, length_time;
  1270.     ngx_buf_t              *atom;
  1271.     ngx_http_mp4_trak_t    *trak;
  1272.     ngx_mp4_tkhd_atom_t    *tkhd_atom;
  1273.     ngx_mp4_tkhd64_atom_t  *tkhd64_atom;

  1274.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 tkhd atom");

  1275.     atom_header = ngx_mp4_atom_header(mp4);
  1276.     tkhd_atom = (ngx_mp4_tkhd_atom_t *) atom_header;
  1277.     tkhd64_atom = (ngx_mp4_tkhd64_atom_t *) atom_header;
  1278.     ngx_mp4_set_atom_name(tkhd_atom, 't', 'k', 'h', 'd');

  1279.     if (ngx_mp4_atom_data_size(ngx_mp4_tkhd_atom_t) > atom_data_size) {
  1280.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1281.                       "\"%s\" mp4 tkhd atom too small", mp4->file.name.data);
  1282.         return NGX_ERROR;
  1283.     }

  1284.     if (tkhd_atom->version[0] == 0) {
  1285.         /* version 0: 32-bit duration */
  1286.         duration = ngx_mp4_get_32value(tkhd_atom->duration);

  1287.     } else {
  1288.         /* version 1: 64-bit duration */

  1289.         if (ngx_mp4_atom_data_size(ngx_mp4_tkhd64_atom_t) > atom_data_size) {
  1290.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1291.                           "\"%s\" mp4 tkhd atom too small",
  1292.                           mp4->file.name.data);
  1293.             return NGX_ERROR;
  1294.         }

  1295.         duration = ngx_mp4_get_64value(tkhd64_atom->duration);
  1296.     }

  1297.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1298.                    "tkhd duration:%uL, time:%.3fs",
  1299.                    duration, (double) duration / mp4->timescale);

  1300.     start_time = (uint64_t) mp4->start * mp4->timescale / 1000;

  1301.     if (duration <= start_time) {
  1302.         ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1303.                        "tkhd duration is less than start time");
  1304.         return NGX_DECLINED;
  1305.     }

  1306.     duration -= start_time;

  1307.     if (mp4->length) {
  1308.         length_time = (uint64_t) mp4->length * mp4->timescale / 1000;

  1309.         if (duration > length_time) {
  1310.             duration = length_time;
  1311.         }
  1312.     }

  1313.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1314.                    "tkhd new duration:%uL, time:%.3fs",
  1315.                    duration, (double) duration / mp4->timescale);

  1316.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;

  1317.     trak = ngx_mp4_last_trak(mp4);

  1318.     if (trak->out[NGX_HTTP_MP4_TKHD_ATOM].buf) {
  1319.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1320.                       "duplicate mp4 tkhd atom in \"%s\"", mp4->file.name.data);
  1321.         return NGX_ERROR;
  1322.     }

  1323.     trak->tkhd_size = atom_size;
  1324.     trak->movie_duration = duration;

  1325.     ngx_mp4_set_32value(tkhd_atom->size, atom_size);

  1326.     if (tkhd_atom->version[0] == 0) {
  1327.         ngx_mp4_set_32value(tkhd_atom->duration, duration);

  1328.     } else {
  1329.         ngx_mp4_set_64value(tkhd64_atom->duration, duration);
  1330.     }

  1331.     atom = &trak->tkhd_atom_buf;
  1332.     atom->temporary = 1;
  1333.     atom->pos = atom_header;
  1334.     atom->last = atom_header + atom_size;

  1335.     trak->out[NGX_HTTP_MP4_TKHD_ATOM].buf = atom;

  1336.     ngx_mp4_atom_next(mp4, atom_data_size);

  1337.     return NGX_OK;
  1338. }


  1339. static ngx_int_t
  1340. ngx_http_mp4_read_mdia_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1341. {
  1342.     u_char               *atom_header;
  1343.     ngx_buf_t            *atom;
  1344.     ngx_http_mp4_trak_t  *trak;

  1345.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "process mdia atom");

  1346.     atom_header = ngx_mp4_atom_header(mp4);
  1347.     ngx_mp4_set_atom_name(atom_header, 'm', 'd', 'i', 'a');

  1348.     trak = ngx_mp4_last_trak(mp4);

  1349.     if (trak->out[NGX_HTTP_MP4_MDIA_ATOM].buf) {
  1350.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1351.                       "duplicate mp4 mdia atom in \"%s\"", mp4->file.name.data);
  1352.         return NGX_ERROR;
  1353.     }

  1354.     atom = &trak->mdia_atom_buf;
  1355.     atom->temporary = 1;
  1356.     atom->pos = atom_header;
  1357.     atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);

  1358.     trak->out[NGX_HTTP_MP4_MDIA_ATOM].buf = atom;

  1359.     return ngx_http_mp4_read_atom(mp4, ngx_http_mp4_mdia_atoms, atom_data_size);
  1360. }


  1361. static void
  1362. ngx_http_mp4_update_mdia_atom(ngx_http_mp4_file_t *mp4,
  1363.     ngx_http_mp4_trak_t *trak)
  1364. {
  1365.     ngx_buf_t  *atom;

  1366.     trak->size += sizeof(ngx_mp4_atom_header_t);
  1367.     atom = &trak->mdia_atom_buf;
  1368.     ngx_mp4_set_32value(atom->pos, trak->size);
  1369. }


  1370. typedef struct {
  1371.     u_char    size[4];
  1372.     u_char    name[4];
  1373.     u_char    version[1];
  1374.     u_char    flags[3];
  1375.     u_char    creation_time[4];
  1376.     u_char    modification_time[4];
  1377.     u_char    timescale[4];
  1378.     u_char    duration[4];
  1379.     u_char    language[2];
  1380.     u_char    quality[2];
  1381. } ngx_mp4_mdhd_atom_t;

  1382. typedef struct {
  1383.     u_char    size[4];
  1384.     u_char    name[4];
  1385.     u_char    version[1];
  1386.     u_char    flags[3];
  1387.     u_char    creation_time[8];
  1388.     u_char    modification_time[8];
  1389.     u_char    timescale[4];
  1390.     u_char    duration[8];
  1391.     u_char    language[2];
  1392.     u_char    quality[2];
  1393. } ngx_mp4_mdhd64_atom_t;


  1394. static ngx_int_t
  1395. ngx_http_mp4_read_mdhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1396. {
  1397.     u_char                 *atom_header;
  1398.     size_t                  atom_size;
  1399.     uint32_t                timescale;
  1400.     uint64_t                duration, start_time, length_time;
  1401.     ngx_buf_t              *atom;
  1402.     ngx_http_mp4_trak_t    *trak;
  1403.     ngx_mp4_mdhd_atom_t    *mdhd_atom;
  1404.     ngx_mp4_mdhd64_atom_t  *mdhd64_atom;

  1405.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 mdhd atom");

  1406.     atom_header = ngx_mp4_atom_header(mp4);
  1407.     mdhd_atom = (ngx_mp4_mdhd_atom_t *) atom_header;
  1408.     mdhd64_atom = (ngx_mp4_mdhd64_atom_t *) atom_header;
  1409.     ngx_mp4_set_atom_name(mdhd_atom, 'm', 'd', 'h', 'd');

  1410.     if (ngx_mp4_atom_data_size(ngx_mp4_mdhd_atom_t) > atom_data_size) {
  1411.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1412.                       "\"%s\" mp4 mdhd atom too small", mp4->file.name.data);
  1413.         return NGX_ERROR;
  1414.     }

  1415.     if (mdhd_atom->version[0] == 0) {
  1416.         /* version 0: everything is 32-bit */
  1417.         timescale = ngx_mp4_get_32value(mdhd_atom->timescale);
  1418.         duration = ngx_mp4_get_32value(mdhd_atom->duration);

  1419.     } else {
  1420.         /* version 1: 64-bit duration and 32-bit timescale */

  1421.         if (ngx_mp4_atom_data_size(ngx_mp4_mdhd64_atom_t) > atom_data_size) {
  1422.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1423.                           "\"%s\" mp4 mdhd atom too small",
  1424.                           mp4->file.name.data);
  1425.             return NGX_ERROR;
  1426.         }

  1427.         timescale = ngx_mp4_get_32value(mdhd64_atom->timescale);
  1428.         duration = ngx_mp4_get_64value(mdhd64_atom->duration);
  1429.     }

  1430.     ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1431.                    "mdhd timescale:%uD, duration:%uL, time:%.3fs",
  1432.                    timescale, duration, (double) duration / timescale);

  1433.     start_time = (uint64_t) mp4->start * timescale / 1000;

  1434.     if (duration <= start_time) {
  1435.         ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1436.                        "mdhd duration is less than start time");
  1437.         return NGX_DECLINED;
  1438.     }

  1439.     duration -= start_time;

  1440.     if (mp4->length) {
  1441.         length_time = (uint64_t) mp4->length * timescale / 1000;

  1442.         if (duration > length_time) {
  1443.             duration = length_time;
  1444.         }
  1445.     }

  1446.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1447.                    "mdhd new duration:%uL, time:%.3fs",
  1448.                    duration, (double) duration / timescale);

  1449.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;

  1450.     trak = ngx_mp4_last_trak(mp4);

  1451.     if (trak->out[NGX_HTTP_MP4_MDHD_ATOM].buf) {
  1452.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1453.                       "duplicate mp4 mdhd atom in \"%s\"", mp4->file.name.data);
  1454.         return NGX_ERROR;
  1455.     }

  1456.     trak->mdhd_size = atom_size;
  1457.     trak->timescale = timescale;
  1458.     trak->duration = duration;

  1459.     ngx_mp4_set_32value(mdhd_atom->size, atom_size);

  1460.     atom = &trak->mdhd_atom_buf;
  1461.     atom->temporary = 1;
  1462.     atom->pos = atom_header;
  1463.     atom->last = atom_header + atom_size;

  1464.     trak->out[NGX_HTTP_MP4_MDHD_ATOM].buf = atom;

  1465.     ngx_mp4_atom_next(mp4, atom_data_size);

  1466.     return NGX_OK;
  1467. }


  1468. static void
  1469. ngx_http_mp4_update_mdhd_atom(ngx_http_mp4_file_t *mp4,
  1470.             ngx_http_mp4_trak_t *trak)
  1471. {
  1472.     ngx_buf_t              *atom;
  1473.     ngx_mp4_mdhd_atom_t    *mdhd_atom;
  1474.     ngx_mp4_mdhd64_atom_t  *mdhd64_atom;

  1475.     atom = trak->out[NGX_HTTP_MP4_MDHD_ATOM].buf;
  1476.     if (atom == NULL) {
  1477.         return;
  1478.     }

  1479.     mdhd_atom = (ngx_mp4_mdhd_atom_t *) atom->pos;
  1480.     mdhd64_atom = (ngx_mp4_mdhd64_atom_t *) atom->pos;

  1481.     if (mdhd_atom->version[0] == 0) {
  1482.         ngx_mp4_set_32value(mdhd_atom->duration, trak->duration);

  1483.     } else {
  1484.         ngx_mp4_set_64value(mdhd64_atom->duration, trak->duration);
  1485.     }

  1486.     trak->size += trak->mdhd_size;
  1487. }


  1488. static ngx_int_t
  1489. ngx_http_mp4_read_hdlr_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1490. {
  1491.     u_char              *atom_header;
  1492.     size_t               atom_size;
  1493.     ngx_buf_t            *atom;
  1494.     ngx_http_mp4_trak_t  *trak;

  1495.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 hdlr atom");

  1496.     atom_header = ngx_mp4_atom_header(mp4);
  1497.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  1498.     ngx_mp4_set_32value(atom_header, atom_size);
  1499.     ngx_mp4_set_atom_name(atom_header, 'h', 'd', 'l', 'r');

  1500.     trak = ngx_mp4_last_trak(mp4);

  1501.     if (trak->out[NGX_HTTP_MP4_HDLR_ATOM].buf) {
  1502.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1503.                       "duplicate mp4 hdlr atom in \"%s\"", mp4->file.name.data);
  1504.         return NGX_ERROR;
  1505.     }

  1506.     atom = &trak->hdlr_atom_buf;
  1507.     atom->temporary = 1;
  1508.     atom->pos = atom_header;
  1509.     atom->last = atom_header + atom_size;

  1510.     trak->hdlr_size = atom_size;
  1511.     trak->out[NGX_HTTP_MP4_HDLR_ATOM].buf = atom;

  1512.     ngx_mp4_atom_next(mp4, atom_data_size);

  1513.     return NGX_OK;
  1514. }


  1515. static ngx_int_t
  1516. ngx_http_mp4_read_minf_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1517. {
  1518.     u_char               *atom_header;
  1519.     ngx_buf_t            *atom;
  1520.     ngx_http_mp4_trak_t  *trak;

  1521.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "process minf atom");

  1522.     atom_header = ngx_mp4_atom_header(mp4);
  1523.     ngx_mp4_set_atom_name(atom_header, 'm', 'i', 'n', 'f');

  1524.     trak = ngx_mp4_last_trak(mp4);

  1525.     if (trak->out[NGX_HTTP_MP4_MINF_ATOM].buf) {
  1526.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1527.                       "duplicate mp4 minf atom in \"%s\"", mp4->file.name.data);
  1528.         return NGX_ERROR;
  1529.     }

  1530.     atom = &trak->minf_atom_buf;
  1531.     atom->temporary = 1;
  1532.     atom->pos = atom_header;
  1533.     atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);

  1534.     trak->out[NGX_HTTP_MP4_MINF_ATOM].buf = atom;

  1535.     return ngx_http_mp4_read_atom(mp4, ngx_http_mp4_minf_atoms, atom_data_size);
  1536. }


  1537. static void
  1538. ngx_http_mp4_update_minf_atom(ngx_http_mp4_file_t *mp4,
  1539.     ngx_http_mp4_trak_t *trak)
  1540. {
  1541.     ngx_buf_t  *atom;

  1542.     trak->size += sizeof(ngx_mp4_atom_header_t)
  1543.                + trak->vmhd_size
  1544.                + trak->smhd_size
  1545.                + trak->dinf_size;
  1546.     atom = &trak->minf_atom_buf;
  1547.     ngx_mp4_set_32value(atom->pos, trak->size);
  1548. }


  1549. static ngx_int_t
  1550. ngx_http_mp4_read_vmhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1551. {
  1552.     u_char              *atom_header;
  1553.     size_t               atom_size;
  1554.     ngx_buf_t            *atom;
  1555.     ngx_http_mp4_trak_t  *trak;

  1556.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 vmhd atom");

  1557.     atom_header = ngx_mp4_atom_header(mp4);
  1558.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  1559.     ngx_mp4_set_32value(atom_header, atom_size);
  1560.     ngx_mp4_set_atom_name(atom_header, 'v', 'm', 'h', 'd');

  1561.     trak = ngx_mp4_last_trak(mp4);

  1562.     if (trak->out[NGX_HTTP_MP4_VMHD_ATOM].buf
  1563.         || trak->out[NGX_HTTP_MP4_SMHD_ATOM].buf)
  1564.     {
  1565.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1566.                       "duplicate mp4 vmhd/smhd atom in \"%s\"",
  1567.                       mp4->file.name.data);
  1568.         return NGX_ERROR;
  1569.     }

  1570.     atom = &trak->vmhd_atom_buf;
  1571.     atom->temporary = 1;
  1572.     atom->pos = atom_header;
  1573.     atom->last = atom_header + atom_size;

  1574.     trak->vmhd_size += atom_size;
  1575.     trak->out[NGX_HTTP_MP4_VMHD_ATOM].buf = atom;

  1576.     ngx_mp4_atom_next(mp4, atom_data_size);

  1577.     return NGX_OK;
  1578. }


  1579. static ngx_int_t
  1580. ngx_http_mp4_read_smhd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1581. {
  1582.     u_char              *atom_header;
  1583.     size_t               atom_size;
  1584.     ngx_buf_t            *atom;
  1585.     ngx_http_mp4_trak_t  *trak;

  1586.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 smhd atom");

  1587.     atom_header = ngx_mp4_atom_header(mp4);
  1588.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  1589.     ngx_mp4_set_32value(atom_header, atom_size);
  1590.     ngx_mp4_set_atom_name(atom_header, 's', 'm', 'h', 'd');

  1591.     trak = ngx_mp4_last_trak(mp4);

  1592.     if (trak->out[NGX_HTTP_MP4_VMHD_ATOM].buf
  1593.         || trak->out[NGX_HTTP_MP4_SMHD_ATOM].buf)
  1594.     {
  1595.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1596.                       "duplicate mp4 vmhd/smhd atom in \"%s\"",
  1597.                       mp4->file.name.data);
  1598.         return NGX_ERROR;
  1599.     }

  1600.     atom = &trak->smhd_atom_buf;
  1601.     atom->temporary = 1;
  1602.     atom->pos = atom_header;
  1603.     atom->last = atom_header + atom_size;

  1604.     trak->smhd_size += atom_size;
  1605.     trak->out[NGX_HTTP_MP4_SMHD_ATOM].buf = atom;

  1606.     ngx_mp4_atom_next(mp4, atom_data_size);

  1607.     return NGX_OK;
  1608. }


  1609. static ngx_int_t
  1610. ngx_http_mp4_read_dinf_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1611. {
  1612.     u_char              *atom_header;
  1613.     size_t               atom_size;
  1614.     ngx_buf_t            *atom;
  1615.     ngx_http_mp4_trak_t  *trak;

  1616.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 dinf atom");

  1617.     atom_header = ngx_mp4_atom_header(mp4);
  1618.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  1619.     ngx_mp4_set_32value(atom_header, atom_size);
  1620.     ngx_mp4_set_atom_name(atom_header, 'd', 'i', 'n', 'f');

  1621.     trak = ngx_mp4_last_trak(mp4);

  1622.     if (trak->out[NGX_HTTP_MP4_DINF_ATOM].buf) {
  1623.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1624.                       "duplicate mp4 dinf atom in \"%s\"", mp4->file.name.data);
  1625.         return NGX_ERROR;
  1626.     }

  1627.     atom = &trak->dinf_atom_buf;
  1628.     atom->temporary = 1;
  1629.     atom->pos = atom_header;
  1630.     atom->last = atom_header + atom_size;

  1631.     trak->dinf_size += atom_size;
  1632.     trak->out[NGX_HTTP_MP4_DINF_ATOM].buf = atom;

  1633.     ngx_mp4_atom_next(mp4, atom_data_size);

  1634.     return NGX_OK;
  1635. }


  1636. static ngx_int_t
  1637. ngx_http_mp4_read_stbl_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1638. {
  1639.     u_char               *atom_header;
  1640.     ngx_buf_t            *atom;
  1641.     ngx_http_mp4_trak_t  *trak;

  1642.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "process stbl atom");

  1643.     atom_header = ngx_mp4_atom_header(mp4);
  1644.     ngx_mp4_set_atom_name(atom_header, 's', 't', 'b', 'l');

  1645.     trak = ngx_mp4_last_trak(mp4);

  1646.     if (trak->out[NGX_HTTP_MP4_STBL_ATOM].buf) {
  1647.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1648.                       "duplicate mp4 stbl atom in \"%s\"", mp4->file.name.data);
  1649.         return NGX_ERROR;
  1650.     }

  1651.     atom = &trak->stbl_atom_buf;
  1652.     atom->temporary = 1;
  1653.     atom->pos = atom_header;
  1654.     atom->last = atom_header + sizeof(ngx_mp4_atom_header_t);

  1655.     trak->out[NGX_HTTP_MP4_STBL_ATOM].buf = atom;

  1656.     return ngx_http_mp4_read_atom(mp4, ngx_http_mp4_stbl_atoms, atom_data_size);
  1657. }


  1658. static void
  1659. ngx_http_mp4_update_edts_atom(ngx_http_mp4_file_t *mp4,
  1660.     ngx_http_mp4_trak_t *trak)
  1661. {
  1662.     ngx_buf_t            *atom;
  1663.     ngx_mp4_elst_atom_t  *elst_atom;
  1664.     ngx_mp4_edts_atom_t  *edts_atom;

  1665.     if (trak->prefix == 0) {
  1666.         return;
  1667.     }

  1668.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1669.                    "mp4 edts atom update prefix:%uL", trak->prefix);

  1670.     edts_atom = &trak->edts_atom;
  1671.     ngx_mp4_set_32value(edts_atom->size, sizeof(ngx_mp4_edts_atom_t)
  1672.                                          + sizeof(ngx_mp4_elst_atom_t));
  1673.     ngx_mp4_set_atom_name(edts_atom, 'e', 'd', 't', 's');

  1674.     atom = &trak->edts_atom_buf;
  1675.     atom->temporary = 1;
  1676.     atom->pos = (u_char *) edts_atom;
  1677.     atom->last = (u_char *) edts_atom + sizeof(ngx_mp4_edts_atom_t);

  1678.     trak->out[NGX_HTTP_MP4_EDTS_ATOM].buf = atom;

  1679.     elst_atom = &trak->elst_atom;
  1680.     ngx_mp4_set_32value(elst_atom->size, sizeof(ngx_mp4_elst_atom_t));
  1681.     ngx_mp4_set_atom_name(elst_atom, 'e', 'l', 's', 't');

  1682.     elst_atom->version[0] = 1;
  1683.     elst_atom->flags[0] = 0;
  1684.     elst_atom->flags[1] = 0;
  1685.     elst_atom->flags[2] = 0;

  1686.     ngx_mp4_set_32value(elst_atom->entries, 1);
  1687.     ngx_mp4_set_64value(elst_atom->duration, trak->movie_duration);
  1688.     ngx_mp4_set_64value(elst_atom->media_time, trak->prefix);
  1689.     ngx_mp4_set_16value(elst_atom->media_rate, 1);
  1690.     ngx_mp4_set_16value(elst_atom->reserved, 0);

  1691.     atom = &trak->elst_atom_buf;
  1692.     atom->temporary = 1;
  1693.     atom->pos = (u_char *) elst_atom;
  1694.     atom->last = (u_char *) elst_atom + sizeof(ngx_mp4_elst_atom_t);

  1695.     trak->out[NGX_HTTP_MP4_ELST_ATOM].buf = atom;

  1696.     trak->size += sizeof(ngx_mp4_edts_atom_t) + sizeof(ngx_mp4_elst_atom_t);
  1697. }


  1698. static void
  1699. ngx_http_mp4_update_stbl_atom(ngx_http_mp4_file_t *mp4,
  1700.     ngx_http_mp4_trak_t *trak)
  1701. {
  1702.     ngx_buf_t  *atom;

  1703.     trak->size += sizeof(ngx_mp4_atom_header_t);
  1704.     atom = &trak->stbl_atom_buf;
  1705.     ngx_mp4_set_32value(atom->pos, trak->size);
  1706. }


  1707. typedef struct {
  1708.     u_char    size[4];
  1709.     u_char    name[4];
  1710.     u_char    version[1];
  1711.     u_char    flags[3];
  1712.     u_char    entries[4];

  1713.     u_char    media_size[4];
  1714.     u_char    media_name[4];
  1715. } ngx_mp4_stsd_atom_t;


  1716. static ngx_int_t
  1717. ngx_http_mp4_read_stsd_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1718. {
  1719.     u_char               *atom_header, *atom_table;
  1720.     size_t                atom_size;
  1721.     ngx_buf_t            *atom;
  1722.     ngx_mp4_stsd_atom_t  *stsd_atom;
  1723.     ngx_http_mp4_trak_t  *trak;

  1724.     /* sample description atom */

  1725.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsd atom");

  1726.     atom_header = ngx_mp4_atom_header(mp4);
  1727.     stsd_atom = (ngx_mp4_stsd_atom_t *) atom_header;
  1728.     atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  1729.     atom_table = atom_header + atom_size;
  1730.     ngx_mp4_set_32value(stsd_atom->size, atom_size);
  1731.     ngx_mp4_set_atom_name(stsd_atom, 's', 't', 's', 'd');

  1732.     if (ngx_mp4_atom_data_size(ngx_mp4_stsd_atom_t) > atom_data_size) {
  1733.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1734.                       "\"%s\" mp4 stsd atom too small", mp4->file.name.data);
  1735.         return NGX_ERROR;
  1736.     }

  1737.     ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1738.                    "stsd entries:%uD, media:%*s",
  1739.                    ngx_mp4_get_32value(stsd_atom->entries),
  1740.                    (size_t) 4, stsd_atom->media_name);

  1741.     trak = ngx_mp4_last_trak(mp4);

  1742.     if (trak->out[NGX_HTTP_MP4_STSD_ATOM].buf) {
  1743.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1744.                       "duplicate mp4 stsd atom in \"%s\"", mp4->file.name.data);
  1745.         return NGX_ERROR;
  1746.     }

  1747.     atom = &trak->stsd_atom_buf;
  1748.     atom->temporary = 1;
  1749.     atom->pos = atom_header;
  1750.     atom->last = atom_table;

  1751.     trak->out[NGX_HTTP_MP4_STSD_ATOM].buf = atom;
  1752.     trak->size += atom_size;

  1753.     ngx_mp4_atom_next(mp4, atom_data_size);

  1754.     return NGX_OK;
  1755. }


  1756. typedef struct {
  1757.     u_char    size[4];
  1758.     u_char    name[4];
  1759.     u_char    version[1];
  1760.     u_char    flags[3];
  1761.     u_char    entries[4];
  1762. } ngx_mp4_stts_atom_t;

  1763. typedef struct {
  1764.     u_char    count[4];
  1765.     u_char    duration[4];
  1766. } ngx_mp4_stts_entry_t;


  1767. static ngx_int_t
  1768. ngx_http_mp4_read_stts_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1769. {
  1770.     u_char               *atom_header, *atom_table, *atom_end;
  1771.     uint32_t              entries;
  1772.     ngx_buf_t            *atom, *data;
  1773.     ngx_mp4_stts_atom_t  *stts_atom;
  1774.     ngx_http_mp4_trak_t  *trak;

  1775.     /* time-to-sample atom */

  1776.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stts atom");

  1777.     atom_header = ngx_mp4_atom_header(mp4);
  1778.     stts_atom = (ngx_mp4_stts_atom_t *) atom_header;
  1779.     ngx_mp4_set_atom_name(stts_atom, 's', 't', 't', 's');

  1780.     if (ngx_mp4_atom_data_size(ngx_mp4_stts_atom_t) > atom_data_size) {
  1781.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1782.                       "\"%s\" mp4 stts atom too small", mp4->file.name.data);
  1783.         return NGX_ERROR;
  1784.     }

  1785.     entries = ngx_mp4_get_32value(stts_atom->entries);

  1786.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1787.                    "mp4 time-to-sample entries:%uD", entries);

  1788.     if (ngx_mp4_atom_data_size(ngx_mp4_stts_atom_t)
  1789.         + entries * sizeof(ngx_mp4_stts_entry_t) > atom_data_size)
  1790.     {
  1791.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1792.                       "\"%s\" mp4 stts atom too small", mp4->file.name.data);
  1793.         return NGX_ERROR;
  1794.     }

  1795.     atom_table = atom_header + sizeof(ngx_mp4_stts_atom_t);
  1796.     atom_end = atom_table + entries * sizeof(ngx_mp4_stts_entry_t);

  1797.     trak = ngx_mp4_last_trak(mp4);

  1798.     if (trak->out[NGX_HTTP_MP4_STTS_ATOM].buf) {
  1799.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1800.                       "duplicate mp4 stts atom in \"%s\"", mp4->file.name.data);
  1801.         return NGX_ERROR;
  1802.     }

  1803.     trak->time_to_sample_entries = entries;

  1804.     atom = &trak->stts_atom_buf;
  1805.     atom->temporary = 1;
  1806.     atom->pos = atom_header;
  1807.     atom->last = atom_table;

  1808.     data = &trak->stts_data_buf;
  1809.     data->temporary = 1;
  1810.     data->pos = atom_table;
  1811.     data->last = atom_end;

  1812.     trak->out[NGX_HTTP_MP4_STTS_ATOM].buf = atom;
  1813.     trak->out[NGX_HTTP_MP4_STTS_DATA].buf = data;

  1814.     ngx_mp4_atom_next(mp4, atom_data_size);

  1815.     return NGX_OK;
  1816. }


  1817. static ngx_int_t
  1818. ngx_http_mp4_update_stts_atom(ngx_http_mp4_file_t *mp4,
  1819.     ngx_http_mp4_trak_t *trak)
  1820. {
  1821.     size_t                atom_size;
  1822.     ngx_buf_t            *atom, *data;
  1823.     ngx_mp4_stts_atom_t  *stts_atom;

  1824.     /*
  1825.      * mdia.minf.stbl.stts updating requires trak->timescale
  1826.      * from mdia.mdhd atom which may reside after mdia.minf
  1827.      */

  1828.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1829.                    "mp4 stts atom update");

  1830.     data = trak->out[NGX_HTTP_MP4_STTS_DATA].buf;

  1831.     if (data == NULL) {
  1832.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1833.                       "no mp4 stts atoms were found in \"%s\"",
  1834.                       mp4->file.name.data);
  1835.         return NGX_ERROR;
  1836.     }

  1837.     if (ngx_http_mp4_crop_stts_data(mp4, trak, 1) != NGX_OK) {
  1838.         return NGX_ERROR;
  1839.     }

  1840.     if (ngx_http_mp4_crop_stts_data(mp4, trak, 0) != NGX_OK) {
  1841.         return NGX_ERROR;
  1842.     }

  1843.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1844.                    "time-to-sample entries:%uD", trak->time_to_sample_entries);

  1845.     atom_size = sizeof(ngx_mp4_stts_atom_t) + (data->last - data->pos);
  1846.     trak->size += atom_size;

  1847.     atom = trak->out[NGX_HTTP_MP4_STTS_ATOM].buf;
  1848.     stts_atom = (ngx_mp4_stts_atom_t *) atom->pos;
  1849.     ngx_mp4_set_32value(stts_atom->size, atom_size);
  1850.     ngx_mp4_set_32value(stts_atom->entries, trak->time_to_sample_entries);

  1851.     return NGX_OK;
  1852. }


  1853. static ngx_int_t
  1854. ngx_http_mp4_crop_stts_data(ngx_http_mp4_file_t *mp4,
  1855.     ngx_http_mp4_trak_t *trak, ngx_uint_t start)
  1856. {
  1857.     uint32_t               count, duration, rest, key_prefix;
  1858.     uint64_t               start_time;
  1859.     ngx_buf_t             *data;
  1860.     ngx_uint_t             start_sample, entries, start_sec;
  1861.     ngx_mp4_stts_entry_t  *entry, *end;

  1862.     if (start) {
  1863.         start_sec = mp4->start;

  1864.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1865.                        "mp4 stts crop start_time:%ui", start_sec);

  1866.     } else if (mp4->length) {
  1867.         start_sec = mp4->length;

  1868.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1869.                        "mp4 stts crop end_time:%ui", start_sec);

  1870.     } else {
  1871.         return NGX_OK;
  1872.     }

  1873.     data = trak->out[NGX_HTTP_MP4_STTS_DATA].buf;

  1874.     start_time = (uint64_t) start_sec * trak->timescale / 1000 + trak->prefix;

  1875.     entries = trak->time_to_sample_entries;
  1876.     start_sample = 0;
  1877.     entry = (ngx_mp4_stts_entry_t *) data->pos;
  1878.     end = (ngx_mp4_stts_entry_t *) data->last;

  1879.     while (entry < end) {
  1880.         count = ngx_mp4_get_32value(entry->count);
  1881.         duration = ngx_mp4_get_32value(entry->duration);

  1882.         ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1883.                        "time:%uL, count:%uD, duration:%uD",
  1884.                        start_time, count, duration);

  1885.         if (start_time < (uint64_t) count * duration) {
  1886.             start_sample += (ngx_uint_t) (start_time / duration);
  1887.             rest = (uint32_t) (start_time / duration);
  1888.             goto found;
  1889.         }

  1890.         start_sample += count;
  1891.         start_time -= (uint64_t) count * duration;
  1892.         entries--;
  1893.         entry++;
  1894.     }

  1895.     if (start) {
  1896.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1897.                       "start time is out mp4 stts samples in \"%s\"",
  1898.                       mp4->file.name.data);

  1899.         return NGX_ERROR;

  1900.     } else {
  1901.         trak->end_sample = trak->start_sample + start_sample;

  1902.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1903.                        "end_sample:%ui", trak->end_sample);

  1904.         return NGX_OK;
  1905.     }

  1906. found:

  1907.     if (start) {
  1908.         key_prefix = ngx_http_mp4_seek_key_frame(mp4, trak, start_sample);

  1909.         start_sample -= key_prefix;

  1910.         while (rest < key_prefix) {
  1911.             trak->prefix += rest * duration;
  1912.             key_prefix -= rest;

  1913.             entry--;
  1914.             entries++;

  1915.             count = ngx_mp4_get_32value(entry->count);
  1916.             duration = ngx_mp4_get_32value(entry->duration);
  1917.             rest = count;
  1918.         }

  1919.         trak->prefix += key_prefix * duration;
  1920.         trak->duration += trak->prefix;
  1921.         rest -= key_prefix;

  1922.         ngx_mp4_set_32value(entry->count, count - rest);
  1923.         data->pos = (u_char *) entry;
  1924.         trak->time_to_sample_entries = entries;
  1925.         trak->start_sample = start_sample;

  1926.         ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1927.                        "start_sample:%ui, new count:%uD",
  1928.                        trak->start_sample, count - rest);

  1929.     } else {
  1930.         ngx_mp4_set_32value(entry->count, rest);
  1931.         data->last = (u_char *) (entry + 1);
  1932.         trak->time_to_sample_entries -= entries - 1;
  1933.         trak->end_sample = trak->start_sample + start_sample;

  1934.         ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1935.                        "end_sample:%ui, new count:%uD",
  1936.                        trak->end_sample, rest);
  1937.     }

  1938.     return NGX_OK;
  1939. }


  1940. static uint32_t
  1941. ngx_http_mp4_seek_key_frame(ngx_http_mp4_file_t *mp4, ngx_http_mp4_trak_t *trak,
  1942.     uint32_t start_sample)
  1943. {
  1944.     uint32_t              key_prefix, sample, *entry, *end;
  1945.     ngx_buf_t            *data;
  1946.     ngx_http_mp4_conf_t  *conf;

  1947.     conf = ngx_http_get_module_loc_conf(mp4->request, ngx_http_mp4_module);
  1948.     if (!conf->start_key_frame) {
  1949.         return 0;
  1950.     }

  1951.     data = trak->out[NGX_HTTP_MP4_STSS_DATA].buf;
  1952.     if (data == NULL) {
  1953.         return 0;
  1954.     }

  1955.     entry = (uint32_t *) data->pos;
  1956.     end = (uint32_t *) data->last;

  1957.     /* sync samples starts from 1 */
  1958.     start_sample++;

  1959.     key_prefix = 0;

  1960.     while (entry < end) {
  1961.         sample = ngx_mp4_get_32value(entry);
  1962.         if (sample > start_sample) {
  1963.             break;
  1964.         }

  1965.         key_prefix = start_sample - sample;
  1966.         entry++;
  1967.     }

  1968.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1969.                    "mp4 key frame prefix:%uD", key_prefix);

  1970.     return key_prefix;
  1971. }


  1972. typedef struct {
  1973.     u_char    size[4];
  1974.     u_char    name[4];
  1975.     u_char    version[1];
  1976.     u_char    flags[3];
  1977.     u_char    entries[4];
  1978. } ngx_http_mp4_stss_atom_t;


  1979. static ngx_int_t
  1980. ngx_http_mp4_read_stss_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  1981. {
  1982.     u_char                    *atom_header, *atom_table, *atom_end;
  1983.     uint32_t                   entries;
  1984.     ngx_buf_t                 *atom, *data;
  1985.     ngx_http_mp4_trak_t       *trak;
  1986.     ngx_http_mp4_stss_atom_t  *stss_atom;

  1987.     /* sync samples atom */

  1988.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stss atom");

  1989.     atom_header = ngx_mp4_atom_header(mp4);
  1990.     stss_atom = (ngx_http_mp4_stss_atom_t *) atom_header;
  1991.     ngx_mp4_set_atom_name(stss_atom, 's', 't', 's', 's');

  1992.     if (ngx_mp4_atom_data_size(ngx_http_mp4_stss_atom_t) > atom_data_size) {
  1993.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  1994.                       "\"%s\" mp4 stss atom too small", mp4->file.name.data);
  1995.         return NGX_ERROR;
  1996.     }

  1997.     entries = ngx_mp4_get_32value(stss_atom->entries);

  1998.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  1999.                    "sync sample entries:%uD", entries);

  2000.     trak = ngx_mp4_last_trak(mp4);

  2001.     if (trak->out[NGX_HTTP_MP4_STSS_ATOM].buf) {
  2002.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2003.                       "duplicate mp4 stss atom in \"%s\"", mp4->file.name.data);
  2004.         return NGX_ERROR;
  2005.     }

  2006.     trak->sync_samples_entries = entries;

  2007.     atom_table = atom_header + sizeof(ngx_http_mp4_stss_atom_t);

  2008.     atom = &trak->stss_atom_buf;
  2009.     atom->temporary = 1;
  2010.     atom->pos = atom_header;
  2011.     atom->last = atom_table;

  2012.     if (ngx_mp4_atom_data_size(ngx_http_mp4_stss_atom_t)
  2013.         + entries * sizeof(uint32_t) > atom_data_size)
  2014.     {
  2015.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2016.                       "\"%s\" mp4 stss atom too small", mp4->file.name.data);
  2017.         return NGX_ERROR;
  2018.     }

  2019.     atom_end = atom_table + entries * sizeof(uint32_t);

  2020.     data = &trak->stss_data_buf;
  2021.     data->temporary = 1;
  2022.     data->pos = atom_table;
  2023.     data->last = atom_end;

  2024.     trak->out[NGX_HTTP_MP4_STSS_ATOM].buf = atom;
  2025.     trak->out[NGX_HTTP_MP4_STSS_DATA].buf = data;

  2026.     ngx_mp4_atom_next(mp4, atom_data_size);

  2027.     return NGX_OK;
  2028. }


  2029. static ngx_int_t
  2030. ngx_http_mp4_update_stss_atom(ngx_http_mp4_file_t *mp4,
  2031.     ngx_http_mp4_trak_t *trak)
  2032. {
  2033.     size_t                     atom_size;
  2034.     uint32_t                   sample, start_sample, *entry, *end;
  2035.     ngx_buf_t                 *atom, *data;
  2036.     ngx_http_mp4_stss_atom_t  *stss_atom;

  2037.     /*
  2038.      * mdia.minf.stbl.stss updating requires trak->start_sample
  2039.      * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
  2040.      * atom which may reside after mdia.minf
  2041.      */

  2042.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2043.                    "mp4 stss atom update");

  2044.     data = trak->out[NGX_HTTP_MP4_STSS_DATA].buf;

  2045.     if (data == NULL) {
  2046.         return NGX_OK;
  2047.     }

  2048.     ngx_http_mp4_crop_stss_data(mp4, trak, 1);
  2049.     ngx_http_mp4_crop_stss_data(mp4, trak, 0);

  2050.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2051.                    "sync sample entries:%uD", trak->sync_samples_entries);

  2052.     if (trak->sync_samples_entries) {
  2053.         entry = (uint32_t *) data->pos;
  2054.         end = (uint32_t *) data->last;

  2055.         start_sample = trak->start_sample;

  2056.         while (entry < end) {
  2057.             sample = ngx_mp4_get_32value(entry);
  2058.             sample -= start_sample;
  2059.             ngx_mp4_set_32value(entry, sample);
  2060.             entry++;
  2061.         }

  2062.     } else {
  2063.         trak->out[NGX_HTTP_MP4_STSS_DATA].buf = NULL;
  2064.     }

  2065.     atom_size = sizeof(ngx_http_mp4_stss_atom_t) + (data->last - data->pos);
  2066.     trak->size += atom_size;

  2067.     atom = trak->out[NGX_HTTP_MP4_STSS_ATOM].buf;
  2068.     stss_atom = (ngx_http_mp4_stss_atom_t *) atom->pos;

  2069.     ngx_mp4_set_32value(stss_atom->size, atom_size);
  2070.     ngx_mp4_set_32value(stss_atom->entries, trak->sync_samples_entries);

  2071.     return NGX_OK;
  2072. }


  2073. static void
  2074. ngx_http_mp4_crop_stss_data(ngx_http_mp4_file_t *mp4,
  2075.     ngx_http_mp4_trak_t *trak, ngx_uint_t start)
  2076. {
  2077.     uint32_t     sample, start_sample, *entry, *end;
  2078.     ngx_buf_t   *data;
  2079.     ngx_uint_t   entries;

  2080.     /* sync samples starts from 1 */

  2081.     if (start) {
  2082.         start_sample = trak->start_sample + 1;

  2083.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2084.                        "mp4 stss crop start_sample:%uD", start_sample);

  2085.     } else if (mp4->length) {
  2086.         start_sample = trak->end_sample + 1;

  2087.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2088.                        "mp4 stss crop end_sample:%uD", start_sample);

  2089.     } else {
  2090.         return;
  2091.     }

  2092.     data = trak->out[NGX_HTTP_MP4_STSS_DATA].buf;

  2093.     entries = trak->sync_samples_entries;
  2094.     entry = (uint32_t *) data->pos;
  2095.     end = (uint32_t *) data->last;

  2096.     while (entry < end) {
  2097.         sample = ngx_mp4_get_32value(entry);

  2098.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2099.                        "sync:%uD", sample);

  2100.         if (sample >= start_sample) {
  2101.             goto found;
  2102.         }

  2103.         entries--;
  2104.         entry++;
  2105.     }

  2106.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2107.                    "sample is out of mp4 stss atom");

  2108. found:

  2109.     if (start) {
  2110.         data->pos = (u_char *) entry;
  2111.         trak->sync_samples_entries = entries;

  2112.     } else {
  2113.         data->last = (u_char *) entry;
  2114.         trak->sync_samples_entries -= entries;
  2115.     }
  2116. }


  2117. typedef struct {
  2118.     u_char    size[4];
  2119.     u_char    name[4];
  2120.     u_char    version[1];
  2121.     u_char    flags[3];
  2122.     u_char    entries[4];
  2123. } ngx_mp4_ctts_atom_t;

  2124. typedef struct {
  2125.     u_char    count[4];
  2126.     u_char    offset[4];
  2127. } ngx_mp4_ctts_entry_t;


  2128. static ngx_int_t
  2129. ngx_http_mp4_read_ctts_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  2130. {
  2131.     u_char               *atom_header, *atom_table, *atom_end;
  2132.     uint32_t              entries;
  2133.     ngx_buf_t            *atom, *data;
  2134.     ngx_mp4_ctts_atom_t  *ctts_atom;
  2135.     ngx_http_mp4_trak_t  *trak;

  2136.     /* composition offsets atom */

  2137.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 ctts atom");

  2138.     atom_header = ngx_mp4_atom_header(mp4);
  2139.     ctts_atom = (ngx_mp4_ctts_atom_t *) atom_header;
  2140.     ngx_mp4_set_atom_name(ctts_atom, 'c', 't', 't', 's');

  2141.     if (ngx_mp4_atom_data_size(ngx_mp4_ctts_atom_t) > atom_data_size) {
  2142.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2143.                       "\"%s\" mp4 ctts atom too small", mp4->file.name.data);
  2144.         return NGX_ERROR;
  2145.     }

  2146.     entries = ngx_mp4_get_32value(ctts_atom->entries);

  2147.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2148.                    "composition offset entries:%uD", entries);

  2149.     trak = ngx_mp4_last_trak(mp4);

  2150.     if (trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf) {
  2151.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2152.                       "duplicate mp4 ctts atom in \"%s\"", mp4->file.name.data);
  2153.         return NGX_ERROR;
  2154.     }

  2155.     trak->composition_offset_entries = entries;

  2156.     atom_table = atom_header + sizeof(ngx_mp4_ctts_atom_t);

  2157.     atom = &trak->ctts_atom_buf;
  2158.     atom->temporary = 1;
  2159.     atom->pos = atom_header;
  2160.     atom->last = atom_table;

  2161.     if (ngx_mp4_atom_data_size(ngx_mp4_ctts_atom_t)
  2162.         + entries * sizeof(ngx_mp4_ctts_entry_t) > atom_data_size)
  2163.     {
  2164.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2165.                       "\"%s\" mp4 ctts atom too small", mp4->file.name.data);
  2166.         return NGX_ERROR;
  2167.     }

  2168.     atom_end = atom_table + entries * sizeof(ngx_mp4_ctts_entry_t);

  2169.     data = &trak->ctts_data_buf;
  2170.     data->temporary = 1;
  2171.     data->pos = atom_table;
  2172.     data->last = atom_end;

  2173.     trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf = atom;
  2174.     trak->out[NGX_HTTP_MP4_CTTS_DATA].buf = data;

  2175.     ngx_mp4_atom_next(mp4, atom_data_size);

  2176.     return NGX_OK;
  2177. }


  2178. static void
  2179. ngx_http_mp4_update_ctts_atom(ngx_http_mp4_file_t *mp4,
  2180.     ngx_http_mp4_trak_t *trak)
  2181. {
  2182.     size_t                atom_size;
  2183.     ngx_buf_t            *atom, *data;
  2184.     ngx_mp4_ctts_atom_t  *ctts_atom;

  2185.     /*
  2186.      * mdia.minf.stbl.ctts updating requires trak->start_sample
  2187.      * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
  2188.      * atom which may reside after mdia.minf
  2189.      */

  2190.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2191.                    "mp4 ctts atom update");

  2192.     data = trak->out[NGX_HTTP_MP4_CTTS_DATA].buf;

  2193.     if (data == NULL) {
  2194.         return;
  2195.     }

  2196.     ngx_http_mp4_crop_ctts_data(mp4, trak, 1);
  2197.     ngx_http_mp4_crop_ctts_data(mp4, trak, 0);

  2198.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2199.                    "composition offset entries:%uD",
  2200.                    trak->composition_offset_entries);

  2201.     if (trak->composition_offset_entries == 0) {
  2202.         trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf = NULL;
  2203.         trak->out[NGX_HTTP_MP4_CTTS_DATA].buf = NULL;
  2204.         return;
  2205.     }

  2206.     atom_size = sizeof(ngx_mp4_ctts_atom_t) + (data->last - data->pos);
  2207.     trak->size += atom_size;

  2208.     atom = trak->out[NGX_HTTP_MP4_CTTS_ATOM].buf;
  2209.     ctts_atom = (ngx_mp4_ctts_atom_t *) atom->pos;

  2210.     ngx_mp4_set_32value(ctts_atom->size, atom_size);
  2211.     ngx_mp4_set_32value(ctts_atom->entries, trak->composition_offset_entries);

  2212.     return;
  2213. }


  2214. static void
  2215. ngx_http_mp4_crop_ctts_data(ngx_http_mp4_file_t *mp4,
  2216.     ngx_http_mp4_trak_t *trak, ngx_uint_t start)
  2217. {
  2218.     uint32_t               count, start_sample, rest;
  2219.     ngx_buf_t             *data;
  2220.     ngx_uint_t             entries;
  2221.     ngx_mp4_ctts_entry_t  *entry, *end;

  2222.     /* sync samples starts from 1 */

  2223.     if (start) {
  2224.         start_sample = trak->start_sample + 1;

  2225.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2226.                        "mp4 ctts crop start_sample:%uD", start_sample);

  2227.     } else if (mp4->length) {
  2228.         start_sample = trak->end_sample - trak->start_sample + 1;

  2229.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2230.                        "mp4 ctts crop end_sample:%uD", start_sample);

  2231.     } else {
  2232.         return;
  2233.     }

  2234.     data = trak->out[NGX_HTTP_MP4_CTTS_DATA].buf;

  2235.     entries = trak->composition_offset_entries;
  2236.     entry = (ngx_mp4_ctts_entry_t *) data->pos;
  2237.     end = (ngx_mp4_ctts_entry_t *) data->last;

  2238.     while (entry < end) {
  2239.         count = ngx_mp4_get_32value(entry->count);

  2240.         ngx_log_debug3(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2241.                        "sample:%uD, count:%uD, offset:%uD",
  2242.                        start_sample, count, ngx_mp4_get_32value(entry->offset));

  2243.         if (start_sample <= count) {
  2244.             rest = start_sample - 1;
  2245.             goto found;
  2246.         }

  2247.         start_sample -= count;
  2248.         entries--;
  2249.         entry++;
  2250.     }

  2251.     if (start) {
  2252.         data->pos = (u_char *) end;
  2253.         trak->composition_offset_entries = 0;
  2254.     }

  2255.     return;

  2256. found:

  2257.     if (start) {
  2258.         ngx_mp4_set_32value(entry->count, count - rest);
  2259.         data->pos = (u_char *) entry;
  2260.         trak->composition_offset_entries = entries;

  2261.     } else {
  2262.         ngx_mp4_set_32value(entry->count, rest);
  2263.         data->last = (u_char *) (entry + 1);
  2264.         trak->composition_offset_entries -= entries - 1;
  2265.     }
  2266. }


  2267. typedef struct {
  2268.     u_char    size[4];
  2269.     u_char    name[4];
  2270.     u_char    version[1];
  2271.     u_char    flags[3];
  2272.     u_char    entries[4];
  2273. } ngx_mp4_stsc_atom_t;


  2274. static ngx_int_t
  2275. ngx_http_mp4_read_stsc_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  2276. {
  2277.     u_char               *atom_header, *atom_table, *atom_end;
  2278.     uint32_t              entries;
  2279.     ngx_buf_t            *atom, *data;
  2280.     ngx_mp4_stsc_atom_t  *stsc_atom;
  2281.     ngx_http_mp4_trak_t  *trak;

  2282.     /* sample-to-chunk atom */

  2283.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsc atom");

  2284.     atom_header = ngx_mp4_atom_header(mp4);
  2285.     stsc_atom = (ngx_mp4_stsc_atom_t *) atom_header;
  2286.     ngx_mp4_set_atom_name(stsc_atom, 's', 't', 's', 'c');

  2287.     if (ngx_mp4_atom_data_size(ngx_mp4_stsc_atom_t) > atom_data_size) {
  2288.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2289.                       "\"%s\" mp4 stsc atom too small", mp4->file.name.data);
  2290.         return NGX_ERROR;
  2291.     }

  2292.     entries = ngx_mp4_get_32value(stsc_atom->entries);

  2293.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2294.                    "sample-to-chunk entries:%uD", entries);

  2295.     if (ngx_mp4_atom_data_size(ngx_mp4_stsc_atom_t)
  2296.         + entries * sizeof(ngx_mp4_stsc_entry_t) > atom_data_size)
  2297.     {
  2298.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2299.                       "\"%s\" mp4 stsc atom too small", mp4->file.name.data);
  2300.         return NGX_ERROR;
  2301.     }

  2302.     atom_table = atom_header + sizeof(ngx_mp4_stsc_atom_t);
  2303.     atom_end = atom_table + entries * sizeof(ngx_mp4_stsc_entry_t);

  2304.     trak = ngx_mp4_last_trak(mp4);

  2305.     if (trak->out[NGX_HTTP_MP4_STSC_ATOM].buf) {
  2306.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2307.                       "duplicate mp4 stsc atom in \"%s\"", mp4->file.name.data);
  2308.         return NGX_ERROR;
  2309.     }

  2310.     trak->sample_to_chunk_entries = entries;

  2311.     atom = &trak->stsc_atom_buf;
  2312.     atom->temporary = 1;
  2313.     atom->pos = atom_header;
  2314.     atom->last = atom_table;

  2315.     data = &trak->stsc_data_buf;
  2316.     data->temporary = 1;
  2317.     data->pos = atom_table;
  2318.     data->last = atom_end;

  2319.     trak->out[NGX_HTTP_MP4_STSC_ATOM].buf = atom;
  2320.     trak->out[NGX_HTTP_MP4_STSC_DATA].buf = data;

  2321.     ngx_mp4_atom_next(mp4, atom_data_size);

  2322.     return NGX_OK;
  2323. }


  2324. static ngx_int_t
  2325. ngx_http_mp4_update_stsc_atom(ngx_http_mp4_file_t *mp4,
  2326.     ngx_http_mp4_trak_t *trak)
  2327. {
  2328.     size_t                 atom_size;
  2329.     uint32_t               chunk;
  2330.     ngx_buf_t             *atom, *data;
  2331.     ngx_mp4_stsc_atom_t   *stsc_atom;
  2332.     ngx_mp4_stsc_entry_t  *entry, *end;

  2333.     /*
  2334.      * mdia.minf.stbl.stsc updating requires trak->start_sample
  2335.      * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
  2336.      * atom which may reside after mdia.minf
  2337.      */

  2338.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2339.                    "mp4 stsc atom update");

  2340.     data = trak->out[NGX_HTTP_MP4_STSC_DATA].buf;

  2341.     if (data == NULL) {
  2342.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2343.                       "no mp4 stsc atoms were found in \"%s\"",
  2344.                       mp4->file.name.data);
  2345.         return NGX_ERROR;
  2346.     }

  2347.     if (trak->sample_to_chunk_entries == 0) {
  2348.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2349.                       "zero number of entries in stsc atom in \"%s\"",
  2350.                       mp4->file.name.data);
  2351.         return NGX_ERROR;
  2352.     }

  2353.     if (ngx_http_mp4_crop_stsc_data(mp4, trak, 1) != NGX_OK) {
  2354.         return NGX_ERROR;
  2355.     }

  2356.     if (ngx_http_mp4_crop_stsc_data(mp4, trak, 0) != NGX_OK) {
  2357.         return NGX_ERROR;
  2358.     }

  2359.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2360.                    "sample-to-chunk entries:%uD",
  2361.                    trak->sample_to_chunk_entries);

  2362.     entry = (ngx_mp4_stsc_entry_t *) data->pos;
  2363.     end = (ngx_mp4_stsc_entry_t *) data->last;

  2364.     while (entry < end) {
  2365.         chunk = ngx_mp4_get_32value(entry->chunk);
  2366.         chunk -= trak->start_chunk;
  2367.         ngx_mp4_set_32value(entry->chunk, chunk);
  2368.         entry++;
  2369.     }

  2370.     atom_size = sizeof(ngx_mp4_stsc_atom_t)
  2371.                 + trak->sample_to_chunk_entries * sizeof(ngx_mp4_stsc_entry_t);

  2372.     trak->size += atom_size;

  2373.     atom = trak->out[NGX_HTTP_MP4_STSC_ATOM].buf;
  2374.     stsc_atom = (ngx_mp4_stsc_atom_t *) atom->pos;

  2375.     ngx_mp4_set_32value(stsc_atom->size, atom_size);
  2376.     ngx_mp4_set_32value(stsc_atom->entries, trak->sample_to_chunk_entries);

  2377.     return NGX_OK;
  2378. }


  2379. static ngx_int_t
  2380. ngx_http_mp4_crop_stsc_data(ngx_http_mp4_file_t *mp4,
  2381.     ngx_http_mp4_trak_t *trak, ngx_uint_t start)
  2382. {
  2383.     uint64_t               n;
  2384.     uint32_t               start_sample, chunk, samples, id, next_chunk,
  2385.                            prev_samples;
  2386.     ngx_buf_t             *data, *buf;
  2387.     ngx_uint_t             entries, target_chunk, chunk_samples;
  2388.     ngx_mp4_stsc_entry_t  *entry, *end, *first;

  2389.     entries = trak->sample_to_chunk_entries - 1;

  2390.     if (start) {
  2391.         start_sample = (uint32_t) trak->start_sample;

  2392.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2393.                        "mp4 stsc crop start_sample:%uD", start_sample);

  2394.     } else if (mp4->length) {
  2395.         start_sample = (uint32_t) (trak->end_sample - trak->start_sample);
  2396.         samples = 0;

  2397.         data = trak->out[NGX_HTTP_MP4_STSC_START].buf;

  2398.         if (data) {
  2399.             entry = (ngx_mp4_stsc_entry_t *) data->pos;
  2400.             samples = ngx_mp4_get_32value(entry->samples);
  2401.             entries--;

  2402.             if (samples > start_sample) {
  2403.                 samples = start_sample;
  2404.                 ngx_mp4_set_32value(entry->samples, samples);
  2405.             }

  2406.             start_sample -= samples;
  2407.         }

  2408.         ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2409.                        "mp4 stsc crop end_sample:%uD, ext_samples:%uD",
  2410.                        start_sample, samples);

  2411.     } else {
  2412.         return NGX_OK;
  2413.     }

  2414.     data = trak->out[NGX_HTTP_MP4_STSC_DATA].buf;

  2415.     entry = (ngx_mp4_stsc_entry_t *) data->pos;
  2416.     end = (ngx_mp4_stsc_entry_t *) data->last;

  2417.     chunk = ngx_mp4_get_32value(entry->chunk);
  2418.     samples = ngx_mp4_get_32value(entry->samples);
  2419.     id = ngx_mp4_get_32value(entry->id);
  2420.     prev_samples = 0;
  2421.     entry++;

  2422.     while (entry < end) {

  2423.         next_chunk = ngx_mp4_get_32value(entry->chunk);

  2424.         if (next_chunk < chunk) {
  2425.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2426.                           "unordered mp4 stsc chunks in \"%s\"",
  2427.                           mp4->file.name.data);
  2428.             return NGX_ERROR;
  2429.         }

  2430.         ngx_log_debug5(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2431.                        "sample:%uD, chunk:%uD, chunks:%uD, "
  2432.                        "samples:%uD, id:%uD",
  2433.                        start_sample, chunk, next_chunk - chunk, samples, id);

  2434.         n = (uint64_t) (next_chunk - chunk) * samples;

  2435.         if (start_sample < n) {
  2436.             goto found;
  2437.         }

  2438.         start_sample -= n;

  2439.         if (next_chunk > chunk) {
  2440.             prev_samples = samples;
  2441.         }

  2442.         chunk = next_chunk;
  2443.         samples = ngx_mp4_get_32value(entry->samples);
  2444.         id = ngx_mp4_get_32value(entry->id);
  2445.         entries--;
  2446.         entry++;
  2447.     }

  2448.     next_chunk = trak->chunks + 1;

  2449.     if (next_chunk < chunk) {
  2450.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2451.                       "unordered mp4 stsc chunks in \"%s\"",
  2452.                       mp4->file.name.data);
  2453.         return NGX_ERROR;
  2454.     }

  2455.     ngx_log_debug4(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2456.                    "sample:%uD, chunk:%uD, chunks:%uD, samples:%uD",
  2457.                    start_sample, chunk, next_chunk - chunk, samples);

  2458.     n = (uint64_t) (next_chunk - chunk) * samples;

  2459.     if (start_sample > n) {
  2460.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2461.                       "%s time is out mp4 stsc chunks in \"%s\"",
  2462.                       start ? "start" : "end", mp4->file.name.data);
  2463.         return NGX_ERROR;
  2464.     }

  2465. found:

  2466.     entries++;
  2467.     entry--;

  2468.     if (samples == 0) {
  2469.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2470.                       "zero number of samples in \"%s\"",
  2471.                       mp4->file.name.data);
  2472.         return NGX_ERROR;
  2473.     }

  2474.     if (chunk == 0) {
  2475.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2476.                       "zero chunk in \"%s\"", mp4->file.name.data);
  2477.         return NGX_ERROR;
  2478.     }

  2479.     target_chunk = chunk - 1;
  2480.     target_chunk += start_sample / samples;
  2481.     chunk_samples = start_sample % samples;

  2482.     if (start) {
  2483.         data->pos = (u_char *) entry;

  2484.         trak->sample_to_chunk_entries = entries;
  2485.         trak->start_chunk = target_chunk;
  2486.         trak->start_chunk_samples = chunk_samples;

  2487.         ngx_mp4_set_32value(entry->chunk, trak->start_chunk + 1);

  2488.         samples -= chunk_samples;

  2489.         ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2490.                        "start_chunk:%ui, start_chunk_samples:%ui",
  2491.                        trak->start_chunk, trak->start_chunk_samples);

  2492.     } else {
  2493.         if (start_sample) {
  2494.             data->last = (u_char *) (entry + 1);
  2495.             trak->sample_to_chunk_entries -= entries - 1;
  2496.             trak->end_chunk_samples = samples;

  2497.         } else {
  2498.             data->last = (u_char *) entry;
  2499.             trak->sample_to_chunk_entries -= entries;
  2500.             trak->end_chunk_samples = prev_samples;
  2501.         }

  2502.         if (chunk_samples) {
  2503.             trak->end_chunk = target_chunk + 1;
  2504.             trak->end_chunk_samples = chunk_samples;

  2505.         } else {
  2506.             trak->end_chunk = target_chunk;
  2507.         }

  2508.         samples = chunk_samples;
  2509.         next_chunk = chunk + 1;

  2510.         ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2511.                        "end_chunk:%ui, end_chunk_samples:%ui",
  2512.                        trak->end_chunk, trak->end_chunk_samples);
  2513.     }

  2514.     if (chunk_samples && next_chunk - target_chunk == 2) {

  2515.         ngx_mp4_set_32value(entry->samples, samples);

  2516.     } else if (chunk_samples && start) {

  2517.         first = &trak->stsc_start_chunk_entry;
  2518.         ngx_mp4_set_32value(first->chunk, 1);
  2519.         ngx_mp4_set_32value(first->samples, samples);
  2520.         ngx_mp4_set_32value(first->id, id);

  2521.         buf = &trak->stsc_start_chunk_buf;
  2522.         buf->temporary = 1;
  2523.         buf->pos = (u_char *) first;
  2524.         buf->last = (u_char *) first + sizeof(ngx_mp4_stsc_entry_t);

  2525.         trak->out[NGX_HTTP_MP4_STSC_START].buf = buf;

  2526.         ngx_mp4_set_32value(entry->chunk, trak->start_chunk + 2);

  2527.         trak->sample_to_chunk_entries++;

  2528.     } else if (chunk_samples) {

  2529.         first = &trak->stsc_end_chunk_entry;
  2530.         ngx_mp4_set_32value(first->chunk, trak->end_chunk - trak->start_chunk);
  2531.         ngx_mp4_set_32value(first->samples, samples);
  2532.         ngx_mp4_set_32value(first->id, id);

  2533.         buf = &trak->stsc_end_chunk_buf;
  2534.         buf->temporary = 1;
  2535.         buf->pos = (u_char *) first;
  2536.         buf->last = (u_char *) first + sizeof(ngx_mp4_stsc_entry_t);

  2537.         trak->out[NGX_HTTP_MP4_STSC_END].buf = buf;

  2538.         trak->sample_to_chunk_entries++;
  2539.     }

  2540.     return NGX_OK;
  2541. }


  2542. typedef struct {
  2543.     u_char    size[4];
  2544.     u_char    name[4];
  2545.     u_char    version[1];
  2546.     u_char    flags[3];
  2547.     u_char    uniform_size[4];
  2548.     u_char    entries[4];
  2549. } ngx_mp4_stsz_atom_t;


  2550. static ngx_int_t
  2551. ngx_http_mp4_read_stsz_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  2552. {
  2553.     u_char               *atom_header, *atom_table, *atom_end;
  2554.     size_t                atom_size;
  2555.     uint32_t              entries, size;
  2556.     ngx_buf_t            *atom, *data;
  2557.     ngx_mp4_stsz_atom_t  *stsz_atom;
  2558.     ngx_http_mp4_trak_t  *trak;

  2559.     /* sample sizes atom */

  2560.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stsz atom");

  2561.     atom_header = ngx_mp4_atom_header(mp4);
  2562.     stsz_atom = (ngx_mp4_stsz_atom_t *) atom_header;
  2563.     ngx_mp4_set_atom_name(stsz_atom, 's', 't', 's', 'z');

  2564.     if (ngx_mp4_atom_data_size(ngx_mp4_stsz_atom_t) > atom_data_size) {
  2565.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2566.                       "\"%s\" mp4 stsz atom too small", mp4->file.name.data);
  2567.         return NGX_ERROR;
  2568.     }

  2569.     size = ngx_mp4_get_32value(stsz_atom->uniform_size);
  2570.     entries = ngx_mp4_get_32value(stsz_atom->entries);

  2571.     ngx_log_debug2(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2572.                    "sample uniform size:%uD, entries:%uD", size, entries);

  2573.     trak = ngx_mp4_last_trak(mp4);

  2574.     if (trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf) {
  2575.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2576.                       "duplicate mp4 stsz atom in \"%s\"", mp4->file.name.data);
  2577.         return NGX_ERROR;
  2578.     }

  2579.     trak->sample_sizes_entries = entries;

  2580.     atom_table = atom_header + sizeof(ngx_mp4_stsz_atom_t);

  2581.     atom = &trak->stsz_atom_buf;
  2582.     atom->temporary = 1;
  2583.     atom->pos = atom_header;
  2584.     atom->last = atom_table;

  2585.     trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf = atom;

  2586.     if (size == 0) {
  2587.         if (ngx_mp4_atom_data_size(ngx_mp4_stsz_atom_t)
  2588.             + entries * sizeof(uint32_t) > atom_data_size)
  2589.         {
  2590.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2591.                           "\"%s\" mp4 stsz atom too small",
  2592.                           mp4->file.name.data);
  2593.             return NGX_ERROR;
  2594.         }

  2595.         atom_end = atom_table + entries * sizeof(uint32_t);

  2596.         data = &trak->stsz_data_buf;
  2597.         data->temporary = 1;
  2598.         data->pos = atom_table;
  2599.         data->last = atom_end;

  2600.         trak->out[NGX_HTTP_MP4_STSZ_DATA].buf = data;

  2601.     } else {
  2602.         /* if size != 0 then all samples are the same size */
  2603.         /* TODO : chunk samples */
  2604.         atom_size = sizeof(ngx_mp4_atom_header_t) + (size_t) atom_data_size;
  2605.         ngx_mp4_set_32value(atom_header, atom_size);
  2606.         trak->size += atom_size;
  2607.     }

  2608.     ngx_mp4_atom_next(mp4, atom_data_size);

  2609.     return NGX_OK;
  2610. }


  2611. static ngx_int_t
  2612. ngx_http_mp4_update_stsz_atom(ngx_http_mp4_file_t *mp4,
  2613.     ngx_http_mp4_trak_t *trak)
  2614. {
  2615.     size_t                atom_size;
  2616.     uint32_t             *pos, *end, entries;
  2617.     ngx_buf_t            *atom, *data;
  2618.     ngx_mp4_stsz_atom_t  *stsz_atom;

  2619.     /*
  2620.      * mdia.minf.stbl.stsz updating requires trak->start_sample
  2621.      * from mdia.minf.stbl.stts which depends on value from mdia.mdhd
  2622.      * atom which may reside after mdia.minf
  2623.      */

  2624.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2625.                    "mp4 stsz atom update");

  2626.     data = trak->out[NGX_HTTP_MP4_STSZ_DATA].buf;

  2627.     if (data) {
  2628.         entries = trak->sample_sizes_entries;

  2629.         if (trak->start_sample > entries) {
  2630.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2631.                           "start time is out mp4 stsz samples in \"%s\"",
  2632.                           mp4->file.name.data);
  2633.             return NGX_ERROR;
  2634.         }

  2635.         entries -= trak->start_sample;
  2636.         data->pos += trak->start_sample * sizeof(uint32_t);
  2637.         end = (uint32_t *) data->pos;

  2638.         for (pos = end - trak->start_chunk_samples; pos < end; pos++) {
  2639.             trak->start_chunk_samples_size += ngx_mp4_get_32value(pos);
  2640.         }

  2641.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2642.                        "chunk samples sizes:%uL",
  2643.                        trak->start_chunk_samples_size);

  2644.         if (trak->start_chunk_samples_size > (uint64_t) mp4->end) {
  2645.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2646.                           "too large mp4 start samples size in \"%s\"",
  2647.                           mp4->file.name.data);
  2648.             return NGX_ERROR;
  2649.         }

  2650.         if (mp4->length) {
  2651.             if (trak->end_sample - trak->start_sample > entries) {
  2652.                 ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2653.                               "end time is out mp4 stsz samples in \"%s\"",
  2654.                               mp4->file.name.data);
  2655.                 return NGX_ERROR;
  2656.             }

  2657.             entries = trak->end_sample - trak->start_sample;
  2658.             data->last = data->pos + entries * sizeof(uint32_t);
  2659.             end = (uint32_t *) data->last;

  2660.             for (pos = end - trak->end_chunk_samples; pos < end; pos++) {
  2661.                 trak->end_chunk_samples_size += ngx_mp4_get_32value(pos);
  2662.             }

  2663.             ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2664.                            "mp4 stsz end_chunk_samples_size:%uL",
  2665.                            trak->end_chunk_samples_size);

  2666.             if (trak->end_chunk_samples_size > (uint64_t) mp4->end) {
  2667.                 ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2668.                               "too large mp4 end samples size in \"%s\"",
  2669.                               mp4->file.name.data);
  2670.                 return NGX_ERROR;
  2671.             }
  2672.         }

  2673.         atom_size = sizeof(ngx_mp4_stsz_atom_t) + (data->last - data->pos);
  2674.         trak->size += atom_size;

  2675.         atom = trak->out[NGX_HTTP_MP4_STSZ_ATOM].buf;
  2676.         stsz_atom = (ngx_mp4_stsz_atom_t *) atom->pos;

  2677.         ngx_mp4_set_32value(stsz_atom->size, atom_size);
  2678.         ngx_mp4_set_32value(stsz_atom->entries, entries);
  2679.     }

  2680.     return NGX_OK;
  2681. }


  2682. typedef struct {
  2683.     u_char    size[4];
  2684.     u_char    name[4];
  2685.     u_char    version[1];
  2686.     u_char    flags[3];
  2687.     u_char    entries[4];
  2688. } ngx_mp4_stco_atom_t;


  2689. static ngx_int_t
  2690. ngx_http_mp4_read_stco_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  2691. {
  2692.     u_char               *atom_header, *atom_table, *atom_end;
  2693.     uint32_t              entries;
  2694.     ngx_buf_t            *atom, *data;
  2695.     ngx_mp4_stco_atom_t  *stco_atom;
  2696.     ngx_http_mp4_trak_t  *trak;

  2697.     /* chunk offsets atom */

  2698.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 stco atom");

  2699.     atom_header = ngx_mp4_atom_header(mp4);
  2700.     stco_atom = (ngx_mp4_stco_atom_t *) atom_header;
  2701.     ngx_mp4_set_atom_name(stco_atom, 's', 't', 'c', 'o');

  2702.     if (ngx_mp4_atom_data_size(ngx_mp4_stco_atom_t) > atom_data_size) {
  2703.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2704.                       "\"%s\" mp4 stco atom too small", mp4->file.name.data);
  2705.         return NGX_ERROR;
  2706.     }

  2707.     entries = ngx_mp4_get_32value(stco_atom->entries);

  2708.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "chunks:%uD", entries);

  2709.     if (ngx_mp4_atom_data_size(ngx_mp4_stco_atom_t)
  2710.         + entries * sizeof(uint32_t) > atom_data_size)
  2711.     {
  2712.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2713.                       "\"%s\" mp4 stco atom too small", mp4->file.name.data);
  2714.         return NGX_ERROR;
  2715.     }

  2716.     atom_table = atom_header + sizeof(ngx_mp4_stco_atom_t);
  2717.     atom_end = atom_table + entries * sizeof(uint32_t);

  2718.     trak = ngx_mp4_last_trak(mp4);

  2719.     if (trak->out[NGX_HTTP_MP4_STCO_ATOM].buf
  2720.         || trak->out[NGX_HTTP_MP4_CO64_ATOM].buf)
  2721.     {
  2722.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2723.                       "duplicate mp4 stco/co64 atom in \"%s\"",
  2724.                       mp4->file.name.data);
  2725.         return NGX_ERROR;
  2726.     }

  2727.     trak->chunks = entries;

  2728.     atom = &trak->stco_atom_buf;
  2729.     atom->temporary = 1;
  2730.     atom->pos = atom_header;
  2731.     atom->last = atom_table;

  2732.     data = &trak->stco_data_buf;
  2733.     data->temporary = 1;
  2734.     data->pos = atom_table;
  2735.     data->last = atom_end;

  2736.     trak->out[NGX_HTTP_MP4_STCO_ATOM].buf = atom;
  2737.     trak->out[NGX_HTTP_MP4_STCO_DATA].buf = data;

  2738.     ngx_mp4_atom_next(mp4, atom_data_size);

  2739.     return NGX_OK;
  2740. }


  2741. static ngx_int_t
  2742. ngx_http_mp4_update_stco_atom(ngx_http_mp4_file_t *mp4,
  2743.     ngx_http_mp4_trak_t *trak)
  2744. {
  2745.     size_t                atom_size;
  2746.     uint32_t              entries;
  2747.     uint64_t              chunk_offset, samples_size;
  2748.     ngx_buf_t            *atom, *data;
  2749.     ngx_mp4_stco_atom_t  *stco_atom;

  2750.     /*
  2751.      * mdia.minf.stbl.stco updating requires trak->start_chunk
  2752.      * from mdia.minf.stbl.stsc which depends on value from mdia.mdhd
  2753.      * atom which may reside after mdia.minf
  2754.      */

  2755.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2756.                    "mp4 stco atom update");

  2757.     data = trak->out[NGX_HTTP_MP4_STCO_DATA].buf;

  2758.     if (data == NULL) {
  2759.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2760.                       "no mp4 stco atoms were found in \"%s\"",
  2761.                       mp4->file.name.data);
  2762.         return NGX_ERROR;
  2763.     }

  2764.     if (trak->start_chunk > trak->chunks) {
  2765.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2766.                       "start time is out mp4 stco chunks in \"%s\"",
  2767.                       mp4->file.name.data);
  2768.         return NGX_ERROR;
  2769.     }

  2770.     data->pos += trak->start_chunk * sizeof(uint32_t);

  2771.     chunk_offset = ngx_mp4_get_32value(data->pos);
  2772.     samples_size = trak->start_chunk_samples_size;

  2773.     if (chunk_offset > (uint64_t) mp4->end - samples_size
  2774.         || chunk_offset + samples_size > NGX_MAX_UINT32_VALUE)
  2775.     {
  2776.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2777.                       "too large chunk offset in \"%s\"",
  2778.                       mp4->file.name.data);
  2779.         return NGX_ERROR;
  2780.     }

  2781.     trak->start_offset = chunk_offset + samples_size;
  2782.     ngx_mp4_set_32value(data->pos, trak->start_offset);

  2783.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2784.                    "start chunk offset:%O", trak->start_offset);

  2785.     if (mp4->length) {

  2786.         if (trak->end_chunk > trak->chunks) {
  2787.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2788.                           "end time is out mp4 stco chunks in \"%s\"",
  2789.                           mp4->file.name.data);
  2790.             return NGX_ERROR;
  2791.         }

  2792.         entries = trak->end_chunk - trak->start_chunk;
  2793.         data->last = data->pos + entries * sizeof(uint32_t);

  2794.         if (entries) {
  2795.             chunk_offset = ngx_mp4_get_32value(data->last - sizeof(uint32_t));
  2796.             samples_size = trak->end_chunk_samples_size;

  2797.             if (chunk_offset > (uint64_t) mp4->end - samples_size
  2798.                 || chunk_offset + samples_size > NGX_MAX_UINT32_VALUE)
  2799.             {
  2800.                 ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2801.                               "too large chunk offset in \"%s\"",
  2802.                               mp4->file.name.data);
  2803.                 return NGX_ERROR;
  2804.             }

  2805.             trak->end_offset = chunk_offset + samples_size;

  2806.             ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2807.                            "end chunk offset:%O", trak->end_offset);
  2808.         }

  2809.     } else {
  2810.         entries = trak->chunks - trak->start_chunk;
  2811.         trak->end_offset = mp4->mdat_data.buf->file_last;
  2812.     }

  2813.     if (entries == 0) {
  2814.         trak->start_offset = mp4->end;
  2815.         trak->end_offset = 0;
  2816.     }

  2817.     atom_size = sizeof(ngx_mp4_stco_atom_t) + (data->last - data->pos);
  2818.     trak->size += atom_size;

  2819.     atom = trak->out[NGX_HTTP_MP4_STCO_ATOM].buf;
  2820.     stco_atom = (ngx_mp4_stco_atom_t *) atom->pos;

  2821.     ngx_mp4_set_32value(stco_atom->size, atom_size);
  2822.     ngx_mp4_set_32value(stco_atom->entries, entries);

  2823.     return NGX_OK;
  2824. }


  2825. static void
  2826. ngx_http_mp4_adjust_stco_atom(ngx_http_mp4_file_t *mp4,
  2827.     ngx_http_mp4_trak_t *trak, int32_t adjustment)
  2828. {
  2829.     uint32_t    offset, *entry, *end;
  2830.     ngx_buf_t  *data;

  2831.     /*
  2832.      * moov.trak.mdia.minf.stbl.stco adjustment requires
  2833.      * minimal start offset of all traks and new moov atom size
  2834.      */

  2835.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2836.                    "mp4 stco atom adjustment");

  2837.     data = trak->out[NGX_HTTP_MP4_STCO_DATA].buf;
  2838.     entry = (uint32_t *) data->pos;
  2839.     end = (uint32_t *) data->last;

  2840.     while (entry < end) {
  2841.         offset = ngx_mp4_get_32value(entry);
  2842.         offset += adjustment;
  2843.         ngx_mp4_set_32value(entry, offset);
  2844.         entry++;
  2845.     }
  2846. }


  2847. typedef struct {
  2848.     u_char    size[4];
  2849.     u_char    name[4];
  2850.     u_char    version[1];
  2851.     u_char    flags[3];
  2852.     u_char    entries[4];
  2853. } ngx_mp4_co64_atom_t;


  2854. static ngx_int_t
  2855. ngx_http_mp4_read_co64_atom(ngx_http_mp4_file_t *mp4, uint64_t atom_data_size)
  2856. {
  2857.     u_char               *atom_header, *atom_table, *atom_end;
  2858.     uint32_t              entries;
  2859.     ngx_buf_t            *atom, *data;
  2860.     ngx_mp4_co64_atom_t  *co64_atom;
  2861.     ngx_http_mp4_trak_t  *trak;

  2862.     /* chunk offsets atom */

  2863.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "mp4 co64 atom");

  2864.     atom_header = ngx_mp4_atom_header(mp4);
  2865.     co64_atom = (ngx_mp4_co64_atom_t *) atom_header;
  2866.     ngx_mp4_set_atom_name(co64_atom, 'c', 'o', '6', '4');

  2867.     if (ngx_mp4_atom_data_size(ngx_mp4_co64_atom_t) > atom_data_size) {
  2868.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2869.                       "\"%s\" mp4 co64 atom too small", mp4->file.name.data);
  2870.         return NGX_ERROR;
  2871.     }

  2872.     entries = ngx_mp4_get_32value(co64_atom->entries);

  2873.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0, "chunks:%uD", entries);

  2874.     if (ngx_mp4_atom_data_size(ngx_mp4_co64_atom_t)
  2875.         + entries * sizeof(uint64_t) > atom_data_size)
  2876.     {
  2877.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2878.                       "\"%s\" mp4 co64 atom too small", mp4->file.name.data);
  2879.         return NGX_ERROR;
  2880.     }

  2881.     atom_table = atom_header + sizeof(ngx_mp4_co64_atom_t);
  2882.     atom_end = atom_table + entries * sizeof(uint64_t);

  2883.     trak = ngx_mp4_last_trak(mp4);

  2884.     if (trak->out[NGX_HTTP_MP4_STCO_ATOM].buf
  2885.         || trak->out[NGX_HTTP_MP4_CO64_ATOM].buf)
  2886.     {
  2887.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2888.                       "duplicate mp4 stco/co64 atom in \"%s\"",
  2889.                       mp4->file.name.data);
  2890.         return NGX_ERROR;
  2891.     }

  2892.     trak->chunks = entries;

  2893.     atom = &trak->co64_atom_buf;
  2894.     atom->temporary = 1;
  2895.     atom->pos = atom_header;
  2896.     atom->last = atom_table;

  2897.     data = &trak->co64_data_buf;
  2898.     data->temporary = 1;
  2899.     data->pos = atom_table;
  2900.     data->last = atom_end;

  2901.     trak->out[NGX_HTTP_MP4_CO64_ATOM].buf = atom;
  2902.     trak->out[NGX_HTTP_MP4_CO64_DATA].buf = data;

  2903.     ngx_mp4_atom_next(mp4, atom_data_size);

  2904.     return NGX_OK;
  2905. }


  2906. static ngx_int_t
  2907. ngx_http_mp4_update_co64_atom(ngx_http_mp4_file_t *mp4,
  2908.     ngx_http_mp4_trak_t *trak)
  2909. {
  2910.     size_t                atom_size;
  2911.     uint64_t              entries, chunk_offset, samples_size;
  2912.     ngx_buf_t            *atom, *data;
  2913.     ngx_mp4_co64_atom_t  *co64_atom;

  2914.     /*
  2915.      * mdia.minf.stbl.co64 updating requires trak->start_chunk
  2916.      * from mdia.minf.stbl.stsc which depends on value from mdia.mdhd
  2917.      * atom which may reside after mdia.minf
  2918.      */

  2919.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2920.                    "mp4 co64 atom update");

  2921.     data = trak->out[NGX_HTTP_MP4_CO64_DATA].buf;

  2922.     if (data == NULL) {
  2923.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2924.                       "no mp4 co64 atoms were found in \"%s\"",
  2925.                       mp4->file.name.data);
  2926.         return NGX_ERROR;
  2927.     }

  2928.     if (trak->start_chunk > trak->chunks) {
  2929.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2930.                       "start time is out mp4 co64 chunks in \"%s\"",
  2931.                       mp4->file.name.data);
  2932.         return NGX_ERROR;
  2933.     }

  2934.     data->pos += trak->start_chunk * sizeof(uint64_t);

  2935.     chunk_offset = ngx_mp4_get_64value(data->pos);
  2936.     samples_size = trak->start_chunk_samples_size;

  2937.     if (chunk_offset > (uint64_t) mp4->end - samples_size) {
  2938.         ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2939.                       "too large chunk offset in \"%s\"",
  2940.                       mp4->file.name.data);
  2941.         return NGX_ERROR;
  2942.     }

  2943.     trak->start_offset = chunk_offset + samples_size;
  2944.     ngx_mp4_set_64value(data->pos, trak->start_offset);

  2945.     ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2946.                    "start chunk offset:%O", trak->start_offset);

  2947.     if (mp4->length) {

  2948.         if (trak->end_chunk > trak->chunks) {
  2949.             ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2950.                           "end time is out mp4 co64 chunks in \"%s\"",
  2951.                           mp4->file.name.data);
  2952.             return NGX_ERROR;
  2953.         }

  2954.         entries = trak->end_chunk - trak->start_chunk;
  2955.         data->last = data->pos + entries * sizeof(uint64_t);

  2956.         if (entries) {
  2957.             chunk_offset = ngx_mp4_get_64value(data->last - sizeof(uint64_t));
  2958.             samples_size = trak->end_chunk_samples_size;

  2959.             if (chunk_offset > (uint64_t) mp4->end - samples_size) {
  2960.                 ngx_log_error(NGX_LOG_ERR, mp4->file.log, 0,
  2961.                               "too large chunk offset in \"%s\"",
  2962.                               mp4->file.name.data);
  2963.                 return NGX_ERROR;
  2964.             }

  2965.             trak->end_offset = chunk_offset + samples_size;

  2966.             ngx_log_debug1(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2967.                            "end chunk offset:%O", trak->end_offset);
  2968.         }

  2969.     } else {
  2970.         entries = trak->chunks - trak->start_chunk;
  2971.         trak->end_offset = mp4->mdat_data.buf->file_last;
  2972.     }

  2973.     if (entries == 0) {
  2974.         trak->start_offset = mp4->end;
  2975.         trak->end_offset = 0;
  2976.     }

  2977.     atom_size = sizeof(ngx_mp4_co64_atom_t) + (data->last - data->pos);
  2978.     trak->size += atom_size;

  2979.     atom = trak->out[NGX_HTTP_MP4_CO64_ATOM].buf;
  2980.     co64_atom = (ngx_mp4_co64_atom_t *) atom->pos;

  2981.     ngx_mp4_set_32value(co64_atom->size, atom_size);
  2982.     ngx_mp4_set_32value(co64_atom->entries, entries);

  2983.     return NGX_OK;
  2984. }


  2985. static void
  2986. ngx_http_mp4_adjust_co64_atom(ngx_http_mp4_file_t *mp4,
  2987.     ngx_http_mp4_trak_t *trak, off_t adjustment)
  2988. {
  2989.     uint64_t    offset, *entry, *end;
  2990.     ngx_buf_t  *data;

  2991.     /*
  2992.      * moov.trak.mdia.minf.stbl.co64 adjustment requires
  2993.      * minimal start offset of all traks and new moov atom size
  2994.      */

  2995.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, mp4->file.log, 0,
  2996.                    "mp4 co64 atom adjustment");

  2997.     data = trak->out[NGX_HTTP_MP4_CO64_DATA].buf;
  2998.     entry = (uint64_t *) data->pos;
  2999.     end = (uint64_t *) data->last;

  3000.     while (entry < end) {
  3001.         offset = ngx_mp4_get_64value(entry);
  3002.         offset += adjustment;
  3003.         ngx_mp4_set_64value(entry, offset);
  3004.         entry++;
  3005.     }
  3006. }


  3007. static char *
  3008. ngx_http_mp4(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
  3009. {
  3010.     ngx_http_core_loc_conf_t  *clcf;

  3011.     clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
  3012.     clcf->handler = ngx_http_mp4_handler;

  3013.     return NGX_CONF_OK;
  3014. }


  3015. static void *
  3016. ngx_http_mp4_create_conf(ngx_conf_t *cf)
  3017. {
  3018.     ngx_http_mp4_conf_t  *conf;

  3019.     conf = ngx_palloc(cf->pool, sizeof(ngx_http_mp4_conf_t));
  3020.     if (conf == NULL) {
  3021.         return NULL;
  3022.     }

  3023.     conf->buffer_size = NGX_CONF_UNSET_SIZE;
  3024.     conf->max_buffer_size = NGX_CONF_UNSET_SIZE;
  3025.     conf->start_key_frame = NGX_CONF_UNSET;

  3026.     return conf;
  3027. }


  3028. static char *
  3029. ngx_http_mp4_merge_conf(ngx_conf_t *cf, void *parent, void *child)
  3030. {
  3031.     ngx_http_mp4_conf_t *prev = parent;
  3032.     ngx_http_mp4_conf_t *conf = child;

  3033.     ngx_conf_merge_size_value(conf->buffer_size, prev->buffer_size, 512 * 1024);
  3034.     ngx_conf_merge_size_value(conf->max_buffer_size, prev->max_buffer_size,
  3035.                               10 * 1024 * 1024);
  3036.     ngx_conf_merge_value(conf->start_key_frame, prev->start_key_frame, 0);

  3037.     return NGX_CONF_OK;
  3038. }