palette.c

 
#include <freerdp/config.h>
 
#include <stdio.h>
 
#include <winpr/crt.h>
 
#include <freerdp/log.h>
 
#include "palette.h"
#include "cache.h"
 
#define TAG FREERDP_TAG("cache.palette")
 
static void* palette_cache_get(rdpPaletteCache* palette, UINT32 index);
 
static void palette_cache_put(rdpPaletteCache* palette, UINT32 index, void* entry);
 
static BOOL update_gdi_cache_color_table(rdpContext* context,
                                         const CACHE_COLOR_TABLE_ORDER* cacheColorTable)
{
  UINT32* colorTable = NULL;
  rdpCache* cache = context->cache;
  colorTable = (UINT32*)malloc(sizeof(UINT32) * 256);
 
  if (!colorTable)
    return FALSE;
 
  CopyMemory(colorTable, cacheColorTable->colorTable, sizeof(UINT32) * 256);
  palette_cache_put(cache->palette, cacheColorTable->cacheIndex, (void*)colorTable);
  return TRUE;
}
 
void* palette_cache_get(rdpPaletteCache* paletteCache, UINT32 index)
{
  void* entry = NULL;
 
  if (index >= paletteCache->maxEntries)
  {
    WLog_ERR(TAG, "invalid color table index: 0x%08" PRIX32 "", index);
    return NULL;
  }
 
  entry = paletteCache->entries[index].entry;
 
  if (!entry)
  {
    WLog_ERR(TAG, "invalid color table at index: 0x%08" PRIX32 "", index);
    return NULL;
  }
 
  return entry;
}
 
void palette_cache_put(rdpPaletteCache* paletteCache, UINT32 index, void* entry)
{
  if (index >= paletteCache->maxEntries)
  {
    WLog_ERR(TAG, "invalid color table index: 0x%08" PRIX32 "", index);
    free(entry);
    return;
  }
 
  free(paletteCache->entries[index].entry);
  paletteCache->entries[index].entry = entry;
}
 
void palette_cache_register_callbacks(rdpUpdate* update)
{
  update->secondary->CacheColorTable = update_gdi_cache_color_table;
}
 
rdpPaletteCache* palette_cache_new(rdpContext* context)
{
  rdpPaletteCache* paletteCache = NULL;
 
  WINPR_ASSERT(context);
 
  paletteCache = (rdpPaletteCache*)calloc(1, sizeof(rdpPaletteCache));
 
  if (paletteCache)
  {
    paletteCache->context = context;
    paletteCache->maxEntries = 6;
    paletteCache->entries =
        (PALETTE_TABLE_ENTRY*)calloc(paletteCache->maxEntries, sizeof(PALETTE_TABLE_ENTRY));
  }
 
  return paletteCache;
}
 
void palette_cache_free(rdpPaletteCache* paletteCache)
{
  if (paletteCache)
  {
    for (UINT32 i = 0; i < paletteCache->maxEntries; i++)
      free(paletteCache->entries[i].entry);
 
    free(paletteCache->entries);
    free(paletteCache);
  }
}
 
void free_palette_update(rdpContext* context, PALETTE_UPDATE* pointer)
{
  free(pointer);
}
 
PALETTE_UPDATE* copy_palette_update(rdpContext* context, const PALETTE_UPDATE* pointer)
{
  PALETTE_UPDATE* dst = calloc(1, sizeof(PALETTE_UPDATE));
 
  if (!dst || !pointer)
    goto fail;
 
  *dst = *pointer;
  return dst;
fail:
  WINPR_PRAGMA_DIAG_PUSH
  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
  free_palette_update(context, dst);
  WINPR_PRAGMA_DIAG_POP
  return NULL;
}