Coverage Report

Created: 2025-07-11 06:43

/src/FreeRDP/libfreerdp/codec/progressive.c
Line
Count
Source (jump to first uncovered line)
1
/**
2
 * FreeRDP: A Remote Desktop Protocol Implementation
3
 * Progressive Codec Bitmap Compression
4
 *
5
 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
6
 * Copyright 2019 Armin Novak <armin.novak@thincast.com>
7
 * Copyright 2019 Thincast Technologies GmbH
8
 *
9
 * Licensed under the Apache License, Version 2.0 (the "License");
10
 * you may not use this file except in compliance with the License.
11
 * You may obtain a copy of the License at
12
 *
13
 *     http://www.apache.org/licenses/LICENSE-2.0
14
 *
15
 * Unless required by applicable law or agreed to in writing, software
16
 * distributed under the License is distributed on an "AS IS" BASIS,
17
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18
 * See the License for the specific language governing permissions and
19
 * limitations under the License.
20
 */
21
22
#include <freerdp/config.h>
23
24
#include <winpr/assert.h>
25
#include <winpr/cast.h>
26
#include <winpr/crt.h>
27
#include <winpr/print.h>
28
#include <winpr/bitstream.h>
29
30
#include <freerdp/primitives.h>
31
#include <freerdp/codec/color.h>
32
#include <freerdp/codec/progressive.h>
33
#include <freerdp/codec/region.h>
34
#include <freerdp/log.h>
35
36
#include "rfx_differential.h"
37
#include "rfx_quantization.h"
38
#include "rfx_dwt.h"
39
#include "rfx_rlgr.h"
40
#include "rfx_constants.h"
41
#include "rfx_types.h"
42
#include "progressive.h"
43
44
5.54k
#define TAG FREERDP_TAG("codec.progressive")
45
46
typedef struct
47
{
48
  BOOL nonLL;
49
  wBitStream* srl;
50
  wBitStream* raw;
51
52
  /* SRL state */
53
54
  UINT32 kp;
55
  int nz;
56
  BOOL mode;
57
} RFX_PROGRESSIVE_UPGRADE_STATE;
58
59
static INLINE void
60
progressive_component_codec_quant_read(wStream* WINPR_RESTRICT s,
61
                                       RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT quantVal)
62
5.24k
{
63
5.24k
  BYTE b = 0;
64
5.24k
  Stream_Read_UINT8(s, b);
65
5.24k
  quantVal->LL3 = b & 0x0F;
66
5.24k
  quantVal->HL3 = b >> 4;
67
5.24k
  Stream_Read_UINT8(s, b);
68
5.24k
  quantVal->LH3 = b & 0x0F;
69
5.24k
  quantVal->HH3 = b >> 4;
70
5.24k
  Stream_Read_UINT8(s, b);
71
5.24k
  quantVal->HL2 = b & 0x0F;
72
5.24k
  quantVal->LH2 = b >> 4;
73
5.24k
  Stream_Read_UINT8(s, b);
74
5.24k
  quantVal->HH2 = b & 0x0F;
75
5.24k
  quantVal->HL1 = b >> 4;
76
5.24k
  Stream_Read_UINT8(s, b);
77
5.24k
  quantVal->LH1 = b & 0x0F;
78
5.24k
  quantVal->HH1 = b >> 4;
79
5.24k
}
80
81
static INLINE void progressive_rfx_quant_add(const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q1,
82
                                             const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q2,
83
                                             RFX_COMPONENT_CODEC_QUANT* dst)
84
0
{
85
0
  dst->HL1 = q1->HL1 + q2->HL1; /* HL1 */
86
0
  dst->LH1 = q1->LH1 + q2->LH1; /* LH1 */
87
0
  dst->HH1 = q1->HH1 + q2->HH1; /* HH1 */
88
0
  dst->HL2 = q1->HL2 + q2->HL2; /* HL2 */
89
0
  dst->LH2 = q1->LH2 + q2->LH2; /* LH2 */
90
0
  dst->HH2 = q1->HH2 + q2->HH2; /* HH2 */
91
0
  dst->HL3 = q1->HL3 + q2->HL3; /* HL3 */
92
0
  dst->LH3 = q1->LH3 + q2->LH3; /* LH3 */
93
0
  dst->HH3 = q1->HH3 + q2->HH3; /* HH3 */
94
0
  dst->LL3 = q1->LL3 + q2->LL3; /* LL3 */
95
0
}
96
97
static INLINE void progressive_rfx_quant_lsub(RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q, int val)
98
0
{
99
0
  q->HL1 -= val; /* HL1 */
100
0
  q->LH1 -= val; /* LH1 */
101
0
  q->HH1 -= val; /* HH1 */
102
0
  q->HL2 -= val; /* HL2 */
103
0
  q->LH2 -= val; /* LH2 */
104
0
  q->HH2 -= val; /* HH2 */
105
0
  q->HL3 -= val; /* HL3 */
106
0
  q->LH3 -= val; /* LH3 */
107
0
  q->HH3 -= val; /* HH3 */
108
0
  q->LL3 -= val; /* LL3 */
109
0
}
110
111
static INLINE void progressive_rfx_quant_sub(const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q1,
112
                                             const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q2,
113
                                             RFX_COMPONENT_CODEC_QUANT* dst)
114
0
{
115
0
  dst->HL1 = q1->HL1 - q2->HL1; /* HL1 */
116
0
  dst->LH1 = q1->LH1 - q2->LH1; /* LH1 */
117
0
  dst->HH1 = q1->HH1 - q2->HH1; /* HH1 */
118
0
  dst->HL2 = q1->HL2 - q2->HL2; /* HL2 */
119
0
  dst->LH2 = q1->LH2 - q2->LH2; /* LH2 */
120
0
  dst->HH2 = q1->HH2 - q2->HH2; /* HH2 */
121
0
  dst->HL3 = q1->HL3 - q2->HL3; /* HL3 */
122
0
  dst->LH3 = q1->LH3 - q2->LH3; /* LH3 */
123
0
  dst->HH3 = q1->HH3 - q2->HH3; /* HH3 */
124
0
  dst->LL3 = q1->LL3 - q2->LL3; /* LL3 */
125
0
}
126
127
static INLINE BOOL
128
progressive_rfx_quant_lcmp_less_equal(const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q, int val)
129
48
{
130
48
  if (q->HL1 > val)
131
0
    return FALSE; /* HL1 */
132
133
48
  if (q->LH1 > val)
134
0
    return FALSE; /* LH1 */
135
136
48
  if (q->HH1 > val)
137
0
    return FALSE; /* HH1 */
138
139
48
  if (q->HL2 > val)
140
0
    return FALSE; /* HL2 */
141
142
48
  if (q->LH2 > val)
143
0
    return FALSE; /* LH2 */
144
145
48
  if (q->HH2 > val)
146
0
    return FALSE; /* HH2 */
147
148
48
  if (q->HL3 > val)
149
0
    return FALSE; /* HL3 */
150
151
48
  if (q->LH3 > val)
152
0
    return FALSE; /* LH3 */
153
154
48
  if (q->HH3 > val)
155
0
    return FALSE; /* HH3 */
156
157
48
  if (q->LL3 > val)
158
0
    return FALSE; /* LL3 */
159
160
48
  return TRUE;
161
48
}
162
163
static INLINE BOOL
164
progressive_rfx_quant_lcmp_greater_equal(const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q, int val)
165
86
{
166
86
  if (q->HL1 < val)
167
9
    return FALSE; /* HL1 */
168
169
77
  if (q->LH1 < val)
170
4
    return FALSE; /* LH1 */
171
172
73
  if (q->HH1 < val)
173
3
    return FALSE; /* HH1 */
174
175
70
  if (q->HL2 < val)
176
4
    return FALSE; /* HL2 */
177
178
66
  if (q->LH2 < val)
179
2
    return FALSE; /* LH2 */
180
181
64
  if (q->HH2 < val)
182
2
    return FALSE; /* HH2 */
183
184
62
  if (q->HL3 < val)
185
3
    return FALSE; /* HL3 */
186
187
59
  if (q->LH3 < val)
188
3
    return FALSE; /* LH3 */
189
190
56
  if (q->HH3 < val)
191
4
    return FALSE; /* HH3 */
192
193
52
  if (q->LL3 < val)
194
4
    return FALSE; /* LL3 */
195
196
48
  return TRUE;
197
52
}
198
199
static INLINE BOOL
200
progressive_rfx_quant_cmp_equal(const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q1,
201
                                const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT q2)
202
0
{
203
0
  if (q1->HL1 != q2->HL1)
204
0
    return FALSE; /* HL1 */
205
206
0
  if (q1->LH1 != q2->LH1)
207
0
    return FALSE; /* LH1 */
208
209
0
  if (q1->HH1 != q2->HH1)
210
0
    return FALSE; /* HH1 */
211
212
0
  if (q1->HL2 != q2->HL2)
213
0
    return FALSE; /* HL2 */
214
215
0
  if (q1->LH2 != q2->LH2)
216
0
    return FALSE; /* LH2 */
217
218
0
  if (q1->HH2 != q2->HH2)
219
0
    return FALSE; /* HH2 */
220
221
0
  if (q1->HL3 != q2->HL3)
222
0
    return FALSE; /* HL3 */
223
224
0
  if (q1->LH3 != q2->LH3)
225
0
    return FALSE; /* LH3 */
226
227
0
  if (q1->HH3 != q2->HH3)
228
0
    return FALSE; /* HH3 */
229
230
0
  if (q1->LL3 != q2->LL3)
231
0
    return FALSE; /* LL3 */
232
233
0
  return TRUE;
234
0
}
235
236
static INLINE BOOL progressive_set_surface_data(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
237
                                                UINT16 surfaceId,
238
                                                PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT pData)
239
5.54k
{
240
5.54k
  ULONG_PTR key = 0;
241
5.54k
  key = ((ULONG_PTR)surfaceId) + 1;
242
243
5.54k
  if (pData)
244
5.54k
    return HashTable_Insert(progressive->SurfaceContexts, (void*)key, pData);
245
246
0
  HashTable_Remove(progressive->SurfaceContexts, (void*)key);
247
0
  return TRUE;
248
5.54k
}
249
250
static INLINE PROGRESSIVE_SURFACE_CONTEXT*
251
progressive_get_surface_data(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive, UINT16 surfaceId)
252
11.0k
{
253
11.0k
  void* key = (void*)(((ULONG_PTR)surfaceId) + 1);
254
255
11.0k
  if (!progressive)
256
0
    return NULL;
257
258
11.0k
  return HashTable_GetItemValue(progressive->SurfaceContexts, key);
259
11.0k
}
260
261
static void progressive_tile_free(RFX_PROGRESSIVE_TILE* WINPR_RESTRICT tile)
262
1.22M
{
263
1.22M
  if (tile)
264
1.22M
  {
265
1.22M
    winpr_aligned_free(tile->sign);
266
1.22M
    winpr_aligned_free(tile->current);
267
1.22M
    winpr_aligned_free(tile->data);
268
1.22M
    winpr_aligned_free(tile);
269
1.22M
  }
270
1.22M
}
271
272
static void progressive_surface_context_free(void* ptr)
273
11.0k
{
274
11.0k
  PROGRESSIVE_SURFACE_CONTEXT* surface = ptr;
275
276
11.0k
  if (!surface)
277
5.54k
    return;
278
279
5.54k
  if (surface->tiles)
280
5.54k
  {
281
1.23M
    for (size_t index = 0; index < surface->tilesSize; index++)
282
1.22M
    {
283
1.22M
      RFX_PROGRESSIVE_TILE* tile = surface->tiles[index];
284
1.22M
      progressive_tile_free(tile);
285
1.22M
    }
286
5.54k
  }
287
288
5.54k
  winpr_aligned_free((void*)surface->tiles);
289
5.54k
  winpr_aligned_free(surface->updatedTileIndices);
290
5.54k
  winpr_aligned_free(surface);
291
5.54k
}
292
293
static INLINE RFX_PROGRESSIVE_TILE* progressive_tile_new(void)
294
1.22M
{
295
1.22M
  RFX_PROGRESSIVE_TILE* tile = winpr_aligned_calloc(1, sizeof(RFX_PROGRESSIVE_TILE), 32);
296
1.22M
  if (!tile)
297
0
    goto fail;
298
299
1.22M
  tile->width = 64;
300
1.22M
  tile->height = 64;
301
1.22M
  tile->stride = 4 * tile->width;
302
303
1.22M
  size_t dataLen = 1ull * tile->stride * tile->height;
304
1.22M
  tile->data = (BYTE*)winpr_aligned_malloc(dataLen, 16);
305
1.22M
  if (!tile->data)
306
0
    goto fail;
307
1.22M
  memset(tile->data, 0xFF, dataLen);
308
309
1.22M
  size_t signLen = (8192ULL + 32ULL) * 3ULL;
310
1.22M
  tile->sign = (BYTE*)winpr_aligned_malloc(signLen, 16);
311
1.22M
  if (!tile->sign)
312
0
    goto fail;
313
314
1.22M
  size_t currentLen = (8192ULL + 32ULL) * 3ULL;
315
1.22M
  tile->current = (BYTE*)winpr_aligned_malloc(currentLen, 16);
316
1.22M
  if (!tile->current)
317
0
    goto fail;
318
319
1.22M
  return tile;
320
321
0
fail:
322
0
  progressive_tile_free(tile);
323
0
  return NULL;
324
1.22M
}
325
326
static INLINE BOOL
327
progressive_allocate_tile_cache(PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface, size_t min)
328
5.54k
{
329
5.54k
  size_t oldIndex = 0;
330
331
5.54k
  WINPR_ASSERT(surface);
332
5.54k
  WINPR_ASSERT(surface->gridSize > 0);
333
334
5.54k
  if (surface->tiles)
335
0
  {
336
0
    oldIndex = surface->gridSize;
337
0
    while (surface->gridSize < min)
338
0
      surface->gridSize += 1024;
339
0
  }
340
341
5.54k
  void* tmp = winpr_aligned_recalloc((void*)surface->tiles, surface->gridSize,
342
5.54k
                                     sizeof(RFX_PROGRESSIVE_TILE*), 32);
343
5.54k
  if (!tmp)
344
0
    return FALSE;
345
5.54k
  surface->tilesSize = surface->gridSize;
346
5.54k
  surface->tiles = (RFX_PROGRESSIVE_TILE**)tmp;
347
348
1.23M
  for (size_t x = oldIndex; x < surface->tilesSize; x++)
349
1.22M
  {
350
1.22M
    surface->tiles[x] = progressive_tile_new();
351
1.22M
    if (!surface->tiles[x])
352
0
      return FALSE;
353
1.22M
  }
354
355
5.54k
  tmp =
356
5.54k
      winpr_aligned_recalloc(surface->updatedTileIndices, surface->gridSize, sizeof(UINT32), 32);
357
5.54k
  if (!tmp)
358
0
    return FALSE;
359
360
5.54k
  surface->updatedTileIndices = tmp;
361
362
5.54k
  return TRUE;
363
5.54k
}
364
365
static PROGRESSIVE_SURFACE_CONTEXT* progressive_surface_context_new(UINT16 surfaceId, UINT32 width,
366
                                                                    UINT32 height)
