Provide safer string copy than strncpy
[openjpeg.git] / src / bin / jp2 / opj_decompress.c
index 2dfb3400ebaae179ffc5579597a6d4daf8ae8ab8..3db36ebe6d53f7c35699ecba9a367a7378e6ddd4 100644 (file)
@@ -75,6 +75,7 @@
 #include "color.h"
 
 #include "format_defs.h"
+#include "opj_string.h"
 
 typedef struct dircnt{
        /** Buffer for holding images read from Directory*/
@@ -121,6 +122,8 @@ typedef struct opj_decompress_params
        int decod_format;
        /** output file format 0: PGX, 1: PxM, 2: BMP */
        int cod_format;
+       /** index file name */
+       char indexfilename[OPJ_PATH_LEN];
        
        /** Decoding area left boundary */
        OPJ_UINT32 DA_x0;
@@ -157,7 +160,7 @@ int get_file_format(const char *filename);
 char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_decompress_parameters *parameters);
 static int infile_format(const char *fname);
 
-int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol, char *indexfilename);
+int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol);
 int parse_DA_values( char* inArg, unsigned int *DA_x0, unsigned int *DA_y0, unsigned int *DA_x1, unsigned int *DA_y1);
 
 static opj_image_t* convert_gray_to_rgb(opj_image_t* original);
@@ -210,13 +213,13 @@ static void decode_help_display(void) {
                       "    By default all tiles are decoded.\n");
        fprintf(stdout,"  -p <comp 0 precision>[C|S][,<comp 1 precision>[C|S][,...]]\n"
                       "    OPTIONAL\n"
-                      "    Force the precision (bit depth) of components.\n"
-                      "    There shall be at least 1 value. Theres no limit on the number of values (comma separated, last values ignored if too much values).\n"
+                      "    Force the precision (bit depth) of components.\n");
+       fprintf(stdout,"    There shall be at least 1 value. Theres no limit on the number of values (comma separated, last values ignored if too much values).\n"
                       "    If there are less values than components, the last value is used for remaining components.\n"
                       "    If 'C' is specified (default), values are clipped.\n"
                       "    If 'S' is specified, values are scaled.\n"
-                      "    A 0 value can be specified (meaning original bit depth).\n"
-                      "  -force-rgb\n"
+                      "    A 0 value can be specified (meaning original bit depth).\n");
+       fprintf(stdout,"  -force-rgb\n"
                       "    Force output image colorspace to RGB\n"
                       "  -upsample\n"
                       "    Downsampled components will be upsampled to image size\n"
@@ -361,6 +364,7 @@ int get_num_images(char *imgdirpath){
                        continue;
                num_images++;
        }
+       closedir(dir);
        return num_images;
 }
 
@@ -387,6 +391,7 @@ int load_images(dircnt_t *dirptr, char *imgdirpath){
                strcpy(dirptr->filename[i],content->d_name);
                i++;
        }
+       closedir(dir);
        return 0;       
 }
 
@@ -427,7 +432,9 @@ char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_decompre
        parameters->decod_format = infile_format(infilename);
        if (parameters->decod_format == -1)
                return 1;
-       strncpy(parameters->infile, infilename, sizeof(infilename));
+       if (opj_strcpy_s(parameters->infile, sizeof(parameters->infile), infilename) != 0) {
+               return 1;
+       }
 
        /*Set output file*/
        strcpy(temp_ofname,strtok(image_filename,"."));
@@ -437,7 +444,9 @@ char get_next_file(int imageno,dircnt_t *dirptr,img_fol_t *img_fol, opj_decompre
        }
        if(img_fol->set_out_format==1){
                sprintf(outfilename,"%s/%s.%s",img_fol->imgdirpath,temp_ofname,img_fol->out_format);
-               strncpy(parameters->outfile, outfilename, sizeof(outfilename));
+               if (opj_strcpy_s(parameters->outfile, sizeof(parameters->outfile), outfilename) != 0) {
+                       return 1;
+               }
        }
        return 0;
 }
@@ -503,15 +512,15 @@ static int infile_format(const char *fname)
  * Parse the command line
  */
 /* -------------------------------------------------------------------------- */
-int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol, char *indexfilename) {
+int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *parameters,img_fol_t *img_fol) {
        /* parse the command line */
        int totlen, c;
        opj_option_t long_option[]={
-               {"ImgDir",    REQ_ARG, NULL ,'y'},
-               {"OutFor",    REQ_ARG, NULL ,'O'},
-               {"force-rgb", NO_ARG,  &(parameters->force_rgb), 1},
-               {"upsample",  NO_ARG,  &(parameters->upsample),  1},
-               {"split-pnm", NO_ARG,  &(parameters->split_pnm), 1}
+               {"ImgDir",    REQ_ARG, NULL,'y'},
+               {"OutFor",    REQ_ARG, NULL,'O'},
+               {"force-rgb", NO_ARG,  NULL, 1},
+               {"upsample",  NO_ARG,  NULL, 1},
+               {"split-pnm", NO_ARG,  NULL, 1}
        };
 
        const char optlist[] = "i:o:r:l:x:d:t:p:"
@@ -522,6 +531,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
 #endif /* USE_JPWL */
 /* <<UniPG */
             "h"                ;
+
+       long_option[2].flag = &(parameters->force_rgb);
+       long_option[3].flag = &(parameters->upsample);
+       long_option[4].flag = &(parameters->split_pnm);
        totlen=sizeof(long_option);
        opj_reset_options_reading();
        img_fol->set_out_format = 0;
@@ -555,7 +568,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
                                                        infile);
                                                return 1;
                                }
