← Back to branch summary

~ubuntu-branches/ubuntu/lucid/graphviz/lucid-security

« back to all changes in this revision

Viewing changes to gd/webpng.c

  • Committer: Bazaar Package Importer
  • Author(s): Stephen M Moraco
  • Date: 2002-02-05 18:52:12 UTC
  • Revision ID: james.westby@ubuntu.com-20020205185212-8i04c70te00rc40y
Tags: upstream-1.7.16
Import upstream version 1.7.16

Show diffs side-by-side

added added

removed removed

Lines of Context:
gd/webpng.c
 
1
/* Bring in the gd library functions */
 
2
#include "gd.h"
 
3
 
 
4
/* Bring in standard I/O and string manipulation functions */
 
5
#include <stdio.h>
 
6
#include <stdlib.h>             /* for atoi() */
 
7
#include <string.h>
 
8
 
 
9
#ifdef _WIN32
 
10
#include <process.h>
 
11
int
 
12
getpid ()
 
13
{
 
14
  return _getpid ();
 
15
}
 
16
#else
 
17
#include <unistd.h>             /* for getpid(), unlink() */
 
18
#endif
 
19
int
 
20
main (int argc, char **argv)
 
21
{
 
22
  FILE *in;
 
23
  FILE *out;
 
24
  char outFn[20];
 
25
  int useStdinStdout = 0;
 
26
 
 
27
  /* Declare our image pointer */
 
28
  gdImagePtr im = 0;
 
29
  int i;
 
30
  /* We'll clear 'no' once we know the user has made a
 
31
     reasonable request. */
 
32
  int no = 1;
 
33
  /* We'll set 'write' once we know the user's request
 
34
     requires that the image be written back to disk. */
 
35
  int write = 0;
 
36
  /* C programs always get at least one argument; we want at
 
37
     least one more (the image), more in practice. */
 
38
  if (argc < 2 || !strcmp (argv[1], "--help"))
 
39
    {
 
40
      no = 1;
 
41
      goto usage;
 
42
    }
 
43
 
 
44
  /* The last argument should be the image. Open the file. */
 
45
  if (strcmp ("-", argv[argc - 1]) == 0)
 
46
    {                           /* - is synonymous with STDIN */
 
47
      useStdinStdout = 1;
 
48
      in = stdin;
 
49
    }
 
50
  else
 
51
    {
 
52
      in = fopen (argv[argc - 1], "rb");
 
53
    }
 
54
  if (!in)
 
55
    {
 
56
      fprintf (stderr,
 
57
               "Error: can't open file %s.\n", argv[argc - 1]);
 
58
      exit (1);
 
59
    }
 
60
  /* Now load the image. */
 
61
  im = gdImageCreateFromPng (in);
 
62
  fclose (in);
 
63
  /* If the load failed, it must not be a PNG file. */
 
64
  if (!im)
 
65
    {
 
66
      fprintf (stderr,
 
67
               "Error: %s is not a valid PNG file.\n", argv[argc - 1]);
 
68
      exit (1);
 
69
    }
 
70
  /* Consider each argument in turn. */
 
71
  for (i = 1; (i < (argc - 1)); i++)
 
72
    {
 
73
      /* -i turns on and off interlacing. */
 
74
      if (!strcmp (argv[i], "--help"))
 
75
        {
 
76
          /* Every program should use this for help! :) */
 
77
          no = 1;
 
78
          goto usage;
 
79
        }
 
80
      else if (!strcmp (argv[i], "-i"))
 
81
        {
 
82
          if (i == (argc - 2))
 
83
            {
 
84
              fprintf (stderr,
 
85
                       "Error: -i specified without y or n.\n");
 
86
              no = 1;
 
87
              goto usage;
 
88
            }
 
89
          if (!strcmp (argv[i + 1], "y"))
 
90
            {
 
91
              /* Set interlace. */
 
92
              gdImageInterlace (im, 1);
 
93
            }
 
94
          else if (!strcmp (argv[i + 1], "n"))
 
95
            {
 
96
              /* Clear interlace. */
 
97
              gdImageInterlace (im, 0);
 
98
            }
 
99
          else
 
100
            {
 
101
              fprintf (stderr,
 
102
                       "Error: -i specified without y or n.\n");
 
103
              no = 1;
 
104
              goto usage;
 
105
            }
 
106
          i++;
 
107
          no = 0;
 
108
          write = 1;
 
109
        }
 
110
      else if (!strcmp (argv[i], "-t"))
 
111
        {
 
112
          /* Set transparent index (or none). */
 
113
          int index;
 
114
          if (i == (argc - 2))
 
115
            {
 
116
              fprintf (stderr,
 
117
                       "Error: -t specified without a color table index.\n");
 
118
              no = 1;
 
119
              goto usage;
 
120
            }
 
121
          if (!strcmp (argv[i + 1], "none"))
 
122
            {
 
123
              /* -1 means not transparent. */
 
124
              gdImageColorTransparent (im, -1);
 
125
            }
 
126
          else
 
127
            {
 
128
              /* OK, get an integer and set the index. */
 
129
              index = atoi (argv[i + 1]);
 
130
              gdImageColorTransparent (im, index);
 
131
            }
 
132
          i++;
 
133
          write = 1;
 
134
          no = 0;
 
135
        }
 
136
      else if (!strcmp (argv[i], "-l"))
 
137
        {
 
138
          /* List the colors in the color table. */
 
139
          int j;
 
140
          if (!im->trueColor)
 
141
            {
 
142
              /* Tabs used below. */
 
143
              printf ("Index    Red     Green   Blue Alpha\n");
 
144
              for (j = 0; (j < gdImageColorsTotal (im)); j++)
 
145
                {
 
146
                  /* Use access macros to learn colors. */
 
147
                  printf ("%d   %d      %d      %d      %d\n",
 
148
                          j,
 
149
                          gdImageRed (im, j),
 
150
                          gdImageGreen (im, j),
 
151
                          gdImageBlue (im, j),
 
152
                          gdImageAlpha (im, j));
 
153
                }
 
154
            }
 
155
          else
 
156
            {
 
157
              printf ("Truecolor image, no palette entries to list.\n");
 
158
            }
 
159
          no = 0;
 
160
        }
 
161
      else if (!strcmp (argv[i], "-d"))
 
162
        {
 
163
          /* Output dimensions, etc. */
 
164
          int t;
 
165
          printf ("Width: %d Height: %d Colors: %d\n",
 
166
                  gdImageSX (im), gdImageSY (im),
 
167
                  gdImageColorsTotal (im));
 
168
          t = gdImageGetTransparent (im);
 
169
          if (t != (-1))
 
170
            {
 
171
              printf ("First 100% transparent index: %d\n", t);
 
172
            }
 
173
          else
 
174
            {
 
175
              /* -1 means the image is not transparent. */
 
176
              printf ("First 100% transparent index: none\n");
 
177
            }
 
178
          if (gdImageGetInterlaced (im))
 
179
            {
 
180
              printf ("Interlaced: yes\n");
 
181
            }
 
182
          else
 
183
            {
 
184
              printf ("Interlaced: no\n");
 
185
            }
 
186
          no = 0;
 
187
        }
 
188
      else
 
189
        {
 
190
          fprintf (stderr, "Unknown argument: %s\n", argv[i]);
 
191
          break;
 
192
        }
 
193
    }
 
194
usage:
 
195
  if (no)
 
196
    {
 
197
      /* If the command failed, output an explanation. */
 
198
      fprintf (stderr,
 
199
          "Usage: webpng [-i y|n ] [-l] [-t index|none ] [-d] pngname.png\n"
 
200
 
 
201
               "  -i [y|n]   Turns on/off interlace\n"
 
202
               "  -l         Prints the table of color indexes\n"
 
203
               "  -t [index] Set the transparent color to the specified index (0-255 or \"none\")\n"
 
204
               "  -d         Reports the dimensions and other characteristics of the image.\n"
 
205
               "\n"
 
206
               "If you specify '-' as the input file, stdin/stdout will be used input/output.\n"
 
207
        );
 
208
    }
 
209
  if (write)
 
210
    {
 
211
      if (useStdinStdout)
 
212
        {
 
213
          out = stdout;
 
214
        }
 
215
      else
 
216
        {
 
217
          /* Open a temporary file. */
 
218
 
 
219
          /* "temp.tmp" is not good temporary filename. */
 
220
          sprintf (outFn, "webpng.tmp%d", getpid ());
 
221
          out = fopen (outFn, "wb");
 
222
 
 
223
          if (!out)
 
224
            {
 
225
              fprintf (stderr,
 
226
                       "Unable to write to %s -- exiting\n", outFn);
 
227
              exit (1);
 
228
            }
 
229
        }
 
230
 
 
231
      /* Write the new PNG. */
 
232
      gdImagePng (im, out);
 
233
 
 
234
      if (!useStdinStdout)
 
235
        {
 
236
          fclose (out);
 
237
          /* Erase the old PNG. */
 
238
          unlink (argv[argc - 1]);
 
239
          /* Rename the new to the old. */
 
240
          if (rename (outFn, argv[argc - 1]) != 0)
 
241
            {
 
242
              perror ("rename");
 
243
              exit (1);
 
244
            }
 
245
        }
 
246
    }
 
247
  /* Delete the image from memory. */
 
248
  if (im)
 
249
    {
 
250
      gdImageDestroy (im);
 
251
    }
 
252
  /* All's well that ends well. */
 
253
  return 0;
 
254
}