lib/libtiff
1
0
Fork 0
mirror of https://gitlab.com/libtiff/libtiff.git synced 2026-10-27 20:32:57 +03:00
Code Activity
Files
requirements
libtiff/tools/tiffsplit.c

446 lines
13 KiB
C
Raw Permalink Blame History

/*
* Copyright (c) 1992-1997 Sam Leffler
* Copyright (c) 1992-1997 Silicon Graphics, Inc.
*
* Permission to use, copy, modify, distribute, and sell this software and
* its documentation for any purpose is hereby granted without fee, provided
* that (i) the above copyright notices and this permission notice appear in
* all copies of the software and related documentation, and (ii) the names of
* Sam Leffler and Silicon Graphics may not be used in any advertising or
* publicity relating to the software without the specific, prior written
* permission of Sam Leffler and Silicon Graphics.
*
* THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
* EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
* WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
*
* IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
* ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
* OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
* WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
* LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
* OF THIS SOFTWARE.
*/
#include "libport.h"
#include "tif_config.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "tiff_tools.h"
#include "tiffio.h"
#ifndef EXIT_SUCCESS
#define EXIT_SUCCESS 0
#endif
#ifndef EXIT_FAILURE
#define EXIT_FAILURE 1
#endif
#define CopyField(tag, v) \
if (TIFFGetField(in, tag, &v)) \
TIFFSetField(out, tag, v)
#define CopyFieldFloat(tag, v) \
if (TIFFGetField(in, tag, &v)) \
TIFFSetField(out, tag, (double)(v))
#define CopyField2(tag, v1, v2) \
if (TIFFGetField(in, tag, &v1, &v2)) \
TIFFSetField(out, tag, v1, v2)
#define CopyField3(tag, v1, v2, v3) \
if (TIFFGetField(in, tag, &v1, &v2, &v3)) \
TIFFSetField(out, tag, v1, v2, v3)
#define PATH_LENGTH 8192
#define DEFAULT_MAX_MALLOC (256 * 1024 * 1024)
/* malloc size limit (in bytes)
* disabled when set to 0 */
static tmsize_t maxMalloc = DEFAULT_MAX_MALLOC;
static const char TIFF_SUFFIX[] = ".tif";
static char fname[PATH_LENGTH];
static int tiffcp(TIFF *, TIFF *);
static void newfilename(void);
static int cpStrips(TIFF *, TIFF *);
static int cpTiles(TIFF *, TIFF *);
static void usage(int);
#define MAXFILES 9999999U
/**
* This custom malloc function enforce a maximum allocation size
*/
static void *limitMalloc(tmsize_t s)
{
/* tmsize_t is signed and _TIFFmalloc() converts s to size_t. Therefore
* check for negative s. */
if (maxMalloc && ((s > maxMalloc) || (s < 0)))
{
fprintf(stderr,
"MemoryLimitError: allocation of %" TIFF_SSIZE_FORMAT
" bytes is forbidden. Limit is %" TIFF_SSIZE_FORMAT ".\n",
s, maxMalloc);
fprintf(stderr, " use -M option to change limit.\n");
return NULL;
}
return _TIFFmalloc(s);
}
static void *limitRealloc(void *buf, tmsize_t s)
{
if (maxMalloc && ((s > maxMalloc) || (s < 0)))
{
fprintf(stderr,
"MemoryLimitError: re-allocation of %" TIFF_SSIZE_FORMAT
" bytes is forbidden. Limit is %" TIFF_SSIZE_FORMAT ".\n",
s, maxMalloc);
fprintf(stderr, " use -M option to change limit.\n");
if (buf != NULL)
_TIFFfree(buf);
return NULL;
}
return _TIFFrealloc(buf, s);
}
int main(int argc, char *argv[])
{
TIFF *in, *out;
#if !HAVE_DECL_OPTARG
extern char *optarg;
extern int optind;
#endif
int c;
while ((c = getopt(argc, argv, "M:")) != -1)
{
switch (c)
{
case 'M':
if (!TIFFToolsParseMemoryLimitMiB(optarg, &maxMalloc))
usage(EXIT_FAILURE);
break;
case '?':
usage(EXIT_SUCCESS);
break;
default:
break;
}
}
c = argc - optind;
if (c < 1 || c > 2)
usage(EXIT_FAILURE);
if (c > 1 && strlen(argv[optind + 1]) > sizeof(fname) - 8)
{
fprintf(stderr, "tiffsplit: filename too long\n");
return EXIT_FAILURE;
}
if (c > 1)
{
strncpy(fname, argv[optind + 1], sizeof(fname));
fname[sizeof(fname) - 1] = '\0';
}
TIFFOpenOptions *opts = TIFFOpenOptionsAlloc();
if (opts == NULL)
{
return EXIT_FAILURE;
}
TIFFOpenOptionsSetMaxSingleMemAlloc(opts, maxMalloc);
in = TIFFOpenExt(argv[optind], "r", opts);
if (in == NULL)
{
fprintf(stderr, "tiffsplit: Error: Could not open %s \n", argv[optind]);
TIFFOpenOptionsFree(opts);
usage(EXIT_FAILURE);
}
do
{
size_t path_len;
char *path = NULL;
newfilename();
path_len = strlen(fname) + sizeof(TIFF_SUFFIX);
path = (char *)limitMalloc((tmsize_t)path_len);
if (!path)
{
fprintf(stderr,
"tiffsplit: Error: Can't allocate %" TIFF_SIZE_FORMAT
" bytes for path-variable.\n",
path_len);
TIFFClose(in);
TIFFOpenOptionsFree(opts);
return (EXIT_FAILURE);
}
strncpy(path, fname, path_len);
path[path_len - 1] = '\0';
strncat(path, TIFF_SUFFIX, path_len - strlen(path) - 1);
out = TIFFOpenExt(path, TIFFIsBigEndian(in) ? "wb" : "wl", opts);
if (out == NULL)
{
TIFFClose(in);
fprintf(stderr,
"tiffsplit: Error: Could not open output file %s \n", path);
_TIFFfree(path);
TIFFOpenOptionsFree(opts);
return (EXIT_FAILURE);
}
printf("%s\n", path);
_TIFFfree(path);
if (!tiffcp(in, out))
{
TIFFClose(in);
TIFFClose(out);
TIFFOpenOptionsFree(opts);
fprintf(stderr, "tiffsplit: Error: Could not copy data from input "
"to output.\n");
return (EXIT_FAILURE);
}
TIFFClose(out);
} while (TIFFReadDirectory(in));
TIFFOpenOptionsFree(opts);
TIFFClose(in);
return (EXIT_SUCCESS);
}
static void newfilename(void)
{
static int first = 1;
static unsigned int fnum;
static short defname;
static char *fpnt;
if (first)
{
if (fname[0])
{
fpnt = fname + strlen(fname);
defname = 0;
}
else
{
fname[0] = 'x';
fpnt = fname + 1;
defname = 1;
}
first = 0;
}
if (fnum == MAXFILES)
{
if (!defname || fname[0] == 'z')
{
fprintf(stderr, "tiffsplit: too many files.\n");
exit(EXIT_FAILURE);
}
fname[0]++;
fnum = 0;
}
fnum++;
snprintf(fpnt, 8, "%07u", fnum);
}
static int tiffcp(TIFF *in, TIFF *out)
{
uint16_t bitspersample, samplesperpixel, compression, shortv, *shortav;
uint32_t w, l;
float floatv;
char *stringv;
uint32_t longv;
CopyField(TIFFTAG_SUBFILETYPE, longv);
CopyField(TIFFTAG_TILEWIDTH, w);
CopyField(TIFFTAG_TILELENGTH, l);
CopyField(TIFFTAG_IMAGEWIDTH, w);
CopyField(TIFFTAG_IMAGELENGTH, l);
CopyField(TIFFTAG_BITSPERSAMPLE, bitspersample);
CopyField(TIFFTAG_SAMPLESPERPIXEL, samplesperpixel);
CopyField(TIFFTAG_COMPRESSION, compression);
if (compression == COMPRESSION_JPEG)
{
uint32_t count = 0;
void *table = NULL;
if (TIFFGetField(in, TIFFTAG_JPEGTABLES, &count, &table) && count > 0 &&
table)
{
TIFFSetField(out, TIFFTAG_JPEGTABLES, count, table);
}
}
CopyField(TIFFTAG_PHOTOMETRIC, shortv);
CopyField(TIFFTAG_PREDICTOR, shortv);
CopyField(TIFFTAG_THRESHHOLDING, shortv);
CopyField(TIFFTAG_FILLORDER, shortv);
CopyField(TIFFTAG_ORIENTATION, shortv);
CopyField(TIFFTAG_MINSAMPLEVALUE, shortv);
CopyField(TIFFTAG_MAXSAMPLEVALUE, shortv);
CopyFieldFloat(TIFFTAG_XRESOLUTION, floatv);
CopyFieldFloat(TIFFTAG_YRESOLUTION, floatv);
CopyField(TIFFTAG_GROUP3OPTIONS, longv);
CopyField(TIFFTAG_GROUP4OPTIONS, longv);
CopyField(TIFFTAG_RESOLUTIONUNIT, shortv);
CopyField(TIFFTAG_PLANARCONFIG, shortv);
CopyField(TIFFTAG_ROWSPERSTRIP, longv);
CopyFieldFloat(TIFFTAG_XPOSITION, floatv);
CopyFieldFloat(TIFFTAG_YPOSITION, floatv);
CopyField(TIFFTAG_IMAGEDEPTH, longv);
CopyField(TIFFTAG_TILEDEPTH, longv);
CopyField(TIFFTAG_SAMPLEFORMAT, shortv);
CopyField2(TIFFTAG_EXTRASAMPLES, shortv, shortav);
{
uint16_t *red, *green, *blue;
CopyField3(TIFFTAG_COLORMAP, red, green, blue);
}
{
uint16_t shortv2;
CopyField2(TIFFTAG_PAGENUMBER, shortv, shortv2);
}
CopyField(TIFFTAG_ARTIST, stringv);
CopyField(TIFFTAG_IMAGEDESCRIPTION, stringv);
CopyField(TIFFTAG_MAKE, stringv);
CopyField(TIFFTAG_MODEL, stringv);
CopyField(TIFFTAG_SOFTWARE, stringv);
CopyField(TIFFTAG_DATETIME, stringv);
CopyField(TIFFTAG_HOSTCOMPUTER, stringv);
CopyField(TIFFTAG_PAGENAME, stringv);
CopyField(TIFFTAG_DOCUMENTNAME, stringv);
CopyField(TIFFTAG_BADFAXLINES, longv);
CopyField(TIFFTAG_CLEANFAXDATA, longv);
CopyField(TIFFTAG_CONSECUTIVEBADFAXLINES, longv);
CopyField(TIFFTAG_FAXRECVPARAMS, longv);
CopyField(TIFFTAG_FAXRECVTIME, longv);
CopyField(TIFFTAG_FAXSUBADDRESS, stringv);
CopyField(TIFFTAG_FAXDCS, stringv);
if (TIFFIsTiled(in))
return (cpTiles(in, out));
else
return (cpStrips(in, out));
}
static int cpStrips(TIFF *in, TIFF *out)
{
tmsize_t bufsize = TIFFStripSize(in);
unsigned char *buf = (unsigned char *)limitMalloc(bufsize);
if (buf)
{
tstrip_t s, ns = TIFFNumberOfStrips(in);
uint64_t *bytecounts;
if (!TIFFGetField(in, TIFFTAG_STRIPBYTECOUNTS, &bytecounts))
{
fprintf(stderr, "tiffsplit: strip byte counts are missing\n");
_TIFFfree(buf);
return (0);
}
for (s = 0; s < ns; s++)
{
if (bytecounts[s] > (uint64_t)bufsize)
{
buf =
(unsigned char *)limitRealloc(buf, (tmsize_t)bytecounts[s]);
if (!buf)
{
fprintf(stderr,
"tiffsplit: Error: Can't re-allocate "
"%" TIFF_SSIZE_FORMAT " bytes for strip-size.\n",
(tmsize_t)bytecounts[s]);
return (0);
}
bufsize = (tmsize_t)bytecounts[s];
}
if (TIFFReadRawStrip(in, s, buf, (tmsize_t)bytecounts[s]) < 0 ||
TIFFWriteRawStrip(out, s, buf, (tmsize_t)bytecounts[s]) < 0)
{
_TIFFfree(buf);
return (0);
}
}
_TIFFfree(buf);
return (1);
}
else
{
fprintf(stderr,
"tiffsplit: Error: Can't allocate %" TIFF_SSIZE_FORMAT
" bytes for strip-size.\n",
bufsize);
}
return (0);
}
static int cpTiles(TIFF *in, TIFF *out)
{
tmsize_t bufsize = TIFFTileSize(in);
unsigned char *buf = (unsigned char *)limitMalloc(bufsize);
if (buf)
{
ttile_t t, nt = TIFFNumberOfTiles(in);
uint64_t *bytecounts;
if (!TIFFGetField(in, TIFFTAG_TILEBYTECOUNTS, &bytecounts))
{
fprintf(stderr, "tiffsplit: tile byte counts are missing\n");
_TIFFfree(buf);
return (0);
}
for (t = 0; t < nt; t++)
{
if (bytecounts[t] > (uint64_t)bufsize)
{
buf =
(unsigned char *)limitRealloc(buf, (tmsize_t)bytecounts[t]);
if (!buf)
{
fprintf(stderr,
"tiffsplit: Error: Can't re-allocate "
"%" TIFF_SSIZE_FORMAT " bytes for tile-size.\n",
(tmsize_t)bytecounts[t]);
return (0);
}
bufsize = (tmsize_t)bytecounts[t];
}
if (TIFFReadRawTile(in, t, buf, (tmsize_t)bytecounts[t]) < 0 ||
TIFFWriteRawTile(out, t, buf, (tmsize_t)bytecounts[t]) < 0)
{
_TIFFfree(buf);
return (0);
}
}
_TIFFfree(buf);
return (1);
}
else
{
fprintf(stderr,
"tiffsplit: Error: Can't allocate %" TIFF_SSIZE_FORMAT
" bytes for tile-size.\n",
bufsize);
}
return (0);
}
static void usage(int code)
{
FILE *out = (code == EXIT_SUCCESS) ? stdout : stderr;
fprintf(out, "\n\n%s\n\n", TIFFGetVersion());
fprintf(out, "Split a multi-image TIFF into single-image TIFF files\n\n");
fprintf(out, "usage: tiffsplit [option] input.tif [prefix]\n");
fprintf(out, "where option is:\n");
fprintf(out, " -M size set the memory allocation limit in MiB. 0 to "
"disable limit.\n");
exit(code);
}
View Git Blame Copy Permalink