Code clean-up

This commit is contained in:
Simon Rozman 2016-10-27 13:59:08 +02:00
parent 654c965851
commit 55f160bce3
4 changed files with 29 additions and 13 deletions

View File

@ -189,7 +189,7 @@ namespace eap
public:
///
/// Constructs an EAP method
/// Constructs a non-EAP method
///
/// \param[in] mod EAP module to use for global services
/// \param[in] cfg Method configuration
@ -198,14 +198,14 @@ namespace eap
method_noneap(_In_ module &module, _In_ config_method &cfg, _In_ credentials &cred);
///
/// Moves an EAP method
/// Moves a non-EAP method
///
/// \param[in] other EAP method to move from
///
method_noneap(_Inout_ method_noneap &&other);
///
/// Moves an EAP method
/// Moves a non-EAP method
///
/// \param[in] other EAP method to move from
///
@ -217,7 +217,7 @@ namespace eap
/// @{
///
/// Obtains a response packet from the EAP method.
/// Obtains a response packet from the non-EAP method.
///
/// \sa [EapPeerGetResponsePacket function](https://msdn.microsoft.com/en-us/library/windows/desktop/aa363610.aspx)
///
@ -236,7 +236,11 @@ namespace eap
/// \param[in] data AVP data (<16777212B)
/// \param[in] size Size of \p data in bytes
///
void append_avp(_In_ unsigned int code, _In_ unsigned char flags, _In_bytecount_(size) const void *data, _In_ unsigned int size);
void append_avp(
_In_ unsigned int code,
_In_ unsigned char flags,
_In_bytecount_(size) const void *data,
_In_ unsigned int size);
///
/// Appends Diameter AVP to response packet
@ -247,7 +251,12 @@ namespace eap
/// \param[in] data AVP data (<16777212B)
/// \param[in] size Size of \p data in bytes
///
void append_avp(_In_ unsigned int code, _In_ unsigned int vendor_id, _In_ unsigned char flags, _In_bytecount_(size) const void *data, _In_ unsigned int size);
void append_avp(
_In_ unsigned int code,
_In_ unsigned int vendor_id,
_In_ unsigned char flags,
_In_bytecount_(size) const void *data,
_In_ unsigned int size);
protected:
sanitizing_blob m_packet_res; ///< Response packet

View File

@ -198,7 +198,11 @@ void eap::method_noneap::get_response_packet(
}
void eap::method_noneap::append_avp(_In_ unsigned int code, _In_ unsigned char flags, _In_bytecount_(size) const void *data, _In_ unsigned int size)
void eap::method_noneap::append_avp(
_In_ unsigned int code,
_In_ unsigned char flags,
_In_bytecount_(size) const void *data,
_In_ unsigned int size)
{
unsigned int
padding = (unsigned int)((4 - size) % 4),
@ -224,7 +228,12 @@ void eap::method_noneap::append_avp(_In_ unsigned int code, _In_ unsigned char f
}
void eap::method_noneap::append_avp(_In_ unsigned int code, _In_ unsigned int vendor_id, _In_ unsigned char flags, _In_bytecount_(size) const void *data, _In_ unsigned int size)
void eap::method_noneap::append_avp(
_In_ unsigned int code,
_In_ unsigned int vendor_id,
_In_ unsigned char flags,
_In_bytecount_(size) const void *data,
_In_ unsigned int size)
{
unsigned int
padding = (unsigned int)((4 - size) % 4),

View File

@ -101,13 +101,13 @@ EapPeerMethodResponseAction eap::method_mschapv2::process_request_packet(
sanitizing_string identity_utf8;
WideCharToMultiByte(CP_UTF8, 0, m_cred.m_identity.c_str(), (int)m_cred.m_identity.length(), identity_utf8, NULL, NULL);
// Randomize Peer-Challenge
// Randomize Peer-Challenge.
m_challenge_client.randomize(m_cp);
// Calculate NT-Response
// Calculate NT-Response.
m_nt_resp = nt_response(m_cp, m_challenge_server, m_challenge_client, identity_utf8.c_str(), m_cred.m_password.c_str());
// Prepare MS-CHAP2-Response
// Prepare MS-CHAP2-Response.
sanitizing_blob response;
response.reserve(
1 + // Ident

View File

@ -89,8 +89,6 @@ EapPeerMethodResponseAction eap::method_pap::process_request_packet(
size_t padding_password_ex = (16 - password_utf8.length()) % 16;
password_utf8.append(padding_password_ex, 0);
m_packet_res.clear();
// Diameter AVP (User-Name=1, User-Password=2)
append_avp(1, diameter_avp_flag_mandatory, identity_utf8.data(), (unsigned int)identity_utf8.size());
append_avp(2, diameter_avp_flag_mandatory, password_utf8.data(), (unsigned int)password_utf8.size());