367
5.54k
{
368
5.54k
  PROGRESSIVE_SURFACE_CONTEXT* surface = (PROGRESSIVE_SURFACE_CONTEXT*)winpr_aligned_calloc(
369
5.54k
      1, sizeof(PROGRESSIVE_SURFACE_CONTEXT), 32);
370
371
5.54k
  if (!surface)
372
0
    return NULL;
373
374
5.54k
  surface->id = surfaceId;
375
5.54k
  surface->width = width;
376
5.54k
  surface->height = height;
377
5.54k
  surface->gridWidth = (width + (64 - width % 64)) / 64;
378
5.54k
  surface->gridHeight = (height + (64 - height % 64)) / 64;
379
5.54k
  surface->gridSize = surface->gridWidth * surface->gridHeight;
380
381
5.54k
  if (!progressive_allocate_tile_cache(surface, surface->gridSize))
382
0
  {
383
0
    progressive_surface_context_free(surface);
384
0
    return NULL;
385
0
  }
386
387
5.54k
  return surface;
388
5.54k
}
389
390
static INLINE BOOL
391
progressive_surface_tile_replace(PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
392
                                 PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
393
                                 const RFX_PROGRESSIVE_TILE* WINPR_RESTRICT tile, BOOL upgrade)
394
108
{
395
108
  RFX_PROGRESSIVE_TILE* t = NULL;
396
397
108
  size_t zIdx = 0;
398
108
  if (!surface || !tile)
399
0
    return FALSE;
400
401
108
  zIdx = (tile->yIdx * surface->gridWidth) + tile->xIdx;
402
403
108
  if (zIdx >= surface->tilesSize)
404
20
  {
405
20
    WLog_ERR(TAG, "Invalid zIndex %" PRIuz, zIdx);
406
20
    return FALSE;
407
20
  }
408
409
88
  t = surface->tiles[zIdx];
410
411
88
  t->blockType = tile->blockType;
412
88
  t->blockLen = tile->blockLen;
413
88
  t->quantIdxY = tile->quantIdxY;
414
88
  t->quantIdxCb = tile->quantIdxCb;
415
88
  t->quantIdxCr = tile->quantIdxCr;
416
88
  t->xIdx = tile->xIdx;
417
88
  t->yIdx = tile->yIdx;
418
88
  t->flags = tile->flags;
419
88
  t->quality = tile->quality;
420
88
  t->x = tile->xIdx * t->width;
421
88
  t->y = tile->yIdx * t->height;
422
423
88
  if (upgrade)
424
53
  {
425
53
    t->ySrlLen = tile->ySrlLen;
426
53
    t->yRawLen = tile->yRawLen;
427
53
    t->cbSrlLen = tile->cbSrlLen;
428
53
    t->cbRawLen = tile->cbRawLen;
429
53
    t->crSrlLen = tile->crSrlLen;
430
53
    t->crRawLen = tile->crRawLen;
431
53
    t->ySrlData = tile->ySrlData;
432
53
    t->yRawData = tile->yRawData;
433
53
    t->cbSrlData = tile->cbSrlData;
434
53
    t->cbRawData = tile->cbRawData;
435
53
    t->crSrlData = tile->crSrlData;
436
53
    t->crRawData = tile->crRawData;
437
53
  }
438
35
  else
439
35
  {
440
35
    t->yLen = tile->yLen;
441
35
    t->cbLen = tile->cbLen;
442
35
    t->crLen = tile->crLen;
443
35
    t->tailLen = tile->tailLen;
444
35
    t->yData = tile->yData;
445
35
    t->cbData = tile->cbData;
446
35
    t->crData = tile->crData;
447
35
    t->tailData = tile->tailData;
448
35
  }
449
450
88
  if (region->usedTiles >= region->numTiles)
451
4
  {
452
4
    WLog_ERR(TAG, "Invalid tile count, only expected %" PRIu16 ", got %" PRIu16,
453
4
             region->numTiles, region->usedTiles);
454
4
    return FALSE;
455
4
  }
456
457
84
  region->tiles[region->usedTiles++] = t;
458
84
  if (!t->dirty)
459
58
  {
460
58
    if (surface->numUpdatedTiles >= surface->gridSize)
461
0
    {
462
0
      if (!progressive_allocate_tile_cache(surface, surface->numUpdatedTiles + 1))
463
0
        return FALSE;
464
0
    }
465
466
58
    surface->updatedTileIndices[surface->numUpdatedTiles++] = (UINT32)zIdx;
467
58
  }
468
469
84
  t->dirty = TRUE;
470
84
  return TRUE;
471
84
}
472
473
INT32 progressive_create_surface_context(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
474
                                         UINT16 surfaceId, UINT32 width, UINT32 height)
475
5.54k
{
476
5.54k
  PROGRESSIVE_SURFACE_CONTEXT* surface = progressive_get_surface_data(progressive, surfaceId);
477
478
5.54k
  if (!surface)
479
5.54k
  {
480
5.54k
    surface = progressive_surface_context_new(surfaceId, width, height);
481
482
5.54k
    if (!surface)
483
0
      return -1;
484
485
5.54k
    if (!progressive_set_surface_data(progressive, surfaceId, (void*)surface))
486
0
    {
487
0
      progressive_surface_context_free(surface);
488
0
      return -1;
489
0
    }
490
5.54k
  }
491
492
5.54k
  return 1;
493
5.54k
}
494
495
int progressive_delete_surface_context(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
496
                                       UINT16 surfaceId)
497
0
{
498
0
  progressive_set_surface_data(progressive, surfaceId, NULL);
499
500
0
  return 1;
501
0
}
502
503
/*
504
 * Band     Offset      Dimensions  Size
505
 *
506
 * HL1      0           31x33       1023
507
 * LH1      1023        33x31       1023
508
 * HH1      2046        31x31       961
509
 *
510
 * HL2      3007        16x17       272
511
 * LH2      3279        17x16       272
512
 * HH2      3551        16x16       256
513
 *
514
 * HL3      3807        8x9         72
515
 * LH3      3879        9x8         72
516
 * HH3      3951        8x8         64
517
 *
518
 * LL3      4015        9x9         81
519
 */
520
521
static int16_t clampi16(int val)
522
0
{
523
0
  if (val < INT16_MIN)
524
0
    return INT16_MIN;
525
0
  if (val > INT16_MAX)
526
0
    return INT16_MAX;
527
0
  return (int16_t)val;
528
0
}
529
530
static INLINE void progressive_rfx_idwt_x(const INT16* WINPR_RESTRICT pLowBand, size_t nLowStep,
531
                                          const INT16* WINPR_RESTRICT pHighBand, size_t nHighStep,
532
                                          INT16* WINPR_RESTRICT pDstBand, size_t nDstStep,
533
                                          size_t nLowCount, size_t nHighCount, size_t nDstCount)
534
0
{
535
0
  INT16 H1 = 0;
536
0
  INT16 X1 = 0;
537
538
0
  for (size_t i = 0; i < nDstCount; i++)
539
0
  {
540
0
    const INT16* pL = pLowBand;
541
0
    const INT16* pH = pHighBand;
542
0
    INT16* pX = pDstBand;
543
0
    INT16 H0 = *pH++;
544
0
    INT16 L0 = *pL++;
545
0
    INT16 X0 = clampi16((int32_t)L0 - H0);
546
0
    INT16 X2 = clampi16((int32_t)L0 - H0);
547
548
0
    for (size_t j = 0; j < (nHighCount - 1); j++)
549
0
    {
550
0
      H1 = *pH;
551
0
      pH++;
552
0
      L0 = *pL;
553
0
      pL++;
554
0
      X2 = clampi16((int32_t)L0 - ((H0 + H1) / 2));
555
0
      X1 = clampi16((int32_t)((X0 + X2) / 2) + (2 * H0));
556
0
      pX[0] = X0;
557
0
      pX[1] = X1;
558
0
      pX += 2;
559
0
      X0 = X2;
560
0
      H0 = H1;
561
0
    }
562
563
0
    if (nLowCount <= (nHighCount + 1))
564
0
    {
565
0
      if (nLowCount <= nHighCount)
566
0
      {
567
0
        pX[0] = X2;
568
0
        pX[1] = clampi16((int32_t)X2 + (2 * H0));
569
0
      }
570
0
      else
571
0
      {
572
0
        L0 = *pL;
573
0
        pL++;
574
0
        X0 = clampi16((int32_t)L0 - H0);
575
0
        pX[0] = X2;
576
0
        pX[1] = clampi16((int32_t)((X0 + X2) / 2) + (2 * H0));
577
0
        pX[2] = X0;
578
0
      }
579
0
    }
580
0
    else
581
0
    {
582
0
      L0 = *pL;
583
0
      pL++;
584
0
      X0 = clampi16((int32_t)L0 - (H0 / 2));
585
0
      pX[0] = X2;
586
0
      pX[1] = clampi16((int32_t)((X0 + X2) / 2) + (2 * H0));
587
0
      pX[2] = X0;
588
0
      L0 = *pL;
589
0
      pL++;
590
0
      pX[3] = clampi16((int32_t)(X0 + L0) / 2);
591
0
    }
592
593
0
    pLowBand += nLowStep;
594
0
    pHighBand += nHighStep;
595
0
    pDstBand += nDstStep;
596
0
  }
597
0
}
598
599
static INLINE void progressive_rfx_idwt_y(const INT16* WINPR_RESTRICT pLowBand, size_t nLowStep,
600
                                          const INT16* WINPR_RESTRICT pHighBand, size_t nHighStep,
601
                                          INT16* WINPR_RESTRICT pDstBand, size_t nDstStep,
602
                                          size_t nLowCount, size_t nHighCount, size_t nDstCount)
603
0
{
604
0
  for (size_t i = 0; i < nDstCount; i++)
605
0
  {
606
0
    INT16 H1 = 0;
607
0
    INT16 X1 = 0;
608
0
    const INT16* pL = pLowBand;
609
0
    const INT16* pH = pHighBand;
610
0
    INT16* pX = pDstBand;
611
0
    INT16 H0 = *pH;
612
0
    pH += nHighStep;
613
0
    INT16 L0 = *pL;
614
0
    pL += nLowStep;
615
0
    int16_t X0 = clampi16((int32_t)L0 - H0);
616
0
    int16_t X2 = clampi16((int32_t)L0 - H0);
617
618
0
    for (size_t j = 0; j < (nHighCount - 1); j++)
619
0
    {
620
0
      H1 = *pH;
621
0
      pH += nHighStep;
622
0
      L0 = *pL;
623
0
      pL += nLowStep;
624
0
      X2 = clampi16((int32_t)L0 - ((H0 + H1) / 2));
625
0
      X1 = clampi16((int32_t)((X0 + X2) / 2) + (2 * H0));
626
0
      *pX = X0;
627
0
      pX += nDstStep;
628
0
      *pX = X1;
629
0
      pX += nDstStep;
630
0
      X0 = X2;
631
0
      H0 = H1;
632
0
    }
633
634
0
    if (nLowCount <= (nHighCount + 1))
635
0
    {
636
0
      if (nLowCount <= nHighCount)
637
0
      {
638
0
        *pX = X2;
639
0
        pX += nDstStep;
640
0
        *pX = clampi16((int32_t)X2 + (2 * H0));
641
0
      }
642
0
      else
643
0
      {
644
0
        L0 = *pL;
645
0
        X0 = clampi16((int32_t)L0 - H0);
646
0
        *pX = X2;
647
0
        pX += nDstStep;
648
0
        *pX = clampi16((int32_t)((X0 + X2) / 2) + (2 * H0));
649
0
        pX += nDstStep;
650
0
        *pX = X0;
651
0
      }
652
0
    }
653
0
    else
654
0
    {
655
0
      L0 = *pL;
656
0
      pL += nLowStep;
657
0
      X0 = clampi16((int32_t)L0 - (H0 / 2));
658
0
      *pX = X2;
659
0
      pX += nDstStep;
660
0
      *pX = clampi16((int32_t)((X0 + X2) / 2) + (2 * H0));
661
0
      pX += nDstStep;
662
0
      *pX = X0;
663
0
      pX += nDstStep;
664
0
      L0 = *pL;
665
0
      *pX = clampi16((int32_t)(X0 + L0) / 2);
666
0
    }
667
668
0
    pLowBand++;
669
0
    pHighBand++;
670
0
    pDstBand++;
671
0
  }
672
0
}
673
674
static INLINE size_t progressive_rfx_get_band_l_count(size_t level)
675
0
{
676
0
  return (64 >> level) + 1;
677
0
}
678
679
static INLINE size_t progressive_rfx_get_band_h_count(size_t level)
680
0
{
681
0
  if (level == 1)
682
0
    return (64 >> 1) - 1;
683
0
  else
684
0
    return (64 + (1 << (level - 1))) >> level;
685
0
}
686
687
static INLINE void progressive_rfx_dwt_2d_decode_block(INT16* WINPR_RESTRICT buffer,
688
                                                       INT16* WINPR_RESTRICT temp, size_t level)
689
0
{
690
0
  size_t nDstStepX = 0;
691
0
  size_t nDstStepY = 0;
692
0
  const INT16* WINPR_RESTRICT HL = NULL;
693
0
  const INT16* WINPR_RESTRICT LH = NULL;
694
0
  const INT16* WINPR_RESTRICT HH = NULL;
695
0
  INT16* WINPR_RESTRICT LL = NULL;
696
0
  INT16* WINPR_RESTRICT L = NULL;
697
0
  INT16* WINPR_RESTRICT H = NULL;
698
0
  INT16* WINPR_RESTRICT LLx = NULL;
699
700
0
  const size_t nBandL = progressive_rfx_get_band_l_count(level);
701
0
  const size_t nBandH = progressive_rfx_get_band_h_count(level);
702
0
  size_t offset = 0;
703
704
0
  HL = &buffer[offset];
705
0
  offset += (nBandH * nBandL);
706
0
  LH = &buffer[offset];
707
0
  offset += (nBandL * nBandH);
708
0
  HH = &buffer[offset];
709
0
  offset += (nBandH * nBandH);
710
0
  LL = &buffer[offset];
711
0
  nDstStepX = (nBandL + nBandH);
712
0
  nDstStepY = (nBandL + nBandH);
713
0
  offset = 0;
714
0
  L = &temp[offset];
715
0
  offset += (nBandL * nDstStepX);
716
0
  H = &temp[offset];
717
0
  LLx = &buffer[0];
718
719
  /* horizontal (LL + HL -> L) */
720
0
  progressive_rfx_idwt_x(LL, nBandL, HL, nBandH, L, nDstStepX, nBandL, nBandH, nBandL);
721
722
  /* horizontal (LH + HH -> H) */
723
0
  progressive_rfx_idwt_x(LH, nBandL, HH, nBandH, H, nDstStepX, nBandL, nBandH, nBandH);
724
725
  /* vertical (L + H -> LL) */
726
0
  progressive_rfx_idwt_y(L, nDstStepX, H, nDstStepX, LLx, nDstStepY, nBandL, nBandH,
727
0
                         nBandL + nBandH);
728
0
}
729
730
void rfx_dwt_2d_extrapolate_decode(INT16* WINPR_RESTRICT buffer, INT16* WINPR_RESTRICT temp)
731
0
{
732
0
  WINPR_ASSERT(buffer);
733
0
  WINPR_ASSERT(temp);
734
0
  progressive_rfx_dwt_2d_decode_block(&buffer[3807], temp, 3);
735
0
  progressive_rfx_dwt_2d_decode_block(&buffer[3007], temp, 2);
736
0
  progressive_rfx_dwt_2d_decode_block(&buffer[0], temp, 1);
737
0
}
738
739
static INLINE int progressive_rfx_dwt_2d_decode(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
740
                                                INT16* WINPR_RESTRICT buffer,
741
                                                INT16* WINPR_RESTRICT current, BOOL coeffDiff,
742
                                                BOOL extrapolate, BOOL reverse)
