started rec file

3 files changed, 289 insertions, 0 deletions

diff --git a/rdp/rdp.h b/rdp/rdp.h
index 83ab798b..be65b52f 100644
--- a/rdp/rdp.h
+++ b/rdp/rdp.h
@@ -138,6 +138,8 @@ struct rdp_rdp
   int bitmap_cache;
   int desktop_save;
   int polygon_ellipse_orders;
+  int rec_mode;
+  int rec_fd;
   /* cache */
   struct rdp_colormap colormap;
   struct rdp_cursor cursors[32];
@@ -410,6 +412,10 @@ int APP_CC
 rdp_rdp_process_demand_active(struct rdp_rdp* self, struct stream* s);
 void APP_CC
 rdp_rdp_out_unistr(struct stream* s, char* text);
+int APP_CC
+rdp_rec_check_file(struct rdp_rdp* self);
+int APP_CC
+rdp_rec_write_item(struct rdp_rdp* self, struct stream* s);
 
 /* rdp_bitmap.c */
 int APP_CC
diff --git a/rdp/rdp_orders.c b/rdp/rdp_orders.c
index 1dafae60..0afb94df 100644
--- a/rdp/rdp_orders.c
+++ b/rdp/rdp_orders.c
@@ -170,6 +170,7 @@ rdp_orders_process_colcache(struct rdp_orders* self, struct stream* s,
                             int flags)
 {
   struct rdp_colormap* colormap;
+  struct stream* rec_s;
   int cache_id;
   int i;
 
@@ -182,6 +183,21 @@ rdp_orders_process_colcache(struct rdp_orders* self, struct stream* s,
   }
   g_free(self->cache_colormap[cache_id]);
   self->cache_colormap[cache_id] = colormap;
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 4096);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 10);
+    out_uint8(rec_s, cache_id);
+    for (i = 0; i < 256; i++)
+    {
+      out_uint32_le(rec_s, colormap->colors[i]);
+    }
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -203,6 +219,7 @@ rdp_orders_process_raw_bmpcache(struct rdp_orders* self, struct stream* s,
   char* src;
   char* dst;
   struct rdp_bitmap* bitmap;
+  struct stream* rec_s;
 
   in_uint8(s, cache_id);
   in_uint8s(s, 1);
@@ -231,6 +248,23 @@ rdp_orders_process_raw_bmpcache(struct rdp_orders* self, struct stream* s,
   }
   g_free(self->cache_bitmap[cache_id][cache_idx]);
   self->cache_bitmap[cache_id][cache_idx] = bitmap;
+  if (self->rdp_layer->rec_mode)
+  {
+    y = width * height * Bpp;
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, y + 256);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 8);
+    out_uint8(rec_s, cache_id);
+    out_uint16_le(rec_s, cache_idx);
+    out_uint16_le(rec_s, width);
+    out_uint16_le(rec_s, height);
+    out_uint16_le(rec_s, y);
+    out_uint8a(rec_s, inverted, y);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -254,6 +288,7 @@ rdp_orders_process_bmpcache(struct rdp_orders* self, struct stream* s,
   int row_size;
   int final_size;
   struct rdp_bitmap* bitmap;
+  struct stream* rec_s;
 
   in_uint8(s, cache_id);
   in_uint8(s, pad1);
@@ -294,6 +329,23 @@ rdp_orders_process_bmpcache(struct rdp_orders* self, struct stream* s,
   }
   g_free(self->cache_bitmap[cache_id][cache_idx]);
   self->cache_bitmap[cache_id][cache_idx] = bitmap;
+  if (self->rdp_layer->rec_mode)
+  {
+    size = width * height * Bpp;
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, size + 256);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 8);
+    out_uint8(rec_s, cache_id);
+    out_uint16_le(rec_s, cache_idx);
+    out_uint16_le(rec_s, width);
+    out_uint16_le(rec_s, height);
+    out_uint16_le(rec_s, size);
+    out_uint8a(rec_s, bmpdata, size);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -302,6 +354,7 @@ static void APP_CC
 rdp_orders_process_fontcache(struct rdp_orders* self, struct stream* s,
                              int flags)
 {
+  struct stream* rec_s;
   int font;
   int nglyphs;
   int character;
@@ -327,6 +380,24 @@ rdp_orders_process_fontcache(struct rdp_orders* self, struct stream* s,
     self->rdp_layer->mod->server_add_char(self->rdp_layer->mod, font,
                                           character, offset, baseline,
                                           width, height, data);
+    if (self->rdp_layer->rec_mode)
+    {
+      rdp_rec_check_file(self->rdp_layer);
+      make_stream(rec_s);
+      init_stream(rec_s, datasize + 256);
+      s_push_layer(rec_s, iso_hdr, 4);
+      out_uint8(rec_s, 9);
+      out_uint8(rec_s, font);
+      out_uint16_le(rec_s, character);
+      out_uint16_le(rec_s, offset);
+      out_uint16_le(rec_s, baseline);
+      out_uint16_le(rec_s, width);
+      out_uint16_le(rec_s, height);
+      out_uint16_le(rec_s, datasize);
+      out_uint8a(rec_s, data, datasize);
+      rdp_rec_write_item(self->rdp_layer, rec_s);
+      free_stream(rec_s);
+    }
   }
 }
 
