xrdp-proprietary/libxrdp/xrdp_fastpath.c

/**
 * xrdp: A Remote Desktop Protocol server.
 *
 * Copyright (C) Jay Sorg 2012-2014
 * Copyright (C) Idan Freiberg 2013-2014
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "libxrdp.h"

/*****************************************************************************/
struct xrdp_fastpath *APP_CC
xrdp_fastpath_create(struct xrdp_sec *owner, struct trans *trans)
{
    struct xrdp_fastpath *self;

    DEBUG(("  in xrdp_fastpath_create"));
    self = (struct xrdp_fastpath *)g_malloc(sizeof(struct xrdp_fastpath), 1);
    self->sec_layer = owner;
    self->trans = trans;
    self->session = owner->rdp_layer->session;
    DEBUG(("  out xrdp_fastpath_create"));
    return self;
}

/*****************************************************************************/
void APP_CC
xrdp_fastpath_delete(struct xrdp_fastpath *self)
{
    if (self == 0)
    {
        return;
    }
    g_free(self);
}

/*****************************************************************************/
/* returns error */
int APP_CC
xrdp_fastpath_reset(struct xrdp_fastpath *self)
{
    return 0;
}

/*****************************************************************************/
int APP_CC
xrdp_fastpath_recv(struct xrdp_fastpath *self, struct stream *s)
{
    int fp_hdr;
    int len = 0; /* unused */
    int byte;
    char *holdp;

    DEBUG(("   in xrdp_fastpath_recv"));
    holdp = s->p;
    if (!s_check_rem(s, 2))
    {
        return 1;
    }
    in_uint8(s, fp_hdr); /* fpInputHeader (1 byte) */
    in_uint8(s, byte); /* length 1 (1 byte) */

    self->numEvents = (fp_hdr & 0x3C) >> 2;
    self->secFlags = (fp_hdr & 0xC0) >> 6;

    if (byte & 0x80)
    {
        byte &= ~(0x80);
        len = (byte << 8);

        if (!s_check_rem(s, 1))
        {
            return 1;
        }
        in_uint8(s, byte); /* length 2 (1 byte) */

        len += byte;
    }
    else
    {
        len = byte;
    }
    s->next_packet = holdp + len;
    DEBUG(("  out xrdp_fastpath_recv"));
    return 0;
}

/*****************************************************************************/
/* no fragmenation */
int APP_CC
xrdp_fastpath_init(struct xrdp_fastpath *self, struct stream *s)
{
    init_stream(s, 32 * 1024);
    return 0;
}

/*****************************************************************************/
/* no fragmenation */
int APP_CC
xrdp_fastpath_send(struct xrdp_fastpath *self, struct stream *s)
{
    if (trans_force_write_s(self->trans, s) != 0)
    {
        return 1;
    }
    return 0;
}

#if 0
/*****************************************************************************/
int APP_CC
xrdp_fastpath_send_update_pdu(struct xrdp_fastpath *self, tui8 updateCode,
                              struct stream *s)
{
    tui16 len;
    tui16 maxLen;
    tui32 payloadLeft;
    tui8 fragment;
    struct stream *s_send;
    int compression;
    int i;
    int i32;

    compression = 0;
//    s_send = self->out_s;
    maxLen = FASTPATH_MAX_PACKET_SIZE - 6; /* 6 bytes for header */
    payloadLeft = (s->end - s->data);

    for (i = 0; payloadLeft > 0; i++)
    {
        if (payloadLeft > maxLen)
        {
            len = maxLen;
        }
        else
        {
            len = payloadLeft;
        }

        payloadLeft -= len;

        if (payloadLeft == 0)
        {
            fragment = i ? FASTPATH_FRAGMENT_LAST : FASTPATH_FRAGMENT_SINGLE;
        }
        else
        {
            fragment = i ? FASTPATH_FRAGMENT_NEXT : FASTPATH_FRAGMENT_FIRST;
        }

        init_stream(s_send, 0);
        out_uint8(s_send, 0); /* fOutputHeader */
        i32 = ((len + 6) >> 8) | 0x80;
        out_uint8(s_send, i32); /* use 2 bytes for length even length < 128 ??? */
        i32 = (len + 6) & 0xff;
        out_uint8(s_send, i32);
        i32 = (updateCode & 0x0f) | ((fragment & 0x03) << 4) |
              ((compression & 0x03) << 6);
        out_uint8(s_send, i32);
        out_uint16_le(s_send, len);
//        s_copy(s_send, s, len);
        s_mark_end(s_send);

//        if (xrdp_tcp_send(self->tcp_layer, s_send) != 0)
//        {
//            return 1;
//        }
    }

    return 0;
}
#endif

