channel changes

4 files changed, 325 insertions, 13 deletions

diff --git a/xrdp/xrdp.h b/xrdp/xrdp.h
index 94e6ed5f..bff2547d 100644
--- a/xrdp/xrdp.h
+++ b/xrdp/xrdp.h
@@ -36,6 +36,7 @@
 #include "list.h"
 #include "file.h"
 #include "file_loc.h"
+#include "trans.h"
 
 /* xrdp.c */
 long APP_CC
@@ -333,6 +334,9 @@ int APP_CC
 xrdp_mm_connect(struct xrdp_mm* self);
 int APP_CC
 xrdp_mm_signal(struct xrdp_mm* self);
+int APP_CC
+xrdp_mm_process_channel_data(struct xrdp_mm* self, tbus param1, tbus param2,
+                             tbus param3, tbus param4);
 
 int DEFAULT_CC
 server_begin_update(struct xrdp_mod* mod);
@@ -394,4 +398,5 @@ int DEFAULT_CC
 server_get_channel_id(struct xrdp_mod* mod, char* name);
 int DEFAULT_CC
 server_send_to_channel(struct xrdp_mod* mod, int channel_id,
-                       char* data, int data_len);
+                       char* data, int data_len,
+                       int total_data_len, int flags);
diff --git a/xrdp/xrdp_mm.c b/xrdp/xrdp_mm.c
index cec71779..6d251d0c 100644
--- a/xrdp/xrdp_mm.c
+++ b/xrdp/xrdp_mm.c
@@ -73,6 +73,7 @@ xrdp_mm_module_cleanup(struct xrdp_mm* self)
   {
     g_xrdp_sync(sync_unload, self->mod_handle, 0);
   }
+  trans_delete(self->chan_trans);
   self->mod_init = 0;
   self->mod_exit = 0;
   self->mod = 0;
@@ -491,6 +492,213 @@ xrdp_mm_setup_mod2(struct xrdp_mm* self)
 }
 
 /*****************************************************************************/
