mimelib/attach.cpp

   1 //=============================================================================
   2 // File:       attach.cpp
   3 // Contents:   Definitions for MessageWithAttachments
   4 // Maintainer: Doug Sauder <dwsauder@fwb.gulf.net>
   5 // WWW:        http://www.fwb.gulf.net/~dwsauder/mimepp.html
   6 // $Revision$
   7 // $Date$
   8 //
   9 // Copyright (c) 1996, 1997 Douglas W. Sauder
  10 // All rights reserved.
  11 //
  12 // IN NO EVENT SHALL DOUGLAS W. SAUDER BE LIABLE TO ANY PARTY FOR DIRECT,
  13 // INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF
  14 // THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF DOUGLAS W. SAUDER
  15 // HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  16 //
  17 // DOUGLAS W. SAUDER SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT
  18 // NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  19 // PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS ON AN "AS IS"
  20 // BASIS, AND DOUGLAS W. SAUDER HAS NO OBLIGATION TO PROVIDE MAINTENANCE,
  21 // SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
  22 //
  23 //=============================================================================
  24
  25 #include <stdio.h>
  26 #include <sys/types.h>
  27 #include <sys/stat.h>
  28 #include <mimelib/string.h>
  29 #include <mimelib/utility.h>
  30 #include "attach.h"
  31
  32
  33 MessageWithAttachments::MessageWithAttachments()
  34 {
  35 }
  36
  37
  38 MessageWithAttachments::~MessageWithAttachments()
  39 {
  40 }
  41
  42
  43 void MessageWithAttachments::SetText(const DwString& aStr)
  44 {
  45     // Create a body part and set the necessary fields
  46
  47     MultipartBodyPart part;
  48     part.SetType(DwMime::kTypeText);
  49     part.SetSubtype(DwMime::kSubtypePlain);
  50     part.SetCte(DwMime::kCte7bit);
  51
  52     // Set the string as the body of the body part
  53
  54     part.SetBody(aStr);
  55
  56     // Set this body part as the first one
  57
  58     SetBodyPart(0, part);
  59 }
  60
  61
  62 int MessageWithAttachments::NumberOfAttachments() const
  63 {
  64     int n = NumberOfParts() - 1;
  65     return (n >= 0) ? n : 0;
  66 }
  67
  68
  69 void MessageWithAttachments::Attach7bitFile(const char* aFilename,
  70     int aType, int aSubtype)
  71 {
  72     // Get the file contents
  73
  74     DwString str;
  75     PutFileInString(aFilename, str);
  76
  77     // Create a body part and set the necessary fields
  78
  79     MultipartBodyPart part;
  80     part.SetType(aType);
  81     part.SetSubtype(aSubtype);
  82     part.SetCte(DwMime::kCte7bit);
  83
  84     // Set content-disposition to attachment, with filename parameter
  85     // (see RFC-1806 for information on this *experimental* header field)
  86
  87     DwString contDisp = "attachment; filename=";
  88     contDisp += '\"';
  89     contDisp += aFilename;
  90     contDisp += '\"';
  91     part.SetContentDisposition(contDisp);
  92
  93     // Set the file contents as the body of the body part
  94
  95     part.SetBody(str);
  96
  97     // Make sure this is not the first part, since that is reserved for
  98     // the text
  99
 100     if (NumberOfParts() == 0) {
 101         SetBodyPart(1, part);
 102     }
 103     else {
 104         AddBodyPart(part);
 105     }
 106 }
 107
 108
 109 void MessageWithAttachments::Attach8bitFile(const char* aFilename,
 110     int aType, int aSubtype)
 111 {
 112     // Get the file contents
 113
 114     DwString str;
 115     PutFileInString(aFilename, str);
 116
 117     // Encode using quoted-printable encoding
 118
 119     DwString encStr;
 120     DwEncodeQuotedPrintable(str, encStr);
 121
 122     // Create a body part and set the necessary fields
 123
 124     MultipartBodyPart part;
 125     part.SetType(aType);
 126     part.SetSubtype(aSubtype);
 127     part.SetCte(DwMime::kCteQuotedPrintable);
 128
 129     // Set content-disposition to attachment, with filename parameter
 130     // (see RFC-1806 for information on this *experimental* header field)
 131
 132     DwString contDisp = "attachment; filename=";
 133     contDisp += '\"';
 134     contDisp += aFilename;
 135     contDisp += '\"';
 136     part.SetContentDisposition(contDisp);
 137
 138     // Set the encoded file contents as the body of the body part
 139
 140     part.SetBody(encStr);
 141
 142     // Make sure this is not the first part, since that is reserved for
 143     // the text
 144
 145     if (NumberOfParts() == 0) {
 146         SetBodyPart(1, part);
 147     }
 148     else {
 149         AddBodyPart(part);
 150     }
 151 }
 152
 153
 154 void MessageWithAttachments::AttachBinaryFile(const char* aFilename,
 155     int aType, int aSubtype)
 156 {
 157     // Get the file contents
 158
 159     DwString str;
 160     PutFileInString(aFilename, str);
 161
 162     // Encode using base64 encoding
 163
 164     DwString encStr;
 165     DwEncodeBase64(str, encStr);
 166
 167     // Create a body part and set the necessary fields
 168
 169     MultipartBodyPart part;
 170     part.SetType(aType);
 171     part.SetSubtype(aSubtype);
 172     part.SetCte(DwMime::kCteBase64);
 173
 174     // Set content-disposition to attachment, with filename parameter
 175     // (see RFC-1806 for information on this *experimental* header field)
 176
 177     DwString contDisp = "attachment; filename=";
 178     contDisp += '\"';
 179     contDisp += aFilename;
 180     contDisp += '\"';
 181     part.SetContentDisposition(contDisp);
 182
 183     // Set the encoded file contents as the body of the body part
 184
 185     part.SetBody(encStr);
 186
 187     // Make sure this is not the first part, since that is reserved for
 188     // the text
 189
 190     if (NumberOfParts() == 0) {
 191         SetBodyPart(1, part);
 192     }
 193     else {
 194         AddBodyPart(part);
 195     }
 196 }
 197
 198
 199 int MessageWithAttachments::PutFileInString(const char* aFilename,
 200     DwString& str)
 201 {
 202     // Get the file size
 203     struct stat statBuf;
 204     int k = stat(aFilename, &statBuf);
 205     if (k < 0) {
 206         str = "";
 207         return -1;
 208     }
 209     int fileSize = (int) statBuf.st_size;
 210
 211     // Allocate a buffer
 212
 213     int bufSize = fileSize + 8; // a little elbow room added
 214     char* buf = new char[bufSize];
 215
 216     // Read the file into the buffer
 217
 218     FILE* fp = fopen(aFilename, "rb");
 219     if (fp == 0) {
 220         delete buf;
 221         str = "";
 222         return -1;
 223     }
 224     int len = 0;
 225     while (1) {
 226         int ch = getc(fp);
 227         if (feof(fp) || len == fileSize) {
 228             break;
 229         }
 230         buf[len++] = ch;
 231     }
 232     buf[len] = 0;
 233     fclose(fp);
 234
 235     // Place the buffer in the string
 236
 237     str.TakeBuffer(buf, bufSize, 0, len);
 238     return 0;
 239 }
 240