/*****************************************************************************/
int
xrdp_fastpath_process_update(struct xrdp_fastpath *self, tui8 updateCode,
                             tui32 size, struct stream *s)
{
    switch (updateCode)
    {
        case FASTPATH_UPDATETYPE_ORDERS:
        case FASTPATH_UPDATETYPE_BITMAP:
        case FASTPATH_UPDATETYPE_PALETTE:
        case FASTPATH_UPDATETYPE_SYNCHRONIZE:
        case FASTPATH_UPDATETYPE_SURFCMDS:
        case FASTPATH_UPDATETYPE_PTR_NULL:
        case FASTPATH_UPDATETYPE_PTR_DEFAULT:
        case FASTPATH_UPDATETYPE_PTR_POSITION:
        case FASTPATH_UPDATETYPE_COLOR:
        case FASTPATH_UPDATETYPE_CACHED:
        case FASTPATH_UPDATETYPE_POINTER:
            break;
        default:
            g_writeln("xrdp_fastpath_process_update: unknown updateCode 0x%X",
                      updateCode);
            break;
    }

    return 0;
}

#if 0
/*****************************************************************************/
int APP_CC
xrdp_fastpath_process_data(struct xrdp_fastpath *self, struct stream *s,
                           tui8 header)
{
    tui8 encryptionFlags;
    tui8 numberEvents;
    tui8 length2;
    tui8 updateHeader;
    tui8 updateCode;
    tui8 updateFrag;
    tui8 updateComp;
    tui16 length;
    tui32 size;

    encryptionFlags = (header & 0xc0) >> 6;
    numberEvents = (header & 0x3c) >> 2;
//    xrdp_tcp_recv(self->tcp_layer, s, 1);
    in_uint8(s, length);

    if (length & 0x80)
    {
//        xrdp_tcp_recv(self->tcp_layer, s, 1);
        in_uint8(s, length2);
        length = (length & 0x7f) << 8 + length2 - 3;
    }
    else
    {
        length -= 2;
    }

//    xrdp_tcp_recv(self->tcp_layer, s, length);

    if (encryptionFlags != 0)
    {
        /* TODO decryption ...*/
    }

    /* parse updateHeader */
    in_uint8(s, updateHeader);
    updateCode = (updateHeader & 0x0f);
    updateFrag = (updateHeader & 0x30) >> 4;
    updateComp = (updateHeader & 0xc0) >> 6;

    if (updateFrag && updateComp)
    {
        /* TODO */
        g_writeln("xrdp_fastpath_process_data: updateFrag=%d, updateComp=%d",
                  updateFrag, updateComp);
        return 1;
    }

    in_uint16_le(s, size);
    return xrdp_fastpath_process_update(self, updateCode, size, s);
}
#endif

/*****************************************************************************/
/* FASTPATH_INPUT_EVENT_SCANCODE */
int APP_CC
xrdp_fastpath_process_EVENT_SCANCODE(struct xrdp_fastpath *self, int eventFlags, struct stream *s)
{
    int flags;
    int code;
    flags = 0;

    if (!s_check_rem(s, 1))
    {
        return 1;
    }
    in_uint8(s, code); /* keyCode (1 byte) */

    if ((eventFlags & FASTPATH_INPUT_KBDFLAGS_RELEASE))
        flags |= KBD_FLAG_UP;
    else
        flags |= KBD_FLAG_DOWN;

    if ((eventFlags & FASTPATH_INPUT_KBDFLAGS_EXTENDED))
        flags |= KBD_FLAG_EXT;

    if (self->session->callback != 0)
    {
        /* msg_type can be
          RDP_INPUT_SYNCHRONIZE - 0
          RDP_INPUT_SCANCODE - 4
          RDP_INPUT_MOUSE - 0x8001
          RDP_INPUT_MOUSEX - 0x8002 */
        /* call to xrdp_wm.c : callback */
        self->session->callback(self->session->id, RDP_INPUT_SCANCODE, code, 0,
                                flags, 0);
    }
    return 0;
}

/*****************************************************************************/
/* FASTPATH_INPUT_EVENT_MOUSE */
int APP_CC
xrdp_fastpath_process_EVENT_MOUSE(struct xrdp_fastpath *self, int eventFlags,
                                  struct stream *s)
{
    int pointerFlags;
    int xPos;
    int yPos;

    if (!s_check_rem(s, 2 + 2 + 2))
    {
        return 1;
    }
    in_uint16_le(s, pointerFlags); /* pointerFlags (2 bytes) */
    in_uint16_le(s, xPos); /* xPos (2 bytes) */
    in_uint16_le(s, yPos); /* yPos (2 bytes) */

    if (self->session->callback != 0)
    {
        /* msg_type can be
          RDP_INPUT_SYNCHRONIZE - 0
          RDP_INPUT_SCANCODE - 4
          RDP_INPUT_MOUSE - 0x8001
          RDP_INPUT_MOUSEX - 0x8002 */
        /* call to xrdp_wm.c : callback */
        self->session->callback(self->session->id, RDP_INPUT_MOUSE,
                                xPos, yPos, pointerFlags, 0);
    }
    return 0;
}

