src/apps/cortex/Persistence/Wrappers/FlatMessageIO.cpp

   1 /*
   2  * Copyright (c) 1999-2000, Eric Moon.
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions, and the following disclaimer.
  11  *
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions, and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * 3. The name of the author may not be used to endorse or promote products
  17  *    derived from this software without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR
  20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  21  * OF TITLE, NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  23  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  24  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  25  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  26  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
  27  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31
  32 // FlatMessageIO.cpp
  33 // e.moon 6jul99
  34
  35 #include "FlatMessageIO.h"
  36 //#include "xml_export_utils.h"
  37
  38 #include "ExportContext.h"
  39
  40 #include <Debug.h>
  41
  42 // base64 encoding tools
  43 #include <E-mail.h>
  44
  45 #include <cstdlib>
  46 #include <cstring>
  47
  48 __USE_CORTEX_NAMESPACE
  49
  50 // -------------------------------------------------------- //
  51 // *** constants
  52 // -------------------------------------------------------- //
  53
  54 const char* const FlatMessageIO::s_element                              = "flat-BMessage";
  55
  56 // -------------------------------------------------------- //
  57 // *** ctor/dtor/accessor
  58 // -------------------------------------------------------- //
  59
  60 FlatMessageIO::~FlatMessageIO() {
  61         if(m_ownMessage && m_message)
  62                 delete m_message;
  63 }
  64 FlatMessageIO::FlatMessageIO() :
  65         m_ownMessage(true),
  66         m_message(0) {}
  67 FlatMessageIO::FlatMessageIO(const BMessage* message) :
  68         m_ownMessage(false),
  69         m_message(const_cast<BMessage*>(message)) {}
  70
  71 void FlatMessageIO::setMessage(BMessage* message) {
  72         if(m_ownMessage && m_message)
  73                 delete m_message;
  74         m_ownMessage = false;
  75         m_message = message;
  76 }
  77
  78 // -------------------------------------------------------- //
  79 // *** static setup method
  80 // -------------------------------------------------------- //
  81
  82 // call this method to install hooks for the tags needed by
  83 // FlatMessageIO into the given document type
  84 /*static*/
  85 void FlatMessageIO::AddTo(XML::DocumentType* pDocType) {
  86         pDocType->addMapping(new Mapping<FlatMessageIO>(s_element));
  87 }
  88
  89 // -------------------------------------------------------- //
  90 // *** IPersistent impl.
  91 // -------------------------------------------------------- //
  92
  93
  94 void FlatMessageIO::xmlExportBegin(
  95         ExportContext& context) const {
  96
  97         context.beginElement(s_element);
  98 }
  99
 100 void FlatMessageIO::xmlExportAttributes(
 101         ExportContext& context) const {
 102
 103         context.writeAttr("encoding", "base64");
 104 }
 105
 106 void FlatMessageIO::xmlExportContent(
 107         ExportContext& context) const {
 108
 109         context.beginContent();
 110
 111         // convert message to base64
 112         ASSERT(m_message);
 113         ssize_t flatSize = m_message->FlattenedSize();
 114         ASSERT(flatSize);
 115         char* flatData = new char[flatSize];
 116         status_t err = m_message->Flatten(flatData, flatSize);
 117         ASSERT(err == B_OK);
 118
 119         // make plenty of room for encoded content (encode_base64 adds newlines)
 120         ssize_t base64Size = ((flatSize * 3) / 2);
 121         char* base64Data = new char[base64Size];
 122         ssize_t base64Used = encode_base64(base64Data, flatData, flatSize);
 123         base64Data[base64Used] = '\0';
 124
 125         // write the data
 126
 127         const char* pos = base64Data;
 128         while(*pos) {
 129                 ssize_t chunk = 0;
 130                 const char* nextBreak = strchr(pos, '\n');
 131                 if(!nextBreak)
 132                         chunk = strlen(pos);
 133                 else
 134                         chunk = nextBreak - pos;
 135
 136                 context.writeString(context.indentString());
 137                 context.writeString(pos, chunk);
 138                 context.writeString("\n");
 139
 140                 pos += chunk;
 141                 if(*pos == '\n')
 142                         ++pos;
 143         }
 144
 145         // clean up
 146         delete [] flatData;
 147         delete [] base64Data;
 148 }
 149
 150 void FlatMessageIO::xmlExportEnd(
 151         ExportContext& context) const {
 152         context.endElement();
 153 }
 154
 155 // -------------------------------------------------------- //
 156
 157 void FlatMessageIO::xmlImportBegin(
 158         ImportContext&          context) {
 159
 160         m_base64Data = "";
 161 }
 162
 163 void FlatMessageIO::xmlImportAttribute(
 164         const char*                                     key,
 165         const char*                                     value,
 166         ImportContext&          context) {
 167         if(!strcmp(key, "encoding")) {
 168                 if(strcmp(value, "base64") != 0)
 169                         context.reportError("Unexpected value of 'encoding'.");
 170         }
 171 }
 172
 173 void FlatMessageIO::xmlImportContent(
 174         const char*                                     data,
 175         uint32                                          length,
 176         ImportContext&          context) {
 177         m_base64Data.Append(data, length);
 178 }
 179
 180 void FlatMessageIO::xmlImportChild(
 181         IPersistent*                    child,
 182         ImportContext&          context) {
 183         delete child;
 184         context.reportError("Unexpected child object.");
 185 }
 186
 187 void FlatMessageIO::xmlImportComplete(
 188         ImportContext&          context) {
 189
 190         if(m_ownMessage && m_message)
 191                 delete m_message;
 192
 193         // decode the message
 194         ssize_t decodedSize = m_base64Data.Length();
 195         char* decodedData = new char[decodedSize];
 196         decodedSize = decode_base64(
 197                 decodedData, const_cast<char*>(m_base64Data.String()),
 198                 m_base64Data.Length(), false);
 199
 200         m_message = new BMessage();
 201         m_ownMessage = true;
 202
 203         status_t err = m_message->Unflatten(decodedData);
 204         if(err < B_OK) {
 205                 // decode failed; report error & clean up
 206                 BString error = "Unflatten(): ";
 207                 error << strerror(err);
 208                 context.reportError(error.String());
 209
 210                 delete m_message;
 211                 m_message = 0;
 212         }
 213
 214         m_base64Data = "";
 215         delete [] decodedData;
 216 }
 217
 218 // END -- FlatMessageIO.cpp --