-                               strncpy(parameters->infile, infile, sizeof(parameters->infile)-1);
+                               if (opj_strcpy_s(parameters->infile, sizeof(parameters->infile), infile) != 0) {
+                                       fprintf(stderr, "[ERROR] Path is too long\n");
+                                       return 1;
+                               }
                        }
                        break;
                                
@@ -586,7 +602,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
                                                fprintf(stderr, "Unknown output format image %s [only *.pnm, *.pgm, *.ppm, *.pgx, *.bmp, *.tif, *.raw or *.tga]!! \n", outfile);
                                                return 1;
                                }
-                               strncpy(parameters->outfile, outfile, sizeof(parameters->outfile)-1);
+                               if (opj_strcpy_s(parameters->outfile, sizeof(parameters->outfile), outfile) != 0) {
+                                       fprintf(stderr, "[ERROR] Path is too long\n");
+                                       return 1;
+                               }
                        }
                        break;
                        
@@ -670,11 +689,14 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
 
                        case 'd':               /* Input decode ROI */
                        {
-                               int size_optarg = (int)strlen(opj_optarg) + 1;
-                               char *ROI_values = (char*) malloc((size_t)size_optarg);
+                               size_t size_optarg = (size_t)strlen(opj_optarg) + 1U;
+                               char *ROI_values = (char*) malloc(size_optarg);
+                               if (ROI_values == NULL) {
+                                       fprintf(stderr, "[ERROR] Couldn't allocate memory\n");
+                                       return 1;
+                               }
                                ROI_values[0] = '\0';
-                               strncpy(ROI_values, opj_optarg, strlen(opj_optarg));
-                               ROI_values[strlen(opj_optarg)] = '\0';
+                               memcpy(ROI_values, opj_optarg, size_optarg);
                                /*printf("ROI_values = %s [%d / %d]\n", ROI_values, strlen(ROI_values), size_optarg ); */
                                parse_DA_values( ROI_values, &parameters->DA_x0, &parameters->DA_y0, &parameters->DA_x1, &parameters->DA_y1);
 
@@ -695,8 +717,10 @@ int parse_cmdline_decoder(int argc, char **argv, opj_decompress_parameters *para
 
                        case 'x':                       /* Creation of index file */
                                {
-                                       char *index = opj_optarg;
-                                       strncpy(indexfilename, index, OPJ_PATH_LEN);
+                                       if (opj_strcpy_s(parameters->indexfilename, sizeof(parameters->indexfilename), opj_optarg) != 0) {
+                                               fprintf(stderr, "[ERROR] Path is too long\n");
+                                               return 1;
+                                       }
                                }
                                break;
                                
@@ -1161,8 +1185,6 @@ int main(int argc, char **argv)
        opj_codec_t* l_codec = NULL;                            /* Handle to a decompressor */
        opj_codestream_index_t* cstr_index = NULL;
 
-       char indexfilename[OPJ_PATH_LEN];       /* index file name */
-
        OPJ_INT32 num_images, imageno;
        img_fol_t img_fol;
        dircnt_t *dirptr = NULL;
@@ -1173,14 +1195,11 @@ int main(int argc, char **argv)
        /* set decoding parameters to default values */
        set_default_parameters(&parameters);
 
-       /* FIXME Initialize indexfilename and img_fol */
-       *indexfilename = 0;
-
        /* Initialize img_fol */
        memset(&img_fol,0,sizeof(img_fol_t));
 
        /* parse input and get user encoding parameters */
-       if(parse_cmdline_decoder(argc, argv, &parameters,&img_fol, indexfilename) == 1) {
+       if(parse_cmdline_decoder(argc, argv, &parameters,&img_fol) == 1) {
                destroy_parameters(&parameters);
                return EXIT_FAILURE;
        }
@@ -1345,10 +1364,6 @@ int main(int argc, char **argv)
                /* Close the byte stream */
                opj_stream_destroy(l_stream);
 
-               if(image->color_space == OPJ_CLRSPC_SYCC){
-                       color_sycc_to_rgb(image); /* FIXME */
-               }
-               
                if( image->color_space != OPJ_CLRSPC_SYCC 
                        && image->numcomps == 3 && image->comps[0].dx == image->comps[0].dy
                        && image->comps[1].dx != 1 )
@@ -1356,9 +1371,22 @@ int main(int argc, char **argv)
                else if (image->numcomps <= 2)
                        image->color_space = OPJ_CLRSPC_GRAY;
 
+               if(image->color_space == OPJ_CLRSPC_SYCC){
+                       color_sycc_to_rgb(image);
+               }
+               else if((image->color_space == OPJ_CLRSPC_CMYK) && (parameters.cod_format != TIF_DFMT)){
+                       color_cmyk_to_rgb(image);
+               }
+               else if(image->color_space == OPJ_CLRSPC_EYCC){
+                       color_esycc_to_rgb(image);
+               }
+               
                if(image->icc_profile_buf) {
 #if defined(OPJ_HAVE_LIBLCMS1) || defined(OPJ_HAVE_LIBLCMS2)
-                       color_apply_icc_profile(image); /* FIXME */
+                       if(image->icc_profile_len)
+                        color_apply_icc_profile(image);
+                       else
+                        color_cielab_to_rgb(image);
 #endif
                        free(image->icc_profile_buf);
                        image->icc_profile_buf = NULL; image->icc_profile_len = 0;
@@ -1537,7 +1565,7 @@ int main(int argc, char **argv)
                /* destroy the codestream index */
                opj_destroy_cstr_index(&cstr_index);
 
-               if(failed) remove(parameters.outfile);
+               if(failed) (void)remove(parameters.outfile); /* ignore return value */
        }
        destroy_parameters(&parameters);
        if (numDecompressedImages) {