s3-libsmb: Use data_blob_talloc to get krb5 ticket and session keys
[ira/wip.git] / source3 / libsmb / clispnego.c
index 2cf276485e71824b1b43b294ac88c69d6c63a48d..539b41105698751c88e647cee3f39692872855a9 100644 (file)
@@ -4,7 +4,8 @@
    Copyright (C) Andrew Tridgell 2001
    Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2002
    Copyright (C) Luke Howard     2003
-   
+   Copyright (C) Jeremy Allison 2010
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
 
 /*
   generate a negTokenInit packet given a list of supported
-  OIDs (the mechanisms) and a principal name string 
+  OIDs (the mechanisms) a blob, and a principal name string
 */
-DATA_BLOB spnego_gen_negTokenInit(const char *OIDs[], 
+
+DATA_BLOB spnego_gen_negTokenInit(TALLOC_CTX *ctx,
+                                 const char *OIDs[],
+                                 DATA_BLOB *psecblob,
                                  const char *principal)
 {
        int i;
@@ -52,61 +56,23 @@ DATA_BLOB spnego_gen_negTokenInit(const char *OIDs[],
        asn1_pop_tag(data);
        asn1_pop_tag(data);
 
-       asn1_push_tag(data, ASN1_CONTEXT(3));
-       asn1_push_tag(data, ASN1_SEQUENCE(0));
-       asn1_push_tag(data, ASN1_CONTEXT(0));
-       asn1_write_GeneralString(data,principal);
-       asn1_pop_tag(data);
-       asn1_pop_tag(data);
-       asn1_pop_tag(data);
-
-       asn1_pop_tag(data);
-       asn1_pop_tag(data);
-
-       asn1_pop_tag(data);
-
-       if (data->has_error) {
-               DEBUG(1,("Failed to build negTokenInit at offset %d\n", (int)data->ofs));
+       if (psecblob && psecblob->length && psecblob->data) {
+               asn1_push_tag(data, ASN1_CONTEXT(2));
+               asn1_write_OctetString(data,psecblob->data,
+                       psecblob->length);
+               asn1_pop_tag(data);
        }
 
-       ret = data_blob(data->data, data->length);
-       asn1_free(data);
-
-       return ret;
-}
-
-/*
-  Generate a negTokenInit as used by the client side ... It has a mechType
-  (OID), and a mechToken (a security blob) ... 
-
-  Really, we need to break out the NTLMSSP stuff as well, because it could be
-  raw in the packets!
-*/
-DATA_BLOB gen_negTokenInit(const char *OID, DATA_BLOB blob)
-{
-       ASN1_DATA *data;
-       DATA_BLOB ret;
-
-       data = asn1_init(talloc_tos());
-       if (data == NULL) {
-               return data_blob_null;
+       if (principal) {
+               asn1_push_tag(data, ASN1_CONTEXT(3));
+               asn1_push_tag(data, ASN1_SEQUENCE(0));
+               asn1_push_tag(data, ASN1_CONTEXT(0));
+               asn1_write_GeneralString(data,principal);
+               asn1_pop_tag(data);
+               asn1_pop_tag(data);
+               asn1_pop_tag(data);
        }
 
-       asn1_push_tag(data, ASN1_APPLICATION(0));
-       asn1_write_OID(data,OID_SPNEGO);
-       asn1_push_tag(data, ASN1_CONTEXT(0));
-       asn1_push_tag(data, ASN1_SEQUENCE(0));
-
-       asn1_push_tag(data, ASN1_CONTEXT(0));
-       asn1_push_tag(data, ASN1_SEQUENCE(0));
-       asn1_write_OID(data, OID);
-       asn1_pop_tag(data);
-       asn1_pop_tag(data);
-
-       asn1_push_tag(data, ASN1_CONTEXT(2));
-       asn1_write_OctetString(data,blob.data,blob.length);
-       asn1_pop_tag(data);
-
        asn1_pop_tag(data);
        asn1_pop_tag(data);
 
@@ -116,7 +82,7 @@ DATA_BLOB gen_negTokenInit(const char *OID, DATA_BLOB blob)
                DEBUG(1,("Failed to build negTokenInit at offset %d\n", (int)data->ofs));
        }
 
-       ret = data_blob(data->data, data->length);
+       ret = data_blob_talloc(ctx, data->data, data->length);
        asn1_free(data);
 
        return ret;
@@ -126,7 +92,8 @@ DATA_BLOB gen_negTokenInit(const char *OID, DATA_BLOB blob)
   parse a negTokenInit packet giving a GUID, a list of supported
   OIDs (the mechanisms) and a principal name string 
 */
-bool spnego_parse_negTokenInit(DATA_BLOB blob,
+bool spnego_parse_negTokenInit(TALLOC_CTX *ctx,
+                              DATA_BLOB blob,
                               char *OIDs[ASN1_MAX_OIDS],
                               char **principal,
                               DATA_BLOB *secblob)
@@ -159,7 +126,7 @@ bool spnego_parse_negTokenInit(DATA_BLOB blob,
        asn1_start_tag(data,ASN1_SEQUENCE(0));
        for (i=0; asn1_tag_remaining(data) > 0 && i < ASN1_MAX_OIDS-1; i++) {
                const char *oid_str = NULL;
-               asn1_read_OID(data,talloc_autofree_context(),&oid_str);
+               asn1_read_OID(data,ctx,&oid_str);
                OIDs[i] = CONST_DISCARD(char *, oid_str);
        }
        OIDs[i] = NULL;
@@ -180,50 +147,46 @@ bool spnego_parse_negTokenInit(DATA_BLOB blob,
          the next tag is ASN1_CONTEXT(3).
        */
 
-       if (asn1_tag_remaining(data) > 0) {
-               if (asn1_peek_tag(data, ASN1_CONTEXT(1))) {
-                       uint8 flags;
+       if (asn1_peek_tag(data, ASN1_CONTEXT(1))) {
+               uint8 flags;
 
-                       /* reqFlags [1] ContextFlags  OPTIONAL */
-                       asn1_start_tag(data, ASN1_CONTEXT(1));
-                       asn1_start_tag(data, ASN1_BIT_STRING);
-                       while (asn1_tag_remaining(data) > 0) {
-                               asn1_read_uint8(data, &flags);
-                       }
-                       asn1_end_tag(data);
-                       asn1_end_tag(data);
+               /* reqFlags [1] ContextFlags  OPTIONAL */
+               asn1_start_tag(data, ASN1_CONTEXT(1));
+               asn1_start_tag(data, ASN1_BIT_STRING);
+               while (asn1_tag_remaining(data) > 0) {
+                       asn1_read_uint8(data, &flags);
                }
+               asn1_end_tag(data);
+               asn1_end_tag(data);
        }
 
-       if (asn1_tag_remaining(data) > 0) {
-               if (asn1_peek_tag(data, ASN1_CONTEXT(2))) {
-                       DATA_BLOB sblob = data_blob_null;
-                       /* mechToken [2] OCTET STRING  OPTIONAL */
-                       asn1_start_tag(data, ASN1_CONTEXT(2));
-                       asn1_read_OctetString(data, talloc_autofree_context(),
-                               &sblob);
-                       asn1_end_tag(data);
-                       if (secblob) {
-                               *secblob = sblob;
-                       }
+       if (asn1_peek_tag(data, ASN1_CONTEXT(2))) {
+               DATA_BLOB sblob = data_blob_null;
+               /* mechToken [2] OCTET STRING  OPTIONAL */
+               asn1_start_tag(data, ASN1_CONTEXT(2));
+               asn1_read_OctetString(data, ctx, &sblob);
+               asn1_end_tag(data);
+               if (secblob) {
+                       *secblob = sblob;
+               } else {
+                       data_blob_free(&sblob);
                }
        }
 
-       if (asn1_tag_remaining(data) > 0) {
-               if (asn1_peek_tag(data, ASN1_CONTEXT(3))) {
-                       char *princ = NULL;
-                       /* mechListMIC [3] OCTET STRING  OPTIONAL */
-                       asn1_start_tag(data, ASN1_CONTEXT(3));
-                       asn1_start_tag(data, ASN1_SEQUENCE(0));
-                       asn1_start_tag(data, ASN1_CONTEXT(0));
-                       asn1_read_GeneralString(data,talloc_autofree_context(),
-                               &princ);
-                       asn1_end_tag(data);
-                       asn1_end_tag(data);
-                       asn1_end_tag(data);
-                       if (principal) {
-                               *principal = princ;
-                       }
+       if (asn1_peek_tag(data, ASN1_CONTEXT(3))) {
+               char *princ = NULL;
+               /* mechListMIC [3] OCTET STRING  OPTIONAL */
+               asn1_start_tag(data, ASN1_CONTEXT(3));
+               asn1_start_tag(data, ASN1_SEQUENCE(0));
+               asn1_start_tag(data, ASN1_CONTEXT(0));
+               asn1_read_GeneralString(data, ctx, &princ);
+               asn1_end_tag(data);
+               asn1_end_tag(data);
+               asn1_end_tag(data);
+               if (principal) {
+                       *principal = princ;
+               } else {
+                       TALLOC_FREE(princ);
                }
        }
 
@@ -250,56 +213,10 @@ bool spnego_parse_negTokenInit(DATA_BLOB blob,
        return ret;
 }
 
-/*
-  generate a negTokenTarg packet given a list of OIDs and a security blob
-*/
-DATA_BLOB gen_negTokenTarg(const char *OIDs[], DATA_BLOB blob)
-{
-       int i;
-       ASN1_DATA *data;
-       DATA_BLOB ret;
-
-       data = asn1_init(talloc_tos());
-       if (data == NULL) {
-               return data_blob_null;
-       }
-
-       asn1_push_tag(data, ASN1_APPLICATION(0));
-       asn1_write_OID(data,OID_SPNEGO);
-       asn1_push_tag(data, ASN1_CONTEXT(0));
-       asn1_push_tag(data, ASN1_SEQUENCE(0));
-
-       asn1_push_tag(data, ASN1_CONTEXT(0));
-       asn1_push_tag(data, ASN1_SEQUENCE(0));
-       for (i=0; OIDs[i]; i++) {
-               asn1_write_OID(data,OIDs[i]);
-       }
-       asn1_pop_tag(data);
-       asn1_pop_tag(data);
-
-       asn1_push_tag(data, ASN1_CONTEXT(2));
-       asn1_write_OctetString(data,blob.data,blob.length);
-       asn1_pop_tag(data);
-
-       asn1_pop_tag(data);
-       asn1_pop_tag(data);
-
-       asn1_pop_tag(data);
-
-       if (data->has_error) {
-               DEBUG(1,("Failed to build negTokenTarg at offset %d\n", (int)data->ofs));
-       }
-
-       ret = data_blob(data->data, data->length);
-       asn1_free(data);
-
-       return ret;
-}
-
 /*
   generate a krb5 GSS-API wrapper packet given a ticket
 */
-DATA_BLOB spnego_gen_krb5_wrap(const DATA_BLOB ticket, const uint8 tok_id[2])
+DATA_BLOB spnego_gen_krb5_wrap(TALLOC_CTX *ctx, const DATA_BLOB ticket, const uint8 tok_id[2])
 {
        ASN1_DATA *data;
        DATA_BLOB ret;
@@ -320,7 +237,7 @@ DATA_BLOB spnego_gen_krb5_wrap(const DATA_BLOB ticket, const uint8 tok_id[2])
                DEBUG(1,("Failed to build krb5 wrapper at offset %d\n", (int)data->ofs));
        }
 
-       ret = data_blob(data->data, data->length);
+       ret = data_blob_talloc(ctx, data->data, data->length);
        asn1_free(data);
 
        return ret;
@@ -329,7 +246,7 @@ DATA_BLOB spnego_gen_krb5_wrap(const DATA_BLOB ticket, const uint8 tok_id[2])
 /*
   parse a krb5 GSS-API wrapper packet giving a ticket
 */
-bool spnego_parse_krb5_wrap(DATA_BLOB blob, DATA_BLOB *ticket, uint8 tok_id[2])
+bool spnego_parse_krb5_wrap(TALLOC_CTX *ctx, DATA_BLOB blob, DATA_BLOB *ticket, uint8 tok_id[2])
 {
        bool ret;
        ASN1_DATA *data;
@@ -351,7 +268,7 @@ bool spnego_parse_krb5_wrap(DATA_BLOB blob, DATA_BLOB *ticket, uint8 tok_id[2])
        } else {
                asn1_read(data, tok_id, 2);
                data_remaining -= 2;
-               *ticket = data_blob(NULL, data_remaining);
+               *ticket = data_blob_talloc(ctx, NULL, data_remaining);
                asn1_read(data, ticket->data, ticket->length);
        }
 
@@ -370,11 +287,12 @@ bool spnego_parse_krb5_wrap(DATA_BLOB blob, DATA_BLOB *ticket, uint8 tok_id[2])
 
 
 /* 
-   generate a SPNEGO negTokenTarg packet, ready for a EXTENDED_SECURITY
-   kerberos session setup 
+   generate a SPNEGO krb5 negTokenInit packet, ready for a EXTENDED_SECURITY
+   kerberos session setup
 */
-int spnego_gen_negTokenTarg(const char *principal, int time_offset, 
-                           DATA_BLOB *targ, 
+int spnego_gen_krb5_negTokenInit(TALLOC_CTX *ctx,
+                           const char *principal, int time_offset,
+                           DATA_BLOB *targ,
                            DATA_BLOB *session_key_krb5, uint32 extra_ap_opts,
                            time_t *expire_time)
 {
@@ -383,18 +301,19 @@ int spnego_gen_negTokenTarg(const char *principal, int time_offset,
        const char *krb_mechs[] = {OID_KERBEROS5_OLD, OID_KERBEROS5, OID_NTLMSSP, NULL};
 
        /* get a kerberos ticket for the service and extract the session key */
-       retval = cli_krb5_get_ticket(principal, time_offset,
-                                       &tkt, session_key_krb5, extra_ap_opts, NULL, 
-                                       expire_time, NULL);
-
-       if (retval)
+       retval = cli_krb5_get_ticket(ctx, principal, time_offset,
+                                         &tkt, session_key_krb5,
+                                         extra_ap_opts, NULL,
+                                         expire_time, NULL);
+       if (retval) {
                return retval;
+       }
 
        /* wrap that up in a nice GSS-API wrapping */
-       tkt_wrapped = spnego_gen_krb5_wrap(tkt, TOK_ID_KRB_AP_REQ);
+       tkt_wrapped = spnego_gen_krb5_wrap(ctx, tkt, TOK_ID_KRB_AP_REQ);
 
        /* and wrap that in a shiny SPNEGO wrapper */
-       *targ = gen_negTokenTarg(krb_mechs, tkt_wrapped);
+       *targ = spnego_gen_negTokenInit(ctx, krb_mechs, &tkt_wrapped, NULL);
 
        data_blob_free(&tkt_wrapped);
        data_blob_free(&tkt);
@@ -406,7 +325,7 @@ int spnego_gen_negTokenTarg(const char *principal, int time_offset,
 /*
   parse a spnego NTLMSSP challenge packet giving two security blobs
 */
-bool spnego_parse_challenge(const DATA_BLOB blob,
+bool spnego_parse_challenge(TALLOC_CTX *ctx, const DATA_BLOB blob,
                            DATA_BLOB *chal1, DATA_BLOB *chal2)
 {
        bool ret;
@@ -433,13 +352,13 @@ bool spnego_parse_challenge(const DATA_BLOB blob,
        asn1_end_tag(data);
 
        asn1_start_tag(data,ASN1_CONTEXT(2));
-       asn1_read_OctetString(data, talloc_autofree_context(), chal1);
+       asn1_read_OctetString(data, ctx, chal1);
        asn1_end_tag(data);
 
        /* the second challenge is optional (XP doesn't send it) */
        if (asn1_tag_remaining(data)) {
                asn1_start_tag(data,ASN1_CONTEXT(3));
-               asn1_read_OctetString(data, talloc_autofree_context(), chal2);
+               asn1_read_OctetString(data, ctx, chal2);
                asn1_end_tag(data);
        }
 
@@ -461,7 +380,7 @@ bool spnego_parse_challenge(const DATA_BLOB blob,
 /*
  generate a SPNEGO auth packet. This will contain the encrypted passwords
 */
-DATA_BLOB spnego_gen_auth(DATA_BLOB blob)
+DATA_BLOB spnego_gen_auth(TALLOC_CTX *ctx, DATA_BLOB blob)
 {
        ASN1_DATA *data;
        DATA_BLOB ret;
@@ -479,7 +398,7 @@ DATA_BLOB spnego_gen_auth(DATA_BLOB blob)
        asn1_pop_tag(data);
        asn1_pop_tag(data);
 
-       ret = data_blob(data->data, data->length);
+       ret = data_blob_talloc(ctx, data->data, data->length);
 
        asn1_free(data);
 
@@ -489,7 +408,7 @@ DATA_BLOB spnego_gen_auth(DATA_BLOB blob)
 /*
  parse a SPNEGO auth packet. This contains the encrypted passwords
 */
-bool spnego_parse_auth(DATA_BLOB blob, DATA_BLOB *auth)
+bool spnego_parse_auth(TALLOC_CTX *ctx, DATA_BLOB blob, DATA_BLOB *auth)
 {
        ssize_t len;
        struct spnego_data token;
@@ -507,7 +426,7 @@ bool spnego_parse_auth(DATA_BLOB blob, DATA_BLOB *auth)
                return false;
        }
 
-       *auth = data_blob_talloc(talloc_tos(),
+       *auth = data_blob_talloc(ctx,
                                 token.negTokenTarg.responseToken.data,
                                 token.negTokenTarg.responseToken.length);
        spnego_free_data(&token);
@@ -518,7 +437,8 @@ bool spnego_parse_auth(DATA_BLOB blob, DATA_BLOB *auth)
 /*
   generate a minimal SPNEGO response packet.  Doesn't contain much.
 */
-DATA_BLOB spnego_gen_auth_response(DATA_BLOB *reply, NTSTATUS nt_status,
+DATA_BLOB spnego_gen_auth_response(TALLOC_CTX *ctx,
+                                  DATA_BLOB *reply, NTSTATUS nt_status,
                                   const char *mechOID)
 {
        ASN1_DATA *data;
@@ -559,7 +479,7 @@ DATA_BLOB spnego_gen_auth_response(DATA_BLOB *reply, NTSTATUS nt_status,
        asn1_pop_tag(data);
        asn1_pop_tag(data);
 
-       ret = data_blob(data->data, data->length);
+       ret = data_blob_talloc(ctx, data->data, data->length);
        asn1_free(data);
        return ret;
 }
@@ -567,7 +487,8 @@ DATA_BLOB spnego_gen_auth_response(DATA_BLOB *reply, NTSTATUS nt_status,
 /*
  parse a SPNEGO auth packet. This contains the encrypted passwords
 */
-bool spnego_parse_auth_response(DATA_BLOB blob, NTSTATUS nt_status,
+bool spnego_parse_auth_response(TALLOC_CTX *ctx,
+                               DATA_BLOB blob, NTSTATUS nt_status,
                                const char *mechOID,
                                DATA_BLOB *auth)
 {
@@ -603,7 +524,7 @@ bool spnego_parse_auth_response(DATA_BLOB blob, NTSTATUS nt_status,
 
                if (asn1_tag_remaining(data)) {
                        asn1_start_tag(data,ASN1_CONTEXT(2));
-                       asn1_read_OctetString(data, talloc_autofree_context(), auth);
+                       asn1_read_OctetString(data, ctx, auth);
                        asn1_end_tag(data);
                }
        } else if (negResult == SPNEGO_ACCEPT_INCOMPLETE) {
@@ -617,7 +538,7 @@ bool spnego_parse_auth_response(DATA_BLOB blob, NTSTATUS nt_status,
        if (asn1_tag_remaining(data)) {
                DATA_BLOB mechList = data_blob_null;
                asn1_start_tag(data, ASN1_CONTEXT(3));
-               asn1_read_OctetString(data, talloc_autofree_context(), &mechList);
+               asn1_read_OctetString(data, ctx, &mechList);
                asn1_end_tag(data);
                data_blob_free(&mechList);
                DEBUG(5,("spnego_parse_auth_response received mechListMIC, "