mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.7 KiB
C++
110 lines
2.7 KiB
C++
#include <llarp/api/messages.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace api
|
|
{
|
|
bool
|
|
IMessage::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
if(!BEncodeWriteDictString("F", FunctionName(), buf))
|
|
return false;
|
|
if(!BEncodeWriteDictInt(buf, "I", sessionID))
|
|
return false;
|
|
if(!BEncodeWriteDictInt(buf, "M", msgID))
|
|
return false;
|
|
if(!BEncodeWriteDictBEncodeList("P", GetParams(), buf))
|
|
return false;
|
|
if(!BEncodeWriteDictEntry("Z", hash, buf))
|
|
return false;
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
bool
|
|
IMessage::DecodeKey(llarp_buffer_t key, llarp_buffer_t* val)
|
|
{
|
|
if(llarp_buffer_eq(key, "P"))
|
|
{
|
|
return DecodeParams(val);
|
|
}
|
|
bool read = false;
|
|
if(!BEncodeMaybeReadDictInt("I", sessionID, read, key, val))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictInt("M", msgID, read, key, val))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictEntry("Z", hash, read, key, val))
|
|
return false;
|
|
return read;
|
|
}
|
|
|
|
bool
|
|
IMessage::IsWellFormed(llarp_crypto* crypto, const std::string& password)
|
|
{
|
|
// hash password
|
|
llarp::ShortHash secret;
|
|
llarp_buffer_t passbuf;
|
|
passbuf.base = (byte_t*)password.c_str();
|
|
passbuf.cur = passbuf.base;
|
|
passbuf.sz = password.size();
|
|
crypto->shorthash(secret, passbuf);
|
|
|
|
llarp::ShortHash digest, tmpHash;
|
|
// save hash
|
|
tmpHash = hash;
|
|
// zero hash
|
|
hash.Zero();
|
|
|
|
// bencode
|
|
byte_t tmp[1500];
|
|
llarp_buffer_t buf;
|
|
buf.base = tmp;
|
|
buf.cur = buf.base;
|
|
buf.sz = sizeof(tmp);
|
|
if(!BEncode(&buf))
|
|
return false;
|
|
|
|
// rewind buffer
|
|
buf.sz = buf.cur - buf.base;
|
|
buf.cur = buf.base;
|
|
// calculate message auth
|
|
crypto->hmac(digest, buf, secret);
|
|
// restore hash
|
|
hash = tmpHash;
|
|
return tmpHash == digest;
|
|
}
|
|
|
|
void
|
|
IMessage::CalculateHash(llarp_crypto* crypto, const std::string& password)
|
|
{
|
|
// hash password
|
|
llarp::ShortHash secret;
|
|
llarp_buffer_t passbuf;
|
|
passbuf.base = (byte_t*)password.c_str();
|
|
passbuf.cur = passbuf.base;
|
|
passbuf.sz = password.size();
|
|
crypto->shorthash(secret, passbuf);
|
|
|
|
//llarp::ShortHash digest;
|
|
// zero hash
|
|
hash.Zero();
|
|
|
|
// bencode
|
|
byte_t tmp[1500];
|
|
llarp_buffer_t buf;
|
|
buf.base = tmp;
|
|
buf.cur = buf.base;
|
|
buf.sz = sizeof(tmp);
|
|
if(BEncode(&buf))
|
|
{
|
|
// rewind buffer
|
|
buf.sz = buf.cur - buf.base;
|
|
buf.cur = buf.base;
|
|
// calculate message auth
|
|
crypto->hmac(hash, buf, secret);
|
|
}
|
|
}
|
|
} // namespace api
|
|
} // namespace llarp
|