client/echo_main.c

 
#include <freerdp/config.h>
 
#include <stdio.h>
#include <stdlib.h>
 
#include <winpr/crt.h>
#include <winpr/assert.h>
#include <winpr/stream.h>
 
#include "echo_main.h"
#include <freerdp/client/channels.h>
#include <freerdp/channels/log.h>
#include <freerdp/channels/echo.h>
 
#define TAG CHANNELS_TAG("echo.client")
 
typedef struct
{
  GENERIC_DYNVC_PLUGIN baseDynPlugin;
} ECHO_PLUGIN;
 
static UINT echo_on_data_received(IWTSVirtualChannelCallback* pChannelCallback, wStream* data)
{
  GENERIC_CHANNEL_CALLBACK* callback = (GENERIC_CHANNEL_CALLBACK*)pChannelCallback;
  const BYTE* pBuffer = Stream_ConstPointer(data);
  const size_t cbSize = Stream_GetRemainingLength(data);
 
  WINPR_ASSERT(callback);
  WINPR_ASSERT(callback->channel);
  WINPR_ASSERT(callback->channel->Write);
 
  if (cbSize > UINT32_MAX)
    return ERROR_INVALID_PARAMETER;
 
  /* echo back what we have received. ECHO does not have any message IDs. */
  return callback->channel->Write(callback->channel, (ULONG)cbSize, pBuffer, NULL);
}
 
static UINT echo_on_close(IWTSVirtualChannelCallback* pChannelCallback)
{
  GENERIC_CHANNEL_CALLBACK* callback = (GENERIC_CHANNEL_CALLBACK*)pChannelCallback;
 
  free(callback);
 
  return CHANNEL_RC_OK;
}
 
static const IWTSVirtualChannelCallback echo_callbacks = { echo_on_data_received, NULL, /* Open */
                                                         echo_on_close, NULL };
 
FREERDP_ENTRY_POINT(UINT VCAPITYPE echo_DVCPluginEntry(IDRDYNVC_ENTRY_POINTS* pEntryPoints))
{
  return freerdp_generic_DVCPluginEntry(pEntryPoints, TAG, ECHO_DVC_CHANNEL_NAME,
                                        sizeof(ECHO_PLUGIN), sizeof(GENERIC_CHANNEL_CALLBACK),
                                        &echo_callbacks, NULL, NULL);
}