1*aa693e99SJason King /*
2*aa693e99SJason King * Copyright 2016 Jakub Klama <jceel@FreeBSD.org>
3*aa693e99SJason King * All rights reserved
4*aa693e99SJason King *
5*aa693e99SJason King * Redistribution and use in source and binary forms, with or without
6*aa693e99SJason King * modification, are permitted providing that the following conditions
7*aa693e99SJason King * are met:
8*aa693e99SJason King * 1. Redistributions of source code must retain the above copyright
9*aa693e99SJason King * notice, this list of conditions and the following disclaimer.
10*aa693e99SJason King * 2. Redistributions in binary form must reproduce the above copyright
11*aa693e99SJason King * notice, this list of conditions and the following disclaimer in the
12*aa693e99SJason King * documentation and/or other materials provided with the distribution.
13*aa693e99SJason King *
14*aa693e99SJason King * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15*aa693e99SJason King * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16*aa693e99SJason King * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*aa693e99SJason King * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
18*aa693e99SJason King * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*aa693e99SJason King * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*aa693e99SJason King * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*aa693e99SJason King * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
22*aa693e99SJason King * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
23*aa693e99SJason King * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24*aa693e99SJason King * POSSIBILITY OF SUCH DAMAGE.
25*aa693e99SJason King *
26*aa693e99SJason King */
27*aa693e99SJason King
28*aa693e99SJason King #include <stdlib.h>
29*aa693e99SJason King #include <string.h>
30*aa693e99SJason King #include <errno.h>
31*aa693e99SJason King #include <assert.h>
32*aa693e99SJason King #include <sys/queue.h>
33*aa693e99SJason King #include "lib9p.h"
34*aa693e99SJason King #include "lib9p_impl.h"
35*aa693e99SJason King #include "fid.h"
36*aa693e99SJason King #include "hashtable.h"
37*aa693e99SJason King #include "log.h"
38*aa693e99SJason King #include "threadpool.h"
39*aa693e99SJason King #include "backend/backend.h"
40*aa693e99SJason King
41*aa693e99SJason King int
l9p_server_init(struct l9p_server ** serverp,struct l9p_backend * backend)42*aa693e99SJason King l9p_server_init(struct l9p_server **serverp, struct l9p_backend *backend)
43*aa693e99SJason King {
44*aa693e99SJason King struct l9p_server *server;
45*aa693e99SJason King
46*aa693e99SJason King server = l9p_calloc(1, sizeof (*server));
47*aa693e99SJason King server->ls_max_version = L9P_2000L;
48*aa693e99SJason King server->ls_backend = backend;
49*aa693e99SJason King LIST_INIT(&server->ls_conns);
50*aa693e99SJason King
51*aa693e99SJason King *serverp = server;
52*aa693e99SJason King return (0);
53*aa693e99SJason King }
54*aa693e99SJason King
55*aa693e99SJason King int
l9p_connection_init(struct l9p_server * server,struct l9p_connection ** conn)56*aa693e99SJason King l9p_connection_init(struct l9p_server *server, struct l9p_connection **conn)
57*aa693e99SJason King {
58*aa693e99SJason King struct l9p_connection *newconn;
59*aa693e99SJason King
60*aa693e99SJason King assert(server != NULL);
61*aa693e99SJason King assert(conn != NULL);
62*aa693e99SJason King
63*aa693e99SJason King newconn = calloc(1, sizeof (*newconn));
64*aa693e99SJason King if (newconn == NULL)
65*aa693e99SJason King return (-1);
66*aa693e99SJason King newconn->lc_server = server;
67*aa693e99SJason King newconn->lc_msize = L9P_DEFAULT_MSIZE;
68*aa693e99SJason King if (l9p_threadpool_init(&newconn->lc_tp, L9P_NUMTHREADS)) {
69*aa693e99SJason King free(newconn);
70*aa693e99SJason King return (-1);
71*aa693e99SJason King }
72*aa693e99SJason King ht_init(&newconn->lc_files, 100);
73*aa693e99SJason King ht_init(&newconn->lc_requests, 100);
74*aa693e99SJason King LIST_INSERT_HEAD(&server->ls_conns, newconn, lc_link);
75*aa693e99SJason King *conn = newconn;
76*aa693e99SJason King
77*aa693e99SJason King return (0);
78*aa693e99SJason King }
79*aa693e99SJason King
80*aa693e99SJason King void
l9p_connection_free(struct l9p_connection * conn)81*aa693e99SJason King l9p_connection_free(struct l9p_connection *conn)
82*aa693e99SJason King {
83*aa693e99SJason King
84*aa693e99SJason King LIST_REMOVE(conn, lc_link);
85*aa693e99SJason King free(conn);
86*aa693e99SJason King }
87*aa693e99SJason King
88*aa693e99SJason King void
l9p_connection_recv(struct l9p_connection * conn,const struct iovec * iov,const size_t niov,void * aux)89*aa693e99SJason King l9p_connection_recv(struct l9p_connection *conn, const struct iovec *iov,
90*aa693e99SJason King const size_t niov, void *aux)
91*aa693e99SJason King {
92*aa693e99SJason King struct l9p_request *req;
93*aa693e99SJason King int error;
94*aa693e99SJason King
95*aa693e99SJason King req = l9p_calloc(1, sizeof (struct l9p_request));
96*aa693e99SJason King req->lr_aux = aux;
97*aa693e99SJason King req->lr_conn = conn;
98*aa693e99SJason King
99*aa693e99SJason King req->lr_req_msg.lm_mode = L9P_UNPACK;
100*aa693e99SJason King req->lr_req_msg.lm_niov = niov;
101*aa693e99SJason King memcpy(req->lr_req_msg.lm_iov, iov, sizeof (struct iovec) * niov);
102*aa693e99SJason King
103*aa693e99SJason King req->lr_resp_msg.lm_mode = L9P_PACK;
104*aa693e99SJason King
105*aa693e99SJason King if (l9p_pufcall(&req->lr_req_msg, &req->lr_req, conn->lc_version) != 0) {
106*aa693e99SJason King L9P_LOG(L9P_WARNING, "cannot unpack received message");
107*aa693e99SJason King l9p_freefcall(&req->lr_req);
108*aa693e99SJason King free(req);
109*aa693e99SJason King return;
110*aa693e99SJason King }
111*aa693e99SJason King
112*aa693e99SJason King if (ht_add(&conn->lc_requests, req->lr_req.hdr.tag, req)) {
113*aa693e99SJason King L9P_LOG(L9P_WARNING, "client reusing outstanding tag %d",
114*aa693e99SJason King req->lr_req.hdr.tag);
115*aa693e99SJason King l9p_freefcall(&req->lr_req);
116*aa693e99SJason King free(req);
117*aa693e99SJason King return;
118*aa693e99SJason King }
119*aa693e99SJason King
120*aa693e99SJason King error = conn->lc_lt.lt_get_response_buffer(req,
121*aa693e99SJason King req->lr_resp_msg.lm_iov,
122*aa693e99SJason King &req->lr_resp_msg.lm_niov,
123*aa693e99SJason King conn->lc_lt.lt_aux);
124*aa693e99SJason King if (error) {
125*aa693e99SJason King L9P_LOG(L9P_WARNING, "cannot obtain buffers for response");
126*aa693e99SJason King ht_remove(&conn->lc_requests, req->lr_req.hdr.tag);
127*aa693e99SJason King l9p_freefcall(&req->lr_req);
128*aa693e99SJason King free(req);
129*aa693e99SJason King return;
130*aa693e99SJason King }
131*aa693e99SJason King
132*aa693e99SJason King /*
133*aa693e99SJason King * NB: it's up to l9p_threadpool_run to decide whether
134*aa693e99SJason King * to queue the work or to run it immediately and wait
135*aa693e99SJason King * (it must do the latter for Tflush requests).
136*aa693e99SJason King */
137*aa693e99SJason King l9p_threadpool_run(&conn->lc_tp, req);
138*aa693e99SJason King }
139*aa693e99SJason King
140*aa693e99SJason King void
l9p_connection_close(struct l9p_connection * conn)141*aa693e99SJason King l9p_connection_close(struct l9p_connection *conn)
142*aa693e99SJason King {
143*aa693e99SJason King struct ht_iter iter;
144*aa693e99SJason King struct l9p_fid *fid;
145*aa693e99SJason King struct l9p_request *req;
146*aa693e99SJason King
147*aa693e99SJason King L9P_LOG(L9P_DEBUG, "waiting for thread pool to shut down");
148*aa693e99SJason King l9p_threadpool_shutdown(&conn->lc_tp);
149*aa693e99SJason King
150*aa693e99SJason King /* Drain pending requests (if any) */
151*aa693e99SJason King L9P_LOG(L9P_DEBUG, "draining pending requests");
152*aa693e99SJason King ht_iter(&conn->lc_requests, &iter);
153*aa693e99SJason King while ((req = ht_next(&iter)) != NULL) {
154*aa693e99SJason King #ifdef notyet
155*aa693e99SJason King /* XXX would be good to know if there is anyone listening */
156*aa693e99SJason King if (anyone listening) {
157*aa693e99SJason King /* XXX crude - ops like Tclunk should succeed */
158*aa693e99SJason King req->lr_error = EINTR;
159*aa693e99SJason King l9p_respond(req, false, false);
160*aa693e99SJason King } else
161*aa693e99SJason King #endif
162*aa693e99SJason King l9p_respond(req, true, false); /* use no-answer path */
163*aa693e99SJason King ht_remove_at_iter(&iter);
164*aa693e99SJason King }
165*aa693e99SJason King
166*aa693e99SJason King /* Close opened files (if any) */
167*aa693e99SJason King L9P_LOG(L9P_DEBUG, "closing opened files");
168*aa693e99SJason King ht_iter(&conn->lc_files, &iter);
169*aa693e99SJason King while ((fid = ht_next(&iter)) != NULL) {
170*aa693e99SJason King conn->lc_server->ls_backend->freefid(
171*aa693e99SJason King conn->lc_server->ls_backend->softc, fid);
172*aa693e99SJason King free(fid);
173*aa693e99SJason King ht_remove_at_iter(&iter);
174*aa693e99SJason King }
175*aa693e99SJason King
176*aa693e99SJason King ht_destroy(&conn->lc_requests);
177*aa693e99SJason King ht_destroy(&conn->lc_files);
178*aa693e99SJason King }
179*aa693e99SJason King
180*aa693e99SJason King struct l9p_fid *
l9p_connection_alloc_fid(struct l9p_connection * conn,uint32_t fid)181*aa693e99SJason King l9p_connection_alloc_fid(struct l9p_connection *conn, uint32_t fid)
182*aa693e99SJason King {
183*aa693e99SJason King struct l9p_fid *file;
184*aa693e99SJason King
185*aa693e99SJason King file = l9p_calloc(1, sizeof (struct l9p_fid));
186*aa693e99SJason King file->lo_fid = fid;
187*aa693e99SJason King /*
188*aa693e99SJason King * Note that the new fid is not marked valid yet.
189*aa693e99SJason King * The insert here will fail if the fid number is
190*aa693e99SJason King * in use, otherwise we have an invalid fid in the
191*aa693e99SJason King * table (as desired).
192*aa693e99SJason King */
193*aa693e99SJason King
194*aa693e99SJason King if (ht_add(&conn->lc_files, fid, file) != 0) {
195*aa693e99SJason King free(file);
196*aa693e99SJason King return (NULL);
197*aa693e99SJason King }
198*aa693e99SJason King
199*aa693e99SJason King return (file);
200*aa693e99SJason King }
201*aa693e99SJason King
202*aa693e99SJason King void
l9p_connection_remove_fid(struct l9p_connection * conn,struct l9p_fid * fid)203*aa693e99SJason King l9p_connection_remove_fid(struct l9p_connection *conn, struct l9p_fid *fid)
204*aa693e99SJason King {
205*aa693e99SJason King struct l9p_backend *be;
206*aa693e99SJason King
207*aa693e99SJason King /* fid should be marked invalid by this point */
208*aa693e99SJason King assert(!l9p_fid_isvalid(fid));
209*aa693e99SJason King
210*aa693e99SJason King be = conn->lc_server->ls_backend;
211*aa693e99SJason King be->freefid(be->softc, fid);
212*aa693e99SJason King
213*aa693e99SJason King ht_remove(&conn->lc_files, fid->lo_fid);
214*aa693e99SJason King free(fid);
215*aa693e99SJason King }
216