Source code

Revision control

Copy as Markdown

Other Tools

#include <combaseapi.h>
#include "gtest/gtest.h"
#include "mozilla/SHA1.h"
#include "nsNotifyAddrListener.h"
using namespace mozilla;
GUID StringToGuid(const std::string& str) {
GUID guid;
sscanf(str.c_str(),
"%8lx-%4hx-%4hx-%2hhx%2hhx-%2hhx%2hhx%2hhx%2hhx%2hhx%2hhx",
&guid.Data1, &guid.Data2, &guid.Data3, &guid.Data4[0], &guid.Data4[1],
&guid.Data4[2], &guid.Data4[3], &guid.Data4[4], &guid.Data4[5],
&guid.Data4[6], &guid.Data4[7]);
return guid;
}
TEST(TestGuidHashWindows, Single)
{
// Setup
SHA1Sum expected_sha1;
SHA1Sum::Hash expected_digest;
GUID g1 = StringToGuid("264555b1-289c-4494-83d1-e158d1d95115");
expected_sha1.update(&g1, sizeof(GUID));
expected_sha1.finish(expected_digest);
std::vector<GUID> nwGUIDS;
nwGUIDS.push_back(g1);
SHA1Sum actual_sha1;
// Run
nsNotifyAddrListener::HashSortedNetworkIds(nwGUIDS, actual_sha1);
SHA1Sum::Hash actual_digest;
actual_sha1.finish(actual_digest);
// Assert
ASSERT_EQ(0, memcmp(&expected_digest, &actual_digest, sizeof(SHA1Sum::Hash)));
}
TEST(TestNetworkLinkIdHashingWindows, Multiple)
{
// Setup
SHA1Sum expected_sha1;
SHA1Sum::Hash expected_digest;
std::vector<GUID> nwGUIDS;
nwGUIDS.push_back(StringToGuid("00000000-0000-0000-0000-000000000001"));
nwGUIDS.push_back(StringToGuid("00000000-0000-0000-0000-000000000002"));
nwGUIDS.push_back(StringToGuid("00000000-0000-0000-0000-000000000003"));
nwGUIDS.push_back(StringToGuid("00000000-0000-0000-0000-000000000004"));
for (const auto& guid : nwGUIDS) {
expected_sha1.update(&guid, sizeof(GUID));
}
expected_sha1.finish(expected_digest);
// Ordered
std::vector<GUID> ordered;
for (const auto& guid : nwGUIDS) {
ordered.push_back(guid);
}
SHA1Sum ordered_sha1;
// Unordered
std::vector<GUID> reversed;
for (auto it = nwGUIDS.rbegin(); it != nwGUIDS.rend(); ++it) {
reversed.push_back(*it);
}
SHA1Sum reversed_sha1;
// Run
nsNotifyAddrListener::HashSortedNetworkIds(ordered, ordered_sha1);
SHA1Sum::Hash ordered_digest;
ordered_sha1.finish(ordered_digest);
nsNotifyAddrListener::HashSortedNetworkIds(reversed, reversed_sha1);
SHA1Sum::Hash reversed_digest;
reversed_sha1.finish(reversed_digest);
// Assert
ASSERT_EQ(0,
memcmp(&expected_digest, &ordered_digest, sizeof(SHA1Sum::Hash)));
ASSERT_EQ(0,
memcmp(&expected_digest, &reversed_digest, sizeof(SHA1Sum::Hash)));
}