text: /* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
  /*
   * Copyright 2014 by the Massachusetts Institute of Technology.
   * All Rights Reserved.
   *
   * Export of this software from the United States of America may
   *   require a specific license from the United States Government.
   *   It is the responsibility of any person or organization contemplating
   *   export to obtain such a license before exporting.
   *
   * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
   * distribute this software and its documentation for any purpose and
   * without fee is hereby granted, provided that the above copyright
   * notice appear in all copies and that both that copyright notice and
   * this permission notice appear in supporting documentation, and that
   * the name of M.I.T. not be used in advertising or publicity pertaining
   * to distribution of the software without specific, written prior
   * permission.  Furthermore if you modify this software you must label
   * your software as modified software and not distribute it in such a
   * fashion that it might be confused with the original M.I.T. software.
   * M.I.T. makes no representations about the suitability of
   * this software for any purpose.  It is provided "as is" without express
   * or implied warranty.
   */
  
  #include "k5-int.h"
  #include "k5-hex.h"
  #include "common.h"
  #include "mglueP.h"
  #include "gssapiP_krb5.h"
  
  static const char inputstr[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
      "abcdefghijklmnopqrstuvwxyz123456789";
  
  /* For each test, out1 corresponds to key1 with an empty input, and out2
   * corresponds to key2 with the above 61-byte input string. */
  static struct {
      krb5_enctype enctype;
      const char *key1;
      const char *out1;
      const char *key2;
      const char *out2;
  } tests[] = {
      { ENCTYPE_ARCFOUR_HMAC,
        "3BB3AE288C12B3B9D06B208A4151B3B6",
        "9AEA11A3BCF3C53F1F91F5A0BA2132E2501ADF5F3C28"
        "3C8A983AB88757CE865A22132D6100EAD63E9E291AFA",
        "6DB7B33A01BD2B72F7655CB7B3D5FA0B",
        "CDA9A544869FC84873B692663A82AFDA101C8611498B"
        "A46138B01E927C9B95EEC953B562807434037837DDDF" },
      { ENCTYPE_AES128_CTS_HMAC_SHA1_96,
        "6C742096EB896230312B73972FA28B5D",
        "94208D982FC1BB7778128BDD77904420B45C9DA699F3"
        "117BCE66E39602128EF0296611A6D191A5828530F20F",
        "FA61138C109D834A477D24C7311BE6DA",
        "0FAEDF0F842CC834FEE750487E1B622739286B975FE5"
        "B7F45AB053143C75CA0DF5D3D4BBB80F6A616C7C9027" },
      { ENCTYPE_AES256_CTS_HMAC_SHA1_96,
        "08FCDAFD5832611B73BA7B497FEBFF8C954B4B58031CAD9B977C3B8C25192FD6",
        "E627EFC14EF5B6D629F830C7109DEA0D3D7D36E8CD57"
        "A1F301C5452494A1928F05AFFBEE3360232209D3BE0D",
        "F5B68B7823D8944F33F41541B4E4D38C9B2934F8D16334A796645B066152B4BE",
        "112F2B2D878590653CCC7DE278E9F0AA46FA5A380B62"
        "59F774CB7C134FCD37F61A50FD0D9F89BF8FE1A6B593" },
      { ENCTYPE_CAMELLIA128_CTS_CMAC,
        "866E0466A178279A32AC0BDA92B72AEB",
        "97FBB354BF341C3A160DCC86A7A910FDA824601DF677"
        "68797BACEEBF5D250AE929DEC9760772084267F50A54",
        "D4893FD37DA1A211E12DD1E03E0F03B7",
        "1DEE2FF126CA563A2A2326B9DD3F0095013257414C83"
        "FAD4398901013D55F367C82681186B7B2FE62F746BA4" },
      { ENCTYPE_CAMELLIA256_CTS_CMAC,
        "203071B1AE77BD3D6FCE70174AF95C225B1CED46B35CF52B6479EFEB47E6B063",
        "9B30020634C10FDA28420CEE7B96B70A90A771CED43A"
        "D8346554163E5949CBAE2FB8EF36AFB6B32CE75116A0",
        "A171AD582C1AFBBAD52ABD622EE6B6A14D19BF95C6914B2BA40FFD99A88EC660",
        "A47CBB6E104DCC77E4DB48A7A474B977F2FB6A7A1AB6"
        "52317D50508AE72B7BE2E4E4BA24164E029CBACF786B" },
      { ENCTYPE_AES128_CTS_HMAC_SHA256_128,
        "089BCA48B105EA6EA77CA5D2F39DC5E7",
        "ED1736209B7C59C9F6A3AE8CCC8A7C97ADFDD11688AD"
        "F304F2F74252CBACD311A2D9253211FDA49745CE4F62",
        "3705D96080C17728A0E800EAB6E0D23C",
        "2BB41B183D76D8D5B30CBB049A7EFE9F350EFA058DC2"
        "C4D868308D354A7B199BE6FD1F22B53C038BC6036581" },
      { ENCTYPE_AES256_CTS_HMAC_SHA384_192,
        "45BD806DBF6A833A9CFFC1C94589A222367A79BC21C413718906E9F578A78467",
        "1C613AE8B77A3B4D783F3DCE6C9178FC025E87F48A44"
        "784A69CB5FC697FE266A6141905067EF78566D309085",
        "6D404D37FAF79F9DF0D33568D320669800EB4836472EA8A026D16B7182460C52",
        "D15944B0A44508D1E61213F6455F292A02298F870C01"
        "A3F74AD0345A4A6651EBE101976E933F32D44F0B5947" },
  };
  
  /* Decode hexstr into out.  No length checking. */
  static size_t
  fromhex(const char *hexstr, unsigned char *out)
  {
      uint8_t *bytes;
      size_t len;
  
      if (k5_hex_decode(hexstr, &bytes, &len) != 0)
          abort();
      memcpy(out, bytes, len);
      free(bytes);
      return len;
  }
  
  int
  main(int argc, char *argv[])
  {
      OM_uint32 minor, major;
      gss_ctx_id_t context;
      gss_union_ctx_id_desc uctx;
      krb5_gss_ctx_id_rec kgctx;
      krb5_key k1, k2;
      krb5_keyblock kb1, kb2;
      gss_buffer_desc in, out;
      unsigned char k1buf[32], k2buf[32], outbuf[44];
      size_t i;
  
      /*
       * Fake up just enough of a krb5 GSS context to make gss_pseudo_random
       * work, with chosen subkeys and acceptor subkeys.  If we implement
       * gss_import_lucid_sec_context, we can rewrite this to use public
       * interfaces and stop using private headers and internal knowledge of the
       * implementation.
       */
      context = (gss_ctx_id_t)&uctx;
      memset(&uctx, 0, sizeof(uctx));
      uctx.mech_type = &mech_krb5;
      uctx.internal_ctx_id = (gss_ctx_id_t)&kgctx;
      memset(&kgctx, 0, sizeof(kgctx));
      kgctx.k5_context = NULL;
      kgctx.established = 1;
      kgctx.have_acceptor_subkey = 1;
      kb1.contents = k1buf;
      kb2.contents = k2buf;
      for (i = 0; i < sizeof(tests) / sizeof(*tests); i++) {
          /* Set up the keys for this test. */
          kb1.enctype = tests[i].enctype;
          kb1.length = fromhex(tests[i].key1, k1buf);
          check_k5err(NULL, "create_key", krb5_k_create_key(NULL, &kb1, &k1));
          kgctx.subkey = k1;
          kb2.enctype = tests[i].enctype;
          kb2.length = fromhex(tests[i].key2, k2buf);
          check_k5err(NULL, "create_key", krb5_k_create_key(NULL, &kb2, &k2));
          kgctx.acceptor_subkey = k2;
  
          /* Generate a PRF value with the subkey and an empty input, and compare
           * it to the first expected output. */
          in.length = 0;
          in.value = NULL;
          major = gss_pseudo_random(&minor, context, GSS_C_PRF_KEY_PARTIAL, &in,
                                    44, &out);
          check_gsserr("gss_pseudo_random", major, minor);
          (void)fromhex(tests[i].out1, outbuf);
          assert(out.length == 44 && memcmp(out.value, outbuf, 44) == 0);
          (void)gss_release_buffer(&minor, &out);
  
          /* Generate a PRF value with the acceptor subkey and the 61-byte input
           * string, and compare it to the second expected output. */
          in.length = strlen(inputstr);
          in.value = (char *)inputstr;
          major = gss_pseudo_random(&minor, context, GSS_C_PRF_KEY_FULL, &in, 44,
                                    &out);
          check_gsserr("gss_pseudo_random", major, minor);
          (void)fromhex(tests[i].out2, outbuf);
          assert(out.length == 44 && memcmp(out.value, outbuf, 44) == 0);
          (void)gss_release_buffer(&minor, &out);
  
          /* Also check that generating zero bytes of output works. */
          major = gss_pseudo_random(&minor, context, GSS_C_PRF_KEY_FULL, &in, 0,
                                    &out);
          check_gsserr("gss_pseudo_random", major, minor);
          assert(out.length == 0);
          (void)gss_release_buffer(&minor, &out);
  
          krb5_k_free_key(NULL, k1);
          krb5_k_free_key(NULL, k2);
      }
      return 0;
  }
  