1/*
2 * Copyright (c) 1997 - 2001 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of the Institute nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#include "krb5_locl.h"
35
36KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
37krb5_read_message (krb5_context context,
38		   krb5_pointer p_fd,
39		   krb5_data *data)
40{
41    krb5_error_code ret;
42    krb5_ssize_t sret;
43    uint32_t len;
44    uint8_t buf[4];
45
46    krb5_data_zero(data);
47
48    sret = krb5_net_read (context, p_fd, buf, 4);
49    if(sret == -1) {
50	ret = errno;
51	krb5_clear_error_message (context);
52	return ret;
53    }
54    if(sret < 4) {
55	krb5_clear_error_message(context);
56	return HEIM_ERR_EOF;
57    }
58    len = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
59    if (len > UINT_MAX / 16) {
60	krb5_set_error_message(context, ERANGE, N_("packet to large", ""));
61	return ERANGE;
62    }
63    ret = krb5_data_alloc (data, len);
64    if (ret) {
65	krb5_clear_error_message(context);
66	return ret;
67    }
68    if (krb5_net_read (context, p_fd, data->data, len) != (ssize_t)len) {
69	ret = errno;
70	krb5_data_free (data);
71	krb5_clear_error_message (context);
72	return ret;
73    }
74    return 0;
75}
76
77KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
78krb5_read_priv_message(krb5_context context,
79		       krb5_auth_context ac,
80		       krb5_pointer p_fd,
81		       krb5_data *data)
82{
83    krb5_error_code ret;
84    krb5_data packet;
85
86    ret = krb5_read_message(context, p_fd, &packet);
87    if(ret)
88	return ret;
89    ret = krb5_rd_priv (context, ac, &packet, data, NULL);
90    krb5_data_free(&packet);
91    return ret;
92}
93
94KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
95krb5_read_safe_message(krb5_context context,
96		       krb5_auth_context ac,
97		       krb5_pointer p_fd,
98		       krb5_data *data)
99{
100    krb5_error_code ret;
101    krb5_data packet;
102
103    ret = krb5_read_message(context, p_fd, &packet);
104    if(ret)
105	return ret;
106    ret = krb5_rd_safe (context, ac, &packet, data, NULL);
107    krb5_data_free(&packet);
108    return ret;
109}
110