eap::credentials_ttls::m_inner is std::unique_ptr managed now

(Also fixes a memory leak caused by missing destructor pointer delete)
This commit is contained in:
Simon Rozman 2016-07-20 11:47:40 +02:00
parent 427e2fb892
commit 07b4ce988b
2 changed files with 29 additions and 26 deletions

View File

@ -59,6 +59,8 @@ namespace eapserial
#include "../../TLS/include/Credentials.h" #include "../../TLS/include/Credentials.h"
#include "../../PAP/include/Credentials.h" #include "../../PAP/include/Credentials.h"
#include <memory>
namespace eap namespace eap
{ {
@ -181,7 +183,7 @@ namespace eap
/// @} /// @}
public: public:
credentials *m_inner; ///< Inner credentials std::unique_ptr<credentials> m_inner; ///< Inner credentials
}; };
} }
@ -192,27 +194,32 @@ namespace eapserial
{ {
pack(cursor, (const eap::credentials_tls&)val); pack(cursor, (const eap::credentials_tls&)val);
if (val.m_inner) { if (val.m_inner) {
if (dynamic_cast<eap::credentials_pap*>(val.m_inner)) { if (dynamic_cast<eap::credentials_pap*>(val.m_inner.get())) {
pack(cursor, (unsigned char)eap::type_pap); pack(cursor, eap::type_pap);
pack(cursor, (const eap::credentials_pap&)*val.m_inner); pack(cursor, (const eap::credentials_pap&)*val.m_inner);
} else { } else {
assert(0); // Unsupported inner authentication method type. assert(0); // Unsupported inner authentication method type.
pack(cursor, (unsigned char)0); pack(cursor, eap::type_undefined);
} }
} else } else
pack(cursor, (unsigned char)0); pack(cursor, eap::type_undefined);
} }
inline size_t get_pk_size(const eap::credentials_ttls &val) inline size_t get_pk_size(const eap::credentials_ttls &val)
{ {
size_t size_inner = sizeof(unsigned char); size_t size_inner;
if (val.m_inner) { if (val.m_inner) {
if (dynamic_cast<eap::credentials_pap*>(val.m_inner)) if (dynamic_cast<eap::credentials_pap*>(val.m_inner.get())) {
size_inner += get_pk_size((const eap::credentials_pap&)*val.m_inner); size_inner =
else get_pk_size(eap::type_pap) +
get_pk_size((const eap::credentials_pap&)*val.m_inner);
} else {
assert(0); // Unsupported inner authentication method type. assert(0); // Unsupported inner authentication method type.
} size_inner = get_pk_size(eap::type_undefined);
}
} else
size_inner = get_pk_size(eap::type_undefined);
return return
get_pk_size((const eap::credentials_tls&)val) + get_pk_size((const eap::credentials_tls&)val) +
@ -224,16 +231,16 @@ namespace eapserial
{ {
unpack(cursor, (eap::credentials_tls&)val); unpack(cursor, (eap::credentials_tls&)val);
assert(!val.m_inner); eap::type_t eap_type;
unsigned char eap_type;
unpack(cursor, eap_type); unpack(cursor, eap_type);
switch (eap_type) { switch (eap_type) {
case eap::type_pap: case eap::type_pap:
val.m_inner = new eap::credentials_pap(val.m_module); val.m_inner.reset(new eap::credentials_pap(val.m_module));
unpack(cursor, (eap::credentials_pap&)*val.m_inner); unpack(cursor, (eap::credentials_pap&)*val.m_inner);
break; break;
case 0 : break; default:
default : assert(0); // Unsupported inner authentication method type. assert(0); // Unsupported inner authentication method type.
val.m_inner.reset(nullptr);
} }
} }
} }

View File

@ -28,23 +28,23 @@ using namespace winstd;
// eap::credentials_ttls // eap::credentials_ttls
////////////////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////////
eap::credentials_ttls::credentials_ttls(_In_ module &mod) : credentials_tls(mod) eap::credentials_ttls::credentials_ttls(_In_ module &mod) :
credentials_tls(mod)
{ {
} }
eap::credentials_ttls::credentials_ttls(_In_ const credentials_ttls &other) : eap::credentials_ttls::credentials_ttls(_In_ const credentials_ttls &other) :
m_inner(other.m_inner ? (credentials*)other.m_inner->clone() : NULL), m_inner(other.m_inner ? (credentials*)other.m_inner->clone() : nullptr),
credentials_tls(other) credentials_tls(other)
{ {
} }
eap::credentials_ttls::credentials_ttls(_Inout_ credentials_ttls &&other) : eap::credentials_ttls::credentials_ttls(_Inout_ credentials_ttls &&other) :
m_inner(other.m_inner), m_inner(std::move(other.m_inner)),
credentials_tls(std::move(other)) credentials_tls(std::move(other))
{ {
other.m_inner = NULL;
} }
@ -52,9 +52,7 @@ eap::credentials_ttls& eap::credentials_ttls::operator=(_In_ const credentials_t
{ {
if (this != &other) { if (this != &other) {
(credentials_tls&)*this = other; (credentials_tls&)*this = other;
m_inner.reset(other.m_inner ? (credentials*)other.m_inner->clone() : nullptr);
if (m_inner) delete m_inner;
m_inner = other.m_inner ? (credentials*)other.m_inner->clone() : NULL;
} }
return *this; return *this;
@ -65,10 +63,7 @@ eap::credentials_ttls& eap::credentials_ttls::operator=(_Inout_ credentials_ttls
{ {
if (this != &other) { if (this != &other) {
(credentials_tls&)*this = std::move(other); (credentials_tls&)*this = std::move(other);
m_inner = std::move(other.m_inner);
if (m_inner) delete m_inner;
m_inner = other.m_inner;
other.m_inner = NULL;
} }
return *this; return *this;
@ -138,6 +133,7 @@ bool eap::credentials_ttls::load(_In_ IXMLDOMNode *pConfigRoot, _Out_ EAP_ERROR
if (!credentials_tls::load(pConfigRoot, ppEapError)) if (!credentials_tls::load(pConfigRoot, ppEapError))
return false; return false;
// TODO: For the time being, there is no detection what type is inner method. Introduce one!
if (m_inner) { if (m_inner) {
com_obj<IXMLDOMNode> pXmlElInnerAuthenticationMethod; com_obj<IXMLDOMNode> pXmlElInnerAuthenticationMethod;
if ((dwResult = eapxml::select_node(pConfigRoot, bstr(L"eap-metadata:InnerAuthenticationMethod"), &pXmlElInnerAuthenticationMethod)) != ERROR_SUCCESS) { if ((dwResult = eapxml::select_node(pConfigRoot, bstr(L"eap-metadata:InnerAuthenticationMethod"), &pXmlElInnerAuthenticationMethod)) != ERROR_SUCCESS) {