added new wait_obj functions

6 files changed, 114 insertions, 3 deletions

diff --git a/rdp/rdp.c b/rdp/rdp.c
index 12edb5a9..b7a279b7 100644
--- a/rdp/rdp.c
+++ b/rdp/rdp.c
@@ -53,6 +53,7 @@ lib_mod_connect(struct mod* mod)
   if (rdp_rdp_connect(mod->rdp_layer, mod->ip, mod->port) == 0)
   {
     mod->sck = mod->rdp_layer->sec_layer->mcs_layer->iso_layer->tcp_layer->sck;
+    mod->sck_obj = g_create_wait_obj_from_socket(mod->sck, 0);
     DEBUG(("out lib_mod_connect"));
     return 0;
   }
@@ -209,6 +210,11 @@ lib_mod_end(struct mod* mod)
   mod->rdp_layer = 0;
   free_stream(mod->in_s);
   mod->in_s = 0;
+  if (mod->sck_obj != 0)
+  {
+    g_delete_wait_obj_from_socket(mod->sck_obj);
+    mod->sck_obj = 0;
+  }
   if (mod->sck != 0)
   {
     g_tcp_close(mod->sck);
@@ -250,6 +256,47 @@ lib_mod_set_param(struct mod* mod, char* name, char* value)
 }
 
 /******************************************************************************/
+/* return error */
+int DEFAULT_CC
+lib_mod_get_wait_objs(struct mod* mod, tbus* read_objs, int* rcount,
+                      tbus* write_objs, int* wcount, int* timeout)
+{
+  int i;
+
+  i = *rcount;
+  if (mod != 0)
+  {
+    if (mod->sck_obj != 0)
+    {
+      read_objs[i++] = mod->sck_obj;
+    }
+  }
+  *rcount = i;
+  return 0;
+}
+
+/******************************************************************************/
+/* return error */
+int DEFAULT_CC
+lib_mod_check_wait_objs(struct mod* mod)
+{
+  int rv;
+
+  rv = 0;
+  if (mod != 0)
+  {
+    if (mod->sck_obj != 0)
+    {
+      if (g_is_wait_obj_set(mod->sck_obj))
+      {
+        rv = lib_mod_signal(mod);
+      }
+    }
+  }
+  return rv;
+}
+
+/******************************************************************************/
 struct mod* EXPORT_CC
 mod_init(void)
 {
@@ -265,6 +312,8 @@ mod_init(void)
   mod->mod_signal = lib_mod_signal;
   mod->mod_end = lib_mod_end;
   mod->mod_set_param = lib_mod_set_param;
+  mod->mod_get_wait_objs = lib_mod_get_wait_objs;
+  mod->mod_check_wait_objs = lib_mod_check_wait_objs;
   mod->rdp_layer = rdp_rdp_create(mod);
   DEBUG(("out mod_init"));
   return mod;
diff --git a/rdp/rdp.h b/rdp/rdp.h
index 2344e591..4e471dc1 100644
--- a/rdp/rdp.h
+++ b/rdp/rdp.h
@@ -263,7 +263,11 @@ struct mod
   int (*mod_signal)(struct mod* v);
   int (*mod_end)(struct mod* v);
   int (*mod_set_param)(struct mod* v, char* name, char* value);
-  long mod_dumby[100 - 6]; /* align, 100 minus the number of mod 
+  int (*mod_session_change)(struct mod* v, int, int);
+  int (*mod_get_wait_objs)(struct mod* v, tbus* read_objs, int* rcount,
+                           tbus* write_objs, int* wcount, int* timeout);
+  int (*mod_check_wait_objs)(struct mod* v);
+  long mod_dumby[100 - 9]; /* align, 100 minus the number of mod 
                               functions above */
   /* server functions */
   int (*server_begin_update)(struct mod* v);
@@ -328,6 +332,7 @@ struct mod
   int keylayout;
   int up_and_running;
   struct stream* in_s;
+  tbus sck_obj;
 };
 
 /* rdp_tcp.c */
diff --git a/vnc/vnc.c b/vnc/vnc.c
index 64da023a..5f9c9bac 100644
--- a/vnc/vnc.c
+++ b/vnc/vnc.c
@@ -1161,6 +1161,7 @@ connections", 0);
   g_sprintf(con_port, "%s", v->port);
   make_stream(pixel_format);
   v->sck = g_tcp_socket();
+  v->sck_obj = g_create_wait_obj_from_socket(v->sck, 0);
   v->sck_closed = 0;
   g_sprintf(text, "connecting to %s %s", v->ip, con_port);
   v->server_msg(v, text, 0);
