size = abs(header->bitpix) >> 3;
for (i = 0; i < header->naxis; i++) {
- if (header->naxisn[i] > SIZE_MAX / size) {
+ if (size && header->naxisn[i] > SIZE_MAX / size) {
av_log(avctx, AV_LOG_ERROR, "unsupported size of FITS image");
return AVERROR_INVALIDDATA;
}
header->data_min = (header->data_min - header->bzero) / header->bscale;
header->data_max = (header->data_max - header->bzero) / header->bscale;
}
+ if (!header->rgb && header->data_min >= header->data_max) {
+ if (header->data_min > header->data_max) {
+ av_log(avctx, AV_LOG_ERROR, "data min/max (%g %g) is invalid\n", header->data_min, header->data_max);
+ return AVERROR_INVALIDDATA;
+ }
+ av_log(avctx, AV_LOG_WARNING, "data min/max indicates a blank image\n");
+ header->data_max ++;
+ }
return 0;
}
uint8_t *dst8;
uint16_t *dst16;
uint64_t t;
+ double scale;
FITSHeader header;
FITSContext * fitsctx = avctx->priv_data;
if (ret < 0)
return ret;
+ scale = header.data_max - header.data_min;
+ if (scale <= 0 || !isfinite(scale)) {
+ scale = 1;
+ }
+ scale = 1/scale;
+
if (header.rgb) {
if (header.bitpix == 8) {
if (header.naxisn[2] == 3) {
for (j = 0; j < avctx->width; j++) { \
t = rd; \
if (!header.blank_found || t != header.blank) { \
- *dst++ = ((t - header.data_min) * ((1 << (sizeof(type) * 8)) - 1)) / (header.data_max - header.data_min); \
+ *dst++ = ((t - header.data_min) * ((1 << (sizeof(type) * 8)) - 1)) * scale; \
} else { \
*dst++ = fitsctx->blank_val; \
} \