heck
4 years ago
3 changed files with 260 additions and 185 deletions
@ -1,2 +1,208 @@ |
|||
#include "utils.hh" |
|||
|
|||
#include <pEp/pEpEngine.h> |
|||
#include <pEp/message_api.h> |
|||
#include <pEp/keymanagement.h> |
|||
#include <pEp/identity_list.h> |
|||
#include <pEp/Adapter.hh> |
|||
#include <pEp/status_to_string.hh> |
|||
#include <pEp/mime.h> |
|||
|
|||
namespace pEp { |
|||
namespace Test { |
|||
namespace Utils { |
|||
|
|||
//Ident
|
|||
pEpIdent wrap(::pEp_identity *const ident) |
|||
{ |
|||
assert(ident); |
|||
auto ret = pEpIdent(ident, [](::pEp_identity *) {}); |
|||
return ret; |
|||
} |
|||
|
|||
pEpIdent appropriate(::pEp_identity *const ident) |
|||
{ |
|||
assert(ident); |
|||
auto ret = pEpIdent(ident, ::free_identity); |
|||
return ret; |
|||
} |
|||
|
|||
pEpIdent dup(const ::pEp_identity *const ident) |
|||
{ |
|||
assert(ident); |
|||
auto ret = pEpIdent(::identity_dup(ident), ::free_identity); |
|||
return ret; |
|||
} |
|||
|
|||
pEpIdent kill(::pEp_identity *const ident) |
|||
{ |
|||
assert(ident); |
|||
auto ret = pEpIdent(::identity_dup(ident), ::free_identity); |
|||
::free_identity(ident); |
|||
return ret; |
|||
} |
|||
|
|||
//Message
|
|||
pEpMessage wrap(::message *const msg) |
|||
{ |
|||
assert(msg); |
|||
auto ret = pEpMessage(msg, [](::message *) {}); |
|||
return ret; |
|||
} |
|||
|
|||
pEpMessage appropriate(::message *const msg) |
|||
{ |
|||
assert(msg); |
|||
auto ret = pEpMessage(msg, ::free_message); |
|||
return ret; |
|||
} |
|||
|
|||
pEpMessage dup(const ::message *const msg) |
|||
{ |
|||
assert(msg); |
|||
auto ret = pEpMessage(::message_dup(msg), ::free_message); |
|||
return ret; |
|||
} |
|||
|
|||
pEpMessage kill(::message *const msg) |
|||
{ |
|||
assert(msg); |
|||
auto ret = pEpMessage(::message_dup(msg), ::free_message); |
|||
::free_message(msg); |
|||
return ret; |
|||
} |
|||
|
|||
// helpers
|
|||
pEpIdent createIdentity(const std::string &address, bool myself) |
|||
{ |
|||
std::string name; |
|||
std::string id; |
|||
::pEp_identity *partner = nullptr; |
|||
if (myself) { |
|||
partner = ::new_identity( |
|||
strdup(address.c_str()), |
|||
"", |
|||
PEP_OWN_USERID, |
|||
("myself " + address).c_str()); |
|||
partner->me = true; |
|||
} else { |
|||
partner = ::new_identity( |
|||
strdup(address.c_str()), |
|||
"", |
|||
"23", |
|||
("partner " + address).c_str()); |
|||
partner->me = false; |
|||
} |
|||
|
|||
return appropriate(partner); |
|||
} |
|||
|
|||
pEpMessage createMessage(pEpIdent from, pEpIdent to, const std::string &longmsg) |
|||
{ |
|||
// create and fill in msg
|
|||
::message *msg = ::new_message(PEP_dir_outgoing); |
|||
msg->from = ::identity_dup(from.get()); |
|||
msg->to = ::new_identity_list(::identity_dup(to.get())); |
|||
msg->longmsg = strdup(longmsg.c_str()); |
|||
|
|||
pEpMessage ret = appropriate(msg); |
|||
return ret; |
|||
} |
|||
|
|||
pEpMessage createMessage(pEpIdent from, const std::string &to_addr, const std::string &longmsg) |
|||
{ |
|||
pEpIdent to_ident = createIdentity(to_addr, false); |
|||
return createMessage(from, to_ident, longmsg); |
|||
} |
|||
|
|||
std::string mimeEncode(const pEpMessage msg) |
|||
{ |
|||
char *mimetext; |
|||
PEP_STATUS status = ::mime_encode_message(msg.get(), false, &mimetext, false); |
|||
throw_status(status); |
|||
std::string text{ mimetext }; |
|||
free(mimetext); |
|||
return text; |
|||
} |
|||
|
|||
pEpMessage mimeDecode(const std::string &mime_text) |
|||
{ |
|||
::message *msg; |
|||
bool has_possible_pEp_msg; |
|||
::PEP_STATUS status = ::mime_decode_message( |
|||
mime_text.c_str(), |
|||
mime_text.length(), |
|||
&msg, |
|||
&has_possible_pEp_msg); |
|||
throw_status(status); |
|||
return pEpMessage(msg, ::free_message); |
|||
} |
|||
|
|||
pEpMessage encryptMessage(const pEpMessage msg) |
|||
{ |
|||
pEpMessage ret; |
|||
::message *msgenc = nullptr; |
|||
PEP_STATUS status = ::encrypt_message( |
|||
Adapter::session(), |
|||
msg.get(), |
|||
nullptr, |
|||
&msgenc, |
|||
PEP_enc_PEP, |
|||
0); |
|||
throw_status(status); |
|||
::message *msg_out = nullptr; |
|||
if (msgenc != nullptr) { |
|||
ret = appropriate(msgenc); |
|||
} else { |
|||
ret = msg; |
|||
} |
|||
// unit.log("encryptAndSend: ENCRYPT - " + status_to_string(status));
|
|||
// unit.log("encryptAndSend" + Utils::to_string(msg_out, false));
|
|||
return ret; |
|||
} |
|||
|
|||
pEpMessage decryptMessage(const pEpMessage msg) |
|||
{ |
|||
pEpMessage ret; |
|||
|
|||
::message *dec{ nullptr }; |
|||
::stringlist_t *kl = ::new_stringlist(""); |
|||
PEP_rating rating; |
|||
unsigned int flags{ 0 }; |
|||
PEP_STATUS status = ::decrypt_message( |
|||
Adapter::session(), |
|||
msg.get(), |
|||
&dec, |
|||
&kl, |
|||
&rating, |
|||
&flags); |
|||
throw_status(status); |
|||
if (dec != nullptr) { |
|||
ret = appropriate(dec); |
|||
} else { |
|||
ret = msg; |
|||
// unit.log("NOT DECRYPTED");
|
|||
} |
|||
// unit.log("decryptMIME" + status_to_string(status));
|
|||
// unit.log("decryptMIME" + Utils::to_string(ret.get(), false));
|
|||
return ret; |
|||
} |
|||
|
|||
std::string encryptAndEncode(const pEpMessage msg) |
|||
{ |
|||
std::string ret; |
|||
pEpMessage msg_enc = encryptMessage(msg); |
|||
ret = mimeEncode(msg_enc); |
|||
return ret; |
|||
} |
|||
|
|||
pEpMessage decryptAndDecode(const std::string &mime_data) |
|||
{ |
|||
pEpMessage ret; |
|||
pEpMessage rx_msg = mimeDecode(mime_data); |
|||
ret = decryptMessage(rx_msg); |
|||
return ret; |
|||
} |
|||
} // namespace Utils
|
|||
} // namespace Test
|
|||
} // namespace pEp
|
|||
Loading…
Reference in new issue