1 files changed, 152 insertions, 0 deletions
diff --git a/mail/src/main/java/org/spongycastle/mail/smime/SMIMECompressedGenerator.java b/mail/src/main/java/org/spongycastle/mail/smime/SMIMECompressedGenerator.java
new file mode 100644
index 00000000..2701e9d0
--- /dev/null
+++ b/mail/src/main/java/org/spongycastle/mail/smime/SMIMECompressedGenerator.java
@@ -0,0 +1,152 @@
+package org.bouncycastle.mail.smime;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.security.AccessController;
+import java.security.PrivilegedAction;
+
+import javax.activation.CommandMap;
+import javax.activation.MailcapCommandMap;
+import javax.mail.MessagingException;
+import javax.mail.internet.MimeBodyPart;
+import javax.mail.internet.MimeMessage;
+
+import org.bouncycastle.cms.CMSCompressedDataGenerator;
+import org.bouncycastle.cms.CMSCompressedDataStreamGenerator;
+import org.bouncycastle.operator.OutputCompressor;
+
+/**
+ * General class for generating a pkcs7-mime compressed message.
+ *
+ * A simple example of usage.
+ *
+ * <pre>
+ *      SMIMECompressedGenerator  fact = new SMIMECompressedGenerator();
+ *
+ *      MimeBodyPart           smime = fact.generate(content, algorithm);
+ * </pre>
+ *
+ * <b>Note:<b> Most clients expect the MimeBodyPart to be in a MimeMultipart
+ * when it's sent.
+ */
+public class SMIMECompressedGenerator
+    extends SMIMEGenerator
+{
+    public static final String  ZLIB    = CMSCompressedDataGenerator.ZLIB;
+
+    private static final String COMPRESSED_CONTENT_TYPE = "application/pkcs7-mime; name=\"smime.p7z\"; smime-type=compressed-data";
+
+    static
+    {
+        final MailcapCommandMap mc = (MailcapCommandMap)CommandMap.getDefaultCommandMap();
+
+        mc.addMailcap("application/pkcs7-mime;; x-java-content-handler=org.bouncycastle.mail.smime.handlers.pkcs7_mime");
+        mc.addMailcap("application/x-pkcs7-mime;; x-java-content-handler=org.bouncycastle.mail.smime.handlers.x_pkcs7_mime");
+
+        AccessController.doPrivileged(new PrivilegedAction()
+        {
+            public Object run()
+            {
+                CommandMap.setDefaultCommandMap(mc);
+
+                return null;
+            }
+        });
+    }
+
+    /**
+     * generate an compressed object that contains an SMIME Compressed
+     * object using the given compression algorithm.
+     */
+    private MimeBodyPart make(
+        MimeBodyPart    content,
+        OutputCompressor compressor)
+        throws SMIMEException
+    {
+        try
+        {  
+            MimeBodyPart data = new MimeBodyPart();
+        
+            data.setContent(new ContentCompressor(content, compressor), COMPRESSED_CONTENT_TYPE);
+            data.addHeader("Content-Type", COMPRESSED_CONTENT_TYPE);
+            data.addHeader("Content-Disposition", "attachment; filename=\"smime.p7z\"");
+            data.addHeader("Content-Description", "S/MIME Compressed Message");
+            data.addHeader("Content-Transfer-Encoding", encoding);
+
+            return data;
+        }
+        catch (MessagingException e)
+        {
+            throw new SMIMEException("exception putting multi-part together.", e);
+        }
+    }
+
+    /**
+     * generate an compressed object that contains an SMIME Compressed
+     * object using the given provider from the contents of the passed in
+     * message
+     */
+    public MimeBodyPart generate(
+        MimeBodyPart    content,
+        OutputCompressor compressor)
+        throws SMIMEException
+    {
+        return make(makeContentBodyPart(content), compressor);
+    }
+
+    /**
+     * generate an compressed object that contains an SMIME Compressed
+     * object using the given provider from the contents of the passed in
+     * message
+     */
+    public MimeBodyPart generate(
+        MimeMessage     message,
+        OutputCompressor compressor)
+        throws SMIMEException
+    {
+        try
+        {
+            message.saveChanges();      // make sure we're up to date.
+        }
+        catch (MessagingException e)
+        {
+            throw new SMIMEException("unable to save message", e);
+        }
+                        
+        return make(makeContentBodyPart(message), compressor);
+    }
+    
+    private class ContentCompressor
+        implements SMIMEStreamingProcessor
+    {
+        private final MimeBodyPart content;
+        private final OutputCompressor compressor;
+        
+        ContentCompressor(
+            MimeBodyPart content,
+            OutputCompressor compressor)
+        {
+            this.content = content;
+            this.compressor = compressor;
+        }
+
+        public void write(OutputStream out)
+            throws IOException
+        {
+            CMSCompressedDataStreamGenerator cGen = new CMSCompressedDataStreamGenerator();
+            
+            OutputStream compressed = cGen.open(out, compressor);
+            
+            try
+            {
+                content.writeTo(compressed);
+                
+                compressed.close();
+            }
+            catch (MessagingException e)
+            {
+                throw new IOException(e.toString());
+            }
+        }
+    }
+}