+/* returns error
+   send a list of channels to the channel handler */
+static int APP_CC
+xrdp_mm_trans_send_channel_setup(struct xrdp_mm* self, struct trans* trans)
+{
+  int index;
+  int chan_id;
+  int chan_flags;
+  int size;
+  struct stream* s;
+  char chan_name[256];
+
+  s = trans_get_out_s(trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  s_push_layer(s, iso_hdr, 8);
+  s_push_layer(s, mcs_hdr, 8);
+  s_push_layer(s, sec_hdr, 2);
+  index = 0;
+  while (server_query_channel(self->mod, index, chan_name,
+                              &chan_flags) == 0)
+  {
+    chan_id = server_get_channel_id(self->mod, chan_name);
+    out_uint8a(s, chan_name, 8);
+    out_uint16_le(s, chan_id);
+    out_uint16_le(s, chan_flags);
+    index++;
+  }
+  s_mark_end(s);
+  s_pop_layer(s, sec_hdr);
+  out_uint16_le(s, index);
+  s_pop_layer(s, mcs_hdr);
+  size = (int)(s->end - s->p);
+  out_uint32_le(s, 3); /* msg id */
+  out_uint32_le(s, size); /* msg size */
+  s_pop_layer(s, iso_hdr);
+  size = (int)(s->end - s->p);
+  out_uint32_le(s, 0); /* version */
+  out_uint32_le(s, size); /* block size */
+  return trans_force_write(trans);
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+xrdp_mm_trans_send_channel_data_response(struct xrdp_mm* self,
+                                         struct trans* trans)
+{
+  struct stream* s;
+
+  s = trans_get_out_s(trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  out_uint32_le(s, 0); /* version */
+  out_uint32_le(s, 8 + 8); /* size */
+  out_uint32_le(s, 7); /* msg id */
+  out_uint32_le(s, 8); /* size */
+  s_mark_end(s);
+  return trans_force_write(trans);
+}
+
+/*****************************************************************************/
+/* returns error
+   init is done, sent channel setup */
+static int APP_CC
+xrdp_mm_trans_process_init_response(struct xrdp_mm* self, struct trans* trans)
+{
+  return xrdp_mm_trans_send_channel_setup(self, trans);
+}
+
+/*****************************************************************************/
+/* returns error
+   data coming in from the channel handler, send it to the client */
+static int APP_CC
+xrdp_mm_trans_process_channel_data(struct xrdp_mm* self, struct trans* trans)
+{
+  struct stream* s;
+  int size;
+  int total_size;
+  int chan_id;
+  int chan_flags;
+  int rv;
+
+  s = trans_get_in_s(trans);
+  if (s == 0)
+  {
+    return 1;
+  }
+  in_uint16_le(s, chan_id);
+  in_uint16_le(s, chan_flags);
+  in_uint16_le(s, size);
+  in_uint32_le(s, total_size);
+  rv = xrdp_mm_trans_send_channel_data_response(self, trans);
+  if (rv == 0)
+  {
+    rv = server_send_to_channel(self->mod, chan_id, s->p, size, total_size,
+                                chan_flags);
+  }
+  return rv;
+}
+
+/*****************************************************************************/
+/* returns error
+   process a message for the channel handler */
+static int APP_CC
+xrdp_mm_chan_process_msg(struct xrdp_mm* self, struct trans* trans,
+                         struct stream* s)
+{
+  int rv;
+  int id;
+  int size;
+  char* next_msg;
+
+  rv = 0;
+  while (s_check_rem(s, 8))
+  {
+    next_msg = s->p;
+    in_uint32_le(s, id);
+    in_uint32_le(s, size);
+    next_msg += size;
+    switch (id)
+    {
+      case 2: /* channel init response */
+        g_writeln("channel init response");
+        rv = xrdp_mm_trans_process_init_response(self, trans);
+        break;
+      case 4: /* channel setup response */
+        g_writeln("channel setup response");
+        break;
+      case 6: /* channel data response */
+        g_writeln("channel data response");
+        break;
+      case 8: /* channel data */
+        g_writeln("channel data");
+        rv = xrdp_mm_trans_process_channel_data(self, trans);
+        break;
+      default:
+        g_writeln("unknown");
+        break;
+    }
+    if (rv != 0)
+    {
+      break;
+    }
+    s->p = next_msg;
+  }
+  return rv;
+}
+
+/*****************************************************************************/
+/* this is callback from trans obj
+   returns error */
+static int APP_CC
+xrdp_mm_chan_data_in(struct trans* trans)
+{
+  struct xrdp_mm* self;
+  struct stream* s;
+  int id;
+  int size;
+  int error;
+
+  if (trans == 0)
+  {
+    return 1;
+  }
+  g_writeln("xrdp_mm_chan_data_in");
+  self = (struct xrdp_mm*)(trans->callback_data);
+  s = trans_get_in_s(trans);
+  if (s == 0)
+  {
+    return 1;
+  }
+  in_uint32_le(s, id);
+  in_uint32_le(s, size);
+  error = trans_force_read(trans, size - 8);
+  if (error == 0)
+  {
+    /* here, the entire message block is read in, process it */
+    error = xrdp_mm_chan_process_msg(self, trans, s);
+  }
+  return error;
+}
+
+/*****************************************************************************/
+static int APP_CC
+xrdp_mm_chan_send_init(struct xrdp_mm* self)
+{
+  struct stream* s;
+
+  s = trans_get_out_s(self->chan_trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  out_uint32_le(s, 0); /* version */
+  out_uint32_le(s, 8 + 8); /* size */
+  out_uint32_le(s, 1); /* msg id */
+  out_uint32_le(s, 8); /* size */
+  s_mark_end(s);
+  return trans_force_write(self->chan_trans);
+}
+
+/*****************************************************************************/
 static int APP_CC
 xrdp_mm_process_login_response(struct xrdp_mm* self, struct stream* s)
 {
@@ -513,6 +721,20 @@ xrdp_mm_process_login_response(struct xrdp_mm* self, struct stream* s)
       {
         xrdp_wm_set_login_mode(self->wm, 10);
         self->wm->dragging = 0;
+        /* connect channel redir */
+        self->chan_trans = trans_create(1, 8192, 8192);
+        self->chan_trans->trans_data_in = xrdp_mm_chan_data_in;
+        self->chan_trans->header_size = 8;
+        self->chan_trans->callback_data = self;
+        g_snprintf(text, 255, "%d", 7200 + display);
+        if (trans_connect(self->chan_trans, "127.0.0.1", text, 3) != 0)
+        {
+          g_writeln("error in trans_connect");
+        }
+        if (xrdp_mm_chan_send_init(self) != 0)
+        {
+          g_writeln("error in xrdp_mm_chan_send_init");
+        }
       }
     }
   }
@@ -643,6 +865,7 @@ xrdp_mm_connect(struct xrdp_mm* self)
       xrdp_mm_module_cleanup(self);
     }
   }
+  self->sesman_controlled = use_sesman;
   return rv;
 }
 
