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

   main rdp process

*/

#include "xrdp.h"

/*****************************************************************************/
struct xrdp_process* xrdp_process_create(struct xrdp_listen* owner)
{
  struct xrdp_process* self;

  self = (struct xrdp_process*)g_malloc(sizeof(struct xrdp_process), 1);
  self->lis_layer = owner;
  return self;
}

/*****************************************************************************/
void xrdp_process_delete(struct xrdp_process* self)
{
  if (self == 0)
    return;
  xrdp_rdp_delete(self->rdp_layer);
  xrdp_orders_delete(self->orders);
  xrdp_wm_delete(self->wm);
  g_free(self->in_s.data);
  g_free(self->out_s.data);
  g_free(self);
}

/*****************************************************************************/
int xrdp_process_main_loop(struct xrdp_process* self)
{
  int code;
  int i;
  int cont;

  self->status = 1;
  self->rdp_layer = xrdp_rdp_create(self);
  g_tcp_set_non_blocking(self->sck);
  if (xrdp_rdp_incoming(self->rdp_layer) == 0)
  {
    while (!g_is_term() && !self->term)
    {
      i = g_tcp_select(self->sck);
      if (i == 1)
      {
        cont = 1;
        while (cont)
        {
          if (xrdp_rdp_recv(self->rdp_layer, &code) != 0)
            break;
          DEBUG(("xrdp_process_main_loop code %d\n\r", code));
          switch (code)
          {
            case -1:
              xrdp_rdp_send_demand_active(self->rdp_layer);
              break;
            case 0:
              break;
            case RDP_PDU_CONFIRM_ACTIVE:
              xrdp_rdp_process_confirm_active(self->rdp_layer); /* 3 */
              break;
            case RDP_PDU_DATA:
              if (xrdp_rdp_process_data(self->rdp_layer) != 0) /* 7 */
              {
                DEBUG(("xrdp_rdp_process_data returned non zero\n\r"));
                cont = 0;
                self->term = 1;
              }
              break;
            default:
              g_printf("unknown in xrdp_process_main_loop\n\r");
              break;
          }
          if (cont)
            cont = self->rdp_layer->next_packet < self->rdp_layer->in_s->end;
        }
        if (cont) /* we must have errored out */
          break;
        if (self->rdp_layer->up_and_running && self->wm == 0)
        {
          /* only do this once */
          DEBUG(("xrdp_process_main_loop up and running\n\r"));
          self->orders = xrdp_orders_create(self);
          self->wm = xrdp_wm_create(self);
          xrdp_wm_init(self->wm);
        }
      }
      else if (i == 0)
        g_sleep(10);
      else
        break;
    }
  }
  xrdp_rdp_disconnect(self->rdp_layer);
  xrdp_rdp_delete(self->rdp_layer);
  self->rdp_layer = 0;
  g_tcp_close(self->sck);
  self->status = -1;
  xrdp_listen_delete_pro(self->lis_layer, self);
  return 0;
}