Progress towards allowing large writes

src/frame/data.rs

@@ -52,9 +52,24 @@ impl<T> Data<T> {
         &self.data
     }
 
+    pub fn payload_mut(&mut self) -> &mut T {
+        &mut self.data
+    }
+
     pub fn into_payload(self) -> T {
         self.data
     }
+
+    pub fn map<F, U>(self, f: F) -> Data<U>
+        where F: FnOnce(T) -> U,
+    {
+        Data {
+            stream_id: self.stream_id,
+            data: f(self.data),
+            flags: self.flags,
+            pad_len: self.pad_len,
+        }
+    }
 }
 
 impl<T: Buf> Data<T> {

src/frame/mod.rs

@@ -80,6 +80,24 @@ impl<T> Frame<T> {
             _ => false,
         }
     }
+
+    pub fn map<F, U>(self, f: F) -> Frame<U>
+        where F: FnOnce(T) -> U
+    {
+        use self::Frame::*;
+
+        match self {
+            Data(frame) => frame.map(f).into(),
+            Headers(frame) => frame.into(),
+            Priority(frame) => frame.into(),
+            PushPromise(frame) => frame.into(),
+            Settings(frame) => frame.into(),
+            Ping(frame) => frame.into(),
+            GoAway(frame) => frame.into(),
+            WindowUpdate(frame) => frame.into(),
+            Reset(frame) => frame.into(),
+        }
+    }
 }
 
 impl<T: Buf> Frame<T> {