diff options
Diffstat (limited to 'src/BtPortMessage.cc')
| -rw-r--r-- | src/BtPortMessage.cc | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/BtPortMessage.cc b/src/BtPortMessage.cc index 99b0d699..6a6c8d0f 100644 --- a/src/BtPortMessage.cc +++ b/src/BtPortMessage.cc @@ -100,7 +100,7 @@ void BtPortMessage::doReceivedAction() } } -unsigned char* BtPortMessage::createMessage() +std::vector<unsigned char> BtPortMessage::createMessage() { /** * len --- 5, 4bytes @@ -108,14 +108,12 @@ unsigned char* BtPortMessage::createMessage() * port --- port number, 2bytes * total: 7bytes */ - auto msg = new unsigned char[MESSAGE_LENGTH]; - bittorrent::createPeerMessageString(msg, MESSAGE_LENGTH, 3, ID); + auto msg = std::vector<unsigned char>(MESSAGE_LENGTH); + bittorrent::createPeerMessageString(msg.data(), MESSAGE_LENGTH, 3, ID); bittorrent::setShortIntParam(&msg[5], port_); return msg; } -size_t BtPortMessage::getMessageLength() { return MESSAGE_LENGTH; } - std::string BtPortMessage::toString() const { return fmt("%s port=%u", NAME, port_); |