/*
   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.

   Copyright (C) Jay Sorg 2004

   rdp layer

*/

#include "xrdp.h"

char unknown1[172] =
{ 0xff, 0x02, 0xb6, 0x00, 0x28, 0x00, 0x00, 0x00,
  0x27, 0x00, 0x27, 0x00, 0x03, 0x00, 0x04, 0x00,
  0x00, 0x00, 0x26, 0x00, 0x01, 0x00, 0x1e, 0x00,
  0x02, 0x00, 0x1f, 0x00, 0x03, 0x00, 0x1d, 0x00,
  0x04, 0x00, 0x27, 0x00, 0x05, 0x00, 0x0b, 0x00,
  0x06, 0x00, 0x28, 0x00, 0x08, 0x00, 0x21, 0x00,
  0x09, 0x00, 0x20, 0x00, 0x0a, 0x00, 0x22, 0x00,
  0x0b, 0x00, 0x25, 0x00, 0x0c, 0x00, 0x24, 0x00,
  0x0d, 0x00, 0x23, 0x00, 0x0e, 0x00, 0x19, 0x00,
  0x0f, 0x00, 0x16, 0x00, 0x10, 0x00, 0x15, 0x00,
  0x11, 0x00, 0x1c, 0x00, 0x12, 0x00, 0x1b, 0x00,
  0x13, 0x00, 0x1a, 0x00, 0x14, 0x00, 0x17, 0x00,
  0x15, 0x00, 0x18, 0x00, 0x16, 0x00, 0x0e, 0x00,
  0x18, 0x00, 0x0c, 0x00, 0x19, 0x00, 0x0d, 0x00,
  0x1a, 0x00, 0x12, 0x00, 0x1b, 0x00, 0x14, 0x00,
  0x1f, 0x00, 0x13, 0x00, 0x20, 0x00, 0x00, 0x00,
  0x21, 0x00, 0x0a, 0x00, 0x22, 0x00, 0x06, 0x00,
  0x23, 0x00, 0x07, 0x00, 0x24, 0x00, 0x08, 0x00,
  0x25, 0x00, 0x09, 0x00, 0x26, 0x00, 0x04, 0x00,
  0x27, 0x00, 0x03, 0x00, 0x28, 0x00, 0x02, 0x00,
  0x29, 0x00, 0x01, 0x00, 0x2a, 0x00, 0x05, 0x00,
  0x2b, 0x00, 0x2a, 0x00 };


/*****************************************************************************/
struct xrdp_rdp* xrdp_rdp_create(struct xrdp_process* owner)
{
  struct xrdp_rdp* self;

  self = (struct xrdp_rdp*)g_malloc(sizeof(struct xrdp_rdp), 1);
  self->pro_layer = owner;
  self->in_s = &owner->in_s;
  self->out_s = &owner->out_s;
  self->share_id = 66538;
  self->sec_layer = xrdp_sec_create(self);
  return self;
}

/*****************************************************************************/
void xrdp_rdp_delete(struct xrdp_rdp* self)
{
  xrdp_sec_delete(self->sec_layer);
  g_free(self);
}

