netwerk/test/gtest/TestIDNA.cpp

   1 #include "gtest/gtest.h"
   2 #include "gtest/MozGTestBench.h"  // For MOZ_GTEST_BENCH
   3 #include "gtest/BlackBox.h"
   4
   5 #include "nsNetUtil.h"
   6 #include "nsIURI.h"
   7 #include "nsCOMPtr.h"
   8 #include "mozilla/Encoding.h"
   9
  10 #define TEST_COUNT 10000
  11
  12 class TestIDNA : public ::testing::Test {
  13  protected:
  14   void SetUp() override {
  15     // Intentionally Assign and not AssignLiteral
  16     // to simulate the heap case.
  17     mPlainASCII.Assign("example.com");
  18     mLeadingDigitASCII.Assign("1test.example");
  19     mUnicodeMixed.Assign("مثال.example");
  20     mPunycodeMixed.Assign("xn--mgbh0fb.example");
  21     mUnicodeLTR.Assign("නම.උදාහරණ");
  22     mPunycodeLTR.Assign("xn--r0co.xn--ozc8dl2c3bxd");
  23     mUnicodeRTL.Assign("الاسم.مثال");
  24     mPunycodeRTL.Assign("xn--mgba0b1dh.xn--mgbh0fb");
  25     // Intentionally not assigning to mEmpty
  26
  27     // For measuring the case inside nsStandardURL
  28     mUrlPlainASCII.Assign("https://example.com/");
  29     mUrlLeadingDigitASCII.Assign("https://1test.example/");
  30     mUrlUnicodeMixed.Assign("https://مثال.example/");
  31     mUrlPunycodeMixed.Assign("https://xn--mgbh0fb.example/");
  32     mUrlUnicodeLTR.Assign("https://නම.උදාහරණ/");
  33     mUrlPunycodeLTR.Assign("https://xn--r0co.xn--ozc8dl2c3bxd/");
  34     mUrlUnicodeRTL.Assign("https://الاسم.مثال/");
  35     mUrlPunycodeRTL.Assign("https://xn--mgba0b1dh.xn--mgbh0fb/");
  36   }
  37
  38  public:
  39   nsCString mPlainASCII;
  40   nsCString mLeadingDigitASCII;
  41   nsCString mUnicodeMixed;
  42   nsCString mPunycodeMixed;
  43   nsCString mUnicodeLTR;
  44   nsCString mPunycodeLTR;
  45   nsCString mUnicodeRTL;
  46   nsCString mPunycodeRTL;
  47   nsCString mEmpty;  // Extremely suspicious measurement!
  48   nsCString mUrlPlainASCII;
  49   nsCString mUrlLeadingDigitASCII;
  50   nsCString mUrlUnicodeMixed;
  51   nsCString mUrlPunycodeMixed;
  52   nsCString mUrlUnicodeLTR;
  53   nsCString mUrlPunycodeLTR;
  54   nsCString mUrlUnicodeRTL;
  55   nsCString mUrlPunycodeRTL;
  56 };
  57
  58 #define IDNA_ITERATIONS 50000
  59
  60 #define IDNA_BENCH(name, func, src)                             \
  61   MOZ_GTEST_BENCH_F(TestIDNA, name, [this] {                    \
  62     for (int i = 0; i < IDNA_ITERATIONS; i++) {                 \
  63       nsCString dst;                                            \
  64       func(*mozilla::BlackBox(&src), *mozilla::BlackBox(&dst)); \
  65     }                                                           \
  66   });
  67
  68 #define IDNA_URL_BENCH(name, src)                                            \
  69   MOZ_GTEST_BENCH_F(TestIDNA, name, [this] {                                 \
  70     for (int i = 0; i < IDNA_ITERATIONS; i++) {                              \
  71       nsCOMPtr<nsIURI> dst;                                                  \
  72       nsresult rv = NS_NewURI(getter_AddRefs(dst), *mozilla::BlackBox(&src), \
  73                               UTF_8_ENCODING);                               \
  74       ASSERT_EQ(NS_OK, rv);                                                  \
  75       mozilla::BlackBox(&dst);                                               \
  76     }                                                                        \
  77   });
  78
  79 IDNA_BENCH(BenchToASCIIPlainASCII, NS_DomainToASCII, mPlainASCII);
  80 IDNA_BENCH(BenchToASCIILeadingDigitASCII, NS_DomainToASCII, mLeadingDigitASCII);
  81 IDNA_BENCH(BenchToASCIIUnicodeMixed, NS_DomainToASCII, mUnicodeMixed);
  82 IDNA_BENCH(BenchToASCIIPunycodeMixed, NS_DomainToASCII, mPunycodeMixed);
  83 IDNA_BENCH(BenchToASCIIUnicodeLTR, NS_DomainToASCII, mUnicodeLTR);
  84 IDNA_BENCH(BenchToASCIIPunycodeLTR, NS_DomainToASCII, mPunycodeLTR);
  85 IDNA_BENCH(BenchToASCIIUnicodeRTL, NS_DomainToASCII, mUnicodeRTL);
  86 IDNA_BENCH(BenchToASCIIPunycodeRTL, NS_DomainToASCII, mPunycodeRTL);
  87 IDNA_BENCH(BenchToASCIIEmpty, NS_DomainToASCII, mEmpty);
  88
  89 IDNA_BENCH(BenchToDisplayPlainASCII, NS_DomainToDisplay, mPlainASCII);
  90 IDNA_BENCH(BenchToDisplayLeadingDigitASCII, NS_DomainToDisplay,
  91            mLeadingDigitASCII);
  92 IDNA_BENCH(BenchToDisplayUnicodeMixed, NS_DomainToDisplay, mUnicodeMixed);
  93 IDNA_BENCH(BenchToDisplayPunycodeMixed, NS_DomainToDisplay, mPunycodeMixed);
  94 IDNA_BENCH(BenchToDisplayUnicodeLTR, NS_DomainToDisplay, mUnicodeLTR);
  95 IDNA_BENCH(BenchToDisplayPunycodeLTR, NS_DomainToDisplay, mPunycodeLTR);
  96 IDNA_BENCH(BenchToDisplayUnicodeRTL, NS_DomainToDisplay, mUnicodeRTL);
  97 IDNA_BENCH(BenchToDisplayPunycodeRTL, NS_DomainToDisplay, mPunycodeRTL);
  98 IDNA_BENCH(BenchToDisplayEmpty, NS_DomainToDisplay, mEmpty);
  99
 100 IDNA_BENCH(BenchToUnicodePlainASCII, NS_DomainToUnicode, mPlainASCII);
 101 IDNA_BENCH(BenchToUnicodeLeadingDigitASCII, NS_DomainToUnicode,
 102            mLeadingDigitASCII);
 103 IDNA_BENCH(BenchToUnicodeUnicodeMixed, NS_DomainToUnicode, mUnicodeMixed);
 104 IDNA_BENCH(BenchToUnicodePunycodeMixed, NS_DomainToUnicode, mPunycodeMixed);
 105 IDNA_BENCH(BenchToUnicodeUnicodeLTR, NS_DomainToUnicode, mUnicodeLTR);
 106 IDNA_BENCH(BenchToUnicodePunycodeLTR, NS_DomainToUnicode, mPunycodeLTR);
 107 IDNA_BENCH(BenchToUnicodeUnicodeRTL, NS_DomainToUnicode, mUnicodeRTL);
 108 IDNA_BENCH(BenchToUnicodePunycodeRTL, NS_DomainToUnicode, mPunycodeRTL);
 109 IDNA_BENCH(BenchToUnicodeEmpty, NS_DomainToUnicode, mEmpty);
 110
 111 IDNA_URL_BENCH(BenchUrlPlainASCII, mUrlPlainASCII);
 112 IDNA_URL_BENCH(BenchUrlLeadingDigitASCII, mUrlLeadingDigitASCII);
 113 IDNA_URL_BENCH(BenchUrlUnicodeMixed, mUrlUnicodeMixed);
 114 IDNA_URL_BENCH(BenchUrlPunycodeMixed, mUrlPunycodeMixed);
 115 IDNA_URL_BENCH(BenchUrlUnicodeLTR, mUrlUnicodeLTR);
 116 IDNA_URL_BENCH(BenchUrlPunycodeLTR, mUrlPunycodeLTR);
 117 IDNA_URL_BENCH(BenchUrlUnicodeRTL, mUrlUnicodeRTL);
 118 IDNA_URL_BENCH(BenchUrlPunycodeRTL, mUrlPunycodeRTL);