1 files changed, 73 insertions, 0 deletions
diff --git a/sesman/chansrv/chansrv_common.c b/sesman/chansrv/chansrv_common.c
new file mode 100644
index 00000000..fd26da7c
--- /dev/null
+++ b/sesman/chansrv/chansrv_common.c
@@ -0,0 +1,73 @@
+/**
+ * xrdp: A Remote Desktop Protocol server.
+ *
+ * Copyright (C) Laxmikant Rashinkar 2009-2014
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "chansrv_common.h"
+
+/**
+ * Assemble fragmented incoming packets into one stream
+ *
+ * @param  src           stream that contains partial data
+ * @param  dest          stream that contains entire data
+ * @param  chan_flags    fragmentation flags
+ * @param  length        bytes in this packet
+ * @param  total_length  total length of assembled packet
+ *
+ * @return 1 when all data has been assembled, 0 otherwise
+ *
+ * NOTE: it is the responsibility of the caller to free dest stream
+ ****************************************************************************/
+int
+read_entire_packet(struct stream *src, struct stream **dest, int chan_flags,
+                   int length, int total_length)
+{
+    struct stream *ls;
+
+    if ((chan_flags & 3) == 3)
+    {
+        /* packet not fragmented */
+        xstream_new(ls, total_length);
+        xstream_copyin(ls, src->p, length);
+        ls->p = ls->data;
+        *dest = ls;
+        return 1;
+    }
+
+    /* is this the first fragmented packet? */
+    if (chan_flags & 1)
+    {
+        xstream_new(ls, total_length);
+        *dest = ls;
+    }
+    else
+    {
+        ls = *dest;
+    }
+
+    xstream_copyin(ls, src->p, length);
+
+    /* in last packet, chan_flags & 0x02 will be true */
+    if (chan_flags & 0x02)
+    {
+        /* rewind stream */
+        ls->p = ls->data;
+        return 1;
+    }
+
+    return 0;
+}
+