Revision a1db009e

View differences:

libavformat/Makefile
284 284
OBJS-$(CONFIG_GOPHER_PROTOCOL)           += gopher.o
285 285
OBJS-$(CONFIG_HTTP_PROTOCOL)             += http.o httpauth.o
286 286
OBJS-$(CONFIG_MMST_PROTOCOL)             += mmst.o asf.o
287
OBJS-$(CONFIG_MD5_PROTOCOL)              += md5proto.o
287 288
OBJS-$(CONFIG_PIPE_PROTOCOL)             += file.o
288 289

  
289 290
# external or internal rtmp
libavformat/allformats.c
223 223
    REGISTER_PROTOCOL (GOPHER, gopher);
224 224
    REGISTER_PROTOCOL (HTTP, http);
225 225
    REGISTER_PROTOCOL (MMST, mmst);
226
    REGISTER_PROTOCOL (MD5,  md5);
226 227
    REGISTER_PROTOCOL (PIPE, pipe);
227 228
    REGISTER_PROTOCOL (RTMP, rtmp);
228 229
#if CONFIG_LIBRTMP
libavformat/md5proto.c
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

  
29
#define PRIV_SIZE 128
30

  
31
static int md5_open(URLContext *h, const char *filename, int flags)
32
{
33
    if (PRIV_SIZE < av_md5_size) {
34
        av_log(NULL, AV_LOG_ERROR, "Insuffient size for MD5 context\n");
35
        return -1;
36
    }
37

  
38
    if (flags != URL_WRONLY)
39
        return AVERROR(EINVAL);
40

  
41
    av_md5_init(h->priv_data);
42

  
43
    return 0;
44
}
45

  
46
static int md5_write(URLContext *h, const unsigned char *buf, int size)
47
{
48
    av_md5_update(h->priv_data, buf, size);
49
    return size;
50
}
51

  
52
static int md5_close(URLContext *h)
53
{
54
    const char *filename = h->filename;
55
    uint8_t md5[16], buf[64];
56
    URLContext *out;
57
    int i, err = 0;
58

  
59
    av_md5_final(h->priv_data, md5);
60
    for (i = 0; i < sizeof(md5); i++)
61
        snprintf(buf + i*2, 3, "%02x", md5[i]);
62
    buf[i*2] = '\n';
63

  
64
    av_strstart(filename, "md5:", &filename);
65

  
66
    if (*filename) {
67
        err = url_open(&out, filename, URL_WRONLY);
68
        if (err)
69
            return err;
70
        err = url_write(out, buf, i*2+1);
71
        url_close(out);
72
    } else {
73
        if (fwrite(buf, 1, i*2+1, stdout) < i*2+1)
74
            err = AVERROR(errno);
75
    }
76

  
77
    return err;
78
}
79

  
80
static int md5_get_handle(URLContext *h)
81
{
82
    return (intptr_t)h->priv_data;
83
}
84

  
85
URLProtocol md5_protocol = {
86
    .name                = "md5",
87
    .url_open            = md5_open,
88
    .url_write           = md5_write,
89
    .url_close           = md5_close,
90
    .url_get_file_handle = md5_get_handle,
91
    .priv_data_size      = PRIV_SIZE,
92
};

Also available in: Unified diff