add some testcases

testprof/base64_test.c

@@ -40,22 +40,24 @@ int base64_test(void)
          {"vuiSPKIl8PiR5O-rC4z9_xTQKZ0=", 1},
          {"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0", 0},
          {"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0=", 0},
+         {"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0==", 0},
+         {"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0===", 0},
+         {"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0====", 0},
+         {"vuiS*=PKIl8P*iR5O-rC4*z9_xTQKZ0=", 0},
+         {"vuiS*==PKIl8P*iR5O-rC4*z9_xTQKZ0=", 0},
+         {"vuiS*===PKIl8P*iR5O-rC4*z9_xTQKZ0=", 0},
    };
 
    for (x = 0; x < sizeof(cases)/sizeof(cases[0]); ++x) {
+       memset(out, 0, sizeof(out));
+       memset(tmp, 0, sizeof(tmp));
        slen1 = strlen(cases[x].s);
        l1 = sizeof(out);
        DO(base64_encode((unsigned char*)cases[x].s, slen1, out, &l1));
        l2 = sizeof(tmp);
-       DO(base64_decode(out, l1, tmp, &l2));
-       if (l2 != slen1 || l1 != strlen(cases[x].b64) || memcmp(tmp, cases[x].s, l2) || memcmp(out, cases[x].b64, l1)) {
-           fprintf(stderr, "\nbase64 failed case %lu", x);
-           fprintf(stderr, "\nbase64 should: %s", cases[x].b64);
-           out[sizeof(out)-1] = '\0';
-           fprintf(stderr, "\nbase64 is:     %s", out);
-           fprintf(stderr, "\nplain  should: %s", cases[x].s);
-           tmp[sizeof(tmp)-1] = '\0';
-           fprintf(stderr, "\nplain  is:     %s\n", tmp);
+       DO(base64_strict_decode(out, l1, tmp, &l2));
+       if (compare_testvector(out, l1, cases[x].b64, strlen(cases[x].b64), "base64 encode", x) ||
+             compare_testvector(tmp, l2, cases[x].s, slen1, "base64 decode", x)) {
            return 1;
        }
    }
@@ -67,15 +69,22 @@ int base64_test(void)
           DO(base64url_strict_decode((unsigned char*)url_cases[x].s, slen1, out, &l1));
        else
           DO(base64url_decode((unsigned char*)url_cases[x].s, slen1, out, &l1));
-       if (l1 != strlen(special_case) ||  memcmp(out, special_case, l1)) {
-           fprintf(stderr, "\nbase64url failed case %lu: %s", x, url_cases[x].s);
-           print_hex("\nbase64url should", special_case, strlen(special_case));
-           out[sizeof(out)-1] = '\0';
-           print_hex("\nbase64url is", out, l1);
+       if (compare_testvector(out, l1, special_case, strlen(special_case), "base64url decode", x)) {
            return 1;
        }
+       if(x < 2) {
+          l2 = sizeof(tmp);
+          if(x == 0)
+             DO(base64url_encode(out, l1, tmp, &l2));
+          else
+             DO(base64url_strict_encode(out, l1, tmp, &l2));
+          if (compare_testvector(tmp, l2, url_cases[x].s, strlen(url_cases[x].s), "base64url encode", x)) {
+              return 1;
+          }
+       }
    }
 
+   DO(base64url_strict_decode((unsigned char*)url_cases[4].s, slen1, out, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_INVALID_PACKET);
 
    for  (x = 0; x < 64; x++) {
        yarrow_read(in, x, &yarrow_prng);
@@ -83,22 +92,18 @@ int base64_test(void)
        DO(base64_encode(in, x, out, &l1));
        l2 = sizeof(tmp);
        DO(base64_decode(out, l1, tmp, &l2));
-       if (l2 != x || memcmp(tmp, in, x)) {
-           fprintf(stderr, "base64 failed %lu %lu %lu", x, l1, l2);
+       if (compare_testvector(tmp, x, in, x, "random base64", x)) {
            return 1;
        }
    }
 
    x--;
    memmove(&out[11], &out[10], l1 - 10);
-   out[10] = '\0';
+   out[10] = '=';
    l1++;
    l2 = sizeof(tmp);
    DO(base64_decode(out, l1, tmp, &l2));
-   if (l2 != x || memcmp(tmp, in, x)) {
-       fprintf(stderr, "relaxed base64 decoding failed %lu %lu %lu", x, l1, l2);
-       print_hex("is    ", tmp, l2);
-       print_hex("should", in, x);
+   if (compare_testvector(tmp, l2, in, l2, "relaxed base64 decoding", -1)) {
        print_hex("input ", out, l1);
        return 1;
    }