| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #include <activemq/commands/ActiveMQBytesMessage.h> |
| |
| #include <activemq/util/CMSExceptionSupport.h> |
| |
| #include <decaf/io/FilterOutputStream.h> |
| #include <decaf/io/ByteArrayInputStream.h> |
| #include <decaf/io/EOFException.h> |
| #include <decaf/io/IOException.h> |
| |
| #include <decaf/util/zip/DeflaterOutputStream.h> |
| #include <decaf/util/zip/InflaterInputStream.h> |
| |
| using namespace std; |
| using namespace activemq; |
| using namespace activemq::util; |
| using namespace activemq::commands; |
| using namespace activemq::exceptions; |
| using namespace decaf::io; |
| using namespace decaf::lang; |
| using namespace decaf::lang::exceptions; |
| using namespace decaf::util; |
| using namespace decaf::util::zip; |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| const unsigned char ActiveMQBytesMessage::ID_ACTIVEMQBYTESMESSAGE = 24; |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| namespace { |
| |
| class ByteCounterOutputStream : public FilterOutputStream { |
| private: |
| |
| int* length; |
| |
| private: |
| |
| ByteCounterOutputStream( const ByteCounterOutputStream& ); |
| ByteCounterOutputStream operator= ( const ByteCounterOutputStream& ); |
| |
| public: |
| |
| ByteCounterOutputStream( int* length, OutputStream* stream, bool own = false ) |
| : FilterOutputStream( stream, own ), length( length ) { |
| } |
| |
| virtual ~ByteCounterOutputStream() {} |
| |
| protected: |
| |
| virtual void doWriteByte( unsigned char value ) { |
| (*length)++; |
| FilterOutputStream::doWriteByte( value ); |
| } |
| |
| }; |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| ActiveMQBytesMessage::ActiveMQBytesMessage() : |
| ActiveMQMessageTemplate<cms::BytesMessage>(), bytesOut(NULL), dataIn(), dataOut(), length(0) { |
| |
| this->clearBody(); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| ActiveMQBytesMessage::~ActiveMQBytesMessage() throw() { |
| try { |
| this->reset(); |
| } |
| AMQ_CATCHALL_NOTHROW() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| unsigned char ActiveMQBytesMessage::getDataStructureType() const { |
| return ActiveMQBytesMessage::ID_ACTIVEMQBYTESMESSAGE; |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| ActiveMQBytesMessage* ActiveMQBytesMessage::cloneDataStructure() const { |
| std::auto_ptr<ActiveMQBytesMessage> message( new ActiveMQBytesMessage() ); |
| message->copyDataStructure( this ); |
| return message.release(); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| cms::BytesMessage* ActiveMQBytesMessage::clone() const { |
| ActiveMQBytesMessage* clone = this->cloneDataStructure(); |
| clone->setReadOnlyBody(false); |
| clone->setReadOnlyProperties(false); |
| return dynamic_cast<cms::BytesMessage*>(clone); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::copyDataStructure(const DataStructure* src) { |
| |
| // Protect against invalid self assignment. |
| if (this == src) { |
| return; |
| } |
| |
| const ActiveMQBytesMessage* srcPtr = dynamic_cast<const ActiveMQBytesMessage*>(src); |
| |
| if (srcPtr == NULL || src == NULL) { |
| throw decaf::lang::exceptions::NullPointerException( |
| __FILE__, __LINE__, |
| "ActiveMQBytesMessage::copyDataStructure - src is NULL or invalid"); |
| } |
| |
| ActiveMQBytesMessage* nonConstSrc = const_cast<ActiveMQBytesMessage*>(srcPtr); |
| nonConstSrc->storeContent(); |
| |
| ActiveMQMessageTemplate<cms::BytesMessage>::copyDataStructure(src); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| std::string ActiveMQBytesMessage::toString() const { |
| return ActiveMQMessageTemplate<cms::BytesMessage>::toString(); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| bool ActiveMQBytesMessage::equals(const DataStructure* value) const { |
| return ActiveMQMessageTemplate<cms::BytesMessage>::equals(value); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::setBodyBytes(const unsigned char* buffer, int numBytes) { |
| |
| try { |
| initializeWriting(); |
| dataOut->write(buffer, (int) numBytes, 0, (int) numBytes); |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| unsigned char* ActiveMQBytesMessage::getBodyBytes() const { |
| |
| try { |
| |
| initializeReading(); |
| |
| int length = this->getBodyLength(); |
| |
| if (length != 0) { |
| unsigned char* buffer = new unsigned char[length]; |
| this->readBytes(buffer, length); |
| return buffer; |
| } else { |
| return NULL; |
| } |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| int ActiveMQBytesMessage::getBodyLength() const { |
| |
| try { |
| initializeReading(); |
| return this->length; |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::clearBody() { |
| |
| // Invoke base class's version. |
| ActiveMQMessageTemplate<cms::BytesMessage>::clearBody(); |
| |
| this->dataOut.reset(NULL); |
| this->bytesOut = NULL; |
| this->dataIn.reset(NULL); |
| this->length = 0; |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::onSend() { |
| this->storeContent(); |
| ActiveMQMessageTemplate<cms::BytesMessage>::onSend(); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::reset() { |
| |
| try { |
| storeContent(); |
| this->bytesOut = NULL; |
| this->dataIn.reset(NULL); |
| this->dataOut.reset(NULL); |
| this->length = 0; |
| this->setReadOnlyBody(true); |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| bool ActiveMQBytesMessage::readBoolean() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readBoolean(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeBoolean(bool value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeBoolean(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| unsigned char ActiveMQBytesMessage::readByte() const { |
| |
| initializeReading(); |
| try { |
| return (unsigned char) this->dataIn->readByte(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeByte(unsigned char value) { |
| |
| initializeWriting(); |
| try{ |
| this->dataOut->writeByte( value ); |
| } catch( EOFException& ex ) { |
| throw CMSExceptionSupport::createMessageEOFException( ex ); |
| } catch( Exception& ex ) { |
| throw CMSExceptionSupport::create( ex ); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| int ActiveMQBytesMessage::readBytes(std::vector<unsigned char>& value) const { |
| return this->readBytes(&value[0], (int) value.size()); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeBytes(const std::vector<unsigned char>& value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->write(&value[0], (int) value.size()); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| int ActiveMQBytesMessage::readBytes(unsigned char* buffer, int length) const { |
| |
| initializeReading(); |
| try { |
| |
| if (length < 0) { |
| throw IndexOutOfBoundsException(__FILE__, __LINE__, "Array length given was negative"); |
| } |
| |
| int n = 0; |
| |
| while (n < length) { |
| int count = this->dataIn->read(buffer, (int) length, (int) n, (int) (length - n)); |
| if (count < 0) { |
| break; |
| } |
| n += count; |
| } |
| |
| if (n == 0 && length > 0) { |
| n = -1; |
| } |
| |
| return n; |
| |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeBytes(const unsigned char* value, int offset, int length) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->write(value, length, offset, length); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| char ActiveMQBytesMessage::readChar() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readChar(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeChar(char value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeChar(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| float ActiveMQBytesMessage::readFloat() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readFloat(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeFloat(float value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeFloat(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| double ActiveMQBytesMessage::readDouble() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readDouble(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeDouble(double value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeDouble(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| short ActiveMQBytesMessage::readShort() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readShort(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeShort(short value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeShort(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| unsigned short ActiveMQBytesMessage::readUnsignedShort() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readUnsignedShort(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeUnsignedShort(unsigned short value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeUnsignedShort(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| int ActiveMQBytesMessage::readInt() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readInt(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeInt(int value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeInt(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| long long ActiveMQBytesMessage::readLong() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readLong(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeLong(long long value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeLong(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| std::string ActiveMQBytesMessage::readString() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readString(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeString(const std::string& value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeChars(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| std::string ActiveMQBytesMessage::readUTF() const { |
| |
| initializeReading(); |
| try { |
| return this->dataIn->readUTF(); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::createMessageFormatException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::writeUTF(const std::string& value) { |
| |
| initializeWriting(); |
| try { |
| this->dataOut->writeUTF(value); |
| } catch (EOFException& ex) { |
| throw CMSExceptionSupport::createMessageEOFException(ex); |
| } catch (Exception& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::storeContent() { |
| |
| try { |
| |
| if (this->dataOut.get() != NULL && this->bytesOut->size() > 0) { |
| |
| this->dataOut->close(); |
| |
| if (!this->compressed) { |
| |
| std::pair<unsigned char*, int> array = this->bytesOut->toByteArray(); |
| this->setContent(std::vector<unsigned char>(array.first, array.first + array.second)); |
| delete[] array.first; |
| |
| } else { |
| |
| ByteArrayOutputStream buffer; |
| DataOutputStream doBuffer(&buffer); |
| |
| // Start by writing the length of the written data before compression. |
| doBuffer.writeInt((int) this->length); |
| |
| // Now write the Compressed bytes. |
| this->bytesOut->writeTo(&doBuffer); |
| |
| // Now store the annotated content. |
| std::pair<unsigned char*, int> array = buffer.toByteArray(); |
| this->setContent(std::vector<unsigned char>(array.first, array.first + array.second)); |
| delete[] array.first; |
| } |
| |
| this->dataOut.reset(NULL); |
| this->bytesOut = NULL; |
| } |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::initializeReading() const { |
| |
| this->failIfWriteOnlyBody(); |
| try { |
| |
| if (this->dataIn.get() == NULL) { |
| InputStream* is = new ByteArrayInputStream(this->getContent()); |
| |
| if (this->isCompressed()) { |
| |
| try { |
| DataInputStream dis(is); |
| this->length = dis.readInt(); |
| } catch (IOException& ex) { |
| throw CMSExceptionSupport::create(ex); |
| } |
| |
| is = new InflaterInputStream(is, true); |
| |
| } else { |
| this->length = (int) this->getContent().size(); |
| } |
| this->dataIn.reset(new DataInputStream(is, true)); |
| } |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void ActiveMQBytesMessage::initializeWriting() { |
| |
| this->failIfReadOnlyBody(); |
| try { |
| if (this->dataOut.get() == NULL) { |
| this->length = 0; |
| this->bytesOut = new ByteArrayOutputStream(); |
| |
| OutputStream* os = this->bytesOut; |
| |
| if (this->connection != NULL && this->connection->isUseCompression()) { |
| this->compressed = true; |
| |
| Deflater* deflator = new Deflater(this->connection->getCompressionLevel()); |
| |
| os = new DeflaterOutputStream(os, deflator, true, true); |
| os = new ByteCounterOutputStream(&length, os, true); |
| } |
| |
| this->dataOut.reset(new DataOutputStream(os, true)); |
| } |
| } |
| AMQ_CATCH_ALL_THROW_CMSEXCEPTION() |
| } |