]> git.sesse.net Git - ffmpeg/blob - pktdumper.c
CPU feature definitions should always be available, not just when the
[ffmpeg] / pktdumper.c
1 /*
2  * Copyright (c) 2005 Francois Revol
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg 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  * FFmpeg 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 FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <avformat.h>
22 #include <limits.h>
23 #include <fcntl.h>
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <unistd.h>
28
29 #define PKTFILESUFF "_%08"PRId64"_%02d_%010"PRId64"_%06d_%c.bin"
30
31 static int usage(int ret)
32 {
33     fprintf(stderr, "dump (up to maxpkts) AVPackets as they are demuxed by libavformat.\n");
34     fprintf(stderr, "each packet is dumped in its own file named like `basename file.ext`_$PKTNUM_$STREAMINDEX_$STAMP_$SIZE_$FLAGS.bin\n");
35     fprintf(stderr, "pktdumper [-nw] file [maxpkts]\n");
36     fprintf(stderr, "-n\twrite No file at all, only demux.\n");
37     fprintf(stderr, "-w\tWait at end of processing instead of quitting.\n");
38     return ret;
39 }
40
41 int main(int argc, char **argv)
42 {
43     char fntemplate[PATH_MAX];
44     char pktfilename[PATH_MAX];
45     AVFormatContext *fctx;
46     AVPacket pkt;
47     int64_t pktnum = 0;
48     int64_t maxpkts = 0;
49     int dontquit = 0;
50     int nowrite = 0;
51     int err;
52
53     if ((argc > 1) && !strncmp(argv[1], "-", 1)) {
54         if (strchr(argv[1], 'w'))
55             dontquit = 1;
56         if (strchr(argv[1], 'n'))
57             nowrite = 1;
58         argv++;
59         argc--;
60     }
61     if (argc < 2)
62         return usage(1);
63     if (argc > 2)
64         maxpkts = atoi(argv[2]);
65     strncpy(fntemplate, argv[1], PATH_MAX-1);
66     if (strrchr(argv[1], '/'))
67         strncpy(fntemplate, strrchr(argv[1], '/')+1, PATH_MAX-1);
68     if (strrchr(fntemplate, '.'))
69         *strrchr(fntemplate, '.') = '\0';
70     if (strchr(fntemplate, '%')) {
71         fprintf(stderr, "can't use filenames containing '%%'\n");
72         return usage(1);
73     }
74     if (strlen(fntemplate) + sizeof(PKTFILESUFF) >= PATH_MAX-1) {
75         fprintf(stderr, "filename too long\n");
76         return usage(1);
77     }
78     strcat(fntemplate, PKTFILESUFF);
79     printf("FNTEMPLATE: '%s'\n", fntemplate);
80
81     // register all file formats
82     av_register_all();
83
84     err = av_open_input_file(&fctx, argv[1], NULL, 0, NULL);
85     if (err < 0) {
86         fprintf(stderr, "av_open_input_file: error %d\n", err);
87         return 1;
88     }
89
90     err = av_find_stream_info(fctx);
91     if (err < 0) {
92         fprintf(stderr, "av_find_stream_info: error %d\n", err);
93         return 1;
94     }
95
96     av_init_packet(&pkt);
97
98     while ((err = av_read_frame(fctx, &pkt)) >= 0) {
99         int fd;
100         snprintf(pktfilename, PATH_MAX-1, fntemplate, pktnum, pkt.stream_index, pkt.pts, pkt.size, (pkt.flags & PKT_FLAG_KEY)?'K':'_');
101         printf(PKTFILESUFF"\n", pktnum, pkt.stream_index, pkt.pts, pkt.size, (pkt.flags & PKT_FLAG_KEY)?'K':'_');
102         //printf("open(\"%s\")\n", pktfilename);
103         if (!nowrite) {
104             fd = open(pktfilename, O_WRONLY|O_CREAT, 0644);
105             write(fd, pkt.data, pkt.size);
106             close(fd);
107         }
108         pktnum++;
109         if (maxpkts && (pktnum >= maxpkts))
110             break;
111     }
112
113     while (dontquit)
114         sleep(60);
115
116     return 0;
117 }