1/*
2 * Copyright (c) 1997 - 2000 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 "test_locl.h"
35#include <gssapi/gssapi.h>
36#include <gssapi/gssapi_krb5.h>
37#include <gssapi/gssapi_spnego.h>
38#include "gss_common.h"
39RCSID("$Id$");
40
41static int
42do_trans (int sock, gss_ctx_id_t context_hdl)
43{
44    OM_uint32 maj_stat, min_stat;
45    gss_buffer_desc real_input_token, real_output_token;
46    gss_buffer_t input_token = &real_input_token,
47	output_token = &real_output_token;
48
49    /* get_mic */
50
51    input_token->length = 3;
52    input_token->value  = strdup("hej");
53
54    maj_stat = gss_get_mic(&min_stat,
55			   context_hdl,
56			   GSS_C_QOP_DEFAULT,
57			   input_token,
58			   output_token);
59    if (GSS_ERROR(maj_stat))
60	gss_err (1, min_stat, "gss_get_mic");
61
62    write_token (sock, input_token);
63    write_token (sock, output_token);
64
65    /* wrap */
66
67    input_token->length = 7;
68    input_token->value  = "hemligt";
69
70    maj_stat = gss_wrap (&min_stat,
71			 context_hdl,
72			 0,
73			 GSS_C_QOP_DEFAULT,
74			 input_token,
75			 NULL,
76			 output_token);
77    if (GSS_ERROR(maj_stat))
78	gss_err (1, min_stat, "gss_wrap");
79
80    write_token (sock, output_token);
81
82    maj_stat = gss_wrap (&min_stat,
83			 context_hdl,
84			 1,
85			 GSS_C_QOP_DEFAULT,
86			 input_token,
87			 NULL,
88			 output_token);
89    if (GSS_ERROR(maj_stat))
90	gss_err (1, min_stat, "gss_wrap");
91
92    write_token (sock, output_token);
93
94    return 0;
95}
96
97extern char *password;
98
99static int
100proto (int sock, const char *hostname, const char *service)
101{
102    struct sockaddr_storage remote, local;
103    socklen_t addrlen;
104
105    int context_established = 0;
106    gss_ctx_id_t context_hdl = GSS_C_NO_CONTEXT;
107    gss_cred_id_t cred = GSS_C_NO_CREDENTIAL;
108    gss_buffer_desc real_input_token, real_output_token;
109    gss_buffer_t input_token = &real_input_token,
110	output_token = &real_output_token;
111    OM_uint32 maj_stat, min_stat;
112    gss_name_t server;
113    gss_buffer_desc name_token;
114    u_char init_buf[4];
115    u_char acct_buf[4];
116    gss_OID mech_oid;
117    char *str;
118
119    mech_oid = select_mech(mech);
120
121    name_token.length = asprintf (&str,
122				  "%s@%s", service, hostname);
123    if (str == NULL)
124	errx(1, "malloc - out of memory");
125    name_token.value = str;
126
127    maj_stat = gss_import_name (&min_stat,
128				&name_token,
129				GSS_C_NT_HOSTBASED_SERVICE,
130				&server);
131    if (GSS_ERROR(maj_stat))
132	gss_err (1, min_stat,
133		 "Error importing name `%s@%s':\n", service, hostname);
134
135    if (password) {
136        gss_buffer_desc pw;
137
138        pw.value = password;
139        pw.length = strlen(password);
140
141        maj_stat = gss_acquire_cred_with_password(&min_stat,
142						  GSS_C_NO_NAME,
143						  &pw,
144						  GSS_C_INDEFINITE,
145						  GSS_C_NO_OID_SET,
146						  GSS_C_INITIATE,
147						  &cred,
148						  NULL,
149						  NULL);
150        if (GSS_ERROR(maj_stat))
151            gss_err (1, min_stat,
152                     "Error acquiring default initiator credentials");
153    }
154
155    addrlen = sizeof(local);
156    if (getsockname (sock, (struct sockaddr *)&local, &addrlen) < 0
157	|| addrlen > sizeof(local))
158	err (1, "getsockname(%s)", hostname);
159
160    addrlen = sizeof(remote);
161    if (getpeername (sock, (struct sockaddr *)&remote, &addrlen) < 0
162	|| addrlen > sizeof(remote))
163	err (1, "getpeername(%s)", hostname);
164
165    input_token->length = 0;
166    output_token->length = 0;
167
168#if 0
169    struct gss_channel_bindings_struct input_chan_bindings;
170
171    input_chan_bindings.initiator_addrtype = GSS_C_AF_INET;
172    input_chan_bindings.initiator_address.length = 4;
173    init_buf[0] = (local.sin_addr.s_addr >> 24) & 0xFF;
174    init_buf[1] = (local.sin_addr.s_addr >> 16) & 0xFF;
175    init_buf[2] = (local.sin_addr.s_addr >>  8) & 0xFF;
176    init_buf[3] = (local.sin_addr.s_addr >>  0) & 0xFF;
177    input_chan_bindings.initiator_address.value = init_buf;
178
179    input_chan_bindings.acceptor_addrtype = GSS_C_AF_INET;
180    input_chan_bindings.acceptor_address.length = 4;
181    acct_buf[0] = (remote.sin_addr.s_addr >> 24) & 0xFF;
182    acct_buf[1] = (remote.sin_addr.s_addr >> 16) & 0xFF;
183    acct_buf[2] = (remote.sin_addr.s_addr >>  8) & 0xFF;
184    acct_buf[3] = (remote.sin_addr.s_addr >>  0) & 0xFF;
185    input_chan_bindings.acceptor_address.value = acct_buf;
186
187    input_chan_bindings.application_data.value = emalloc(4);
188    * (unsigned short*)input_chan_bindings.application_data.value = local.sin_port;
189    * ((unsigned short *)input_chan_bindings.application_data.value + 1) = remote.sin_port;
190    input_chan_bindings.application_data.length = 4;
191
192    input_chan_bindings.application_data.length = 0;
193    input_chan_bindings.application_data.value = NULL;
194#endif
195
196    while(!context_established) {
197	maj_stat =
198	    gss_init_sec_context(&min_stat,
199				 cred,
200				 &context_hdl,
201				 server,
202				 mech_oid,
203				 GSS_C_MUTUAL_FLAG | GSS_C_SEQUENCE_FLAG,
204				 0,
205				 NULL,
206				 input_token,
207				 NULL,
208				 output_token,
209				 NULL,
210				 NULL);
211	if (GSS_ERROR(maj_stat))
212	    gss_err (1, min_stat, "gss_init_sec_context");
213	if (output_token->length != 0)
214	    write_token (sock, output_token);
215	if (GSS_ERROR(maj_stat)) {
216	    if (context_hdl != GSS_C_NO_CONTEXT)
217		gss_delete_sec_context (&min_stat,
218					&context_hdl,
219					GSS_C_NO_BUFFER);
220	    break;
221	}
222	if (maj_stat & GSS_S_CONTINUE_NEEDED) {
223	    read_token (sock, input_token);
224	} else {
225	    context_established = 1;
226	}
227
228    }
229    if (fork_flag) {
230	pid_t pid;
231	int pipefd[2];
232
233	if (pipe (pipefd) < 0)
234	    err (1, "pipe");
235
236	pid = fork ();
237	if (pid < 0)
238	    err (1, "fork");
239	if (pid != 0) {
240	    gss_buffer_desc buf;
241
242	    maj_stat = gss_export_sec_context (&min_stat,
243					       &context_hdl,
244					       &buf);
245	    if (GSS_ERROR(maj_stat))
246		gss_err (1, min_stat, "gss_export_sec_context");
247	    write_token (pipefd[1], &buf);
248	    exit (0);
249	} else {
250	    gss_ctx_id_t context_hdl;
251	    gss_buffer_desc buf;
252
253	    close (pipefd[1]);
254	    read_token (pipefd[0], &buf);
255	    close (pipefd[0]);
256	    maj_stat = gss_import_sec_context (&min_stat, &buf, &context_hdl);
257	    if (GSS_ERROR(maj_stat))
258		gss_err (1, min_stat, "gss_import_sec_context");
259	    gss_release_buffer (&min_stat, &buf);
260	    return do_trans (sock, context_hdl);
261	}
262    } else {
263	return do_trans (sock, context_hdl);
264    }
265}
266
267int
268main(int argc, char **argv)
269{
270    krb5_context context; /* XXX */
271    int port = client_setup(&context, &argc, argv);
272    return client_doit (argv[argc], port, service, proto);
273}
274