cache/brush.c

 
#include <freerdp/config.h>
 
#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/assert.h>
 
#include <freerdp/log.h>
#include <freerdp/update.h>
#include <freerdp/freerdp.h>
#include <winpr/stream.h>
 
#include "brush.h"
#include "cache.h"
 
#define TAG FREERDP_TAG("cache.brush")
 
typedef struct
{
  UINT32 bpp;
  void* entry;
} BRUSH_ENTRY;
 
struct rdp_brush_cache
{
  pPatBlt PatBlt;          /* 0 */
  pCacheBrush CacheBrush;  /* 1 */
  pPolygonSC PolygonSC;    /* 2 */
  pPolygonCB PolygonCB;    /* 3 */
  UINT32 paddingA[16 - 4]; /* 4 */
 
  UINT32 maxEntries;        /* 16 */
  UINT32 maxMonoEntries;    /* 17 */
  BRUSH_ENTRY* entries;     /* 18 */
  BRUSH_ENTRY* monoEntries; /* 19 */
  UINT32 paddingB[32 - 20]; /* 20 */
 
  rdpContext* context;
};
 
static BOOL update_gdi_patblt(rdpContext* context, PATBLT_ORDER* patblt)
{
  BOOL ret = TRUE;
 
  WINPR_ASSERT(context);
  WINPR_ASSERT(patblt);
 
  const rdpCache* cache = context->cache;
  WINPR_ASSERT(cache);
 
  rdpBrush* brush = &patblt->brush;
  WINPR_ASSERT(brush->style <= UINT8_MAX);
  const BYTE style = (BYTE)brush->style;
  BYTE* data = brush->data;
 
  if (brush->style & CACHED_BRUSH)
  {
    brush->data = brush_cache_get(cache->brush, brush->index, &brush->bpp);
    brush->style = 0x03;
  }
 
  WINPR_ASSERT(cache->brush);
  IFCALLRET(cache->brush->PatBlt, ret, context, patblt);
  brush->style = style;
  brush->data = data;
  return ret;
}
 
static BOOL update_gdi_polygon_sc(rdpContext* context, const POLYGON_SC_ORDER* polygon_sc)
{
  rdpCache* cache = nullptr;
  WINPR_ASSERT(context);
  cache = context->cache;
  WINPR_ASSERT(cache);
  WINPR_ASSERT(cache->brush);
  return IFCALLRESULT(TRUE, cache->brush->PolygonSC, context, polygon_sc);
}
 
static BOOL update_gdi_polygon_cb(rdpContext* context, POLYGON_CB_ORDER* polygon_cb)
{
  BOOL ret = TRUE;
 
  WINPR_ASSERT(context);
  WINPR_ASSERT(polygon_cb);
 
  rdpCache* cache = context->cache;
  WINPR_ASSERT(cache);
 
  rdpBrush* brush = &polygon_cb->brush;
  WINPR_ASSERT(brush->style <= UINT8_MAX);
  const BYTE style = (UINT8)brush->style;
  BYTE* data = brush->data;
 
  if (brush->style & CACHED_BRUSH)
  {
    brush->data = brush_cache_get(cache->brush, brush->index, &brush->bpp);
    brush->style = 0x03;
  }
 
  WINPR_ASSERT(cache->brush);
  IFCALLRET(cache->brush->PolygonCB, ret, context, polygon_cb);
  brush->style = style;
  brush->data = data;
  return ret;
}
 
static BOOL update_gdi_cache_brush(rdpContext* context, const CACHE_BRUSH_ORDER* cacheBrush)
{
  UINT32 length = 0;
  void* data = nullptr;
  rdpCache* cache = nullptr;
 
  WINPR_ASSERT(context);
  WINPR_ASSERT(cacheBrush);
 
  cache = context->cache;
  WINPR_ASSERT(cache);
 
  length = cacheBrush->bpp * 64 / 8;
  data = malloc(length);
 
  if (!data)
    return FALSE;
 
  CopyMemory(data, cacheBrush->data, length);
  brush_cache_put(cache->brush, cacheBrush->index, data, cacheBrush->bpp);
  return TRUE;
}
 
void* brush_cache_get(rdpBrushCache* brushCache, UINT32 index, UINT32* bpp)
{
  void* entry = nullptr;
 
  if (!brushCache)
    return nullptr;
 
  if (!bpp)
    return nullptr;
 
  if (*bpp == 1)
  {
    if (index >= brushCache->maxMonoEntries)
    {
      WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", *bpp, index);
      return nullptr;
    }
 
    *bpp = brushCache->monoEntries[index].bpp;
    entry = brushCache->monoEntries[index].entry;
  }
  else
  {
    if (index >= brushCache->maxEntries)
    {
      WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", *bpp, index);
      return nullptr;
    }
 
    *bpp = brushCache->entries[index].bpp;
    entry = brushCache->entries[index].entry;
  }
 
  if (entry == nullptr)
  {
    WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) at index: 0x%08" PRIX32 "", *bpp, index);
    return nullptr;
  }
 
  return entry;
}
 