743
0
{
744
0
  const primitives_t* prims = primitives_get();
745
746
0
  if (!progressive || !buffer || !current)
747
0
    return -1;
748
749
0
  const uint32_t belements = 4096;
750
0
  const uint32_t bsize = belements * sizeof(INT16);
751
0
  if (reverse)
752
0
    memcpy(buffer, current, bsize);
753
0
  else if (!coeffDiff)
754
0
    memcpy(current, buffer, bsize);
755
0
  else
756
0
    prims->add_16s_inplace(buffer, current, belements);
757
758
0
  INT16* temp = (INT16*)BufferPool_Take(progressive->bufferPool, -1); /* DWT buffer */
759
760
0
  if (!temp)
761
0
    return -2;
762
763
0
  if (!extrapolate)
764
0
  {
765
0
    progressive->rfx_context->dwt_2d_decode(buffer, temp);
766
0
  }
767
0
  else
768
0
  {
769
0
    WINPR_ASSERT(progressive->rfx_context->dwt_2d_extrapolate_decode);
770
0
    progressive->rfx_context->dwt_2d_extrapolate_decode(buffer, temp);
771
0
  }
772
0
  BufferPool_Return(progressive->bufferPool, temp);
773
0
  return 1;
774
0
}
775
776
static INLINE void progressive_rfx_decode_block(const primitives_t* prims,
777
                                                INT16* WINPR_RESTRICT buffer, UINT32 length,
778
                                                UINT32 shift)
779
0
{
780
0
  if (!shift)
781
0
    return;
782
783
0
  prims->lShiftC_16s_inplace(buffer, shift, length);
784
0
}
785
786
static INLINE int
787
progressive_rfx_decode_component(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
788
                                 const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT shift,
789
                                 const BYTE* WINPR_RESTRICT data, UINT32 length,
790
                                 INT16* WINPR_RESTRICT buffer, INT16* WINPR_RESTRICT current,
791
                                 INT16* WINPR_RESTRICT sign, BOOL coeffDiff,
792
                                 WINPR_ATTR_UNUSED BOOL subbandDiff, BOOL extrapolate)
793
0
{
794
0
  int status = 0;
795
0
  const primitives_t* prims = primitives_get();
796
797
0
  status = progressive->rfx_context->rlgr_decode(RLGR1, data, length, buffer, 4096);
798
799
0
  if (status < 0)
800
0
    return status;
801
802
0
  CopyMemory(sign, buffer, 4096ULL * 2ULL);
803
0
  if (!extrapolate)
804
0
  {
805
0
    rfx_differential_decode(buffer + 4032, 64);
806
0
    progressive_rfx_decode_block(prims, &buffer[0], 1024, shift->HL1);    /* HL1 */
807
0
    progressive_rfx_decode_block(prims, &buffer[1024], 1024, shift->LH1); /* LH1 */
808
0
    progressive_rfx_decode_block(prims, &buffer[2048], 1024, shift->HH1); /* HH1 */
809
0
    progressive_rfx_decode_block(prims, &buffer[3072], 256, shift->HL2);  /* HL2 */
810
0
    progressive_rfx_decode_block(prims, &buffer[3328], 256, shift->LH2);  /* LH2 */
811
0
    progressive_rfx_decode_block(prims, &buffer[3584], 256, shift->HH2);  /* HH2 */
812
0
    progressive_rfx_decode_block(prims, &buffer[3840], 64, shift->HL3);   /* HL3 */
813
0
    progressive_rfx_decode_block(prims, &buffer[3904], 64, shift->LH3);   /* LH3 */
814
0
    progressive_rfx_decode_block(prims, &buffer[3968], 64, shift->HH3);   /* HH3 */
815
0
    progressive_rfx_decode_block(prims, &buffer[4032], 64, shift->LL3);   /* LL3 */
816
0
  }
817
0
  else
818
0
  {
819
0
    progressive_rfx_decode_block(prims, &buffer[0], 1023, shift->HL1);    /* HL1 */
820
0
    progressive_rfx_decode_block(prims, &buffer[1023], 1023, shift->LH1); /* LH1 */
821
0
    progressive_rfx_decode_block(prims, &buffer[2046], 961, shift->HH1);  /* HH1 */
822
0
    progressive_rfx_decode_block(prims, &buffer[3007], 272, shift->HL2);  /* HL2 */
823
0
    progressive_rfx_decode_block(prims, &buffer[3279], 272, shift->LH2);  /* LH2 */
824
0
    progressive_rfx_decode_block(prims, &buffer[3551], 256, shift->HH2);  /* HH2 */
825
0
    progressive_rfx_decode_block(prims, &buffer[3807], 72, shift->HL3);   /* HL3 */
826
0
    progressive_rfx_decode_block(prims, &buffer[3879], 72, shift->LH3);   /* LH3 */
827
0
    progressive_rfx_decode_block(prims, &buffer[3951], 64, shift->HH3);   /* HH3 */
828
0
    rfx_differential_decode(&buffer[4015], 81);                           /* LL3 */
829
0
    progressive_rfx_decode_block(prims, &buffer[4015], 81, shift->LL3);   /* LL3 */
830
0
  }
831
0
  return progressive_rfx_dwt_2d_decode(progressive, buffer, current, coeffDiff, extrapolate,
832
0
                                       FALSE);
833
0
}
834
835
static INLINE int
836
progressive_decompress_tile_first(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
837
                                  RFX_PROGRESSIVE_TILE* WINPR_RESTRICT tile,
838
                                  PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
839
                                  const PROGRESSIVE_BLOCK_CONTEXT* WINPR_RESTRICT context)
