source: mainline/uspace/lib/c/generic/dnsr.c@ 401840b

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 401840b was 401840b, checked in by Jakub Jermar <jakub@…>, 9 years ago

Fix memory leak in dnsr_name2host and missing call to async_forget

  • Property mode set to 100644
File size: 4.6 KB
RevLine 
[31e9fe0]1/*
2 * Copyright (c) 2013 Jiri Svoboda
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 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * - The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#include <async.h>
30#include <assert.h>
31#include <errno.h>
[fff7ef4]32#include <fibril_synch.h>
[31e9fe0]33#include <inet/dnsr.h>
34#include <ipc/dnsr.h>
35#include <ipc/services.h>
36#include <loc.h>
37#include <stdlib.h>
38#include <str.h>
39
[fff7ef4]40static FIBRIL_MUTEX_INITIALIZE(dnsr_sess_mutex);
41
[31e9fe0]42static async_sess_t *dnsr_sess = NULL;
43
[fff7ef4]44static async_exch_t *dnsr_exchange_begin(void)
[31e9fe0]45{
[fff7ef4]46 fibril_mutex_lock(&dnsr_sess_mutex);
[3e66428]47
[fff7ef4]48 if (dnsr_sess == NULL) {
[3e66428]49 service_id_t dnsr_svc;
50
[fff7ef4]51 (void) loc_service_get_id(SERVICE_NAME_DNSR, &dnsr_svc,
52 IPC_FLAG_BLOCKING);
[3e66428]53
[f9b2cb4c]54 dnsr_sess = loc_service_connect(dnsr_svc, INTERFACE_DNSR,
[fff7ef4]55 IPC_FLAG_BLOCKING);
56 }
[3e66428]57
58 async_sess_t *sess = dnsr_sess;
[fff7ef4]59 fibril_mutex_unlock(&dnsr_sess_mutex);
[3e66428]60
[fff7ef4]61 return async_exchange_begin(sess);
62}
63
64static void dnsr_exchange_end(async_exch_t *exch)
65{
66 async_exchange_end(exch);
[31e9fe0]67}
68
[e948fde]69int dnsr_name2host(const char *name, dnsr_hostinfo_t **rinfo, ip_ver_t ver)
[31e9fe0]70{
[fff7ef4]71 async_exch_t *exch = dnsr_exchange_begin();
[02a09ed]72
[31e9fe0]73 ipc_call_t answer;
[e948fde]74 aid_t req = async_send_1(exch, DNSR_NAME2HOST, (sysarg_t) ver,
[0aa70f4]75 &answer);
[02a09ed]76
77 int rc = async_data_write_start(exch, name, str_size(name));
78 if (rc != EOK) {
79 async_exchange_end(exch);
80 async_forget(req);
81 return rc;
82 }
[401840b]83
84 dnsr_hostinfo_t *info = calloc(1, sizeof(dnsr_hostinfo_t));
85 if (info == NULL)
86 return ENOMEM;
[02a09ed]87
88 ipc_call_t answer_addr;
89 aid_t req_addr = async_data_read(exch, &info->addr,
90 sizeof(inet_addr_t), &answer_addr);
91
92 sysarg_t retval_addr;
93 async_wait_for(req_addr, &retval_addr);
94
95 if (retval_addr != EOK) {
96 async_exchange_end(exch);
97 async_forget(req);
[401840b]98 free(info);
[02a09ed]99 return (int) retval_addr;
100 }
101
102 ipc_call_t answer_cname;
103 char cname_buf[DNSR_NAME_MAX_SIZE + 1];
104 aid_t req_cname = async_data_read(exch, cname_buf, DNSR_NAME_MAX_SIZE,
105 &answer_cname);
106
[fff7ef4]107 dnsr_exchange_end(exch);
[02a09ed]108
109 sysarg_t retval_cname;
110 async_wait_for(req_cname, &retval_cname);
111
112 if (retval_cname != EOK) {
[31e9fe0]113 async_forget(req);
[401840b]114 free(info);
[02a09ed]115 return (int) retval_cname;
[31e9fe0]116 }
[02a09ed]117
118 sysarg_t retval;
[31e9fe0]119 async_wait_for(req, &retval);
[02a09ed]120
[401840b]121 if (retval != EOK) {
122 async_forget(req);
123 free(info);
[02a09ed]124 return (int) retval;
[401840b]125 }
126
[02a09ed]127 size_t act_size = IPC_GET_ARG2(answer_cname);
[959d2ec]128 assert(act_size <= DNSR_NAME_MAX_SIZE);
[02a09ed]129
[959d2ec]130 cname_buf[act_size] = '\0';
[401840b]131
[959d2ec]132 info->cname = str_dup(cname_buf);
[401840b]133
[31e9fe0]134 *rinfo = info;
135 return EOK;
136}
137
138void dnsr_hostinfo_destroy(dnsr_hostinfo_t *info)
139{
[c55cbbf]140 if (info == NULL)
141 return;
[3e66428]142
[959d2ec]143 free(info->cname);
[31e9fe0]144 free(info);
145}
146
[a2e3ee6]147int dnsr_get_srvaddr(inet_addr_t *srvaddr)
[31e9fe0]148{
[fff7ef4]149 async_exch_t *exch = dnsr_exchange_begin();
[3e66428]150
151 ipc_call_t answer;
152 aid_t req = async_send_0(exch, DNSR_GET_SRVADDR, &answer);
[a2e3ee6]153 int rc = async_data_read_start(exch, srvaddr, sizeof(inet_addr_t));
[3e66428]154
155 loc_exchange_end(exch);
156
157 if (rc != EOK) {
158 async_forget(req);
[31e9fe0]159 return rc;
[3e66428]160 }
161
162 sysarg_t retval;
163 async_wait_for(req, &retval);
164
165 return (int) retval;
[31e9fe0]166}
167
[a2e3ee6]168int dnsr_set_srvaddr(inet_addr_t *srvaddr)
[31e9fe0]169{
[fff7ef4]170 async_exch_t *exch = dnsr_exchange_begin();
[3e66428]171
172 ipc_call_t answer;
173 aid_t req = async_send_0(exch, DNSR_SET_SRVADDR, &answer);
[a2e3ee6]174 int rc = async_data_write_start(exch, srvaddr, sizeof(inet_addr_t));
[3e66428]175
176 loc_exchange_end(exch);
177
178 if (rc != EOK) {
179 async_forget(req);
[31e9fe0]180 return rc;
[3e66428]181 }
182
183 sysarg_t retval;
184 async_wait_for(req, &retval);
185
186 return (int) retval;
[31e9fe0]187}
188
189/** @}
190 */
Note: See TracBrowser for help on using the repository browser.