void brush_cache_put(rdpBrushCache* brushCache, UINT32 index, void* entry, UINT32 bpp)
{
  WINPR_ASSERT(brushCache);
 
  if (bpp == 1)
  {
    if (index >= brushCache->maxMonoEntries)
    {
      WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", bpp, index);
      free(entry);
      return;
    }
 
    WINPR_ASSERT(brushCache->monoEntries);
    free(brushCache->monoEntries[index].entry);
    brushCache->monoEntries[index].bpp = bpp;
    brushCache->monoEntries[index].entry = entry;
  }
  else
  {
    if (index >= brushCache->maxEntries)
    {
      WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", bpp, index);
      free(entry);
      return;
    }
 
    WINPR_ASSERT(brushCache->entries);
    free(brushCache->entries[index].entry);
    brushCache->entries[index].bpp = bpp;
    brushCache->entries[index].entry = entry;
  }
}
 
void brush_cache_register_callbacks(rdpUpdate* update)
{
  WINPR_ASSERT(update);
  WINPR_ASSERT(update->context);
  WINPR_ASSERT(update->primary);
  WINPR_ASSERT(update->secondary);
 
  if (!freerdp_settings_get_bool(update->context->settings, FreeRDP_DeactivateClientDecoding))
  {
    rdpCache* cache = update->context->cache;
    WINPR_ASSERT(cache);
    WINPR_ASSERT(cache->brush);
 
    cache->brush->PatBlt = update->primary->PatBlt;
    cache->brush->PolygonSC = update->primary->PolygonSC;
    cache->brush->PolygonCB = update->primary->PolygonCB;
    update->primary->PatBlt = update_gdi_patblt;
    update->primary->PolygonSC = update_gdi_polygon_sc;
    update->primary->PolygonCB = update_gdi_polygon_cb;
    update->secondary->CacheBrush = update_gdi_cache_brush;
  }
}
 
rdpBrushCache* brush_cache_new(rdpContext* context)
{
  rdpBrushCache* brushCache = nullptr;
 
  WINPR_ASSERT(context);
 
  brushCache = (rdpBrushCache*)calloc(1, sizeof(rdpBrushCache));
 
  if (!brushCache)
    return nullptr;
 
  brushCache->context = context;
  brushCache->maxEntries = 64;
  brushCache->maxMonoEntries = 64;
  brushCache->entries = (BRUSH_ENTRY*)calloc(brushCache->maxEntries, sizeof(BRUSH_ENTRY));
 
  if (!brushCache->entries)
    goto fail;
 
  brushCache->monoEntries = (BRUSH_ENTRY*)calloc(brushCache->maxMonoEntries, sizeof(BRUSH_ENTRY));
 
  if (!brushCache->monoEntries)
    goto fail;
 
  return brushCache;
fail:
  WINPR_PRAGMA_DIAG_PUSH
  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
  brush_cache_free(brushCache);
  WINPR_PRAGMA_DIAG_POP
  return nullptr;
}
 
void brush_cache_free(rdpBrushCache* brushCache)
{
  if (brushCache)
  {
    if (brushCache->entries)
    {
      for (size_t i = 0; i < brushCache->maxEntries; i++)
        free(brushCache->entries[i].entry);
 
      free(brushCache->entries);
    }
 
    if (brushCache->monoEntries)
    {
      for (size_t i = 0; i < brushCache->maxMonoEntries; i++)
        free(brushCache->monoEntries[i].entry);
 
      free(brushCache->monoEntries);
    }
 
    free(brushCache);
  }
}
 
void free_cache_brush_order(rdpContext* context, CACHE_BRUSH_ORDER* order)
{
  WINPR_UNUSED(context);
  free(order);
}
 
CACHE_BRUSH_ORDER* copy_cache_brush_order(rdpContext* context, const CACHE_BRUSH_ORDER* order)
{
  CACHE_BRUSH_ORDER* dst = nullptr;
 
  WINPR_ASSERT(context);
 
  dst = calloc(1, sizeof(CACHE_BRUSH_ORDER));
 
  if (!dst || !order)
    goto fail;
 
  *dst = *order;
  return dst;
fail:
  free_cache_brush_order(context, dst);
  return nullptr;
}