@@ -435,6 +506,7 @@ rdp_orders_process_text2(struct rdp_orders* self, struct stream* s,
 {
   int fgcolor;
   int bgcolor;
+  struct stream* rec_s;
 
   if (present & 0x000001)
   {
@@ -535,6 +607,34 @@ rdp_orders_process_text2(struct rdp_orders* self, struct stream* s,
                                          self->state.text_text,
                                          self->state.text_length);
   self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod, 0xcc);
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 512);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 7);
+    out_uint8(rec_s, self->state.text_font);
+    out_uint8(rec_s, self->state.text_flags);
+    out_uint8(rec_s, self->state.text_opcode);
+    out_uint8(rec_s, self->state.text_mixmode);
+    out_uint32_le(rec_s, self->state.text_fgcolor);
+    out_uint32_le(rec_s, self->state.text_bgcolor);
+    out_uint16_le(rec_s, self->state.text_clipleft);
+    out_uint16_le(rec_s, self->state.text_cliptop);
+    out_uint16_le(rec_s, self->state.text_clipright);
+    out_uint16_le(rec_s, self->state.text_clipbottom);
+    out_uint16_le(rec_s, self->state.text_boxleft);
+    out_uint16_le(rec_s, self->state.text_boxtop);
+    out_uint16_le(rec_s, self->state.text_boxright);
+    out_uint16_le(rec_s, self->state.text_boxbottom);
+    out_uint16_le(rec_s, self->state.text_x);
+    out_uint16_le(rec_s, self->state.text_y);
+    out_uint16_le(rec_s, self->state.text_length);
+    out_uint8a(rec_s, self->state.text_text, self->state.text_length);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -543,6 +643,8 @@ static void APP_CC
 rdp_orders_process_destblt(struct rdp_orders* self, struct stream* s,
                            int present, int delta)
 {
+  struct stream* rec_s;
+
   if (present & 0x01)
   {
     rdp_orders_in_coord(s, &self->state.dest_x, delta);
@@ -571,6 +673,21 @@ rdp_orders_process_destblt(struct rdp_orders* self, struct stream* s,
                                          self->state.dest_cx,
                                          self->state.dest_cy);
   self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod, 0xcc);
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 512);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 6);
+    out_uint16_le(rec_s, self->state.dest_x);
+    out_uint16_le(rec_s, self->state.dest_y);
+    out_uint16_le(rec_s, self->state.dest_cx);
+    out_uint16_le(rec_s, self->state.dest_cy);
+    out_uint8(rec_s, self->state.dest_opcode);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -581,6 +698,7 @@ rdp_orders_process_patblt(struct rdp_orders* self, struct stream* s,
 {
   int fgcolor;
   int bgcolor;
+  struct stream* rec_s;
 
   if (present & 0x0001)
   {
@@ -636,6 +754,27 @@ rdp_orders_process_patblt(struct rdp_orders* self, struct stream* s,
                                          self->state.pat_cy);
   self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod, 0xcc);
   self->rdp_layer->mod->server_set_mixmode(self->rdp_layer->mod, 0);
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 512);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 5);
+    out_uint16_le(rec_s, self->state.pat_x);
+    out_uint16_le(rec_s, self->state.pat_y);
+    out_uint16_le(rec_s, self->state.pat_cx);
+    out_uint16_le(rec_s, self->state.pat_cy);
+    out_uint8(rec_s, self->state.pat_opcode);
+    out_uint32_le(rec_s, self->state.pat_fgcolor);
+    out_uint32_le(rec_s, self->state.pat_bgcolor);
+    out_uint8(rec_s, self->state.pat_brush.xorigin);
+    out_uint8(rec_s, self->state.pat_brush.yorigin);
+    out_uint8(rec_s, self->state.pat_brush.style);
+    out_uint8a(rec_s, self->state.pat_brush.pattern, 8);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -644,6 +783,8 @@ static void APP_CC
 rdp_orders_process_screenblt(struct rdp_orders* self, struct stream* s,
                              int present, int delta)
 {
+  struct stream* rec_s;
+
   if (present & 0x0001)
   {
     rdp_orders_in_coord(s, &self->state.screenblt_x, delta);
@@ -682,6 +823,23 @@ rdp_orders_process_screenblt(struct rdp_orders* self, struct stream* s,
                                           self->state.screenblt_srcx,
                                           self->state.screenblt_srcy);
   self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod, 0xcc);
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 512);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 4);
+    out_uint16_le(rec_s, self->state.screenblt_x);
+    out_uint16_le(rec_s, self->state.screenblt_y);
+    out_uint16_le(rec_s, self->state.screenblt_cx);
+    out_uint16_le(rec_s, self->state.screenblt_cy);
+    out_uint16_le(rec_s, self->state.screenblt_srcx);
+    out_uint16_le(rec_s, self->state.screenblt_srcy);
+    out_uint8(rec_s, self->state.screenblt_opcode);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -690,7 +848,9 @@ static void APP_CC
 rdp_orders_process_line(struct rdp_orders* self, struct stream* s,
                         int present, int delta)
 {
+  int bgcolor;
   int fgcolor;
+  struct stream* rec_s;
 
   if (present & 0x0001)
   {
@@ -723,11 +883,16 @@ rdp_orders_process_line(struct rdp_orders* self, struct stream* s,
   rdp_orders_parse_pen(s, &self->state.line_pen, present >> 7);
   self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod,
                                           self->state.line_opcode);
+  bgcolor = rdp_orders_convert_color(self->rdp_layer->mod->rdp_bpp,
+                                     self->rdp_layer->mod->xrdp_bpp,
+                                     self->state.line_bgcolor,
+                                     self->rdp_layer->colormap.colors);
   fgcolor = rdp_orders_convert_color(self->rdp_layer->mod->rdp_bpp,
                                      self->rdp_layer->mod->xrdp_bpp,
                                      self->state.line_pen.color,
                                      self->rdp_layer->colormap.colors);
   self->rdp_layer->mod->server_set_fgcolor(self->rdp_layer->mod, fgcolor);
+  self->rdp_layer->mod->server_set_bgcolor(self->rdp_layer->mod, bgcolor);
   self->rdp_layer->mod->server_set_pen(self->rdp_layer->mod,
                                        self->state.line_pen.style,
                                        self->state.line_pen.width);
@@ -737,6 +902,26 @@ rdp_orders_process_line(struct rdp_orders* self, struct stream* s,
                                          self->state.line_endx,
                                          self->state.line_endy);
   self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod, 0xcc);
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 512);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 3);
+    out_uint16_le(rec_s, self->state.line_mixmode);
+    out_uint16_le(rec_s, self->state.line_startx);
+    out_uint16_le(rec_s, self->state.line_starty);
+    out_uint16_le(rec_s, self->state.line_endx);
+    out_uint16_le(rec_s, self->state.line_endy);
+    out_uint32_le(rec_s, self->state.line_bgcolor);
+    out_uint8(rec_s, self->state.line_opcode);
+    out_uint8(rec_s, self->state.line_pen.style);
+    out_uint8(rec_s, self->state.line_pen.width);
+    out_uint32_le(rec_s, self->state.line_pen.color);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -747,6 +932,7 @@ rdp_orders_process_rect(struct rdp_orders* self, struct stream* s,
 {
   int i;
   int fgcolor;
+  struct stream* rec_s;
 
   if (present & 0x01)
   {
@@ -789,6 +975,21 @@ rdp_orders_process_rect(struct rdp_orders* self, struct stream* s,
                                          self->state.rect_y,
                                          self->state.rect_cx,
                                          self->state.rect_cy);
+  if (self->rdp_layer->rec_mode)
+  {
+    rdp_rec_check_file(self->rdp_layer);
+    make_stream(rec_s);
+    init_stream(rec_s, 512);
+    s_push_layer(rec_s, iso_hdr, 4);
+    out_uint8(rec_s, 1);
+    out_uint16_le(rec_s, self->state.rect_x);
+    out_uint16_le(rec_s, self->state.rect_y);
+    out_uint16_le(rec_s, self->state.rect_cx);
+    out_uint16_le(rec_s, self->state.rect_cy);
+    out_uint32_le(rec_s, self->state.rect_color);
+    rdp_rec_write_item(self->rdp_layer, rec_s);
+    free_stream(rec_s);
+  }
 }
 
 /*****************************************************************************/
@@ -843,6 +1044,7 @@ rdp_orders_process_memblt(struct rdp_orders* self, struct stream* s,
                           int present, int delta)
 {
   struct rdp_bitmap* bitmap;
+  struct stream* rec_s;
   char* bmpdata;
 
   if (present & 0x0001)
@@ -905,6 +1107,25 @@ rdp_orders_process_memblt(struct rdp_orders* self, struct stream* s,
                                             self->state.memblt_srcx,
                                             self->state.memblt_srcy);
     self->rdp_layer->mod->server_set_opcode(self->rdp_layer->mod, 0xcc);
+    if (self->rdp_layer->rec_mode)
+    {
+      rdp_rec_check_file(self->rdp_layer);
+      make_stream(rec_s);
+      init_stream(rec_s, 512);
+      s_push_layer(rec_s, iso_hdr, 4);
+      out_uint8(rec_s, 2);
+      out_uint8(rec_s, self->state.memblt_opcode);
+      out_uint16_le(rec_s, self->state.memblt_x);
+      out_uint16_le(rec_s, self->state.memblt_y);
+      out_uint16_le(rec_s, self->state.memblt_cx);
+      out_uint16_le(rec_s, self->state.memblt_cy);
+      out_uint16_le(rec_s, self->state.memblt_cache_id);
+      out_uint16_le(rec_s, self->state.memblt_cache_idx);
+      out_uint16_le(rec_s, self->state.memblt_srcx);
+      out_uint16_le(rec_s, self->state.memblt_srcy);
+      rdp_rec_write_item(self->rdp_layer, rec_s);
+      free_stream(rec_s);
+    }
     if (bmpdata != bitmap->data)
     {
       g_free(bmpdata);
diff --git a/rdp/rdp_rdp.c b/rdp/rdp_rdp.c
index 62cef281..e4e07e05 100644
--- a/rdp/rdp_rdp.c
+++ b/rdp/rdp_rdp.c
@@ -35,6 +35,7 @@ rdp_rdp_create(struct mod* owner)
   self->bitmap_cache = 1;
   self->desktop_save = 0;
   self->orders = rdp_orders_create(self);
+  self->rec_mode = 0;
   return self;
 }
 
@@ -48,6 +49,11 @@ rdp_rdp_delete(struct rdp_rdp* self)
   }
   rdp_orders_delete(self->orders);
   rdp_sec_delete(self->sec_layer);
+  if (self->rec_fd != 0)
+  {
+    g_file_close(self->rec_fd);
+    self->rec_fd = 0;
+  }
   g_free(self);
 }
 
@@ -1010,3 +1016,59 @@ rdp_rdp_process_demand_active(struct rdp_rdp* self, struct stream* s)
   rdp_orders_reset_state(self->orders);
   return 0;
 }
