tools: Simplify the mime-maker container creation.
[gnupg.git] / tools / gpgsplit.c
index 856e9aa..93dd8ed 100644 (file)
@@ -5,7 +5,7 @@
  *
  * GnuPG is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
+ * the Free Software Foundation; either version 3 of the License, or
  * (at your option) any later version.
  *
  * GnuPG is distributed in the hope that it will be useful,
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
- */
-
-/* 
- * TODO: Add an option to uncompress packets.  This should come quite handy.
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
  */
 
 #include <config.h>
 #include <ctype.h>
 #include <unistd.h>
 #include <assert.h>
+#include <sys/types.h>
 #ifdef HAVE_DOSISH_SYSTEM
 # include <fcntl.h> /* for setmode() */
 #endif
-#include <zlib.h>
+#ifdef HAVE_ZIP
+# include <zlib.h>
+#endif
 #ifdef HAVE_BZIP2
-#include <bzlib.h>
+# include <bzlib.h>
 #endif /* HAVE_BZIP2 */
 #if defined(__riscos__) && defined(USE_ZLIBRISCOS)
 # include "zlib-riscos.h"
 #endif
 
 #define INCLUDED_BY_MAIN_MODULE 1
-#include "../g10/packet.h"
 #include "util.h"
+#include "openpgpdefs.h"
 
 static int opt_verbose;
 static const char *opt_prefix = "";
@@ -58,9 +56,9 @@ static void split_packets (const char *fname);
 enum cmd_and_opt_values {
   aNull = 0,
   oVerbose       = 'v',
-  oPrefix       = 'p',                          
-  oUncompress   = 500,                      
-  oSecretToPublic,                      
+  oPrefix       = 'p',
+  oUncompress   = 500,
+  oSecretToPublic,
   oNoSplit,
 
   aTest
@@ -79,19 +77,18 @@ static ARGPARSE_OPTS opts[] = {
 {0} };
 
 
-const char *
-strusage( int level )
+static const char *
+my_strusage (int level)
 {
   const char *p;
   switch (level)
     {
-    case 11: p = "gpgsplit (GnuPG)";
+    case 11: p = "gpgsplit (@GNUPG@)";
       break;
     case 13: p = VERSION; break;
     case 17: p = PRINTABLE_OS_NAME; break;
-    case 19: p =
-               "Please report bugs to <bug-gnupg@gnu.org>.\n";
-    break;
+    case 19: p = "Please report bugs to <@EMAIL@>.\n"; break;
+
     case 1:
     case 40:   p =
                   "Usage: gpgsplit [options] [files] (-h for help)";
@@ -100,8 +97,8 @@ strusage( int level )
                   "Syntax: gpgsplit [options] [files]\n"
                   "Split an OpenPGP message into packets\n";
     break;
-    
-    default:   p = default_strusage(level);
+
+    default:   p = NULL;
     }
   return p;
 }
@@ -109,7 +106,7 @@ strusage( int level )
 
 
 int
-main( int argc, char **argv )
+main (int argc, char **argv)
 {
   ARGPARSE_ARGS pargs;
 
@@ -117,8 +114,9 @@ main( int argc, char **argv )
   setmode( fileno(stdin), O_BINARY );
   setmode( fileno(stdout), O_BINARY );
 #endif
-  log_set_name("gpgsplit");
-  
+  log_set_prefix ("gpgsplit", GPGRT_LOG_WITH_PREFIX);
+  set_strusage (my_strusage);
+
   pargs.argc = &argc;
   pargs.argv = &argv;
   pargs.flags=  1;  /* do not remove the args */
@@ -134,7 +132,7 @@ main( int argc, char **argv )
         default : pargs.err = 2; break;
        }
     }
-  
+
   if (log_get_errorcount(0))
     g10_exit (2);
 
@@ -142,12 +140,12 @@ main( int argc, char **argv )
     split_packets (NULL);
   else
     {
-      for ( ;argc; argc--, argv++) 
+      for ( ;argc; argc--, argv++)
         split_packets (*argv);
     }
-  
+
   g10_exit (0);
-  return 0; 
+  return 0;
 }
 
 
@@ -172,7 +170,7 @@ pkttype_to_string (int pkttype)
     case PKT_SECRET_KEY    : s = "secret_key"; break;
     case PKT_PUBLIC_KEY    : s = "public_key"; break;
     case PKT_SECRET_SUBKEY : s = "secret_subkey"; break;
