summaryrefslogtreecommitdiff
path: root/ffmpeg1/libavformat/md5proto.c
diff options
context:
space:
mode:
Diffstat (limited to 'ffmpeg1/libavformat/md5proto.c')
-rw-r--r--ffmpeg1/libavformat/md5proto.c95
1 files changed, 95 insertions, 0 deletions
diff --git a/ffmpeg1/libavformat/md5proto.c b/ffmpeg1/libavformat/md5proto.c
new file mode 100644
index 0000000..6af0a6e
--- /dev/null
+++ b/ffmpeg1/libavformat/md5proto.c
@@ -0,0 +1,95 @@
+/*
+ * Copyright (c) 2010 Mans Rullgard
+ *
+ * This file is part of FFmpeg.
+ *
+ * FFmpeg is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * FFmpeg is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with FFmpeg; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include <stdio.h>
+#include "libavutil/avstring.h"
+#include "libavutil/md5.h"
+#include "libavutil/mem.h"
+#include "libavutil/error.h"
+#include "avformat.h"
+#include "avio.h"
+#include "url.h"
+
+struct MD5Context {
+ struct AVMD5 *md5;
+};
+
+static int md5_open(URLContext *h, const char *filename, int flags)
+{
+ struct MD5Context *c = h->priv_data;
+
+ if (!(flags & AVIO_FLAG_WRITE))
+ return AVERROR(EINVAL);
+
+ c->md5 = av_md5_alloc();
+ if (!c->md5)
+ return AVERROR(ENOMEM);
+ av_md5_init(c->md5);
+
+ return 0;
+}
+
+static int md5_write(URLContext *h, const unsigned char *buf, int size)
+{
+ struct MD5Context *c = h->priv_data;
+ av_md5_update(c->md5, buf, size);
+ return size;
+}
+
+static int md5_close(URLContext *h)
+{
+ struct MD5Context *c = h->priv_data;
+ const char *filename = h->filename;
+ uint8_t md5[16], buf[64];
+ URLContext *out;
+ int i, err = 0;
+
+ av_md5_final(c->md5, md5);
+ for (i = 0; i < sizeof(md5); i++)
+ snprintf(buf + i*2, 3, "%02x", md5[i]);
+ buf[i*2] = '\n';
+
+ av_strstart(filename, "md5:", &filename);
+
+ if (*filename) {
+ err = ffurl_open(&out, filename, AVIO_FLAG_WRITE,
+ &h->interrupt_callback, NULL);
+ if (err)
+ return err;
+ err = ffurl_write(out, buf, i*2+1);
+ ffurl_close(out);
+ } else {
+ if (fwrite(buf, 1, i*2+1, stdout) < i*2+1)
+ err = AVERROR(errno);
+ }
+
+ av_freep(&c->md5);
+
+ return err;
+}
+
+
+URLProtocol ff_md5_protocol = {
+ .name = "md5",
+ .url_open = md5_open,
+ .url_write = md5_write,
+ .url_close = md5_close,
+ .priv_data_size = sizeof(struct MD5Context),
+};