840
0
{
841
0
  int rc = 0;
842
0
  BOOL diff = 0;
843
0
  BOOL sub = 0;
844
0
  BOOL extrapolate = 0;
845
0
  BYTE* pBuffer = NULL;
846
0
  INT16* pSign[3];
847
0
  INT16* pSrcDst[3];
848
0
  INT16* pCurrent[3];
849
0
  RFX_COMPONENT_CODEC_QUANT shiftY = { 0 };
850
0
  RFX_COMPONENT_CODEC_QUANT shiftCb = { 0 };
851
0
  RFX_COMPONENT_CODEC_QUANT shiftCr = { 0 };
852
0
  RFX_COMPONENT_CODEC_QUANT* quantY = NULL;
853
0
  RFX_COMPONENT_CODEC_QUANT* quantCb = NULL;
854
0
  RFX_COMPONENT_CODEC_QUANT* quantCr = NULL;
855
0
  RFX_COMPONENT_CODEC_QUANT* quantProgY = NULL;
856
0
  RFX_COMPONENT_CODEC_QUANT* quantProgCb = NULL;
857
0
  RFX_COMPONENT_CODEC_QUANT* quantProgCr = NULL;
858
0
  RFX_PROGRESSIVE_CODEC_QUANT* quantProgVal = NULL;
859
0
  static const prim_size_t roi_64x64 = { 64, 64 };
860
0
  const primitives_t* prims = primitives_get();
861
862
0
  tile->pass = 1;
863
0
  diff = tile->flags & RFX_TILE_DIFFERENCE;
864
0
  sub = context->flags & RFX_SUBBAND_DIFFING;
865
0
  extrapolate = region->flags & RFX_DWT_REDUCE_EXTRAPOLATE;
866
867
#if defined(WITH_DEBUG_CODECS)
868
  WLog_Print(progressive->log, WLOG_DEBUG,
869
             "ProgressiveTile%s: quantIdx Y: %" PRIu8 " Cb: %" PRIu8 " Cr: %" PRIu8
870
             " xIdx: %" PRIu16 " yIdx: %" PRIu16 " flags: 0x%02" PRIX8 " quality: %" PRIu8
871
             " yLen: %" PRIu16 " cbLen: %" PRIu16 " crLen: %" PRIu16 " tailLen: %" PRIu16 "",
872
             (tile->blockType == PROGRESSIVE_WBT_TILE_FIRST) ? "First" : "Simple",
873
             tile->quantIdxY, tile->quantIdxCb, tile->quantIdxCr, tile->xIdx, tile->yIdx,
874
             tile->flags, tile->quality, tile->yLen, tile->cbLen, tile->crLen, tile->tailLen);
875
#endif
876
877
0
  if (tile->quantIdxY >= region->numQuant)
878
0
  {
879
0
    WLog_ERR(TAG, "quantIdxY %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxY, region->numQuant);
880
0
    return -1;
881
0
  }
882
883
0
  quantY = &(region->quantVals[tile->quantIdxY]);
884
885
0
  if (tile->quantIdxCb >= region->numQuant)
886
0
  {
887
0
    WLog_ERR(TAG, "quantIdxCb %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCb,
888
0
             region->numQuant);
889
0
    return -1;
890
0
  }
891
892
0
  quantCb = &(region->quantVals[tile->quantIdxCb]);
893
894
0
  if (tile->quantIdxCr >= region->numQuant)
895
0
  {
896
0
    WLog_ERR(TAG, "quantIdxCr %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCr,
897
0
             region->numQuant);
898
0
    return -1;
899
0
  }
900
901
0
  quantCr = &(region->quantVals[tile->quantIdxCr]);
902
903
0
  if (tile->quality == 0xFF)
904
0
  {
905
0
    quantProgVal = &(progressive->quantProgValFull);
906
0
  }
907
0
  else
908
0
  {
909
0
    if (tile->quality >= region->numProgQuant)
910
0
    {
911
0
      WLog_ERR(TAG, "quality %" PRIu8 " > numProgQuant %" PRIu8, tile->quality,
912
0
               region->numProgQuant);
913
0
      return -1;
914
0
    }
915
916
0
    quantProgVal = &(region->quantProgVals[tile->quality]);
917
0
  }
918
919
0
  quantProgY = &(quantProgVal->yQuantValues);
920
0
  quantProgCb = &(quantProgVal->cbQuantValues);
921
0
  quantProgCr = &(quantProgVal->crQuantValues);
922
923
0
  tile->yQuant = *quantY;
924
0
  tile->cbQuant = *quantCb;
925
0
  tile->crQuant = *quantCr;
926
0
  tile->yProgQuant = *quantProgY;
927
0
  tile->cbProgQuant = *quantProgCb;
928
0
  tile->crProgQuant = *quantProgCr;
929
930
0
  progressive_rfx_quant_add(quantY, quantProgY, &(tile->yBitPos));
931
0
  progressive_rfx_quant_add(quantCb, quantProgCb, &(tile->cbBitPos));
932
0
  progressive_rfx_quant_add(quantCr, quantProgCr, &(tile->crBitPos));
933
0
  progressive_rfx_quant_add(quantY, quantProgY, &shiftY);
934
0
  progressive_rfx_quant_lsub(&shiftY, 1); /* -6 + 5 = -1 */
935
0
  progressive_rfx_quant_add(quantCb, quantProgCb, &shiftCb);
936
0
  progressive_rfx_quant_lsub(&shiftCb, 1); /* -6 + 5 = -1 */
937
0
  progressive_rfx_quant_add(quantCr, quantProgCr, &shiftCr);
938
0
  progressive_rfx_quant_lsub(&shiftCr, 1); /* -6 + 5 = -1 */
939
940
0
  pSign[0] = (INT16*)((&tile->sign[((8192 + 32) * 0) + 16])); /* Y/R buffer */
941
0
  pSign[1] = (INT16*)((&tile->sign[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
942
0
  pSign[2] = (INT16*)((&tile->sign[((8192 + 32) * 2) + 16])); /* Cr/B buffer */
943
944
0
  pCurrent[0] = (INT16*)((&tile->current[((8192 + 32) * 0) + 16])); /* Y/R buffer */
945
0
  pCurrent[1] = (INT16*)((&tile->current[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
946
0
  pCurrent[2] = (INT16*)((&tile->current[((8192 + 32) * 2) + 16])); /* Cr/B buffer */
947
948
0
  pBuffer = (BYTE*)BufferPool_Take(progressive->bufferPool, -1);
949
0
  pSrcDst[0] = (INT16*)((&pBuffer[((8192 + 32) * 0) + 16])); /* Y/R buffer */
950
0
  pSrcDst[1] = (INT16*)((&pBuffer[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
951
0
  pSrcDst[2] = (INT16*)((&pBuffer[((8192 + 32) * 2) + 16])); /* Cr/B buffer */
952
953
0
  rc = progressive_rfx_decode_component(progressive, &shiftY, tile->yData, tile->yLen, pSrcDst[0],
954
0
                                        pCurrent[0], pSign[0], diff, sub, extrapolate); /* Y */
955
0
  if (rc < 0)
956
0
    goto fail;
957
0
  rc = progressive_rfx_decode_component(progressive, &shiftCb, tile->cbData, tile->cbLen,
958
0
                                        pSrcDst[1], pCurrent[1], pSign[1], diff, sub,
959
0
                                        extrapolate); /* Cb */
960
0
  if (rc < 0)
961
0
    goto fail;
962
0
  rc = progressive_rfx_decode_component(progressive, &shiftCr, tile->crData, tile->crLen,
963
0
                                        pSrcDst[2], pCurrent[2], pSign[2], diff, sub,
964
0
                                        extrapolate); /* Cr */
965
0
  if (rc < 0)
966
0
    goto fail;
967
968
0
  const INT16** ptr = WINPR_REINTERPRET_CAST(pSrcDst, INT16**, const INT16**);
969
0
  rc = prims->yCbCrToRGB_16s8u_P3AC4R(ptr, 64 * 2, tile->data, tile->stride, progressive->format,
970
0
                                      &roi_64x64);
971
0
fail:
972
0
  BufferPool_Return(progressive->bufferPool, pBuffer);
973
0
  return rc;
974
0
}
975
976
static INLINE INT16 progressive_rfx_srl_read(RFX_PROGRESSIVE_UPGRADE_STATE* WINPR_RESTRICT state,
977
                                             UINT32 numBits)
978
0
{
979
0
  WINPR_ASSERT(state);
980
981
0
  wBitStream* bs = state->srl;
982
0
  WINPR_ASSERT(bs);
983
984
0
  if (state->nz)
985
0
  {
986
0
    state->nz--;
987
0
    return 0;
988
0
  }
989
990
0
  const UINT32 k = state->kp / 8;
991
992
0
  if (!state->mode)
993
0
  {
994
    /* zero encoding */
995
0
    const UINT32 bit = (bs->accumulator & 0x80000000) ? 1 : 0;
996
0
    BitStream_Shift(bs, 1);
997
998
0
    if (!bit)
999
0
    {
1000
      /* '0' bit, nz >= (1 << k), nz = (1 << k) */
1001
0
      state->nz = (1 << k);
1002
0
      state->kp += 4;
1003
1004
0
      if (state->kp > 80)
1005
0
        state->kp = 80;
1006
1007
0
      state->nz--;
1008
0
      return 0;
1009
0
    }
1010
0
    else
1011
0
    {
1012
      /* '1' bit, nz < (1 << k), nz = next k bits */
1013
0
      state->nz = 0;
1014
0
      state->mode = 1; /* unary encoding is next */
1015
1016
0
      if (k)
1017
0
      {
1018
0
        bs->mask = ((1 << k) - 1);
1019
0
        state->nz =
1020
0
            WINPR_ASSERTING_INT_CAST(int16_t, ((bs->accumulator >> (32u - k)) & bs->mask));
1021
0
        BitStream_Shift(bs, k);
1022
0
      }
1023
1024
0
      if (state->nz)
1025
0
      {
1026
0
        state->nz--;
1027
0
        return 0;
1028
0
      }
1029
0
    }
1030
0
  }
1031
1032
0
  state->mode = 0; /* zero encoding is next */
1033
  /* unary encoding */
1034
  /* read sign bit */
1035
0
  const UINT32 sign = (bs->accumulator & 0x80000000) ? 1 : 0;
1036
0
  BitStream_Shift(bs, 1);
1037
1038
0
  if (state->kp < 6)
1039
0
    state->kp = 0;
1040
0
  else
1041
0
    state->kp -= 6;
1042
1043
0
  if (numBits == 1)
1044
0
    return sign ? -1 : 1;
1045
1046
0
  UINT32 mag = 1;
1047
0
  const UINT32 max = (1 << numBits) - 1;
1048
1049
0
  while (mag < max)
1050
0
  {
1051
0
    const UINT32 bit = (bs->accumulator & 0x80000000) ? 1 : 0;
1052
0
    BitStream_Shift(bs, 1);
1053
1054
0
    if (bit)
1055
0
      break;
1056
1057
0
    mag++;
1058
0
  }
1059
1060
0
  if (mag > INT16_MAX)
1061
0
    mag = INT16_MAX;
1062
0
  return (INT16)(sign ? -1 * (int)mag : (INT16)mag);
1063
0
}
1064
1065
static INLINE int
1066
progressive_rfx_upgrade_state_finish(RFX_PROGRESSIVE_UPGRADE_STATE* WINPR_RESTRICT state)
1067
0
{
1068
0
  UINT32 pad = 0;
1069
0
  wBitStream* srl = NULL;
1070
0
  wBitStream* raw = NULL;
1071
0
  if (!state)
1072
0
    return -1;
1073
1074
0
  srl = state->srl;
1075
0
  raw = state->raw;
1076
  /* Read trailing bits from RAW/SRL bit streams */
1077
0
  pad = (raw->position % 8) ? (8 - (raw->position % 8)) : 0;
1078
1079
0
  if (pad)
1080
0
    BitStream_Shift(raw, pad);
1081
1082
0
  pad = (srl->position % 8) ? (8 - (srl->position % 8)) : 0;
1083
1084
0
  if (pad)
1085
0
    BitStream_Shift(srl, pad);
1086
1087
0
  if (BitStream_GetRemainingLength(srl) == 8)
1088
0
    BitStream_Shift(srl, 8);
1089
1090
0
  return 1;
1091
0
}
1092
1093
static INLINE int progressive_rfx_upgrade_block(RFX_PROGRESSIVE_UPGRADE_STATE* WINPR_RESTRICT state,
1094
                                                INT16* WINPR_RESTRICT buffer,
1095
                                                INT16* WINPR_RESTRICT sign, UINT32 length,
1096
                                                UINT32 shift, WINPR_ATTR_UNUSED UINT32 bitPos,
1097
                                                UINT32 numBits)
1098
0
{
1099
0
  if (!numBits)
1100
0
    return 1;
1101
1102
0
  wBitStream* raw = state->raw;
1103
0
  int32_t input = 0;
1104
1105
0
  if (!state->nonLL)
1106
0
  {
1107
0
    for (UINT32 index = 0; index < length; index++)
1108
0
    {
1109
0
      raw->mask = ((1 << numBits) - 1);
1110
0
      input = (INT16)((raw->accumulator >> (32 - numBits)) & raw->mask);
1111
0
      BitStream_Shift(raw, numBits);
1112
1113
0
      const int32_t shifted = input << shift;
1114
0
      const int32_t val = buffer[index] + shifted;
1115
0
      const int16_t ival = WINPR_ASSERTING_INT_CAST(int16_t, val);
1116
0
      buffer[index] = ival;
1117
0
    }
1118
1119
0
    return 1;
1120
0
  }
1121
1122
0
  for (UINT32 index = 0; index < length; index++)
1123
0
  {
1124
0
    if (sign[index] > 0)
1125
0
    {
1126
      /* sign > 0, read from raw */
1127
0
      raw->mask = ((1 << numBits) - 1);
1128
0
      input = (INT16)((raw->accumulator >> (32 - numBits)) & raw->mask);
1129
0
      BitStream_Shift(raw, numBits);
1130
0
    }
1131
0
    else if (sign[index] < 0)
1132
0
    {
1133
      /* sign < 0, read from raw */
1134
0
      raw->mask = ((1 << numBits) - 1);
1135
0
      input = (INT16)((raw->accumulator >> (32 - numBits)) & raw->mask);
1136
0
      BitStream_Shift(raw, numBits);
1137
0
      input *= -1;
1138
0
    }
1139
0
    else
1140
0
    {
1141
      /* sign == 0, read from srl */
1142
0
      input = progressive_rfx_srl_read(state, numBits);
1143
0
      sign[index] = WINPR_ASSERTING_INT_CAST(int16_t, input);
1144
0
    }
1145
1146
0
    const int32_t val = input << shift;
1147
0
    const int32_t ival = buffer[index] + val;
1148
0
    buffer[index] = WINPR_ASSERTING_INT_CAST(INT16, ival);
1149
0
  }
1150
1151
0
  return 1;
1152
0
}
1153
1154
static INLINE int progressive_rfx_upgrade_component(
1155
    PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
1156
    const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT shift,
1157
    const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT bitPos,
1158
    const RFX_COMPONENT_CODEC_QUANT* WINPR_RESTRICT numBits, INT16* WINPR_RESTRICT buffer,
1159
    INT16* WINPR_RESTRICT current, INT16* WINPR_RESTRICT sign, const BYTE* WINPR_RESTRICT srlData,
1160
    UINT32 srlLen, const BYTE* WINPR_RESTRICT rawData, UINT32 rawLen, BOOL coeffDiff,
1161
    WINPR_ATTR_UNUSED BOOL subbandDiff, BOOL extrapolate)
1162
0
{
1163
0
  int rc = 0;
1164
0
  UINT32 aRawLen = 0;
1165
0
  UINT32 aSrlLen = 0;
1166
0
  wBitStream s_srl = { 0 };
1167
0
  wBitStream s_raw = { 0 };
1168
0
  RFX_PROGRESSIVE_UPGRADE_STATE state = { 0 };
1169
1170
0
  state.kp = 8;
1171
0
  state.mode = 0;
1172
0
  state.srl = &s_srl;
1173
0
  state.raw = &s_raw;
1174
0
  BitStream_Attach(state.srl, srlData, srlLen);
1175
0
  BitStream_Fetch(state.srl);
1176
0
  BitStream_Attach(state.raw, rawData, rawLen);
1177
0
  BitStream_Fetch(state.raw);
1178
1179
0
  state.nonLL = TRUE;
1180
0
  rc = progressive_rfx_upgrade_block(&state, &current[0], &sign[0], 1023, shift->HL1, bitPos->HL1,
1181
0
                                     numBits->HL1); /* HL1 */
1182
0
  if (rc < 0)
1183
0
    return rc;
1184
0
  rc = progressive_rfx_upgrade_block(&state, &current[1023], &sign[1023], 1023, shift->LH1,
1185
0
                                     bitPos->LH1, numBits->LH1); /* LH1 */
1186
0
  if (rc < 0)
1187
0
    return rc;
1188
0
  rc = progressive_rfx_upgrade_block(&state, &current[2046], &sign[2046], 961, shift->HH1,
1189
0
                                     bitPos->HH1, numBits->HH1); /* HH1 */
1190
0
  if (rc < 0)
1191
0
    return rc;
1192
0
  rc = progressive_rfx_upgrade_block(&state, &current[3007], &sign[3007], 272, shift->HL2,
1193
0
                                     bitPos->HL2, numBits->HL2); /* HL2 */
1194
0
  if (rc < 0)
1195
0
    return rc;
1196
0
  rc = progressive_rfx_upgrade_block(&state, &current[3279], &sign[3279], 272, shift->LH2,
1197
0
                                     bitPos->LH2, numBits->LH2); /* LH2 */
1198
0
  if (rc < 0)
1199
0
    return rc;
1200
0
  rc = progressive_rfx_upgrade_block(&state, &current[3551], &sign[3551], 256, shift->HH2,
1201
0
                                     bitPos->HH2, numBits->HH2); /* HH2 */
1202
0
  if (rc < 0)
1203
0
    return rc;
1204
0
  rc = progressive_rfx_upgrade_block(&state, &current[3807], &sign[3807], 72, shift->HL3,
1205
0
                                     bitPos->HL3, numBits->HL3); /* HL3 */
1206
0
  if (rc < 0)
1207
0
    return rc;
1208
0
  rc = progressive_rfx_upgrade_block(&state, &current[3879], &sign[3879], 72, shift->LH3,
1209
0
                                     bitPos->LH3, numBits->LH3); /* LH3 */
1210
0
  if (rc < 0)
1211
0
    return rc;
1212
0
  rc = progressive_rfx_upgrade_block(&state, &current[3951], &sign[3951], 64, shift->HH3,
1213
0
                                     bitPos->HH3, numBits->HH3); /* HH3 */
1214
0
  if (rc < 0)
1215
0
    return rc;
1216
1217
0
  state.nonLL = FALSE;
1218
0
  rc = progressive_rfx_upgrade_block(&state, &current[4015], &sign[4015], 81, shift->LL3,
1219
0
                                     bitPos->LL3, numBits->LL3); /* LL3 */
1220
0
  if (rc < 0)
1221
0
    return rc;
1222
0
  rc = progressive_rfx_upgrade_state_finish(&state);
1223
0
  if (rc < 0)
1224
0
    return rc;
1225
0
  aRawLen = (state.raw->position + 7) / 8;
1226
0
  aSrlLen = (state.srl->position + 7) / 8;
1227
1228
0
  if ((aRawLen != rawLen) || (aSrlLen != srlLen))
1229
0
  {
1230
0
    int pRawLen = 0;
1231
0
    int pSrlLen = 0;
1232
1233
0
    if (rawLen)
1234
0
      pRawLen = (int)((((float)aRawLen) / ((float)rawLen)) * 100.0f);
1235
1236
0
    if (srlLen)
1237
0
      pSrlLen = (int)((((float)aSrlLen) / ((float)srlLen)) * 100.0f);
1238
1239
0
    WLog_Print(progressive->log, WLOG_WARN,
1240
0
               "RAW: %" PRIu32 "/%" PRIu32 " %d%% (%" PRIu32 "/%" PRIu32 ":%" PRIu32
1241
0
               ")\tSRL: %" PRIu32 "/%" PRIu32 " %d%% (%" PRIu32 "/%" PRIu32 ":%" PRIu32 ")",
1242
0
               aRawLen, rawLen, pRawLen, state.raw->position, rawLen * 8,
1243
0
               (rawLen * 8) - state.raw->position, aSrlLen, srlLen, pSrlLen,
1244
0
               state.srl->position, srlLen * 8, (srlLen * 8) - state.srl->position);
1245
0
    return -1;
1246
0
  }
1247
1248
0
  return progressive_rfx_dwt_2d_decode(progressive, buffer, current, coeffDiff, extrapolate,
1249
0
                                       TRUE);
1250
0
}
1251
1252
static INLINE int
1253
progressive_decompress_tile_upgrade(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
1254
                                    RFX_PROGRESSIVE_TILE* WINPR_RESTRICT tile,
1255
                                    PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
1256
                                    const PROGRESSIVE_BLOCK_CONTEXT* WINPR_RESTRICT context)
1257
0
{
1258
0
  int status = 0;
1259
0
  BOOL coeffDiff = 0;
1260
0
  BOOL sub = 0;
1261
0
  BOOL extrapolate = 0;
1262
0
  BYTE* pBuffer = NULL;
1263
0
  INT16* pSign[3] = { 0 };
1264
0
  INT16* pSrcDst[3] = { 0 };
1265
0
  INT16* pCurrent[3] = { 0 };
1266
0
  RFX_COMPONENT_CODEC_QUANT shiftY = { 0 };
1267
0
  RFX_COMPONENT_CODEC_QUANT shiftCb = { 0 };
1268
0
  RFX_COMPONENT_CODEC_QUANT shiftCr = { 0 };
1269
0
  RFX_COMPONENT_CODEC_QUANT yBitPos = { 0 };
1270
0
  RFX_COMPONENT_CODEC_QUANT cbBitPos = { 0 };
1271
0
  RFX_COMPONENT_CODEC_QUANT crBitPos = { 0 };
1272
0
  RFX_COMPONENT_CODEC_QUANT yNumBits = { 0 };
1273
0
  RFX_COMPONENT_CODEC_QUANT cbNumBits = { 0 };
1274
0
  RFX_COMPONENT_CODEC_QUANT crNumBits = { 0 };
1275
0
  RFX_COMPONENT_CODEC_QUANT* quantY = NULL;
1276
0
  RFX_COMPONENT_CODEC_QUANT* quantCb = NULL;
1277
0
  RFX_COMPONENT_CODEC_QUANT* quantCr = NULL;
1278
0
  RFX_COMPONENT_CODEC_QUANT* quantProgY = NULL;
1279
0
  RFX_COMPONENT_CODEC_QUANT* quantProgCb = NULL;
1280
0
  RFX_COMPONENT_CODEC_QUANT* quantProgCr = NULL;
1281
0
  RFX_PROGRESSIVE_CODEC_QUANT* quantProg = NULL;
1282
0
  static const prim_size_t roi_64x64 = { 64, 64 };
1283
0
  const primitives_t* prims = primitives_get();
1284
1285
0
  coeffDiff = tile->flags & RFX_TILE_DIFFERENCE;
1286
0
  sub = context->flags & RFX_SUBBAND_DIFFING;
1287
0
  extrapolate = region->flags & RFX_DWT_REDUCE_EXTRAPOLATE;
1288
1289
0
  tile->pass++;
1290
1291
#if defined(WITH_DEBUG_CODECS)
1292
  WLog_Print(progressive->log, WLOG_DEBUG,
1293
             "ProgressiveTileUpgrade: pass: %" PRIu16 " quantIdx Y: %" PRIu8 " Cb: %" PRIu8
1294
             " Cr: %" PRIu8 " xIdx: %" PRIu16 " yIdx: %" PRIu16 " quality: %" PRIu8
1295
             " ySrlLen: %" PRIu16 " yRawLen: %" PRIu16 " cbSrlLen: %" PRIu16 " cbRawLen: %" PRIu16
1296
             " crSrlLen: %" PRIu16 " crRawLen: %" PRIu16 "",
1297
             tile->pass, tile->quantIdxY, tile->quantIdxCb, tile->quantIdxCr, tile->xIdx,
1298
             tile->yIdx, tile->quality, tile->ySrlLen, tile->yRawLen, tile->cbSrlLen,
1299
             tile->cbRawLen, tile->crSrlLen, tile->crRawLen);
1300
#endif
1301
1302
0
  if (tile->quantIdxY >= region->numQuant)
1303
0
  {
1304
0
    WLog_ERR(TAG, "quantIdxY %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxY, region->numQuant);
1305
0
    return -1;
1306
0
  }
1307
1308
0
  quantY = &(region->quantVals[tile->quantIdxY]);
1309
1310
0
  if (tile->quantIdxCb >= region->numQuant)
1311
0
  {
1312
0
    WLog_ERR(TAG, "quantIdxCb %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCb,
1313
0
             region->numQuant);
1314
0
    return -1;
1315
0
  }
1316
1317
0
  quantCb = &(region->quantVals[tile->quantIdxCb]);
1318
1319
0
  if (tile->quantIdxCr >= region->numQuant)
1320
0
  {
1321
0
    WLog_ERR(TAG, "quantIdxCr %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCr,
1322
0
             region->numQuant);
1323
0
    return -1;
1324
0
  }
1325
1326
0
  quantCr = &(region->quantVals[tile->quantIdxCr]);
1327
1328
0
  if (tile->quality == 0xFF)
1329
0
  {
1330
0
    quantProg = &(progressive->quantProgValFull);
1331
0
  }
1332
0
  else
1333
0
  {
1334
0
    if (tile->quality >= region->numProgQuant)
1335
0
    {
1336
0
      WLog_ERR(TAG, "quality %" PRIu8 " > numProgQuant %" PRIu8, tile->quality,
1337
0
               region->numProgQuant);
1338
0
      return -1;
1339
0
    }
1340
1341
0
    quantProg = &(region->quantProgVals[tile->quality]);
1342
0
  }
1343
1344
0
  quantProgY = &(quantProg->yQuantValues);
1345
0
  quantProgCb = &(quantProg->cbQuantValues);
1346
0
  quantProgCr = &(quantProg->crQuantValues);
1347
1348
0
  if (!progressive_rfx_quant_cmp_equal(quantY, &(tile->yQuant)))
1349
0
    WLog_Print(progressive->log, WLOG_WARN, "non-progressive quantY has changed!");
1350
1351
0
  if (!progressive_rfx_quant_cmp_equal(quantCb, &(tile->cbQuant)))
1352
0
    WLog_Print(progressive->log, WLOG_WARN, "non-progressive quantCb has changed!");
1353
1354
0
  if (!progressive_rfx_quant_cmp_equal(quantCr, &(tile->crQuant)))
1355
0
    WLog_Print(progressive->log, WLOG_WARN, "non-progressive quantCr has changed!");
1356
1357
0
  if (!(context->flags & RFX_SUBBAND_DIFFING))
1358
0
    WLog_WARN(TAG, "PROGRESSIVE_BLOCK_CONTEXT::flags & RFX_SUBBAND_DIFFING not set");
1359
1360
0
  progressive_rfx_quant_add(quantY, quantProgY, &yBitPos);
1361
0
  progressive_rfx_quant_add(quantCb, quantProgCb, &cbBitPos);
1362
0
  progressive_rfx_quant_add(quantCr, quantProgCr, &crBitPos);
1363
0
  progressive_rfx_quant_sub(&(tile->yBitPos), &yBitPos, &yNumBits);
1364
0
  progressive_rfx_quant_sub(&(tile->cbBitPos), &cbBitPos, &cbNumBits);
1365
0
  progressive_rfx_quant_sub(&(tile->crBitPos), &crBitPos, &crNumBits);
1366
0
  progressive_rfx_quant_add(quantY, quantProgY, &shiftY);
1367
0
  progressive_rfx_quant_lsub(&shiftY, 1); /* -6 + 5 = -1 */
1368
0
  progressive_rfx_quant_add(quantCb, quantProgCb, &shiftCb);
1369
0
  progressive_rfx_quant_lsub(&shiftCb, 1); /* -6 + 5 = -1 */
1370
0
  progressive_rfx_quant_add(quantCr, quantProgCr, &shiftCr);
1371
0
  progressive_rfx_quant_lsub(&shiftCr, 1); /* -6 + 5 = -1 */
1372
1373
0
  tile->yBitPos = yBitPos;
1374
0
  tile->cbBitPos = cbBitPos;
1375
0
  tile->crBitPos = crBitPos;
1376
0
  tile->yQuant = *quantY;
1377
0
  tile->cbQuant = *quantCb;
1378
0
  tile->crQuant = *quantCr;
1379
0
  tile->yProgQuant = *quantProgY;
1380
0
  tile->cbProgQuant = *quantProgCb;
1381
0
  tile->crProgQuant = *quantProgCr;
1382
1383
0
  pSign[0] = (INT16*)((&tile->sign[((8192 + 32) * 0) + 16])); /* Y/R buffer */
1384
0
  pSign[1] = (INT16*)((&tile->sign[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
1385
0
  pSign[2] = (INT16*)((&tile->sign[((8192 + 32) * 2) + 16])); /* Cr/B buffer */
1386
1387
0
  pCurrent[0] = (INT16*)((&tile->current[((8192 + 32) * 0) + 16])); /* Y/R buffer */
1388
0
  pCurrent[1] = (INT16*)((&tile->current[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
1389
0
  pCurrent[2] = (INT16*)((&tile->current[((8192 + 32) * 2) + 16])); /* Cr/B buffer */
1390
1391
0
  pBuffer = (BYTE*)BufferPool_Take(progressive->bufferPool, -1);
1392
0
  pSrcDst[0] = (INT16*)((&pBuffer[((8192 + 32) * 0) + 16])); /* Y/R buffer */
1393
0
  pSrcDst[1] = (INT16*)((&pBuffer[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
1394
0
  pSrcDst[2] = (INT16*)((&pBuffer[((8192 + 32) * 2) + 16])); /* Cr/B buffer */
1395
1396
0
  status = progressive_rfx_upgrade_component(progressive, &shiftY, quantProgY, &yNumBits,
1397
0
                                             pSrcDst[0], pCurrent[0], pSign[0], tile->ySrlData,
1398
0
                                             tile->ySrlLen, tile->yRawData, tile->yRawLen,
1399
0
                                             coeffDiff, sub, extrapolate); /* Y */
1400
1401
0
  if (status < 0)
1402
0
    goto fail;
1403
1404
0
  status = progressive_rfx_upgrade_component(progressive, &shiftCb, quantProgCb, &cbNumBits,
1405
0
                                             pSrcDst[1], pCurrent[1], pSign[1], tile->cbSrlData,
1406
0
                                             tile->cbSrlLen, tile->cbRawData, tile->cbRawLen,
1407
0
                                             coeffDiff, sub, extrapolate); /* Cb */
1408
1409
0
  if (status < 0)
1410
0
    goto fail;
1411
1412
0
  status = progressive_rfx_upgrade_component(progressive, &shiftCr, quantProgCr, &crNumBits,
1413
0
                                             pSrcDst[2], pCurrent[2], pSign[2], tile->crSrlData,
1414
0
                                             tile->crSrlLen, tile->crRawData, tile->crRawLen,
1415
0
                                             coeffDiff, sub, extrapolate); /* Cr */
1416
1417
0
  if (status < 0)
1418
0
    goto fail;
1419
1420
0
  const INT16** ptr = WINPR_REINTERPRET_CAST(pSrcDst, INT16**, const INT16**);
1421
0
  status = prims->yCbCrToRGB_16s8u_P3AC4R(ptr, 64 * 2, tile->data, tile->stride,
1422
0
                                          progressive->format, &roi_64x64);
1423
0
fail:
1424
0
  BufferPool_Return(progressive->bufferPool, pBuffer);
1425
0
  return status;
1426
0
}
1427
1428
static INLINE BOOL progressive_tile_read_upgrade(
1429
    PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive, wStream* WINPR_RESTRICT s, UINT16 blockType,
1430
    UINT32 blockLen, PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
1431
    PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
1432
    WINPR_ATTR_UNUSED const PROGRESSIVE_BLOCK_CONTEXT* WINPR_RESTRICT context)
1433
80
{
1434
80
  RFX_PROGRESSIVE_TILE tile = { 0 };
1435
80
  const size_t expect = 20;
1436
1437
80
  if (!Stream_CheckAndLogRequiredLength(TAG, s, expect))
1438
1
    return FALSE;
1439
1440
79
  tile.blockType = blockType;
1441
79
  tile.blockLen = blockLen;
1442
79
  tile.flags = 0;
1443
1444
79
  Stream_Read_UINT8(s, tile.quantIdxY);
1445
79
  Stream_Read_UINT8(s, tile.quantIdxCb);
1446
79
  Stream_Read_UINT8(s, tile.quantIdxCr);
1447
79
  Stream_Read_UINT16(s, tile.xIdx);
1448
79
  Stream_Read_UINT16(s, tile.yIdx);
1449
79
  Stream_Read_UINT8(s, tile.quality);
1450
79
  Stream_Read_UINT16(s, tile.ySrlLen);
1451
79
  Stream_Read_UINT16(s, tile.yRawLen);
1452
79
  Stream_Read_UINT16(s, tile.cbSrlLen);
1453
79
  Stream_Read_UINT16(s, tile.cbRawLen);
1454
79
  Stream_Read_UINT16(s, tile.crSrlLen);
1455
79
  Stream_Read_UINT16(s, tile.crRawLen);
1456
1457
79
  tile.ySrlData = Stream_Pointer(s);
1458
79
  if (!Stream_SafeSeek(s, tile.ySrlLen))
1459
6
  {
1460
6
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.ySrlLen);
1461
6
    return FALSE;
1462
6
  }
1463
1464
73
  tile.yRawData = Stream_Pointer(s);
1465
73
  if (!Stream_SafeSeek(s, tile.yRawLen))
1466
2
  {
1467
2
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.yRawLen);
1468
2
    return FALSE;
1469
2
  }
1470
1471
71
  tile.cbSrlData = Stream_Pointer(s);
1472
71
  if (!Stream_SafeSeek(s, tile.cbSrlLen))
1473
3
  {
1474
3
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
1475
3
               tile.cbSrlLen);
1476
3
    return FALSE;
1477
3
  }
1478
1479
68
  tile.cbRawData = Stream_Pointer(s);
1480
68
  if (!Stream_SafeSeek(s, tile.cbRawLen))
1481
4
  {
1482
4
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
1483
4
               tile.cbRawLen);
1484
4
    return FALSE;
1485
4
  }
1486
1487
64
  tile.crSrlData = Stream_Pointer(s);
1488
64
  if (!Stream_SafeSeek(s, tile.crSrlLen))
1489
1
  {
1490
1
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
1491
1
               tile.crSrlLen);
1492
1
    return FALSE;
1493
1
  }
1494
1495
63
  tile.crRawData = Stream_Pointer(s);
1496
63
  if (!Stream_SafeSeek(s, tile.crRawLen))
1497
6
  {
1498
6
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
1499
6
               tile.crRawLen);
1500
6
    return FALSE;
1501
6
  }
1502
1503
57
  return progressive_surface_tile_replace(surface, region, &tile, TRUE);
1504
63
}
1505
1506
static INLINE BOOL progressive_tile_read(
1507
    PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive, BOOL simple, wStream* WINPR_RESTRICT s,
1508
    UINT16 blockType, UINT32 blockLen, PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
1509
    PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
1510
    WINPR_ATTR_UNUSED const PROGRESSIVE_BLOCK_CONTEXT* WINPR_RESTRICT context)
1511
68
{
1512
68
  RFX_PROGRESSIVE_TILE tile = { 0 };
1513
68
  size_t expect = simple ? 16 : 17;
1514
1515
68
  if (!Stream_CheckAndLogRequiredLength(TAG, s, expect))
1516
2
    return FALSE;
1517
1518
66
  tile.blockType = blockType;
1519
66
  tile.blockLen = blockLen;
1520
1521
66
  Stream_Read_UINT8(s, tile.quantIdxY);
1522
66
  Stream_Read_UINT8(s, tile.quantIdxCb);
1523
66
  Stream_Read_UINT8(s, tile.quantIdxCr);
1524
66
  Stream_Read_UINT16(s, tile.xIdx);
1525
66
  Stream_Read_UINT16(s, tile.yIdx);
1526
66
  Stream_Read_UINT8(s, tile.flags);
1527
1528
66
  if (!simple)
1529
39
    Stream_Read_UINT8(s, tile.quality);
1530
27
  else
1531
27
    tile.quality = 0xFF;
1532
66
  Stream_Read_UINT16(s, tile.yLen);
1533
66
  Stream_Read_UINT16(s, tile.cbLen);
1534
66
  Stream_Read_UINT16(s, tile.crLen);
1535
66
  Stream_Read_UINT16(s, tile.tailLen);
1536
1537
66
  tile.yData = Stream_Pointer(s);
1538
66
  if (!Stream_SafeSeek(s, tile.yLen))
1539
4
  {
1540
4
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.yLen);
1541
4
    return FALSE;
1542
4
  }
1543
1544
62
  tile.cbData = Stream_Pointer(s);
1545
62
  if (!Stream_SafeSeek(s, tile.cbLen))
1546
5
  {
1547
5
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.cbLen);
1548
5
    return FALSE;
1549
5
  }
1550
1551
57
  tile.crData = Stream_Pointer(s);
1552
57
  if (!Stream_SafeSeek(s, tile.crLen))
1553
3
  {
1554
3
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.crLen);
1555
3
    return FALSE;
1556
3
  }
1557
1558
54
  tile.tailData = Stream_Pointer(s);
1559
54
  if (!Stream_SafeSeek(s, tile.tailLen))
1560
3
  {
1561
3
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.tailLen);
1562
3
    return FALSE;
1563
3
  }
1564
1565
51
  return progressive_surface_tile_replace(surface, region, &tile, FALSE);
1566
54
}
1567
1568
static void CALLBACK progressive_process_tiles_tile_work_callback(PTP_CALLBACK_INSTANCE instance,
1569
                                                                  void* context, PTP_WORK work)
1570
0
{
1571
0
  PROGRESSIVE_TILE_PROCESS_WORK_PARAM* param = (PROGRESSIVE_TILE_PROCESS_WORK_PARAM*)context;
1572
1573
0
  WINPR_UNUSED(instance);
1574
0
  WINPR_UNUSED(work);
1575
1576
0
  switch (param->tile->blockType)
1577
0
  {
1578
0
    case PROGRESSIVE_WBT_TILE_SIMPLE:
1579
0
    case PROGRESSIVE_WBT_TILE_FIRST:
1580
0
      progressive_decompress_tile_first(param->progressive, param->tile, param->region,
1581
0
                                        param->context);
1582
0
      break;
1583
1584
0
    case PROGRESSIVE_WBT_TILE_UPGRADE:
1585
0
      progressive_decompress_tile_upgrade(param->progressive, param->tile, param->region,
1586
0
                                          param->context);
1587
0
      break;
1588
0
    default:
1589
0
      WLog_Print(param->progressive->log, WLOG_ERROR, "Invalid block type %04" PRIx16 " (%s)",
1590
0
                 param->tile->blockType,
1591
0
                 rfx_get_progressive_block_type_string(param->tile->blockType));
1592
0
      break;
1593
0
  }
1594
0
}
1595
1596
static INLINE SSIZE_T progressive_process_tiles(
1597
    PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive, wStream* WINPR_RESTRICT s,
1598
    PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
1599
    PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
1600
    const PROGRESSIVE_BLOCK_CONTEXT* WINPR_RESTRICT context)
1601
562
{
1602
562
  int status = 0;
1603
562
  size_t end = 0;
1604
562
  const size_t start = Stream_GetPosition(s);
1605
562
  UINT16 blockType = 0;
1606
562
  UINT32 blockLen = 0;
1607
562
  UINT32 count = 0;
1608
562
  UINT16 close_cnt = 0;
1609
1610
562
  WINPR_ASSERT(progressive);
1611
562
  WINPR_ASSERT(region);
1612
1613
562
  if (!Stream_CheckAndLogRequiredLength(TAG, s, region->tileDataSize))
1614
0
    return -1;
1615
1616
620
  while ((Stream_GetRemainingLength(s) >= 6) &&
1617
620
         (region->tileDataSize > (Stream_GetPosition(s) - start)))
1618
255
  {
1619
255
    const size_t pos = Stream_GetPosition(s);
1620
1621
255
    Stream_Read_UINT16(s, blockType);
1622
255
    Stream_Read_UINT32(s, blockLen);
1623
1624
#if defined(WITH_DEBUG_CODECS)
1625
    WLog_Print(progressive->log, WLOG_DEBUG, "%s",
1626
               rfx_get_progressive_block_type_string(blockType));
1627
#endif
1628
1629
255
    if (blockLen < 6)
1630
9
    {
1631
9
      WLog_Print(progressive->log, WLOG_ERROR, "Expected >= %" PRIu32 " remaining %" PRIuz, 6,
1632
9
                 blockLen);
1633
9
      return -1003;
1634
9
    }
1635
246
    if (!Stream_CheckAndLogRequiredLength(TAG, s, blockLen - 6))
1636
59
      return -1003;
1637
1638
187
    switch (blockType)
1639
187
    {
1640
28
      case PROGRESSIVE_WBT_TILE_SIMPLE:
1641
28
        if (!progressive_tile_read(progressive, TRUE, s, blockType, blockLen, surface,
1642
28
                                   region, context))
1643
22
          return -1022;
1644
6
        break;
1645
1646
40
      case PROGRESSIVE_WBT_TILE_FIRST:
1647
40
        if (!progressive_tile_read(progressive, FALSE, s, blockType, blockLen, surface,
1648
40
                                   region, context))
1649
15
          return -1027;
1650
25
        break;
1651
1652
80
      case PROGRESSIVE_WBT_TILE_UPGRADE:
1653
80
        if (!progressive_tile_read_upgrade(progressive, s, blockType, blockLen, surface,
1654
80
                                           region, context))
1655
27
          return -1032;
1656
53
        break;
1657
53
      default:
1658
39
        WLog_ERR(TAG, "Invalid block type %04" PRIx16 " (%s)", blockType,
1659
39
                 rfx_get_progressive_block_type_string(blockType));
1660
39
        return -1039;
1661
187
    }
1662
1663
84
    size_t rem = Stream_GetPosition(s);
1664
84
    if ((rem - pos) != blockLen)
1665
26
    {
1666
26
      WLog_Print(progressive->log, WLOG_ERROR,
1667
26
                 "Actual block read %" PRIuz " but expected %" PRIu32, rem - pos, blockLen);
1668
26
      return -1040;
1669
26
    }
1670
58
    count++;
1671
58
  }
1672
1673
365
  end = Stream_GetPosition(s);
1674
365
  if ((end - start) != region->tileDataSize)
1675
14
  {
1676
14
    WLog_Print(progressive->log, WLOG_ERROR,
1677
14
               "Actual total blocks read %" PRIuz " but expected %" PRIu32, end - start,
1678
14
               region->tileDataSize);
1679
14
    return -1041;
1680
14
  }
1681
1682
351
  if (count != region->numTiles)
1683
25
  {
1684
25
    WLog_Print(progressive->log, WLOG_WARN,
1685
25
               "numTiles inconsistency: actual: %" PRIu32 ", expected: %" PRIu16 "\n", count,
1686
25
               region->numTiles);
1687
25
    return -1044;
1688
25
  }
1689
1690
326
  for (UINT32 idx = 0; idx < region->numTiles; idx++)
1691
0
  {
1692
0
    RFX_PROGRESSIVE_TILE* tile = region->tiles[idx];
1693
0
    PROGRESSIVE_TILE_PROCESS_WORK_PARAM* param = &progressive->params[idx];
1694
0
    param->progressive = progressive;
1695
0
    param->region = region;
1696
0
    param->context = context;
1697
0
    param->tile = tile;
1698
1699
0
    if (progressive->rfx_context->priv->UseThreads)
1700
0
    {
1701
0
      progressive->work_objects[idx] =
1702
0
          CreateThreadpoolWork(progressive_process_tiles_tile_work_callback, (void*)param,
1703
0
                               &progressive->rfx_context->priv->ThreadPoolEnv);
1704
0
      if (!progressive->work_objects[idx])
1705
0
      {
1706
0
        WLog_Print(progressive->log, WLOG_ERROR,
1707
0
                   "Failed to create ThreadpoolWork for tile %" PRIu32, idx);
1708
0
        status = -1;
1709
0
        break;
1710
0
      }
1711
1712
0
      SubmitThreadpoolWork(progressive->work_objects[idx]);
1713
1714
0
      close_cnt = WINPR_ASSERTING_INT_CAST(UINT16, idx + 1);
1715
0
    }
1716
0
    else
1717
0
    {
1718
0
      progressive_process_tiles_tile_work_callback(0, param, 0);
1719
0
    }
1720
1721
0
    if (status < 0)
1722
0
    {
1723
0
      WLog_Print(progressive->log, WLOG_ERROR, "Failed to decompress %s at %" PRIu16,
1724
0
                 rfx_get_progressive_block_type_string(tile->blockType), idx);
1725
0
      goto fail;
1726
0
    }
1727
0
  }
1728
1729
326
  if (progressive->rfx_context->priv->UseThreads)
1730
326
  {
1731
326
    for (UINT32 idx = 0; idx < close_cnt; idx++)
1732
0
    {
1733
0
      WaitForThreadpoolWorkCallbacks(progressive->work_objects[idx], FALSE);
1734
0
      CloseThreadpoolWork(progressive->work_objects[idx]);
1735
0
    }
1736
326
  }
1737
1738
326
fail:
1739
1740
326
  if (status < 0)
1741
0
    return -1;
1742
1743
326
  return (SSIZE_T)(end - start);
1744
326
}
1745
1746
static INLINE SSIZE_T progressive_wb_sync(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
1747
                                          wStream* WINPR_RESTRICT s, UINT16 blockType,
1748
                                          UINT32 blockLen)
1749
775
{
1750
775
  const UINT32 magic = 0xCACCACCA;
1751
775
  const UINT16 version = 0x0100;
1752
775
  PROGRESSIVE_BLOCK_SYNC sync = { 0 };
1753
1754
775
  sync.blockType = blockType;
1755
775
  sync.blockLen = blockLen;
1756
1757
775
  if (sync.blockLen != 12)
1758
26
  {
1759
26
    WLog_Print(progressive->log, WLOG_ERROR,
1760
26
               "PROGRESSIVE_BLOCK_SYNC::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
1761
26
               sync.blockLen, 12);
1762
26
    return -1005;
1763
26
  }
1764
1765
749
  if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
1766
0
    return -1004;
1767
1768
#if defined(WITH_DEBUG_CODECS)
1769
  WLog_Print(progressive->log, WLOG_DEBUG, "ProgressiveSync");
1770
#endif
1771
1772
749
  Stream_Read_UINT32(s, sync.magic);
1773
749
  Stream_Read_UINT16(s, sync.version);
1774
1775
749
  if (sync.magic != magic)
1776
50
  {
1777
50
    WLog_Print(progressive->log, WLOG_ERROR,
1778
50
               "PROGRESSIVE_BLOCK_SYNC::magic = 0x%08" PRIx32 " != 0x%08" PRIx32, sync.magic,
1779
50
               magic);
1780
50
    return -1005;
1781
50
  }
1782
1783
699
  if (sync.version != 0x0100)
1784
19
  {
1785
19
    WLog_Print(progressive->log, WLOG_ERROR,
1786
19
               "PROGRESSIVE_BLOCK_SYNC::version = 0x%04" PRIx16 " != 0x%04" PRIu16,
1787
19
               sync.version, version);
1788
19
    return -1006;
1789
19
  }
1790
1791
680
  if ((progressive->state & FLAG_WBT_SYNC) != 0)
1792
655
    WLog_WARN(TAG, "Duplicate PROGRESSIVE_BLOCK_SYNC, ignoring");
1793
1794
680
  progressive->state |= FLAG_WBT_SYNC;
1795
680
  return 0;
1796
699
}
1797
1798
static INLINE SSIZE_T progressive_wb_frame_begin(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
1799
                                                 wStream* WINPR_RESTRICT s, UINT16 blockType,
1800
                                                 UINT32 blockLen)
1801
627
{
1802
627
  PROGRESSIVE_BLOCK_FRAME_BEGIN frameBegin = { 0 };
1803
1804
627
  frameBegin.blockType = blockType;
1805
627
  frameBegin.blockLen = blockLen;
1806
1807
627
  if (frameBegin.blockLen != 12)
1808
62
  {
1809
62
    WLog_Print(progressive->log, WLOG_ERROR,
1810
62
               " RFX_PROGRESSIVE_FRAME_BEGIN::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
1811
62
               frameBegin.blockLen, 12);
1812
62
    return -1005;
1813
62
  }
1814
1815
565
  if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
1816
0
    return -1007;
1817
1818
565
  Stream_Read_UINT32(s, frameBegin.frameIndex);
1819
565
  Stream_Read_UINT16(s, frameBegin.regionCount);
1820
1821
#if defined(WITH_DEBUG_CODECS)
1822
  WLog_Print(progressive->log, WLOG_DEBUG,
1823
             "ProgressiveFrameBegin: frameIndex: %" PRIu32 " regionCount: %" PRIu16 "",
1824
             frameBegin.frameIndex, frameBegin.regionCount);
1825
#endif
1826
1827
  /**
1828
   * If the number of elements specified by the regionCount field is
1829
   * larger than the actual number of elements in the regions field,
1830
   * the decoder SHOULD ignore this inconsistency.
1831
   */
1832
1833
565
  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) != 0)
1834
1
  {
1835
1
    WLog_ERR(TAG, "Duplicate RFX_PROGRESSIVE_FRAME_BEGIN in stream, this is not allowed!");
1836
1
    return -1008;
1837
1
  }
1838
1839
564
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
1840
1
  {
1841
1
    WLog_ERR(TAG, "RFX_PROGRESSIVE_FRAME_BEGIN after RFX_PROGRESSIVE_FRAME_END in stream, this "
1842
1
                  "is not allowed!");
1843
1
    return -1008;
1844
1
  }
1845
1846
563
  progressive->state |= FLAG_WBT_FRAME_BEGIN;
1847
563
  return 0;
1848
564
}
1849
1850
static INLINE SSIZE_T progressive_wb_frame_end(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
1851
                                               wStream* WINPR_RESTRICT s, UINT16 blockType,
1852
                                               UINT32 blockLen)
1853
2.15k
{
1854
2.15k
  PROGRESSIVE_BLOCK_FRAME_END frameEnd = { 0 };
1855
1856
2.15k
  frameEnd.blockType = blockType;
1857
2.15k
  frameEnd.blockLen = blockLen;
1858
1859
2.15k
  if (frameEnd.blockLen != 6)
1860
63
  {
1861
63
    WLog_Print(progressive->log, WLOG_ERROR,
1862
63
               " RFX_PROGRESSIVE_FRAME_END::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
1863
63
               frameEnd.blockLen, 6);
1864
63
    return -1005;
1865
63
  }
1866
1867
2.08k
  if (Stream_GetRemainingLength(s) != 0)
1868
0
  {
1869
0
    WLog_Print(progressive->log, WLOG_ERROR,
1870
0
               "ProgressiveFrameEnd short %" PRIuz ", expected %" PRIuz,
1871
0
               Stream_GetRemainingLength(s), 0);
1872
0
    return -1008;
1873
0
  }
1874
1875
#if defined(WITH_DEBUG_CODECS)
1876
  WLog_Print(progressive->log, WLOG_DEBUG, "ProgressiveFrameEnd");
1877
#endif
1878
1879
2.08k
  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) == 0)
1880
1.87k
    WLog_WARN(TAG, "RFX_PROGRESSIVE_FRAME_END before RFX_PROGRESSIVE_FRAME_BEGIN, ignoring");
1881
2.08k
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
1882
2.01k
    WLog_WARN(TAG, "Duplicate RFX_PROGRESSIVE_FRAME_END, ignoring");
1883
1884
2.08k
  progressive->state |= FLAG_WBT_FRAME_END;
1885
2.08k
  return 0;
1886
2.08k
}
1887
1888
static INLINE SSIZE_T progressive_wb_context(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
1889
                                             wStream* WINPR_RESTRICT s, UINT16 blockType,
1890
                                             UINT32 blockLen)
1891
1.02k
{
1892
1.02k
  PROGRESSIVE_BLOCK_CONTEXT* context = &progressive->context;
1893
1.02k
  context->blockType = blockType;
1894
1.02k
  context->blockLen = blockLen;
1895
1896
1.02k
  if (context->blockLen != 10)
1897
76
  {
1898
76
    WLog_Print(progressive->log, WLOG_ERROR,
1899
76
               "RFX_PROGRESSIVE_CONTEXT::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
1900
76
               context->blockLen, 10);
1901
76
    return -1005;
1902
76
  }
1903
1904
952
  if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
1905
0
    return -1009;
1906
1907
952
  Stream_Read_UINT8(s, context->ctxId);
1908
952
  Stream_Read_UINT16(s, context->tileSize);
1909
952
  Stream_Read_UINT8(s, context->flags);
1910
1911
952
  if (context->ctxId != 0x00)
1912
441
    WLog_WARN(TAG, "RFX_PROGRESSIVE_CONTEXT::ctxId != 0x00: %" PRIu8, context->ctxId);
1913
1914
952
  if (context->tileSize != 64)
1915
27
  {
1916
27
    WLog_ERR(TAG, "RFX_PROGRESSIVE_CONTEXT::tileSize != 0x40: %" PRIu16, context->tileSize);
1917
27
    return -1010;
1918
27
  }
1919
1920
925
  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) != 0)
1921
425
    WLog_WARN(TAG, "RFX_PROGRESSIVE_CONTEXT received after RFX_PROGRESSIVE_FRAME_BEGIN");
1922
925
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
1923
115
    WLog_WARN(TAG, "RFX_PROGRESSIVE_CONTEXT received after RFX_PROGRESSIVE_FRAME_END");
1924
925
  if ((progressive->state & FLAG_WBT_CONTEXT) != 0)
1925
882
    WLog_WARN(TAG, "Duplicate RFX_PROGRESSIVE_CONTEXT received, ignoring.");
1926
1927
#if defined(WITH_DEBUG_CODECS)
1928
  WLog_Print(progressive->log, WLOG_DEBUG, "ProgressiveContext: flags: 0x%02" PRIX8 "",
1929
             context->flags);
1930
#endif
1931
1932
925
  progressive->state |= FLAG_WBT_CONTEXT;
1933
925
  return 0;
1934
952
}
1935
1936
static INLINE SSIZE_T progressive_wb_read_region_header(
1937
    PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive, wStream* WINPR_RESTRICT s, UINT16 blockType,
1938
    UINT32 blockLen, PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region)
1939
1.70k
{
1940
1.70k
  region->usedTiles = 0;
1941
1942
1.70k
  if (!Stream_CheckAndLogRequiredLength(TAG, s, 12))
1943
7
    return -1011;
1944
1945
1.70k
  region->blockType = blockType;
1946
1.70k
  region->blockLen = blockLen;
1947
1.70k
  Stream_Read_UINT8(s, region->tileSize);
1948
1.70k
  Stream_Read_UINT16(s, region->numRects);
1949
1.70k
  Stream_Read_UINT8(s, region->numQuant);
1950
1.70k
  Stream_Read_UINT8(s, region->numProgQuant);
1951
1.70k
  Stream_Read_UINT8(s, region->flags);
1952
1.70k
  Stream_Read_UINT16(s, region->numTiles);
1953
1.70k
  Stream_Read_UINT32(s, region->tileDataSize);
1954
1955
1.70k
  if (region->tileSize != 64)
1956
19
  {
1957
19
    WLog_Print(progressive->log, WLOG_ERROR,
1958
19
               "ProgressiveRegion tile size %" PRIu8 ", expected %" PRIuz, region->tileSize,
1959
19
               64);
1960
19
    return -1012;
1961
19
  }
1962
1963
1.68k
  if (region->numRects < 1)
1964
2
  {
1965
2
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion missing rect count %" PRIu16,
1966
2
               region->numRects);
1967
2
    return -1013;
1968
2
  }
1969
1970
1.68k
  if (region->numQuant > 7)
1971
8
  {
1972
8
    WLog_Print(progressive->log, WLOG_ERROR,
1973
8
               "ProgressiveRegion quant count too high %" PRIu8 ", expected < %" PRIuz,
1974
8
               region->numQuant, 7);
1975
8
    return -1014;
1976
8
  }
1977
1978
3.34k
  const SSIZE_T rc = WINPR_ASSERTING_INT_CAST(SSIZE_T, Stream_GetRemainingLength(s));
1979
0
  const SSIZE_T expect = region->numRects * 8ll + region->numQuant * 5ll +
1980
3.34k
                         region->numProgQuant * 16ll + region->tileDataSize;
1981
3.34k
  SSIZE_T len = rc;
1982
3.34k
  if (expect != rc)
1983
879
  {
1984
879
    if (len / 8LL < region->numRects)
1985
35
    {
1986
35
      WLog_Print(progressive->log, WLOG_ERROR,
1987
35
                 "ProgressiveRegion data short for region->rects");
1988
35
      return -1015;
1989
35
    }
1990
844
    len -= region->numRects * 8LL;
1991
1992
844
    if (len / 5LL < region->numQuant)
1993
1
    {
1994
1
      WLog_Print(progressive->log, WLOG_ERROR,
1995
1
                 "ProgressiveRegion data short for region->cQuant");
1996
1
      return -1018;
1997
1
    }
1998
843
    len -= region->numQuant * 5LL;
1999
2000
843
    if (len / 16LL < region->numProgQuant)
2001
7
    {
2002
7
      WLog_Print(progressive->log, WLOG_ERROR,
2003
7
                 "ProgressiveRegion data short for region->cProgQuant");
2004
7
      return -1021;
2005
7
    }
2006
836
    len -= region->numProgQuant * 16LL;
2007
2008
836
    if (len < region->tileDataSize * 1ll)
2009
41
    {
2010
41
      WLog_Print(progressive->log, WLOG_ERROR,
2011
41
                 "ProgressiveRegion data short for region->tiles");
2012
41
      return -1024;
2013
41
    }
2014
795
    len -= region->tileDataSize;
2015
2016
795
    if (len > 0)
2017
795
      WLog_Print(progressive->log, WLOG_WARN,
2018
795
                 "Unused bytes detected, %" PRIdz " bytes not processed", len);
2019
795
  }
2020
2021
1.58k
  return rc;
2022
3.34k
}
2023
2024
static INLINE SSIZE_T progressive_wb_skip_region(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
2025
                                                 wStream* WINPR_RESTRICT s, UINT16 blockType,
2026
                                                 UINT32 blockLen)
2027
1.09k
{
2028
1.09k
  PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region = &progressive->region;
2029
2030
1.09k
  const SSIZE_T rc =
2031
1.09k
      progressive_wb_read_region_header(progressive, s, blockType, blockLen, region);
2032
1.09k
  if (rc < 0)
2033
108
    return rc;
2034
2035
989
  if (!Stream_SafeSeek(s, WINPR_ASSERTING_INT_CAST(size_t, rc)))
2036
0
    return -1111;
2037
2038
989
  return rc;
2039
989
}
2040
2041
static INLINE SSIZE_T progressive_wb_region(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
2042
                                            wStream* WINPR_RESTRICT s, UINT16 blockType,
2043
                                            UINT32 blockLen,
2044
                                            PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
2045
                                            PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region)
2046
1.70k
{
2047
1.70k
  SSIZE_T rc = -1;
2048
1.70k
  UINT16 boxLeft = 0;
2049
1.70k
  UINT16 boxTop = 0;
2050
1.70k
  UINT16 boxRight = 0;
2051
1.70k
  UINT16 boxBottom = 0;
2052
1.70k
  UINT16 idxLeft = 0;
2053
1.70k
  UINT16 idxTop = 0;
2054
1.70k
  UINT16 idxRight = 0;
2055
1.70k
  UINT16 idxBottom = 0;
2056
1.70k
  const PROGRESSIVE_BLOCK_CONTEXT* context = &progressive->context;
2057
2058
1.70k
  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) == 0)
2059
799
  {
2060
799
    WLog_WARN(TAG, "RFX_PROGRESSIVE_REGION before RFX_PROGRESSIVE_FRAME_BEGIN, ignoring");
2061
799
    return progressive_wb_skip_region(progressive, s, blockType, blockLen);
2062
799
  }
2063
910
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
2064
298
  {
2065
298
    WLog_WARN(TAG, "RFX_PROGRESSIVE_REGION after RFX_PROGRESSIVE_FRAME_END, ignoring");
2066
298
    return progressive_wb_skip_region(progressive, s, blockType, blockLen);
2067
298
  }
2068
2069
612
  progressive->state |= FLAG_WBT_REGION;
2070
2071
612
  rc = progressive_wb_read_region_header(progressive, s, blockType, blockLen, region);
2072
612
  if (rc < 0)
2073
12
    return rc;
2074
2075
27.0k
  for (UINT16 index = 0; index < region->numRects; index++)
2076
26.4k
  {
2077
26.4k
    RFX_RECT* rect = &(region->rects[index]);
2078
26.4k
    Stream_Read_UINT16(s, rect->x);
2079
26.4k
    Stream_Read_UINT16(s, rect->y);
2080
26.4k
    Stream_Read_UINT16(s, rect->width);
2081
26.4k
    Stream_Read_UINT16(s, rect->height);
2082
26.4k
  }
2083
2084
648
  for (BYTE index = 0; index < region->numQuant; index++)
2085
86
  {
2086
86
    RFX_COMPONENT_CODEC_QUANT* quantVal = &(region->quantVals[index]);
2087
86
    progressive_component_codec_quant_read(s, quantVal);
2088
2089
86
    if (!progressive_rfx_quant_lcmp_greater_equal(quantVal, 6))
2090
38
    {
2091
38
      WLog_Print(progressive->log, WLOG_ERROR,
2092
38
                 "ProgressiveRegion region->cQuant[%" PRIu32 "] < 6", index);
2093
38
      return -1;
2094
38
    }
2095
2096
48
    if (!progressive_rfx_quant_lcmp_less_equal(quantVal, 15))
2097
0
    {
2098
0
      WLog_Print(progressive->log, WLOG_ERROR,
2099
0
                 "ProgressiveRegion region->cQuant[%" PRIu32 "] > 15", index);
2100
0
      return -1;
2101
0
    }
2102
48
  }
2103
2104
2.28k
  for (BYTE index = 0; index < region->numProgQuant; index++)
2105
1.71k
  {
2106
1.71k
    RFX_PROGRESSIVE_CODEC_QUANT* quantProgVal = &(region->quantProgVals[index]);
2107
2108
1.71k
    Stream_Read_UINT8(s, quantProgVal->quality);
2109
2110
1.71k
    progressive_component_codec_quant_read(s, &(quantProgVal->yQuantValues));
2111
1.71k
    progressive_component_codec_quant_read(s, &(quantProgVal->cbQuantValues));
2112
1.71k
    progressive_component_codec_quant_read(s, &(quantProgVal->crQuantValues));
2113
1.71k
  }
2114
2115
#if defined(WITH_DEBUG_CODECS)
2116
  WLog_Print(progressive->log, WLOG_DEBUG,
2117
             "ProgressiveRegion: numRects: %" PRIu16 " numTiles: %" PRIu16
2118
             " tileDataSize: %" PRIu32 " flags: 0x%02" PRIX8 " numQuant: %" PRIu8
2119
             " numProgQuant: %" PRIu8 "",
2120
             region->numRects, region->numTiles, region->tileDataSize, region->flags,
2121
             region->numQuant, region->numProgQuant);
2122
#endif
2123
2124
1.12k
  boxLeft = WINPR_ASSERTING_INT_CAST(UINT16, surface->gridWidth);
2125
1.12k
  boxTop = WINPR_ASSERTING_INT_CAST(UINT16, surface->gridHeight);
2126
0
  boxRight = 0;
2127
1.12k
  boxBottom = 0;
2128
2129
27.0k
  for (UINT16 index = 0; index < region->numRects; index++)
2130
26.4k
  {
2131
26.4k
    RFX_RECT* rect = &(region->rects[index]);
2132
26.4k
    idxLeft = rect->x / 64;
2133
26.4k
    idxTop = rect->y / 64;
2134
26.4k
    idxRight = (rect->x + rect->width + 63) / 64;
2135
26.4k
    idxBottom = (rect->y + rect->height + 63) / 64;
2136
2137
26.4k
    if (idxLeft < boxLeft)
2138
504
      boxLeft = idxLeft;
2139
2140
26.4k
    if (idxTop < boxTop)
2141
456
      boxTop = idxTop;
2142
2143
26.4k
    if (idxRight > boxRight)
2144
1.14k
      boxRight = idxRight;
2145
2146
26.4k
    if (idxBottom > boxBottom)
2147
1.09k
      boxBottom = idxBottom;
2148
2149
#if defined(WITH_DEBUG_CODECS)
2150
    WLog_Print(progressive->log, WLOG_DEBUG,
2151
               "rect[%" PRIu16 "]: x: %" PRIu16 " y: %" PRIu16 " w: %" PRIu16 " h: %" PRIu16 "",
2152
               index, rect->x, rect->y, rect->width, rect->height);
2153
#endif
2154
26.4k
  }
2155
2156
1.12k
  const SSIZE_T res = progressive_process_tiles(progressive, s, region, surface, context);
2157
1.12k
  if (res < 0)
2158
236
    return -1;
2159
326
  return rc;
2160
1.12k
}
2161
2162
static INLINE SSIZE_T progressive_parse_block(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
2163
                                              wStream* WINPR_RESTRICT s,
2164
                                              PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
2165
                                              PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region)
2166
10.7k
{
2167
10.7k
  UINT16 blockType = 0;
2168
10.7k
  UINT32 blockLen = 0;
2169
10.7k
  SSIZE_T rc = -1;
2170
10.7k
  wStream sub = { 0 };
2171
2172
10.7k
  WINPR_ASSERT(progressive);
2173
2174
10.7k
  if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
2175
544
    return -1;
2176
2177
10.2k
  Stream_Read_UINT16(s, blockType);
2178
10.2k
  Stream_Read_UINT32(s, blockLen);
2179
2180
10.2k
  if (blockLen < 6)
2181
1.31k
  {
2182
1.31k
    WLog_WARN(TAG, "Invalid blockLen %" PRIu32 ", expected >= 6", blockLen);
2183
1.31k
    return -1;
2184
1.31k
  }
2185
8.92k
  if (!Stream_CheckAndLogRequiredLength(TAG, s, blockLen - 6))
2186
2.30k
    return -1;
2187
6.62k
  Stream_StaticConstInit(&sub, Stream_Pointer(s), blockLen - 6);
2188
6.62k
  Stream_Seek(s, blockLen - 6);
2189
2190
6.62k
  switch (blockType)
2191
6.62k
  {
2192
775
    case PROGRESSIVE_WBT_SYNC:
2193
775
      rc = progressive_wb_sync(progressive, &sub, blockType, blockLen);
2194
775
      break;
2195
2196
627
    case PROGRESSIVE_WBT_FRAME_BEGIN:
2197
627
      rc = progressive_wb_frame_begin(progressive, &sub, blockType, blockLen);
2198
627
      break;
2199
2200
2.15k
    case PROGRESSIVE_WBT_FRAME_END:
2201
2.15k
      rc = progressive_wb_frame_end(progressive, &sub, blockType, blockLen);
2202
2.15k
      break;
2203
2204
1.02k
    case PROGRESSIVE_WBT_CONTEXT:
2205
1.02k
      rc = progressive_wb_context(progressive, &sub, blockType, blockLen);
2206
1.02k
      break;
2207
2208
1.70k
    case PROGRESSIVE_WBT_REGION:
2209
1.70k
      rc = progressive_wb_region(progressive, &sub, blockType, blockLen, surface, region);
2210
1.70k
      break;
2211
2212
330
    default:
2213
330
      WLog_Print(progressive->log, WLOG_ERROR, "Invalid block type %04" PRIx16, blockType);
2214
330
      return -1;
2215
6.62k
  }
2216
2217
6.29k
  if (rc < 0)
2218
719
    return -1;
2219
2220
5.57k
  if (Stream_GetRemainingLength(&sub) > 0)
2221
25
  {
2222
25
    WLog_Print(progressive->log, WLOG_ERROR,
2223
25
               "block len %" PRIu32 " does not match read data %" PRIuz, blockLen,
2224
25
               blockLen - Stream_GetRemainingLength(&sub));
2225
25
    return -1;
2226
25
  }
2227
2228
5.54k
  return rc;
2229
5.57k
}
2230
2231
static INLINE BOOL update_tiles(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
2232
                                PROGRESSIVE_SURFACE_CONTEXT* WINPR_RESTRICT surface,
2233
                                BYTE* WINPR_RESTRICT pDstData, UINT32 DstFormat, UINT32 nDstStep,
2234
                                UINT32 nXDst, UINT32 nYDst,
2235
                                PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region,
2236
                                REGION16* WINPR_RESTRICT invalidRegion)
2237
299
{
2238
299
  BOOL rc = TRUE;
2239
299
  REGION16 clippingRects = { 0 };
2240
299
  region16_init(&clippingRects);
2241
2242
12.4k
  for (UINT32 i = 0; i < region->numRects; i++)
2243
12.1k
  {
2244
12.1k
    RECTANGLE_16 clippingRect = { 0 };
2245
12.1k
    const RFX_RECT* rect = &(region->rects[i]);
2246
2247
12.1k
    clippingRect.left = (UINT16)nXDst + rect->x;
2248
12.1k
    clippingRect.top = (UINT16)nYDst + rect->y;
2249
12.1k
    clippingRect.right = clippingRect.left + rect->width;
2250
12.1k
    clippingRect.bottom = clippingRect.top + rect->height;
2251
12.1k
    region16_union_rect(&clippingRects, &clippingRects, &clippingRect);
2252
12.1k
  }
2253
2254
299
  for (UINT32 i = 0; i < surface->numUpdatedTiles; i++)
2255
0
  {
2256
0
    UINT32 nbUpdateRects = 0;
2257
0
    const RECTANGLE_16* updateRects = NULL;
2258
0
    RECTANGLE_16 updateRect = { 0 };
2259
2260
0
    WINPR_ASSERT(surface->updatedTileIndices);
2261
0
    const UINT32 index = surface->updatedTileIndices[i];
2262
2263
0
    WINPR_ASSERT(index < surface->tilesSize);
2264
0
    RFX_PROGRESSIVE_TILE* tile = surface->tiles[index];
2265
0
    WINPR_ASSERT(tile);
2266
2267
0
    const UINT32 dl = nXDst + tile->x;
2268
0
    updateRect.left = WINPR_ASSERTING_INT_CAST(UINT16, dl);
2269
2270
0
    const UINT32 dt = nYDst + tile->y;
2271
0
    updateRect.top = WINPR_ASSERTING_INT_CAST(UINT16, dt);
2272
0
    updateRect.right = updateRect.left + 64;
2273
0
    updateRect.bottom = updateRect.top + 64;
2274
2275
0
    REGION16 updateRegion = { 0 };
2276
0
    region16_init(&updateRegion);
2277
0
    region16_intersect_rect(&updateRegion, &clippingRects, &updateRect);
2278
0
    updateRects = region16_rects(&updateRegion, &nbUpdateRects);
2279
2280
0
    for (UINT32 j = 0; j < nbUpdateRects; j++)
2281
0
    {
2282
0
      rc = FALSE;
2283
0
      const RECTANGLE_16* rect = &updateRects[j];
2284
0
      if (rect->left < updateRect.left)
2285
0
        break;
2286
0
      const UINT32 nXSrc = rect->left - updateRect.left;
2287
0
      const UINT32 nYSrc = rect->top - updateRect.top;
2288
0
      const UINT32 width = rect->right - rect->left;
2289
0
      const UINT32 height = rect->bottom - rect->top;
2290
2291
0
      if (rect->left + width > surface->width)
2292
0
        break;
2293
0
      if (rect->top + height > surface->height)
2294
0
        break;
2295
0
      rc = freerdp_image_copy_no_overlap(
2296
0
          pDstData, DstFormat, nDstStep, rect->left, rect->top, width, height, tile->data,
2297
0
          progressive->format, tile->stride, nXSrc, nYSrc, NULL, FREERDP_KEEP_DST_ALPHA);
2298
0
      if (!rc)
2299
0
        break;
2300
2301
0
      if (invalidRegion)
2302
0
        region16_union_rect(invalidRegion, invalidRegion, rect);
2303
0
    }
2304
2305
0
    region16_uninit(&updateRegion);
2306
0
    if (!rc)
2307
0
      goto fail;
2308
0
    tile->dirty = FALSE;
2309
0
  }
2310
2311
299
fail:
2312
299
  region16_uninit(&clippingRects);
2313
299
  return rc;
2314
299
}
2315
2316
INT32 progressive_decompress(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
2317
                             const BYTE* WINPR_RESTRICT pSrcData, UINT32 SrcSize,
2318
                             BYTE* WINPR_RESTRICT pDstData, UINT32 DstFormat, UINT32 nDstStep,
2319
                             UINT32 nXDst, UINT32 nYDst, REGION16* WINPR_RESTRICT invalidRegion,
2320
                             UINT16 surfaceId, UINT32 frameId)
2321
5.54k
{
2322
5.54k
  INT32 rc = 1;
2323
2324
5.54k
  WINPR_ASSERT(progressive);
2325
5.54k
  PROGRESSIVE_SURFACE_CONTEXT* surface = progressive_get_surface_data(progressive, surfaceId);
2326
2327
5.54k
  if (!surface)
2328
0
  {
2329
0
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion no surface for %" PRIu16,
2330
0
               surfaceId);
2331
0
    return -1001;
2332
0
  }
2333
2334
5.54k
  PROGRESSIVE_BLOCK_REGION* WINPR_RESTRICT region = &progressive->region;
2335
5.54k
  WINPR_ASSERT(region);
2336
2337
5.54k
  if (surface->frameId != frameId)
2338
0
  {
2339
0
    surface->frameId = frameId;
2340
0
    surface->numUpdatedTiles = 0;
2341
0
  }
2342
2343
5.54k
  wStream ss = { 0 };
2344
5.54k
  wStream* s = Stream_StaticConstInit(&ss, pSrcData, SrcSize);
2345
5.54k
  WINPR_ASSERT(s);
2346
2347
5.54k
  switch (DstFormat)
2348
5.54k
  {
2349
0
    case PIXEL_FORMAT_RGBA32:
2350
0
    case PIXEL_FORMAT_RGBX32:
2351
0
    case PIXEL_FORMAT_BGRA32:
2352
5.54k
    case PIXEL_FORMAT_BGRX32:
2353
5.54k
      progressive->format = DstFormat;
2354
5.54k
      break;
2355
0
    default:
2356
0
      progressive->format = PIXEL_FORMAT_XRGB32;
2357
0
      break;
2358
5.54k
  }
2359
2360
5.54k
  const size_t start = Stream_GetPosition(s);
2361
5.54k
  progressive->state = 0; /* Set state to not initialized */
2362
11.0k
  while (Stream_GetRemainingLength(s) > 0)
2363
10.7k
  {
2364
10.7k
    if (progressive_parse_block(progressive, s, surface, region) < 0)
2365
5.24k
      goto fail;
2366
10.7k
  }
2367
2368
299
  const size_t end = Stream_GetPosition(s);
2369
299
  if ((end - start) != SrcSize)
2370
0
  {
2371
0
    WLog_Print(progressive->log, WLOG_ERROR,
2372
0
               "total block len %" PRIuz " does not match read data %" PRIu32, end - start,
2373
0
               SrcSize);
2374
0
    rc = -1041;
2375
0
    goto fail;
2376
0
  }
2377
2378
299
  if (!update_tiles(progressive, surface, pDstData, DstFormat, nDstStep, nXDst, nYDst, region,
2379
299
                    invalidRegion))
2380
0
    return -2002;
2381
5.54k
fail:
2382
5.54k
  return rc;
2383
299
}
2384
2385
BOOL progressive_rfx_write_message_progressive_simple(
2386
    PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive, wStream* WINPR_RESTRICT s,
2387
    const RFX_MESSAGE* WINPR_RESTRICT msg)
