PRBool
test_atomtable()
{
nsrefcnt count = NS_GetNumberOfAtoms();
nsCOMPtr<nsIAtom> thirdNonPerm = do_GetAtom(THIRD_ATOM_STR);
if (isStaticAtom(thirdNonPerm))
return PR_FALSE;
if (!thirdNonPerm || NS_GetNumberOfAtoms() != count + 1)
return PR_FALSE;
NS_RegisterStaticAtoms(sAtoms_info, NS_ARRAY_LENGTH(sAtoms_info));
return sAtom1 &&
sAtom1->Equals(NS_LITERAL_STRING(FIRST_ATOM_STR)) &&
isStaticAtom(sAtom1) &&
sAtom2 &&
sAtom2->Equals(NS_LITERAL_STRING(SECOND_ATOM_STR)) &&
isStaticAtom(sAtom2) &&
sAtom3 &&
sAtom3->Equals(NS_LITERAL_STRING(THIRD_ATOM_STR)) &&
isStaticAtom(sAtom3) &&
NS_GetNumberOfAtoms() == count + 3 &&
thirdNonPerm == sAtom3;
}
PRBool
test_permanent()
{
nsrefcnt count = NS_GetNumberOfAtoms();
{
nsCOMPtr<nsIAtom> first = do_GetAtom(FIRST_PERM_ATOM_STR);
if (!first->Equals(NS_LITERAL_STRING(FIRST_PERM_ATOM_STR)) ||
isStaticAtom(first))
return PR_FALSE;
nsCOMPtr<nsIAtom> first_p =
NS_NewPermanentAtom(NS_LITERAL_STRING(FIRST_PERM_ATOM_STR));
if (!first_p->Equals(NS_LITERAL_STRING(FIRST_PERM_ATOM_STR)) ||
!isStaticAtom(first_p) ||
first != first_p)
return PR_FALSE;
nsCOMPtr<nsIAtom> second_p =
NS_NewPermanentAtom(NS_LITERAL_STRING(SECOND_PERM_ATOM_STR));
if (!second_p->Equals(NS_LITERAL_STRING(SECOND_PERM_ATOM_STR)) ||
!isStaticAtom(second_p))
return PR_FALSE;
nsCOMPtr<nsIAtom> second = do_GetAtom(SECOND_PERM_ATOM_STR);
if (!second->Equals(NS_LITERAL_STRING(SECOND_PERM_ATOM_STR)) ||
!isStaticAtom(second) ||
second != second_p)
return PR_FALSE;
}
return NS_GetNumberOfAtoms() == count + 2;
}
PRBool
test_invalid()
{
for (unsigned int i = 0; i < NS_ARRAY_LENGTH(Invalid16Strings); ++i) {
nsrefcnt count = NS_GetNumberOfAtoms();
{
nsCOMPtr<nsIAtom> atom16 = do_GetAtom(Invalid16Strings[i].m16);
if (!atom16->Equals(nsDependentString(Invalid16Strings[i].m16)))
return PR_FALSE;
}
if (count != NS_GetNumberOfAtoms())
return PR_FALSE;
}
for (unsigned int i = 0; i < NS_ARRAY_LENGTH(Invalid8Strings); ++i) {
nsrefcnt count = NS_GetNumberOfAtoms();
{
nsCOMPtr<nsIAtom> atom8 = do_GetAtom(Invalid8Strings[i].m8);
nsCOMPtr<nsIAtom> atom16 = do_GetAtom(Invalid8Strings[i].m16);
if (atom16 != atom8 ||
!atom16->Equals(nsDependentString(Invalid8Strings[i].m16)))
return PR_FALSE;
}
if (count != NS_GetNumberOfAtoms())
return PR_FALSE;
}
// Don't run this test in debug builds as that intentionally asserts.
#ifndef DEBUG
nsCOMPtr<nsIAtom> emptyAtom = do_GetAtom("");
for (unsigned int i = 0; i < NS_ARRAY_LENGTH(Malformed8Strings); ++i) {
nsrefcnt count = NS_GetNumberOfAtoms();
nsCOMPtr<nsIAtom> atom8 = do_GetAtom(Malformed8Strings[i]);
if (atom8 != emptyAtom ||
count != NS_GetNumberOfAtoms())
return PR_FALSE;
}
#endif
return PR_TRUE;
}
int main(int argc, char** argv)
{
FILE* fp = fopen("words.txt", "r");
if (nsnull == fp) {
printf("can't open words.txt\n");
return -1;
}
PRInt32 count = 0;
PRUnichar** strings = new PRUnichar*[60000];
nsIAtom** ids = new nsIAtom*[60000];
nsAutoString s1, s2;
PRTime start = PR_Now();
PRInt32 i;
for (i = 0; i < 60000; i++) {
char buf[1000];
char* s = fgets(buf, sizeof(buf), fp);
if (nsnull == s) {
break;
}
nsCAutoString sb;
sb.Assign(buf);
strings[count++] = ToNewUnicode(sb);
ToUpperCase(sb);
strings[count++] = ToNewUnicode(sb);
}
PRTime end0 = PR_Now();
// Find and create idents
for (i = 0; i < count; i++) {
ids[i] = NS_NewAtom(strings[i]);
}
PRUnichar qqs[1]; qqs[0] = 0;
nsIAtom* qq = NS_NewAtom(qqs);
PRTime end1 = PR_Now();
// Now make sure we can find all the idents we just made
for (i = 0; i < count; i++) {
const char *utf8String;
ids[i]->GetUTF8String(&utf8String);
nsIAtom* id = NS_NewAtom(utf8String);
if (id != ids[i]) {
id->ToString(s1);
ids[i]->ToString(s2);
printf("find failed: id='%s' ids[%d]='%s'\n",
NS_LossyConvertUTF16toASCII(s1).get(), i, NS_LossyConvertUTF16toASCII(s2).get());
return -1;
}
NS_RELEASE(id);
}
PRTime end2 = PR_Now();
// Destroy all the atoms we just made
NS_RELEASE(qq);
for (i = 0; i < count; i++) {
NS_RELEASE(ids[i]);
}
// Print out timings
PRTime end3 = PR_Now();
PRTime creates, finds, lookups, dtor, ustoms;
LL_I2L(ustoms, 1000);
LL_SUB(creates, end0, start);
LL_DIV(creates, creates, ustoms);
LL_SUB(finds, end1, end0);
LL_DIV(finds, finds, ustoms);
LL_SUB(lookups, end2, end1);
LL_DIV(lookups, lookups, ustoms);
LL_SUB(dtor, end3, end2);
char buf[500];
PR_snprintf(buf, sizeof(buf), "making %d ident strings took %lldms",
count, creates);
puts(buf);
PR_snprintf(buf, sizeof(buf), "%d new idents took %lldms",
count, finds);
puts(buf);
PR_snprintf(buf, sizeof(buf), "%d ident lookups took %lldms",
count, lookups);
puts(buf);
PR_snprintf(buf, sizeof(buf), "dtor took %lldusec", dtor);
puts(buf);
printf("%d live atoms\n", NS_GetNumberOfAtoms());
NS_POSTCONDITION(0 == NS_GetNumberOfAtoms(), "dangling atoms");
return 0;
}