summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--rdp/rdp.c49
-rw-r--r--rdp/rdp.h7
-rw-r--r--vnc/vnc.c45
-rw-r--r--vnc/vnc.h7
-rw-r--r--xup/xup.c2
-rw-r--r--xup/xup.h7
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;
};