c3a3a218ef903b2ad5d01eaf06c6fa2455fc04bd
[libav.git] / tools / pktdumper.c
1 /*
2 * Copyright (c) 2005 Francois Revol
3 *
4 * This file is part of Libav.
5 *
6 * Libav is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * Libav is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with Libav; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <limits.h>
22 #include <fcntl.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <unistd.h>
27
28 #include "libavformat/avformat.h"
29
30 #define PKTFILESUFF "_%08" PRId64 "_%02d_%010" PRId64 "_%06d_%c.bin"
31
32 static int usage(int ret)
33 {
34 fprintf(stderr, "dump (up to maxpkts) AVPackets as they are demuxed by libavformat.\n");
35 fprintf(stderr, "each packet is dumped in its own file named like `basename file.ext`_$PKTNUM_$STREAMINDEX_$STAMP_$SIZE_$FLAGS.bin\n");
36 fprintf(stderr, "pktdumper [-nw] file [maxpkts]\n");
37 fprintf(stderr, "-n\twrite No file at all, only demux.\n");
38 fprintf(stderr, "-w\tWait at end of processing instead of quitting.\n");
39 return ret;
40 }
41
42 int main(int argc, char **argv)
43 {
44 char fntemplate[PATH_MAX];
45 char pktfilename[PATH_MAX];
46 AVFormatContext *fctx = NULL;
47 AVPacket pkt;
48 int64_t pktnum = 0;
49 int64_t maxpkts = 0;
50 int donotquit = 0;
51 int nowrite = 0;
52 int err;
53
54 if ((argc > 1) && !strncmp(argv[1], "-", 1)) {
55 if (strchr(argv[1], 'w'))
56 donotquit = 1;
57 if (strchr(argv[1], 'n'))
58 nowrite = 1;
59 argv++;
60 argc--;
61 }
62 if (argc < 2)
63 return usage(1);
64 if (argc > 2)
65 maxpkts = atoi(argv[2]);
66 strncpy(fntemplate, argv[1], PATH_MAX - 1);
67 if (strrchr(argv[1], '/'))
68 strncpy(fntemplate, strrchr(argv[1], '/') + 1, PATH_MAX - 1);
69 if (strrchr(fntemplate, '.'))
70 *strrchr(fntemplate, '.') = '\0';
71 if (strchr(fntemplate, '%')) {
72 fprintf(stderr, "can't use filenames containing '%%'\n");
73 return usage(1);
74 }
75 if (strlen(fntemplate) + sizeof(PKTFILESUFF) >= PATH_MAX - 1) {
76 fprintf(stderr, "filename too long\n");
77 return usage(1);
78 }
79 strcat(fntemplate, PKTFILESUFF);
80 printf("FNTEMPLATE: '%s'\n", fntemplate);
81
82 // register all file formats
83 av_register_all();
84
85 err = avformat_open_input(&fctx, argv[1], NULL, NULL);
86 if (err < 0) {
87 fprintf(stderr, "cannot open input: error %d\n", err);
88 return 1;
89 }
90
91 err = avformat_find_stream_info(fctx, NULL);
92 if (err < 0) {
93 fprintf(stderr, "avformat_find_stream_info: error %d\n", err);
94 return 1;
95 }
96
97 av_init_packet(&pkt);
98
99 while ((err = av_read_frame(fctx, &pkt)) >= 0) {
100 int fd;
101 snprintf(pktfilename, PATH_MAX - 1, fntemplate, pktnum,
102 pkt.stream_index, pkt.pts, pkt.size,
103 (pkt.flags & AV_PKT_FLAG_KEY) ? 'K' : '_');
104 printf(PKTFILESUFF "\n", pktnum, pkt.stream_index, pkt.pts, pkt.size,
105 (pkt.flags & AV_PKT_FLAG_KEY) ? 'K' : '_');
106 //printf("open(\"%s\")\n", pktfilename);
107 if (!nowrite) {
108 fd = open(pktfilename, O_WRONLY | O_CREAT, 0644);
109 err = write(fd, pkt.data, pkt.size);
110 if (err < 0) {
111 fprintf(stderr, "write: error %d\n", err);
112 return 1;
113 }
114 close(fd);
115 }
116 av_free_packet(&pkt);
117 pktnum++;
118 if (maxpkts && (pktnum >= maxpkts))
119 break;
120 }
121
122 avformat_close_input(&fctx);
123
124 while (donotquit)
125 usleep(60 * 1000000);
126
127 return 0;
128 }