channel changes
This commit is contained in:
parent
af83cb7cae
commit
12b2ea66fc
@ -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);
|
||||
|
289
xrdp/xrdp_mm.c
289
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;
|
||||
@ -490,6 +491,213 @@ xrdp_mm_setup_mod2(struct xrdp_mm* self)
|
||||
return rv;
|
||||
}
|
||||
|
||||
/*****************************************************************************/
|
||||
/* 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;
|
||||
}
|
||||
|
||||
@ -692,6 +915,54 @@ xrdp_mm_signal(struct xrdp_mm* self)
|
||||
return rv;
|
||||
}
|
||||
|
||||
/*****************************************************************************/
|
||||
/* 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);
|
||||
}
|
||||
|
@ -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
|
||||
|
@ -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);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user