Add configure checks for sys/socket.h
[deb_libnfs.git] / lib / init.c
... / ...
CommitLineData
1/*
2 Copyright (C) 2010 by Ronnie Sahlberg <ronniesahlberg@gmail.com>
3
4
5
6 This program is distributed in the hope that it will be useful,
7 but WITHOUT ANY WARRANTY; without even the implied warranty of
8 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
9 GNU Lesser General Public License for more details.
10
11 You should have received a copy of the GNU Lesser General Public License
12 along with this program; if not, see <http://www.gnu.org/licenses/>.
13*/
14#ifdef HAVE_CONFIG_H
15#include "config.h"
16#endif
17
18#ifdef AROS
19#include "aros_compat.h"
20#endif
21
22#ifdef WIN32
23#include "win32_compat.h"
24#else
25#include <strings.h>
26#endif/*WIN32*/
27#define _GNU_SOURCE
28
29#ifdef HAVE_UNISTD_H
30#include <unistd.h>
31#endif
32
33#include <stdio.h>
34#include <stdarg.h>
35#include <string.h>
36#include <stdlib.h>
37#include <assert.h>
38#include <time.h>
39#include "slist.h"
40#include "libnfs-zdr.h"
41#include "libnfs.h"
42#include "libnfs-raw.h"
43#include "libnfs-private.h"
44
45struct rpc_context *rpc_init_context(void)
46{
47 struct rpc_context *rpc;
48 static uint32_t salt = 0;
49
50 rpc = malloc(sizeof(struct rpc_context));
51 if (rpc == NULL) {
52 return NULL;
53 }
54 memset(rpc, 0, sizeof(struct rpc_context));
55
56 rpc->magic = RPC_CONTEXT_MAGIC;
57 rpc->encodebuflen = 65536;
58 rpc->encodebuf = malloc(rpc->encodebuflen);
59 if (rpc->encodebuf == NULL) {
60 free(rpc);
61 return NULL;
62 }
63
64 rpc->auth = authunix_create_default();
65 if (rpc->auth == NULL) {
66 free(rpc->encodebuf);
67 free(rpc);
68 return NULL;
69 }
70 rpc->xid = salt + time(NULL);
71 salt += 0x01000000;
72 rpc->fd = -1;
73
74 return rpc;
75}
76
77
78struct rpc_context *rpc_init_udp_context(void)
79{
80 struct rpc_context *rpc;
81
82 rpc = rpc_init_context();
83 if (rpc != NULL) {
84 rpc->is_udp = 1;
85 }
86
87 return rpc;
88}
89
90void rpc_set_auth(struct rpc_context *rpc, struct AUTH *auth)
91{
92 assert(rpc->magic == RPC_CONTEXT_MAGIC);
93
94 if (rpc->auth != NULL) {
95 auth_destroy(rpc->auth);
96 }
97 rpc->auth = auth;
98}
99
100
101void rpc_set_error(struct rpc_context *rpc, char *error_string, ...)
102{
103 va_list ap;
104
105 assert(rpc->magic == RPC_CONTEXT_MAGIC);
106
107 if (rpc->error_string != NULL) {
108 free(rpc->error_string);
109 }
110 va_start(ap, error_string);
111 rpc->error_string = malloc(1024);
112 vsnprintf(rpc->error_string, 1024, error_string, ap);
113 va_end(ap);
114}
115
116char *rpc_get_error(struct rpc_context *rpc)
117{
118 assert(rpc->magic == RPC_CONTEXT_MAGIC);
119
120 return rpc->error_string;
121}
122
123void rpc_error_all_pdus(struct rpc_context *rpc, char *error)
124{
125 struct rpc_pdu *pdu;
126
127 assert(rpc->magic == RPC_CONTEXT_MAGIC);
128
129 while((pdu = rpc->outqueue) != NULL) {
130 pdu->cb(rpc, RPC_STATUS_ERROR, error, pdu->private_data);
131 SLIST_REMOVE(&rpc->outqueue, pdu);
132 rpc_free_pdu(rpc, pdu);
133 }
134 while((pdu = rpc->waitpdu) != NULL) {
135 pdu->cb(rpc, RPC_STATUS_ERROR, error, pdu->private_data);
136 SLIST_REMOVE(&rpc->waitpdu, pdu);
137 rpc_free_pdu(rpc, pdu);
138 }
139}
140
141static void rpc_free_fragment(struct rpc_fragment *fragment)
142{
143 if (fragment->data != NULL) {
144 free(fragment->data);
145 }
146 free(fragment);
147}
148
149void rpc_free_all_fragments(struct rpc_context *rpc)
150{
151 assert(rpc->magic == RPC_CONTEXT_MAGIC);
152
153 while (rpc->fragments != NULL) {
154 struct rpc_fragment *fragment = rpc->fragments;
155
156 SLIST_REMOVE(&rpc->fragments, fragment);
157 rpc_free_fragment(fragment);
158 }
159}
160
161int rpc_add_fragment(struct rpc_context *rpc, char *data, uint64_t size)
162{
163 struct rpc_fragment *fragment;
164
165 assert(rpc->magic == RPC_CONTEXT_MAGIC);
166
167 fragment = malloc(sizeof(struct rpc_fragment));
168 if (fragment == NULL) {
169 return -1;
170 }
171
172 fragment->size = size;
173 fragment->data = malloc(fragment->size);
174 if(fragment->data == NULL) {
175 free(fragment);
176 return -1;
177 }
178
179 memcpy(fragment->data, data, fragment->size);
180 SLIST_ADD_END(&rpc->fragments, fragment);
181 return 0;
182}
183
184void rpc_destroy_context(struct rpc_context *rpc)
185{
186 struct rpc_pdu *pdu;
187
188 assert(rpc->magic == RPC_CONTEXT_MAGIC);
189
190 while((pdu = rpc->outqueue) != NULL) {
191 pdu->cb(rpc, RPC_STATUS_CANCEL, NULL, pdu->private_data);
192 SLIST_REMOVE(&rpc->outqueue, pdu);
193 rpc_free_pdu(rpc, pdu);
194 }
195 while((pdu = rpc->waitpdu) != NULL) {
196 pdu->cb(rpc, RPC_STATUS_CANCEL, NULL, pdu->private_data);
197 SLIST_REMOVE(&rpc->waitpdu, pdu);
198 rpc_free_pdu(rpc, pdu);
199 }
200
201 rpc_free_all_fragments(rpc);
202
203 auth_destroy(rpc->auth);
204 rpc->auth =NULL;
205
206 if (rpc->fd != -1) {
207 close(rpc->fd);
208 }
209
210 if (rpc->encodebuf != NULL) {
211 free(rpc->encodebuf);
212 rpc->encodebuf = NULL;
213 }
214
215 if (rpc->error_string != NULL) {
216 free(rpc->error_string);
217 rpc->error_string = NULL;
218 }
219
220 if (rpc->udp_dest != NULL) {
221 free(rpc->udp_dest);
222 rpc->udp_dest = NULL;
223 }
224
225 rpc->magic = 0;
226 free(rpc);
227}
228
229