155682Smarkm/*
2233294Sstas * Copyright (c) 1997 - 2003 Kungliga Tekniska H��gskolan
3233294Sstas * (Royal Institute of Technology, Stockholm, Sweden).
4233294Sstas * All rights reserved.
555682Smarkm *
6233294Sstas * Redistribution and use in source and binary forms, with or without
7233294Sstas * modification, are permitted provided that the following conditions
8233294Sstas * are met:
955682Smarkm *
10233294Sstas * 1. Redistributions of source code must retain the above copyright
11233294Sstas *    notice, this list of conditions and the following disclaimer.
1255682Smarkm *
13233294Sstas * 2. Redistributions in binary form must reproduce the above copyright
14233294Sstas *    notice, this list of conditions and the following disclaimer in the
15233294Sstas *    documentation and/or other materials provided with the distribution.
1655682Smarkm *
17233294Sstas * 3. Neither the name of the Institute nor the names of its contributors
18233294Sstas *    may be used to endorse or promote products derived from this software
19233294Sstas *    without specific prior written permission.
2055682Smarkm *
21233294Sstas * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22233294Sstas * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23233294Sstas * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24233294Sstas * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25233294Sstas * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26233294Sstas * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27233294Sstas * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28233294Sstas * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29233294Sstas * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30233294Sstas * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31233294Sstas * SUCH DAMAGE.
3255682Smarkm */
3355682Smarkm
34233294Sstas#include "krb5_locl.h"
3555682Smarkm
36233294SstasKRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
3755682Smarkmkrb5_mk_safe(krb5_context context,
3855682Smarkm	     krb5_auth_context auth_context,
3955682Smarkm	     const krb5_data *userdata,
4055682Smarkm	     krb5_data *outbuf,
41178825Sdfr	     krb5_replay_data *outdata)
4255682Smarkm{
43178825Sdfr    krb5_error_code ret;
44178825Sdfr    KRB_SAFE s;
45178825Sdfr    u_char *buf = NULL;
46178825Sdfr    size_t buf_size;
47233294Sstas    size_t len = 0;
48178825Sdfr    krb5_crypto crypto;
49178825Sdfr    krb5_keyblock *key;
50178825Sdfr    krb5_replay_data rdata;
5155682Smarkm
52233294Sstas    if ((auth_context->flags &
53178825Sdfr	 (KRB5_AUTH_CONTEXT_RET_TIME | KRB5_AUTH_CONTEXT_RET_SEQUENCE)) &&
54178825Sdfr	outdata == NULL)
55178825Sdfr	return KRB5_RC_REQUIRED; /* XXX better error, MIT returns this */
5690926Snectar
57178825Sdfr    if (auth_context->local_subkey)
58178825Sdfr	key = auth_context->local_subkey;
59178825Sdfr    else if (auth_context->remote_subkey)
60178825Sdfr	key = auth_context->remote_subkey;
61178825Sdfr    else
62178825Sdfr	key = auth_context->keyblock;
6355682Smarkm
64178825Sdfr    s.pvno = 5;
65178825Sdfr    s.msg_type = krb_safe;
6655682Smarkm
67178825Sdfr    memset(&rdata, 0, sizeof(rdata));
6855682Smarkm
69178825Sdfr    s.safe_body.user_data = *userdata;
7055682Smarkm
71178825Sdfr    krb5_us_timeofday (context, &rdata.timestamp, &rdata.usec);
7255682Smarkm
73178825Sdfr    if (auth_context->flags & KRB5_AUTH_CONTEXT_DO_TIME) {
74178825Sdfr	s.safe_body.timestamp  = &rdata.timestamp;
75178825Sdfr	s.safe_body.usec       = &rdata.usec;
76178825Sdfr    } else {
77178825Sdfr	s.safe_body.timestamp  = NULL;
78178825Sdfr	s.safe_body.usec       = NULL;
79178825Sdfr    }
80233294Sstas
81178825Sdfr    if (auth_context->flags & KRB5_AUTH_CONTEXT_RET_TIME) {
82178825Sdfr	outdata->timestamp = rdata.timestamp;
83178825Sdfr	outdata->usec = rdata.usec;
84178825Sdfr    }
8555682Smarkm
86178825Sdfr    if (auth_context->flags & KRB5_AUTH_CONTEXT_DO_SEQUENCE) {
87178825Sdfr	rdata.seq = auth_context->local_seqnumber;
88178825Sdfr	s.safe_body.seq_number = &rdata.seq;
89233294Sstas    } else
90178825Sdfr	s.safe_body.seq_number = NULL;
9155682Smarkm
92178825Sdfr    if (auth_context->flags & KRB5_AUTH_CONTEXT_RET_SEQUENCE)
93178825Sdfr	outdata->seq = auth_context->local_seqnumber;
94233294Sstas
95178825Sdfr    s.safe_body.s_address = auth_context->local_address;
96178825Sdfr    s.safe_body.r_address = auth_context->remote_address;
97178825Sdfr
98178825Sdfr    s.cksum.cksumtype       = 0;
99178825Sdfr    s.cksum.checksum.data   = NULL;
100178825Sdfr    s.cksum.checksum.length = 0;
101178825Sdfr
102178825Sdfr    ASN1_MALLOC_ENCODE(KRB_SAFE, buf, buf_size, &s, &len, ret);
103178825Sdfr    if (ret)
104178825Sdfr	return ret;
105178825Sdfr    if(buf_size != len)
106178825Sdfr	krb5_abortx(context, "internal error in ASN.1 encoder");
107178825Sdfr    ret = krb5_crypto_init(context, key, 0, &crypto);
108178825Sdfr    if (ret) {
109178825Sdfr	free (buf);
110178825Sdfr	return ret;
111178825Sdfr    }
112233294Sstas    ret = krb5_create_checksum(context,
113178825Sdfr			       crypto,
114178825Sdfr			       KRB5_KU_KRB_SAFE_CKSUM,
115178825Sdfr			       0,
116178825Sdfr			       buf,
117178825Sdfr			       len,
118178825Sdfr			       &s.cksum);
119178825Sdfr    krb5_crypto_destroy(context, crypto);
120178825Sdfr    if (ret) {
121178825Sdfr	free (buf);
122178825Sdfr	return ret;
123178825Sdfr    }
124178825Sdfr
125178825Sdfr    free(buf);
126178825Sdfr    ASN1_MALLOC_ENCODE(KRB_SAFE, buf, buf_size, &s, &len, ret);
127178825Sdfr    free_Checksum (&s.cksum);
128178825Sdfr    if(ret)
129178825Sdfr	return ret;
130178825Sdfr    if(buf_size != len)
131178825Sdfr	krb5_abortx(context, "internal error in ASN.1 encoder");
132178825Sdfr
133178825Sdfr    outbuf->length = len;
134178825Sdfr    outbuf->data   = buf;
135178825Sdfr    if (auth_context->flags & KRB5_AUTH_CONTEXT_DO_SEQUENCE)
136178825Sdfr	auth_context->local_seqnumber =
137178825Sdfr	    (auth_context->local_seqnumber + 1) & 0xFFFFFFFF;
138178825Sdfr    return 0;
13955682Smarkm}
140