]> granicus.if.org Git - imagemagick/blob - coders/pdb.c
https://github.com/ImageMagick/ImageMagick/issues/876
[imagemagick] / coders / pdb.c
1 /*
2 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
3 %                                                                             %
4 %                                                                             %
5 %                                                                             %
6 %                            PPPP   DDDD   BBBB                               %
7 %                            P   P  D   D  B   B                              %
8 %                            PPPP   D   D  BBBB                               %
9 %                            P      D   D  B   B                              %
10 %                            P      DDDD   BBBB                               %
11 %                                                                             %
12 %                                                                             %
13 %               Read/Write Palm Database ImageViewer Image Format             %
14 %                                                                             %
15 %                              Software Design                                %
16 %                                   Cristy                                    %
17 %                                 July 1992                                   %
18 %                                                                             %
19 %                                                                             %
20 %  Copyright 1999-2017 ImageMagick Studio LLC, a non-profit organization      %
21 %  dedicated to making software imaging solutions freely available.           %
22 %                                                                             %
23 %  You may not use this file except in compliance with the License.  You may  %
24 %  obtain a copy of the License at                                            %
25 %                                                                             %
26 %    https://www.imagemagick.org/script/license.php                           %
27 %                                                                             %
28 %  Unless required by applicable law or agreed to in writing, software        %
29 %  distributed under the License is distributed on an "AS IS" BASIS,          %
30 %  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   %
31 %  See the License for the specific language governing permissions and        %
32 %  limitations under the License.                                             %
33 %                                                                             %
34 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
35 %
36 %
37 %   20071202 TS * rewrote RLE decoder - old version could cause buffer overflows
38 %               * failure of RLE decoding now thows error RLEDecoderError
39 %               * fixed bug in RLE decoding - now all rows are decoded, not just
40 %     the first one
41 %   * fixed bug in reader - record offsets now handled correctly
42 %   * fixed bug in reader - only bits 0..2 indicate compression type
43 %               * in writer: now using image color count instead of depth
44 */
45 \f
46 /*
47   Include declarations.
48 */
49 #include "MagickCore/studio.h"
50 #include "MagickCore/attribute.h"
51 #include "MagickCore/blob.h"
52 #include "MagickCore/blob-private.h"
53 #include "MagickCore/cache.h"
54 #include "MagickCore/colormap-private.h"
55 #include "MagickCore/color-private.h"
56 #include "MagickCore/colormap.h"
57 #include "MagickCore/colorspace.h"
58 #include "MagickCore/colorspace-private.h"
59 #include "MagickCore/constitute.h"
60 #include "MagickCore/exception.h"
61 #include "MagickCore/exception-private.h"
62 #include "MagickCore/image.h"
63 #include "MagickCore/image-private.h"
64 #include "MagickCore/list.h"
65 #include "MagickCore/magick.h"
66 #include "MagickCore/memory_.h"
67 #include "MagickCore/monitor.h"
68 #include "MagickCore/monitor-private.h"
69 #include "MagickCore/pixel-accessor.h"
70 #include "MagickCore/property.h"
71 #include "MagickCore/quantum-private.h"
72 #include "MagickCore/quantum-private.h"
73 #include "MagickCore/static.h"
74 #include "MagickCore/string_.h"
75 #include "MagickCore/module.h"
76 \f
77 /*
78   Typedef declarations.
79 */
80 typedef struct _PDBInfo
81 {
82   char
83     name[32];
84
85   short int
86     attributes,
87     version;
88
89   size_t
90     create_time,
91     modify_time,
92     archive_time,
93     modify_number,
94     application_info,
95     sort_info;
96
97   char
98     type[4],  /* database type identifier "vIMG" */
99     id[4];    /* database creator identifier "View" */
100
101   size_t
102     seed,
103     next_record;
104
105   short int
106     number_records;
107 } PDBInfo;
108
109 typedef struct _PDBImage
110 {
111   char
112     name[32],
113     version;
114
115   size_t
116     reserved_1,
117     note;
118
119   short int
120     x_last,
121     y_last;
122
123   size_t
124     reserved_2;
125
126   short int
127     width,
128     height;
129
130   unsigned char
131     type;
132
133   unsigned short
134     x_anchor,
135     y_anchor;
136 } PDBImage;
137 /*
138   Forward declarations.
139 */
140 static MagickBooleanType
141   WritePDBImage(const ImageInfo *,Image *,ExceptionInfo *);
142 \f
143 /*
144 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
145 %                                                                             %
146 %                                                                             %
147 %                                                                             %
148 %   D e c o d e I m a g e                                                     %
149 %                                                                             %
150 %                                                                             %
151 %                                                                             %
152 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
153 %
154 %  DecodeImage unpacks the packed image pixels into runlength-encoded
155 %  pixel packets.
156 %
157 %  The format of the DecodeImage method is:
158 %
159 %      MagickBooleanType DecodeImage(Image *image,unsigned char *pixels,
160 %        const size_t length)
161 %
162 %  A description of each parameter follows:
163 %
164 %    o image: the address of a structure of type Image.
165 %
166 %    o pixels:  The address of a byte (8 bits) array of pixel data created by
167 %      the decoding process.
168 %
169 %    o length:  Number of bytes to read into buffer 'pixels'.
170 %
171 */
172 static MagickBooleanType DecodeImage(Image *image, unsigned char *pixels,
173   const size_t length)
174 {
175 #define RLE_MODE_NONE -1
176 #define RLE_MODE_COPY  0
177 #define RLE_MODE_RUN   1
178
179   int           data = 0, count = 0;
180   unsigned char *p;
181   int           mode = RLE_MODE_NONE;
182
183   for (p = pixels; p < pixels + length; p++) {
184     if (0 == count) {
185       data = ReadBlobByte( image );
186       if (-1 == data) return MagickFalse;
187       if (data > 128) {
188         mode  = RLE_MODE_RUN;
189         count = data - 128 + 1;
190         data  = ReadBlobByte( image );
191         if (-1 == data) return MagickFalse;
192       } else {
193         mode  = RLE_MODE_COPY;
194         count = data + 1;
195       }
196     }
197
198     if (RLE_MODE_COPY == mode) {
199       data = ReadBlobByte( image );
200       if (-1 == data) return MagickFalse;
201     }
202     *p = (unsigned char)data;
203     --count;
204   }
205   return MagickTrue;
206 }
207 \f
208 /*
209 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
210 %                                                                             %
211 %                                                                             %
212 %                                                                             %
213 %   I s P D B                                                                 %
214 %                                                                             %
215 %                                                                             %
216 %                                                                             %
217 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
218 %
219 %  IsPDB() returns MagickTrue if the image format type, identified by the
220 %  magick string, is PDB.
221 %
222 %  The format of the ReadPDBImage method is:
223 %
224 %      MagickBooleanType IsPDB(const unsigned char *magick,const size_t length)
225 %
226 %  A description of each parameter follows:
227 %
228 %    o magick: compare image format pattern against these bytes.
229 %
230 %    o length: Specifies the length of the magick string.
231 %
232 */
233 static MagickBooleanType IsPDB(const unsigned char *magick,const size_t length)
234 {
235   if (length < 68)
236     return(MagickFalse);
237   if (memcmp(magick+60,"vIMGView",8) == 0)
238     return(MagickTrue);
239   return(MagickFalse);
240 }
241 \f
242 /*
243 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
244 %                                                                             %
245 %                                                                             %
246 %                                                                             %
247 %   R e a d P D B I m a g e                                                   %
248 %                                                                             %
249 %                                                                             %
250 %                                                                             %
251 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
252 %
253 %  ReadPDBImage() reads an Pilot image file and returns it.  It
254 %  allocates the memory necessary for the new Image structure and returns a
255 %  pointer to the new image.
256 %
257 %  The format of the ReadPDBImage method is:
258 %
259 %      Image *ReadPDBImage(const ImageInfo *image_info,ExceptionInfo *exception)
260 %
261 %  A description of each parameter follows:
262 %
263 %    o image_info: the image info.
264 %
265 %    o exception: return any errors or warnings in this structure.
266 %
267 */
268 static Image *ReadPDBImage(const ImageInfo *image_info,ExceptionInfo *exception)
269 {
270   unsigned char
271     attributes,
272     tag[3];
273
274   Image
275     *image;
276
277   MagickBooleanType
278     status;
279
280   PDBImage
281     pdb_image;
282
283   PDBInfo
284     pdb_info;
285
286   Quantum
287     index;
288
289   register ssize_t
290     x;
291
292   register Quantum
293     *q;
294
295   register unsigned char
296     *p;
297
298   size_t
299     bits_per_pixel,
300     num_pad_bytes,
301     one,
302     packets;
303
304   ssize_t
305     count,
306     img_offset,
307     comment_offset = 0,
308     y;
309
310   unsigned char
311     *pixels;
312
313   /*
314     Open image file.
315   */
316   assert(image_info != (const ImageInfo *) NULL);
317   assert(image_info->signature == MagickCoreSignature);
318   if (image_info->debug != MagickFalse)
319     (void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",
320       image_info->filename);
321   assert(exception != (ExceptionInfo *) NULL);
322   assert(exception->signature == MagickCoreSignature);
323   image=AcquireImage(image_info,exception);
324   status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
325   if (status == MagickFalse)
326     {
327       image=DestroyImageList(image);
328       return((Image *) NULL);
329     }
330   /*
331     Determine if this a PDB image file.
332   */
333   count=ReadBlob(image,sizeof(pdb_info.name),(unsigned char *) pdb_info.name);
334   if (count != sizeof(pdb_info.name))
335     ThrowReaderException(CorruptImageError,"ImproperImageHeader");
336   pdb_info.attributes=(short) ReadBlobMSBShort(image);
337   pdb_info.version=(short) ReadBlobMSBShort(image);
338   pdb_info.create_time=ReadBlobMSBLong(image);
339   pdb_info.modify_time=ReadBlobMSBLong(image);
340   pdb_info.archive_time=ReadBlobMSBLong(image);
341   pdb_info.modify_number=ReadBlobMSBLong(image);
342   pdb_info.application_info=ReadBlobMSBLong(image);
343   pdb_info.sort_info=ReadBlobMSBLong(image);
344   (void) ReadBlob(image,4,(unsigned char *) pdb_info.type);
345   (void) ReadBlob(image,4,(unsigned char *) pdb_info.id);
346   pdb_info.seed=ReadBlobMSBLong(image);
347   pdb_info.next_record=ReadBlobMSBLong(image);
348   pdb_info.number_records=(short) ReadBlobMSBShort(image);
349   if ((memcmp(pdb_info.type,"vIMG",4) != 0) ||
350       (memcmp(pdb_info.id,"View",4) != 0))
351     ThrowReaderException(CorruptImageError,"ImproperImageHeader");
352   if (pdb_info.next_record != 0)
353     ThrowReaderException(CoderError,"MultipleRecordListNotSupported");
354   /*
355     Read record header.
356   */
357   img_offset=(ssize_t) ReadBlobMSBSignedLong(image);
358   attributes=(unsigned char) ReadBlobByte(image);
359   (void) attributes;
360   count=ReadBlob(image,3,(unsigned char *) tag);
361   if (count != 3  ||  memcmp(tag,"\x6f\x80\x00",3) != 0)
362     ThrowReaderException(CorruptImageError,"CorruptImage");
363   if (pdb_info.number_records > 1)
364     {
365       comment_offset=(ssize_t) ReadBlobMSBSignedLong(image);
366       attributes=(unsigned char) ReadBlobByte(image);
367       count=ReadBlob(image,3,(unsigned char *) tag);
368       if (count != 3  ||  memcmp(tag,"\x6f\x80\x01",3) != 0)
369         ThrowReaderException(CorruptImageError,"CorruptImage");
370     }
371   num_pad_bytes = (size_t) (img_offset - TellBlob( image ));
372   while (num_pad_bytes-- != 0)
373   {
374     int
375       c;
376
377     c=ReadBlobByte(image);
378     if (c == EOF)
379       break;
380   }
381   /*
382     Read image header.
383   */
384   count=ReadBlob(image,sizeof(pdb_image.name),(unsigned char *) pdb_image.name);
385   if (count != sizeof(pdb_image.name))
386     ThrowReaderException(CorruptImageError,"ImproperImageHeader");
387   pdb_image.version=ReadBlobByte(image);
388   pdb_image.type=(unsigned char) (ReadBlobByte(image));
389   pdb_image.reserved_1=ReadBlobMSBLong(image);
390   pdb_image.note=ReadBlobMSBLong(image);
391   pdb_image.x_last=(short) ReadBlobMSBShort(image);
392   pdb_image.y_last=(short) ReadBlobMSBShort(image);
393   pdb_image.reserved_2=ReadBlobMSBLong(image);
394   pdb_image.x_anchor=ReadBlobMSBShort(image);
395   pdb_image.y_anchor=ReadBlobMSBShort(image);
396   pdb_image.width=(short) ReadBlobMSBShort(image);
397   pdb_image.height=(short) ReadBlobMSBShort(image);
398   /*
399     Initialize image structure.
400   */
401   image->columns=(size_t) pdb_image.width;
402   image->rows=(size_t) pdb_image.height;
403   image->depth=8;
404   image->storage_class=PseudoClass;
405   bits_per_pixel=pdb_image.type == 0 ? 2UL : pdb_image.type == 2 ? 4UL : 1UL;
406   one=1;
407   if (AcquireImageColormap(image,one << bits_per_pixel,exception) == MagickFalse)
408     ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
409   if (image_info->ping != MagickFalse)
410     {
411       (void) CloseBlob(image);
412       return(GetFirstImageInList(image));
413     }
414   status=SetImageExtent(image,image->columns,image->rows,exception);
415   if (status == MagickFalse)
416     return(DestroyImageList(image));
417   packets=(bits_per_pixel*image->columns+7)/8;
418   pixels=(unsigned char *) AcquireQuantumMemory(packets+257UL,image->rows*
419     sizeof(*pixels));
420   if (pixels == (unsigned char *) NULL)
421     ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
422   switch (pdb_image.version & 0x07) 
423   {
424     case 0:
425     {
426       image->compression=NoCompression;
427       count=(ssize_t) ReadBlob(image,packets*image->rows,pixels);
428       break;
429     }
430     case 1:
431     {
432       image->compression=RLECompression;
433       if (!DecodeImage(image,pixels,packets*image->rows))
434         {
435           pixels=(unsigned char *) RelinquishMagickMemory(pixels);
436           ThrowReaderException(CorruptImageError,"RLEDecoderError");
437         }
438       break;
439     }
440     default:
441     {
442       pixels=(unsigned char *) RelinquishMagickMemory(pixels);
443       ThrowReaderException(CorruptImageError,
444         "UnrecognizedImageCompressionType");
445     }
446   }
447   p=pixels;
448   switch (bits_per_pixel)
449   {
450     case 1:
451     {
452       int
453         bit;
454
455       /*
456         Read 1-bit PDB image.
457       */
458       for (y=0; y < (ssize_t) image->rows; y++)
459       {
460         q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
461         if (q == (Quantum *) NULL)
462           break;
463         for (x=0; x < ((ssize_t) image->columns-7); x+=8)
464         {
465           for (bit=0; bit < 8; bit++)
466           {
467             index=(Quantum) (*p & (0x80 >> bit) ? 0x00 : 0x01);
468             SetPixelIndex(image,index,q);
469             q+=GetPixelChannels(image);
470           }
471           p++;
472         }
473         if (SyncAuthenticPixels(image,exception) == MagickFalse)
474           break;
475         status=SetImageProgress(image,LoadImageTag,(MagickOffsetType) y,
476           image->rows);
477         if (status == MagickFalse)
478           break;
479       }
480       (void) SyncImage(image,exception);
481       break;
482     }
483     case 2:
484     {
485       /*
486         Read 2-bit PDB image.
487       */
488       for (y=0; y < (ssize_t) image->rows; y++)
489       {
490         q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
491         if (q == (Quantum *) NULL)
492           break;
493         for (x=0; x < (ssize_t) image->columns-3; x+=4)
494         {
495           index=ConstrainColormapIndex(image,3UL-((*p >> 6) & 0x03),exception);
496           SetPixelIndex(image,index,q);
497           q+=GetPixelChannels(image);
498           index=ConstrainColormapIndex(image,3UL-((*p >> 4) & 0x03),exception);
499           SetPixelIndex(image,index,q);
500           q+=GetPixelChannels(image);
501           index=ConstrainColormapIndex(image,3UL-((*p >> 2) & 0x03),exception);
502           SetPixelIndex(image,index,q);
503           q+=GetPixelChannels(image);
504           index=ConstrainColormapIndex(image,3UL-((*p) & 0x03),exception);
505           SetPixelIndex(image,index,q);
506           p++;
507           q+=GetPixelChannels(image);
508         }
509         if (SyncAuthenticPixels(image,exception) == MagickFalse)
510           break;
511         status=SetImageProgress(image,LoadImageTag,(MagickOffsetType) y,
512           image->rows);
513         if (status == MagickFalse)
514           break;
515       }
516       (void) SyncImage(image,exception);
517       break;
518     }
519     case 4:
520     {
521       /*
522         Read 4-bit PDB image.
523       */
524       for (y=0; y < (ssize_t) image->rows; y++)
525       {
526         q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
527         if (q == (Quantum *) NULL)
528           break;
529         for (x=0; x < (ssize_t) image->columns-1; x+=2)
530         {
531           index=ConstrainColormapIndex(image,15UL-((*p >> 4) & 0x0f),exception);
532           SetPixelIndex(image,index,q);
533           q+=GetPixelChannels(image);
534           index=ConstrainColormapIndex(image,15UL-((*p) & 0x0f),exception);
535           SetPixelIndex(image,index,q);
536           p++;
537           q+=GetPixelChannels(image);
538         }
539         if (SyncAuthenticPixels(image,exception) == MagickFalse)
540           break;
541         status=SetImageProgress(image,LoadImageTag,(MagickOffsetType) y,
542           image->rows);
543         if (status == MagickFalse)
544           break;
545       }
546       (void) SyncImage(image,exception);
547       break;
548     }
549     default:
550     {
551       pixels=(unsigned char *) RelinquishMagickMemory(pixels);
552       ThrowReaderException(CorruptImageError,"ImproperImageHeader");
553     }
554   }
555   pixels=(unsigned char *) RelinquishMagickMemory(pixels);
556   if (EOFBlob(image) != MagickFalse)
557     ThrowFileException(exception,CorruptImageError,"UnexpectedEndOfFile",
558       image->filename);
559   if (pdb_info.number_records > 1)
560     {
561       char
562         *comment;
563
564       int
565         c;
566
567       register char
568         *p;
569
570       size_t
571         length;
572
573       num_pad_bytes = (size_t) (comment_offset - TellBlob( image ));
574       while (num_pad_bytes-- != 0)
575       {
576         int
577           c;
578
579         c=ReadBlobByte(image);
580         if (c == EOF)
581           break;
582       }
583
584       /*
585         Read comment.
586       */
587       c=ReadBlobByte(image);
588       length=MagickPathExtent;
589       comment=AcquireString((char *) NULL);
590       for (p=comment; c != EOF; p++)
591       {
592         if ((size_t) (p-comment+MagickPathExtent) >= length)
593           {
594             *p='\0';
595             length<<=1;
596             length+=MagickPathExtent;
597             comment=(char *) ResizeQuantumMemory(comment,length+MagickPathExtent,
598               sizeof(*comment));
599             if (comment == (char *) NULL)
600               break;
601             p=comment+strlen(comment);
602           }
603         *p=c;
604         c=ReadBlobByte(image);
605       }
606       *p='\0';
607       if (comment == (char *) NULL)
608         ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
609       (void) SetImageProperty(image,"comment",comment,exception);
610       comment=DestroyString(comment);
611     }
612   (void) CloseBlob(image);
613   return(GetFirstImageInList(image));
614 }
615 \f
616 /*
617 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
618 %                                                                             %
619 %                                                                             %
620 %                                                                             %
621 %   R e g i s t e r P D B I m a g e                                           %
622 %                                                                             %
623 %                                                                             %
624 %                                                                             %
625 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
626 %
627 %  RegisterPDBImage() adds properties for the PDB image format to
628 %  the list of supported formats.  The properties include the image format
629 %  tag, a method to read and/or write the format, whether the format
630 %  supports the saving of more than one frame to the same file or blob,
631 %  whether the format supports native in-memory I/O, and a brief
632 %  description of the format.
633 %
634 %  The format of the RegisterPDBImage method is:
635 %
636 %      size_t RegisterPDBImage(void)
637 %
638 */
639 ModuleExport size_t RegisterPDBImage(void)
640 {
641   MagickInfo
642     *entry;
643
644   entry=AcquireMagickInfo("PDB","PDB","Palm Database ImageViewer Format");
645   entry->decoder=(DecodeImageHandler *) ReadPDBImage;
646   entry->encoder=(EncodeImageHandler *) WritePDBImage;
647   entry->magick=(IsImageFormatHandler *) IsPDB;
648   (void) RegisterMagickInfo(entry);
649   return(MagickImageCoderSignature);
650 }
651 \f
652 /*
653 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
654 %                                                                             %
655 %                                                                             %
656 %                                                                             %
657 %   U n r e g i s t e r P D B I m a g e                                       %
658 %                                                                             %
659 %                                                                             %
660 %                                                                             %
661 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
662 %
663 %  UnregisterPDBImage() removes format registrations made by the
664 %  PDB module from the list of supported formats.
665 %
666 %  The format of the UnregisterPDBImage method is:
667 %
668 %      UnregisterPDBImage(void)
669 %
670 */
671 ModuleExport void UnregisterPDBImage(void)
672 {
673   (void) UnregisterMagickInfo("PDB");
674 }
675 \f
676 /*
677 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
678 %                                                                             %
679 %                                                                             %
680 %                                                                             %
681 %   W r i t e P D B I m a g e                                                 %
682 %                                                                             %
683 %                                                                             %
684 %                                                                             %
685 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
686 %
687 %  WritePDBImage() writes an image
688 %
689 %  The format of the WritePDBImage method is:
690 %
691 %      MagickBooleanType WritePDBImage(const ImageInfo *image_info,
692 %        Image *image,ExceptionInfo *exception)
693 %
694 %  A description of each parameter follows.
695 %
696 %    o image_info: the image info.
697 %
698 %    o image:  The image.
699 %
700 %    o exception: return any errors or warnings in this structure.
701 %
702 */
703
704 static unsigned char *EncodeRLE(unsigned char *destination,
705   unsigned char *source,size_t literal,size_t repeat)
706 {
707   if (literal > 0)
708     *destination++=(unsigned char) (literal-1);
709   (void) CopyMagickMemory(destination,source,literal);
710   destination+=literal;
711   if (repeat > 0)
712     {
713       *destination++=(unsigned char) (0x80 | (repeat-1));
714       *destination++=source[literal];
715     }
716   return(destination);
717 }
718
719 static MagickBooleanType WritePDBImage(const ImageInfo *image_info,Image *image,
720   ExceptionInfo *exception)
721 {
722   const char
723     *comment;
724
725   int
726     bits;
727
728   MagickBooleanType
729     status;
730
731   PDBImage
732     pdb_image;
733
734   PDBInfo
735     pdb_info;
736
737   QuantumInfo
738     *quantum_info;
739
740   register const Quantum
741     *p;
742
743   register ssize_t
744     x;
745
746   register unsigned char
747     *q;
748
749   size_t
750     bits_per_pixel,
751     literal,
752     packets,
753     packet_size,
754     repeat;
755
756   ssize_t
757     y;
758
759   unsigned char
760     *buffer,
761     *runlength,
762     *scanline;
763
764   /*
765     Open output image file.
766   */
767   assert(image_info != (const ImageInfo *) NULL);
768   assert(image_info->signature == MagickCoreSignature);
769   assert(image != (Image *) NULL);
770   assert(image->signature == MagickCoreSignature);
771   if (image->debug != MagickFalse)
772     (void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",image->filename);
773   assert(exception != (ExceptionInfo *) NULL);
774   assert(exception->signature == MagickCoreSignature);
775   status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
776   if (status == MagickFalse)
777     return(status);
778   (void) TransformImageColorspace(image,sRGBColorspace,exception);
779   if (SetImageMonochrome(image,exception) != MagickFalse) {
780     bits_per_pixel=1;
781   } else if (image->colors <= 4) {
782     bits_per_pixel=2;
783   } else if (image->colors <= 8) {
784     bits_per_pixel=3;
785   } else {
786     bits_per_pixel=4;
787   }
788   (void) ResetMagickMemory(&pdb_info,0,sizeof(pdb_info));
789   (void) CopyMagickString(pdb_info.name,image_info->filename,
790     sizeof(pdb_info.name));
791   pdb_info.attributes=0;
792   pdb_info.version=0;
793   pdb_info.create_time=time(NULL);
794   pdb_info.modify_time=pdb_info.create_time;
795   pdb_info.archive_time=0;
796   pdb_info.modify_number=0;
797   pdb_info.application_info=0;
798   pdb_info.sort_info=0;
799   (void) CopyMagickMemory(pdb_info.type,"vIMG",4);
800   (void) CopyMagickMemory(pdb_info.id,"View",4);
801   pdb_info.seed=0;
802   pdb_info.next_record=0;
803   comment=GetImageProperty(image,"comment",exception);
804   pdb_info.number_records=(comment == (const char *) NULL ? 1 : 2);
805   (void) WriteBlob(image,sizeof(pdb_info.name),(unsigned char *) pdb_info.name);
806   (void) WriteBlobMSBShort(image,(unsigned short) pdb_info.attributes);
807   (void) WriteBlobMSBShort(image,(unsigned short) pdb_info.version);
808   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.create_time);
809   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.modify_time);
810   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.archive_time);
811   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.modify_number);
812   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.application_info);
813   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.sort_info);
814   (void) WriteBlob(image,4,(unsigned char *) pdb_info.type);
815   (void) WriteBlob(image,4,(unsigned char *) pdb_info.id);
816   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.seed);
817   (void) WriteBlobMSBLong(image,(unsigned int) pdb_info.next_record);
818   (void) WriteBlobMSBShort(image,(unsigned short) pdb_info.number_records);
819   (void) CopyMagickString(pdb_image.name,pdb_info.name,sizeof(pdb_image.name));
820   pdb_image.version=1;  /* RLE Compressed */
821   switch (bits_per_pixel)
822   {
823     case 1: pdb_image.type=(unsigned char) 0xff; break;  /* monochrome */
824     case 2: pdb_image.type=(unsigned char) 0x00; break;  /* 2 bit gray */
825     default: pdb_image.type=(unsigned char) 0x02;  /* 4 bit gray */
826   }
827   pdb_image.reserved_1=0;
828   pdb_image.note=0;
829   pdb_image.x_last=0;
830   pdb_image.y_last=0;
831   pdb_image.reserved_2=0;
832   pdb_image.x_anchor=(unsigned short) 0xffff;
833   pdb_image.y_anchor=(unsigned short) 0xffff;
834   pdb_image.width=(short) image->columns;
835   if (image->columns % 16)
836     pdb_image.width=(short) (16*(image->columns/16+1));
837   pdb_image.height=(short) image->rows;
838   packets=((bits_per_pixel*image->columns+7)/8);
839   packet_size=(size_t) (image->depth > 8 ? 2 : 1);
840   runlength=(unsigned char *) AcquireQuantumMemory(9UL*packets,
841     image->rows*sizeof(*runlength));
842   buffer=(unsigned char *) AcquireQuantumMemory(512,sizeof(*buffer));
843   scanline=(unsigned char *) AcquireQuantumMemory(image->columns,packet_size*
844     sizeof(*scanline));
845   if ((runlength == (unsigned char *) NULL) ||
846       (buffer == (unsigned char *) NULL) ||
847       (scanline == (unsigned char *) NULL))
848     {
849       if (runlength != (unsigned char *) NULL)
850         runlength=(unsigned char *) RelinquishMagickMemory(runlength);
851       if (buffer != (unsigned char *) NULL)
852         buffer=(unsigned char *) RelinquishMagickMemory(buffer);
853       if (scanline != (unsigned char *) NULL)
854         scanline=(unsigned char *) RelinquishMagickMemory(scanline);
855       ThrowWriterException(ResourceLimitError,"MemoryAllocationFailed");
856     }
857   if (IssRGBCompatibleColorspace(image->colorspace) == MagickFalse)
858     (void) TransformImageColorspace(image,sRGBColorspace,exception);
859   /*
860     Convert to GRAY raster scanline.
861   */
862   quantum_info=AcquireQuantumInfo(image_info,image);
863   if (quantum_info == (QuantumInfo *) NULL)
864     ThrowWriterException(ResourceLimitError,"MemoryAllocationFailed");
865   status=SetQuantumDepth(image,quantum_info,image->depth > 8 ? 16 : 8);
866   bits=8/(int) bits_per_pixel-1;  /* start at most significant bits */
867   literal=0;
868   repeat=0;
869   q=runlength;
870   buffer[0]=0x00;
871   for (y=0; y < (ssize_t) image->rows; y++)
872   {
873     p=GetVirtualPixels(image,0,y,image->columns,1,exception);
874     if (p == (const Quantum *) NULL)
875       break;
876     (void) ExportQuantumPixels(image,(CacheView *) NULL,quantum_info,
877       GrayQuantum,scanline,exception);
878     for (x=0; x < (ssize_t) pdb_image.width; x++)
879     {
880       if (x < (ssize_t) image->columns)
881         buffer[literal+repeat]|=(0xff-scanline[x*packet_size]) >>
882           (8-bits_per_pixel) << bits*bits_per_pixel;
883       bits--;
884       if (bits < 0)
885         {
886           if (((literal+repeat) > 0) &&
887               (buffer[literal+repeat] == buffer[literal+repeat-1]))
888             {
889               if (repeat == 0)
890                 {
891                   literal--;
892                   repeat++;
893                 }
894               repeat++;
895               if (0x7f < repeat)
896                 {
897                   q=EncodeRLE(q,buffer,literal,repeat);
898                   literal=0;
899                   repeat=0;
900                 }
901             }
902           else
903             {
904               if (repeat >= 2)
905                 literal+=repeat;
906               else
907                 {
908                   q=EncodeRLE(q,buffer,literal,repeat);
909                   buffer[0]=buffer[literal+repeat];
910                   literal=0;
911                 }
912               literal++;
913               repeat=0;
914               if (0x7f < literal)
915                 {
916                   q=EncodeRLE(q,buffer,(literal < 0x80 ? literal : 0x80),0);
917                   (void) CopyMagickMemory(buffer,buffer+literal+repeat,0x80);
918                   literal-=0x80;
919                 }
920             }
921         bits=8/(int) bits_per_pixel-1;
922         buffer[literal+repeat]=0x00;
923       }
924     }
925     status=SetImageProgress(image,SaveImageTag,(MagickOffsetType) y,
926       image->rows);
927     if (status == MagickFalse)
928       break;
929   }
930   q=EncodeRLE(q,buffer,literal,repeat);
931   scanline=(unsigned char *) RelinquishMagickMemory(scanline);
932   buffer=(unsigned char *) RelinquishMagickMemory(buffer);
933   quantum_info=DestroyQuantumInfo(quantum_info);
934   /*
935     Write the Image record header.
936   */
937   (void) WriteBlobMSBLong(image,(unsigned int)
938     (TellBlob(image)+8*pdb_info.number_records));
939   (void) WriteBlobByte(image,0x40);
940   (void) WriteBlobByte(image,0x6f);
941   (void) WriteBlobByte(image,0x80);
942   (void) WriteBlobByte(image,0);
943   if (pdb_info.number_records > 1)
944     {
945       /*
946         Write the comment record header.
947       */
948       (void) WriteBlobMSBLong(image,(unsigned int) (TellBlob(image)+8+58+q-
949         runlength));
950       (void) WriteBlobByte(image,0x40);
951       (void) WriteBlobByte(image,0x6f);
952       (void) WriteBlobByte(image,0x80);
953       (void) WriteBlobByte(image,1);
954     }
955   /*
956     Write the Image data.
957   */
958   (void) WriteBlob(image,sizeof(pdb_image.name),(unsigned char *)
959     pdb_image.name);
960   (void) WriteBlobByte(image,(unsigned char) pdb_image.version);
961   (void) WriteBlobByte(image,pdb_image.type);
962   (void) WriteBlobMSBLong(image,(unsigned int) pdb_image.reserved_1);
963   (void) WriteBlobMSBLong(image,(unsigned int) pdb_image.note);
964   (void) WriteBlobMSBShort(image,(unsigned short) pdb_image.x_last);
965   (void) WriteBlobMSBShort(image,(unsigned short) pdb_image.y_last);
966   (void) WriteBlobMSBLong(image,(unsigned int) pdb_image.reserved_2);
967   (void) WriteBlobMSBShort(image,pdb_image.x_anchor);
968   (void) WriteBlobMSBShort(image,pdb_image.y_anchor);
969   (void) WriteBlobMSBShort(image,(unsigned short) pdb_image.width);
970   (void) WriteBlobMSBShort(image,(unsigned short) pdb_image.height);
971   (void) WriteBlob(image,(size_t) (q-runlength),runlength);
972   runlength=(unsigned char *) RelinquishMagickMemory(runlength);
973   if (pdb_info.number_records > 1)
974     (void) WriteBlobString(image,comment);
975   (void) CloseBlob(image);
976   return(MagickTrue);
977 }