Merge pull request #86 from painless-security/jennifer/aaa_server_port
[trust_router.git] / common / tr_gss_client.c
1 /*
2  * Copyright (c) 2012, 2014-2018, JANET(UK)
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * 3. Neither the name of JANET(UK) nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
25  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
29  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
31  * OF THE POSSIBILITY OF SUCH DAMAGE.
32  *
33  */
34
35 #include <talloc.h>
36
37 #include <trust_router/tr_dh.h>
38 #include <tr_msg.h>
39 #include <gsscon.h>
40 #include <tr_debug.h>
41 #include <tr_gss_client.h>
42
43 TR_GSSC_INSTANCE *tr_gssc_instance_new(TALLOC_CTX *mem_ctx)
44 {
45   TR_GSSC_INSTANCE *gssc=talloc(NULL, TR_GSSC_INSTANCE);
46   if (gssc != NULL) {
47     gssc->service_name = NULL;
48     gssc->conn = -1;
49     gssc->gss_ctx = talloc(gssc, gss_ctx_id_t);
50     if (gssc->gss_ctx == NULL) {
51       talloc_free(gssc);
52       return NULL;
53     }
54   }
55   return gssc;
56 }
57
58 void tr_gssc_instance_free(TR_GSSC_INSTANCE *tr_gssc)
59 {
60   talloc_free(tr_gssc);
61 }
62
63 /**
64  * Open a connection to the requested server:port
65  *
66  * @param gssc client instance
67  * @param server server name/address
68  * @param port TCP port to connect
69  * @return 0 on success, -1 on failure
70  */
71 int tr_gssc_open_connection(TR_GSSC_INSTANCE *gssc, const char *server, int port)
72 {
73   if ((port <= 0) || (port > 65535)) {
74     tr_err("tr_gssc_open_connection: invalid port requested (%d)", port);
75     return -1;
76   }
77
78   tr_debug("tr_gssc_open_connection: opening connection to %s:%d", server, port);
79   if (0 != gsscon_connect(server, (unsigned int) port, gssc->service_name, &(gssc->conn), gssc->gss_ctx))
80     return -1;
81
82   return 0; /* success */
83 }
84
85 /**
86  * Send a request message and retrieve a response message
87  *
88  * @param mem_ctx
89  * @param gssc
90  * @param req_msg
91  * @return decoded message, or null on error
92  */
93 TR_MSG *tr_gssc_exchange_msgs(TALLOC_CTX *mem_ctx, TR_GSSC_INSTANCE *gssc, TR_MSG *req_msg)
94 {
95   TALLOC_CTX *tmp_ctx = talloc_new(NULL);
96   char *req_buf = NULL;
97   char *resp_buf = NULL;
98   size_t resp_buflen = 0;
99   TR_MSG *resp_msg = NULL; /* this is the return value */
100   int err;
101
102   /* Validate inputs */
103   if ((gssc == NULL) || (req_msg == NULL))
104     goto cleanup;
105
106   /* Encode the request into a json string */
107   if (!(req_buf = tr_msg_encode(tmp_ctx, req_msg))) {
108     tr_err("tr_gssc_exchange_msgs: Error encoding request message.\n");
109     goto cleanup;
110   }
111
112   tr_debug( "tr_gssc_exchange_msgs: Sending request message:\n%s\n", req_buf);
113
114   /* Send the request over the connection */
115   err = gsscon_write_encrypted_token(gssc->conn, *(gssc->gss_ctx), req_buf, strlen(req_buf));
116   if (err) {
117     tr_err( "tr_gssc_exchange_msgs: Error sending request.\n");
118     goto cleanup;
119   }
120
121   /* Read the response from the connection */
122   /* TBD -- timeout? */
123   if (gsscon_read_encrypted_token(gssc->conn, *(gssc->gss_ctx), &resp_buf, &resp_buflen))
124     goto cleanup;
125
126   tr_debug( "tr_gssc_exchange_msgs: Response Received (%u bytes).\n%s\n", (unsigned) resp_buflen, resp_buf);
127   resp_msg = tr_msg_decode(mem_ctx, resp_buf, resp_buflen);
128   free(resp_buf);
129
130   if (resp_msg == NULL) {
131     tr_err( "tr_gssc_exchange_msgs: Error decoding response.\n");
132     goto cleanup;
133   }
134
135   /* If we get here, then we decoded the message and resp_msg is not null. Nothing more to do. */
136
137 cleanup:
138   talloc_free(tmp_ctx);
139   return resp_msg;
140 }