+
+/******************************************************************************/
+int APP_CC
+rdp_rec_check_file(struct rdp_rdp* self)
+{
+  char file_name[256];
+  int index;
+  int len;
+  struct stream* s;
+
+  if (self->rec_fd == 0)
+  {
+    index = 1;
+    g_sprintf(file_name, "rec%8.8d.rec", index);
+    while (g_file_exist(file_name))
+    {
+      index++;
+      if (index >= 9999)
+      {
+        return 1;
+      }
+      g_sprintf(file_name, "rec%8.8d.rec", index);
+    }
+    self->rec_fd = g_file_open(file_name);
+    make_stream(s);
+    init_stream(s, 8192);
+    out_uint8a(s, "XRDPREC1", 8);
+    out_uint8s(s, 8);
+    s_mark_end(s);
+    len = s->end - s->data;
+    g_file_write(self->rec_fd, s->data, len);
+    free_stream(s);
+  }
+  return 0;
+}
+
+/******************************************************************************/
+int APP_CC
+rdp_rec_write_item(struct rdp_rdp* self, struct stream* s)
+{
+  int len;
+  int time;
+
+  if (self->rec_fd == 0)
+  {
+    return 1;
+  }
+  time = g_time1();
+  out_uint32_le(s, time);
+  s_mark_end(s);
+  len = s->end - s->data;
+  s_pop_layer(s, iso_hdr);
+  out_uint32_le(s, len);
+  g_file_write(self->rec_fd, s->data, len);
+  return 0;
+}