/*****************************************************************************/
int xrdp_rdp_init(struct xrdp_rdp* self, int len)
{
  if (xrdp_sec_init(self->sec_layer, len + 6) != 0)
    return 1;
  s_push_layer(self->out_s, rdp_hdr, 6);
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_init_data(struct xrdp_rdp* self, int len)
{
  if (xrdp_sec_init(self->sec_layer, len + 18) != 0)
    return 1;
  s_push_layer(self->out_s, rdp_hdr, 18);
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_recv(struct xrdp_rdp* self, int* code)
{
  int error;
  int len;
  int pdu_code;

  if (self->next_packet == 0 || self->next_packet >= self->in_s->end)
  {
    error = xrdp_sec_recv(self->sec_layer);
    if (error == -1) /* special code for send demand active */
    {
      self->next_packet = 0;
      *code = -1;
      return 0;
    }
    if (error != 0)
      return 1;
    self->next_packet = self->in_s->p;
  }
  else
    self->in_s->p = self->next_packet;
  in_uint16_le(self->in_s, len);
  if (len == 0x8000)
  {
    self->next_packet += 8;
    *code = 0;
    return 0;
  }
  in_uint16_le(self->in_s, pdu_code);
  *code = pdu_code & 0xf;
  in_uint8s(self->in_s, 2); /* mcs user id */
  self->next_packet += len;
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_send(struct xrdp_rdp* self, int pdu_type)
{
  int len;

  s_pop_layer(self->out_s, rdp_hdr);
  len = self->out_s->end - self->out_s->p;
  out_uint16_le(self->out_s, len);
  out_uint16_le(self->out_s, 0x10 | pdu_type);
  out_uint16_le(self->out_s, self->mcs_channel);
  if (xrdp_sec_send(self->sec_layer, 0) != 0)
    return 1;
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_send_data(struct xrdp_rdp* self, int data_pdu_type)
{
  int len;

  s_pop_layer(self->out_s, rdp_hdr);
  len = self->out_s->end - self->out_s->p;
  out_uint16_le(self->out_s, len);
  out_uint16_le(self->out_s, 0x10 | RDP_PDU_DATA);
  out_uint16_le(self->out_s, self->mcs_channel);
  out_uint32_le(self->out_s, self->share_id);
  out_uint8(self->out_s, 0);
  out_uint8(self->out_s, 1);
  out_uint16_le(self->out_s, len - 14);
  out_uint8(self->out_s, data_pdu_type);
  out_uint8(self->out_s, 0);
  out_uint16_le(self->out_s, 0);
  if (xrdp_sec_send(self->sec_layer, 0) != 0)
    return 1;
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_parse_client_mcs_data(struct xrdp_rdp* self)
{
  struct stream* p;
  int i;

  p = &self->sec_layer->client_mcs_data;
  p->p = p->data;
  in_uint8s(p, 31);
  in_uint16_le(p, self->width);
  in_uint16_le(p, self->height);
  in_uint8s(p, 120);
  self->bpp = 8;
  in_uint16_le(p, i);
  switch (i)
  {
    case 0xca01:
      in_uint8s(p, 6);
      in_uint8(p, i);
      if (i > 8)
        self->bpp = i;
      break;
    case 0xca02:
      self->bpp = 15;
      break;
    case 0xca03:
      self->bpp = 16;
      break;
    case 0xca04:
      self->bpp = 24;
      break;
  }
  p->p = p->data;
  DEBUG(("client width %d, client height %d bpp %d\n\r",
         self->width, self->height, self->bpp));
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_incoming(struct xrdp_rdp* self)
{
  DEBUG(("in xrdp_rdp_incoming\n"));
  if (xrdp_sec_incoming(self->sec_layer) != 0)
    return 1;
  self->mcs_channel = self->sec_layer->mcs_layer->userid +
                      MCS_USERCHANNEL_BASE;
  xrdp_rdp_parse_client_mcs_data(self);
  DEBUG(("out xrdp_rdp_incoming mcs channel %d\n", self->mcs_channel));
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_send_demand_active(struct xrdp_rdp* self)
{
  if (xrdp_rdp_init(self, 512) != 0)
    return 1;

  out_uint32_le(self->out_s, self->share_id);
  out_uint32_be(self->out_s, 0x04000401);
  out_uint32_be(self->out_s, 0x524e5300);
  out_uint32_be(self->out_s, 0x08000000);
  out_uint32_be(self->out_s, 0x09000800);
  out_uint16_le(self->out_s, self->mcs_channel);
  out_uint16_be(self->out_s, 0xb5e2);

  /* Output general capability set */
  out_uint16_le(self->out_s, RDP_CAPSET_GENERAL); /* 1 */
  out_uint16_le(self->out_s, RDP_CAPLEN_GENERAL); /* 24(0x18) */
  out_uint16_le(self->out_s, 1); /* OS major type */
  out_uint16_le(self->out_s, 3); /* OS minor type */
  out_uint16_le(self->out_s, 0x200); /* Protocol version */
  out_uint16_le(self->out_s, 0); /* pad */
  out_uint16_le(self->out_s, 0); /* Compression types */
  out_uint16_le(self->out_s, 0); /* pad */
  out_uint16_le(self->out_s, 0); /* Update capability */
  out_uint16_le(self->out_s, 0); /* Remote unshare capability */
  out_uint16_le(self->out_s, 0); /* Compression level */
  out_uint16_le(self->out_s, 0); /* Pad */

  /* Output bitmap capability set */
  out_uint16_le(self->out_s, RDP_CAPSET_BITMAP); /* 2 */
  out_uint16_le(self->out_s, RDP_CAPLEN_BITMAP); /* 28(0x1c) */
  out_uint16_le(self->out_s, self->bpp); /* Preferred BPP */
  out_uint16_le(self->out_s, 1); /* Receive 1 BPP */
  out_uint16_le(self->out_s, 1); /* Receive 4 BPP */
  out_uint16_le(self->out_s, 1); /* Receive 8 BPP */
  out_uint16_le(self->out_s, self->width); /* width */
  out_uint16_le(self->out_s, self->height); /* height */
  out_uint16_le(self->out_s, 0); /* Pad */
  out_uint16_le(self->out_s, 1); /* Allow resize */
  out_uint16_le(self->out_s, 1); /* bitmap compression */
  out_uint16_le(self->out_s, 0); /* unknown */
  out_uint16_le(self->out_s, 0); /* unknown */
  out_uint16_le(self->out_s, 0); /* pad */

  /* Output ? */
  out_uint16_le(self->out_s, 14);
  out_uint16_le(self->out_s, 4);

  /* Output order capability set */
  out_uint16_le(self->out_s, RDP_CAPSET_ORDER); /* 3 */
  out_uint16_le(self->out_s, RDP_CAPLEN_ORDER); /* 88(0x58) */
  out_uint8s(self->out_s, 16);
  out_uint32_be(self->out_s, 0x40420f00);
  out_uint16_le(self->out_s, 1); /* Cache X granularity */
  out_uint16_le(self->out_s, 20); /* Cache Y granularity */
  out_uint16_le(self->out_s, 0); /* Pad */
  out_uint16_le(self->out_s, 1); /* Max order level */
  out_uint16_le(self->out_s, 0x2f); /* Number of fonts */
  out_uint16_le(self->out_s, 0x22); /* Capability flags */
  /* caps */
  out_uint8(self->out_s, 1); /* dest blt */
  out_uint8(self->out_s, 1); /* pat blt */
  out_uint8(self->out_s, 1); /* screen blt */
  out_uint8(self->out_s, 1); /* memblt */
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 0);
  out_uint8(self->out_s, 0);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 1);
  out_uint8(self->out_s, 0);
  out_uint8(self->out_s, 0);
  out_uint8(self->out_s, 0);
  out_uint16_le(self->out_s, 0x6a1);
  out_uint8s(self->out_s, 6); /* ? */
  out_uint32_le(self->out_s, 0x0f4240); /* desk save */
  out_uint32_le(self->out_s, 0); /* ? */
  out_uint32_le(self->out_s, 0); /* ? */

  /* Output color cache capability set */
  out_uint16_le(self->out_s, RDP_CAPSET_COLCACHE);
  out_uint16_le(self->out_s, RDP_CAPLEN_COLCACHE);
  out_uint16_le(self->out_s, 6); /* cache size */
  out_uint16_le(self->out_s, 0); /* pad */

  /* Output pointer capability set */
  out_uint16_le(self->out_s, RDP_CAPSET_POINTER);
  out_uint16_le(self->out_s, RDP_CAPLEN_POINTER);
  out_uint16_le(self->out_s, 1); /* Colour pointer */
  out_uint16_le(self->out_s, 0x19); /* Cache size */

  /* Output ? */
  out_uint16_le(self->out_s, 0xd);
  out_uint16_le(self->out_s, 0x58); /* 88 */
  out_uint8(self->out_s, 1);
  out_uint8s(self->out_s, 83);

  s_mark_end(self->out_s);

  if (xrdp_rdp_send(self, RDP_PDU_DEMAND_ACTIVE) != 0)
    return 1;

  return 0;
}

/*****************************************************************************/
int xrdp_rdp_process_confirm_active(struct xrdp_rdp* self)
{
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_process_data_pointer(struct xrdp_rdp* self)
{
  return 0;
}

/*****************************************************************************/
/* RDP_INPUT_SYNCHRONIZE */
int xrdp_rdp_process_input_sync(struct xrdp_rdp* self, int device_flags,
                                int key_flags)
{
  return 0;
}

/*****************************************************************************/
/* RDP_INPUT_SCANCODE */
int xrdp_rdp_process_input_scancode(struct xrdp_rdp* self, int device_flags,
                                    int ext_flags, int scan_code)
{
  return 0;
}

/*****************************************************************************/
/* RDP_INPUT_MOUSE */
int xrdp_rdp_process_input_mouse(struct xrdp_rdp* self, int device_flags,
                                 int x, int y)
{
  return 0;
}

/*****************************************************************************/
/* RDP_DATA_PDU_INPUT */
int xrdp_rdp_process_data_input(struct xrdp_rdp* self)
{
  int num_events;
  int index;
  int msg_type;
  int device_flags;
  int param1;
  int param2;

  in_uint16_le(self->in_s, num_events);
  in_uint8s(self->in_s, 2); /* pad */
  for (index = 0; index < num_events; index++)
  {
    in_uint8s(self->in_s, 4); /* time */
    in_uint16_le(self->in_s, msg_type);
    in_uint16_le(self->in_s, device_flags);
    in_uint16_le(self->in_s, param1);
    in_uint16_le(self->in_s, param2);
    switch (msg_type)
    {
      case RDP_INPUT_SYNCHRONIZE: /* 0 */
        xrdp_rdp_process_input_sync(self, device_flags, param1);
        break;
      case RDP_INPUT_SCANCODE: /* 4 */
        xrdp_rdp_process_input_scancode(self, device_flags, param1, param2);
        break;
      case RDP_INPUT_MOUSE: /* 8001 */
        xrdp_rdp_process_input_mouse(self, device_flags, param1, param2);
        break;
      default:
        g_printf("unknown in xrdp_rdp_process_data_input\n");
        break;
    }
  }
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_send_synchronise(struct xrdp_rdp* self)
{
  if (xrdp_rdp_init_data(self, 4) != 0)
    return 1;
  out_uint16_le(self->out_s, 1);
  out_uint16_le(self->out_s, 1002);
  s_mark_end(self->out_s);
  if (xrdp_rdp_send_data(self, RDP_DATA_PDU_SYNCHRONISE) != 0)
    return 1;
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_send_control(struct xrdp_rdp* self, int action)
{
  if (xrdp_rdp_init_data(self, 8) != 0)
    return 1;
  out_uint16_le(self->out_s, action);
  out_uint16_le(self->out_s, 0); /* userid */
  out_uint32_le(self->out_s, 1002); /* control id */
  s_mark_end(self->out_s);
  if (xrdp_rdp_send_data(self, RDP_DATA_PDU_CONTROL) != 0)
    return 1;
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_process_data_control(struct xrdp_rdp* self)
{
  int action;

  in_uint16_le(self->in_s, action);
  in_uint8s(self->in_s, 2); /* user id */
  in_uint8s(self->in_s, 4); /* control id */
  if (action == RDP_CTL_REQUEST_CONTROL)
  {
    xrdp_rdp_send_synchronise(self);
    xrdp_rdp_send_control(self, RDP_CTL_COOPERATE);
    xrdp_rdp_send_control(self, RDP_CTL_GRANT_CONTROL);
  }
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_process_data_sync(struct xrdp_rdp* self)
{
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_process_screen_update(struct xrdp_rdp* self)
{
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_send_unknown1(struct xrdp_rdp* self)
{
  if (xrdp_rdp_init_data(self, 300) != 0)
    return 1;
  out_uint8a(self->out_s, unknown1, 172);
  s_mark_end(self->out_s);
  if (xrdp_rdp_send_data(self, 0x28) != 0)
    return 1;
  return 0;
}

/*****************************************************************************/
int xrdp_rdp_process_data_font(struct xrdp_rdp* self)
{
  int seq;

  in_uint8s(self->in_s, 2); /* num of fonts */
  in_uint8s(self->in_s, 2); // unknown
  in_uint16_le(self->in_s, seq);
  /* 419 client sends Seq 1, then 2 */
  /* 2600 clients sends only Seq 3 */
  if (seq == 2 || seq == 3) /* after second font message, we are up and */
  {                                           /* running */
    xrdp_rdp_send_unknown1(self);
  }
  return 0;
}

/*****************************************************************************/
/* RDP_PDU_DATA */
int xrdp_rdp_process_data(struct xrdp_rdp* self)
{
  int len;
  int data_type;
  int ctype;
  int clen;

  in_uint8s(self->in_s, 6);
  in_uint16_le(self->in_s, len);
  in_uint8(self->in_s, data_type);
  in_uint8(self->in_s, ctype);
  in_uint16_le(self->in_s, clen);
  DEBUG(("xrdp_rdp_process_data code %d\n", data_type));
  switch (data_type)
  {
    case RDP_DATA_PDU_POINTER: /* 27 */
      xrdp_rdp_process_data_pointer(self);
      break;
    case RDP_DATA_PDU_INPUT: /* 28 */
      xrdp_rdp_process_data_input(self);
      break;
    case RDP_DATA_PDU_CONTROL: /* 20 */
      xrdp_rdp_process_data_control(self);
      break;
    case RDP_DATA_PDU_SYNCHRONISE: /* 31 */
      xrdp_rdp_process_data_sync(self);
      break;
    case 33: /* 33 ?? */
      xrdp_rdp_process_screen_update(self);
      break;
    case 36: /* 36 ?? */
      return 1;
      break;
    case RDP_DATA_PDU_FONT2: /* 39 */
      xrdp_rdp_process_data_font(self);
      break;
    default:
      g_printf("unknown in xrdp_rdp_process_data\n");
      break;
  }
  return 0;
}