added chansrv

1 files changed, 511 insertions, 0 deletions

diff --git a/sesman/chansrv/chansrv.c b/sesman/chansrv/chansrv.c
new file mode 100644
index 00000000..0ce6c74c
--- /dev/null
+++ b/sesman/chansrv/chansrv.c
@@ -0,0 +1,511 @@
+/*
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+   xrdp: A Remote Desktop Protocol server.
+   Copyright (C) Jay Sorg 2009
+*/
+
+#include "arch.h"
+#include "os_calls.h"
+#include "thread_calls.h"
+#include "trans.h"
+#include "chansrv.h"
+#include "defines.h"
+#include "sound.h"
+
+static tbus g_thread_done_event = 0;
+static struct trans* g_lis_trans = 0;
+static struct trans* g_con_trans = 0;
+static struct chan_item g_chan_items[32];
+static int g_num_chan_items = 0;
+static int g_clip_index = -1;
+static int g_sound_index = -1;
+
+tbus g_term_event = 0;
+int g_display = 0;
+int g_clip_chan_id = -1;
+int g_sound_chan_id = -1;
+
+/*****************************************************************************/
+/* returns error */
+int APP_CC
+send_channel_data(int chan_id, char* data, int size)
+{
+  struct stream* s;
+  int chan_flags;
+  int total_size;
+  int sent;
+  int rv;
+
+  s = trans_get_out_s(g_con_trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  rv = 0;
+  sent = 0;
+  total_size = size;
+  while (sent < total_size)
+  {
+    size = MIN(1600, total_size - sent);
+    chan_flags = 0;
+    if (sent == 0)
+    {
+      chan_flags |= 1; /* first */
+    }
+    if (size + sent == total_size)
+    {
+      chan_flags |= 2; /* last */
+    }
+    out_uint32_le(s, 0); /* version */
+    out_uint32_le(s, 8 + 8 + 2 + 2 + 2 + 4 + size); /* size */
+    out_uint32_le(s, 8); /* msg id */
+    out_uint32_le(s, 8 + 2 + 2 + 2 + 4 + size); /* size */
+    out_uint16_le(s, chan_id);
+    out_uint16_le(s, chan_flags);
+    out_uint16_le(s, size);
+    out_uint32_le(s, total_size);
+    out_uint8a(s, data + sent, size);
+    s_mark_end(s);
+    rv = trans_force_write(g_con_trans);
+    if (rv != 0)
+    {
+      break;
+    }
+    sent += size;
+  }
+  return rv;
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+send_init_response_message(void)
+{
+  struct stream* s;
+
+  g_writeln("xrdp-chansrv: in send_init_response_message");
+  s = trans_get_out_s(g_con_trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  out_uint32_le(s, 0); /* version */
+  out_uint32_le(s, 8 + 8); /* size */
+  out_uint32_le(s, 2); /* msg id */
+  out_uint32_le(s, 8); /* size */
+  s_mark_end(s);
+  return trans_force_write(g_con_trans);
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+send_channel_setup_response_message(void)
+{
+  struct stream* s;
+
+  g_writeln("xrdp-chansrv: in send_channel_setup_response_message");
+  s = trans_get_out_s(g_con_trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  out_uint32_le(s, 0); /* version */
+  out_uint32_le(s, 8 + 8); /* size */
+  out_uint32_le(s, 4); /* msg id */
+  out_uint32_le(s, 8); /* size */
+  s_mark_end(s);
+  return trans_force_write(g_con_trans);
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+send_channel_data_response_message(void)
+{
+  struct stream* s;
+
+  g_writeln("xrdp-chansrv: in send_channel_data_response_message");
+  s = trans_get_out_s(g_con_trans, 8192);
+  if (s == 0)
+  {
+    return 1;
+  }
+  out_uint32_le(s, 0); /* version */
+  out_uint32_le(s, 8 + 8); /* size */
+  out_uint32_le(s, 6); /* msg id */
+  out_uint32_le(s, 8); /* size */
+  s_mark_end(s);
+  return trans_force_write(g_con_trans);
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+process_message_init(struct stream* s)
+{
+  g_writeln("xrdp-chansrv: in process_message_init");
+  return send_init_response_message();
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+process_message_channel_setup(struct stream* s)
+{
+  int num_chans;
+  int index;
+  int rv;
+  struct chan_item* ci;
+
+  g_writeln("xrdp-chansrv: in process_message_channel_setup");
+  in_uint16_le(s, num_chans);
+  for (index = 0; index < num_chans; index++)
+  {
+    ci = &(g_chan_items[g_num_chan_items]);
+    g_memset(ci->name, 0, sizeof(ci->name));
+    in_uint8a(s, ci->name, 8);
+    in_uint16_le(s, ci->id);
+    in_uint16_le(s, ci->flags);
+    g_writeln("xrdp-chansrv: chan name %s id %d flags %d", ci->name, ci->id,
+              ci->flags);
+    if (g_strcasecmp(ci->name, "cliprdr") == 0)
+    {
+      g_clip_index = g_num_chan_items;
+      g_clip_chan_id = ci->id;
+    }
+    else if (g_strcasecmp(ci->name, "rdpsnd") == 0)
+    {
+      g_sound_index = g_num_chan_items;
+      g_sound_chan_id = ci->id;
+    }
+    g_num_chan_items++;
+  }
+  rv = send_channel_setup_response_message();
+  if (g_clip_index >= 0)
+  {
+    clipboard_init();
+  }
+  if (g_sound_index >= 0)
+  {
+    sound_init();
+  }
+  return rv;
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+process_message_channel_data(struct stream* s)
+{
+  int chan_id;
+  int chan_flags;
+  int rv;
+  int length;
+  int total_length;
+
+  in_uint16_le(s, chan_id);
+  in_uint16_le(s, chan_flags);
+  in_uint16_le(s, length);
+  in_uint32_le(s, total_length);
+  g_writeln("xrdp-chansrv: log channel data chan_id %d chan_flags %d",
+            chan_id, chan_flags);
+  rv = send_channel_data_response_message();
+  if (rv == 0)
+  {
+    if (chan_id == g_clip_chan_id)
+    {
+      rv = clipboard_data_in(s, chan_id, chan_flags, length, total_length);
+    }
+    else if (chan_id == g_sound_chan_id)
+    {
+      rv = sound_data_in(s, chan_id, chan_flags, length, total_length);
+    }
+  }
+  return rv;
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+process_message_channel_data_response(struct stream* s)
+{
+  g_writeln("xrdp-chansrv: in process_message_channel_data_response");
+  return 0;
+}
+
+/*****************************************************************************/
+/* returns error */
+static int APP_CC
+process_message(void)
+{
+  struct stream* s;
+  int size;
+  int id;
+  int rv;
+  char* next_msg;
+
+  if (g_con_trans == 0)
+  {
+    return 1;
+  }
+  s = trans_get_in_s(g_con_trans);
+  if (s == 0)
+  {
+    return 1;
+  }
+  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 1: /* init */
+        rv = process_message_init(s);
+        break;
+      case 3: /* channel setup */
+        rv = process_message_channel_setup(s);
+        break;
+      case 5: /* channel data */
+        rv = process_message_channel_data(s);
+        break;
+      case 7: /* channel data response */
+        rv = process_message_channel_data_response(s);
+        break;
+      default:
+        g_writeln("xrdp-chansrv: error in process_message unknown msg %d",
+                  id);
+        break;
+    }
+    if (rv != 0)
+    {
+      break;
+    }
+    s->p = next_msg;
+  }
+  return rv;
+}
+
+/*****************************************************************************/
+/* returns error */
+int DEFAULT_CC
+my_trans_data_in(struct trans* trans)
+{
+  struct stream* s;
+  int id;
+  int size;
+  int error;
+
+  if (trans == 0)
+  {
+    return 0;
+  }
+  if (trans != g_con_trans)
+  {
+    return 1;
+  }
+  g_writeln("xrdp-chansrv: my_trans_data_in");
+  s = trans_get_in_s(trans);
+  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 = process_message();
+  }
+  return error;
+}
+
+/*****************************************************************************/
+int DEFAULT_CC
+my_trans_conn_in(struct trans* trans, struct trans* new_trans)
+{
+  if (trans == 0)
+  {
+    return 1;
+  }
+  if (trans != g_lis_trans)
+  {
+    return 1;
+  }
+  if (g_con_trans != 0) /* if already set, error */
+  {
+    return 1;
+  }
+  if (new_trans == 0)
+  {
+    return 1;
+  }
+  g_writeln("xrdp-chansrv: my_trans_conn_in");
+  g_con_trans = new_trans;
+  g_con_trans->trans_data_in = my_trans_data_in;
+  g_con_trans->header_size = 8;
+  /* stop listening */
+  trans_delete(g_lis_trans);
+  g_lis_trans = 0;
+  return 0;
+}
+
+/*****************************************************************************/
+THREAD_RV THREAD_CC
+channel_thread_loop(void* in_val)
+{
+  tbus objs[32];
+  int num_objs;
+  int timeout;
+  int error;
+  char text[256];
+  THREAD_RV rv;
+
+  g_writeln("xrdp-chansrv: thread start");
+  rv = 0;
+  g_lis_trans = trans_create(1, 8192, 8192);
+  g_lis_trans->trans_conn_in = my_trans_conn_in;
+  g_snprintf(text, 255, "%d", 7200 + g_display);
+  error = trans_listen(g_lis_trans, text);
+  if (error != 0)
+  {
+    g_writeln("xrdp-chansrv: trans_listen failed");
+  }
+  if (error == 0)
+  {
+    timeout = 0;
+    num_objs = 0;
+    objs[num_objs] = g_term_event;
+    num_objs++;
+    trans_get_wait_objs(g_lis_trans, objs, &num_objs, &timeout);
+    while (g_obj_wait(objs, num_objs, 0, 0, timeout) == 0)
+    {
+      if (g_is_wait_obj_set(g_term_event))
+      {
+        g_writeln("xrdp-chansrv: g_term_event set");
+        break;
+      }
+      if (g_lis_trans != 0)
+      {
+        if (trans_check_wait_objs(g_lis_trans) != 0)
+        {
+          g_writeln("xrdp-chansrv: trans_check_wait_objs error");
+        }
+      }
+      if (g_con_trans != 0)
+      {
+        if (trans_check_wait_objs(g_con_trans) != 0)
+        {
+          g_writeln("xrdp-chansrv: trans_check_wait_objs error resetting");
+          /* delete g_con_trans */
+          trans_delete(g_con_trans);
+          g_con_trans = 0;
+          /* create new listener */
+          g_lis_trans = trans_create(1, 8192, 8192);
+          g_lis_trans->trans_conn_in = my_trans_conn_in;
+          g_snprintf(text, 255, "%d", 7200 + g_display);
+          error = trans_listen(g_lis_trans, text);
+          if (error != 0)
+          {
+            g_writeln("xrdp-chansrv: trans_listen failed");
+            break;
+          }
+        }
+      }
+      clipboard_check_wait_objs();
+      sound_check_wait_objs();
+      timeout = 0;
+      num_objs = 0;
+      objs[num_objs] = g_term_event;
+      num_objs++;
+      trans_get_wait_objs(g_lis_trans, objs, &num_objs, &timeout);
+      trans_get_wait_objs(g_con_trans, objs, &num_objs, &timeout);
+      clipboard_get_wait_objs(objs, &num_objs, &timeout);
+      sound_get_wait_objs(objs, &num_objs, &timeout);
+    }
+  }
+  trans_delete(g_lis_trans);
+  g_lis_trans = 0;
+  trans_delete(g_con_trans);
+  g_con_trans = 0;
+  g_writeln("xrdp-chansrv: thread stop");
+  g_set_wait_obj(g_thread_done_event);
+  return rv;
+}
+
+/*****************************************************************************/
+void DEFAULT_CC
+term_signal_handler(int sig)
+{
+  g_writeln("xrdp-chansrv: term_signal_handler: got signal %d", sig);
+  g_set_wait_obj(g_term_event);
+}
+
+/*****************************************************************************/
+void DEFAULT_CC
+nil_signal_handler(int sig)
+{
+  g_writeln("xrdp-chansrv: nil_signal_handler: got signal %d", sig);
+}
+
+/*****************************************************************************/
+int DEFAULT_CC
+main(int argc, char** argv)
+{
+  int pid;
+  char text[256];
+  char* display_text;
+
+  pid = g_getpid();
+  g_writeln("xrdp-chansrv: started pid %d(0x%8.8x)", pid, pid);
+  g_signal_kill(term_signal_handler); /* SIGKILL */
+  g_signal_terminate(term_signal_handler); /* SIGTERM */
+  g_signal_user_interrupt(term_signal_handler); /* SIGINT */
+  g_signal_pipe(nil_signal_handler); /* SIGPIPE */
+  display_text = g_getenv("XRDP_SESSVC_DISPLAY");
+  if (display_text == 0)
+  {
+    g_writeln("xrdp-chansrv: error, XRDP_SESSVC_DISPLAY not set");
+    return 1;
+  }
+  else
+  {
+    g_display = g_atoi(display_text);
+  }
+  if (g_display == 0)
+  {
+    g_writeln("xrdp-chansrv: error, display is zero");
+    return 1;
+  }
+  g_snprintf(text, 255, "xrdp_chansrv_term_%8.8x", pid);
+  g_term_event = g_create_wait_obj(text);
+  g_snprintf(text, 255, "xrdp_chansrv_thread_done_%8.8x", pid);
+  g_thread_done_event = g_create_wait_obj(text);
+  tc_thread_create(channel_thread_loop, 0);
+  if (g_obj_wait(&g_term_event, 1, 0, 0, 0) != 0)
+  {
+    g_writeln("xrdp-chansrv: error, g_obj_wait failed");
+  }
+  /* wait for thread to exit */
+  if (g_obj_wait(&g_thread_done_event, 1, 0, 0, 0) != 0)
+  {
+    g_writeln("xrdp-chansrv: error, g_obj_wait failed");
+  }
+  /* cleanup */
+  g_delete_wait_obj(g_term_event);
+  g_delete_wait_obj(g_thread_done_event);
+  return 0;
+}