dany/libwebp
1
0
Fork 0
mirror of https://github.com/webmproject/libwebp.git synced 2025-08-29 07:12:05 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge "Add fbounds-safety annotations in palette.c/.h." into main

Browse Source
This commit is contained in:
James Zern
2025-08-20 15:10:00 -07:00
committed by Gerrit Code Review
parent b7d30cfd94 52135b8e00
commit dab2cf21fa
2 changed files with 50 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
src/utils/palette.c
View File

@@ -65,7 +65,8 @@ static WEBP_INLINE void SwapColor(uint32_t* const col1, uint32_t* const col2) {
*col2 = tmp; *col2 = tmp;
} }
int SearchColorNoIdx(const uint32_t sorted[], uint32_t color, int num_colors) { int SearchColorNoIdx(const uint32_t WEBP_COUNTED_BY(num_colors) sorted[],
uint32_t color, int num_colors) {
int low = 0, hi = num_colors; int low = 0, hi = num_colors;
if (sorted[low] == color) return low; // loop invariant: sorted[low] != color if (sorted[low] == color) return low; // loop invariant: sorted[low] != color
while (1) { while (1) {
@@ -82,10 +83,12 @@ int SearchColorNoIdx(const uint32_t sorted[], uint32_t color, int num_colors) {
return 0; return 0;
} }
void PrepareMapToPalette(const uint32_t palette[], uint32_t num_colors, void PrepareMapToPalette(const uint32_t WEBP_COUNTED_BY(num_colors) palette[],
uint32_t sorted[], uint32_t idx_map[]) { uint32_t num_colors,
uint32_t WEBP_COUNTED_BY(num_colors) sorted[],
uint32_t WEBP_COUNTED_BY(num_colors) idx_map[]) {
uint32_t i; uint32_t i;
WEBP_UNSAFE_MEMCPY(sorted, palette, num_colors * sizeof(*sorted)); memcpy(sorted, palette, num_colors * sizeof(*sorted));
qsort(sorted, num_colors, sizeof(*sorted), PaletteCompareColorsForQsort); qsort(sorted, num_colors, sizeof(*sorted), PaletteCompareColorsForQsort);
for (i = 0; i < num_colors; ++i) { for (i = 0; i < num_colors; ++i) {
idx_map[SearchColorNoIdx(sorted, palette[i], num_colors)] = i; idx_map[SearchColorNoIdx(sorted, palette[i], num_colors)] = i;
@@ -165,8 +168,8 @@ int GetColorPalette(const WebPPicture* const pic,
// no benefit to re-organize them greedily. A monotonic development // no benefit to re-organize them greedily. A monotonic development
// would be spotted in green-only situations (like lossy alpha) or gray-scale // would be spotted in green-only situations (like lossy alpha) or gray-scale
// images. // images.
static int PaletteHasNonMonotonousDeltas(const uint32_t* const palette, static int PaletteHasNonMonotonousDeltas(
int num_colors) { const uint32_t* const WEBP_COUNTED_BY(num_colors) palette, int num_colors) {
uint32_t predict = 0x000000; uint32_t predict = 0x000000;
int i; int i;
uint8_t sign_found = 0x00; uint8_t sign_found = 0x00;
@@ -189,11 +192,12 @@ static int PaletteHasNonMonotonousDeltas(const uint32_t* const palette,
return (sign_found & (sign_found << 1)) != 0; // two consequent signs. return (sign_found & (sign_found << 1)) != 0; // two consequent signs.
} }
static void PaletteSortMinimizeDeltas(const uint32_t* const palette_sorted, static void PaletteSortMinimizeDeltas(
int num_colors, uint32_t* const palette) { const uint32_t* const WEBP_COUNTED_BY(num_colors) palette_sorted,
int num_colors, uint32_t* const WEBP_COUNTED_BY(num_colors) palette) {
uint32_t predict = 0x00000000; uint32_t predict = 0x00000000;
int i, k; int i, k;
WEBP_UNSAFE_MEMCPY(palette, palette_sorted, num_colors * sizeof(*palette)); memcpy(palette, palette_sorted, num_colors * sizeof(*palette));
if (!PaletteHasNonMonotonousDeltas(palette_sorted, num_colors)) return; if (!PaletteHasNonMonotonousDeltas(palette_sorted, num_colors)) return;
// Find greedily always the closest color of the predicted color to minimize // Find greedily always the closest color of the predicted color to minimize
// deltas in the palette. This reduces storage needs since the // deltas in the palette. This reduces storage needs since the
@@ -225,9 +229,9 @@ static void PaletteSortMinimizeDeltas(const uint32_t* const palette_sorted,
// Pinho and Antonio J. R. Neves. // Pinho and Antonio J. R. Neves.
// Finds the biggest cooccurrence in the matrix. // Finds the biggest cooccurrence in the matrix.
static void CoOccurrenceFindMax(const uint32_t* const cooccurrence, static void CoOccurrenceFindMax(
uint32_t num_colors, uint8_t* const c1, const uint32_t* const WEBP_COUNTED_BY(num_colors* num_colors) cooccurrence,
uint8_t* const c2) { uint32_t num_colors, uint8_t* const c1, uint8_t* const c2) {
// Find the index that is most frequently located adjacent to other // Find the index that is most frequently located adjacent to other
// (different) indexes. // (different) indexes.
uint32_t best_sum = 0u; uint32_t best_sum = 0u;
@@ -255,8 +259,11 @@ static void CoOccurrenceFindMax(const uint32_t* const cooccurrence,
// Builds the cooccurrence matrix // Builds the cooccurrence matrix
static int CoOccurrenceBuild(const WebPPicture* const pic, static int CoOccurrenceBuild(const WebPPicture* const pic,
const uint32_t* const palette, uint32_t num_colors, const uint32_t* const WEBP_COUNTED_BY(num_colors)
uint32_t* cooccurrence) { palette,
uint32_t num_colors,
uint32_t* WEBP_COUNTED_BY(num_colors* num_colors)
cooccurrence) {
uint32_t *lines, *line_top, *line_current, *line_tmp; uint32_t *lines, *line_top, *line_current, *line_tmp;
int x, y; int x, y;
const uint32_t* src = pic->argb; const uint32_t* src = pic->argb;
@@ -306,10 +313,10 @@ struct Sum {
uint32_t sum; uint32_t sum;
}; };
static int PaletteSortModifiedZeng(const WebPPicture* const pic, static int PaletteSortModifiedZeng(
const uint32_t* const palette_in, const WebPPicture* const pic,
uint32_t num_colors, const uint32_t* const WEBP_COUNTED_BY(num_colors) palette_in,
uint32_t* const palette) { uint32_t num_colors, uint32_t* const WEBP_COUNTED_BY(num_colors) palette) {
uint32_t i, j, ind; uint32_t i, j, ind;
uint8_t remapping[MAX_PALETTE_SIZE]; uint8_t remapping[MAX_PALETTE_SIZE];
uint32_t* cooccurrence; uint32_t* cooccurrence;
@@ -324,13 +331,19 @@ static int PaletteSortModifiedZeng(const WebPPicture* const pic,
if (cooccurrence == NULL) { if (cooccurrence == NULL) {
return 0; return 0;
} }
if (!CoOccurrenceBuild(pic, palette_in, num_colors, cooccurrence)) { if (!CoOccurrenceBuild(pic, palette_in, num_colors,
WEBP_UNSAFE_FORGE_BIDI_INDEXABLE(
uint32_t*, cooccurrence,
num_colors* num_colors * sizeof(*cooccurrence)))) {
WebPSafeFree(cooccurrence); WebPSafeFree(cooccurrence);
return 0; return 0;
} }
// Initialize the mapping list with the two best indices. // Initialize the mapping list with the two best indices.
CoOccurrenceFindMax(cooccurrence, num_colors, &remapping[0], &remapping[1]); CoOccurrenceFindMax(WEBP_UNSAFE_FORGE_BIDI_INDEXABLE(
const uint32_t*, cooccurrence,
num_colors* num_colors * sizeof(*cooccurrence)),
num_colors, &remapping[0], &remapping[1]);
// We need to append and prepend to the list of remapping. To this end, we // We need to append and prepend to the list of remapping. To this end, we
// actually define the next start/end of the list as indices in a vector (with // actually define the next start/end of the list as indices in a vector (with
@@ -393,17 +406,18 @@ static int PaletteSortModifiedZeng(const WebPPicture* const pic,
// ----------------------------------------------------------------------------- // -----------------------------------------------------------------------------
int PaletteSort(PaletteSorting method, const struct WebPPicture* const pic, int PaletteSort(PaletteSorting method, const struct WebPPicture* const pic,
const uint32_t* const palette_sorted, uint32_t num_colors, const uint32_t* const WEBP_COUNTED_BY(num_colors)
uint32_t* const palette) { palette_sorted,
uint32_t num_colors,
uint32_t* const WEBP_COUNTED_BY(num_colors) palette) {
switch (method) { switch (method) {
case kSortedDefault: case kSortedDefault:
if (palette_sorted[0] == 0 && num_colors > 17) { if (palette_sorted[0] == 0 && num_colors > 17) {
WEBP_UNSAFE_MEMCPY(palette, palette_sorted + 1, memcpy(palette, palette_sorted + 1,
(num_colors - 1) * sizeof(*palette_sorted)); (num_colors - 1) * sizeof(*palette_sorted));
palette[num_colors - 1] = 0; palette[num_colors - 1] = 0;
} else { } else {
WEBP_UNSAFE_MEMCPY(palette, palette_sorted, memcpy(palette, palette_sorted, num_colors * sizeof(*palette));
num_colors * sizeof(*palette));
} }
return 1; return 1;
case kMinimizeDelta: case kMinimizeDelta:

15
src/utils/palette.h
View File

@@ -38,11 +38,14 @@ typedef enum PaletteSorting {
// Returns the index of 'color' in the sorted palette 'sorted' of size // Returns the index of 'color' in the sorted palette 'sorted' of size
// 'num_colors'. // 'num_colors'.
int SearchColorNoIdx(const uint32_t sorted[], uint32_t color, int num_colors); int SearchColorNoIdx(const uint32_t WEBP_COUNTED_BY(num_colors) sorted[],
uint32_t color, int num_colors);
// Sort palette in increasing order and prepare an inverse mapping array. // Sort palette in increasing order and prepare an inverse mapping array.
void PrepareMapToPalette(const uint32_t palette[], uint32_t num_colors, void PrepareMapToPalette(const uint32_t WEBP_COUNTED_BY(num_colors) palette[],
uint32_t sorted[], uint32_t idx_map[]); uint32_t num_colors,
uint32_t WEBP_COUNTED_BY(num_colors) sorted[],
uint32_t WEBP_COUNTED_BY(num_colors) idx_map[]);
// Returns count of unique colors in 'pic', assuming pic->use_argb is true. // Returns count of unique colors in 'pic', assuming pic->use_argb is true.
// If the unique color count is more than MAX_PALETTE_SIZE, returns // If the unique color count is more than MAX_PALETTE_SIZE, returns
@@ -61,7 +64,9 @@ int GetColorPalette(const struct WebPPicture* const pic,
// For kSortedDefault and kMinimizeDelta methods, 0 (if present) is set as the // For kSortedDefault and kMinimizeDelta methods, 0 (if present) is set as the
// last element to optimize later storage. // last element to optimize later storage.
int PaletteSort(PaletteSorting method, const struct WebPPicture* const pic, int PaletteSort(PaletteSorting method, const struct WebPPicture* const pic,
const uint32_t* const palette_sorted, uint32_t num_colors, const uint32_t* const WEBP_COUNTED_BY(num_colors)
uint32_t* const palette); palette_sorted,
uint32_t num_colors,
uint32_t* const WEBP_COUNTED_BY(num_colors) palette);
#endif // WEBP_UTILS_PALETTE_H_ #endif // WEBP_UTILS_PALETTE_H_