Merge branch 'master' into dynamic-select-file

1 files changed, 3 insertions, 5 deletions

diff --git a/src/ZeroBtMessage.cc b/src/ZeroBtMessage.cc
index 36c92201..dbe10f94 100644
--- a/src/ZeroBtMessage.cc
+++ b/src/ZeroBtMessage.cc
@@ -42,20 +42,18 @@ ZeroBtMessage::ZeroBtMessage(uint8_t id, const char* name)
 {
 }
 
-unsigned char* ZeroBtMessage::createMessage()
+std::vector<unsigned char> ZeroBtMessage::createMessage()
 {
   /**
    * len --- 1, 4bytes
    * id --- ?, 1byte
    * total: 5bytes
    */
-  auto msg = new unsigned char[MESSAGE_LENGTH];
-  bittorrent::createPeerMessageString(msg, MESSAGE_LENGTH, 1, getId());
+  auto msg = std::vector<unsigned char>(MESSAGE_LENGTH);
+  bittorrent::createPeerMessageString(msg.data(), MESSAGE_LENGTH, 1, getId());
   return msg;
 }
 
-size_t ZeroBtMessage::getMessageLength() { return MESSAGE_LENGTH; }
-
 std::string ZeroBtMessage::toString() const { return getName(); }
 
 } // namespace aria2