API improvements needed by freeradius
[trust_router.git] / tid / tidc.c
1 /*
2  * Copyright (c) 2012, 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 <stdio.h>
36 #include <jansson.h>
37
38 #include <trust_router/tr_dh.h>
39 #include <tid_internal.h>
40 #include <tr_msg.h>
41 #include <gsscon.h>
42
43 int tmp_len = 32;
44
45 TIDC_INSTANCE *tidc_create ()
46 {
47   TIDC_INSTANCE *tidc = NULL;
48
49   if (tidc = malloc(sizeof(TIDC_INSTANCE))) 
50     memset(tidc, 0, sizeof(TIDC_INSTANCE));
51   else
52     return NULL;
53
54   return tidc;
55 }
56
57 void tidc_destroy (TIDC_INSTANCE *tidc)
58 {
59   if (tidc)
60     free(tidc);
61 }
62
63 int tidc_open_connection (TIDC_INSTANCE *tidc, 
64                           char *server,
65                           unsigned int port,
66                           gss_ctx_id_t *gssctx)
67 {
68   int err = 0;
69   int conn = -1;
70   unsigned int use_port = 0;
71
72   if (0 == port)
73     use_port = TID_PORT;
74   else 
75     use_port = port;
76
77   fprintf(stderr, "tidc_open_connection: Opening GSS connection to %s:%u.", server, use_port);  
78   err = gsscon_connect(server, use_port, "trustidentity", &conn, gssctx);
79
80   if (!err)
81     return conn;
82   else
83     return -1;
84 }
85
86 int tidc_send_request (TIDC_INSTANCE *tidc, 
87                        int conn, 
88                        gss_ctx_id_t gssctx,
89                        char *rp_realm,
90                        char *realm, 
91                        char *comm,
92                        TIDC_RESP_FUNC *resp_handler,
93                        void *cookie)
94
95 {
96   TID_REQ *tid_req = NULL;
97
98   /* Create and populate a TID req structure */
99   if (!(tid_req = tid_req_new()))
100     return -1;
101
102   tid_req->conn = conn;
103   tid_req->gssctx = gssctx;
104
105   if ((NULL == (tid_req->rp_realm = tr_new_name(rp_realm))) ||
106       (NULL == (tid_req->realm = tr_new_name(realm))) ||
107       (NULL == (tid_req->comm = tr_new_name(comm)))) {
108     fprintf (stderr, "tidc_send_request: Error duplicating names.\n");
109     return -1;
110   }
111
112   tid_req->tidc_dh = tidc->client_dh;
113
114   return (tidc_fwd_request(tidc, tid_req, resp_handler, cookie));
115 }
116
117 int tidc_fwd_request (TIDC_INSTANCE *tidc, 
118                       TID_REQ *tid_req, 
119                       TIDC_RESP_FUNC *resp_handler,
120                       void *cookie)
121
122 {
123   char *req_buf = NULL;
124   char *resp_buf = NULL;
125   size_t resp_buflen = 0;
126   TR_MSG *msg = NULL;
127   TR_MSG *resp_msg = NULL;
128   int err;
129
130   /* Create and populate a TID msg structure */
131   if (!(msg = malloc(sizeof(TR_MSG))))
132     return -1;
133
134   msg->msg_type = TID_REQUEST;
135   tr_msg_set_req(msg, tid_req);
136
137   /* store the response function and cookie */
138   // tid_req->resp_func = resp_handler;
139   // tid_req->cookie = cookie;
140   
141
142   /* Encode the request into a json string */
143   if (!(req_buf = tr_msg_encode(msg))) {
144     fprintf(stderr, "tidc_fwd_request: Error encoding TID request.\n");
145     return -1;
146   }
147
148   fprintf (stderr, "tidc_fwd_request: Sending TID request:\n");
149   fprintf (stderr, "%s\n", req_buf);
150
151   /* Send the request over the connection */
152   if (err = gsscon_write_encrypted_token (tid_req->conn, tid_req->gssctx, req_buf, 
153                                           strlen(req_buf))) {
154     fprintf(stderr, "tidc_fwd_request: Error sending request over connection.\n");
155     return -1;
156   }
157
158   /* TBD -- queue request on instance, read resps in separate thread */
159
160   /* Read the response from the connection */
161   /* TBD -- timeout? */
162   if (err = gsscon_read_encrypted_token(tid_req->conn, tid_req->gssctx, &resp_buf, &resp_buflen)) {
163     if (resp_buf)
164       free(resp_buf);
165     return -1;
166   }
167
168   fprintf(stdout, "tidc_fwd_request: Response Received (%u bytes).\n", (unsigned) resp_buflen);
169   fprintf(stdout, "%s\n", resp_buf);
170
171   if (NULL == (resp_msg = tr_msg_decode(resp_buf, resp_buflen))) {
172     fprintf(stderr, "tidc_fwd_request: Error decoding response.\n");
173     return -1;
174   }
175
176   /* TBD -- Check if this is actually a valid response */
177   if (TID_RESPONSE != tr_msg_get_msg_type(resp_msg)) {
178     fprintf(stderr, "tidc_fwd_request: Error, no response in the response!\n");
179     return -1;
180   }
181   
182   if (resp_handler)
183     /* Call the caller's response function */
184     (*resp_handler)(tidc, tid_req, tr_msg_get_resp(resp_msg), cookie);
185   else
186     fprintf(stderr, "tidc_fwd_request: NULL response function.\n");
187
188   if (msg)
189     free(msg);
190   if (tid_req)
191     tid_req_free(tid_req);
192   if (req_buf)
193     free(req_buf);
194   if (resp_buf)
195     free(resp_buf);
196
197   /* TBD -- free the decoded response */
198
199   return 0;
200 }
201
202
203 DH * tidc_get_dh(TIDC_INSTANCE *inst)
204 {
205   return inst->client_dh;
206 }
207
208 DH *tidc_set_dh(TIDC_INSTANCE *inst, DH *dh)
209 {
210   inst->client_dh = dh;
211   return dh;
212 }