2388
0
{
2389
0
  RFX_CONTEXT* context = NULL;
2390
2391
0
  WINPR_ASSERT(progressive);
2392
0
  WINPR_ASSERT(s);
2393
0
  WINPR_ASSERT(msg);
2394
0
  context = progressive->rfx_context;
2395
0
  return rfx_write_message_progressive_simple(context, s, msg);
2396
0
}
2397
2398
int progressive_compress(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive,
2399
                         const BYTE* WINPR_RESTRICT pSrcData, UINT32 SrcSize, UINT32 SrcFormat,
2400
                         UINT32 Width, UINT32 Height, UINT32 ScanLine,
2401
                         const REGION16* WINPR_RESTRICT invalidRegion,
2402
                         BYTE** WINPR_RESTRICT ppDstData, UINT32* WINPR_RESTRICT pDstSize)
2403
0
{
2404
0
  BOOL rc = FALSE;
2405
0
  int res = -6;
2406
0
  wStream* s = NULL;
2407
0
  UINT32 numRects = 0;
2408
0
  RFX_RECT* rects = NULL;
2409
0
  RFX_MESSAGE* message = NULL;
2410
2411
0
  if (!progressive || !pSrcData || !ppDstData || !pDstSize)
2412
0
  {
2413
0
    return -1;
2414
0
  }
2415
2416
0
  if (ScanLine == 0)
2417
0
  {
2418
0
    switch (SrcFormat)
2419
0
    {
2420
0
      case PIXEL_FORMAT_ABGR32:
2421
0
      case PIXEL_FORMAT_ARGB32:
2422
0
      case PIXEL_FORMAT_XBGR32:
2423
0
      case PIXEL_FORMAT_XRGB32:
2424
0
      case PIXEL_FORMAT_BGRA32:
2425
0
      case PIXEL_FORMAT_BGRX32:
2426
0
      case PIXEL_FORMAT_RGBA32:
2427
0
      case PIXEL_FORMAT_RGBX32:
2428
0
        ScanLine = Width * 4;
2429
0
        break;
2430
0
      default:
2431
0
        return -2;
2432
0
    }
2433
0
  }
2434
2435
0
  if (SrcSize < Height * ScanLine)
2436
0
    return -4;
2437
2438
0
  if (!invalidRegion)
2439
0
  {
2440
0
    numRects = (Width + 63) / 64;
2441
0
    numRects *= (Height + 63) / 64;
2442
0
  }
2443
0
  else
2444
0
  {
2445
0
    const int nr = region16_n_rects(invalidRegion);
2446
0
    numRects = WINPR_ASSERTING_INT_CAST(uint32_t, nr);
2447
0
  }
2448
2449
0
  if (numRects == 0)
2450
0
    return 0;
2451
2452
0
  if (!Stream_EnsureRemainingCapacity(progressive->rects, numRects * sizeof(RFX_RECT)))
2453
0
    return -5;
2454
0
  rects = Stream_BufferAs(progressive->rects, RFX_RECT);
2455
0
  if (invalidRegion)
2456
0
  {
2457
0
    const RECTANGLE_16* region_rects = region16_rects(invalidRegion, NULL);
2458
0
    for (UINT32 idx = 0; idx < numRects; idx++)
2459
0
    {
2460
0
      const RECTANGLE_16* r = &region_rects[idx];
2461
0
      RFX_RECT* rect = &rects[idx];
2462
2463
0
      rect->x = r->left;
2464
0
      rect->y = r->top;
2465
0
      rect->width = r->right - r->left;
2466
0
      rect->height = r->bottom - r->top;
2467
0
    }
2468
0
  }
2469
0
  else
2470
0
  {
2471
0
    UINT16 x = 0;
2472
0
    UINT16 y = 0;
2473
2474
0
    for (UINT32 i = 0; i < numRects; i++)
2475
0
    {
2476
0
      RFX_RECT* r = &rects[i];
2477
0
      r->x = x;
2478
0
      r->y = y;
2479
2480
0
      WINPR_ASSERT(Width >= x);
2481
0
      WINPR_ASSERT(Height >= y);
2482
0
      r->width = MIN(64, WINPR_ASSERTING_INT_CAST(UINT16, Width - x));
2483
0
      r->height = MIN(64, WINPR_ASSERTING_INT_CAST(UINT16, Height - y));
2484
2485
0
      if (x + 64UL >= Width)
2486
0
      {
2487
0
        y += 64;
2488
0
        x = 0;
2489
0
      }
2490
0
      else
2491
0
        x += 64;
2492
2493
0
      WINPR_ASSERT(r->x % 64 == 0);
2494
0
      WINPR_ASSERT(r->y % 64 == 0);
2495
0
      WINPR_ASSERT(r->width <= 64);
2496
0
      WINPR_ASSERT(r->height <= 64);
2497
0
    }
2498
0
  }
2499
0
  s = progressive->buffer;
2500
0
  Stream_SetPosition(s, 0);
2501
2502
0
  progressive->rfx_context->mode = RLGR1;
2503
2504
0
  progressive->rfx_context->width = WINPR_ASSERTING_INT_CAST(UINT16, Width);
2505
0
  progressive->rfx_context->height = WINPR_ASSERTING_INT_CAST(UINT16, Height);
2506
0
  rfx_context_set_pixel_format(progressive->rfx_context, SrcFormat);
2507
0
  message = rfx_encode_message(progressive->rfx_context, rects, numRects, pSrcData, Width, Height,
2508
0
                               ScanLine);
2509
0
  if (!message)
2510
0
  {
2511
0
    WLog_ERR(TAG, "failed to encode rfx message");
2512
0
    goto fail;
2513
0
  }
2514
2515
0
  rc = progressive_rfx_write_message_progressive_simple(progressive, s, message);
2516
0
  rfx_message_free(progressive->rfx_context, message);
2517
0
  if (!rc)
2518
0
    goto fail;
2519
2520
0
  const size_t pos = Stream_GetPosition(s);
2521
0
  WINPR_ASSERT(pos <= UINT32_MAX);
2522
0
  *pDstSize = (UINT32)pos;
2523
0
  *ppDstData = Stream_Buffer(s);
2524
0
  res = 1;
2525
0
fail:
2526
0
  return res;
2527
0
}
2528
2529
BOOL progressive_context_reset(PROGRESSIVE_CONTEXT* WINPR_RESTRICT progressive)
2530
0
{
2531
0
  if (!progressive)
2532
0
    return FALSE;
2533
2534
0
  return TRUE;
2535
0
}
2536
2537
PROGRESSIVE_CONTEXT* progressive_context_new(BOOL Compressor)
2538
5.54k
{
2539
5.54k
  return progressive_context_new_ex(Compressor, 0);
2540
5.54k
}
2541
2542
PROGRESSIVE_CONTEXT* progressive_context_new_ex(BOOL Compressor, UINT32 ThreadingFlags)
2543
5.54k
{
2544
5.54k
  PROGRESSIVE_CONTEXT* progressive =
2545
5.54k
      (PROGRESSIVE_CONTEXT*)winpr_aligned_calloc(1, sizeof(PROGRESSIVE_CONTEXT), 32);
2546
2547
5.54k
  if (!progressive)
2548
0
    return NULL;
2549
2550
5.54k
  progressive->Compressor = Compressor;
2551
5.54k
  progressive->quantProgValFull.quality = 100;
2552
5.54k
  progressive->log = WLog_Get(TAG);
2553
5.54k
  if (!progressive->log)
2554
0
    goto fail;
2555
5.54k
  progressive->rfx_context = rfx_context_new_ex(Compressor, ThreadingFlags);
2556
5.54k
  if (!progressive->rfx_context)
2557
0
    goto fail;
2558
5.54k
  progressive->buffer = Stream_New(NULL, 1024);
2559
5.54k
  if (!progressive->buffer)
2560
0
    goto fail;
2561
5.54k
  progressive->rects = Stream_New(NULL, 1024);
2562
5.54k
  if (!progressive->rects)
2563
0
    goto fail;
2564
5.54k
  progressive->bufferPool = BufferPool_New(TRUE, (8192LL + 32LL) * 3LL, 16);
2565
5.54k
  if (!progressive->bufferPool)
2566
0
    goto fail;
2567
5.54k
  progressive->SurfaceContexts = HashTable_New(TRUE);
2568
5.54k
  if (!progressive->SurfaceContexts)
2569
0
    goto fail;
2570
2571
5.54k
  {
2572
5.54k
    wObject* obj = HashTable_ValueObject(progressive->SurfaceContexts);
2573
5.54k
    WINPR_ASSERT(obj);
2574
5.54k
    obj->fnObjectFree = progressive_surface_context_free;
2575
5.54k
  }
2576
0
  return progressive;
2577
0
fail:
2578
0
  WINPR_PRAGMA_DIAG_PUSH
2579
0
  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
2580
0
  progressive_context_free(progressive);
2581
0
  WINPR_PRAGMA_DIAG_POP
2582
0
  return NULL;
2583
5.54k
}
2584
2585
void progressive_context_free(PROGRESSIVE_CONTEXT* progressive)
2586
5.54k
{
2587
5.54k
  if (!progressive)
2588
0
    return;
2589
2590
5.54k
  Stream_Free(progressive->buffer, TRUE);
2591
5.54k
  Stream_Free(progressive->rects, TRUE);
2592
5.54k
  rfx_context_free(progressive->rfx_context);
2593
2594
5.54k
  BufferPool_Free(progressive->bufferPool);
2595
5.54k
  HashTable_Free(progressive->SurfaceContexts);
2596
2597
5.54k
  winpr_aligned_free(progressive);
2598
5.54k
}