|
Packit Service |
fa4841 |
/**
|
|
Packit Service |
fa4841 |
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
Packit Service |
fa4841 |
* Brush Cache
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
Packit Service |
fa4841 |
* you may not use this file except in compliance with the License.
|
|
Packit Service |
fa4841 |
* You may obtain a copy of the License at
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* http://www.apache.org/licenses/LICENSE-2.0
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* Unless required by applicable law or agreed to in writing, software
|
|
Packit Service |
fa4841 |
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
Packit Service |
fa4841 |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
Packit Service |
fa4841 |
* See the License for the specific language governing permissions and
|
|
Packit Service |
fa4841 |
* limitations under the License.
|
|
Packit Service |
fa4841 |
*/
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#ifdef HAVE_CONFIG_H
|
|
Packit Service |
fa4841 |
#include "config.h"
|
|
Packit Service |
fa4841 |
#endif
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include <stdio.h>
|
|
Packit Service |
fa4841 |
#include <winpr/crt.h>
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include <freerdp/log.h>
|
|
Packit Service |
fa4841 |
#include <freerdp/update.h>
|
|
Packit Service |
fa4841 |
#include <freerdp/freerdp.h>
|
|
Packit Service |
fa4841 |
#include <winpr/stream.h>
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include <freerdp/cache/brush.h>
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include "brush.h"
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#define TAG FREERDP_TAG("cache.brush")
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
struct _BRUSH_ENTRY
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
UINT32 bpp;
|
|
Packit Service |
fa4841 |
void* entry;
|
|
Packit Service |
fa4841 |
};
|
|
Packit Service |
fa4841 |
typedef struct _BRUSH_ENTRY BRUSH_ENTRY;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
struct rdp_brush_cache
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
pPatBlt PatBlt; /* 0 */
|
|
Packit Service |
fa4841 |
pCacheBrush CacheBrush; /* 1 */
|
|
Packit Service |
fa4841 |
pPolygonSC PolygonSC; /* 2 */
|
|
Packit Service |
fa4841 |
pPolygonCB PolygonCB; /* 3 */
|
|
Packit Service |
fa4841 |
UINT32 paddingA[16 - 4]; /* 4 */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
UINT32 maxEntries; /* 16 */
|
|
Packit Service |
fa4841 |
UINT32 maxMonoEntries; /* 17 */
|
|
Packit Service |
fa4841 |
BRUSH_ENTRY* entries; /* 18 */
|
|
Packit Service |
fa4841 |
BRUSH_ENTRY* monoEntries; /* 19 */
|
|
Packit Service |
fa4841 |
UINT32 paddingB[32 - 20]; /* 20 */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* internal */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rdpSettings* settings;
|
|
Packit Service |
fa4841 |
};
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static BOOL update_gdi_patblt(rdpContext* context, PATBLT_ORDER* patblt)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
BYTE style;
|
|
Packit Service |
fa4841 |
BOOL ret = TRUE;
|
|
Packit Service |
fa4841 |
rdpBrush* brush = &patblt->brush;
|
|
Packit Service |
fa4841 |
const rdpCache* cache = context->cache;
|
|
Packit Service |
fa4841 |
style = brush->style;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (brush->style & CACHED_BRUSH)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
brush->data = brush_cache_get(cache->brush, brush->index, &brush->bpp);
|
|
Packit Service |
fa4841 |
brush->style = 0x03;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
IFCALLRET(cache->brush->PatBlt, ret, context, patblt);
|
|
Packit Service |
fa4841 |
brush->style = style;
|
|
Packit Service |
fa4841 |
return ret;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static BOOL update_gdi_polygon_sc(rdpContext* context, const POLYGON_SC_ORDER* polygon_sc)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rdpCache* cache = context->cache;
|
|
Packit Service |
fa4841 |
return IFCALLRESULT(TRUE, cache->brush->PolygonSC, context, polygon_sc);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static BOOL update_gdi_polygon_cb(rdpContext* context, POLYGON_CB_ORDER* polygon_cb)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
BYTE style;
|
|
Packit Service |
fa4841 |
rdpBrush* brush = &polygon_cb->brush;
|
|
Packit Service |
fa4841 |
rdpCache* cache = context->cache;
|
|
Packit Service |
fa4841 |
BOOL ret = TRUE;
|
|
Packit Service |
fa4841 |
style = brush->style;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (brush->style & CACHED_BRUSH)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
brush->data = brush_cache_get(cache->brush, brush->index, &brush->bpp);
|
|
Packit Service |
fa4841 |
brush->style = 0x03;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
IFCALLRET(cache->brush->PolygonCB, ret, context, polygon_cb);
|
|
Packit Service |
fa4841 |
brush->style = style;
|
|
Packit Service |
fa4841 |
return ret;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static BOOL update_gdi_cache_brush(rdpContext* context, const CACHE_BRUSH_ORDER* cacheBrush)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
UINT32 length;
|
|
Packit Service |
fa4841 |
void* data = NULL;
|
|
Packit Service |
fa4841 |
rdpCache* cache = context->cache;
|
|
Packit Service |
fa4841 |
length = cacheBrush->bpp * 64 / 8;
|
|
Packit Service |
fa4841 |
data = malloc(length);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!data)
|
|
Packit Service |
fa4841 |
return FALSE;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
CopyMemory(data, cacheBrush->data, length);
|
|
Packit Service |
fa4841 |
brush_cache_put(cache->brush, cacheBrush->index, data, cacheBrush->bpp);
|
|
Packit Service |
fa4841 |
return TRUE;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void* brush_cache_get(rdpBrushCache* brushCache, UINT32 index, UINT32* bpp)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
void* entry;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!brushCache)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!bpp)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (*bpp == 1)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (index >= brushCache->maxMonoEntries)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", *bpp, index);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
*bpp = brushCache->monoEntries[index].bpp;
|
|
Packit Service |
fa4841 |
entry = brushCache->monoEntries[index].entry;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (index >= brushCache->maxEntries)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", *bpp, index);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
*bpp = brushCache->entries[index].bpp;
|
|
Packit Service |
fa4841 |
entry = brushCache->entries[index].entry;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (entry == NULL)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) at index: 0x%08" PRIX32 "", *bpp, index);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return entry;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void brush_cache_put(rdpBrushCache* brushCache, UINT32 index, void* entry, UINT32 bpp)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (bpp == 1)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (index >= brushCache->maxMonoEntries)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", bpp, index);
|
|
Packit Service |
fa4841 |
free(entry);
|
|
Packit Service |
fa4841 |
return;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(brushCache->monoEntries[index].entry);
|
|
Packit Service |
fa4841 |
brushCache->monoEntries[index].bpp = bpp;
|
|
Packit Service |
fa4841 |
brushCache->monoEntries[index].entry = entry;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (index >= brushCache->maxEntries)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "invalid brush (%" PRIu32 " bpp) index: 0x%08" PRIX32 "", bpp, index);
|
|
Packit Service |
fa4841 |
free(entry);
|
|
Packit Service |
fa4841 |
return;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(brushCache->entries[index].entry);
|
|
Packit Service |
fa4841 |
brushCache->entries[index].bpp = bpp;
|
|
Packit Service |
fa4841 |
brushCache->entries[index].entry = entry;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void brush_cache_register_callbacks(rdpUpdate* update)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rdpCache* cache = update->context->cache;
|
|
Packit Service |
fa4841 |
cache->brush->PatBlt = update->primary->PatBlt;
|
|
Packit Service |
fa4841 |
cache->brush->PolygonSC = update->primary->PolygonSC;
|
|
Packit Service |
fa4841 |
cache->brush->PolygonCB = update->primary->PolygonCB;
|
|
Packit Service |
fa4841 |
update->primary->PatBlt = update_gdi_patblt;
|
|
Packit Service |
fa4841 |
update->primary->PolygonSC = update_gdi_polygon_sc;
|
|
Packit Service |
fa4841 |
update->primary->PolygonCB = update_gdi_polygon_cb;
|
|
Packit Service |
fa4841 |
update->secondary->CacheBrush = update_gdi_cache_brush;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rdpBrushCache* brush_cache_new(rdpSettings* settings)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rdpBrushCache* brushCache;
|
|
Packit Service |
fa4841 |
brushCache = (rdpBrushCache*)calloc(1, sizeof(rdpBrushCache));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!brushCache)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
brushCache->settings = settings;
|
|
Packit Service |
fa4841 |
brushCache->maxEntries = 64;
|
|
Packit Service |
fa4841 |
brushCache->maxMonoEntries = 64;
|
|
Packit Service |
fa4841 |
brushCache->entries = (BRUSH_ENTRY*)calloc(brushCache->maxEntries, sizeof(BRUSH_ENTRY));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!brushCache->entries)
|
|
Packit Service |
fa4841 |
goto error_entries;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
brushCache->monoEntries = (BRUSH_ENTRY*)calloc(brushCache->maxMonoEntries, sizeof(BRUSH_ENTRY));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!brushCache->monoEntries)
|
|
Packit Service |
fa4841 |
goto error_mono;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return brushCache;
|
|
Packit Service |
fa4841 |
error_mono:
|
|
Packit Service |
fa4841 |
free(brushCache->entries);
|
|
Packit Service |
fa4841 |
error_entries:
|
|
Packit Service |
fa4841 |
free(brushCache);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void brush_cache_free(rdpBrushCache* brushCache)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int i;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (brushCache)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (brushCache->entries)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
for (i = 0; i < (int)brushCache->maxEntries; i++)
|
|
Packit Service |
fa4841 |
free(brushCache->entries[i].entry);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(brushCache->entries);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (brushCache->monoEntries)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
for (i = 0; i < (int)brushCache->maxMonoEntries; i++)
|
|
Packit Service |
fa4841 |
free(brushCache->monoEntries[i].entry);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(brushCache->monoEntries);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(brushCache);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void free_cache_brush_order(rdpContext* context, CACHE_BRUSH_ORDER* order)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
free(order);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
CACHE_BRUSH_ORDER* copy_cache_brush_order(rdpContext* context, const CACHE_BRUSH_ORDER* order)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
CACHE_BRUSH_ORDER* dst = calloc(1, sizeof(CACHE_BRUSH_ORDER));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!dst || !order)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
*dst = *order;
|
|
Packit Service |
fa4841 |
return dst;
|
|
Packit Service |
fa4841 |
fail:
|
|
Packit Service |
fa4841 |
free_cache_brush_order(context, dst);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|