fixed undefined behaviour in image codecs initialization
This commit is contained in:
parent
8c9a9f1c94
commit
d88bbb3d9d
@ -54,101 +54,6 @@
|
|||||||
namespace cv
|
namespace cv
|
||||||
{
|
{
|
||||||
|
|
||||||
static vector<ImageDecoder> decoders;
|
|
||||||
static vector<ImageEncoder> encoders;
|
|
||||||
|
|
||||||
static ImageDecoder findDecoder( const string& filename )
|
|
||||||
{
|
|
||||||
size_t i, maxlen = 0;
|
|
||||||
for( i = 0; i < decoders.size(); i++ )
|
|
||||||
{
|
|
||||||
size_t len = decoders[i]->signatureLength();
|
|
||||||
maxlen = std::max(maxlen, len);
|
|
||||||
}
|
|
||||||
|
|
||||||
FILE* f= fopen( filename.c_str(), "rb" );
|
|
||||||
if( !f )
|
|
||||||
return ImageDecoder();
|
|
||||||
string signature(maxlen, ' ');
|
|
||||||
maxlen = fread( &signature[0], 1, maxlen, f );
|
|
||||||
fclose(f);
|
|
||||||
signature = signature.substr(0, maxlen);
|
|
||||||
|
|
||||||
for( i = 0; i < decoders.size(); i++ )
|
|
||||||
{
|
|
||||||
if( decoders[i]->checkSignature(signature) )
|
|
||||||
return decoders[i]->newDecoder();
|
|
||||||
}
|
|
||||||
|
|
||||||
return ImageDecoder();
|
|
||||||
}
|
|
||||||
|
|
||||||
static ImageDecoder findDecoder( const Mat& buf )
|
|
||||||
{
|
|
||||||
size_t i, maxlen = 0;
|
|
||||||
|
|
||||||
if( buf.rows*buf.cols < 1 || !buf.isContinuous() )
|
|
||||||
return ImageDecoder();
|
|
||||||
|
|
||||||
for( i = 0; i < decoders.size(); i++ )
|
|
||||||
{
|
|
||||||
size_t len = decoders[i]->signatureLength();
|
|
||||||
maxlen = std::max(maxlen, len);
|
|
||||||
}
|
|
||||||
|
|
||||||
size_t bufSize = buf.rows*buf.cols*buf.elemSize();
|
|
||||||
maxlen = std::min(maxlen, bufSize);
|
|
||||||
string signature(maxlen, ' ');
|
|
||||||
memcpy( &signature[0], buf.data, maxlen );
|
|
||||||
|
|
||||||
for( i = 0; i < decoders.size(); i++ )
|
|
||||||
{
|
|
||||||
if( decoders[i]->checkSignature(signature) )
|
|
||||||
return decoders[i]->newDecoder();
|
|
||||||
}
|
|
||||||
|
|
||||||
return ImageDecoder();
|
|
||||||
}
|
|
||||||
|
|
||||||
static ImageEncoder findEncoder( const string& _ext )
|
|
||||||
{
|
|
||||||
if( _ext.size() <= 1 )
|
|
||||||
return ImageEncoder();
|
|
||||||
|
|
||||||
const char* ext = strrchr( _ext.c_str(), '.' );
|
|
||||||
if( !ext )
|
|
||||||
return ImageEncoder();
|
|
||||||
int len = 0;
|
|
||||||
for( ext++; isalnum(ext[len]) && len < 128; len++ )
|
|
||||||
;
|
|
||||||
|
|
||||||
for( size_t i = 0; i < encoders.size(); i++ )
|
|
||||||
{
|
|
||||||
string description = encoders[i]->getDescription();
|
|
||||||
const char* descr = strchr( description.c_str(), '(' );
|
|
||||||
|
|
||||||
while( descr )
|
|
||||||
{
|
|
||||||
descr = strchr( descr + 1, '.' );
|
|
||||||
if( !descr )
|
|
||||||
break;
|
|
||||||
int j = 0;
|
|
||||||
for( descr++; isalnum(descr[j]) && j < len; j++ )
|
|
||||||
{
|
|
||||||
int c1 = tolower(ext[j]);
|
|
||||||
int c2 = tolower(descr[j]);
|
|
||||||
if( c1 != c2 )
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if( j == len && !isalnum(descr[j]))
|
|
||||||
return encoders[i]->newEncoder();
|
|
||||||
descr += j;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return ImageEncoder();
|
|
||||||
}
|
|
||||||
|
|
||||||
struct ImageCodecInitializer
|
struct ImageCodecInitializer
|
||||||
{
|
{
|
||||||
ImageCodecInitializer()
|
ImageCodecInitializer()
|
||||||
@ -186,10 +91,104 @@ struct ImageCodecInitializer
|
|||||||
encoders.push_back( new ImageIOEncoder );
|
encoders.push_back( new ImageIOEncoder );
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
vector<ImageDecoder> decoders;
|
||||||
|
vector<ImageEncoder> encoders;
|
||||||
};
|
};
|
||||||
|
|
||||||
static ImageCodecInitializer initialize_codecs;
|
static ImageCodecInitializer codecs;
|
||||||
|
|
||||||
|
static ImageDecoder findDecoder( const string& filename )
|
||||||
|
{
|
||||||
|
size_t i, maxlen = 0;
|
||||||
|
for( i = 0; i < codecs.decoders.size(); i++ )
|
||||||
|
{
|
||||||
|
size_t len = codecs.decoders[i]->signatureLength();
|
||||||
|
maxlen = std::max(maxlen, len);
|
||||||
|
}
|
||||||
|
|
||||||
|
FILE* f= fopen( filename.c_str(), "rb" );
|
||||||
|
if( !f )
|
||||||
|
return ImageDecoder();
|
||||||
|
string signature(maxlen, ' ');
|
||||||
|
maxlen = fread( &signature[0], 1, maxlen, f );
|
||||||
|
fclose(f);
|
||||||
|
signature = signature.substr(0, maxlen);
|
||||||
|
|
||||||
|
for( i = 0; i < codecs.decoders.size(); i++ )
|
||||||
|
{
|
||||||
|
if( codecs.decoders[i]->checkSignature(signature) )
|
||||||
|
return codecs.decoders[i]->newDecoder();
|
||||||
|
}
|
||||||
|
|
||||||
|
return ImageDecoder();
|
||||||
|
}
|
||||||
|
|
||||||
|
static ImageDecoder findDecoder( const Mat& buf )
|
||||||
|
{
|
||||||
|
size_t i, maxlen = 0;
|
||||||
|
|
||||||
|
if( buf.rows*buf.cols < 1 || !buf.isContinuous() )
|
||||||
|
return ImageDecoder();
|
||||||
|
|
||||||
|
for( i = 0; i < codecs.decoders.size(); i++ )
|
||||||
|
{
|
||||||
|
size_t len = codecs.decoders[i]->signatureLength();
|
||||||
|
maxlen = std::max(maxlen, len);
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t bufSize = buf.rows*buf.cols*buf.elemSize();
|
||||||
|
maxlen = std::min(maxlen, bufSize);
|
||||||
|
string signature(maxlen, ' ');
|
||||||
|
memcpy( &signature[0], buf.data, maxlen );
|
||||||
|
|
||||||
|
for( i = 0; i < codecs.decoders.size(); i++ )
|
||||||
|
{
|
||||||
|
if( codecs.decoders[i]->checkSignature(signature) )
|
||||||
|
return codecs.decoders[i]->newDecoder();
|
||||||
|
}
|
||||||
|
|
||||||
|
return ImageDecoder();
|
||||||
|
}
|
||||||
|
|
||||||
|
static ImageEncoder findEncoder( const string& _ext )
|
||||||
|
{
|
||||||
|
if( _ext.size() <= 1 )
|
||||||
|
return ImageEncoder();
|
||||||
|
|
||||||
|
const char* ext = strrchr( _ext.c_str(), '.' );
|
||||||
|
if( !ext )
|
||||||
|
return ImageEncoder();
|
||||||
|
int len = 0;
|
||||||
|
for( ext++; isalnum(ext[len]) && len < 128; len++ )
|
||||||
|
;
|
||||||
|
|
||||||
|
for( size_t i = 0; i < codecs.encoders.size(); i++ )
|
||||||
|
{
|
||||||
|
string description = codecs.encoders[i]->getDescription();
|
||||||
|
const char* descr = strchr( description.c_str(), '(' );
|
||||||
|
|
||||||
|
while( descr )
|
||||||
|
{
|
||||||
|
descr = strchr( descr + 1, '.' );
|
||||||
|
if( !descr )
|
||||||
|
break;
|
||||||
|
int j = 0;
|
||||||
|
for( descr++; isalnum(descr[j]) && j < len; j++ )
|
||||||
|
{
|
||||||
|
int c1 = tolower(ext[j]);
|
||||||
|
int c2 = tolower(descr[j]);
|
||||||
|
if( c1 != c2 )
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if( j == len && !isalnum(descr[j]))
|
||||||
|
return codecs.encoders[i]->newEncoder();
|
||||||
|
descr += j;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ImageEncoder();
|
||||||
|
}
|
||||||
|
|
||||||
enum { LOAD_CVMAT=0, LOAD_IMAGE=1, LOAD_MAT=2 };
|
enum { LOAD_CVMAT=0, LOAD_IMAGE=1, LOAD_MAT=2 };
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user