Merge branch 'refactor-bt-msg'

1 files changed, 3 insertions, 5 deletions

diff --git a/src/BtHandshakeMessage.cc b/src/BtHandshakeMessage.cc
index 95f99e0d..ce6023e7 100644
--- a/src/BtHandshakeMessage.cc
+++ b/src/BtHandshakeMessage.cc
@@ -81,10 +81,10 @@ BtHandshakeMessage::create(const unsigned char* data, size_t dataLength)
   return msg;
 }
 
-unsigned char* BtHandshakeMessage::createMessage()
+std::vector<unsigned char> BtHandshakeMessage::createMessage()
 {
-  auto msg = new unsigned char[MESSAGE_LENGTH];
-  auto dst = msg;
+  auto msg = std::vector<unsigned char>(MESSAGE_LENGTH);
+  auto dst = msg.data();
   *dst++ = pstrlen_;
   dst = std::copy(std::begin(pstr_), std::end(pstr_), dst);
   dst = std::copy(std::begin(reserved_), std::end(reserved_), dst);
@@ -93,8 +93,6 @@ unsigned char* BtHandshakeMessage::createMessage()
   return msg;
 }
 
-size_t BtHandshakeMessage::getMessageLength() { return MESSAGE_LENGTH; }
-
 std::string BtHandshakeMessage::toString() const
 {
   return fmt("%s peerId=%s, reserved=%s", NAME,