diff options
author | Jens Geyer <jensg@apache.org> | 2021-01-28 08:48:24 +0100 |
---|---|---|
committer | Mario Emmenlauer <memmenlauer@biodataanalysis.de> | 2021-02-03 13:35:36 +0100 |
commit | 1f73455940c440cd9b4226c92e01ae0f0378705c (patch) | |
tree | 68315a0737e3bd6bc68c04d296a6a6b7d9155721 | |
parent | 0f21e39c9ba1b20a50d035f01c14836885678d08 (diff) | |
download | thrift-1f73455940c440cd9b4226c92e01ae0f0378705c.tar.gz |
THRIFT-5237 Implement MAX_MESSAGE_SIZE and consolidate limits into a TConfiguration class (cpp)
Client: cpp
Patch: Jens Geyer
Fixes an issue introduced with that ticket
-rw-r--r-- | lib/cpp/src/thrift/transport/TTransport.h | 6 | ||||
-rw-r--r-- | lib/cpp/test/ThrifttReadCheckTests.cpp | 36 |
2 files changed, 21 insertions, 21 deletions
diff --git a/lib/cpp/src/thrift/transport/TTransport.h b/lib/cpp/src/thrift/transport/TTransport.h index 5f657f89b..52b3a0a4d 100644 --- a/lib/cpp/src/thrift/transport/TTransport.h +++ b/lib/cpp/src/thrift/transport/TTransport.h @@ -275,7 +275,7 @@ public: void checkReadBytesAvailable(long int numBytes) { if (remainingMessageSize_ < numBytes) - throw new TTransportException(TTransportException::END_OF_FILE, "MaxMessageSize reached"); + throw TTransportException(TTransportException::END_OF_FILE, "MaxMessageSize reached"); } protected: @@ -306,7 +306,7 @@ protected: // update only: message size can shrink, but not grow if (newSize > knownMessageSize_) - throw new TTransportException(TTransportException::END_OF_FILE, "MaxMessageSize reached"); + throw TTransportException(TTransportException::END_OF_FILE, "MaxMessageSize reached"); knownMessageSize_ = newSize; remainingMessageSize_ = newSize; @@ -326,7 +326,7 @@ protected: else { remainingMessageSize_ = 0; - throw new TTransportException(TTransportException::END_OF_FILE, "MaxMessageSize reached"); + throw TTransportException(TTransportException::END_OF_FILE, "MaxMessageSize reached"); } } }; diff --git a/lib/cpp/test/ThrifttReadCheckTests.cpp b/lib/cpp/test/ThrifttReadCheckTests.cpp index 4a594e6ca..eb4ca01b2 100644 --- a/lib/cpp/test/ThrifttReadCheckTests.cpp +++ b/lib/cpp/test/ThrifttReadCheckTests.cpp @@ -72,7 +72,7 @@ BOOST_AUTO_TEST_CASE(test_tmemorybuffer_read_check_exception) { TMemoryBuffer trans_in(config); memset(buffer, 0, sizeof(buffer)); - BOOST_CHECK_THROW(trans_in.read(buffer, sizeof(buffer)), TTransportException*); + BOOST_CHECK_THROW(trans_in.read(buffer, sizeof(buffer)), TTransportException); trans_in.close(); } @@ -85,13 +85,13 @@ BOOST_AUTO_TEST_CASE(test_tpipedtransport_read_check_exception) { uint8_t buffer[4]; underlying->write((uint8_t*)"abcd", 4); - BOOST_CHECK_THROW(trans->read(buffer, sizeof(buffer)), TTransportException*); - BOOST_CHECK_THROW(trans->readAll(buffer, sizeof(buffer)), TTransportException*); + BOOST_CHECK_THROW(trans->read(buffer, sizeof(buffer)), TTransportException); + BOOST_CHECK_THROW(trans->readAll(buffer, sizeof(buffer)), TTransportException); trans->readEnd(); pipe->resetBuffer(); underlying->write((uint8_t*)"ef", 2); - BOOST_CHECK_THROW(trans->read(buffer, sizeof(buffer)), TTransportException*); - BOOST_CHECK_THROW(trans->readAll(buffer, sizeof(buffer)), TTransportException*); + BOOST_CHECK_THROW(trans->read(buffer, sizeof(buffer)), TTransportException); + BOOST_CHECK_THROW(trans->readAll(buffer, sizeof(buffer)), TTransportException); trans->readEnd(); } @@ -104,7 +104,7 @@ BOOST_AUTO_TEST_CASE(test_tsimplefiletransport_read_check_exception) { TSimpleFileTransport trans_in("data",true, false, config); memset(buffer, 0, sizeof(buffer)); - BOOST_CHECK_THROW(trans_in.read(buffer, sizeof(buffer)), TTransportException*); + BOOST_CHECK_THROW(trans_in.read(buffer, sizeof(buffer)), TTransportException); trans_in.close(); remove("./data"); @@ -118,7 +118,7 @@ BOOST_AUTO_TEST_CASE(test_tfiletransport_read_check_exception) { TFileTransport trans_in("data", false, config); memset(buffer, 0, sizeof(buffer)); - BOOST_CHECK_THROW(trans_in.read(buffer, sizeof(buffer)), TTransportException*); + BOOST_CHECK_THROW(trans_in.read(buffer, sizeof(buffer)), TTransportException); remove("./data"); } @@ -130,7 +130,7 @@ BOOST_AUTO_TEST_CASE(test_tbufferedtransport_read_check_exception) { std::shared_ptr<TBufferedTransport> trans (new TBufferedTransport(buffer, config)); trans->write((const uint8_t*)arr, sizeof(arr)); - BOOST_CHECK_THROW(trans->read(arr, sizeof(arr)), TTransportException*); + BOOST_CHECK_THROW(trans->read(arr, sizeof(arr)), TTransportException); } BOOST_AUTO_TEST_CASE(test_tframedtransport_read_check_exception) { @@ -140,7 +140,7 @@ BOOST_AUTO_TEST_CASE(test_tframedtransport_read_check_exception) { std::shared_ptr<TFramedTransport> trans (new TFramedTransport(buffer, config)); trans->write((const uint8_t*)arr, sizeof(arr)); - BOOST_CHECK_THROW(trans->read(arr, sizeof(arr)), TTransportException*); + BOOST_CHECK_THROW(trans->read(arr, sizeof(arr)), TTransportException); } BOOST_AUTO_TEST_CASE(test_tthriftbinaryprotocol_read_check_exception) { @@ -154,19 +154,19 @@ BOOST_AUTO_TEST_CASE(test_tthriftbinaryprotocol_read_check_exception) { TList list(T_I32, 8); protocol->writeListBegin(list.elemType_, list.size_); protocol->writeListEnd(); - BOOST_CHECK_THROW(protocol->readListBegin(elemType, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readListBegin(elemType, val), TTransportException); protocol->readListEnd(); TSet set(T_I32, 8); protocol->writeSetBegin(set.elemType_, set.size_); protocol->writeSetEnd(); - BOOST_CHECK_THROW(protocol->readSetBegin(elemType, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readSetBegin(elemType, val), TTransportException); protocol->readSetEnd(); TMap map(T_I32, T_I32, 8); protocol->writeMapBegin(map.keyType_, map.valueType_, map.size_); protocol->writeMapEnd(); - BOOST_CHECK_THROW(protocol->readMapBegin(elemType, elemType1, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readMapBegin(elemType, elemType1, val), TTransportException); protocol->readMapEnd(); } @@ -181,19 +181,19 @@ BOOST_AUTO_TEST_CASE(test_tthriftcompactprotocol_read_check_exception) { TList list(T_I32, 8); protocol->writeListBegin(list.elemType_, list.size_); protocol->writeListEnd(); - BOOST_CHECK_THROW(protocol->readListBegin(elemType, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readListBegin(elemType, val), TTransportException); protocol->readListEnd(); TSet set(T_I32, 8); protocol->writeSetBegin(set.elemType_, set.size_); protocol->writeSetEnd(); - BOOST_CHECK_THROW(protocol->readSetBegin(elemType, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readSetBegin(elemType, val), TTransportException); protocol->readSetEnd(); TMap map(T_I32, T_I32, 8); protocol->writeMapBegin(map.keyType_, map.valueType_, map.size_); protocol->writeMapEnd(); - BOOST_CHECK_THROW(protocol->readMapBegin(elemType, elemType1, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readMapBegin(elemType, elemType1, val), TTransportException); protocol->readMapEnd(); } @@ -208,19 +208,19 @@ BOOST_AUTO_TEST_CASE(test_tthriftjsonprotocol_read_check_exception) { TList list(T_I32, 8); protocol->writeListBegin(list.elemType_, list.size_); protocol->writeListEnd(); - BOOST_CHECK_THROW(protocol->readListBegin(elemType, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readListBegin(elemType, val), TTransportException); protocol->readListEnd(); TSet set(T_I32, 8); protocol->writeSetBegin(set.elemType_, set.size_); protocol->writeSetEnd(); - BOOST_CHECK_THROW(protocol->readSetBegin(elemType, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readSetBegin(elemType, val), TTransportException); protocol->readSetEnd(); TMap map(T_I32, T_I32, 8); protocol->writeMapBegin(map.keyType_, map.valueType_, map.size_); protocol->writeMapEnd(); - BOOST_CHECK_THROW(protocol->readMapBegin(elemType, elemType1, val), TTransportException*); + BOOST_CHECK_THROW(protocol->readMapBegin(elemType, elemType1, val), TTransportException); protocol->readMapEnd(); } |