6275
|
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 };
|