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 }