@@ -693,6 +916,54 @@ xrdp_mm_signal(struct xrdp_mm* self)
 }
 
 /*****************************************************************************/
+/* returns error
+   data coming from client that need to go to channel handler */
+int APP_CC
+xrdp_mm_process_channel_data(struct xrdp_mm* self, tbus param1, tbus param2,
+                             tbus param3, tbus param4)
+{
+  struct stream* s;
+  int rv;
+  int length;
+  int total_length;
+  int flags;
+  int id;
+  char* data;
+
+  g_writeln("in xrdp_mm_process_channel_data %d %d %d %d", param1, param2, param3, param4);
+  rv = 0;
+  if (self->chan_trans != 0)
+  {
+    s = trans_get_out_s(self->chan_trans, 8192);
+    if (s != 0)
+    {
+      id = LOWORD(param1);
+      flags = HIWORD(param1);
+      length = param2;
+      data = (char*)param3;
+      total_length = param4;
+      if (total_length < length)
+      {
+        g_writeln("warning in xrdp_mm_process_channel_data total_len < length");
+        total_length = length;
+      }
+      out_uint32_le(s, 0); /* version */
+      out_uint32_le(s, 8 + 8 + 2 + 2 + 2 + 4 + length);
+      out_uint32_le(s, 5); /* msg id */
+      out_uint32_le(s, 8 + 2 + 2 + 2 + 4 + length);
+      out_uint16_le(s, id);
+      out_uint16_le(s, flags);
+      out_uint16_le(s, length);
+      out_uint32_le(s, total_length);
+      out_uint8a(s, data, length);
+      s_mark_end(s);
+      rv = trans_force_write(self->chan_trans);
+    }
+  }
+  return rv;
+}
+
+/*****************************************************************************/
 int DEFAULT_CC
 server_begin_update(struct xrdp_mod* mod)
 {
@@ -1004,6 +1275,10 @@ server_query_channel(struct xrdp_mod* mod, int index, char* channel_name,
   struct xrdp_wm* wm;
 
   wm = (struct xrdp_wm*)mod->wm;
+  if (wm->mm->sesman_controlled)
+  {
+    return 1;
+  }
   return libxrdp_query_channel(wm->session, index, channel_name,
                                channel_flags);
 }
@@ -1015,16 +1290,26 @@ server_get_channel_id(struct xrdp_mod* mod, char* name)
   struct xrdp_wm* wm;
 
   wm = (struct xrdp_wm*)mod->wm;
+  if (wm->mm->sesman_controlled)
+  {
+    return 1;
+  }
   return libxrdp_get_channel_id(wm->session, name);
 }
 
 /*****************************************************************************/
 int DEFAULT_CC
 server_send_to_channel(struct xrdp_mod* mod, int channel_id,
-                       char* data, int data_len)
+                       char* data, int data_len,
+                       int total_data_len, int flags)
 {
   struct xrdp_wm* wm;
 
   wm = (struct xrdp_wm*)mod->wm;
-  return libxrdp_send_to_channel(wm->session, channel_id, data, data_len);
+  if (wm->mm->sesman_controlled)
+  {
+    return 1;
+  }
+  return libxrdp_send_to_channel(wm->session, channel_id, data, data_len,
+                                 total_data_len, flags);
 }