@@ -1440,6 +1441,47 @@ lib_mod_set_param(struct vnc* v, char* name, char* value)
 }
 
 /******************************************************************************/
+/* return error */
+int DEFAULT_CC
+lib_mod_get_wait_objs(struct vnc* v, tbus* read_objs, int* rcount,
+                      tbus* write_objs, int* wcount, int* timeout)
+{
+  int i;
+
+  i = *rcount;
+  if (v != 0)
+  {
+    if (v->sck_obj != 0)
+    {
+      read_objs[i++] = v->sck_obj;
+    }
+  }
+  *rcount = i;
+  return 0;
+}
+
+/******************************************************************************/
+/* return error */
+int DEFAULT_CC
+lib_mod_check_wait_objs(struct vnc* v)
+{
+  int rv;
+
+  rv = 0;
+  if (v != 0)
+  {
+    if (v->sck_obj != 0)
+    {
+      if (g_is_wait_obj_set(v->sck_obj))
+      {
+        rv = lib_mod_signal(v);
+      }
+    }
+  }
+  return rv;
+}
+
+/******************************************************************************/
 struct vnc* EXPORT_CC
 mod_init(void)
 {
@@ -1455,6 +1497,8 @@ mod_init(void)
   v->mod_signal = lib_mod_signal;
   v->mod_end = lib_mod_end;
   v->mod_set_param = lib_mod_set_param;
+  v->mod_get_wait_objs = lib_mod_get_wait_objs;
+  v->mod_check_wait_objs = lib_mod_check_wait_objs;
   return v;
 }
 
@@ -1466,6 +1510,7 @@ mod_exit(struct vnc* v)
   {
     return 0;
   }
+  g_delete_wait_obj_from_socket(v->sck_obj);
   g_tcp_close(v->sck);
   g_free(v);
   return 0;
diff --git a/vnc/vnc.h b/vnc/vnc.h
index f7642daf..1d664eb3 100644
--- a/vnc/vnc.h
+++ b/vnc/vnc.h
@@ -38,7 +38,11 @@ struct vnc
   int (*mod_signal)(struct vnc* v);
   int (*mod_end)(struct vnc* v);
   int (*mod_set_param)(struct vnc* v, char* name, char* value);
-  long mod_dumby[100 - 6]; /* align, 100 minus the number of mod
+  int (*mod_session_change)(struct vnc* v, int, int);
+  int (*mod_get_wait_objs)(struct vnc* v, tbus* read_objs, int* rcount,
+                           tbus* write_objs, int* wcount, int* timeout);
+  int (*mod_check_wait_objs)(struct vnc* v);
+  long mod_dumby[100 - 9]; /* align, 100 minus the number of mod
                               functions above */
   /* server functions */
   int (*server_begin_update)(struct vnc* v);
@@ -107,4 +111,5 @@ struct vnc
   int clip_chanid;
   char* clip_data;
   int clip_data_size;
+  tbus sck_obj;
 };
diff --git a/xup/xup.c b/xup/xup.c
index ff570314..e4e57e44 100644
--- a/xup/xup.c
+++ b/xup/xup.c
@@ -455,6 +455,8 @@ mod_init(void)
   mod->mod_signal = lib_mod_signal;
   mod->mod_end = lib_mod_end;
   mod->mod_set_param = lib_mod_set_param;
+  mod->mod_get_wait_objs = lib_mod_get_wait_objs;
+  mod->mod_check_wait_objs = lib_mod_check_wait_objs;
   return mod;
 }
 
diff --git a/xup/xup.h b/xup/xup.h
index b82e6d1d..96584c5f 100644
--- a/xup/xup.h
+++ b/xup/xup.h
@@ -38,7 +38,11 @@ struct mod
   int (*mod_signal)(struct mod* v);
   int (*mod_end)(struct mod* v);
   int (*mod_set_param)(struct mod* v, char* name, char* value);
-  long mod_dumby[100 - 6]; /* align, 100 minus the number of mod
+  int (*mod_session_change)(struct mod* v, int, int);
+  int (*mod_get_wait_objs)(struct mod* v, tbus* read_objs, int* rcount,
+                           tbus* write_objs, int* wcount, int* timeout);
+  int (*mod_check_wait_objs)(struct mod* v);
+  long mod_dumby[100 - 9]; /* align, 100 minus the number of mod
                               functions above */
   /* server functions */
   int (*server_begin_update)(struct mod* v);
@@ -95,4 +99,5 @@ struct mod
   char password[256];
   char ip[256];
   char port[256];
+  tbus sck_obj;
 };