blob: 0e04b90aac47f25a8580cfc61c5eb2e89b12bc07
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 | static int md5_open(URLContext *h, const char *filename, int flags) |
35 | { |
36 | struct MD5Context *c = h->priv_data; |
37 | |
38 | if (!(flags & AVIO_FLAG_WRITE)) |
39 | return AVERROR(EINVAL); |
40 | |
41 | c->md5 = av_md5_alloc(); |
42 | if (!c->md5) |
43 | return AVERROR(ENOMEM); |
44 | av_md5_init(c->md5); |
45 | |
46 | return 0; |
47 | } |
48 | |
49 | static int md5_write(URLContext *h, const unsigned char *buf, int size) |
50 | { |
51 | struct MD5Context *c = h->priv_data; |
52 | av_md5_update(c->md5, buf, size); |
53 | return size; |
54 | } |
55 | |
56 | static int md5_close(URLContext *h) |
57 | { |
58 | struct MD5Context *c = h->priv_data; |
59 | const char *filename = h->filename; |
60 | uint8_t md5[16], buf[64]; |
61 | URLContext *out; |
62 | int i, err = 0; |
63 | |
64 | av_md5_final(c->md5, md5); |
65 | for (i = 0; i < sizeof(md5); i++) |
66 | snprintf(buf + i*2, 3, "%02x", md5[i]); |
67 | buf[i*2] = '\n'; |
68 | |
69 | av_strstart(filename, "md5:", &filename); |
70 | |
71 | if (*filename) { |
72 | err = ffurl_open_whitelist(&out, filename, AVIO_FLAG_WRITE, |
73 | &h->interrupt_callback, NULL, |
74 | h->protocol_whitelist, h->protocol_blacklist, h); |
75 | if (err) |
76 | return err; |
77 | err = ffurl_write(out, buf, i*2+1); |
78 | ffurl_close(out); |
79 | } else { |
80 | if (fwrite(buf, 1, i*2+1, stdout) < i*2+1) |
81 | err = AVERROR(errno); |
82 | } |
83 | |
84 | av_freep(&c->md5); |
85 | |
86 | 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 | }; |
97 |