-    case PKT_COMPRESSED    : 
+    case PKT_COMPRESSED    :
       s = opt_uncompress? "uncompressed":"compressed";
       break;
     case PKT_ENCRYPTED     : s = "encrypted"; break;
@@ -195,17 +193,17 @@ pkttype_to_string (int pkttype)
 
 /*
  * Create a new filename and a return a pointer to a statically
- * allocated buffer 
+ * allocated buffer
  */
 static char *
 create_filename (int pkttype)
 {
   static unsigned int partno = 0;
   static char *name;
-  
-  if (!name) 
-    name = m_alloc (strlen (opt_prefix) + 100 );
-  
+
+  if (!name)
+    name = xmalloc (strlen (opt_prefix) + 100 );
+
   assert (pkttype < 1000 && pkttype >= 0 );
   partno++;
   sprintf (name, "%s%06u-%03d" EXTSEP_S "%.40s",
@@ -231,7 +229,7 @@ static int
 read_u32 (FILE *fp, unsigned long *rn)
 {
   size_t tmp;
-  
+
   if (read_u16 (fp, &tmp))
     return -1;
   *rn = tmp << 16;
@@ -243,9 +241,9 @@ read_u32 (FILE *fp, unsigned long *rn)
 
 static int
 write_old_header (FILE *fp, int pkttype, unsigned int len)
-{     
+{
   int ctb = (0x80 | ((pkttype & 15)<<2));
-  
+
   if (len < 256)
     ;
   else if (len < 65536)
@@ -275,7 +273,7 @@ write_old_header (FILE *fp, int pkttype, unsigned int len)
 
 static int
 write_new_header (FILE *fp, int pkttype, unsigned int len)
-{     
+{
   if ( putc ((0xc0 | (pkttype & 0x3f)), fp) == EOF )
     return -1;
 
@@ -317,9 +315,9 @@ public_key_length (const unsigned char *buf, size_t buflen)
   int nmpis;
 
   /*   byte version number (3 or 4)
-       u32  creation time 
+       u32  creation time
        [u16  valid days (version 3 only)]
-       byte algorithm 
+       byte algorithm
        n    MPIs (n and e) */
   if (!buflen)
     return 0;
@@ -364,6 +362,7 @@ public_key_length (const unsigned char *buf, size_t buflen)
   return s - buf;
 }
 
+#ifdef HAVE_ZIP
 static int
 handle_zlib(int algo,FILE *fpin,FILE *fpout)
 {
@@ -372,15 +371,15 @@ handle_zlib(int algo,FILE *fpin,FILE *fpout)
   unsigned int inbufsize, outbufsize;
   int c,zinit_done, zrc, nread, count;
   size_t n;
-              
+
   memset (&zs, 0, sizeof zs);
   inbufsize = 2048;
-  inbuf = m_alloc (inbufsize);
+  inbuf = xmalloc (inbufsize);
   outbufsize = 8192;
-  outbuf = m_alloc (outbufsize);
+  outbuf = xmalloc (outbufsize);
   zs.avail_in = 0;
   zinit_done = 0;
-              
+
   do
     {
       if (zs.avail_in < inbufsize)
@@ -391,11 +390,11 @@ handle_zlib(int algo,FILE *fpin,FILE *fpout)
          count = inbufsize - n;
          for (nread=0;
               nread < count && (c=getc (fpin)) != EOF;
-              nread++) 
+              nread++)
            inbuf[n+nread] = c;
-                      
+
          n += nread;
-         if (nread < count && algo == 1) 
+         if (nread < count && algo == 1)
            {
              inbuf[n] = 0xFF; /* chew dummy byte */
              n++;
@@ -404,12 +403,12 @@ handle_zlib(int algo,FILE *fpin,FILE *fpout)
        }
       zs.next_out = (Bytef *) outbuf;
       zs.avail_out = outbufsize;
-                    
-      if (!zinit_done) 
+
+      if (!zinit_done)
        {
          zrc = (algo == 1? inflateInit2 ( &zs, -13)
                 : inflateInit ( &zs ));
-         if (zrc != Z_OK) 
+         if (zrc != Z_OK)
            {
              log_fatal ("zlib problem: %s\n", zs.msg? zs.msg :
                         zrc == Z_MEM_ERROR ? "out of core" :
@@ -435,18 +434,28 @@ handle_zlib(int algo,FILE *fpin,FILE *fpout)
              else
                log_fatal ("zlib inflate problem: rc=%d\n", zrc );
            }
-         for (n=0; n < outbufsize - zs.avail_out; n++) 
+         for (n=0; n < outbufsize - zs.avail_out; n++)
            {
              if (putc (outbuf[n], fpout) == EOF )
                return 1;
            }
        }
-    } 
+    }
   while (zrc != Z_STREAM_END && zrc != Z_BUF_ERROR);
+  {
+    int i;
+
+    fputs ("Left over bytes:", stderr);
+    for (i=0; i < zs.avail_in; i++)
+      fprintf (stderr, " %02X", zs.next_in[i]);
+    putc ('\n', stderr);
+
+  }
   inflateEnd (&zs);
 
   return 0;
 }
+#endif /*HAVE_ZIP*/
 
 #ifdef HAVE_BZIP2
 static int
@@ -457,43 +466,43 @@ handle_bzip2(int algo,FILE *fpin,FILE *fpout)
   unsigned int inbufsize, outbufsize;
   int c,zinit_done, zrc, nread, count;
   size_t n;
-              
+
   memset (&bzs, 0, sizeof bzs);
   inbufsize = 2048;
-  inbuf = m_alloc (inbufsize);
+  inbuf = xmalloc (inbufsize);
   outbufsize = 8192;
-  outbuf = m_alloc (outbufsize);
+  outbuf = xmalloc (outbufsize);
   bzs.avail_in = 0;
   zinit_done = 0;
-              
+
   do
     {
       if (bzs.avail_in < inbufsize)
        {
          n = bzs.avail_in;
          if (!n)
-           bzs.next_in = (Bytef *) inbuf;
+           bzs.next_in = inbuf;
          count = inbufsize - n;
          for (nread=0;
               nread < count && (c=getc (fpin)) != EOF;
-              nread++) 
+              nread++)
            inbuf[n+nread] = c;
-                      
+
          n += nread;
-         if (nread < count && algo == 1) 
+         if (nread < count && algo == 1)
            {
              inbuf[n] = 0xFF; /* chew dummy byte */
              n++;
            }
          bzs.avail_in = n;
        }
-      bzs.next_out = (Bytef *) outbuf;
+      bzs.next_out = outbuf;
       bzs.avail_out = outbufsize;
-                    
-      if (!zinit_done) 
+
+      if (!zinit_done)
        {
          zrc = BZ2_bzDecompressInit(&bzs,0,0);
-         if (zrc != BZ_OK) 
+         if (zrc != BZ_OK)
            log_fatal ("bz2lib problem: %d\n",zrc);
          zinit_done = 1;
        }
@@ -504,13 +513,13 @@ handle_bzip2(int algo,FILE *fpin,FILE *fpout)
            ; /* eof */
          else if (zrc != BZ_OK && zrc != BZ_PARAM_ERROR)
            log_fatal ("bz2lib inflate problem: %d\n", zrc );
-         for (n=0; n < outbufsize - bzs.avail_out; n++) 
+         for (n=0; n < outbufsize - bzs.avail_out; n++)
            {
              if (putc (outbuf[n], fpout) == EOF )
                return 1;
            }
        }
-    } 
+    }
   while (zrc != BZ_STREAM_END && zrc != BZ_PARAM_ERROR);
   BZ2_bzDecompressEnd(&bzs);
 
@@ -520,14 +529,14 @@ handle_bzip2(int algo,FILE *fpin,FILE *fpout)
 
 /* hdr must point to a buffer large enough to hold all header bytes */
 static int
-write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
-             int pkttype, int partial, unsigned char *hdr, size_t hdrlen)
+write_part (FILE *fpin, unsigned long pktlen,
+            int pkttype, int partial, unsigned char *hdr, size_t hdrlen)
 {
   FILE *fpout;
   int c, first;
   unsigned char *p;
   const char *outname = create_filename (pkttype);
-  
+
 #if defined(__riscos__) && defined(USE_ZLIBRISCOS)
   static int initialized = 0;
 
@@ -539,11 +548,11 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
   else
     {
       if (opt_verbose)
-        log_info ("writing `%s'\n", outname);
+        log_info ("writing '%s'\n", outname);
       fpout = fopen (outname, "wb");
-      if (!fpout) 
+      if (!fpout)
         {
-          log_error ("error creating `%s': %s\n", outname, strerror(errno));
+          log_error ("error creating '%s': %s\n", outname, strerror(errno));
           /* stop right now, otherwise we would mess up the sequence
              of the part numbers */
           g10_exit (1);
@@ -553,36 +562,36 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
   if (opt_secret_to_public
       && (pkttype == PKT_SECRET_KEY || pkttype == PKT_SECRET_SUBKEY))
     {
-      unsigned char *blob = m_alloc (pktlen);
+      unsigned char *blob = xmalloc (pktlen);
       int i, len;
 
       pkttype = pkttype == PKT_SECRET_KEY? PKT_PUBLIC_KEY:PKT_PUBLIC_SUBKEY;
 
-      for (i=0; i < pktlen; i++) 
+      for (i=0; i < pktlen; i++)
         {
           c = getc (fpin);
-          if (c == EOF) 
+          if (c == EOF)
             goto read_error;
           blob[i] = c;
         }
       len = public_key_length (blob, pktlen);
       if (!len)
         {
-          log_error ("error calcualting public key length\n");
+          log_error ("error calculating public key length\n");
           g10_exit (1);
         }
       if ( (hdr[0] & 0x40) )
-        { 
+        {
           if (write_new_header (fpout, pkttype, len))
             goto write_error;
         }
       else
-        { 
+        {
           if (write_old_header (fpout, pkttype, len))
             goto write_error;
         }
 
-      for (i=0; i < len; i++) 
+      for (i=0; i < len; i++)
         {
           if ( putc (blob[i], fpout) == EOF )
             goto write_error;
@@ -600,12 +609,12 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
             goto write_error;
         }
     }
-  
+
   first = 1;
   while (partial)
     {
       size_t partlen;
-      
+
       if (partial == 1)
         { /* openpgp */
           if (first )
@@ -614,11 +623,11 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
               assert( c >= 224 && c < 255 );
               first = 0;
             }
-          else if ((c = getc (fpin)) == EOF ) 
+          else if ((c = getc (fpin)) == EOF )
             goto read_error;
           else
             hdr[hdrlen++] = c;
-            
+
           if (c < 192)
             {
               pktlen = c;
@@ -627,7 +636,7 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
           else if (c < 224 )
             {
               pktlen = (c - 192) * 256;
-              if ((c = getc (fpin)) == EOF) 
+              if ((c = getc (fpin)) == EOF)
                 goto read_error;
               hdr[hdrlen++] = c;
               pktlen += c + 192;
@@ -651,9 +660,9 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
                     goto write_error;
                 }
               partlen = 1 << (c & 0x1f);
-              for (; partlen; partlen--) 
+              for (; partlen; partlen--)
                 {
-                  if ((c = getc (fpin)) == EOF) 
+                  if ((c = getc (fpin)) == EOF)
                     goto read_error;
                   if ( putc (c, fpout) == EOF )
                     goto write_error;
@@ -667,17 +676,17 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
             goto read_error;
           hdr[hdrlen++] = partlen >> 8;
           hdr[hdrlen++] = partlen;
-          for (p=hdr; hdrlen; p++, hdrlen--) 
+          for (p=hdr; hdrlen; p++, hdrlen--)
             {
               if ( putc (*p, fpout) == EOF )
                 goto write_error;
             }
           if (!partlen)
             partial = 0; /* end of packet */
-          for (; partlen; partlen--) 
+          for (; partlen; partlen--)
             {
               c = getc (fpin);
-              if (c == EOF) 
+              if (c == EOF)
                 goto read_error;
               if ( putc (c, fpout) == EOF )
                 goto write_error;
@@ -688,16 +697,20 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
           pktlen = 0;
           partial = 0;
           hdrlen = 0;
-          if (opt_uncompress) 
+          if (opt_uncompress)
             {
               if ((c = getc (fpin)) == EOF)
                 goto read_error;
 
-             if(c==1 || c==2)
+             if (0)
+                ;
+#ifdef HAVE_ZIP
+             else if(c==1 || c==2)
                {
                  if(handle_zlib(c,fpin,fpout))
                    goto write_error;
                }
+#endif /* HAVE_ZIP */
 #ifdef HAVE_BZIP2
              else if(c==3)
                {
@@ -713,7 +726,7 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
             }
           else
             {
-              while ( (c=getc (fpin)) != EOF ) 
+              while ( (c=getc (fpin)) != EOF )
                 {
                   if ( putc (c, fpout) == EOF )
                     goto write_error;
@@ -724,33 +737,33 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
        }
     }
 
-  for (p=hdr; hdrlen; p++, hdrlen--) 
+  for (p=hdr; hdrlen; p++, hdrlen--)
     {
       if ( putc (*p, fpout) == EOF )
         goto write_error;
     }
-  
+
   /* standard packet or last segment of partial length encoded packet */
-  for (; pktlen; pktlen--) 
+  for (; pktlen; pktlen--)
     {
       c = getc (fpin);
-      if (c == EOF) 
+      if (c == EOF)
         goto read_error;
       if ( putc (c, fpout) == EOF )
         goto write_error;
     }
-  
+
  ready:
   if ( !opt_no_split && fclose (fpout) )
-    log_error ("error closing `%s': %s\n", outname, strerror (errno));
+    log_error ("error closing '%s': %s\n", outname, strerror (errno));
   return 0;
-  
- write_error:    
-  log_error ("error writing `%s': %s\n", outname, strerror (errno));
+
+ write_error:
+  log_error ("error writing '%s': %s\n", outname, strerror (errno));
   if (!opt_no_split)
     fclose (fpout);
   return 2;
-  
+
  read_error:
   if (!opt_no_split)
     {
@@ -764,19 +777,19 @@ write_part ( const char *fname, FILE *fpin, unsigned long pktlen,
 
 
 static int
-do_split (const char *fname, FILE *fp)
+do_split (FILE *fp)
 {
   int c, ctb, pkttype;
   unsigned long pktlen = 0;
   int partial = 0;
   unsigned char header[20];
   int header_idx = 0;
-  
+
   ctb = getc (fp);
   if (ctb == EOF)
     return 3; /* ready */
   header[header_idx++] = ctb;
-  
+
   if (!(ctb & 0x80))
     {
       log_error("invalid CTB %02x\n", ctb );
@@ -794,19 +807,19 @@ do_split (const char *fname, FILE *fp)
       else if ( c < 224 )
         {
           pktlen = (c - 192) * 256;
-          if( (c = getc (fp)) == EOF ) 
+          if( (c = getc (fp)) == EOF )
             return -1;
           header[header_idx++] = c;
           pktlen += c + 192;
        }
-      else if ( c == 255 ) 
+      else if ( c == 255 )
         {
           if (read_u32 (fp, &pktlen))
             return -1;
           header[header_idx++] = pktlen >> 24;
           header[header_idx++] = pktlen >> 16;
           header[header_idx++] = pktlen >> 8;
-          header[header_idx++] = pktlen; 
+          header[header_idx++] = pktlen;
        }
       else
         { /* partial body length */
@@ -817,7 +830,7 @@ do_split (const char *fname, FILE *fp)
   else
     {
       int lenbytes;
-      
+
       pkttype = (ctb>>2)&0xf;
       lenbytes = ((ctb&3)==3)? 0 : (1<<(ctb & 3));
       if (!lenbytes )
@@ -830,20 +843,19 @@ do_split (const char *fname, FILE *fp)
        }
       else
         {
-          for ( ; lenbytes; lenbytes-- ) 
+          for ( ; lenbytes; lenbytes-- )
             {
               pktlen <<= 8;
-              if( (c = getc (fp)) == EOF ) 
+              if( (c = getc (fp)) == EOF )
                 return -1;
               header[header_idx++] = c;
-              
+
               pktlen |= c;
            }
        }
     }
 
-  return write_part (fname, fp, pktlen, pkttype, partial,
-                     header, header_idx);
+  return write_part (fp, pktlen, pkttype, partial, header, header_idx);
 }
 
 
@@ -852,27 +864,27 @@ split_packets (const char *fname)
 {
   FILE *fp;
   int rc;
-  
+
   if (!fname || !strcmp (fname, "-"))
     {
       fp = stdin;
       fname = "-";
     }
-  else if ( !(fp = fopen (fname,"rb")) ) 
+  else if ( !(fp = fopen (fname,"rb")) )
     {
-      log_error ("can't open `%s': %s\n", fname, strerror (errno));
+      log_error ("can't open '%s': %s\n", fname, strerror (errno));
       return;
     }
-  
-  while ( !(rc = do_split (fname, fp)) )
+
+  while ( !(rc = do_split (fp)) )
     ;
   if ( rc > 0 )
     ; /* error already handled */
   else if ( ferror (fp) )
-    log_error ("error reading `%s': %s\n", fname, strerror (errno));
+    log_error ("error reading '%s': %s\n", fname, strerror (errno));
   else
-    log_error ("premature EOF while reading `%s'\n", fname );
-  
+    log_error ("premature EOF while reading '%s'\n", fname );
+
   if ( fp != stdin )
     fclose (fp);
 }