add rpc_context field to track if a UDP connection was requested
[deb_libnfs.git] / lib / init.c
... / ...
CommitLineData
1/*
2 Copyright (C) 2010 by Ronnie Sahlberg <ronniesahlberg@gmail.com>
3
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU Lesser General Public License as published by
6 the Free Software Foundation; either version 2.1 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU Lesser General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public License
15 along with this program; if not, see <http://www.gnu.org/licenses/>.
16*/
17
18#define _GNU_SOURCE
19#include <stdio.h>
20#include <stdarg.h>
21#include <unistd.h>
22#include <string.h>
23#include <strings.h>
24#include <stdlib.h>
25#include <rpc/rpc.h>
26#include <rpc/xdr.h>
27#include "slist.h"
28#include "libnfs.h"
29#include "libnfs-raw.h"
30#include "libnfs-private.h"
31
32struct rpc_context *rpc_init_context(void)
33{
34 struct rpc_context *rpc;
35
36 rpc = malloc(sizeof(struct rpc_context));
37 if (rpc == NULL) {
38 return NULL;
39 }
40 bzero(rpc, sizeof(struct rpc_context));
41
42 rpc->encodebuflen = 65536;
43 rpc->encodebuf = malloc(rpc->encodebuflen);
44 if (rpc->encodebuf == NULL) {
45 free(rpc);
46 return NULL;
47 }
48
49 rpc->auth = authunix_create_default();
50 if (rpc->auth == NULL) {
51 free(rpc->encodebuf);
52 free(rpc);
53 return NULL;
54 }
55 rpc->xid = 1;
56 rpc->fd = -1;
57
58 return rpc;
59}
60
61
62void rpc_set_auth(struct rpc_context *rpc, struct AUTH *auth)
63{
64 if (rpc->auth != NULL) {
65 auth_destroy(rpc->auth);
66 }
67 rpc->auth = auth;
68}
69
70
71void rpc_set_error(struct rpc_context *rpc, char *error_string, ...)
72{
73 va_list ap;
74 char *str;
75
76 if (rpc->error_string != NULL) {
77 free(rpc->error_string);
78 }
79 va_start(ap, error_string);
80 vasprintf(&str, error_string, ap);
81 rpc->error_string = str;
82 va_end(ap);
83}
84
85char *rpc_get_error(struct rpc_context *rpc)
86{
87 return rpc->error_string;
88}
89
90void rpc_error_all_pdus(struct rpc_context *rpc, char *error)
91{
92 struct rpc_pdu *pdu;
93
94 while((pdu = rpc->outqueue) != NULL) {
95 pdu->cb(rpc, RPC_STATUS_ERROR, error, pdu->private_data);
96 SLIST_REMOVE(&rpc->outqueue, pdu);
97 rpc_free_pdu(rpc, pdu);
98 }
99 while((pdu = rpc->waitpdu) != NULL) {
100 pdu->cb(rpc, RPC_STATUS_ERROR, error, pdu->private_data);
101 SLIST_REMOVE(&rpc->waitpdu, pdu);
102 rpc_free_pdu(rpc, pdu);
103 }
104}
105
106
107void rpc_destroy_context(struct rpc_context *rpc)
108{
109 struct rpc_pdu *pdu;
110
111 while((pdu = rpc->outqueue) != NULL) {
112 pdu->cb(rpc, RPC_STATUS_CANCEL, NULL, pdu->private_data);
113 SLIST_REMOVE(&rpc->outqueue, pdu);
114 rpc_free_pdu(rpc, pdu);
115 }
116 while((pdu = rpc->waitpdu) != NULL) {
117 pdu->cb(rpc, RPC_STATUS_CANCEL, NULL, pdu->private_data);
118 SLIST_REMOVE(&rpc->waitpdu, pdu);
119 rpc_free_pdu(rpc, pdu);
120 }
121
122 auth_destroy(rpc->auth);
123 rpc->auth =NULL;
124
125 if (rpc->fd != -1) {
126 close(rpc->fd);
127 }
128
129 if (rpc->encodebuf != NULL) {
130 free(rpc->encodebuf);
131 rpc->encodebuf = NULL;
132 }
133
134 if (rpc->error_string != NULL) {
135 free(rpc->error_string);
136 rpc->error_string = NULL;
137 }
138
139 free(rpc);
140}
141
142