From 4a855c053e7c811a2eac0ca8a5772f6b37cc47a2 Mon Sep 17 00:00:00 2001
From: meh <meh@schizofreni.co>
Date: Sun, 21 Mar 2021 12:47:40 +0100
Subject: [PATCH] fix: enable buffering for the writer

---
 src/format/io.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/format/io.rs b/src/format/io.rs
index 7dc2714..c0373d7 100644
--- a/src/format/io.rs
+++ b/src/format/io.rs
@@ -172,10 +172,11 @@ impl Io {
 	pub fn output(value: impl Write + 'static) -> Self {
 		unsafe {
 			let proxy = Box::into_raw(Box::new(Proxy::Output(Box::new(value))));
+			let buffer = av_malloc(4096);
 			let ptr = avio_alloc_context(
-				ptr::null_mut(),
-				0,
-				AVIO_FLAG_WRITE & AVIO_FLAG_DIRECT,
+				buffer.cast(),
+				4096,
+				AVIO_FLAG_WRITE,
 				proxy.cast(),
 				None,
 				Some(write_packet),
@@ -190,6 +191,10 @@ impl Io {
 impl Drop for Io {
 	fn drop(&mut self) {
 		unsafe {
+			if !(*self.ptr).buffer.is_null() {
+				av_free((*self.ptr).buffer.cast());
+			}
+
 			avio_context_free(&mut self.ptr);
 			Box::from_raw(self.proxy);
 		}