diff options
Diffstat (limited to 'src/ZeroBtMessage.cc')
| -rw-r--r-- | src/ZeroBtMessage.cc | 8 |
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 |