LCOV - code coverage report
Current view: top level - libavformat - md5proto.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 24 33 72.7 %
Date: 2017-12-16 13:57:32 Functions: 3 3 100.0 %

          Line data    Source code
       1             : /*
       2             :  * Copyright (c) 2010 Mans Rullgard
       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 <stdio.h>
      22             : #include "libavutil/avstring.h"
      23             : #include "libavutil/md5.h"
      24             : #include "libavutil/mem.h"
      25             : #include "libavutil/error.h"
      26             : #include "avformat.h"
      27             : #include "avio.h"
      28             : #include "url.h"
      29             : 
      30             : struct MD5Context {
      31             :     struct AVMD5 *md5;
      32             : };
      33             : 
      34        1762 : static int md5_open(URLContext *h, const char *filename, int flags)
      35             : {
      36        1762 :     struct MD5Context *c = h->priv_data;
      37             : 
      38        1762 :     if (!(flags & AVIO_FLAG_WRITE))
      39           0 :         return AVERROR(EINVAL);
      40             : 
      41        1762 :     c->md5 = av_md5_alloc();
      42        1762 :     if (!c->md5)
      43           0 :         return AVERROR(ENOMEM);
      44        1762 :     av_md5_init(c->md5);
      45             : 
      46        1762 :     return 0;
      47             : }
      48             : 
      49       37572 : static int md5_write(URLContext *h, const unsigned char *buf, int size)
      50             : {
      51       37572 :     struct MD5Context *c = h->priv_data;
      52       37572 :     av_md5_update(c->md5, buf, size);
      53       37572 :     return size;
      54             : }
      55             : 
      56        1762 : static int md5_close(URLContext *h)
      57             : {
      58        1762 :     struct MD5Context *c = h->priv_data;
      59        1762 :     const char *filename = h->filename;
      60             :     uint8_t md5[16], buf[64];
      61             :     URLContext *out;
      62        1762 :     int i, err = 0;
      63             : 
      64        1762 :     av_md5_final(c->md5, md5);
      65       29954 :     for (i = 0; i < sizeof(md5); i++)
      66       28192 :         snprintf(buf + i*2, 3, "%02x", md5[i]);
      67        1762 :     buf[i*2] = '\n';
      68             : 
      69        1762 :     av_strstart(filename, "md5:", &filename);
      70             : 
      71        1762 :     if (*filename) {
      72           0 :         err = ffurl_open_whitelist(&out, filename, AVIO_FLAG_WRITE,
      73           0 :                                    &h->interrupt_callback, NULL,
      74             :                                    h->protocol_whitelist, h->protocol_blacklist, h);
      75           0 :         if (err)
      76           0 :             return err;
      77           0 :         err = ffurl_write(out, buf, i*2+1);
      78           0 :         ffurl_close(out);
      79             :     } else {
      80        1762 :         if (fwrite(buf, 1, i*2+1, stdout) < i*2+1)
      81           0 :             err = AVERROR(errno);
      82             :     }
      83             : 
      84        1762 :     av_freep(&c->md5);
      85             : 
      86        1762 :     return err;
      87             : }
      88             : 
      89             : 
      90             : const URLProtocol ff_md5_protocol = {
      91             :     .name                = "md5",
      92             :     .url_open            = md5_open,
      93             :     .url_write           = md5_write,
      94             :     .url_close           = md5_close,
      95             :     .priv_data_size      = sizeof(struct MD5Context),
      96             : };

Generated by: LCOV version 1.13