diff --git a/xrdp/xrdp_types.h b/xrdp/xrdp_types.h
index 7d2b2b7c..16fe9a5b 100644
--- a/xrdp/xrdp_types.h
+++ b/xrdp/xrdp_types.h
@@ -77,7 +77,8 @@ struct xrdp_mod
                               int* channel_flags);
   int (*server_get_channel_id)(struct xrdp_mod* v, char* name);
   int (*server_send_to_channel)(struct xrdp_mod* v, int channel_id,
-                                char* data, int data_len);
+                                char* data, int data_len,
+                                int total_data_len, int flags);
   long server_dumby[100 - 24]; /* align, 100 minus the number of server 
                                   functions above */
   /* common */
@@ -185,6 +186,8 @@ struct xrdp_mm
   struct xrdp_mod* mod;
   int display;
   int code;
+  int sesman_controlled;
+  struct trans* chan_trans;
 };
 
 struct xrdp_keymap
diff --git a/xrdp/xrdp_wm.c b/xrdp/xrdp_wm.c
index 3a7c52ab..cc479cd1 100644
--- a/xrdp/xrdp_wm.c
+++ b/xrdp/xrdp_wm.c
@@ -1250,18 +1250,29 @@ xrdp_wm_process_input_mouse(struct xrdp_wm* self, int device_flags,
 
 /******************************************************************************/
 static int APP_CC
-xrdp_wm_process_channel_data(struct xrdp_wm* self, int channel_id,
-                             char* data, int data_len)
+xrdp_wm_process_channel_data(struct xrdp_wm* self,
+                            tbus param1, tbus param2, tbus param3, tbus param4)
 {
- if (self->mm->mod != 0)
+  int rv;
+
+  rv = 1;
+  if (self->mm->mod != 0)
   {
-    if (self->mm->mod->mod_event != 0)
+    if (self->mm->sesman_controlled)
     {
-      self->mm->mod->mod_event(self->mm->mod, 0x5555, channel_id, (long)data,
-                               data_len, 0);
+      rv = xrdp_mm_process_channel_data(self->mm, param1, param2,
+                                        param3, param4);
+    }
+    else
+    {
+      if (self->mm->mod->mod_event != 0)
+      {
+        rv = self->mm->mod->mod_event(self->mm->mod, 0x5555, param1, param2,
+                                      param3, param4);
+      }
     }
   }
-  return 0;
+  return rv;
 }
 
 /******************************************************************************/
@@ -1302,7 +1313,7 @@ callback(long id, int msg, long param1, long param2, long param3, long param4)
       break;
     case 0x5555: /* called from xrdp_channel.c, channel data has come in,
                     pass it to module if there is one */
-      rv = xrdp_wm_process_channel_data(wm, param1, (char*)param2, param3);
+      rv = xrdp_wm_process_channel_data(wm, param1, param2, param3, param4);
       break;
   }
   return rv;
@@ -1479,6 +1490,10 @@ xrdp_wm_mod_get_wait_objs(struct xrdp_wm* self,
 {
   if (self->mm != 0)
   {
+    if (self->mm->chan_trans != 0)
+    {
+      trans_get_wait_objs(self->mm->chan_trans, read_objs, rcount, timeout);
+    }
     if (self->mm->mod != 0)
     {
       if (self->mm->mod->mod_get_wait_objs != 0)
@@ -1498,6 +1513,10 @@ xrdp_wm_mod_check_wait_objs(struct xrdp_wm* self)
 {
   if (self->mm != 0)
   {
+    if (self->mm->chan_trans != 0)
+    {
+      trans_check_wait_objs(self->mm->chan_trans);
+    }
     if (self->mm->mod != 0)
     {
       if (self->mm->mod->mod_check_wait_objs != 0)
@@ -1555,7 +1574,7 @@ xrdp_wm_check_wait_objs(struct xrdp_wm* self)
     {
       if (g_is_wait_obj_set(self->mm->sck_obj))
       {
-        rv = xrdp_mm_signal(self->mm); 
+        rv = xrdp_mm_signal(self->mm);
       }
     }
   }