/*****************************************************************************/
/* FASTPATH_INPUT_EVENT_MOUSEX */
int APP_CC
xrdp_fastpath_process_EVENT_MOUSEX(struct xrdp_fastpath *self,
                                   int eventFlags, struct stream *s)
{
    int pointerFlags;
    int xPos;
    int yPos;

    if (!s_check_rem(s, 2 + 2 + 2))
    {
        return 1;
    }
    in_uint16_le(s, pointerFlags); /* pointerFlags (2 bytes) */
    in_uint16_le(s, xPos); /* xPos (2 bytes) */
    in_uint16_le(s, yPos); /* yPos (2 bytes) */

    if (self->session->callback != 0)
    {
        /* msg_type can be
          RDP_INPUT_SYNCHRONIZE - 0
          RDP_INPUT_SCANCODE - 4
          RDP_INPUT_MOUSE - 0x8001
          RDP_INPUT_MOUSEX - 0x8002 */
        /* call to xrdp_wm.c : callback */
        self->session->callback(self->session->id, RDP_INPUT_MOUSEX,
                                xPos, yPos, pointerFlags, 0);
    }
    return 0;
}

/*****************************************************************************/
/* FASTPATH_INPUT_EVENT_SYNC */
int APP_CC
xrdp_fastpath_process_EVENT_SYNC(struct xrdp_fastpath *self, int eventCode, int eventFlags, struct stream *s)
{
  /*
    * The eventCode bitfield (3 bits in size) MUST be set to FASTPATH_INPUT_EVENT_SYNC (3).
    * The eventFlags bitfield (5 bits in size) contains flags indicating the "on"
    * status of the keyboard toggle keys.
    */
    if (self->session->callback != 0)
    {
        /* msg_type can be
           RDP_INPUT_SYNCHRONIZE - 0
           RDP_INPUT_SCANCODE - 4
           RDP_INPUT_MOUSE - 0x8001
           RDP_INPUT_MOUSEX - 0x8002 */
        /* call to xrdp_wm.c : callback */
        self->session->callback(self->session->id, RDP_INPUT_SYNCHRONIZE,
                                eventCode, 0, eventFlags, 0);
    }
    return 0;
}

/*****************************************************************************/
/* FASTPATH_INPUT_EVENT_UNICODE */
int APP_CC
xrdp_fastpath_process_EVENT_UNICODE(struct xrdp_fastpath *self,
                                    int eventFlags, struct stream *s)
{
  if (!s_check_rem(s, 2))
  {
      return 1;
  }
  in_uint8s(s, 2);
  return 0;
}

/*****************************************************************************/
/* FASTPATH_INPUT_EVENT */
int APP_CC
xrdp_fastpath_process_input_event(struct xrdp_fastpath *self, struct stream *s)
{
    int i;
    int eventHeader;
    int eventCode;
    int eventFlags;

    // process fastpath input events
    for (i = 0; i < self->numEvents; i++)
    {
        if (!s_check_rem(s, 1))
        {
            return 1;
        }
        in_uint8(s, eventHeader);

        eventFlags = (eventHeader & 0x1F);
        eventCode = (eventHeader >> 5);

        //DEBUG(("xrdp_fastpath_process_input_event: eventCode= %d, eventFlags= %d, numEvents= %d",
        //          eventCode, eventFlags, self->sec_layer->fastpath_layer->numEvents));

        switch (eventCode)
        {
            case FASTPATH_INPUT_EVENT_SCANCODE:
                if (xrdp_fastpath_process_EVENT_SCANCODE(self, eventFlags, s) != 0)
                {
                    return 1;
                }
                break;

            case FASTPATH_INPUT_EVENT_MOUSE:
                if (xrdp_fastpath_process_EVENT_MOUSE(self, eventFlags, s) != 0)
                {
                    return 1;
                }
                break;
            case FASTPATH_INPUT_EVENT_MOUSEX:
                if (xrdp_fastpath_process_EVENT_MOUSEX(self, eventFlags, s) != 0)
                {
                    return 1;
                }
                break;
            case FASTPATH_INPUT_EVENT_SYNC:
                if (xrdp_fastpath_process_EVENT_SYNC(self, eventCode, eventFlags, s) != 0)
                {
                    return 1;
                }
                break;
            case FASTPATH_INPUT_EVENT_UNICODE:
                if (xrdp_fastpath_process_EVENT_UNICODE(self, eventFlags, s) != 0)
                {
                    return 1;
                }
                break;
            default:
                g_writeln("xrdp_fastpath_process_input_event: unknown eventCode %d", eventCode);
                break;
          }
    }
    return 0;
}