1 files changed, 44 insertions, 0 deletions
diff --git a/src/udt_extras.rs b/src/udt_extras.rs
new file mode 100644
index 0000000..f4a714b
--- /dev/null
+++ b/src/udt_extras.rs
@@ -0,0 +1,44 @@
+
+extern crate udt;
+
+use std::str::{self, FromStr};
+use udt::{UdtSocket};
+use std::io;
+use std::io::{Read, Write, ErrorKind};
+
+pub struct UdtStream {
+    socket: UdtSocket,
+}
+
+impl UdtStream {
+
+    pub fn new(socket: UdtSocket) -> UdtStream {
+        UdtStream { socket: socket }
+
+    }
+}
+
+impl Read for UdtStream {
+    fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
+        let buf_len = buf.len();
+        match self.socket.recv(buf, buf_len) {
+            Ok(len) => Ok(len as usize),
+            Err(err) => Err(io::Error::new(
+                ErrorKind::Other, err.err_msg)),
+        }
+    }
+}
+
+impl Write for UdtStream {
+    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
+        match self.socket.send(buf) {
+            Ok(len) => Ok(len as usize),
+            Err(err) => Err(io::Error::new(
+                ErrorKind::Other, err.err_msg)),
+        }
+    }
+
+    fn flush(&mut self) -> io::Result<()> {
+        Ok(())
+    }
+}